#!/bin/bash

# Configuration
BACKUP_SCRIPT="/home/litespeed/mb-backups/backup_all.sh"
LOG_DIR="/home/litespeed/mb-backups/logs/auto"
ACTION_LOG_FILE="${LOG_DIR}/schedule_actions.log"
BACKUP_LOG_PREFIX="${LOG_DIR}/backup_"

# Ensure the log directory exists
mkdir -p "$LOG_DIR"

log_action() {
    echo "[$(date +'%Y-%m-%d %H:%M:%S')] $1" >> "$ACTION_LOG_FILE"
}

# Function to add or update the cron job with dynamic logging
add_update_cron_job() {
    # Verify Restic password is provided
    if [ -z "$3" ]; then
        echo "Restic password is required."
        log_action "Attempted to add/update a schedule without providing a Restic password."
        exit 1
    fi

    # Prepare the cron command to include dynamic date in the backup log filename
    CMD="RESTIC_PASSWORD=\"$3\" $BACKUP_SCRIPT > \"${BACKUP_LOG_PREFIX}\$(date +\\%Y-\\%m-\\%d_\\%H-\\%M-\\%S).log\" 2>&1"

    # Add or update the cron job
    (crontab -l | grep -v "$BACKUP_SCRIPT" 2>/dev/null; echo "$2 $CMD") | crontab -
    local update_msg="Backup schedule updated to: $2"
    echo "$update_msg"
    log_action "$update_msg"
}

# Function to remove the cron job
remove_cron_job() {
    crontab -l | grep -v "$BACKUP_SCRIPT" | crontab -
    local remove_msg="Backup schedule removed."
    echo "$remove_msg"
    log_action "$remove_msg"
}

# Main logic to add, update, or remove the cron job based on user input
case $1 in
    add|update)
        if [ "$#" -ne 3 ]; then
            echo "Usage for add/update: $0 {add|update} 'schedule' 'restic_password'"
            echo "Example: $0 add '0 1 * * *' 'secret_password'"
            log_action "Incorrect usage for add/update. Correct format not followed."
        else
            add_update_cron_job "$@"
        fi
        ;;
    remove)
        remove_cron_job
        ;;
    *)
        echo "Invalid action: $1. Use add, update, or remove."
        log_action "Invalid action attempted: $1"
        exit 1
        ;;
esac