#!/bin/bash
set -e

# Colors
BLUE='\033[1;34m'
YELLOW='\033[1;33m'
GREEN='\033[1;32m'
RED='\033[1;31m'
NC='\033[0m'

SECRETS_FILE="$HOME/.zsh_secrets"
REMOTE_NAME="hetzner-box"

# Dependency Check
if ! command -v rclone &> /dev/null; then
    echo -e "${RED}Error: rclone is not installed.${NC}"
    exit 1
fi

# Load Secrets
if [ -f "$SECRETS_FILE" ]; then
    source "$SECRETS_FILE"
fi

# Configure Remote (If missing)
if ! rclone listremotes | grep -q "^${REMOTE_NAME}:$"; then
    echo -e "${BLUE}Configuration missing. Starting setup...${NC}"

    if [ -z "$HETZNER_USER" ]; then
        DEFAULT_USER="u513875"
        read -p "Enter Username [$DEFAULT_USER]: " INPUT_USER
        HETZNER_USER="${INPUT_USER:-$DEFAULT_USER}"
    fi
    
    if [ -z "$HETZNER_HOST" ]; then
        DEFAULT_HOST="${HETZNER_USER}.your-storagebox.de"
        read -p "Enter Host [$DEFAULT_HOST]: " INPUT_HOST
        HETZNER_HOST="${INPUT_HOST:-$DEFAULT_HOST}"
    fi
    
    if [ -z "$HETZNER_PASS" ]; then
        echo -e "${YELLOW}Note: Passwords are obscured in config, but saving to secrets file uses plain text.${NC}"
        read -s -p "Enter Password: " HETZNER_PASS
        echo ""
    fi

    # Save to secrets
    if ! grep -q "HETZNER_USER" "$SECRETS_FILE" 2>/dev/null; then
        echo -e "${BLUE}Saving credentials to $SECRETS_FILE...${NC}"
        {
            echo ""
            echo "# Hetzner Storage Box"
            echo "export HETZNER_USER=\"$HETZNER_USER\""
            echo "export HETZNER_HOST=\"$HETZNER_HOST\""
            echo "export HETZNER_PASS=\"$HETZNER_PASS\""
        } >> "$SECRETS_FILE"
    fi

    OBSCURED_PASS=$(rclone obscure "$HETZNER_PASS")
    rclone config create "$REMOTE_NAME" sftp \
        host "$HETZNER_HOST" \
        user "$HETZNER_USER" \
        pass "$OBSCURED_PASS" \
        port 23 \
        --non-interactive
        
    echo -e "${GREEN}Configuration created!${NC}"
fi

# Interactive Menu
echo -e "${BLUE}--- Hetzner Storage Box Manager ---${NC}"
echo "1) List files (ls)"
echo "2) Upload (Copy LOCAL -> REMOTE)"
echo "3) Download (Copy REMOTE -> LOCAL)"
echo "4) Mount (Fast NVMe Cache)"
echo "q) Quit"

read -p "Select option: " OPTION

case $OPTION in
    1)
        read -p "Enter depth [2]: " INPUT_DEPTH
        DEPTH="${INPUT_DEPTH:-2}"
        rclone ls "${REMOTE_NAME}:" --max-depth "$DEPTH"
        ;;
    2)
        read -p "Enter local folder to upload: " LOCAL_FOLDER
        LOCAL_FOLDER=${LOCAL_FOLDER%/} 
        read -p "Enter remote folder name (e.g. Backup): " REMOTE_FOLDER
        
        echo -e "${YELLOW}Uploading (Copying)...${NC}"
        rclone copy "$LOCAL_FOLDER" "${REMOTE_NAME}:/${REMOTE_FOLDER}" -P
        ;;
    3)
        read -p "Enter remote folder to download: " REMOTE_FOLDER
        read -p "Enter local destination: " LOCAL_FOLDER
        
        echo -e "${YELLOW}Downloading (Copying)...${NC}"
        rclone copy "${REMOTE_NAME}:/${REMOTE_FOLDER}" "$LOCAL_FOLDER" -P
        ;;
    4)
        if [ -d "/mnt/resource" ]; then
            BASE_DIR="/mnt/resource"
        else
            BASE_DIR="/mnt"
        fi
        
        MOUNT_POINT="$BASE_DIR/storagebox"
        CACHE_DIR="$BASE_DIR/rclone_cache"

        echo -e "${BLUE}Target: $MOUNT_POINT${NC}"
        echo -e "${BLUE}Cache:  $CACHE_DIR${NC}"

        if [ ! -d "$MOUNT_POINT" ] || [ ! -w "$MOUNT_POINT" ]; then
            echo -e "${YELLOW}Setting up permissions for $BASE_DIR... (sudo required)${NC}"
            sudo mkdir -p "$MOUNT_POINT" "$CACHE_DIR"
            sudo chown $(whoami):$(id -gn) "$MOUNT_POINT" "$CACHE_DIR"
        fi

        echo -e "${YELLOW}Mounting... (Ctrl+C to stop)${NC}"
        
        rclone mount "${REMOTE_NAME}:" "$MOUNT_POINT" \
            --vfs-cache-mode full \
            --cache-dir "$CACHE_DIR" \
            --vfs-cache-max-size 10G \
            --vfs-cache-max-age 24h
        ;;
    q)
        exit 0
        ;;
    *)
        echo "Invalid option"
        ;;
esac