From 2cad495f4a2a33f311d1132beeb2dec32be8ecef Mon Sep 17 00:00:00 2001
From: kraanzu <kraanzu@gmail.com>
Date: Mon, 26 Feb 2024 14:46:30 +0530
Subject: [PATCH] fix: workflow for app build

---
 .github/workflows/app.yml |  2 +-
 smassh/__init__.py        | 34 ++++++++++++++++++++++++++++++++++
 2 files changed, 35 insertions(+), 1 deletion(-)
 create mode 100644 smassh/__init__.py

diff --git a/.github/workflows/app.yml b/.github/workflows/app.yml
index 1f2be0c3..e6f2bac9 100644
--- a/.github/workflows/app.yml
+++ b/.github/workflows/app.yml
@@ -31,7 +31,7 @@ jobs:
 
       - name: Package app with pyinstaller
         run: |
-          python -c "with open('smassh.py', 'w') as f: f.write('from smassh.__main__ import main\nmain()\n')"
+          python -c "with open('smassh.py', 'w') as f: f.write('from smassh.__init__ import main\nmain()\n')"
           pyinstaller -F smassh.py --add-data="smassh/ui/css/:smassh/ui/css"
 
       # Setup File name
diff --git a/smassh/__init__.py b/smassh/__init__.py
new file mode 100644
index 00000000..549d8d1d
--- /dev/null
+++ b/smassh/__init__.py
@@ -0,0 +1,34 @@
+import click
+from smassh.src.plugins.add_language import AddLanguage
+from smassh.ui.tui import Smassh
+
+PKG_VERSION = "3.1.0"
+
+
+@click.group(
+    context_settings={"help_option_names": ["-h", "--help"]},
+    invoke_without_command=True,
+)
+@click.option(
+    "--version",
+    "-v",
+    is_flag=True,
+    help="Show version and exit.",
+)
+@click.pass_context
+def main(ctx, version: bool) -> None:
+    if version:
+        return print(f"smassh - v{PKG_VERSION}")
+
+    if ctx.invoked_subcommand is None:
+        Smassh().run()
+
+
+@main.command(help="Add a language to smassh")
+@click.argument("name")
+def add(name: str) -> None:
+    AddLanguage().add(name)
+
+
+if __name__ == "__main__":
+    main()