Merge pull request #36 from Wytamma/share

Add --share to cli

backend/beastiary/cli.py

@@ -2,6 +2,7 @@
 import uuid
 import typer
 import uvicorn
+from with_cloudflared import cloudflared
 
 from pathlib import Path
 from typing import List, Optional
@@ -14,7 +15,7 @@
 app = typer.Typer()
 
 
-@app.command()
+@app.command(context_settings={"help_option_names": ["-h", "--help"]})
 def main(
     log_files: Optional[List[Path]] = typer.Argument(
         None,
@@ -27,10 +28,11 @@ def main(
     token: str = typer.Option(str(uuid.uuid4()), "--token", "-t"),
     host: str = typer.Option("127.0.0.1", "--host"),
     port: str = typer.Option(5000, "--port"),
-    security: bool = typer.Option(True, help="Turn off token requirement."),
-    debug: bool = typer.Option(False, help="Set debug mode."),
+    share: bool = typer.Option(False, "--share", help="Create a publicly shareable link."),
+    no_security: bool = typer.Option(False, "--no-security", help="Turn off token requirement."),
+    delimiter: str = typer.Option("\t", "--delimiter", help="Delimiter to split file columns on. Default is tab."),
+    debug: bool = typer.Option(False, "--debug", help="Set debug mode."),
     testing: bool = typer.Option(False, help="Only for testing.", hidden=True),
-    delimiter: str = typer.Option("\t", "--delimiter"),
 ) -> None:
     """
     Realtime and remote trace inspection with BEASTIARY.
@@ -62,14 +64,22 @@ def main(
         f"http://{host}:{port}/login?token={token}", fg=typer.colors.GREEN, bold=False
     )
     typer.echo(f"Go to: {url}\n")
-    if security:
-        typer.echo(f"If prompted enter token: {token}")
-    else:
+    if no_security:
         warning = typer.style("WARNING", fg=typer.colors.YELLOW, bold=True)
         typer.echo(f"{warning}: Security disabled!")
         setattr(api, "security", False)
+    else:
+        typer.echo(f"If prompted enter token: {token}\n")
     log_level = "warning"
     if debug:
         log_level = "debug"
-    if not testing:
+    if testing:
+        typer.Exit()
+    if share:
+        typer.echo("Creating public shareable link...")
+        with cloudflared(port=port) as cloudflared_url:
+            url_with_token = typer.style(f"{cloudflared_url}", fg=typer.colors.GREEN, bold=False)
+            typer.echo(f"\nBeastiary is now publicly accessible at: {url_with_token}")
+            uvicorn.run(api, host=host, port=port, log_level=log_level)
+    else:
         uvicorn.run(api, host=host, port=port, log_level=log_level)