#!/bin/bash
set -euo pipefail

# --- Resolve to this script's directory ---
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
VENV_DIR="$SCRIPT_DIR/venv"
PYTHON_SCRIPT="export_w1hd9_raw.py"
OUTPUT_DIR="$SCRIPT_DIR"
CALLSIGN="W1HD-9"

# 8-hour window in UTC
START_TIME=$(date -u -d "8 hours ago" +"%Y-%m-%d %H:%M")
END_TIME=$(date -u +"%Y-%m-%d %H:%M")

# Safe filename
START_SAFE="${START_TIME//[: ]/_}"
END_SAFE="${END_TIME//[: ]/_}"
OUTPUT_FILE="w1hd9_raw_${START_SAFE}_${END_SAFE}.csv"

# --- System deps (Chrome + Python venv) ---
sudo apt-get update
sudo apt-get install -y google-chrome-stable python3-venv python3-full

# --- Work as bbl (venv + run) ---
sudo -u bbl -H bash -lc '
  set -euo pipefail

  SCRIPT_DIR="'"$SCRIPT_DIR"'"
  VENV_DIR="'"$VENV_DIR"'"
  OUTPUT_DIR="'"$OUTPUT_DIR"'"
  PYTHON_SCRIPT="'"$PYTHON_SCRIPT"'"
  CALLSIGN="'"$CALLSIGN"'"
  START_TIME="'"$START_TIME"'"
  END_TIME="'"$END_TIME"'"
  OUTPUT_FILE="'"$OUTPUT_FILE"'"

  # Load env (credentials) explicitly
  if [ -f "$HOME/.bashrc" ]; then
    # shellcheck disable=SC1090
    source "$HOME/.bashrc"
  fi
  if [ -f "$HOME/.aprs.env" ]; then
    set -a
    # shellcheck disable=SC1090
    source "$HOME/.aprs.env"
    set +a
  fi

  if [ -z "${APRS_USERNAME-}" ] || [ -z "${APRS_PASSWORD-}" ]; then
    echo "ERROR: APRS_USERNAME/APRS_PASSWORD not set for user bbl (in ~/.bashrc or ~/.aprs.env)" >&2
    exit 2
  fi

  # Create venv if needed
  if [ ! -d "$VENV_DIR" ]; then
    python3 -m venv "$VENV_DIR"
  fi
  source "$VENV_DIR/bin/activate"

  python -m pip install --upgrade pip
  pip install --upgrade selenium webdriver-manager

  # --- Write Python driver (keep PYEOF exact; no spaces) ---
  cat > "$SCRIPT_DIR/$PYTHON_SCRIPT" <<PYEOF
#!/usr/bin/env python3
import os, time, glob, sys, traceback
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.keys import Keys
from webdriver_manager.chrome import ChromeDriverManager
from selenium.webdriver.chrome.service import Service as ChromeService
from selenium.webdriver.chrome.options import Options as ChromeOptions

USERNAME = os.environ.get("APRS_USERNAME")
PASSWORD = os.environ.get("APRS_PASSWORD")
OUTPUT_DIR = os.environ.get("OUTPUT_DIR", ".")
CALLSIGN   = os.environ.get("CALLSIGN", "W1HD-9")
START_TIME = os.environ.get("START_TIME")
END_TIME   = os.environ.get("END_TIME")
OUTPUT_FILE= os.environ.get("OUTPUT_FILE", "export.csv")
DEBUG_DIR  = os.path.join(OUTPUT_DIR, "debug")
os.makedirs(OUTPUT_DIR, exist_ok=True)
os.makedirs(DEBUG_DIR, exist_ok=True)

def snap(tag, driver):
    try:
        ts = str(int(time.time()))
        png  = os.path.join(DEBUG_DIR, f"{ts}_{tag}.png")
        html = os.path.join(DEBUG_DIR, f"{ts}_{tag}.html")
        driver.save_screenshot(png)
        open(html, "w", encoding="utf-8").write(driver.page_source)
        print("DEBUG wrote", png, "and", html)
    except Exception as e:
        print("DEBUG snap failed:", e)

def die(msg, driver=None):
    print("ERROR:", msg)
    if driver: snap("exception", driver)
    sys.exit(1)

if not USERNAME or not PASSWORD:
    die("APRS_USERNAME/APRS_PASSWORD missing in environment")

prefs = {
    "download.default_directory": OUTPUT_DIR,
    "download.prompt_for_download": False,
    "profile.default_content_setting_values.automatic_downloads": 1,
    "safebrowsing.enabled": True,
}
opts = ChromeOptions()
opts.add_argument("--headless=new")
opts.add_argument("--no-sandbox")
opts.add_argument("--disable-dev-shm-usage")
opts.add_argument("--disable-gpu")
opts.add_argument("--window-size=1280,1200")
opts.add_argument("--disable-blink-features=AutomationControlled")
opts.add_argument("--user-agent=Mozilla/5.0 (X11; Linux x86_64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121 Safari/537.36")
opts.add_experimental_option("prefs", prefs)

driver = None
try:
    driver = webdriver.Chrome(service=ChromeService(ChromeDriverManager().install()), options=opts)
    wait = WebDriverWait(driver, 25)
    try:
        caps = driver.capabilities
        print("Chrome version:", caps.get("browserVersion"))
        print("Driver:", caps.get("chrome", {}).get("chromedriverVersion"))
    except Exception:
        pass

    try:
        driver.execute_cdp_cmd("Page.setDownloadBehavior", {"behavior": "allow", "downloadPath": OUTPUT_DIR})
    except Exception:
        pass

    # 1) Go directly to login form (avoids modal)
    driver.get("https://aprs.fi/?c=login")
    time.sleep(1)
    snap("login_direct", driver)

    # 2) Robust selectors (properly quoted XPaths)
    selectors_user = [
        '//*[@id="username"]',
        '//*[@name="username"]',
        '//*[@name="login"]',
        '//input[@type="email"]',
        '//input[contains(translate(@placeholder,"EMAIL","email"), "email")]',
        '//input[@type="text" and (contains(@name,"user") or contains(@id,"user") or contains(@name,"email"))]'
    ]
    selectors_pw = [
        '//*[@id="password"]',
        '//*[@name="password"]',
        '//input[@type="password"]'
    ]
    def find_first(xps):
        for xp in xps:
            try:
                return wait.until(EC.presence_of_element_located((By.XPATH, xp)))
            except Exception:
                pass
        snap("login_no_fields", driver)
        die("Could not find login fields", driver)

    user = find_first(selectors_user)
    pw   = find_first(selectors_pw)

    user.clear(); user.send_keys(USERNAME)
    time.sleep(0.3)
    pw.clear(); pw.send_keys(PASSWORD); pw.send_keys(Keys.RETURN)
    time.sleep(2)
    snap("after_login", driver)

    # 3) Export page (positions)
    driver.get("https://aprs.fi/export/")
    time.sleep(1)
    snap("export_page", driver)

    call_el  = wait.until(EC.presence_of_element_located((By.XPATH, '//*[@name="call" or @id="call" or contains(@placeholder,"callsign")]')))
    start_el = wait.until(EC.presence_of_element_located((By.XPATH, '//*[@name="start" or @id="start"]')))
    end_el   = wait.until(EC.presence_of_element_located((By.XPATH, '//*[@name="end" or @id="end"]')))

    call_el.clear();  call_el.send_keys(CALLSIGN)
    start_el.clear(); start_el.send_keys(START_TIME)
    end_el.clear();   end_el.send_keys(END_TIME)
    snap("filled_form", driver)

    # Format radio (CSV) if present
    try:
        radios = driver.find_elements(By.XPATH, '//input[@type="radio" and (contains(@value,"csv") or contains(@id,"csv"))]')
        if radios:
            radios[0].click()
    except Exception:
        pass

    # Submit export
    try:
        btn = wait.until(EC.element_to_be_clickable((By.XPATH, '//button[contains(.,"Export") or contains(.,"Download") or @type="submit"]')))
        btn.click()
    except Exception:
        try:
            form = driver.find_element(By.XPATH, "//form[.//*[@name='call']]")
            form.submit()
        except Exception:
            snap("no_submit", driver)
            die("Could not locate export/submit control", driver)

    time.sleep(2)
    snap("after_submit", driver)

    # 4) Wait for download to finish
    before = set(glob.glob(os.path.join(OUTPUT_DIR, "*")))
    deadline = time.time() + 40
    downloaded = None
    while time.time() < deadline:
        time.sleep(1)
        after = set(glob.glob(os.path.join(OUTPUT_DIR, "*")))
        new = [f for f in (after - before) if not f.endswith(".crdownload")]
        if new:
            downloaded = sorted(new, key=os.path.getmtime)[-1]
            break

    if not downloaded:
        snap("no_download", driver)
        die(f"No file appeared in {OUTPUT_DIR}", driver)

    dest = os.path.join(OUTPUT_DIR, OUTPUT_FILE)
    if downloaded != dest:
        try:
            os.replace(downloaded, dest)
        except Exception as e:
            print("Downloaded but rename failed:", e)
            dest = downloaded
    print("Saved", dest)

except Exception:
    traceback.print_exc()
    die("Unhandled exception", driver)
finally:
    if driver:
        driver.quit()
PYEOF

  chmod +x "$SCRIPT_DIR/$PYTHON_SCRIPT"

  export OUTPUT_DIR="$OUTPUT_DIR"
  export CALLSIGN="$CALLSIGN"
  export START_TIME="$START_TIME"
  export END_TIME="$END_TIME"
  export OUTPUT_FILE="$OUTPUT_FILE"

  echo "Running export at $(date -u) ..."
  "$VENV_DIR/bin/python" "$SCRIPT_DIR/$PYTHON_SCRIPT"
'

echo "Done. Check: $OUTPUT_DIR/$OUTPUT_FILE (and $OUTPUT_DIR/debug/ for screenshots & HTML)"