plugins

#!/bin/bash
set -e

PLUGINS_DIR="/app/fuseki/builtin-plugins"
RUN_PLUGINS_DIR="/app/fuseki/run/plugins"
EXTRA_DIR="/app/fuseki/run/extra"

# Ensure directories exist
mkdir -p "$PLUGINS_DIR" "$RUN_PLUGINS_DIR" "$EXTRA_DIR"

# Get all available plugin filenames
get_available_plugins() {
    local files=()
    if [[ -d "$PLUGINS_DIR" ]]; then
        for f in "$PLUGINS_DIR"/*.jar; do
            [[ -e "$f" ]] && files+=("$(basename "$f")")
        done
    fi
    if [[ -d "$RUN_PLUGINS_DIR" ]]; then
        for f in "$RUN_PLUGINS_DIR"/*.jar; do
            [[ -e "$f" ]] && files+=("$(basename "$f")")
        done
    fi
    printf '%s\n' "${files[@]}" | sort -u
}

# Check if plugin is installed (exists in EXTRA_DIR)
is_installed() {
    local plugin="$1"
    [[ -f "$EXTRA_DIR/$plugin" ]]
}

# Check if plugin exists in either plugins directory
plugin_exists() {
    local plugin="$1"
    [[ -f "$PLUGINS_DIR/$plugin" ]] || [[ -f "$RUN_PLUGINS_DIR/$plugin" ]]
}

# List available plugins (filenames only)
cmd_list() {
    get_available_plugins
}

# Show status of all plugins
cmd_status() {
    local plugins
    plugins=$(get_available_plugins)
    
    if [[ -z "$plugins" ]]; then
        echo "No plugins available."
        return 0
    fi
    
    echo "$plugins" | while read -r plugin; do
        if is_installed "$plugin"; then
            echo "$plugin (installed)"
        else
            echo "$plugin"
        fi
    done
}

# Add plugin from URL
cmd_add() {
    local url="$1"
    
    if [[ -z "$url" ]]; then
        echo "Error: URL required" >&2
        echo "Usage: plugins add <url>" >&2
        exit 1
    fi
    
    # Validate URL ends with .jar
    if [[ ! "$url" =~ \.jar$ ]]; then
        echo "Error: URL must end with .jar" >&2
        exit 1
    fi
    
    # Extract filename from URL
    local filename
    filename=$(basename "$url")
    
    # Check if file already exists
    if [[ -f "$RUN_PLUGINS_DIR/$filename" ]]; then
        echo "Error: Plugin already exists: $filename" >&2
        exit 1
    fi
    
    echo "Downloading $filename..."
    if ! curl -L --proto =https,http,file -o "$RUN_PLUGINS_DIR/$filename" "$url"; then
        echo "Error: Download failed for $url" >&2
        rm -f "$RUN_PLUGINS_DIR/$filename"
        exit 1
    fi
    
    if [ ! -s "$RUN_PLUGINS_DIR/$filename" ]; then
        echo "Error: Downloaded file is empty" >&2
        rm -f "$RUN_PLUGINS_DIR/$filename"
        exit 1
    fi
    
    echo "Added: $filename"
}

# Remove plugin
cmd_remove() {
    local plugins=("$@")
    
    if [[ ${#plugins[@]} -eq 0 ]]; then
        echo "Error: At least one plugin name required" >&2
        echo "Usage: plugins remove <filename> [filename...]" >&2
        exit 1
    fi
    
    # Validate all plugins exist first
    for plugin in "${plugins[@]}"; do
        if ! plugin_exists "$plugin"; then
            echo "Error: Plugin not found: $plugin" >&2
            exit 1
        fi
    done
    
    # Remove from both locations
    for plugin in "${plugins[@]}"; do
        local removed=0
        if [[ -f "$RUN_PLUGINS_DIR/$plugin" ]]; then
            rm "$RUN_PLUGINS_DIR/$plugin"
            removed=1
        fi
        if [[ -f "$EXTRA_DIR/$plugin" ]]; then
            rm "$EXTRA_DIR/$plugin"
            removed=1
        fi
        if [[ $removed -eq 1 ]]; then
            echo "Removed: $plugin"
        fi
    done
}

# Enable plugins
cmd_enable() {
    local plugins=("$@")
    
    if [[ ${#plugins[@]} -eq 0 ]]; then
        echo "Error: At least one plugin name required" >&2
        echo "Usage: plugins enable <filename> [filename...]" >&2
        exit 1
    fi
    
    # Validate all plugins exist first
    for plugin in "${plugins[@]}"; do
        if ! plugin_exists "$plugin"; then
            echo "Error: Plugin not found: $plugin" >&2
            exit 1
        fi
    done
    
    # Copy to extra directory
    for plugin in "${plugins[@]}"; do
        local source
        if [[ -f "$RUN_PLUGINS_DIR/$plugin" ]]; then
            source="$RUN_PLUGINS_DIR/$plugin"
        else
            source="$PLUGINS_DIR/$plugin"
        fi
        cp "$source" "$EXTRA_DIR/$plugin"
        echo "Enabled: $plugin"
    done
}

# Disable plugins
cmd_disable() {
    local plugins=("$@")
    
    if [[ ${#plugins[@]} -eq 0 ]]; then
        echo "Error: At least one plugin name required" >&2
        echo "Usage: plugins disable <filename> [filename...]" >&2
        exit 1
    fi
    
    # Validate all plugins are installed first
    for plugin in "${plugins[@]}"; do
        if ! is_installed "$plugin"; then
            echo "Error: Plugin not installed: $plugin" >&2
            exit 1
        fi
    done
    
    # Remove from extra directory
    for plugin in "${plugins[@]}"; do
        rm "$EXTRA_DIR/$plugin"
        echo "Disabled: $plugin"
    done
}

# Show usage
usage() {
    cat <<EOF
Usage: plugins <command> [options]

Commands:
  list              List available plugins
  status            List all plugins with installation status
  add <url>         Download plugin from URL to /app/fuseki/run/plugins/
  remove <name>     Remove plugin from /app/fuseki/run/plugins/ and /app/fuseki/run/extra/
  enable <names>    Copy plugin(s) to /app/fuseki/run/extra/ (activate)
  disable <names>   Remove plugin(s) from /app/fuseki/run/extra/ (deactivate)

Examples:
  plugins list
  plugins status
  plugins add https://example.com/plugin-1.0.0.jar
  plugins enable exectracker.jar
  plugins disable exectracker.jar graphql.jar
  plugins remove exectracker.jar
EOF
}

# Main command dispatcher
main() {
    local cmd="$1"
    shift || true
    
    case "$cmd" in
        list)
            cmd_list
            ;;
        status)
            cmd_status
            ;;
        add)
            cmd_add "$@"
            ;;
        remove)
            cmd_remove "$@"
            ;;
        enable)
            cmd_enable "$@"
            ;;
        disable)
            cmd_disable "$@"
            ;;
        help|--help|-h)
            usage
            ;;
        "")
            echo "Error: No command specified" >&2
            usage >&2
            exit 1
            ;;
        *)
            echo "Error: Unknown command: $cmd" >&2
            usage >&2
            exit 1
            ;;
    esac
}

# Run main only if script is executed directly (not sourced)
if [[ "${BASH_SOURCE[0]}" == "${0}" ]]; then
    main "$@"
fi