功能修改

2026-07-20 16:42:33 +08:00
parent 48b8ad8d6c
commit a0e69417db
224 changed files with 22015 additions and 884 deletions
@@ -0,0 +1,51 @@
+package main
+
+import (
+	"context"
+	"encoding/json"
+	"os"
+	"time"
+
+	companion "browser.local/plugins/scum-server-plugin/companion"
+)
+
+func main() {
+	result := companion.SmokeResult{CompletedCommandIDs: []string{}, UnsupportedCommandIDs: []string{}}
+	if len(os.Args) != 1 {
+		writeResult(result)
+		os.Exit(1)
+	}
+	file, err := os.Open("config.yaml")
+	if err != nil {
+		writeResult(result)
+		os.Exit(1)
+	}
+	config, err := companion.LoadConfig(file)
+	closeErr := file.Close()
+	if err != nil || closeErr != nil {
+		writeResult(result)
+		os.Exit(1)
+	}
+	client, err := companion.NewClient(config, companion.Options{})
+	if err != nil {
+		writeResult(result)
+		os.Exit(1)
+	}
+	_ = os.Unsetenv(companion.ProofEnvironment)
+	isolated := os.Getenv(companion.SmokeIsolationEnvironment) == companion.SmokeIsolationValue
+	requestBudget := time.Duration(config.Timing.RequestTimeoutSeconds) * time.Second * time.Duration(2*companion.SmokeClaimLimit+4)
+	if requestBudget > 45*time.Second {
+		requestBudget = 45 * time.Second
+	}
+	ctx, cancel := context.WithTimeout(context.Background(), requestBudget)
+	defer cancel()
+	result, err = companion.RunOneShotSmoke(ctx, client, companion.SmokeOptions{IsolatedNonProduction: isolated})
+	writeResult(result)
+	if err != nil {
+		os.Exit(1)
+	}
+}
+
+func writeResult(result companion.SmokeResult) {
+	_ = json.NewEncoder(os.Stdout).Encode(result)
+}