#!/bin/bash

# Ensure a session ID is provided as the first argument
if [ -z "$1" ]; then
  echo "Usage: $0 <session-id>"
  exit 1
fi

# Function to print the ASCII art banner with a blue-ish/purple-ish gradient
print_banner() {
  local colors=(
    "\033[38;5;111m" "\033[38;5;105m" "\033[38;5;99m" "\033[38;5;63m"
    "\033[38;5;27m" "\033[38;5;93m" "\033[38;5;129m" "\033[38;5;165m"
    "\033[38;5;201m" "\033[38;5;200m" "\033[38;5;199m"
  )
  local banner=(
    "                  _ _               _ " 
    "                 | | |             | |"
    "  _ __ ___   __ _| | | __ _ _ __ __| |"
    " | '_ \` _ \ / _\` | | |/ _\` | '__/ _\` |"
    " | | | | | | (_| | | | (_| | | | (_| |"
    " |_| |_| |_|\__,_|_|_|\__,_|_|  \__,_|"
    "                                      "
    "                                      "
  )

  for i in "${!banner[@]}"; do
    printf "%b%s%b\n" "${colors[$((i % ${#colors[@]}))]}" "${banner[$i]}" "\033[0m"
  done
}

# Function to find Chrome or Chromium executable path from known locations
find_chrome_path() {
  local chrome_paths=(
#    "/usr/bin/google-chrome"
#    "/usr/local/bin/google-chrome"
#    "/usr/bin/chromium"
#    "/usr/local/bin/chromium"
    "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome"
#    "/Applications/Chromium.app/Contents/MacOS/Chromium"
  )

  for path in "${chrome_paths[@]}"; do
    if [ -x "$path" ]; then
      echo "$path"
      return 0
    fi
  done
  
  if command -v google-chrome > /dev/null; then
    echo "$(command -v google-chrome)"
    return 0
  fi

  if command -v chromium-browser > /dev/null; then
    echo "$(command -v chromium-browser)"
    return 0
  fi

  printf "Google Chrome or Chromium not found on this system.\n"
  exit 1
}

# Function to launch Chrome with specific arguments
launch_chrome() {
  local chrome_path=$(find_chrome_path)
  printf "Mallard using chrome at: %s" "$chrome_path"
  local user_data_dir="/tmp/mallard/chromeProfile"
  local devtools_file="$user_data_dir/DevToolsActivePort"

  # Remove existing DevToolsActivePort file if it exists
    rm -rf "$user_data_dir"


  local chrome_args=(
    "--remote-debugging-port=7777"
    "--user-data-dir=$user_data_dir"
    "--remote-allow-origins=https://127.0.0.1"
    "--no-default-browser-check"
    "--no-first-run"
   "--disable-features=MediaRouter,PrivacySandbox,PrivacyGuide,Translation,InterestFeedContentSuggestions,HeavyAdPrivacyMitigations,PrivacySandboxSettings4"
#    "--ash-no-nudges"
    "--disable-search-engine-choice-screen"
#    "--propagate-iph-for-testing"
    "--disable-first-run-ui"
    "--app"
#    "--force-dark-mode"
    "--enable-devtools-experiments"
    "--password-store=basic"
    "--use-mock-keychain"
    "--disable-breakpad"
   "--disable-domain-reliability"
    "--disable-sync"
#    "--no-pings"
    "https://pato.academy"
  )

  printf "➤ Launching Chrome...\n" 
  #echo -e "with args: ${chrome_args[*]}"
  "$chrome_path" "${chrome_args[@]}" > /dev/null 2>&1 &
  chrome_pid=$!
  printf " Chrome PID: %s\n" "$chrome_pid"
  sleep 2.0
  # Wait for the DevToolsActivePort file to appear with a timeout
  local timeout=60
  local waited=0
  while [ ! -f "$devtools_file" ] && [ $waited -lt $timeout ]; do
    sleep 1
    waited=$((waited + 1))
    if (( $(echo "$waited % 5" | bc -l) == 0 )); then
      printf "Waiting for DevToolsActivePort file (%d seconds elapsed)...\n" "$waited"
    fi
  done

  if [ ! -f "$devtools_file" ]; then
    printf "Error: DevToolsActivePort file not found after %d seconds.\n" "$timeout"
    kill $chrome_pid
    exit 1
  fi

  #echo "DevToolsActivePort file found"

  # Read the DevToolsActivePort file
  local debug_port
  local debug_nonce
  IFS=$'\n' read -d '' -r debug_port debug_nonce < "$devtools_file"
  local nonce=$(echo "$debug_nonce" | awk -F/ '{print $NF}')
  local ws_endpoint="ws://localhost:$debug_port/devtools/browser/$debug_nonce"
  printf "➤ Mallard ready @ %s : %d [ %s ]\n" "$nonce" "$debug_port" "$ws_endpoint"

  # Send the HTTP request to mallard.run
  local session_id=$1
  #echo "Sending request to mallard.run"
  #curl -s "https://mallard.run/session/$session_id/$nonce/$debug_port"

  # Print the final message
  printf "$(tput setaf 2)➤ Mallard has started. You can continue working on the opened Chrome screen.$(tput sgr0)\n"

  # Exit the script
  exit 0
}

# Print the ASCII art banner
print_banner

# Launch Chrome with hardcoded arguments
launch_chrome "$1"