Medios-Macina/cmdlets/adjective.py

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
)