Something went wrong. Try again.
A privacy-first, self-hosted, fully open source personal knowledge management software, written in typescript and golang. (PERSONAL FORK)
Something went wrong. Try again.
Go
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318// SiYuan - Refactor your thinking// Copyright (c) 2020-present, b3log.org//// This program is free software: you can redistribute it and/or modify// it under the terms of the GNU Affero General Public License as published by// the Free Software Foundation, either version 3 of the License, or// (at your option) any later version.//// This program is distributed in the hope that it will be useful,// but WITHOUT ANY WARRANTY; without even the implied warranty of// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the// GNU Affero General Public License for more details.//// You should have received a copy of the GNU Affero General Public License// along with this program. If not, see <https://www.gnu.org/licenses/>.
package api
import ( "encoding/base32" "encoding/base64" "encoding/hex" "fmt" "io" "math" "net/http" "net/textproto" "net/url" "strings" "time"
"github.com/88250/gulu" "github.com/gin-gonic/gin" "github.com/imroc/req/v3" "github.com/siyuan-note/logging" "github.com/siyuan-note/siyuan/kernel/util")
type File struct { Filename string Header textproto.MIMEHeader Size int64 Content string}
type MultipartForm struct { Value map[string][]string File map[string][]File}
func echo(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret)
var ( password string passwordSet bool multipartForm *MultipartForm rawData any )
password, passwordSet = c.Request.URL.User.Password()
if form, err := c.MultipartForm(); err != nil || nil == form { multipartForm = nil } else { multipartForm = &MultipartForm{ Value: form.Value, File: map[string][]File{}, } for k, handlers := range form.File { files := make([]File, len(handlers)) multipartForm.File[k] = files for i, handler := range handlers { files[i].Filename = handler.Filename files[i].Header = handler.Header files[i].Size = handler.Size if file, err := handler.Open(); err != nil { logging.LogWarnf("echo open form [%s] file [%s] error: %s", k, handler.Filename, err.Error()) } else { content := make([]byte, handler.Size) if _, err := file.Read(content); err != nil { logging.LogWarnf("echo read form [%s] file [%s] error: %s", k, handler.Filename, err.Error()) } else { files[i].Content = base64.StdEncoding.EncodeToString(content) } } } } }
if data, err := c.GetRawData(); err == nil { rawData = base64.StdEncoding.EncodeToString(data) } else { logging.LogWarnf("echo get raw data error: %s", err.Error()) rawData = nil } c.Request.ParseForm() c.Request.ParseMultipartForm(math.MaxInt64)
ret.Data = map[string]interface{}{ "Context": map[string]interface{}{ "Params": c.Params, "HandlerNames": c.HandlerNames(), "FullPath": c.FullPath(), "ClientIP": c.ClientIP(), "RemoteIP": c.RemoteIP(), "ContentType": c.ContentType(), "IsWebsocket": c.IsWebsocket(), "RawData": rawData, }, "Request": map[string]interface{}{ "Method": c.Request.Method, "URL": c.Request.URL, "Proto": c.Request.Proto, "ProtoMajor": c.Request.ProtoMajor, "ProtoMinor": c.Request.ProtoMinor, "Header": c.Request.Header, "ContentLength": c.Request.ContentLength, "TransferEncoding": c.Request.TransferEncoding, "Close": c.Request.Close, "Host": c.Request.Host, "Form": c.Request.Form, "PostForm": c.Request.PostForm, "MultipartForm": multipartForm, "Trailer": c.Request.Trailer, "RemoteAddr": c.Request.RemoteAddr, "TLS": c.Request.TLS, "UserAgent": c.Request.UserAgent(), "Cookies": c.Request.Cookies(), "Referer": c.Request.Referer(), }, "URL": map[string]interface{}{ "EscapedPath": c.Request.URL.EscapedPath(), "EscapedFragment": c.Request.URL.EscapedFragment(), "String": c.Request.URL.String(), "Redacted": c.Request.URL.Redacted(), "IsAbs": c.Request.URL.IsAbs(), "Query": c.Request.URL.Query(), "RequestURI": c.Request.URL.RequestURI(), "Hostname": c.Request.URL.Hostname(), "Port": c.Request.URL.Port(), }, "User": map[string]interface{}{ "Username": c.Request.URL.User.Username(), "Password": password, "PasswordSet": passwordSet, "String": c.Request.URL.User.String(), }, }}
func forwardProxy(c *gin.Context) { ret := gulu.Ret.NewResult() defer c.JSON(http.StatusOK, ret)
arg, ok := util.JsonArg(c, ret) if !ok { return }
destURL := arg["url"].(string) if _, e := url.ParseRequestURI(destURL); nil != e { ret.Code = -1 ret.Msg = "invalid [url]" return }
method := "POST" if methodArg := arg["method"]; nil != methodArg { method = strings.ToUpper(methodArg.(string)) } timeout := 7 * 1000 if timeoutArg := arg["timeout"]; nil != timeoutArg { timeout = int(timeoutArg.(float64)) if 1 > timeout { timeout = 7 * 1000 } }
client := req.C() client.SetTimeout(time.Duration(timeout) * time.Millisecond) request := client.R() headers := arg["headers"].([]interface{}) for _, pair := range headers { for k, v := range pair.(map[string]interface{}) { request.SetHeader(k, fmt.Sprintf("%s", v)) } }
contentType := "application/json" if contentTypeArg := arg["contentType"]; nil != contentTypeArg { contentType = contentTypeArg.(string) } request.SetHeader("Content-Type", contentType)
payloadEncoding := "json" if payloadEncodingArg := arg["payloadEncoding"]; nil != payloadEncodingArg { payloadEncoding = payloadEncodingArg.(string) }
switch payloadEncoding { case "base64": fallthrough case "base64-std": if payload, err := base64.StdEncoding.DecodeString(arg["payload"].(string)); err != nil { ret.Code = -2 ret.Msg = "decode base64-std payload failed: " + err.Error() return } else { request.SetBody(payload) } case "base64-url": if payload, err := base64.URLEncoding.DecodeString(arg["payload"].(string)); err != nil { ret.Code = -2 ret.Msg = "decode base64-url payload failed: " + err.Error() return } else { request.SetBody(payload) } case "base32": fallthrough case "base32-std": if payload, err := base32.StdEncoding.DecodeString(arg["payload"].(string)); err != nil { ret.Code = -2 ret.Msg = "decode base32-std payload failed: " + err.Error() return } else { request.SetBody(payload) } case "base32-hex": if payload, err := base32.HexEncoding.DecodeString(arg["payload"].(string)); err != nil { ret.Code = -2 ret.Msg = "decode base32-hex payload failed: " + err.Error() return } else { request.SetBody(payload) } case "hex": if payload, err := hex.DecodeString(arg["payload"].(string)); err != nil { ret.Code = -2 ret.Msg = "decode hex payload failed: " + err.Error() return } else { request.SetBody(payload) } case "text": default: request.SetBody(arg["payload"]) }
started := time.Now() resp, err := request.Send(method, destURL) if err != nil { ret.Code = -1 ret.Msg = "forward request failed: " + err.Error() return }
bodyData, err := io.ReadAll(resp.Body) if err != nil { ret.Code = -1 ret.Msg = "read response body failed: " + err.Error() return }
elapsed := time.Now().Sub(started)
responseEncoding := "text" if responseEncodingArg := arg["responseEncoding"]; nil != responseEncodingArg { responseEncoding = responseEncodingArg.(string) }
body := "" switch responseEncoding { case "base64": fallthrough case "base64-std": body = base64.StdEncoding.EncodeToString(bodyData) case "base64-url": body = base64.URLEncoding.EncodeToString(bodyData) case "base32": fallthrough case "base32-std": body = base32.StdEncoding.EncodeToString(bodyData) case "base32-hex": body = base32.HexEncoding.EncodeToString(bodyData) case "hex": body = hex.EncodeToString(bodyData) case "text": fallthrough default: responseEncoding = "text" body = string(bodyData) }
data := map[string]interface{}{ "url": destURL, "status": resp.StatusCode, "contentType": resp.GetHeader("content-type"), "body": body, "bodyEncoding": responseEncoding, "headers": resp.Header, "elapsed": elapsed.Milliseconds(), } ret.Data = data
//shortBody := "" //if 64 > len(body) { // shortBody = body //} else { // shortBody = body[:64] //} // //logging.LogInfof("elapsed [%.1fs], length [%d], request [url=%s, headers=%s, content-type=%s, body=%s], status [%d], body [%s]", // elapsed.Seconds(), len(bodyData), data["url"], headers, contentType, arg["payload"], data["status"], shortBody)}