#!/bin/sh -e

# Load common script functions
. ../common-script.sh  

# Function to install packages based on the package manager
install_package() {
    PACKAGE=$1
    if ! command_exists "$PACKAGE"; then
        case "$PACKAGER" in
            pacman)
                $ESCALATION_TOOL "$PACKAGER" -S --noconfirm "$PACKAGE"
                ;;
            *)
                $ESCALATION_TOOL "$PACKAGER" install -y "$PACKAGE"
                ;;
        esac
    else
        echo "$PACKAGE is already installed."
    fi
}       

# Function to setup and configure SSH
setup_ssh() {
    printf "%b\n" "${YELLOW}Setting up SSH...${RC}"

    # Detect package manager and install appropriate SSH package
    case "$PACKAGER" in
    "apt-get")
        install_package openssh-server
        SSH_SERVICE="ssh"
        ;;
    "pacman")
        install_package openssh
        SSH_SERVICE="sshd"
        ;;
    *)
        install_package openssh-server
        SSH_SERVICE="sshd"
        ;;
    esac

    # Enable and start the appropriate SSH service
    $ESCALATION_TOOL systemctl enable "$SSH_SERVICE"
    $ESCALATION_TOOL systemctl start "$SSH_SERVICE"

    # Get the local IP address
    LOCAL_IP=$(ip -4 addr show | awk '/inet / {print $2}' | tail -n 1)

    printf "%b\n" "${GREEN}Your local IP address is: $LOCAL_IP${RC}"

    # Check if SSH is running
    if systemctl is-active --quiet "$SSH_SERVICE"; then
        printf "%b\n" "${GREEN}SSH is up and running.${RC}"
    else
        printf "%b\n" "${RED}Failed to start SSH.${RC}"
    fi
}

# Function to setup and configure Samba
setup_samba() {
    printf "%b\n" "${YELLOW}Setting up Samba...${RC}"
    
    # Install Samba if not installed
    install_package samba

    SAMBA_CONFIG="/etc/samba/smb.conf"

    if [ -f "$SAMBA_CONFIG" ]; then
        printf "%b\n" "${YELLOW}Samba configuration file already exists in $SAMBA_CONFIG.${RC}"
        echo "Do you want to modify the existing Samba configuration? (yes/no): "
        read MODIFY_SAMBA
        if [ "$MODIFY_SAMBA" = "yes" ]; then
            $ESCALATION_TOOL nano "$SAMBA_CONFIG"
        fi
    else
        printf "%b\n" "${YELLOW}No existing Samba configuration found. Setting up a new one...${RC}"

        # Prompt user for shared directory path
        echo "Enter the path for the Samba share (default: /srv/samba/share): "
        read SHARED_DIR
        SHARED_DIR=${SHARED_DIR:-/srv/samba/share}

        # Create the shared directory if it doesn't exist
        $ESCALATION_TOOL mkdir -p "$SHARED_DIR"
        $ESCALATION_TOOL chmod -R 0777 "$SHARED_DIR"

        # Add a new Samba user
        echo "Enter Samba username: "
        read SAMBA_USER

        # Loop until the passwords match
        while true; do
            echo "Enter Samba password: "
            stty -echo
            read SAMBA_PASSWORD
            stty echo
            echo "Confirm Samba password: "
            stty -echo
            read SAMBA_PASSWORD_CONFIRM
            stty echo
            echo ""
            if [ "$SAMBA_PASSWORD" = "$SAMBA_PASSWORD_CONFIRM" ]; then
                printf "%b\n" "${GREEN}Passwords match.${RC}"
                break
            else
                printf "%b\n" "${RED}Passwords do not match. Please try again.${RC}"
            fi
        done

        # Add the user and set the password
        $ESCALATION_TOOL smbpasswd -a "$SAMBA_USER"

        # Configure Samba settings
        $ESCALATION_TOOL sh -c "cat > $SAMBA_CONFIG" <<EOL
[global]
   workgroup = WORKGROUP
   server string = Samba Server
   security = user
   map to guest = bad user
   dns proxy = no

[Share]
   path = $SHARED_DIR
   browsable = yes
   writable = yes
   guest ok = no
   read only = no
EOL
    fi

    # Enable and start Samba services
    $ESCALATION_TOOL systemctl enable smb nmb
    $ESCALATION_TOOL systemctl start smb nmb

    # Check if Samba is running
    if systemctl is-active --quiet smb && systemctl is-active --quiet nmb; then
        printf "%b\n" "${GREEN}Samba is up and running.${RC}"
        printf "%b\n" "${YELLOW}Samba share available at: $SHARED_DIR${RC}"
    else
        printf "%b\n" "${RED}Failed to start Samba.${RC}"
    fi
}

# Function to configure firewall (optional)
configure_firewall() {
    printf "%b\n" "${BLUE}Configuring firewall...${RC}"

    if command_exists ufw; then
        $ESCALATION_TOOL ufw allow OpenSSH
        $ESCALATION_TOOL ufw allow Samba
        $ESCALATION_TOOL ufw enable
        printf "%b\n" "${GREEN}Firewall configured for SSH and Samba.${RC}"
    else
        printf "%b\n" "${YELLOW}UFW is not installed. Skipping firewall configuration.${RC}"
    fi
}

setup_ssh_samba(){
    echo "Samba and SSH Setup Script"
    echo "----------------------------"
    clear

    # Display menu
    echo "Select an option:"
    echo "1. Setup SSH"
    echo "2. Setup Samba"
    echo "3. Configure Firewall"
    echo "4. Setup All"
    echo "5. Exit"

    echo "Enter your choice [1-5]: "
    read CHOICE

    case "$CHOICE" in
        1)
            setup_ssh
            ;;
        2)
            setup_samba
            ;;
        3)
            configure_firewall
            ;;
        4)
            setup_ssh
            setup_samba
            configure_firewall
            ;;
        5)
            printf "%b\n" "${GREEN}Exiting.${RC}"
            exit 0
            ;;
        *)
            printf "%b\n" "${RED}Invalid choice. Please enter a number between 1 and 5.${RC}"
            exit 1
            ;;
    esac

    printf "%b\n" "${GREEN}Setup completed.${RC}"
}

checkEnv
checkEscalationTool
setup_ssh_samba