From 338d12e60cd7f2fe9a15efecac07ce7ae8449d57 Mon Sep 17 00:00:00 2001
From: Jakub Trllo <jakub.trllo@gmail.com>
Date: Thu, 25 Aug 2022 18:03:09 +0200
Subject: [PATCH] added cli command for webserver

---
 openpype/hosts/webpublisher/addon.py          | 19 +++++++++++++++++++
 .../webserver_service/__init__.py             |  6 ++++++
 .../webserver_service/webserver_cli.py        | 16 ++++++++++------
 3 files changed, 35 insertions(+), 6 deletions(-)
 create mode 100644 openpype/hosts/webpublisher/webserver_service/__init__.py

diff --git a/openpype/hosts/webpublisher/addon.py b/openpype/hosts/webpublisher/addon.py
index cb639db3fa..85e16de4a6 100644
--- a/openpype/hosts/webpublisher/addon.py
+++ b/openpype/hosts/webpublisher/addon.py
@@ -85,3 +85,22 @@ def publishfromapp(project, path, user=None, targets=None):
     from .publish_functions import cli_publish_from_app
 
     cli_publish_from_app(project, path, user, targets)
+
+
+@cli_main.command()
+@click.option("-h", "--host", help="Host", default=None)
+@click.option("-p", "--port", help="Port", default=None)
+@click.option("-e", "--executable", help="Executable")
+@click.option("-u", "--upload_dir", help="Upload dir")
+def webserver(executable, upload_dir, host=None, port=None):
+    """Starts webserver for communication with Webpublish FR via command line
+
+        OP must be congigured on a machine, eg. OPENPYPE_MONGO filled AND
+        FTRACK_BOT_API_KEY provided with api key from Ftrack.
+
+        Expect "pype.club" user created on Ftrack.
+    """
+
+    from .webserver_service import run_webserver
+
+    run_webserver(executable, upload_dir, host, port)
diff --git a/openpype/hosts/webpublisher/webserver_service/__init__.py b/openpype/hosts/webpublisher/webserver_service/__init__.py
new file mode 100644
index 0000000000..e43f3f063a
--- /dev/null
+++ b/openpype/hosts/webpublisher/webserver_service/__init__.py
@@ -0,0 +1,6 @@
+from .webserver_cli import run_webserver
+
+
+__all__ = (
+    "run_webserver",
+)
diff --git a/openpype/hosts/webpublisher/webserver_service/webserver_cli.py b/openpype/hosts/webpublisher/webserver_service/webserver_cli.py
index 47c792a575..093b53d9d3 100644
--- a/openpype/hosts/webpublisher/webserver_service/webserver_cli.py
+++ b/openpype/hosts/webpublisher/webserver_service/webserver_cli.py
@@ -32,21 +32,25 @@ from .webpublish_routes import (
 log = Logger.get_logger("webserver_gui")
 
 
-def run_webserver(*args, **kwargs):
+def run_webserver(executable, upload_dir, host=None, port=None):
     """Runs webserver in command line, adds routes."""
 
+    if not host:
+        host = "localhost"
+    if not port:
+        port = 8079
+
     manager = ModulesManager()
     webserver_module = manager.modules_by_name["webserver"]
-    host = kwargs.get("host") or "localhost"
-    port = kwargs.get("port") or 8079
+
     server_manager = webserver_module.create_new_server_manager(port, host)
     webserver_url = server_manager.url
     # queue for remotepublishfromapp tasks
     studio_task_queue = collections.deque()
 
     resource = RestApiResource(server_manager,
-                               upload_dir=kwargs["upload_dir"],
-                               executable=kwargs["executable"],
+                               upload_dir=upload_dir,
+                               executable=executable,
                                studio_task_queue=studio_task_queue)
     projects_endpoint = ProjectsEndpoint(resource)
     server_manager.add_route(
@@ -111,7 +115,7 @@ def run_webserver(*args, **kwargs):
     last_reprocessed = time.time()
     while True:
         if time.time() - last_reprocessed > 20:
-            reprocess_failed(kwargs["upload_dir"], webserver_url)
+            reprocess_failed(upload_dir, webserver_url)
             last_reprocessed = time.time()
         if studio_task_queue:
             args = studio_task_queue.popleft()