#!/usr/bin/env bash
# =============================================================================
# uninstall-plugin.sh - Remove marketplace plugin from a consumer project
# =============================================================================
#
# Usage: ./scripts/uninstall-plugin.sh <plugin-name> <target-project-path>
#
# This script:
# 1. Removes MCP server entries from target project's .mcp.json
# 2. Removes CLAUDE.md integration section for the plugin
# 3. Is idempotent (safe to run multiple times)
#
# Examples:
#   ./scripts/uninstall-plugin.sh data-platform ~/projects/personal-portfolio
#   ./scripts/uninstall-plugin.sh projman /home/user/my-project
#
# =============================================================================
set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
REPO_ROOT="$(dirname "$SCRIPT_DIR")"

# --- Color Definitions ---
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
CYAN='\033[0;36m'
NC='\033[0m' # No Color

# --- Logging Functions ---
log_info() { echo -e "${BLUE}[INFO]${NC} $1"; }
log_success() { echo -e "${GREEN}[OK]${NC} $1"; }
log_skip() { echo -e "${YELLOW}[SKIP]${NC} $1"; }
log_error() { echo -e "${RED}[ERROR]${NC} $1"; }
log_warning() { echo -e "${YELLOW}[WARN]${NC} $1"; }

# --- Track Changes ---
CHANGES_MADE=()
SKIPPED=()

# --- Usage ---
usage() {
    echo "Usage: $0 <plugin-name> <target-project-path>"
    echo ""
    echo "Remove a marketplace plugin from a consumer project."
    echo ""
    echo "Arguments:"
    echo "  plugin-name         Name of the plugin (e.g., data-platform, viz-platform, projman)"
    echo "  target-project-path Path to the target project (absolute or relative)"
    echo ""
    echo "Examples:"
    echo "  $0 data-platform ~/projects/personal-portfolio"
    echo "  $0 projman /home/user/my-project"
    exit 1
}

# --- Prerequisite Check ---
check_prerequisites() {
    if ! command -v jq &> /dev/null; then
        log_error "jq is required but not installed."
        echo "Install with: sudo apt install jq"
        exit 1
    fi
}

# --- Validate Target Project ---
validate_target() {
    local target_path="$1"

    if [[ ! -d "$target_path" ]]; then
        log_error "Target project path does not exist: $target_path"
        exit 1
    fi

    log_success "Target project found: $target_path"
}

# --- Get MCP Servers for Plugin ---
# Reads the mcp_servers array from plugin.json
# Returns newline-separated list of MCP server names, or empty if none
get_mcp_servers() {
    local plugin_name="$1"
    local plugin_json="$REPO_ROOT/plugins/$plugin_name/.claude-plugin/plugin.json"

    if [[ ! -f "$plugin_json" ]]; then
        return
    fi

    # Read mcp_servers array from plugin.json
    # Returns empty if field doesn't exist or is empty
    jq -r '.mcp_servers // [] | .[]' "$plugin_json" 2>/dev/null || true
}

# --- Remove from .mcp.json ---
remove_from_mcp_json() {
    local plugin_name="$1"
    local target_path="$2"
    local mcp_json="$target_path/.mcp.json"

    # Check if .mcp.json exists
    if [[ ! -f "$mcp_json" ]]; then
        log_skip "No .mcp.json found - nothing to remove"
        SKIPPED+=(".mcp.json: File does not exist")
        return 0
    fi

    # Get MCP servers for this plugin
    local mcp_servers
    mcp_servers=$(get_mcp_servers "$plugin_name")

    if [[ -z "$mcp_servers" ]]; then
        # Fallback: try to remove entry with plugin name (backward compatibility)
        if jq -e ".mcpServers[\"$plugin_name\"]" "$mcp_json" > /dev/null 2>&1; then
            log_info "Removing MCP server '$plugin_name' from .mcp.json"
            local tmp_file=$(mktemp)
            jq "del(.mcpServers[\"$plugin_name\"])" "$mcp_json" > "$tmp_file"
            mv "$tmp_file" "$mcp_json"
            CHANGES_MADE+=("Removed $plugin_name from .mcp.json")
            log_success "Removed MCP server entry for '$plugin_name'"
        else
            log_skip "Plugin '$plugin_name' has no MCP servers configured"
            SKIPPED+=(".mcp.json: No MCP servers for $plugin_name")
        fi
        return 0
    fi

    # Remove each MCP server
    local servers_removed=0
    while IFS= read -r server_name; do
        [[ -z "$server_name" ]] && continue

        # Check if entry exists
        if ! jq -e ".mcpServers[\"$server_name\"]" "$mcp_json" > /dev/null 2>&1; then
            log_skip "MCP server '$server_name' not in .mcp.json"
            SKIPPED+=(".mcp.json: $server_name not present")
            continue
        fi

        # Remove MCP server entry
        log_info "Removing MCP server '$server_name' from .mcp.json"
        local tmp_file=$(mktemp)
        jq "del(.mcpServers[\"$server_name\"])" "$mcp_json" > "$tmp_file"
        mv "$tmp_file" "$mcp_json"

        CHANGES_MADE+=("Removed $server_name from .mcp.json")
        log_success "Removed MCP server entry for '$server_name'"
        ((++servers_removed))
    done <<< "$mcp_servers"
}

# --- Remove from CLAUDE.md ---
remove_from_claude_md() {
    local plugin_name="$1"
    local target_path="$2"
    local target_claude_md="$target_path/CLAUDE.md"

    # Check if CLAUDE.md exists
    if [[ ! -f "$target_claude_md" ]]; then
        log_skip "No CLAUDE.md found - nothing to remove"
        SKIPPED+=("CLAUDE.md: File does not exist")
        return 0
    fi

    # Try HTML comment markers first (preferred method)
    local begin_marker="<!-- BEGIN marketplace-plugin: $plugin_name -->"
    local end_marker="<!-- END marketplace-plugin: $plugin_name -->"

    if grep -qF "$begin_marker" "$target_claude_md" 2>/dev/null; then
        log_info "Removing '$plugin_name' section from CLAUDE.md (using markers)"

        # Remove everything between markers (inclusive) and preceding ---
        local tmp_file=$(mktemp)
        awk -v begin="$begin_marker" -v end="$end_marker" '
        BEGIN { skip = 0; prev_hr = 0; buffer = "" }
        {
            is_hr = /^---[[:space:]]*$/

            if ($0 == begin) {
                skip = 1
                # If previous line was ---, dont print it
                if (prev_hr) {
                    buffer = ""
                }
                next
            }

            if (skip) {
                if ($0 == end) {
                    skip = 0
                }
                next
            }

            # Print buffered content
            if (buffer != "") {
                print buffer
            }

            # Buffer current line (in case its --- before a marker)
            buffer = $0
            prev_hr = is_hr
        }
        END {
            # Print final buffered content
            if (buffer != "") {
                print buffer
            }
        }
        ' "$target_claude_md" > "$tmp_file"

        # Clean up multiple consecutive blank lines
        awk 'NF{blank=0} !NF{blank++} blank<=2' "$tmp_file" > "${tmp_file}.clean"
        mv "${tmp_file}.clean" "$target_claude_md"
        rm -f "$tmp_file"

        CHANGES_MADE+=("Removed $plugin_name section from CLAUDE.md")
        log_success "Removed CLAUDE.md section for '$plugin_name'"
        return 0
    fi

    # Fallback: try legacy header-based detection
    local section_header
    section_header=$(grep -E "^# ${plugin_name}( Plugin)? -? ?CLAUDE\.md Integration" "$target_claude_md" 2>/dev/null | head -1)

    if [[ -z "$section_header" ]]; then
        log_skip "Plugin '$plugin_name' section not found in CLAUDE.md"
        SKIPPED+=("CLAUDE.md: $plugin_name section not found")
        return 0
    fi

    log_info "Removing '$plugin_name' section from CLAUDE.md (legacy format)"

    # Create temp file and use awk to remove section
    local tmp_file=$(mktemp)

    awk -v header="$section_header" '
    BEGIN { skip = 0; found = 0; in_code_block = 0 }
    {
        # Track code blocks (``` markers)
        if (/^```/) {
            in_code_block = !in_code_block
        }

        # Check if this is the section header we want to remove
        if ($0 == header) {
            skip = 1
            found = 1
            next
        }

        # Check if this is a horizontal rule (---) - only count if not in code block
        is_hr = /^---[[:space:]]*$/ && !in_code_block

        # Check if this is a new plugin section header (only outside code blocks)
        is_new_plugin_section = /^# [a-z-]+( Plugin)? -? ?CLAUDE\.md Integration/ && !in_code_block && $0 != header

        # Check for HTML marker (new format)
        is_begin_marker = /^<!-- BEGIN marketplace-plugin:/ && !in_code_block

        if (skip) {
            # Stop skipping when we hit --- or a new section
            if (is_hr) {
                skip = 0
                next
            }
            if (is_new_plugin_section || is_begin_marker) {
                skip = 0
                print
            }
            next
        }

        print
    }
    END { if (!found) exit 1 }
    ' "$target_claude_md" > "$tmp_file" 2>/dev/null

    if [[ $? -eq 0 ]]; then
        # Clean up multiple consecutive blank lines
        awk 'NF{blank=0} !NF{blank++} blank<=2' "$tmp_file" > "${tmp_file}.clean"
        mv "${tmp_file}.clean" "$target_claude_md"
        rm -f "$tmp_file"
        CHANGES_MADE+=("Removed $plugin_name section from CLAUDE.md")
        log_success "Removed CLAUDE.md section for '$plugin_name'"
    else
        rm -f "$tmp_file"
        log_skip "Could not locate exact section boundaries in CLAUDE.md"
        log_warning "You may need to manually remove the $plugin_name section"
        SKIPPED+=("CLAUDE.md: Manual removal may be needed")
    fi
}

# --- Print Summary ---
print_summary() {
    local plugin_name="$1"
    local target_path="$2"

    echo ""
    echo "=============================================="
    echo -e "${GREEN}Uninstallation Summary${NC}"
    echo "=============================================="
    echo ""
    echo -e "${CYAN}Plugin:${NC} $plugin_name"
    echo -e "${CYAN}Target:${NC} $target_path"
    echo ""

    if [[ ${#CHANGES_MADE[@]} -gt 0 ]]; then
        echo -e "${GREEN}Changes Made:${NC}"
        for change in "${CHANGES_MADE[@]}"; do
            echo "  ✓ $change"
        done
        echo ""
    fi

    if [[ ${#SKIPPED[@]} -gt 0 ]]; then
        echo -e "${YELLOW}Skipped (not present or N/A):${NC}"
        for skip in "${SKIPPED[@]}"; do
            echo "  - $skip"
        done
        echo ""
    fi

    if [[ ${#CHANGES_MADE[@]} -gt 0 ]]; then
        echo -e "${YELLOW}⚠️  IMPORTANT:${NC}"
        echo "  Restart your Claude Code session for changes to take effect."
        echo ""
    fi
}

# =============================================================================
# Main Execution
# =============================================================================

# Check arguments
if [[ $# -lt 2 ]]; then
    usage
fi

PLUGIN_NAME="$1"
TARGET_PATH="$2"

# Resolve target path to absolute
TARGET_PATH=$(cd "$TARGET_PATH" 2>/dev/null && pwd || echo "$TARGET_PATH")

echo ""
echo "=============================================="
echo -e "${BLUE}Uninstalling Plugin: $PLUGIN_NAME${NC}"
echo "=============================================="
echo ""

# Run checks
check_prerequisites
validate_target "$TARGET_PATH"

echo ""

# Perform uninstallation
remove_from_mcp_json "$PLUGIN_NAME" "$TARGET_PATH"
remove_from_claude_md "$PLUGIN_NAME" "$TARGET_PATH"

# Print summary
print_summary "$PLUGIN_NAME" "$TARGET_PATH"