#!/bin/bash

# Run this script with:
# wget https://git.hhf.technology/hhf/script-management-cloudpanel/raw/branch/main/master.sh && sudo bash master.sh

# Variables for URLs and script info
this_script_url="https://git.hhf.technology/hhf/script-management-cloudpanel/raw/branch/main/master.sh"
this_script_name="Script Management CloudPanel"
formatter_url="https://git.hhf.technology/hhf/TaskFormatter/raw/branch/main/bash_task_formatter/task_formatter.sh"
BRANCH="main"
forgejo_instance="git.hhf.technology"
forgejo_owner="hhf"
forgejo_repo="script-management-cloudpanel"

# Temporary file tracking and log file list
tmp_files=()
master_log_file="/tmp/${this_script_name// /_}_$(date +'%Y%m%d_%H%M%S').log"
script_log_files=()

# Trap to clean up script on exit or interruption
trap remove_script EXIT

# Function to log messages to the master log file
log() {
    if [[ "$verbose" == "true" ]]; then
        echo -e "$1" | tee -a "$master_log_file"
    else
        echo -e "$1"
    fi
}

# Function to execute and log commands
exec_and_log() {
    if [[ "$verbose" == "true" ]]; then
        "$@" 2>&1 | tee -a "$master_log_file"
    else
        "$@"
    fi
}

# Download and source the formatter script
exec_and_log wget $formatter_url -O task_formatter.sh > /dev/null 2>&1
source ./task_formatter.sh

# Ensure sudo is available
install_sudo() {
    if ! command -v sudo &> /dev/null; then
        log "${CROSS_MARK} sudo is not installed. Installing sudo..."
        if [ -x "$(command -v apt-get)" ]; then
            exec_and_log sudo apt-get update > /dev/null 2>&1 && exec_and_log sudo apt-get install -y sudo > /dev/null 2>&1
        elif [ -x "$(command -v yum)" ]; then
            exec_and_log sudo yum install -y sudo > /dev/null 2>&1
        else
            log "${CROSS_MARK} Could not install sudo. Please install it manually."
            exit 1
        fi
    fi
}

# Ensure curl is available
install_curl() {
    if ! command -v curl &> /dev/null; then
        log "${CROSS_MARK} curl is not installed. Installing curl..."
        if [ -x "$(command -v apt-get)" ]; then
            exec_and_log sudo apt-get update > /dev/null 2>&1 && exec_and_log sudo apt-get install -y curl > /dev/null 2>&1
        elif [ -x "$(command -v yum)" ]; then
            exec_and_log sudo yum install -y curl > /dev/null 2>&1
        else
            log "${CROSS_MARK} Could not install curl. Please install it manually."
            exit 1
        fi
    fi
}

# Ensure jq is available
install_jq() {
    if ! command -v jq &> /dev/null; then
        log "${CROSS_MARK} jq is not installed. Installing jq..."
        if [ -x "$(command -v apt-get)" ]; then
            exec_and_log sudo apt-get update > /dev/null 2>&1 && exec_and_log sudo apt-get install -y jq > /dev/null 2>&1
        elif [ -x "$(command -v yum)" ]; then
            exec_and_log sudo yum install -y jq > /dev/null 2>&1
        else
            log "${CROSS_MARK} Could not install jq. Please install it manually."
            exit 1
        fi
    fi
}

# Clean up temporary files
cleanup_tmp_files() {
    log "${CHECK_MARK} Cleaning up temporary files..."
    for tmp_file in "${tmp_files[@]}"; do
        exec_and_log sudo rm -rf "$tmp_file"
    done
    log "${CHECK_MARK} Temporary files cleaned."
}

# Remove the master script and formatter
remove_script() {
    if [ -f "$0" ]; then
        rm -- "$0"
    fi
    if [ -f "task_formatter.sh" ]; then
        rm task_formatter.sh
    fi
    log "${CHECK_MARK} Script and formatter cleaned up."
}

# Modified fetch_scripts function for Forgejo API
fetch_scripts() {
    # Forgejo API endpoint for getting repository contents
    api_url="https://$forgejo_instance/api/v1/repos/$forgejo_owner/$forgejo_repo/git/trees/$BRANCH?recursive=1"
    
    # Get all files recursively and filter for .sh files
    all_scripts=$(curl -s "$api_url" | jq -r '.tree[] | select(.path | endswith(".sh")) | .path' | sort -u)

    # Log silently without displaying
    if [[ "$verbose" == "true" ]]; then
        echo "$all_scripts" >> "$master_log_file"
    fi

    # Return the scripts to be displayed later
    echo "$all_scripts"
}

# Download and run the selected script
run_script() {
    local script_path=$1
    local script_name=$(basename "$script_path")
    local url="https://$forgejo_instance/$forgejo_owner/$forgejo_repo/raw/branch/$BRANCH/$script_path"
    local script_log_file="/tmp/${script_name}_$(date +'%Y%m%d_%H%M%S').log"

    log "Requesting URL: $url"

    exec_and_log mkdir -p "/tmp/$(dirname "$script_path")"
    tmp_files+=("/tmp/$(dirname "$script_path")")

    http_status=$(exec_and_log curl -sL -w "%{http_code}" -o "/tmp/$script_path" "$url")
    tmp_files+=("/tmp/$script_path")

    log "Request completed with status code: $http_status"

    if [[ "$http_status" == "200" ]]; then
        exec_and_log chmod +x "/tmp/$script_path"
        log "Running script: /tmp/$script_path"
        sudo bash "/tmp/$script_path" "$ORIGINAL_USER" > >(tee -a "$script_log_file") 2>&1
        script_log_files+=("$script_name: $script_log_file")
    else
        log "${CROSS_MARK} Failed to download script: $script_name (HTTP status code: $http_status)"
    fi
}

# Main function to run selected scripts
run_scripts() {
    log "Fetching list of available scripts from Forgejo repository...\n"
    scripts=$(fetch_scripts)

    if [ -z "$scripts" ]; then
        log "${COLOR_RED}No scripts found in the repository.${COLOR_RESET}\n"
        exit 1
    fi

    # Display formatted list to the user
    echo -e "${COLOR_BLUE}Available scripts:${COLOR_RESET}"
    select script in $scripts "Quit"; do
        if [ "$script" == "Quit" ]; then
            break
        elif [ -n "$script" ]; then
            log "You selected $script. Running script...\n"
            run_script "$script"
            break
        else
            log "${COLOR_RED}Invalid selection. Please try again.${COLOR_RESET}\n"
        fi
    done

    printf "${COLOR_BLUE}Would you like to run more scripts? (y/n)${COLOR_RESET}\n"
    read -r choice
    if [[ "$choice" != "y" ]]; then
        return
    fi
}

# Main script logic
clear

verbose="false"
if [[ "$1" == "-v" ]]; then
    verbose="true"
    log "Verbose mode enabled. Logging to $master_log_file"
    sleep 2
fi

print_header "$this_script_name" "$this_script_url"

success=0
install_sudo
install_curl
install_jq
run_scripts

cleanup_tmp_files
remove_script

# Print all script log file locations at the end
if [[ "${#script_log_files[@]}" -gt 0 ]]; then
    echo -e "\nScript execution completed. Log files:"
    for log_file in "${script_log_files[@]}"; do
        echo -e "$log_file"
    done | tee -a "$master_log_file"
fi
if [[ "$verbose" == "true" ]]; then
    log "Master log file saved at: $master_log_file"
fi

final_message() {
    local script_name=$1
    local success=$2
    if [[ $success -eq 0 ]]; then
        log "${CHECK_MARK} $script_name completed successfully."
    else
        log "${CROSS_MARK} $script_name encountered errors."
    fi
}

exit $success