use click library for cli

Author: djcopley

README.md

@@ -45,7 +45,7 @@ Installing ShellOracle is easy!
     ```
 2. Configure ShellOracle and follow the prompts
     ```shell
-    shor configure
+    shor config init
     ```
 3. Refer to the [providers](#providers) section for specific details regarding your chosen provider.
 
@@ -97,11 +97,13 @@ Refer to the [Ollama docs](https://ollama.ai) for installation, available models
 
 To use ShellOracle with OpenAI's models, create an [API key](https://platform.openai.com/account/api-keys). Edit
 your `~/.shelloracle/config.toml` to change your provider and enter your API key.
+You can use `shor config edit` as a shorthand to edit this file.
 
 ### Deepseek
 
 To use ShellOracle with Deepseek's models, create an [API key](https://platform.deepseek.com/api_keys). Edit
 your `~/.shelloracle/config.toml` to change your provider and enter your API key.
+You can use `shor config edit` as a shorthand to edit this file.
 
 ### LocalAI
 
@@ -110,18 +112,21 @@ Refer to the [LocalAI docs](https://localai.io/) for installation, available mod
 ### XAI
 
 To use ShellOracle with XAI's models, create an [API key](https://docs.x.ai/docs/quickstart#creating-an-api-key).
+
 Edit your `~/.shelloracle/config.toml` to change your provider and enter your API key.
+You can use `shor config edit` as a shorthand to edit this file.
 
 ### Google
 
 To use ShellOracle with Google's models, create an [API key](https://aistudio.google.com/app/apikey).
 Edit your `~/.shelloracle/config.toml` to change your provider and enter your API key.
+You can use `shor config edit` as a shorthand to edit this file.
 
 ## Configuration
 
 ShellOracle's configuration is your gateway to tailoring the utility to match your preferences and requirements.
 The `~/.shelloracle/config.toml` file serves as the control center for customizing various aspects of ShellOracle's
-behavior.
+behavior. You can quickly access and edit this file using the command `shor config edit`.
 
 ## System Requirements
 

pyproject.toml

@@ -35,10 +35,11 @@ dependencies = [
     "tomlkit",
     "tomli >= 1.1.0; python_version < '3.11'",
     "google-generativeai",
+    "click>=8.1.8",
 ]
 
 [project.scripts]
-shor = "shelloracle.__main__:main"
+shor = "shelloracle.cli:main"
 
 [project.urls]
 Homepage = "https://github.com/djcopley/ShellOracle"

src/shelloracle/cli/__init__.py

@@ -0,0 +1,56 @@
+import logging
+import sys
+from importlib.metadata import version
+
+import click
+
+from shelloracle import shelloracle
+from shelloracle.cli.config import config
+from shelloracle.config import initialize_config
+from shelloracle.settings import Settings
+from shelloracle.tty_log_handler import TtyLogHandler
+
+logger = logging.getLogger(__name__)
+
+
+def configure_logging():
+    root_logger = logging.getLogger()
+    root_logger.setLevel(logging.DEBUG)
+
+    file_formatter = logging.Formatter("%(asctime)s - %(levelname)s - %(name)s - %(message)s")
+    file_handler = logging.FileHandler(Settings.shelloracle_home / "shelloracle.log")
+    file_handler.setLevel(logging.DEBUG)
+    file_handler.setFormatter(file_formatter)
+
+    tty_formatter = logging.Formatter("%(message)s")
+    tty_handler = TtyLogHandler()
+    tty_handler.setLevel(logging.WARNING)
+    tty_handler.setFormatter(tty_formatter)
+
+    root_logger.addHandler(file_handler)
+    root_logger.addHandler(tty_handler)
+
+
+@click.group(invoke_without_command=True)
+@click.version_option(version=version("shelloracle"))
+@click.pass_context
+def cli(ctx):
+    """ShellOracle command line interface."""
+    configure_logging()
+
+    # If no subcommand is invoked, run the main CLI
+    if ctx.invoked_subcommand is None:
+        try:
+            initialize_config()
+        except FileNotFoundError:
+            logger.warning("ShellOracle configuration not found. Run `shor config init` to initialize.")
+            sys.exit(1)
+
+        shelloracle.cli()
+
+
+cli.add_command(config)
+
+
+def main():
+    cli()

src/shelloracle/cli/config/__init__.py

@@ -0,0 +1,10 @@
+import click
+
+from shelloracle.cli.config.init import init
+
+
+@click.group()
+def config(): ...
+
+
+config.add_command(init)

src/shelloracle/cli/config/init.py

@@ -0,0 +1,10 @@
+import click
+
+
+@click.command()
+def init():
+    """Install shelloracle keybindings."""
+    # nest this import in a function to avoid expensive module loads
+    from shelloracle.bootstrap import bootstrap_shelloracle
+
+    bootstrap_shelloracle()