#!/bin/bash

[[ ! -d "$CONFIG_DIR" ]] && mkdir -p "$CONFIG_DIR"
if [[ ! -f "$CONFIG_FILE" || ! -s "$CONFIG_FILE" ]]; then
    echo "{}" > "$CONFIG_FILE"
    NEEDS_SETUP=1
fi

validate_config() {
    local is_valid=true
    local temp_file=$(mktemp)

    local keys
    keys=$(jq -r 'keys[]' "$CONFIG_FILE")

    for key in $keys; do
        if [[ -z "${ALLOWED_KEYS[$key]}" ]]; then
            jq "del(.\"$key\")" "$CONFIG_FILE" > "$temp_file" && mv "$temp_file" "$CONFIG_FILE"
        else
            local value
            value=$(jq -r --arg key "$key" '.[$key]' "$CONFIG_FILE")
            if ! [[ "$value" =~ ${ALLOWED_KEYS[$key]} ]]; then
                jq "del(.\"$key\")" "$CONFIG_FILE" > "$temp_file" && mv "$temp_file" "$CONFIG_FILE"
            fi
        fi
    done

    for req_key in "${REQUIRED_KEYS[@]}"; do
        if ! jq -e --arg key "$req_key" 'has($key)' "$CONFIG_FILE" >/dev/null; then
            NEEDS_SETUP=1
        fi
    done
}

get_value() {
    local key="$1"
    jq -r ".${key} // empty" "$CONFIG_FILE"
}

get_starting_with() {
    local prefix="$1"
    jq -r "to_entries | map(select(.key | startswith(\"$prefix\"))) | from_entries" "$CONFIG_FILE"
}

set_value() {
    local key="$1"
    local value="$2"

    if [[ -z "$key" || -z "$value" ]]; then
        echo "Usage: $0 set <key> <value>"
        exit 1
    fi
    if [[ -z "${ALLOWED_KEYS[$key]}" ]]; then
        echo "Invalid key: $key"
        exit 1
    fi
    if ! [[ "$value" =~ ${ALLOWED_KEYS[$key]} ]]; then
        echo "Invalid value for $key: $value"
        exit 1
    fi

    jq ". + {\"$key\": \"$value\"}" "$CONFIG_FILE" > temp.json && mv temp.json "$CONFIG_FILE"
}

remove_value() {
    local key="$1"
    jq "del(.${key})" "$CONFIG_FILE" > temp.json && mv temp.json "$CONFIG_FILE"
    echo "Removed $key."
}

list_values() {
    cat "$CONFIG_FILE" | jq .
}

setup_domain() {
    local domain
    local service
    local auth

    if [[ -z "$SERVICE" ]]; then
        while true; do
            echo -ne "${CYAN}Choose a service (${valid_services[*]}): ${RESET}"
            read service

            if [[ " ${valid_services[*]} " =~ " $service " ]]; then
                break
            else
                echo -e "${RED}Invalid service. Please choose from: ${valid_services[*]}${RESET}"
            fi
        done
        
        set_value "SERVICE" "$service"
    else
        service="$SERVICE"
    fi

    if [[ "$service" == "zipline" ]]; then
        while true; do
            echo -ne "${CYAN}Enter the domain (e.g. example.com): ${RESET}"
            read domain

            if [[ "$domain" =~ ^[a-zA-Z0-9.-]+$ ]]; then
                break
            else
                echo -e "${RED}Invalid domain format. Please enter a valid domain (e.g. example.com).${RESET}"
            fi
        done

        domain="https://$domain"
        domain="${domain%/}"
        [[ "$domain" != */api/upload ]] && domain="$domain/api/upload"

        set_value "DOMAIN" "$domain"
        echo -e "${GREEN}Domain set to: $domain${RESET}"

        # extra options for zipline
        echo -ne "${CYAN}Do you want to change upload options? (y/n): ${RESET}"
        read change_options
        while ! [[ "$change_options" =~ ^(y|Y|n|N|true|false)$ ]]; do
            echo -e "${RED}Invalid option. Please enter 'y' or 'n'.${RESET}"
            echo -ne "${CYAN}Do you want to change upload options? (y/n): ${RESET}"
            read change_options
        done

        if [[ "$change_options" =~ ^(y|Y)$ ]]; then
            echo -ne "${CYAN}Enter the number of max views (default: 0): ${RESET}"
            read max_views
            while [[ -n "$max_views" && ! "$max_views" =~ ^[0-9]+$ ]]; do
                echo -e "${RED}Invalid number. Please enter a valid number.${RESET}"
                echo -ne "${CYAN}Enter the number of views (default: 0): ${RESET}"
                read max_views
            done
            if [[ -n "$max_views" && "$max_views" -ne 0 ]]; then
                set_value "x-zipline-max-views" "$max_views"
                echo -e "${GREEN}Max views set to: $max_views${RESET}"
            else
                echo -e "${YELLOW}Using default max views (0). Not saving.${RESET}"
            fi

            echo -ne "${CYAN}Enter the image compression percent (default: 0): ${RESET}"
            read compression_percent
            while [[ -n "$compression_percent" && ! "$compression_percent" =~ ^[0-9]+$ ]]; do
                echo -e "${RED}Invalid number. Please enter a valid number.${RESET}"
                echo -ne "${CYAN}Enter the image compression percent (default: 0): ${RESET}"
                read compression_percent
            done
            if [[ -n "$compression_percent" && "$compression_percent" -ne 0 ]]; then
                set_value "x-zipline-image-compression-percent" "$compression_percent"
                echo -e "${GREEN}Image compression percent set to: $compression_percent${RESET}"
            else
                echo -e "${YELLOW}Using default compression percent (0). Not saving.${RESET}"
            fi

            echo -ne "${CYAN}Do you want to keep the original name on download? (y/n): ${RESET}"
            read keep_original_name
            while [[ -n "$keep_original_name" && ! "$keep_original_name" =~ ^(y|Y|n|N|true|false)$ ]]; do
                echo -e "${RED}Invalid option. Please enter 'y' or 'n'.${RESET}"
                echo -ne "${CYAN}Do you want to keep the original name on download? (y/n): ${RESET}"
                read keep_original_name
            done
            if [[ "$keep_original_name" =~ ^(y|Y)$ ]]; then
                keep_original_name="true"
            else
                keep_original_name="false"
            fi
            set_value "x-zipline-original-name" "$keep_original_name"
            echo -e "${GREEN}Keep original name set to: $keep_original_name${RESET}"

            echo -ne "${CYAN}What do you want the name format to be? (default: date): ${RESET}"
            read name_format
            while [[ -n "$name_format" && ! "$name_format" =~ ^(date|random|uuid|name|gfycat)$ ]]; do
                echo -e "${RED}Invalid option. Please enter 'date', 'random', 'uuid', 'name', or 'gfycat'.${RESET}"
                echo -ne "${CYAN}What do you want the name format to be? (default: date): ${RESET}"
                read name_format
            done
            if [[ -n "$name_format" ]]; then
                set_value "x-zipline-format" "$name_format"
                echo -e "${GREEN}Name format set to: $name_format${RESET}"
            else
                echo -e "${YELLOW}Using default name format (date). Not saving.${RESET}"
            fi

            echo -e "${GREEN}Upload options set successfully.${RESET}"
        fi
    fi

    while [[ -z "$auth" ]]; do
        echo -ne "${CYAN}Enter your auth token: ${RESET}"
        read auth
        [[ -z "$auth" ]] && echo -e "${RED}Auth token cannot be empty. Please enter a valid token.${RESET}"
    done

    set_value "${service}_auth" "$auth"
    echo -e "${GREEN}Auth token set successfully.${RESET}"
}

setup_config() {
    local save_dir
    local show_capture_preview
    local show_notifications
    local save_images
    local default_option

    echo -ne "${CYAN}What do you want the default option to be? (upload/save/copy): ${RESET}"
    read default_option
    while ! [[ "$default_option" =~ ^(upload|save|copy)$ ]]; do
        echo -e "${RED}Invalid option. Please enter 'upload', 'save', or 'copy'.${RESET}"
        echo -ne "${CYAN}Choose (upload/save/copy): ${RESET}"
        read default_option
    done
    set_value "DEFAULT_OPTION" "$default_option"

    [[ "$default_option" == "upload" ]] && setup_domain

    echo -ne "${CYAN}Do you want to save images? (y/n): ${RESET}"
    read save_images
    while ! [[ "$save_images" =~ ^(y|Y|n|N|true|false)$ ]]; do
        echo -e "${RED}Invalid option. Please enter 'y' or 'n'.${RESET}"
        echo -ne "${CYAN}Do you want to save images? (y/n): ${RESET}"
        read save_images
    done
    [[ "$save_images" =~ ^(y|Y)$ ]] && save_images="true" || save_images="false"
    set_value "SAVE_IMAGES" "$save_images"

    if [[ "$save_images" == "true" ]]; then
        echo -ne "${CYAN}Enter the directory to save images (default: $HOME/Pictures): ${RESET}"
        read save_dir
        save_dir="${save_dir:-$HOME/Pictures}"
        while [[ ! -d "$save_dir" ]]; do
            echo -e "${RED}Directory does not exist. Please enter a valid directory.${RESET}"
            echo -ne "${CYAN}Enter the directory to save images: ${RESET}"
            read save_dir
        done
        set_value "SAVE_DIR" "$save_dir"
    fi  

    echo -ne "${CYAN}Do you want to show capture notifications? (y/n): ${RESET}"
    read show_notifications
    while ! [[ "$show_notifications" =~ ^(y|Y|n|N|true|false)$ ]]; do
        echo -e "${RED}Invalid option. Please enter 'y' or 'n'.${RESET}"
        echo -ne "${CYAN}Do you want to show capture notifications? (y/n): ${RESET}"
        read show_notifications
    done
    [[ "$show_notifications" =~ ^(y|Y)$ ]] && show_notifications="true" || show_notifications="false"
    set_value "SHOW_NOTIFICATIONS" "$show_notifications"

    echo -ne "${CYAN}Do you want to show capture preview? (y/n): ${RESET}"
    read show_capture_preview
    while ! [[ "$show_capture_preview" =~ ^(y|Y|n|N|true|false)$ ]]; do
        echo -e "${RED}Invalid option. Please enter 'y' or 'n'.${RESET}"
        echo -ne "${CYAN}Do you want to show capture preview? (y/n): ${RESET}"
        read show_capture_preview
    done
    [[ "$show_capture_preview" =~ ^(y|Y)$ ]] && show_capture_preview="true" || show_capture_preview="false"
    set_value "SHOW_CAPTURE_PREVIEW" "$show_capture_preview"

    echo -e "${GREEN}${BOLD}Configuration completed successfully!${RESET}"
    echo -e "${YELLOW}You can change the configuration later by running the script with the --config option.${RESET}"

    [[ "$show_capture_preview" == "true" ]] && setup_venv
}