This commit is contained in:
cdle
2023-06-14 20:50:09 +08:00
parent aa4c3fbaaa
commit 05ffe94894
4 changed files with 55 additions and 75 deletions
+36 -50
View File
@@ -14,8 +14,7 @@ import (
"github.com/goccy/go-json" "github.com/goccy/go-json"
) )
func fetch(vm *goja.Runtime, resolve func(result interface{}), reject func(reason interface{}), wts ...interface{}) interface{} { func fetch(vm *goja.Runtime, wts ...interface{}) (interface{}, error) {
var block = resolve == nil
var method = "get" var method = "get"
var url = "" var url = ""
var req *http.Request var req *http.Request
@@ -93,22 +92,14 @@ func fetch(vm *goja.Runtime, resolve func(result interface{}), reject func(reaso
var err error var err error
transport, err = GetTransport(url, user, password) transport, err = GetTransport(url, user, password)
if err != nil { if err != nil {
info := "proxy config error " + err.Error() return nil, err
console.Error()
if block {
panic(Error(vm, info))
} else {
reject(Error(vm, info))
// return nil
}
return nil
} }
} }
} }
} }
} }
method = strings.ToUpper(method) method = strings.ToUpper(method)
var err error
if len(formData) > 0 { if len(formData) > 0 {
data := u.Values{} data := u.Values{}
for key, value := range formData { for key, value := range formData {
@@ -125,10 +116,16 @@ func fetch(vm *goja.Runtime, resolve func(result interface{}), reject func(reaso
} }
} }
req, _ = http.NewRequest(method, url, strings.NewReader(data.Encode())) req, err = http.NewRequest(method, url, strings.NewReader(data.Encode()))
if err != nil {
return nil, err
}
req.Header.Set("Content-Type", "application/x-www-form-urlencoded") req.Header.Set("Content-Type", "application/x-www-form-urlencoded")
} else { } else {
req, _ = http.NewRequest(method, url, bytes.NewBuffer([]byte(body))) req, err = http.NewRequest(method, url, bytes.NewBuffer([]byte(body)))
if err != nil {
return nil, err
}
if isJsonBody { if isJsonBody {
req.Header.Set("Content-Type", "application/json") req.Header.Set("Content-Type", "application/json")
} }
@@ -158,41 +155,30 @@ func fetch(vm *goja.Runtime, resolve func(result interface{}), reject func(reaso
} }
} }
rsp, err = client.Do(req) rsp, err = client.Do(req)
if err == nil { if err != nil {
defer rsp.Body.Close() return nil, err
rspObj = vm.NewProxy(MakeResponseObject(vm, reject, rsp, responseType), &goja.ProxyTrapConfig{
Get: func(target *goja.Object, property string, receiver goja.Value) (value goja.Value) {
obj := target.Get(property)
if obj != nil {
return obj
}
switch property {
case "statusText", "statusMessage":
return vm.ToValue("")
case "statusCode":
return target.Get("status")
// case "body":
// return vm.ToValue(target.Get("getBody").Export().(func() interface{})())
case "then":
return goja.Undefined()
}
console.Error("response has no property " + property)
return goja.Undefined()
},
})
if block {
return rspObj
} else {
resolve(rspObj)
}
} else {
if block {
return rspObj
} else {
reject(Error(vm, err))
}
} }
return nil defer rsp.Body.Close()
obj, err := MakeResponseObject(vm, rsp, responseType)
rspObj = vm.NewProxy(obj, &goja.ProxyTrapConfig{
Get: func(target *goja.Object, property string, receiver goja.Value) (value goja.Value) {
obj := target.Get(property)
if obj != nil {
return obj
}
switch property {
case "statusText", "statusMessage":
return vm.ToValue("")
case "statusCode":
return target.Get("status")
// case "body":
// return vm.ToValue(target.Get("getBody").Export().(func() interface{})())
case "then":
return goja.Undefined()
}
console.Error("response has no property " + property)
return goja.Undefined()
},
})
return rspObj, err
} }
+6 -10
View File
@@ -11,12 +11,11 @@ import (
// type Reason map[string]interface{} // type Reason map[string]interface{}
func MakeResponseObject(vm *goja.Runtime, reject func(reason interface{}), resp *http.Response, responseType string) *goja.Object { func MakeResponseObject(vm *goja.Runtime, resp *http.Response, responseType string) (*goja.Object, error) {
obj := vm.NewObject() obj := vm.NewObject()
data, err := ioutil.ReadAll(resp.Body) data, err := ioutil.ReadAll(resp.Body)
if err != nil { if err != nil { ///////
reject(Error(vm, err)) return obj, err
return obj
} }
var body interface{} var body interface{}
if Contains([]string{"blob", "arraybuffer"}, responseType) { if Contains([]string{"blob", "arraybuffer"}, responseType) {
@@ -26,11 +25,8 @@ func MakeResponseObject(vm *goja.Runtime, reject func(reason interface{}), resp
} else if responseType == "json" { } else if responseType == "json" {
var v interface{} var v interface{}
err := json.Unmarshal(data, &v) err := json.Unmarshal(data, &v)
if err != nil { if err != nil { /////
// console.Error("response body is not json data") return obj, err
// panic("response body is not json data")
// reject()
defer reject(Error(vm, "response body is not json data"))
} else { } else {
body = v body = v
} }
@@ -89,5 +85,5 @@ func MakeResponseObject(vm *goja.Runtime, reject func(reason interface{}), resp
return true return true
}, },
})) }))
return obj return obj, nil
} }
+1 -1
View File
@@ -12,7 +12,7 @@ import (
func GetTransport(proxyUrl string, user, password string) (*http.Transport, error) { func GetTransport(proxyUrl string, user, password string) (*http.Transport, error) {
var auth *proxy.Auth var auth *proxy.Auth
if user != "" && password != "" { if user != "" || password != "" {
auth = &proxy.Auth{User: user, Password: password} auth = &proxy.Auth{User: user, Password: password}
} }
switch { switch {
+12 -14
View File
@@ -365,24 +365,22 @@ func SetPluginMethod(vm *goja.Runtime, uuid string, on_start bool, running func(
}) })
vm.Set("fetch", func(wts ...interface{}) interface{} { vm.Set("fetch", func(wts ...interface{}) interface{} {
promise, resolve, reject := vm.NewPromise() promise, resolve, reject := vm.NewPromise()
func() { go func() {
func() { obj, err := fetch(vm, wts...)
v := recover() if err != nil {
if v != nil { reject(err)
if err, ok := v.(error); ok { } else {
reject(Error(vm, err)) resolve(obj)
} else { }
reject(Error(vm, fmt.Sprint(v)))
}
}
}()
fetch(vm, resolve, reject, wts...)
}() }()
return promise return promise
}) })
vm.Set("request", func(wts ...interface{}) interface{} { vm.Set("request", func(wts ...interface{}) interface{} {
return fetch(vm, nil, nil, wts...) obj, err := fetch(vm, wts...)
if err != nil {
panic(Error(vm, err))
}
return obj
}) })
// for _, method := range []string{"get", "post", "delete", "put", "fetch"} { // for _, method := range []string{"get", "post", "delete", "put", "fetch"} {
// vm.Set(method, ) // vm.Set(method, )