cli-mem 0.4.0__tar.gz → 0.4.1__tar.gz

{cli_mem-0.4.0 → cli_mem-0.4.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: cli-mem
-Version: 0.4.0
+Version: 0.4.1
 Summary: Privacy-first CLI that turns shell history into searchable memory
 Project-URL: GitHub, https://github.com/matinsaurralde/mem
 Author: Matias Insaurralde

{cli_mem-0.4.0 → cli_mem-0.4.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "cli-mem"
-version = "0.4.0"
+version = "0.4.1"
 description = "Privacy-first CLI that turns shell history into searchable memory"
 license = "MIT"
 readme = "README.md"

{cli_mem-0.4.0 → cli_mem-0.4.1}/src/mem/__init__.py

@@ -1,3 +1,3 @@
 """mem — your shell history, understood."""
 
-__version__ = "0.4.0"
+__version__ = "0.4.1"

{cli_mem-0.4.0 → cli_mem-0.4.1}/src/mem/groups.py

@@ -24,7 +24,11 @@ import click
 from mem import storage
 from mem.capture import get_git_repo
 from mem.models import Group, GroupCommand, GroupFile, SavedCommand, VarDeclaration
-from mem.variables import merge_var_declarations, parse_variables, process_escapes
+from mem.variables import (
+    merge_var_declarations,
+    parse_variables,
+    process_escapes,
+)
 
 GROUP_NAME_PATTERN = re.compile(r"^[a-z][a-z0-9-]*$")
 
@@ -239,6 +243,21 @@ def export_json(group_name: str, group: Group) -> str:
     )
 
 
+def _auto_detect_vars(commands: list[GroupCommand]) -> None:
+    """Populate the vars field on imported commands by scanning for $VAR_NAME tokens.
+
+    Commands that already have vars declared (e.g., from a full mem export)
+    are left untouched. This ensures imported runbooks work with mem run
+    variable substitution out of the box.
+    """
+    for cmd in commands:
+        if cmd.vars is not None:
+            continue
+        detected = parse_variables(cmd.cmd)
+        if detected:
+            cmd.vars = [VarDeclaration(name=n) for n in detected]
+
+
 def import_from_json_str(content: str) -> tuple[str | None, list[GroupCommand]]:
     """Parse a JSON string and extract commands.
 
@@ -276,10 +295,18 @@ def import_from_json_str(content: str) -> tuple[str | None, list[GroupCommand]]:
         raise click.ClickException("Expected 'commands' to be a list.")
 
     try:
-        commands = [GroupCommand(cmd=c["cmd"], comment=c.get("comment")) for c in cmds]
+        commands = [
+            GroupCommand(
+                cmd=c["cmd"],
+                comment=c.get("comment"),
+                vars=c.get("vars"),
+            )
+            for c in cmds
+        ]
     except (KeyError, TypeError) as e:
         raise click.ClickException(f"Malformed command entry: {e}")
 
+    _auto_detect_vars(commands)
     return group_name, commands
 
 
@@ -338,6 +365,7 @@ def import_from_markdown_str(content: str) -> tuple[str | None, list[GroupComman
     if not commands:
         raise click.ClickException("No commands found in markdown table.")
 
+    _auto_detect_vars(commands)
     return group_name, commands
 
 

{cli_mem-0.4.0 → cli_mem-0.4.1}/tests/test_groups.py

@@ -2053,6 +2053,91 @@ class TestImportFromMarkdownStr:
             groups.import_from_markdown_str("# Just text\nNo table here.")
 
 
+class TestImportAutoDetectVars:
+    """Tests for automatic variable detection during import."""
+
+    def test_json_detects_single_var(self):
+        content = json.dumps({"recon": {"commands": [{"cmd": "dig $DOMAIN A +short"}]}})
+        _, cmds = groups.import_from_json_str(content)
+        assert cmds[0].vars is not None
+        assert len(cmds[0].vars) == 1
+        assert cmds[0].vars[0].name == "DOMAIN"
+
+    def test_json_detects_multiple_vars(self):
+        content = json.dumps(
+            {"commands": [{"cmd": "curl -u $USER_NAME:$API_TOKEN https://$HOST/api"}]}
+        )
+        _, cmds = groups.import_from_json_str(content)
+        var_names = [v.name for v in cmds[0].vars]
+        assert "USER_NAME" in var_names
+        assert "API_TOKEN" in var_names
+        assert "HOST" in var_names
+
+    def test_json_no_vars_leaves_none(self):
+        content = json.dumps({"commands": [{"cmd": "ls -la"}]})
+        _, cmds = groups.import_from_json_str(content)
+        assert cmds[0].vars is None
+
+    def test_json_excludes_shell_vars(self):
+        """$HOME and $PATH are common shell vars and should not be detected."""
+        content = json.dumps({"commands": [{"cmd": "ls $HOME/.config && echo $PATH"}]})
+        _, cmds = groups.import_from_json_str(content)
+        assert cmds[0].vars is None
+
+    def test_json_preserves_existing_vars(self):
+        """Commands that already declare vars (from a full export) should not be overwritten."""
+        content = json.dumps(
+            {
+                "commands": [
+                    {
+                        "cmd": "curl https://$HOST/api",
+                        "vars": [{"name": "HOST", "default": "localhost"}],
+                    }
+                ]
+            }
+        )
+        _, cmds = groups.import_from_json_str(content)
+        assert len(cmds[0].vars) == 1
+        assert cmds[0].vars[0].name == "HOST"
+        assert cmds[0].vars[0].default == "localhost"
+
+    def test_markdown_detects_vars(self):
+        md = (
+            "## ops\n\n"
+            "| Command | Description |\n"
+            "|---|---|\n"
+            "| `kubectl get pods -n $NAMESPACE` | list pods |\n"
+        )
+        _, cmds = groups.import_from_markdown_str(md)
+        assert cmds[0].vars is not None
+        assert cmds[0].vars[0].name == "NAMESPACE"
+
+    def test_markdown_no_vars(self):
+        md = (
+            "## test\n\n"
+            "| Command | Description |\n"
+            "|---|---|\n"
+            "| `echo hello` | greeting |\n"
+        )
+        _, cmds = groups.import_from_markdown_str(md)
+        assert cmds[0].vars is None
+
+    def test_mixed_commands_some_with_vars(self):
+        content = json.dumps(
+            {
+                "commands": [
+                    {"cmd": "echo starting"},
+                    {"cmd": "curl https://$HOST:$PORT/health"},
+                    {"cmd": "echo done"},
+                ]
+            }
+        )
+        _, cmds = groups.import_from_json_str(content)
+        assert cmds[0].vars is None
+        assert len(cmds[1].vars) == 2
+        assert cmds[2].vars is None
+
+
 class TestImportFromJsonDelegation:
     """Ensure the file-based function still works via delegation."""