/
request.go
112 lines (90 loc) · 1.96 KB
/
request.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
package client
import (
"bytes"
"context"
"encoding/json"
"fmt"
"io"
"mime/multipart"
"net"
"net/http"
"net/url"
"path/filepath"
)
type RequestHandler struct {
client *http.Client
localAdd string
}
type HTTPPreviewData struct {
Data io.Reader
FileName string
}
type HTTPResponseFormat struct {
Status string
}
const (
SockTypeTCP = iota
SockTypeUNIX
)
func NewRequestHandler(addr string, sockType int) *RequestHandler {
var client *http.Client
switch sockType {
case SockTypeUNIX:
var sock = filepath.Join(addr, "http.sock")
client = &http.Client{
Transport: &http.Transport{
DialContext: func(_ context.Context, _, _ string) (net.Conn, error) {
return net.Dial("unix", sock)
},
},
}
addr = "."
case SockTypeTCP:
client = new(http.Client)
}
return &RequestHandler{client, "http://" + addr}
}
func (r RequestHandler) ChromeShow() (err error) {
_, err = r.client.Get(r.localAdd + "/chromeshow")
return
}
func (r RequestHandler) ChromeOff() (err error) {
_, err = r.client.Get(r.localAdd + "/chromeoff")
return
}
func (r *RequestHandler) Send(data HTTPPreviewData) (err error) {
var body = new(bytes.Buffer)
var mw = multipart.NewWriter(body)
fw, err := mw.CreateFormFile("file", data.FileName)
if err != nil {
return
}
io.Copy(fw, data.Data)
var contentType = mw.FormDataContentType()
mw.Close()
resp, err := r.client.Post(r.localAdd+"/remote", contentType, body)
if err != nil {
return
}
defer resp.Body.Close()
var status HTTPResponseFormat
json.NewDecoder(resp.Body).Decode(&status)
if status.Status != "OK" {
fmt.Println(status.Status)
}
return
}
func (r *RequestHandler) Verify(fileName string) (err error) {
resp, err := r.client.Get(r.localAdd + "/verify?name=" + url.QueryEscape(fileName))
if err != nil {
return
}
if resp.StatusCode != 200 {
return fmt.Errorf("Invalid")
}
return
}
func (r *RequestHandler) Ping() (err error) {
_, err = r.client.Get(r.localAdd + "/ping")
return err
}