x
This commit is contained in:
cdle
@@ -8,6 +8,6 @@
|
||||
</head>
|
||||
<body>
|
||||
<div id="root"></div>
|
||||
<script src="/admin/umi.d87f33ac.js"></script>
|
||||
<script src="/admin/umi.4dec6745.js"></script>
|
||||
|
||||
</body></html>
|
||||
+3
-3
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
File diff suppressed because one or more lines are too long
+142
-1
@@ -3,6 +3,7 @@ package core
|
||||
import (
|
||||
"bytes"
|
||||
"fmt"
|
||||
"io/ioutil"
|
||||
"net/http"
|
||||
u "net/url"
|
||||
"strconv"
|
||||
@@ -14,7 +15,147 @@ import (
|
||||
"github.com/goccy/go-json"
|
||||
)
|
||||
|
||||
func request(vm *goja.Runtime, resolve func(result interface{}), reject func(reason interface{}), wts ...interface{}) {
|
||||
func request(wt interface{}, handles ...func(error, map[string]interface{}, interface{}) interface{}) interface{} {
|
||||
var method = "get"
|
||||
var url = ""
|
||||
var req *http.Request
|
||||
|
||||
var headers map[string]interface{}
|
||||
// var formData map[string]interface{}
|
||||
var isJson bool
|
||||
var isJsonBody bool
|
||||
var body string
|
||||
var allow_redirects bool = true
|
||||
// var useproxy bool
|
||||
// var timeout time.Duration = 0
|
||||
var goroutine = false
|
||||
var uuid = ""
|
||||
switch wt := wt.(type) {
|
||||
case string:
|
||||
url = wt
|
||||
default:
|
||||
props := wt.(map[string]interface{})
|
||||
for i := range props {
|
||||
switch strings.ToLower(i) {
|
||||
// case "timeout":
|
||||
// timeout = time.Duration(utils.Int64(props[i])) * time.Millisecond
|
||||
case "headers":
|
||||
headers = props[i].(map[string]interface{})
|
||||
case "method":
|
||||
method = strings.ToLower(props[i].(string))
|
||||
case "url":
|
||||
if f, ok := props[i].(func(i goja.FunctionCall) goja.Value); ok {
|
||||
url = f(goja.FunctionCall{}).ToString().String()
|
||||
} else {
|
||||
url = props[i].(string)
|
||||
}
|
||||
case "json":
|
||||
isJson = props[i].(bool)
|
||||
case "uuid":
|
||||
uuid = props[i].(string)
|
||||
case "goroutine":
|
||||
goroutine = true
|
||||
case "datatype":
|
||||
switch props[i].(type) {
|
||||
case string:
|
||||
switch strings.ToLower(props[i].(string)) {
|
||||
case "json":
|
||||
isJson = true
|
||||
}
|
||||
}
|
||||
case "allowredirects":
|
||||
allow_redirects = props[i].(bool)
|
||||
case "body":
|
||||
if v, ok := props[i].(string); !ok {
|
||||
d, _ := json.Marshal(props[i])
|
||||
body = string(d)
|
||||
isJsonBody = true
|
||||
} else {
|
||||
body = v
|
||||
}
|
||||
// case "formdata":
|
||||
// formData = props[i].(map[string]interface{})
|
||||
// case "useproxy":
|
||||
// useproxy = props[i].(bool)
|
||||
}
|
||||
}
|
||||
}
|
||||
method = strings.ToUpper(method)
|
||||
req, _ = http.NewRequest(method, url, bytes.NewBuffer([]byte(body)))
|
||||
// if timeout != 0 {
|
||||
// req.SetTimeout(timeout, timeout)
|
||||
// }
|
||||
|
||||
if isJsonBody {
|
||||
req.Header.Set("Content-Type", "application/json")
|
||||
}
|
||||
for i := range headers {
|
||||
req.Header.Set(i, fmt.Sprint(headers[i]))
|
||||
}
|
||||
// for i := range formData {
|
||||
// req.Param(i, fmt.Sprint(formData[i]))
|
||||
// }
|
||||
|
||||
// if useproxy && Transport != nil {
|
||||
// req.SetTransport(Transport)
|
||||
// }
|
||||
|
||||
rspObj := map[string]interface{}{}
|
||||
var rsp *http.Response
|
||||
var err error
|
||||
// if !allow_redirects {
|
||||
// req.SetCheckRedirect(func(req *http.Request, via []*http.Request) error {
|
||||
// return http.ErrUseLastResponse
|
||||
// })
|
||||
// }
|
||||
var dddd = func() interface{} {
|
||||
client := &http.Client{}
|
||||
if !allow_redirects {
|
||||
client.CheckRedirect = func(req *http.Request, via []*http.Request) error {
|
||||
return http.ErrUseLastResponse
|
||||
}
|
||||
}
|
||||
rsp, err = client.Do(req)
|
||||
var bd interface{}
|
||||
if err == nil {
|
||||
defer rsp.Body.Close()
|
||||
rspObj["status"] = rsp.StatusCode
|
||||
rspObj["statusCode"] = rsp.StatusCode
|
||||
data, _ := ioutil.ReadAll(rsp.Body)
|
||||
if isJson {
|
||||
var v interface{}
|
||||
json.Unmarshal(data, &v)
|
||||
bd = v
|
||||
} else {
|
||||
bd = string(data)
|
||||
}
|
||||
rspObj["body"] = bd
|
||||
h := make(map[string][]string)
|
||||
for k := range rsp.Header {
|
||||
h[k] = rsp.Header[k]
|
||||
}
|
||||
rspObj["headers"] = h
|
||||
} else {
|
||||
rspObj["error"] = err.Error()
|
||||
}
|
||||
if uuid != "" {
|
||||
rspObj["uuid"] = uuid
|
||||
}
|
||||
if len(handles) > 0 {
|
||||
return handles[0](err, rspObj, bd)
|
||||
} else {
|
||||
return rspObj
|
||||
}
|
||||
}
|
||||
if goroutine {
|
||||
go dddd()
|
||||
} else {
|
||||
return dddd()
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func fetch(vm *goja.Runtime, resolve func(result interface{}), reject func(reason interface{}), wts ...interface{}) {
|
||||
var method = "get"
|
||||
var url = ""
|
||||
var req *http.Request
|
||||
|
||||
@@ -359,7 +359,7 @@ func SetPluginMethod(vm *goja.Runtime, uuid string, on_start bool) {
|
||||
vm.Set("Buffer", func(call goja.ConstructorCall) *goja.Object {
|
||||
return Buffer(vm, call)
|
||||
})
|
||||
vm.Set("request", func(wts ...interface{}) interface{} {
|
||||
vm.Set("fetch", func(wts ...interface{}) interface{} {
|
||||
promise, resolve, reject := vm.NewPromise()
|
||||
func() {
|
||||
func() {
|
||||
@@ -373,10 +373,11 @@ func SetPluginMethod(vm *goja.Runtime, uuid string, on_start bool) {
|
||||
|
||||
}
|
||||
}()
|
||||
request(vm, resolve, reject, wts...)
|
||||
fetch(vm, resolve, reject, wts...)
|
||||
}()
|
||||
return promise
|
||||
})
|
||||
vm.Set("request", request)
|
||||
}
|
||||
|
||||
func EncryptPlugin(script string) string {
|
||||
|
||||
-36
@@ -41,8 +41,6 @@ func Cors() gin.HandlerFunc {
|
||||
var Server = gin.New()
|
||||
|
||||
func init() {
|
||||
gin.SetMode(gin.ReleaseMode)
|
||||
// Server.Use(gin.Recovery())
|
||||
Server.Use(Cors())
|
||||
Server.Use(gzip.Gzip(gzip.DefaultCompression))
|
||||
Server.NoRoute(func(c *gin.Context) {
|
||||
@@ -131,9 +129,6 @@ func init() {
|
||||
}
|
||||
|
||||
c.String(404, "页面被喵咪劫走了。") //
|
||||
//开启代理模式
|
||||
|
||||
// handleHTTP(c.Writer, c.Request)
|
||||
})
|
||||
|
||||
port := sillyGirl.GetString("port", "8080")
|
||||
@@ -189,37 +184,6 @@ func init() {
|
||||
}()
|
||||
}
|
||||
|
||||
// var httpProxys = MakeBucket("httpProxys")
|
||||
|
||||
func handleHTTP(w http.ResponseWriter, req *http.Request) {
|
||||
// console.Info("%s", utils.JsonMarshal(req.Header))
|
||||
// u, err := url.Parse("addr")
|
||||
// if err != nil {
|
||||
// core.Logs.Warn("can't connect to the http proxy:", err)
|
||||
// return
|
||||
// }
|
||||
// Transport = &http.Transport{Proxy: http.ProxyURL(u)}
|
||||
// resp, err := Transport.RoundTrip(req)
|
||||
// fmt.Println("====")
|
||||
resp, err := http.DefaultTransport.RoundTrip(req)
|
||||
if err != nil {
|
||||
http.Error(w, err.Error(), http.StatusServiceUnavailable)
|
||||
return
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
copyHeader(w.Header(), resp.Header)
|
||||
w.WriteHeader(resp.StatusCode)
|
||||
io.Copy(w, resp.Body)
|
||||
}
|
||||
|
||||
func copyHeader(dst, src http.Header) {
|
||||
for k, vv := range src {
|
||||
for _, v := range vv {
|
||||
dst.Add(k, v)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
type Req struct {
|
||||
Method string
|
||||
Path string
|
||||
|
||||
Reference in New Issue
Block a user