Command "version"

SigmaHQ · May 13, 2023 · 651cdc3 · 651cdc3
1 parent e49b929
commit 651cdc3
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 2 deletions.
diff --git a/sigma/cli/main.py b/sigma/cli/main.py
@@ -1,5 +1,6 @@
 import imp
 import click
+import pkg_resources
 from .list import list_group
 from .convert import convert
 from .check import check
@@ -10,12 +11,18 @@
 def cli():
     pass
 
+@click.command()
+def version():
+    """Print version of Sigma CLI."""
+    click.echo(pkg_resources.get_distribution("sigma-cli").version)
+
 def main():
     cli.add_command(analyze_group)
     cli.add_command(plugin_group)
     cli.add_command(list_group)
     cli.add_command(convert)
     cli.add_command(check)
+    cli.add_command(version)
     cli()
 
 

diff --git a/tests/test_main.py b/tests/test_main.py
@@ -1,8 +1,15 @@
-from sigma.cli.main import cli as main
+from sigma.cli.main import cli as main, version
 from click.testing import CliRunner
+import re
 
 def test_help():
     cli = CliRunner()
     result = cli.invoke(main, ["--help"])
     assert result.exit_code == 0
-    assert len(result.stdout.split()) > 10
+    assert len(result.stdout.split()) > 10
+
+def test_version():
+    cli = CliRunner()
+    result = cli.invoke(version)
+    assert result.exit_code == 0
+    assert re.search("\\d+\\.\\d+\\.\\d+", result.stdout)