feat: sync deployed v1.0.0-rc.38-custom.42

Sync production source including routing, billing, Astra detection and public uptime. Update custom build instructions and license packaging.

controller/manxue_public.go

@@ -0,0 +1,64 @@
+package controller
+
+import (
+	"context"
+	"errors"
+	"net/http"
+	"net/http/cookiejar"
+	"net/url"
+	"strings"
+	"time"
+
+	"github.com/QuantumNous/new-api/common"
+	"github.com/QuantumNous/new-api/model"
+	"github.com/tidwall/gjson"
+)
+
+// Visitor sessions belong to a single detection. Keep them in its private task
+// metadata until completion so polling can resume after a process restart.
+type manxueVisitorSession struct {
+	Cookies []*http.Cookie `json:"manxue_cookies,omitempty"`
+}
+
+func manxueRunRequest(ctx context.Context, run *model.DetectionRun, method, path string, payload any, key string) (gjson.Result, error) {
+	var session manxueVisitorSession
+	_ = common.UnmarshalJsonStr(run.Result, &session)
+	if len(session.Cookies) == 0 && !(method == http.MethodPost && run.RemoteID == "") {
+		return manxueRequest(ctx, method, path, payload, key)
+	}
+	jar, _ := cookiejar.New(nil)
+	origin, _ := url.Parse("https://manxue.ai/api/visitor/")
+	client := &http.Client{Timeout: 35 * time.Second, Jar: jar, CheckRedirect: func(*http.Request, []*http.Request) error { return http.ErrUseLastResponse }}
+	if len(session.Cookies) == 0 {
+		if _, err := manxueRequestWithClient(ctx, client, http.MethodGet, "/api/visitor/config", nil, ""); err != nil {
+			return gjson.Result{}, errDetectionServiceUnavailable
+		}
+		session.Cookies = jar.Cookies(origin)
+		if len(session.Cookies) == 0 {
+			return gjson.Result{}, errDetectionServiceUnavailable
+		}
+		encoded, err := common.Marshal(session)
+		if err != nil {
+			return gjson.Result{}, err
+		}
+		if err = model.DB.Model(&model.DetectionRun{}).Where("id = ?", run.ID).Update("result", string(encoded)).Error; err != nil {
+			return gjson.Result{}, err
+		}
+		run.Result = string(encoded)
+	} else {
+		jar.SetCookies(origin, session.Cookies)
+	}
+	if method == http.MethodPost {
+		fields, ok := payload.(map[string]any)
+		if !ok {
+			return gjson.Result{}, errors.New("invalid pelican payload")
+		}
+		public := make(map[string]any, len(fields)+1)
+		for k, v := range fields {
+			public[k] = v
+		}
+		public["public"] = true
+		payload = public
+	}
+	return manxueRequestWithClient(ctx, client, method, strings.Replace(path, "/api/v1/tests", "/api/visitor/tests", 1), payload, key)
+}