Upload 12 files

extensions/sd-webui-tunnels/.gitignore

@@ -0,0 +1,176 @@
+# Created by https://www.toptal.com/developers/gitignore/api/python
+# Edit at https://www.toptal.com/developers/gitignore?templates=python
+
+### Python ###
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/#use-with-ide
+.pdm.toml
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/
+
+### Python Patch ###
+# Poetry local configuration file - https://python-poetry.org/docs/configuration/#local-configuration
+poetry.toml
+
+# ruff
+.ruff_cache/
+
+# End of https://www.toptal.com/developers/gitignore/api/python
+
+id_rsa
+id_rsa.pub

extensions/sd-webui-tunnels/.pre-commit-config.yaml

@@ -0,0 +1,25 @@
+repos:
+  - repo: https://github.com/pre-commit/pre-commit-hooks
+    rev: v4.4.0
+    hooks:
+      - id: trailing-whitespace
+        args: [--markdown-linebreak-ext=md]
+      - id: end-of-file-fixer
+
+  - repo: https://github.com/asottile/pyupgrade
+    rev: v3.4.0
+    hooks:
+      - id: pyupgrade
+        args: [--py310-plus]
+
+  - repo: https://github.com/charliermarsh/ruff-pre-commit
+    # Ruff version.
+    rev: "v0.0.265"
+    hooks:
+      - id: ruff
+        args: [--fix, --exit-non-zero-on-fix]
+
+  - repo: https://github.com/psf/black
+    rev: 23.3.0
+    hooks:
+      - id: black

extensions/sd-webui-tunnels/LICENSE.md

@@ -0,0 +1,22 @@
+
+The MIT License (MIT)
+
+Copyright (c) 2023 Bingsu
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

extensions/sd-webui-tunnels/README.md

@@ -0,0 +1,46 @@
+# sd-webui-tunnels
+
+Tunneling extension for [AUTOMATIC1111/stable-diffusion-webui](https://github.com/AUTOMATIC1111/stable-diffusion-webui)
+
+## Usage
+
+### [cloudflared](https://try.cloudflare.com/)
+
+add `--cloudflared` to commandline options.
+
+### [localhost.run](https://localhost.run/)
+
+add `--localhostrun` to commandline options.
+
+### [remote.moe](https://github.com/fasmide/remotemoe)
+
+add `--remotemoe` to commandline options.
+
+The feature of `remote.moe` is that as long as the same ssh key is used, the same url is generated.
+
+The ssh keys for `localhost.run` and `remote.moe` are created with the name `id_rsa` in the script's root folder. However, if there is a problem with the write permission, it is created in a temporary folder instead, so a different url is created each time.
+
+### [jprq](https://github.com/azimjohn/jprq)
+
+add `--jprq "YOUR_JPRQ_TOKEN"` to commandline options.
+
+You can get a token [here](https://jprq.io/auth).
+
+### [bore](https://github.com/ekzhang/bore)
+
+add `--bore` to commandline options.
+
+add `--bore_url "URL"` for custom bore url. url without 'http://' and (optional) port. example: myboreserver.com or myboreserver.com:12345
+
+### googleusercontent
+
+add `--googleusercontent` to commandline options. It must be used with `--no-gradio-queue`, otherwise it will not work.
+
+If not in google colab, it will be ignored.
+
+-----
+### Discord webhook
+
+add `--tunnel-webhook <webhookurl>` to commandline options.
+
+This feature was taken from [nur-zaman/sd-webui-tunnels](https://github.com/nur-zaman/sd-webui-tunnels) fork. thanks.

extensions/sd-webui-tunnels/discord_webhook.py

@@ -0,0 +1,27 @@
+import requests
+
+
+def send_to_discord(message: str, webhook_url: str) -> bool:
+    """
+    Sends a message to a Discord channel using a webhook URL.
+    Args:
+        message (str): The message to send to the Discord channel.
+        webhook_url (str): The Discord webhook URL for the channel to send the message to.
+    Returns:
+        bool: True if the message was successfully sent, False otherwise.
+    """
+    try:
+        # Define the JSON payload to send to the webhook URL.
+        payload = {"content": message}
+
+        # Make a POST request to the webhook URL with the JSON payload.
+        response = requests.post(webhook_url, json=payload)
+
+        # Check the response status code and return True if it was successful.
+        response.raise_for_status()
+        return True
+
+    except Exception as e:
+        # If there was an error sending the message, print the error message and return False.
+        print(f"Error sending message to Discord channel: {e}")
+        return False

extensions/sd-webui-tunnels/install.py

@@ -0,0 +1,7 @@
+import launch
+
+if not launch.is_installed("pycloudflared"):
+    launch.run_pip("install pycloudflared", "pycloudflared")
+
+if not launch.is_installed("tntn"):
+    launch.run_pip("install tntn", "tntn")

extensions/sd-webui-tunnels/preload.py

@@ -0,0 +1,49 @@
+import argparse
+
+
+def preload(parser: argparse.ArgumentParser):
+    parser.add_argument(
+        "--cloudflared",
+        action="store_true",
+        help="use trycloudflare, alternative to gradio --share",
+    )
+
+    parser.add_argument(
+        "--localhostrun",
+        action="store_true",
+        help="use localhost.run, alternative to gradio --share",
+    )
+
+    parser.add_argument(
+        "--remotemoe",
+        action="store_true",
+        help="use remote.moe, alternative to gradio --share",
+    )
+
+    parser.add_argument(
+        "--jprq",
+        type=str,
+        help="use jprq, alternative to gradio --share, requires a token. https://jprq.io/auth",
+    )
+
+    parser.add_argument(
+        "--bore",
+        action="store_true",
+        help="use bore, alternative to gradio --share",
+    )
+
+    parser.add_argument(
+        "--bore_url",
+        type=str,
+        help="custom bore url. url without 'http://' and (optional) port. example: myboreserver.com or myboreserver.com:12345",
+    )
+
+    parser.add_argument(
+        "--googleusercontent",
+        action="store_true",
+        help="use googleusercontent, only available in the google colab. must be used with --no-gradio-queue",
+    )
+
+    parser.add_argument(
+        "--tunnel-webhook", type=str, help="discord webhook to send tunnel url to"
+    )

extensions/sd-webui-tunnels/pyproject.toml

@@ -0,0 +1,25 @@
+[project]
+name = "sd-webui-tunnels"
+version = "23.5.1"
+description = "Tunneling extension for automatic1111 sd-webui"
+authors = [
+    {name = "dowon", email = "ks2515@naver.com"},
+]
+requires-python = ">=3.8"
+readme = "README.md"
+license = {text = "MIT"}
+
+[project.urls]
+repository = "https://github.com/Bing-su/sd-webui-tunnels"
+
+[tool.isort]
+profile = "black"
+known_first_party = ["modules", "launch", "discord_webhook"]
+
+[tool.ruff]
+select = ["A", "B", "C4", "E", "F", "I001", "N", "PT", "UP", "W"]
+ignore = ["B008", "B905", "E501"]
+unfixable = ["F401"]
+
+[tool.ruff.isort]
+known-first-party = ["modules", "launch", "discord_webhook"]

extensions/sd-webui-tunnels/scripts/googleusercontent.py

@@ -0,0 +1,24 @@
+import os
+
+from discord_webhook import send_to_discord
+from modules.shared import cmd_opts
+
+
+is_colab = "COLAB_RELEASE_TAG" in os.environ or "COLAB_BACKEND_VERSION" in os.environ
+
+if is_colab and cmd_opts.googleusercontent:
+    print("googleusercontent detected, trying to connect...")
+
+    if not getattr(cmd_opts, "no_gradio_queue", True):
+        msg = " * If without `--no-gradio-queue` option, it will not work on google colab."
+        print(msg)
+
+    from google.colab.output import eval_js
+
+    port = cmd_opts.port if cmd_opts.port else 7860
+    js = "google.colab.kernel.proxyPort(" + str(port) + ", {'cache': false})"
+    tunnel_url = eval_js(js)
+    print(f" * Running on {tunnel_url}")
+
+    if cmd_opts.tunnel_webhook:
+        send_to_discord(tunnel_url, cmd_opts.tunnel_webhook)

extensions/sd-webui-tunnels/scripts/ssh_tunnel.py

@@ -0,0 +1,103 @@
+from __future__ import annotations
+
+import atexit
+import re
+import subprocess
+from pathlib import Path
+from tempfile import TemporaryDirectory
+
+from discord_webhook import send_to_discord
+from modules.shared import cmd_opts
+
+LOCALHOST_RUN = "localhost.run"
+REMOTE_MOE = "remote.moe"
+localhostrun_pattern = re.compile(r"(?P<url>https?://\S+\.lhr\.life)")
+remotemoe_pattern = re.compile(r"(?P<url>https?://\S+\.remote\.moe)")
+
+
+def gen_key(path: str | Path) -> None:
+    path = Path(path)
+    args = [
+        "ssh-keygen",
+        "-t",
+        "rsa",
+        "-b",
+        "4096",
+        "-q",
+        "-f",
+        path.as_posix(),
+        "-N",
+        "",
+    ]
+    subprocess.run(args, check=True)
+    path.chmod(0o600)
+
+
+def ssh_tunnel(host: str = LOCALHOST_RUN) -> str:
+    ssh_name = "id_rsa"
+    ssh_path = Path(__file__).parent.parent / ssh_name
+
+    tmp = None
+    if not ssh_path.exists():
+        try:
+            gen_key(ssh_path)
+        # write permission error or etc
+        except subprocess.CalledProcessError:
+            tmp = TemporaryDirectory()
+            ssh_path = Path(tmp.name) / ssh_name
+            gen_key(ssh_path)
+
+    port = cmd_opts.port if cmd_opts.port else 7860
+
+    args = [
+        "ssh",
+        "-R",
+        f"80:127.0.0.1:{port}",
+        "-o",
+        "StrictHostKeyChecking=no",
+        "-i",
+        ssh_path.as_posix(),
+        host,
+    ]
+
+    tunnel = subprocess.Popen(
+        args, stdout=subprocess.PIPE, stderr=subprocess.STDOUT, encoding="utf-8"
+    )
+
+    atexit.register(tunnel.terminate)
+    if tmp is not None:
+        atexit.register(tmp.cleanup)
+
+    tunnel_url = ""
+    lines = 27 if host == LOCALHOST_RUN else 5
+    pattern = localhostrun_pattern if host == LOCALHOST_RUN else remotemoe_pattern
+
+    for _ in range(lines):
+        line = tunnel.stdout.readline()
+        if line.startswith("Warning"):
+            print(line, end="")
+
+        url_match = pattern.search(line)
+        if url_match:
+            tunnel_url = url_match.group("url")
+            break
+    else:
+        raise RuntimeError(f"Failed to run {host}")
+
+    print(f" * Running on {tunnel_url}")
+    return tunnel_url
+
+
+if cmd_opts.localhostrun:
+    print("localhost.run detected, trying to connect...")
+    lhr_url = ssh_tunnel(LOCALHOST_RUN)
+
+    if cmd_opts.tunnel_webhook:
+        send_to_discord(lhr_url, cmd_opts.tunnel_webhook)
+
+if cmd_opts.remotemoe:
+    print("remote.moe detected, trying to connect...")
+    moe_url = ssh_tunnel(REMOTE_MOE)
+
+    if cmd_opts.tunnel_webhook:
+        send_to_discord(moe_url, cmd_opts.tunnel_webhook)

extensions/sd-webui-tunnels/scripts/tntn_tunnel.py

@@ -0,0 +1,42 @@
+# Note: tntn is also my package. https://github.com/Bing-su/tntn
+from tntn import bore, jprq
+
+from discord_webhook import send_to_discord
+from modules.shared import cmd_opts
+
+
+def bore_url(s: str):
+    if ":" in s:
+        host, port = s.split(":")
+        port = int(port)
+    else:
+        host = s
+        port = None
+    return host, port
+
+
+port = cmd_opts.port if cmd_opts.port else 7860
+
+if cmd_opts.jprq:
+    # jprq will raise an error unless that port is actually running.
+    import modules.script_callbacks as script_callbacks
+
+    def jprq_callback(*args, **kwargs):
+        jprq_urls = jprq(port, cmd_opts.jprq)
+
+        if cmd_opts.tunnel_webhook:
+            send_to_discord(jprq_urls.tunnel, cmd_opts.tunnel_webhook)
+
+    script_callbacks.on_app_started(jprq_callback)
+
+if cmd_opts.bore:
+    host, bore_port = bore_url(cmd_opts.bore_url) if cmd_opts.bore_url else (None, None)
+    kwargs = {}
+    if host is not None:
+        kwargs["bore_url"] = host
+    if bore_port is not None:
+        kwargs["bore_port"] = bore_port
+    bore_urls = bore(port, **kwargs)
+
+    if cmd_opts.tunnel_webhook:
+        send_to_discord(bore_urls.tunnel, cmd_opts.tunnel_webhook)

extensions/sd-webui-tunnels/scripts/try_cloudflare.py

@@ -0,0 +1,13 @@
+# Note: pycloudflared is also my package. https://github.com/Bing-su/pycloudflared
+from pycloudflared import try_cloudflare
+
+from discord_webhook import send_to_discord
+from modules.shared import cmd_opts
+
+if cmd_opts.cloudflared:
+    print("cloudflared detected, trying to connect...")
+    port = cmd_opts.port if cmd_opts.port else 7860
+
+    urls = try_cloudflare(port)
+    if cmd_opts.tunnel_webhook:
+        send_to_discord(urls.tunnel, cmd_opts.tunnel_webhook)