Add site onboarding config generator

2026-10-17 07:31:32 +02:00 · 2026-07-06 17:55:19 -04:00
parent 0426ca673c
commit db5d9737c0
6 changed files with 557 additions and 10 deletions
@@ -0,0 +1,141 @@
+import contextlib
+import io
+import tempfile
+import unittest
+from pathlib import Path
+
+from ampg.cli import main
+from ampg.config import load_config
+
+
+def _source(root: Path) -> Path:
+    source = root / "site"
+    source.mkdir()
+    (source / "index.html").write_text("<h1>Hello</h1>", encoding="utf-8")
+    return source
+
+
+def _run_cli(args: list[str]) -> tuple[int, str, str]:
+    stdout = io.StringIO()
+    stderr = io.StringIO()
+    with contextlib.redirect_stdout(stdout), contextlib.redirect_stderr(stderr):
+        status = main(args)
+    return status, stdout.getvalue(), stderr.getvalue()
+
+
+class OnboardingTest(unittest.TestCase):
+    def test_init_site_writes_loadable_gateway_config(self):
+        with tempfile.TemporaryDirectory() as tmp:
+            root = Path(tmp).resolve()
+            config_path = root / "gateway.toml"
+
+            status, output, error = _run_cli(
+                [
+                    "--config",
+                    str(config_path),
+                    "init",
+                    "site",
+                    "example",
+                    "--domain",
+                    "example.test",
+                    "--source",
+                    str(_source(root)),
+                    "--preset",
+                    "privacy",
+                ]
+            )
+
+            config = load_config(config_path)
+
+        self.assertEqual(0, status, error)
+        self.assertIn("AMPG_INIT_SITE status=written", output)
+        self.assertIn("AMPG_INIT_PROFILE name=tor-i2p", output)
+        self.assertIn("AMPG_INIT_TOGGLE protocol=clearnet enabled=false", output)
+        self.assertIn("AMPG_INIT_TOGGLE protocol=tor enabled=true", output)
+        self.assertEqual("example", config.sites[0].id)
+        self.assertTrue(config.sites[0].protocols["tor"].enabled)
+        self.assertTrue(config.sites[0].protocols["i2p"].enabled)
+        self.assertFalse(config.sites[0].protocols["clearnet"].enabled)
+        self.assertFalse(config.sites[0].protocols["gemini"].enabled)
+
+    def test_init_site_protocol_override_can_make_i2p_only_config(self):
+        with tempfile.TemporaryDirectory() as tmp:
+            root = Path(tmp).resolve()
+            config_path = root / "gateway.toml"
+
+            status, output, error = _run_cli(
+                [
+                    "--config",
+                    str(config_path),
+                    "init",
+                    "site",
+                    "example",
+                    "--domain",
+                    "example.test",
+                    "--source",
+                    str(_source(root)),
+                    "--protocol",
+                    "i2p",
+                ]
+            )
+
+            config = load_config(config_path)
+
+        self.assertEqual(0, status, error)
+        self.assertIn("protocols=i2p", output)
+        self.assertTrue(config.sites[0].protocols["i2p"].enabled)
+        self.assertFalse(config.sites[0].protocols["tor"].enabled)
+        self.assertIn("mobile-i2p", config.profiles)
+
+    def test_init_site_refuses_overwrite_without_force(self):
+        with tempfile.TemporaryDirectory() as tmp:
+            root = Path(tmp).resolve()
+            config_path = root / "gateway.toml"
+            source = _source(root)
+            args = [
+                "--config",
+                str(config_path),
+                "init",
+                "site",
+                "example",
+                "--domain",
+                "example.test",
+                "--source",
+                str(source),
+            ]
+            self.assertEqual(0, _run_cli(args)[0])
+
+            status, _, error = _run_cli(args)
+
+        self.assertEqual(1, status)
+        self.assertIn("config already exists", error)
+
+    def test_init_site_output_can_build_immediately(self):
+        with tempfile.TemporaryDirectory() as tmp:
+            root = Path(tmp).resolve()
+            config_path = root / "gateway.toml"
+            status, _, error = _run_cli(
+                [
+                    "--config",
+                    str(config_path),
+                    "init",
+                    "site",
+                    "example",
+                    "--domain",
+                    "example.test",
+                    "--source",
+                    str(_source(root)),
+                    "--protocol",
+                    "i2p",
+                ]
+            )
+            self.assertEqual(0, status, error)
+
+            status, output, error = _run_cli(["--config", str(config_path), "build"])
+
+        self.assertEqual(0, status, error)
+        self.assertIn("AMPG_BUILD site=example protocol=i2p", output)
+
+
+if __name__ == "__main__":
+    unittest.main()