#!/bin/bash

# Function to handle errors
handle_error() {
    echo "Error on line $1"
    exit 1
}

# Trap errors and call handle_error function
trap 'handle_error $LINENO' ERR

# Function to check if a package is installed
is_installed() {
    dpkg -l | grep -i "$1" > /dev/null 2>&1
}

# Function to check if a command is available
command_exists() {
    command -v "$1" > /dev/null 2>&1
}

# Update system packages
echo "Updating system packages..."
sudo apt update

# Install curl if it's not already installed (required for NodeSource setup script)
if ! command_exists curl; then
    echo "Installing curl..."
    sudo apt install -y curl
else
    echo "curl is already installed."
fi

# Install Node.js from NodeSource if not already installed
if ! command_exists node; then
    echo "Adding Node.js 16.x source to package manager..."
    curl -fsSL https://deb.nodesource.com/setup_16.x | sudo -E bash -
    echo "Installing Node.js and npm..."
    sudo apt install -y nodejs
else
    echo "Node.js is already installed."
    node_version=$(node -v)
    echo "Installed Node.js version: $node_version"
fi

# Verify npm installation
if ! command_exists npm; then
    echo "Installing npm (Node.js package manager)..."
    sudo apt install -y npm
else
    echo "npm is already installed."
    npm_version=$(npm -v)
    echo "Installed npm version: $npm_version"
fi

# Install Yarn globally if not already installed
if ! command_exists yarn; then
    echo "Installing Yarn globally using npm..."
    sudo npm install --global yarn
else
    echo "Yarn is already installed."
    yarn_version=$(yarn -v)
    echo "Installed Yarn version: $yarn_version"
fi

# Clone the Daedal OS repository from GitHub if not already present
if [ ! -d "daedalOS" ]; then
    echo "Cloning Daedal OS repository..."
    git clone https://github.com/DustinBrett/daedalOS.git
else
    echo "Daedal OS repository already exists. Pulling latest changes..."
    cd daedalOS
    git pull
    cd ..
fi

# Navigate to the Daedal OS directory
cd daedalOS

# Install dependencies using Yarn, handle OpenSSL issue if it occurs
echo "Installing dependencies with Yarn..."
if ! yarn install; then
    echo "Encountered OpenSSL error. Setting NODE_OPTIONS to --openssl-legacy-provider."
    export NODE_OPTIONS=--openssl-legacy-provider
    yarn install
fi

# Ask the user whether to run in Development or Production mode
echo "Do you want to run Daedal OS in (d)evelopment or (p)roduction mode? (d/p)"
read mode

if [[ "$mode" == "d" ]]; then
    echo "Building Daedal OS for development..."
    yarn build:prebuild
    echo "Starting Daedal OS in development mode..."
    yarn dev
elif [[ "$mode" == "p" ]]; then
    echo "Building Daedal OS for production..."
    yarn build
    echo "Starting Daedal OS in production mode..."
    yarn serve
else
    echo "Invalid option. Exiting."
    exit 1
fi

# untested option to run Daedal OS with Docker
echo "Do you want to build and run Daedal OS using Docker? (y/n)"
read docker_mode

if [[ "$docker_mode" == "y" ]]; then
    echo "Building Docker image for Daedal OS..."
    docker build -t daedalos .
    echo "Running Daedal OS container on port 3000..."
    docker run -dp 3000:3000 --rm --name daedalos daedalos
else
    echo "Skipping Docker setup."
fi

echo "Daedal OS setup complete!"