#!/bin/bash

# Path to the Qortal folder
QORTAL_DIR=~/qortal
LOG_FILE="$QORTAL_DIR/restart_log.txt"

# Logging function
log() {
  echo "$(date '+%Y-%m-%d %H:%M:%S') - $1" | tee -a "$LOG_FILE"
}

# Check if screen is installed
if command -v screen &> /dev/null; then
  log "Screen is installed, running script in a screen session..."
  SCRIPT_NAME="restart-qortal-every-2-hours.sh"
  SCRIPT_PATH="$(realpath "$BASH_SOURCE")"
  log "Copying script from '$SCRIPT_PATH' to '$QORTAL_DIR/$SCRIPT_NAME'..."
  cp "$SCRIPT_PATH" "$QORTAL_DIR/$SCRIPT_NAME"
  log "Attempting to run script in screen..."
  screen -S qortal_restart -dm bash -c "cd $QORTAL_DIR && bash $SCRIPT_NAME"
  if [ $? -eq 0 ]; then
    log "Script successfully started in screen session 'qortal_restart'."
  else
    log "Failed to start script in screen session."
  fi
  exit 0
else
  log "Screen is not installed, running script normally..."
fi

while true; do
  # Navigate to Qortal directory
  log "Navigating to Qortal directory..."
  cd "$QORTAL_DIR" || { log "Failed to navigate to Qortal directory."; exit 1; }

  # Stop Qortal core
  log "Stopping Qortal core..."
  ./stop.sh &> stop_output.log &
  stop_pid=$!

  # Wait for 30 seconds
  sleep 30

  # Check if stop script succeeded
  if ! grep -q "Qortal ended gracefully" stop_output.log; then
    log "Stop script did not complete successfully, force killing Java..."
    killall -9 java

    # Remove blockchain lock file
    log "Removing blockchain lock file..."
    rm -rf "$QORTAL_DIR/db/blockchain.lck"
  else
    log "Qortal stopped gracefully."
  fi

  # Ensure stop process completes
  log "Waiting for stop process to complete..."
  wait $stop_pid

  # Start Qortal core
  log "Starting Qortal core..."
  ./start.sh

  # Wait for 2 hours while logging output
  log "Waiting for 2 hours before restarting..."
  sleep 2h &
  sleep_pid=$!
  tail -f "$QORTAL_DIR/qortal.log" &
  tail_pid=$!

  # Wait for the sleep to finish, then kill the tail process
  wait $sleep_pid
  log "2-hour wait complete, killing tail process..."
  kill $tail_pid
done