#!/bin/bash

#Author: stirelshka8
#Description: Backup script
#License: MIT
#Version: 0.0.1test

WORK_DIR="/opt/adminlabs_backup"
TEMP_DIR="/tmp/backup_$TIMESTAMP"
CONFIG_DIR="$WORK_DIR/backup.ini"
LOG_FILE="$WORK_DIR/backup_log_$(hostname)_$(date +"%Y-%m-%d_%H-%M-%S").txt"

cd $WORK_DIR

mkdir -p "$WORK_DIR"

# Инициализация логирования
exec &> >(tee -a "$LOG_FILE")

# Проверка на выполнение скрипта с правами суперпользователя
if [ "$EUID" -ne 0 ]; then
    echo "Error: the script must be run with superuser rights (sudo)." >> "$LOG_FILE"
    exit 1
fi

# Проверка наличия необходимых утилит
for cmd in tar ftp lftp curl; do
    if ! command -v "$cmd" &> /dev/null; then
        echo "Error: $cmd is not installed." >> "$LOG_FILE"
        exit 1
    fi
done

get_ini_value() {
    local ini_file="$CONFIG_DIR"
    local section="$1"
    local key="$2"

    # Найти секцию
    local section_found=0
    while IFS= read -r line; do
        # Проверка на начало секции
        if [[ "$line" == "[$section]" ]]; then
            section_found=1
            continue
        fi

        # Если секция найдена, ищем ключ
        if [[ $section_found -eq 1 ]]; then
            # Проверка на конец секции
            if [[ "$line" =~ ^[.*]$ ]]; then
                break
            fi

            # Проверка на наличие ключа
            if [[ "$line" == "$key="* ]]; then
                echo "${line#*=}"
                return
            fi
        fi
    done < "$ini_file"

    echo "Error: Key '$key' not found in section '$section'." >> "$LOG_FILE"
    exit 1
}

get_sent_message() {
    # Настройки для Telegram
    TELEGRAM_TOKEN=$(get_ini_value "SENT" "TELEGRAM_TOKEN")
    CHAT_ID=$(get_ini_value "SENT" "CHAT_ID")

    # Настройки для Gotify
    GOTIFY_URL="$(get_ini_value "SENT" "GOTIFY_URL")/message"
    GOTIFY_APP_TOKEN=$(get_ini_value "SENT" "GOTIFY_APP_TOKEN")

    # Сообщение для отправки
    local message="$1"
    local title="$(hostname)"
    local escaped_message=$(printf '%s\n' "$message" | jq -R .)

    if [[ "$ENABLE_SENT" == "true" ]]; then
        if [[ "$ENABLE_SENT_GOTIFY" == "true" ]]; then
            response=$(curl -s -o /dev/null -w "%{http_code}" -X POST "$GOTIFY_URL" 
                -H "X-Gotify-Key: $GOTIFY_APP_TOKEN" 
                -H "Content-Type: application/json" 
                -d "{"title":$title,"message":$escaped_message,"priority":5}")
            if [[ "$response" -ne 200 ]]; then
                echo "Gotify error: $response" >> "$LOG_FILE"
            fi
        fi

        if [[ "$ENABLE_SENT_TELEGRAM" == "true" ]]; then
            response=$(curl -s -o /dev/null -w "%{http_code}" -X POST "https://api.telegram.org/bot$TELEGRAM_TOKEN/sendMessage" 
                -d "chat_id=$CHAT_ID&text=$title: $message")
            if [[ "$response" -ne 200 ]]; then
                echo "Telegram error: $response" >> "$LOG_FILE"
            fi
        fi
    fi
}


# Указываем директории и файлы для копирования
DIRECTORIES=$(get_ini_value "Backup" "DIRS")
FILES=$(get_ini_value "Backup" "FILES")
SKIP_FILES=$(get_ini_value "Backup" "SKIP_FILES")
SKIP_DIRS=$(get_ini_value "Backup" "SKIP_DIRS")
ENABLE_DIRS=$(get_ini_value "Backup" "ENABLE_DIRS")
ENABLE_FILES=$(get_ini_value "Backup" "ENABLE_FILES")
TIMESTAMP=$(date +"%Y-%m-%d_%H-%M-%S")
TAR_NAME="BACKUP_$(hostname)_$TIMESTAMP.tar.gz"
FTP_SERVER="$(get_ini_value "FTP" "FTP_SERVER")"
FTP_USER="$(get_ini_value "FTP" "FTP_USER")"
FTP_PASS="$(get_ini_value "FTP" "FTP_PASS")"
DAYS_TO_KEEP="$(get_ini_value "FTP" "DAYS_TO_KEEP")"

# Создаем временную директорию
mkdir -p "$TEMP_DIR"
echo "Create temp directory: $TEMP_DIR" >> "$LOG_FILE"

# Преобразуем списки исключений в массивы
IFS=',' read -r -a SKIP_FILE_ARRAY <<< "$SKIP_FILES"
IFS=',' read -r -a SKIP_DIR_ARRAY <<< "$SKIP_DIRS"

# Проверка, включено ли резервное копирование директорий
echo "Enable backup of directories: $ENABLE_DIRS"
if [[ "$ENABLE_DIRS" == "true" ]]; then
    IFS=',' read -r -a DIR_ARRAY <<< "$DIRECTORIES"
    for DIR in "${DIR_ARRAY[@]}"; do
        # Проверка на исключение директорий
        if [[ " ${SKIP_DIR_ARRAY[*]} " =~ " $DIR " ]]; then
            echo "Directory $DIR is in the skip list, skipping." >> "$LOG_FILE"
            continue
        fi
        
        if [ -d "$DIR" ]; then
            # Проверка наличия файлов и директорий для исключения в текущей директории
            for EXCLUDED in "${SKIP_DIR_ARRAY[@]}"; do
                if [ -e "$DIR/$EXCLUDED" ]; then
                    echo "Excluded item $EXCLUDED found in directory $DIR, skipping backup." >> "$LOG_FILE"
                    continue 2 # Пропускаем всю директорию, если найдено исключение
                fi
            done
            echo "Copy directory $DIR started."
            cp -r "$DIR" "$TEMP_DIR/" && echo "Copy directory $DIR completed." >> "$LOG_FILE" || echo "Error while copying directory $DIR." >> "$LOG_FILE"
        else
            echo "Directory $DIR not found, skipping." >> "$LOG_FILE"
        fi
    done
else
    echo "Backup of directories is disabled." >> "$LOG_FILE"
fi

# Проверка, включено ли резервное копирование файлов
echo "Enable backup of files: $ENABLE_FILES" >> "$LOG_FILE"
if [[ "$ENABLE_FILES" == "true" ]]; then
    IFS=',' read -r -a FILE_ARRAY <<< "$FILES"
    for FILE in "${FILE_ARRAY[@]}"; do
        # Проверка на исключение файлов
        if [[ " ${SKIP_FILE_ARRAY[*]} " =~ " $FILE " ]]; then
            echo "File $FILE is in the skip list, skipping." >> "$LOG_FILE"
            continue
        fi
        
        if [ -f "$FILE" ]; then
            cp "$FILE" "$TEMP_DIR/" && echo "Copy file $FILE completed." >> "$LOG_FILE" || echo "Error while copying file $FILE." >> "$LOG_FILE"
        else
            echo "File $FILE not found, skipping." >> "$LOG_FILE"
        fi
    done
else
    echo "Backup of files is disabled." >> "$LOG_FILE"
fi

# Упаковываем временную директорию в tar.gz архив
echo "Creating tar.gz archive..."
tar -czf "$WORK_DIR/$TAR_NAME" -C "$TEMP_DIR" . && echo "Created tar.gz archive: $TAR_NAME" || echo "Error while creating tar.gz archive."; rm -rf "$TEMP_DIR"; exit 1;
echo "Created tar.gz archive: $TAR_NAME"

# Проверяем наличие файлов
if [[ ! -f "$TAR_NAME" ]]; then
    echo "Fайл $TAR_NAME не найден."
    exit 1
fi

if [[ ! -f "$LOG_FILE" ]]; then
    echo "Fайл $LOG_FILE не найден."
    exit 1
fi

# Загружаем файлы на FTP-сервер
echo "Deleting old files and sending files ..." >> "$LOG_FILE"
lftp -u "$FTP_USER","$FTP_PASS" "$FTP_SERVER" <<EOF > ftp_log.txt 2>&1
set ssl:verify-certificate no
set net:timeout 10
set net:max-retries 3
set net:reconnect-interval-base 5
set net:reconnect-interval-multiplier 2
find . -type f -mtime +$DAYS_TO_KEEP -exec rm {} ;
put "$TAR_NAME"
put "$LOG_FILE"
bye
EOF

# Проверка статуса выполнения FTP команды
if [ $? -ne 0 ]; then
    echo "Error while sending files."
    rm "$TAR_NAME"
    exit 1
else
    echo "Files sent successfully."
fi

# Удаление временной директории и архива после завершения работы скрипта
rm -rf "$TEMP_DIR" || echo "Error while deleting temp directory."
echo "Temp directory deleted."
rm "$TAR_NAME" || echo "Error while deleting tar.gz archive."
echo "Tar.gz archive deleted."
rm "$LOG_FILE" || echo "Error while deleting log file."
echo "Log file deleted."

echo "Sending message..."
get_sent_message "$(get_ini_value "SENT" "TEXT_FINISH_MESSAGE")" || echo "Error while sending message."
echo "Backup completed."

cd ~
exit 0