#!/bin/bash

TOOLS_DIR="tools"
declare -A tool_scripts
declare -A tool_outputs
domain=""

# Load tool scripts and previous outputs
index=1
for tool_script in $(ls "$TOOLS_DIR"/*.sh); do
    source "$tool_script"
    tool_name=$(get_tool_name)
    tool_scripts["$index"]=$tool_script
    tool_outputs["$index"]="${tool_outputs["$index"]}"
    index=$((index + 1))
done

# Main menu loop
while true; do
    echo "Available tools:"
    echo "[D] Add domain or file of domains"
    for number in $(seq 1 ${#tool_scripts[@]}); do
        source "${tool_scripts[$number]}"
        tool_name=$(get_tool_name)
        echo "[$number] $tool_name"
    done
    echo "[0] Exit and Print Output"

    read -p "Enter the number of the tool to execute (0 to exit): " choice

    if [ "$choice" == "D" ] || [ "$choice" == "d" ]; then
        echo "Choose an option to load domains:"
        echo "[1] Load domains from domains.txt"
        echo "[2] Load a single domain"
        echo "[3] Load multiple single domains"

        read -p "Enter the number of the option: " load_choice

        if [ "$load_choice" -eq 1 ]; then
            if [ -f "domains.txt" ]; then
                read -p "Do you want to load domains from domains.txt? (y/n): " confirm
                if [ "$confirm" == "y" ]; then
                    domain=$(cat "domains.txt")
                    echo "Loaded domains from domains.txt."
                fi
            else
                read -p "Enter the path to the domains.txt file: " domains_file
                if [ -f "$domains_file" ]; then
                    domain=$(cat "$domains_file")
                    echo "Loaded domains from $domains_file."
                else
                    echo "File not found: $domains_file"
                fi
            fi
        elif [ "$load_choice" -eq 2 ]; then
            read -p "Enter a single domain: " single_domain
            domain="$single_domain"
            echo "Added single domain: $domain"
        elif [ "$load_choice" -eq 3 ]; then
            read -p "Enter multiple single domains (separated by spaces): " multiple_domains
            domain="$multiple_domains"
            echo "Added multiple single domains: $domain"
        else
            echo "Invalid option: $load_choice"
        fi
    elif [[ "$choice" =~ ^[0-9]+$ ]]; then
        if [ "$choice" -eq 0 ]; then
            echo -e "\nResults:"
            for number in $(seq 1 ${#tool_scripts[@]}); do
                if [ "${tool_outputs["$number"]}" ]; then
                    echo "Tool $number Output:"
                    echo -e "${tool_outputs["$number"]}"
                    echo "------------------------------------"
                fi
            done
            exit
        elif [ "${tool_scripts[$choice]}" ]; then
            selected_script="${tool_scripts[$choice]}"
            source "$selected_script"
            tool_output=$(execute_tool)
            tool_outputs["$choice"]+="$tool_output"
            echo "Tool $choice executed."
        else
            echo "Invalid tool number: $choice"
        fi
    else
        echo "Invalid input."
    fi
done