#!/usr/bin/env python3
# SPDX-License-Identifier: Apache-2.0
# Copyright 2024-2026 Lukasz Krzemien (biuro@softspark.eu)
# Source: https://github.com/softspark/ai-toolkit

"""remove-rule -- Unregister a rule (opposite of add-rule).

Removes the rule file from ~/.softspark/ai-toolkit/rules/ (so it is no longer
re-applied on future 'ai-toolkit install' runs), removes the generated Claude
Code user-level rule file, and strips the legacy injected block from
CLAUDE.md.

Usage:
  remove_rule.py <rule-name> [target-dir]

Arguments:
  rule-name   Name of the rule (filename without .md)
  target-dir  Directory containing .claude/CLAUDE.md (default: $HOME)
"""
from __future__ import annotations

import sys
from pathlib import Path
import re

sys.path.insert(0, str(Path(__file__).resolve().parent))
from _common import remove_rule_section


def main() -> None:
    """Unregister a rule and remove generated Claude rule artifacts."""
    if len(sys.argv) < 2:
        print("Usage: remove_rule.py <rule-name> [target-dir]", file=sys.stderr)
        sys.exit(1)

    rule_name = sys.argv[1]
    target_dir = Path(sys.argv[2]) if len(sys.argv) > 2 else Path.home()
    from paths import RULES_DIR
    rules_dir = RULES_DIR

    removed = 0

    # 1. Unregister from ~/.softspark/ai-toolkit/rules/
    rule_file = rules_dir / f"{rule_name}.md"
    if rule_file.is_file():
        rule_file.unlink()
        print(f"Unregistered: '{rule_name}' (removed from {rules_dir})")
        removed += 1
    else:
        print(f"Not registered: '{rule_name}' not found in {rules_dir}")

    # 1b. Clean up URL source metadata (if any)
    from rule_sources import unregister_source
    if unregister_source(rules_dir, rule_name):
        print(f"Removed URL source for '{rule_name}'")

    safe_rule_name = re.sub(r"[^a-zA-Z0-9_-]", "", rule_name)

    # 2. Remove generated Claude Code user-level rule file
    generated_rule = target_dir / ".claude" / "rules" / f"ai-toolkit-registered-{safe_rule_name}.md"
    if generated_rule.is_file():
        generated_rule.unlink()
        print(f"Removed generated Claude rule: {generated_rule}")
        removed += 1

    # 3. Strip legacy injected block from .claude/CLAUDE.md
    found = remove_rule_section(rule_name, target_dir)
    if found:
        print(f"Removed rule '{rule_name}' from {target_dir / '.claude' / 'CLAUDE.md'}")
        removed += 1

    if removed == 0:
        print()
        print("Nothing to unregister. To list registered rules:")
        print(f"  ls {rules_dir}/")


if __name__ == "__main__":
    main()