#!/usr/bin/env bash
set -euo pipefail

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
COMPOSE_FILE="docker-compose-multitenant.yml"

# Directories
INVOICE_ARCHIVE_DIR="${INVOICE_ARCHIVE_DIR:-/var/backups/invoice-archive}"
FULL_BACKUP_DIR="${FULL_BACKUP_DIR:-/var/backups/inventarsystem}"
LOG_DIR="$SCRIPT_DIR/logs"

KEEP_DAYS="${INVOICE_KEEP_DAYS:-3650}"
MODE="auto"
BASE_NAME=""

usage() {
    cat <<EOF
Usage: $0 [options]

Options:
  --invoice-archive-dir DIR   Directory for invoice archive files
  --invoice-keep-days N       Remove archived files older than N days
  --mode auto|docker|host     Backup mode (default: auto)
  --base-name NAME            Base filename prefix for archive files
  -h, --help                  Show this help message
EOF
}

parse_args() {
    while [[ $# -gt 0 ]]; do
        case "$1" in
            --invoice-archive-dir)
                INVOICE_ARCHIVE_DIR="$2"
                shift 2
                ;;
            --invoice-keep-days)
                KEEP_DAYS="$2"
                shift 2
                ;;
            --mode)
                MODE="$2"
                shift 2
                ;;
            --base-name)
                BASE_NAME="$2"
                shift 2
                ;;
            -h|--help)
                usage
                exit 0
                ;;
            *)
                echo "Error: unknown option '$1'" >&2
                usage
                exit 2
                ;;
        esac
    done
}

ensure_directory() {
    mkdir -p "$INVOICE_ARCHIVE_DIR"
    mkdir -p "$FULL_BACKUP_DIR"
}

cleanup_old_archives() {
    if [[ -n "$KEEP_DAYS" ]]; then
        find "$INVOICE_ARCHIVE_DIR" -maxdepth 1 -type f -mtime +"$KEEP_DAYS" -print -delete 2>/dev/null || true
    fi
    find "$FULL_BACKUP_DIR" -maxdepth 1 -type f -mtime +30 -print -delete 2>/dev/null || true
}

docker_backup() {
    if ! command -v docker >/dev/null 2>&1; then
        return 1
    fi
    if ! docker compose -f "$COMPOSE_FILE" ps -q app >/dev/null 2>&1; then
        return 1
    fi

    local timestamp
    timestamp="$(date +'%Y-%m-%d_%H-%M-%S')"

    echo "[$(date +'%T')] Starting full system backup..."

    # 1. Full MongoDB Dump
    if docker compose -f "$COMPOSE_FILE" ps -q mongodb >/dev/null 2>&1; then
        echo "[$(date +'%T')] Creating MongoDB dump..."
        docker compose -f "$COMPOSE_FILE" exec -T mongodb mongodump --archive --gzip > "$FULL_BACKUP_DIR/full_db_${timestamp}.archive.gz"
    else
        echo "Warning: MongoDB container not found. Skipping DB dump."
    fi

    # 2. Archive Logs
    if [ -d "$LOG_DIR" ]; then
        echo "[$(date +'%T')] Archiving logs..."
        tar -czf "$FULL_BACKUP_DIR/logs_${timestamp}.tar.gz" -C "$(dirname "$LOG_DIR")" "$(basename "$LOG_DIR")"
    fi

    # 3. Legacy Invoice Export
    local app_container
    app_container="$(docker compose -f "$COMPOSE_FILE" ps -q app | tr -d '[:space:]')"
    if [[ -n "$app_container" ]]; then
        local output_dir="/tmp/invoice_archive_${timestamp}"
        local remote_base="${output_dir}/${BASE_NAME}"

        docker compose -f "$COMPOSE_FILE" exec -T app mkdir -p "$output_dir"
        
        docker compose -f "$COMPOSE_FILE" exec -T app python3 /app/Web/modules/inventarsystem/backup_invoices.py \
            --archive-dir "$output_dir" \
            --base-name "$BASE_NAME" >/dev/null 2>&1 || true
        
        if docker compose -f "$COMPOSE_FILE" exec -T app test -f "${remote_base}.jsonl"; then
            docker cp "$app_container":"${remote_base}.jsonl" "$INVOICE_ARCHIVE_DIR/"
            docker cp "$app_container":"${remote_base}.csv" "$INVOICE_ARCHIVE_DIR/"
            docker cp "$app_container":"${remote_base}.meta.json" "$INVOICE_ARCHIVE_DIR/"
        else
            echo "[$(date +'%T')] Info: No invoice records found to export."
        fi

        docker compose -f "$COMPOSE_FILE" exec -T app rm -rf "$output_dir"
    fi

    return 0
}

main() {
    parse_args "$@"

    if [[ -z "$BASE_NAME" ]]; then
        BASE_NAME="invoices-$(date +'%Y-%m-%d_%H-%M-%S')"
    fi

    ensure_directory

    case "$MODE" in
        auto|docker)
            if docker_backup; then
                cleanup_old_archives
                return 0
            fi
            echo "Error: Docker backup failed." >&2
            return 1
            ;;
        *)
            echo "Error: unsupported mode '$MODE'" >&2
            usage
            return 2
            ;;
    esac
}

main "$@"