AST

cmdlets/adjective.py

@@ -0,0 +1,148 @@
+import json
+import os
+import sys
+from typing import List, Dict, Any, Optional, Sequence
+from ._shared import Cmdlet, CmdletArg, parse_cmdlet_args
+from helper.logger import log
+from result_table import ResultTable
+import pipeline as ctx
+
+ADJECTIVE_FILE = os.path.join(os.path.dirname(os.path.dirname(__file__)), "helper", "adjective.json")
+
+def _load_adjectives() -> Dict[str, List[str]]:
+    try:
+        if os.path.exists(ADJECTIVE_FILE):
+            with open(ADJECTIVE_FILE, 'r', encoding='utf-8') as f:
+                return json.load(f)
+    except Exception as e:
+        log(f"Error loading adjectives: {e}", file=sys.stderr)
+    return {}
+
+def _save_adjectives(data: Dict[str, List[str]]) -> bool:
+    try:
+        with open(ADJECTIVE_FILE, 'w', encoding='utf-8') as f:
+            json.dump(data, f, indent=2)
+        return True
+    except Exception as e:
+        log(f"Error saving adjectives: {e}", file=sys.stderr)
+        return False
+
+def _run(result: Any, args: Sequence[str], config: Dict[str, Any]) -> int:
+    data = _load_adjectives()
+    
+    # Parse arguments manually first to handle positional args
+    # We expect: .adjective [category] [tag] [-add] [-delete]
+    
+    # If no args, list categories
+    if not args:
+        table = ResultTable("Adjective Categories")
+        for i, (category, tags) in enumerate(data.items()):
+            row = table.add_row()
+            row.add_column("#", str(i + 1))
+            row.add_column("Category", category)
+            row.add_column("Tag Amount", str(len(tags)))
+            
+            # Selection expands to: .adjective "Category Name"
+            table.set_row_selection_args(i, [category])
+            
+        table.set_source_command(".adjective")
+        ctx.set_last_result_table_overlay(table, list(data.keys()))
+        ctx.set_current_stage_table(table)
+        print(table)
+        return 0
+
+    # We have args. First arg is likely category.
+    category = args[0]
+    
+    # Check if we are adding a new category (implicit if it doesn't exist)
+    if category not in data:
+        # If only category provided, create it
+        if len(args) == 1:
+            data[category] = []
+            _save_adjectives(data)
+            log(f"Created new category: {category}")
+        # If more args, we might be trying to add to a non-existent category
+        elif "-add" in args:
+             data[category] = []
+             # Continue to add logic
+    
+    # Handle operations within category
+    remaining_args = list(args[1:])
+    
+    # Check for -add flag
+    if "-add" in remaining_args:
+        # .adjective category -add tag
+        # or .adjective category tag -add
+        add_idx = remaining_args.index("-add")
+        # Tag could be before or after
+        tag = None
+        if add_idx + 1 < len(remaining_args):
+            tag = remaining_args[add_idx + 1]
+        elif add_idx > 0:
+            tag = remaining_args[add_idx - 1]
+            
+        if tag:
+            if tag not in data[category]:
+                data[category].append(tag)
+                _save_adjectives(data)
+                log(f"Added '{tag}' to '{category}'")
+            else:
+                log(f"Tag '{tag}' already exists in '{category}'")
+        else:
+            log("Error: No tag specified to add")
+            return 1
+            
+    # Check for -delete flag
+    elif "-delete" in remaining_args:
+        # .adjective category -delete tag
+        # or .adjective category tag -delete
+        del_idx = remaining_args.index("-delete")
+        tag = None
+        if del_idx + 1 < len(remaining_args):
+            tag = remaining_args[del_idx + 1]
+        elif del_idx > 0:
+            tag = remaining_args[del_idx - 1]
+            
+        if tag:
+            if tag in data[category]:
+                data[category].remove(tag)
+                _save_adjectives(data)
+                log(f"Deleted '{tag}' from '{category}'")
+            else:
+                log(f"Tag '{tag}' not found in '{category}'")
+        else:
+            log("Error: No tag specified to delete")
+            return 1
+            
+    # List tags in category (Default action if no flags or after modification)
+    tags = data.get(category, [])
+    table = ResultTable(f"Tags in '{category}'")
+    for i, tag in enumerate(tags):
+        row = table.add_row()
+        row.add_column("#", str(i + 1))
+        row.add_column("Tag", tag)
+        
+        # Selection expands to: .adjective "Category" "Tag"
+        # This allows typing @N -delete to delete it
+        table.set_row_selection_args(i, [category, tag])
+        
+    table.set_source_command(".adjective")
+    ctx.set_last_result_table_overlay(table, tags)
+    ctx.set_current_stage_table(table)
+    print(table)
+    
+    return 0
+
+CMDLET = Cmdlet(
+    name=".adjective",
+    aliases=["adj"],
+    summary="Manage adjective categories and tags",
+    usage=".adjective [category] [-add tag] [-delete tag]",
+    args=[
+        CmdletArg(name="category", type="string", description="Category name", required=False),
+        CmdletArg(name="tag", type="string", description="Tag name", required=False),
+        CmdletArg(name="add", type="flag", description="Add tag"),
+        CmdletArg(name="delete", type="flag", description="Delete tag"),
+    ],
+    exec=_run
+)