-
-
Notifications
You must be signed in to change notification settings - Fork 19
/
main.go
113 lines (96 loc) · 2.28 KB
/
main.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
113
package main
import (
"bytes"
"flag"
"fmt"
"io/ioutil"
"log"
"net/http"
)
type handler struct{}
func (*handler) ServeHTTP(w http.ResponseWriter, r *http.Request) {
defer fmt.Println()
fmt.Printf("Proxy Request Over: %s - %s \n", r.Method, r.URL.String())
pfr := addHeaders(w, r)
if pfr { // pre-flight request
return
}
URL := getRequestURL(w, r)
if URL == nil { // invalid URL
return
}
// extract request body
b, err := ioutil.ReadAll(r.Body)
if err != nil {
fmt.Println("Invalid Body:", err)
Return(w, &Error{
Code: http.StatusPreconditionFailed,
Message: err.Error(),
Detail: map[string]interface{}{
"method": r.Method,
"requestedURL": URL.String(),
},
})
return
}
// create proxy request
req, err := http.NewRequest(r.Method, URL.String(), bytes.NewReader(b))
if err != nil {
fmt.Println("Request cannot be created:", err)
Return(w, &Error{
Code: http.StatusPreconditionFailed,
Message: err.Error(),
Detail: map[string]interface{}{
"body": b,
"method": r.Method,
"requestedURL": URL.String(),
},
})
return
}
fmt.Println("UserClient --> bypass-cors -->", req.URL.Host)
// Populate the rest of the header
for k, v := range r.Header {
req.Header.Add(k, v[0])
}
res, err := http.DefaultClient.Do(req)
if err != nil {
fmt.Println("Request Failed:", err)
Return(w, &Error{
Code: http.StatusUnprocessableEntity,
Message: err.Error(),
Detail: map[string]interface{}{
"body": b,
"method": r.Method,
"requestedURL": URL.String(),
"response": res,
},
})
return
}
body, err := ioutil.ReadAll(res.Body)
if err != nil {
fmt.Println("Failed to read:", err)
Return(w, &Error{
Code: res.StatusCode,
Message: err.Error(),
Detail: map[string]interface{}{
"method": r.Method,
"requestedURL": URL.String(),
"body": b,
"response": res,
"responseCode": res.StatusCode,
},
})
return
}
Return(w, &ValuerStruct{res.StatusCode, string(body)})
}
func main() {
// parse all flags set in `init`
flag.Parse()
fmt.Printf("\nStarting Proxy ByPass-Cors Server at port(:%v)...\n\n", PORT)
if err := http.ListenAndServe(":"+PORT, License(&handler{})); err != nil {
log.Println("\n\nPanics", err)
}
}