PLUS: serveo --subdomain

plus/serve_muscriptor.py

@@ -168,6 +168,16 @@ class Tunnel:
             pass
 
 
+SUBDOMAIN = os.environ.get("MUSCRIPTOR_SUBDOMAIN", "").strip()
+
+
+def forward_spec(port: int, wants_name: bool) -> str:
+    """Строка проброса: с постоянным поддоменом, если он задан."""
+    if wants_name and SUBDOMAIN:
+        return "%s:80:127.0.0.1:%d" % (SUBDOMAIN, port)
+    return "80:127.0.0.1:%d" % port
+
+
 SSH_OPTS = ["-o", "StrictHostKeyChecking=no", "-o", "UserKnownHostsFile=/dev/null",
             "-o", "ServerAliveInterval=15", "-o", "ServerAliveCountMax=3",
             "-o", "TCPKeepAlive=yes", "-o", "ExitOnForwardFailure=yes"]
@@ -229,7 +239,7 @@ def specs(port: int) -> list[tuple[str, list[str], str]]:
          ["ssh", "-T", *SSH_OPTS, "-R", "80:127.0.0.1:%d" % port, SSH_USER + "@localhost.run"],
          r"https://[a-z0-9]+\.lhr\.life"),
         ("serveo",
-         ["ssh", "-T", *SSH_OPTS, "-R", "80:127.0.0.1:%d" % port, "serveo.net"],
+         ["ssh", "-T", *SSH_OPTS, "-R", forward_spec(port, True), "serveo.net"],
          r"https://[a-z0-9-]+(?:\.[a-z0-9-]+)*\."
          r"(?:serveousercontent|serveo)\.[a-z]+"),
         ("pinggy",
@@ -468,6 +478,8 @@ def main(argv: list[str] | None = None) -> int:
                         help="путь к safetensors, имя размера или hf://org/repo/file")
     parser.add_argument("--device", default="auto", help="auto | cpu | cuda | cuda:0")
     parser.add_argument("--dtype", default=None, help="float32 | float16 | bfloat16")
+    parser.add_argument("--subdomain", default="",
+                        help="постоянный поддомен serveo, например nokia1709")
     parser.add_argument("--tunnel", default=None, help="только этот туннель")
     parser.add_argument("--ssh", default=None,
                         help="свой ssh-туннель: пользователь@хост или пользователь@хост:порт")
@@ -491,7 +503,10 @@ def main(argv: list[str] | None = None) -> int:
                         help="инструменты для модели: auto (по умолчанию), band (8 групп)"
                              " или список через запятую")
     args = parser.parse_args(argv)
-    global SSH_USER, CUSTOM_SSH
+    global SSH_USER, CUSTOM_SSH, SUBDOMAIN
+    if args.subdomain:
+        SUBDOMAIN = args.subdomain.strip()
+        os.environ["MUSCRIPTOR_SUBDOMAIN"] = SUBDOMAIN
     if args.ssh_user:
         SSH_USER = args.ssh_user
     if args.ssh: