sync_ccr.sh

#!/bin/bash

# Check for uv
if ! command -v uv &> /dev/null; then
    echo "Error: 'uv' command not found. Please install uv to run this script."
    exit 1
fi

HELPER="ccr_helper.py"
SCRIPT_DIR="$(dirname "$0")"
HELPER_PATH="$SCRIPT_DIR/$HELPER"

if [ ! -f "$HELPER_PATH" ]; then
    echo "Error: $HELPER not found in $SCRIPT_DIR"
    exit 1
fi

run_helper() {
    uv run python "$HELPER_PATH" "$@"
}

apply_changes() {
    echo "Applying changes..."
    echo "Restarting CCR..."
    ccr restart
    if [ $? -eq 0 ]; then
        echo "CCR restarted successfully."
    else
        echo "Failed to restart CCR."
    fi

    echo "Updating Claude Code settings..."
    run_helper update_settings
}

select_model_interactive() {
    echo "Fetching available models..."
    models_output=$(run_helper list)

    if [ -z "$models_output" ]; then
        echo "No models found."
        return 1
    fi

    echo "Available Models:"
    i=1
    declare -a p_names
    declare -a m_names

    while IFS=, read -r p m; do
        if [ -z "$p" ] || [ -z "$m" ]; then continue; fi

        echo "$i) $p / $m"
        p_names[i]="$p"
        m_names[i]="$m"
        ((i++))
    done <<< "$models_output"

    count=$((i-1))
    if [ "$count" -eq 0 ]; then
        echo "No valid models found."
        return 1
    fi

    read -p "Select a model (1-$count): " selection

    if [[ "$selection" =~ ^[0-9]+$ ]] && [ "$selection" -ge 1 ] && [ "$selection" -le "$count" ]; then
        SELECTED_PROVIDER="${p_names[$selection]}"
        SELECTED_MODEL="${m_names[$selection]}"
        return 0
    else
        echo "Invalid selection."
        return 1
    fi
}

select_route_interactive() {
    echo "Fetching available routes..."
    routes_output=$(run_helper get_router_keys)

    if [ -z "$routes_output" ]; then
        echo "No routes found."
        return 1
    fi

    echo "Available Routes:"
    i=1
    declare -a route_keys

    # get_router_keys returns comma separated keys
    IFS=',' read -r -a keys_array <<< "$routes_output"

    for key in "${keys_array[@]}"; do
        # Trim whitespace just in case
        key=$(echo "$key" | xargs)
        if [ -z "$key" ]; then continue; fi

        echo "$i) $key"
        route_keys[i]="$key"
        ((i++))
    done

    count=$((i-1))
    if [ "$count" -eq 0 ]; then
        echo "No valid routes found."
        return 1
    fi

    read -p "Select a route to modify (1-$count): " selection

    if [[ "$selection" =~ ^[0-9]+$ ]] && [ "$selection" -ge 1 ] && [ "$selection" -le "$count" ]; then
        SELECTED_ROUTE="${route_keys[$selection]}"
        return 0
    else
        echo "Invalid selection."
        return 1
    fi
}

select_provider_interactive() {
    echo "Fetching available providers..."
    providers_output=$(run_helper list_providers)

    if [ -z "$providers_output" ]; then
        echo "No providers found."
        return 1
    fi

    echo "Available Providers:"
    i=1
    declare -a provider_names

    while IFS= read -r provider; do
        provider=$(echo "$provider" | xargs)
        if [ -z "$provider" ]; then continue; fi
        echo "$i) $provider"
        provider_names[i]="$provider"
        ((i++))
    done <<< "$providers_output"

    count=$((i-1))
    if [ "$count" -eq 0 ]; then
        echo "No valid providers found."
        return 1
    fi

    read -p "Select a provider (1-$count): " selection

    if [[ "$selection" =~ ^[0-9]+$ ]] && [ "$selection" -ge 1 ] && [ "$selection" -le "$count" ]; then
        SELECTED_PROVIDER="${provider_names[$selection]}"
        return 0
    else
        echo "Invalid selection."
        return 1
    fi
}

select_preset_interactive() {
    echo "Fetching available presets..."
    presets_output=$(run_helper list_presets)

    if [ -z "$presets_output" ]; then
        echo "No presets found."
        return 1
    fi

    echo "Available Presets:"
    i=1
    declare -a preset_names

    while IFS='|' read -r name desc updated; do
        # Trim whitespace
        name=$(echo "$name" | xargs)
        desc=$(echo "$desc" | xargs)

        if [ -z "$name" ]; then continue; fi

        echo "$i) $name - $desc"
        preset_names[i]="$name"
        ((i++))
    done <<< "$presets_output"

    count=$((i-1))
    if [ "$count" -eq 0 ]; then
        echo "No valid presets found."
        return 1
    fi

    read -p "Select a preset (1-$count): " selection

    if [[ "$selection" =~ ^[0-9]+$ ]] && [ "$selection" -ge 1 ] && [ "$selection" -le "$count" ]; then
        SELECTED_PRESET="${preset_names[$selection]}"
        return 0
    else
        echo "Invalid selection."
        return 1
    fi
}

save_preset_interactive() {
    read -p "Enter preset name: " preset_name
    if [ -z "$preset_name" ]; then
        echo "Error: Preset name cannot be empty."
        return 1
    fi

    read -p "Enter preset description (optional): " preset_desc

    if [ -n "$preset_desc" ]; then
        run_helper save_preset "$preset_name" "$preset_desc"
    else
        run_helper save_preset "$preset_name"
    fi

    if [ $? -eq 0 ]; then
        echo "✓ Preset saved successfully."
    else
        echo "✗ Failed to save preset."
        return 1
    fi
}

load_preset_interactive() {
    if select_preset_interactive; then
        run_helper load_preset "$SELECTED_PRESET"
        if [ $? -eq 0 ]; then
            echo "✓ Preset loaded successfully."
            apply_changes
        else
            echo "✗ Failed to load preset."
            return 1
        fi
    fi
}

delete_preset_interactive() {
    if select_preset_interactive; then
        read -p "Are you sure you want to delete preset '$SELECTED_PRESET'? (y/n): " confirm
        if [ "$confirm" = "y" ] || [ "$confirm" = "Y" ]; then
            run_helper delete_preset "$SELECTED_PRESET"
            if [ $? -eq 0 ]; then
                echo "✓ Preset deleted successfully."
            else
                echo "✗ Failed to delete preset."
            fi
        else
            echo "Deletion cancelled."
        fi
    fi
}

view_preset_interactive() {
    if select_preset_interactive; then
        run_helper show_preset "$SELECTED_PRESET"
    fi
}

view_router_interactive() {
    echo "Current Router Configuration:"
    echo ""
    run_helper show_router
    echo ""
    read -p "Press Enter to continue..."
}

while true; do
    echo "----------------------------------------"
    echo "CCR Model Manager"
    echo "----------------------------------------"
    echo "1. View Current Router Config"
    echo "2. View Models"
    echo "3. Add Model to Provider"
    echo "4. Update Router (All Routes)"
    echo "5. Update Router (Single Route)"
    echo "6. Apply Changes & Exit (Update Configs & Restart)"
    echo "----------------------------------------"
    echo "Presets Management:"
    echo "7. View Presets"
    echo "8. Save Current Config as Preset"
    echo "9. Load Preset"
    echo "0. View Preset Details"
    echo "----------------------------------------"
    echo "d. Delete Preset"
    echo "q. Quit (Without Applying)"
    echo "----------------------------------------"
    read -p "Select an option: " choice

    case $choice in
        1)
            view_router_interactive
            ;;
        2)
            echo "Current Models:"
            run_helper list
            ;;
        3)
            if select_provider_interactive; then
                read -p "Enter New Model Name: " m_name
                if [ -n "$m_name" ]; then
                    run_helper add_model "$SELECTED_PROVIDER" "$m_name"
                else
                    echo "Invalid model name."
                fi
            fi
            ;;
        4)
            echo "Select the model to use for ALL routes:"
            if select_model_interactive; then
                run_helper update_router_all "$SELECTED_PROVIDER" "$SELECTED_MODEL"
                apply_changes
            fi
            ;;
        5)
            if select_route_interactive; then
                echo "Select the model for route '$SELECTED_ROUTE':"
                if select_model_interactive; then
                     run_helper update_router "$SELECTED_ROUTE" "$SELECTED_PROVIDER" "$SELECTED_MODEL"
                     apply_changes
                fi
            fi
            ;;
        6)
            apply_changes
            break
            ;;
        7)
            echo "Available Presets:"
            run_helper list_presets
            ;;
        8)
            save_preset_interactive
            ;;
        9)
            load_preset_interactive
            ;;
        0)
            view_preset_interactive
            ;;
        d|D)
            delete_preset_interactive
            ;;
        q)
            echo "Exiting..."
            break
            ;;
        *)
            echo "Invalid option."
            ;;
    esac
    echo ""
done