Add version flag

Author: sbalian

ape_linux.py

@@ -2,15 +2,24 @@
 
 import os
 import subprocess
+from importlib.metadata import version
 from typing import Annotated
 
 import openai
 import rich.console
 import typer
 
+__version__ = version("ape_linux")
+
 app = typer.Typer(add_completion=False, pretty_exceptions_enable=False)
 
 
+def version_callback(value: bool) -> None:
+    if value:
+        typer.echo(__version__)
+        raise typer.Exit()
+
+
 def call_llm(
     api_key: str, model: str, system_prompt: str, user_prompt: str
 ) -> str | None:
@@ -49,6 +58,16 @@ def main(
             help="Run the command if suggested. Dangerous!",
         ),
     ] = False,
+    version: Annotated[
+        bool | None,
+        typer.Option(
+            "--version",
+            "-v",
+            callback=version_callback,
+            help="Show the version and exit.",
+            is_eager=True,
+        ),
+    ] = None,
 ):
     """Suggest a command for a Linux task described in QUERY.
 

tests/test_app.py

@@ -52,3 +52,10 @@ def test_app_for_suggestion_with_execute(mockenv, monkeypatch):
     assert result.stdout == "ls\n"  # careful, this will actually run
     assert result.stderr == ""
     assert result.exit_code == 0
+
+
+def test_app_for_version(mockenv):
+    result = runner.invoke(ape_linux.app, ["--version"])
+    assert result.stdout == f"{ape_linux.__version__}\n"
+    assert result.stderr == ""
+    assert result.exit_code == 0