#!/bin/bash

set -e  # Exit immediately if any command exits with a non-zero status

update_config() {
    CONFIG_FILE="instance/config.py"
    TODO_FILE="ownchatbot/defaults/todo.py"
    TEMP_FILE=$(mktemp)  # Create a temporary file

        read -p "Enter the name of the existing Owncast access token used by OwnchatBot: " TOKEN_NAME

    if grep -q "ACCESS_ID =" "$CONFIG_FILE"; then  # Check if the variable already exists in CONFIG_FILE
        echo "ACCESS_ID variable already exists in '$CONFIG_FILE'."
    else
        NEW_VARIABLES="ACCESS_ID = '$TOKEN_NAME'  # The name of your access token"
    fi

    if [[ -f "$TODO_FILE" ]]; then  # Copy todo.py to the instance directory
        cp "$TODO_FILE" "instance/" || { echo "Error copying $TODO_FILE to instance/"; exit 1; }
    else
        echo "Error: '$TODO_FILE' does not exist."
        exit 1
    fi

    while IFS= read -r line; do  # Read the original config file and output to the temp file
        echo "$line" >> "$TEMP_FILE" || { echo "Error writing to temp file"; exit 1; }
    done < "$CONFIG_FILE"

    if [[ -n "$NEW_VARIABLES" ]]; then  # Append new variables if they are not already present
        echo "$NEW_VARIABLES" >> "$TEMP_FILE" || { echo "Error writing new variables to temp file"; exit 1; }
    fi

    mv "$TEMP_FILE" "$CONFIG_FILE" || { echo "Error replacing the original config file"; exit 1; }  # Move the temp file to the config file

    echo "Config file updated successfully."
}

activate_venv() {
    source env/bin/activate || { echo "Failed to activate virtual environment"; exit 1; }
}

install_module() {
    pip install --upgrade pip || { echo "Failed to upgrade pip"; exit 1; }  # Upgrade pip
    pip install -e . || { echo "Failed to install/upgrade module"; exit 1; }  # Install/upgrade modules
}

update_db() {  # Create the database. This also populates it with some goals and votes from the default rewards.py
    export FLASK_APP=ownchatbot
    if python -m flask update-db; then
        echo "Database updated successfully."
    else
        echo "Failed to update the database. Please check for errors."
        exit 1  # Exit the script with a non-zero status
    fi
}

update_config
activate_venv
install_module
deactivate

echo -e "***IMPORTANT***\n"
echo -e "Look for new \"Access Token Name\" option in the Settings menu of your OwnchatBot management panel.\n"
echo "The way you access your OwnchatBot management panel has changed. Authentication is now handled using your Owncast server as an IndieAuth server. You log in using the same credentials you use to log in to your Owncast Admin page."