#!/bin/bash
# Script to set up ActivityWatch on Arch Linux with i3
# Handles installation, startup, autostart, and i3blocks status
# Handles sudo privileges automatically

set -e # Exit on any error

# Source common library for shared functions
SCRIPT_DIR="$(dirname "$(readlink -f "$0")")"
# shellcheck source=../lib/common.sh
source "$SCRIPT_DIR/../lib/common.sh"

# Function to check and request sudo privileges for package installation
check_sudo() {
  if [[ $EUID -ne 0 ]] && [[ $1 == "install" ]]; then
    echo "Package installation requires sudo privileges."
    echo "Requesting sudo access..."
    exec sudo "$0" "$@"
  fi
}

# Get the actual user (even when running with sudo)
set_actual_user_vars

echo "ActivityWatch Setup for Arch Linux + i3"
echo "======================================="
echo "Current Date: $(date)"
echo "User: $ACTUAL_USER"
echo "Target user: $ACTUAL_USER"
echo "User home: $USER_HOME"

# Function to check if ActivityWatch is installed
check_activitywatch_installed() {
  echo ""
  echo "1. Checking ActivityWatch Installation..."
  echo "========================================"

  # Check if activitywatch-bin is installed via pacman
  if pacman -Qi activitywatch-bin &> /dev/null; then
    echo "✓ activitywatch-bin package is installed"
    return 0
  fi

  # Check if aw-qt binary exists in common locations
  local common_paths=(
    "/usr/bin/aw-qt"
    "/usr/local/bin/aw-qt"
    "$USER_HOME/.local/bin/aw-qt"
    "$USER_HOME/activitywatch/aw-qt"
  )

  for path in "${common_paths[@]}"; do
    if [[ -x $path ]]; then
      echo "✓ ActivityWatch found at: $path"
      return 0
    fi
  done

  echo "✗ ActivityWatch not found"
  return 1
}

# Function to install ActivityWatch
install_activitywatch() {
  echo ""
  echo "2. Installing ActivityWatch..."
  echo "============================="

  # Check if we need sudo for installation
  check_sudo "install"

  echo "Installing activitywatch-bin from AUR..."

  # Check if an AUR helper is available
  local aur_helpers=("yay" "paru" "makepkg")
  local helper_found=""

  for helper in "${aur_helpers[@]}"; do
    if command -v "$helper" &> /dev/null; then
      helper_found="$helper"
      break
    fi
  done

  if [[ -n $helper_found && $helper_found != "makepkg" ]]; then
    echo "Using AUR helper: $helper_found"
    if [[ $EUID -eq 0 ]]; then
      # Running as root, need to install as user
      sudo -u "$ACTUAL_USER" "$helper_found" -S --noconfirm activitywatch-bin
    else
      "$helper_found" -S --noconfirm activitywatch-bin
    fi
  else
    echo "No AUR helper found. Installing manually with makepkg..."
    install_activitywatch_manual
  fi

  echo "✓ ActivityWatch installation completed"
}

# Function to manually install ActivityWatch via makepkg
install_activitywatch_manual() {
  local temp_dir="/tmp/activitywatch-install"
  local original_user="$ACTUAL_USER"

  # Create temp directory
  mkdir -p "$temp_dir"
  cd "$temp_dir"

  # Download PKGBUILD
  if command -v git &> /dev/null; then
    sudo -u "$original_user" git clone https://aur.archlinux.org/activitywatch-bin.git .
  else
    echo "Installing git..."
    pacman -S --noconfirm git
    sudo -u "$original_user" git clone https://aur.archlinux.org/activitywatch-bin.git .
  fi

  # Build and install package
  sudo -u "$original_user" makepkg -si --noconfirm

  # Cleanup
  cd /
  rm -rf "$temp_dir"
}

# Function to check if ActivityWatch is running
check_activitywatch_running() {
  echo ""
  echo "3. Checking ActivityWatch Status..."
  echo "=================================="

  # Check for aw-qt process
  if pgrep -f "aw-qt" > /dev/null; then
    echo "✓ ActivityWatch (aw-qt) is running"
    return 0
  fi

  # Check for aw-server process
  if pgrep -f "aw-server" > /dev/null; then
    echo "✓ ActivityWatch server is running"
    return 0
  fi

  echo "✗ ActivityWatch is not running"
  return 1
}

# Function to start ActivityWatch
start_activitywatch() {
  echo ""
  echo "4. Starting ActivityWatch..."
  echo "==========================="

  # Find aw-qt executable
  local aw_qt_path=""

  if command -v aw-qt &> /dev/null; then
    aw_qt_path="$(which aw-qt)"
  elif [[ -x "/usr/bin/aw-qt" ]]; then
    aw_qt_path="/usr/bin/aw-qt"
  else
    echo "✗ Could not find aw-qt executable"
    return 1
  fi

  echo "Starting ActivityWatch as user: $ACTUAL_USER"
  echo "Using aw-qt from: $aw_qt_path"

  # Start as the actual user in the background
  if [[ $EUID -eq 0 ]]; then
    # Running as root, start as user
    sudo -u "$ACTUAL_USER" env DISPLAY=:0 "$aw_qt_path" &
  else
    # Running as user
    "$aw_qt_path" &
  fi

  # Give it time to start
  sleep 3

  if check_activitywatch_running > /dev/null 2>&1; then
    echo "✓ ActivityWatch started successfully"
  else
    echo "! ActivityWatch may be starting (check system tray)"
  fi
}

# Function to setup autostart
setup_autostart() {
  echo ""
  echo "5. Setting Up Autostart..."
  echo "========================="

  local autostart_dir="$USER_HOME/.config/autostart"
  local desktop_file="$autostart_dir/activitywatch.desktop"
  local i3_config="$USER_HOME/.config/i3/config"

  # Method 1: XDG Autostart (works with most desktop environments)
  if [[ $EUID -eq 0 ]]; then
    sudo -u "$ACTUAL_USER" mkdir -p "$autostart_dir"
  else
    mkdir -p "$autostart_dir"
  fi

  # Create desktop file for autostart
  cat > "$desktop_file" << EOF
[Desktop Entry]
Type=Application
Name=ActivityWatch
Comment=Automated time tracker
Exec=aw-qt
Icon=activitywatch
Hidden=false
NoDisplay=false
X-GNOME-Autostart-enabled=true
StartupNotify=false
Terminal=false
Categories=Utility;
EOF

  # Set proper ownership if running as root
  if [[ $EUID -eq 0 ]]; then
    chown "$ACTUAL_USER:$ACTUAL_USER" "$desktop_file"
  fi

  echo "✓ Created XDG autostart entry: $desktop_file"

  # Method 2: i3 config autostart (specific to i3)
  if [[ -f $i3_config ]]; then
    # Check if autostart entry already exists
    if ! grep -q "aw-qt" "$i3_config"; then
      # Add autostart entry to i3 config
      if [[ $EUID -eq 0 ]]; then
        # Running as root
        sudo -u "$ACTUAL_USER" bash -c "cat <<'EOF' >> '$i3_config'

# Auto-start ActivityWatch
exec --no-startup-id aw-qt
EOF"
      else
        {
          printf '\n'
          printf '# Auto-start ActivityWatch\n'
          printf 'exec --no-startup-id aw-qt\n'
        } >> "$i3_config"
      fi

      echo "✓ Added ActivityWatch to i3 config autostart"
    else
      echo "✓ ActivityWatch autostart already exists in i3 config"
    fi
  else
    echo "! i3 config not found at $i3_config"
  fi
}

# Function to create i3blocks status script
create_i3blocks_status() {
  echo ""
  echo "6. Creating i3blocks Status Script..."
  echo "===================================="

  local i3blocks_dir="$USER_HOME/.config/i3blocks"
  local status_script="$i3blocks_dir/activitywatch_status.sh"

  # Create i3blocks directory if it doesn't exist
  if [[ $EUID -eq 0 ]]; then
    sudo -u "$ACTUAL_USER" mkdir -p "$i3blocks_dir"
  else
    mkdir -p "$i3blocks_dir"
  fi

  # Create the status script
  cat > "$status_script" << 'EOF'
#!/bin/bash
# ActivityWatch status script for i3blocks
# Shows ActivityWatch installation and running status

# Check if ActivityWatch is installed
check_installed() {
    # Check if activitywatch-bin package is installed
    if pacman -Qi activitywatch-bin &>/dev/null; then
        return 0
    fi
    
    # Check if aw-qt binary exists
    if command -v aw-qt &>/dev/null; then
        return 0
    fi
    
    return 1
}

# Check if ActivityWatch is running
check_running() {
    # Check for aw-qt process
    if pgrep -f "aw-qt" >/dev/null 2>&1; then
        return 0
    fi
    
    # Check for aw-server process
    if pgrep -f "aw-server" >/dev/null 2>&1; then
        return 0
    fi
    
    return 1
}

# Main logic
if ! check_installed; then
    echo "AW uninstalled"
    echo
    echo "#FF0000"  # Red
elif check_running; then
    echo "AW on"
    echo
    echo "#00FF00"  # Green
else
    echo "AW off"
    echo
    echo "#FF0000"  # Red
fi
EOF

  chmod +x "$status_script"

  # Set proper ownership if running as root
  if [[ $EUID -eq 0 ]]; then
    chown "$ACTUAL_USER:$ACTUAL_USER" "$status_script"
  fi

  echo "✓ Created i3blocks status script: $status_script"

  # Show configuration instructions
  echo ""
  echo "To add to your i3blocks config, add this block:"
  echo ""
  echo "[activitywatch]"
  echo "command=~/.config/i3blocks/activitywatch_status.sh"
  echo "interval=10"
  echo "color=#FFFFFF"
  echo ""
}

# Function to test the setup
test_setup() {
  echo ""
  echo "7. Testing Setup..."
  echo "=================="

  echo "Installation status:"
  if check_activitywatch_installed > /dev/null 2>&1; then
    echo "✓ ActivityWatch is installed"
  else
    echo "✗ ActivityWatch is not installed"
  fi

  echo "Running status:"
  if check_activitywatch_running > /dev/null 2>&1; then
    echo "✓ ActivityWatch is running"
  else
    echo "✗ ActivityWatch is not running"
  fi

  echo "Autostart files:"
  if [[ -f "$USER_HOME/.config/autostart/activitywatch.desktop" ]]; then
    echo "✓ XDG autostart file exists"
  else
    echo "✗ XDG autostart file missing"
  fi

  if [[ -f "$USER_HOME/.config/i3/config" ]] && grep -q "aw-qt" "$USER_HOME/.config/i3/config"; then
    echo "✓ i3 autostart configured"
  else
    echo "! i3 autostart may not be configured"
  fi

  echo "i3blocks status script:"
  if [[ -x "$USER_HOME/.config/i3blocks/activitywatch_status.sh" ]]; then
    echo "✓ i3blocks status script created"
    echo "Testing status script:"
    if [[ $EUID -eq 0 ]]; then
      sudo -u "$ACTUAL_USER" "$USER_HOME/.config/i3blocks/activitywatch_status.sh"
    else
      "$USER_HOME/.config/i3blocks/activitywatch_status.sh"
    fi
  else
    echo "✗ i3blocks status script missing"
  fi
}

# Function to show final instructions
show_instructions() {
  echo ""
  echo "=========================================="
  echo "ActivityWatch Setup Complete"
  echo "=========================================="
  echo "Summary:"
  echo "✓ ActivityWatch installation checked/completed"
  echo "✓ ActivityWatch startup configured"
  echo "✓ Autostart configured (XDG + i3)"
  echo "✓ i3blocks status script created"
  echo ""
  echo "Next steps:"
  echo "1. Add the i3blocks configuration to your config file:"
  echo "   ~/.config/i3blocks/config"
  echo ""
  echo "2. Reload i3 configuration:"
  echo "   Super+Shift+R"
  echo ""
  echo "3. ActivityWatch web interface should be available at:"
  echo "   http://localhost:5600"
  echo ""
  echo "4. Check system tray for ActivityWatch icon"
  echo ""
  echo "Files created:"
  echo "  ~/.config/autostart/activitywatch.desktop"
  echo "  ~/.config/i3blocks/activitywatch_status.sh"
  echo "  ~/.config/i3/config (modified)"
  echo ""
}

# Main execution flow
main() {
  local need_install=false
  local need_start=false

  # Check installation
  if ! check_activitywatch_installed; then
    need_install=true
  fi

  # Install if needed
  if [[ $need_install == true ]]; then
    install_activitywatch
  fi

  # Check if running
  if ! check_activitywatch_running; then
    need_start=true
  fi

  # Start if needed
  if [[ $need_start == true ]]; then
    start_activitywatch
  fi

  # Always set up autostart and i3blocks (in case they're missing)
  setup_autostart
  create_i3blocks_status
  test_setup
  show_instructions
}

# Run main function
main "$@"