import { copyToClipboard } from "./clipboard";
import { setupAutoLock, touchAutoLock } from "./alarms";
import * as vault from "./vault";
import { activateLicense, getLicenseInfo } from "./license";
import {
  getPasskeyProviderState,
  getPendingPasskeyRequest,
  resolvePendingPasskeyRequest,
  setPasskeyProviderEnabled,
} from "./passkey";
import type { Message, MessageResponse, MessageType } from "../shared/types";
import type { SyncStatus } from "../shared/sync-types";

const PRIVATE_IPV4_RANGES = [
  { prefix: "10." },
  { prefix: "127." },
  { prefix: "169.254." },
  { prefix: "192.168." },
];

export function setupMessageHandler(): void {
  chrome.runtime.onMessage.addListener(
    (msg: unknown, sender, sendResponse) => {
      if (!isMessage(msg)) {
        sendResponse({ ok: false, error: "invalid message" });
        return true;
      }
      handleMessage(msg, sender)
        .then(sendResponse)
        .catch((err) =>
          sendResponse({
            ok: false,
            error: err instanceof Error ? err.message : String(err),
          })
        );
      return true;
    }
  );
}

function isMessage(m: unknown): m is Message {
  return typeof m === "object" && m !== null && "type" in m;
}

async function handleMessage(
  msg: Message,
  sender: chrome.runtime.MessageSender
): Promise<MessageResponse> {
  try {
    // Validate sender — only allow messages from our extension pages/scripts.
    // `sender.id` is undefined for messages originating from extension UI
    // (popup/options), which is legitimate; reject only when it is defined
    // and does not match our extension id.
    if (sender.id !== undefined && sender.id !== chrome.runtime.id) {
      return { ok: false, error: "invalid sender" };
    }

    if (shouldRefreshAutoLock(msg.type) && vault.isUnlocked()) {
      touchAutoLock();
    }

    switch (msg.type) {
      case "CREATE_VAULT": {
        const { password } = msg.payload as { password: string };
        await vault.createVault(password);
        await setupAutoLock();
        broadcastToContentScripts({ type: "VAULT_UNLOCKED" });
        return { ok: true };
      }
      case "UNLOCK": {
        const { password } = msg.payload as { password: string };
        const ok = await vault.unlock(password);
        if (ok) {
          await setupAutoLock();
          broadcastToContentScripts({ type: "VAULT_UNLOCKED" });
        }
        return { ok };
      }
      case "LOCK":
        vault.lock();
        return { ok: true };
      case "IS_UNLOCKED":
        return { ok: true, data: vault.isUnlocked() };
      case "HAS_VAULT":
        return { ok: true, data: await vault.hasVault() };
      case "GET_ENTRIES":
        return { ok: true, data: vault.getEntries() };
      case "ADD_ENTRY": {
        const entry = await vault.addEntry(msg.payload as any);
        return { ok: true, data: entry };
      }
      case "BATCH_ADD_ENTRIES": {
        const { entries } = msg.payload as { entries: any[] };
        const added = await vault.batchAddEntries(entries);
        return { ok: true, data: added };
      }
      case "RESTORE_ENTRIES": {
        const { entries, mode, settings } = msg.payload as {
          entries: any[];
          mode: "replace" | "merge";
          settings?: vault.VaultSettings;
        };
        const summary = await vault.restoreEntries(entries, mode);
        if (settings && typeof settings === "object") {
          await vault.updateSettings(settings);
        }
        return { ok: true, data: summary };
      }
      case "CLEAR_VAULT_ENTRIES": {
        await vault.clearVaultEntries("REPLACE_ALL");
        return { ok: true };
      }
      case "LIST_VAULT_SNAPSHOTS": {
        const snapshots = await vault.listSnapshots();
        return { ok: true, data: snapshots };
      }
      case "RESTORE_VAULT_SNAPSHOT": {
        const { index, password } = msg.payload as { index: number; password: string };
        const ok = await vault.restoreFromSnapshot(index, password);
        if (ok) broadcastToContentScripts({ type: "VAULT_UNLOCKED" });
        return { ok };
      }
      case "COUNT_DUPLICATES": {
        return { ok: true, data: vault.countDuplicates() };
      }
      case "DEDUPLICATE_ENTRIES": {
        const removed = await vault.deduplicateEntries();
        return { ok: true, data: removed };
      }
      case "UPDATE_ENTRY": {
        const { id, patch } = msg.payload as any;
        await vault.updateEntry(id, patch);
        return { ok: true };
      }
      case "DELETE_ENTRY": {
        const { id } = msg.payload as { id: string };
        await vault.deleteEntry(id);
        return { ok: true };
      }
      case "GET_DELETED_ENTRIES":
        return { ok: true, data: vault.getDeletedEntries() };
      case "RESTORE_ENTRY": {
        const { id } = msg.payload as { id: string };
        await vault.restoreEntry(id);
        return { ok: true };
      }
      case "PURGE_ENTRY": {
        const { id } = msg.payload as { id: string };
        await vault.purgeEntry(id);
        return { ok: true };
      }
      case "PURGE_EXPIRED_TRASH": {
        const { retentionMs } = msg.payload as { retentionMs: number };
        const purged = await vault.purgeExpiredTrash(retentionMs);
        return { ok: true, data: purged };
      }
      case "AUTOFILL_QUERY": {
        if (!vault.isUnlocked()) {
          return { ok: false, error: "VAULT_LOCKED" };
        }
        const { domain, includeAll } = msg.payload as {
          domain: string;
          includeAll?: boolean;
        };
        const matches = includeAll
          ? vault
              .getEntries()
              .filter((entry) => entry.type !== "note" && entry.type !== "passkey")
          : vault.queryDomain(domain);
        return {
          ok: true,
          data: matches.map((e) =>
            includeAll
              ? {
                  id: e.id,
                  title: e.title,
                }
              : {
                  id: e.id,
                  title: e.title,
                  username: e.username,
                  url: e.url,
                  hasTotp: Boolean(e.totp),
                }
          ),
        };
      }
      case "GET_ENTRY_FOR_FILL": {
        if (!vault.isUnlocked()) return { ok: false, error: "vault locked" };
        const { id } = msg.payload as { id: string };
        const entry = vault.getEntryById(id);
        if (!entry) return { ok: false, error: "entry not found" };
        if (entry.type === "passkey") return { ok: false, error: "passkey not fillable" };
        // Content scripts never need the passkey private key; strip it
        // defensively even though login entries do not carry one.
        const { privateKeyJwk: _stripped, ...safeEntry } = entry;
        return { ok: true, data: safeEntry };
      }
      case "COPY": {
        const { text, label } = msg.payload as { text: string; label: string };
        await copyToClipboard(text, label);
        return { ok: true };
      }
      case "CHECK_BREACH": {
        const { password } = msg.payload as { password: string };
        const count = await vault.checkBreach(password);
        return { ok: true, data: count };
      }
      case "CHECK_ALL_BREACHES": {
        const allEntries = vault.getEntries();
        const results: Record<string, number> = {};
        for (const entry of allEntries) {
          if (entry.type === "note") continue;
          if (!entry.password) continue;
          results[entry.id] = await vault.checkBreach(entry.password);
          // Small delay — be a good citizen to the HIBP API
          await new Promise((r) => setTimeout(r, 200));
        }
        // Persist any non-zero counts back to entries
        for (const [id, count] of Object.entries(results)) {
          const entry = vault.getEntryById(id);
          if (entry && (entry.breach ?? 0) !== count) {
            await vault.updateEntry(id, { breach: count > 0 ? count : undefined });
          }
        }
        return { ok: true, data: results };
      }
      case "GET_SETTINGS":
        return { ok: true, data: await vault.ensureSettingsLoaded() };
      case "UPDATE_SETTINGS": {
        const patch = msg.payload as Record<string, unknown>;
        await vault.updateSettings(patch as any);
        if (typeof patch.autoLockMinutes === "number") {
          await setupAutoLock();
          if (vault.isUnlocked()) touchAutoLock();
        }
        return { ok: true };
      }
      case "GET_PENDING_PASSKEY":
        return { ok: true, data: await getPendingPasskeyRequest() };
      case "APPROVE_PASSKEY": {
        const { requestId } = msg.payload as { requestId: number };
        await resolvePendingPasskeyRequest(requestId, true);
        return { ok: true };
      }
      case "DENY_PASSKEY": {
        const { requestId } = msg.payload as { requestId: number };
        await resolvePendingPasskeyRequest(requestId, false);
        return { ok: true };
      }
      case "GET_PASSKEYS":
        return {
          ok: true,
          data: vault.getEntries().filter((entry) => entry.type === "passkey"),
        };
      case "DELETE_PASSKEY": {
        const { id } = msg.payload as { id: string };
        await vault.deleteEntry(id);
        return { ok: true };
      }
      case "GET_PASSKEY_PROVIDER_STATE":
        return { ok: true, data: await getPasskeyProviderState() };
      case "SET_PASSKEY_PROVIDER_ENABLED": {
        const { enabled } = msg.payload as { enabled: boolean };
        return { ok: true, data: await setPasskeyProviderEnabled(enabled) };
      }
      case "FIRE_CANARY_PING": {
        const { webhookUrl, entryTitle, trigger } = msg.payload as {
          webhookUrl: string;
          entryTitle: string;
          trigger: "password_copied" | "username_copied" | "autofilled" | "submitted" | "test";
        };
        if (!isAllowedCanaryWebhookUrl(webhookUrl)) {
          return { ok: false, error: "invalid webhook url" };
        }
        const body = JSON.stringify({
          event: trigger === "test" ? "peach_canary_test" : "peach_canary_accessed",
          entry: entryTitle,
          trigger,
          timestamp: new Date().toISOString(),
          source: "peach-extension",
        });
        if (trigger === "test") {
          // Test pings are awaited so the UI can show success/failure
          try {
            const res = await fetch(webhookUrl, {
              method: "POST",
              headers: { "Content-Type": "application/json" },
              body,
            });
            return { ok: res.ok, error: res.ok ? undefined : `HTTP ${res.status}` };
          } catch (err) {
            return { ok: false, error: err instanceof Error ? err.message : "network error" };
          }
        }
        // Non-test triggers are fire-and-forget
        fetch(webhookUrl, {
          method: "POST",
          headers: { "Content-Type": "application/json" },
          body,
        }).catch(() => {});
        return { ok: true };
      }
      case "ACTIVATE_LICENSE": {
        const { key } = msg.payload as { key: string };
        const licenseInfo = await activateLicense(key);
        return { ok: true, data: licenseInfo };
      }
      case "GET_LICENSE": {
        const licenseInfo = await getLicenseInfo();
        return { ok: true, data: licenseInfo };
      }
      case "IS_DURESS_ENABLED": {
        const enabled = await vault.isDuressEnabled();
        return { ok: true, data: enabled };
      }
      case "SETUP_DURESS": {
        const { duressPassword } = msg.payload as { duressPassword: string };
        await vault.setupDuress(duressPassword);
        return { ok: true };
      }
      case "DISABLE_DURESS": {
        await vault.disableDuress();
        return { ok: true };
      }
      case "VERIFY_MAIN_PASSWORD": {
        const { password } = msg.payload as { password: string };
        const ok = await vault.verifyMainPassword(password);
        return { ok: true, data: ok };
      }
      case "CHANGE_PASSWORD": {
        const { currentPassword, newPassword } = msg.payload as {
          currentPassword: string;
          newPassword: string;
        };
        const changed = await vault.changePassword(currentPassword, newPassword);
        return changed
          ? { ok: true }
          : { ok: false, error: "incorrect current password" };
      }
      case "GET_SITE_PASSWORD_POLICY": {
        const [tab] = await chrome.tabs.query({ active: true, currentWindow: true });
        if (!tab?.id) return { ok: true, data: null };
        try {
          const response = await chrome.tabs.sendMessage(tab.id, {
            type: "GET_SITE_PASSWORD_POLICY",
          });
          return (response as MessageResponse) ?? { ok: true, data: null };
        } catch {
          return { ok: true, data: null };
        }
      }
      case "GET_SYNC_STATE": {
        const { getPairedDevices, getTrustedNetworks } = await import('./sync-storage');
        const { isOnTailscale } = await import('./sync-network');
        const { isProUser } = await import('./license');
        const pairedDevices = await getPairedDevices();
        const trustedNetworks = await getTrustedNetworks();
        const tailscale = await isOnTailscale();
        const pro = await isProUser();
        
        let status: SyncStatus = "offline";
        if (tailscale || trustedNetworks.some((n: { isTailscale?: boolean }) => !n.isTailscale)) {
          if (pairedDevices.length === 0) {
            status = "no_peers";
          } else {
            status = "synced";
          }
        } else {
          status = "not_trusted";
        }
        
        return { ok: true, data: { pairedDevices, trustedNetworks, status, isPro: pro } };
      }
      case "TRIGGER_SYNC": {
        return { ok: true };
      }
      case "START_PAIRING": {
        return { ok: true };
      }
      case "CANCEL_PAIRING": {
        return { ok: true };
      }
      case "ACCEPT_PAIRING": {
        return { ok: true };
      }
      case "REMOVE_PAIRED_DEVICE": {
        const { deviceId } = msg.payload as { deviceId: string };
        const { removePairedDevice } = await import('./sync-storage');
        await removePairedDevice(deviceId);
        return { ok: true };
      }
      case "REMOVE_TRUSTED_NETWORK": {
        const { ssid, bssid, isTailscale } = msg.payload as any;
        const { removeTrustedNetwork, getTrustedNetworks } = await import('./sync-storage');
        const networks = await getTrustedNetworks();
        const network = networks.find((n: { ssid?: string; bssid?: string; isTailscale?: boolean }) => 
          (isTailscale ? n.isTailscale : n.ssid === ssid) && 
          (!bssid || n.bssid === bssid)
        );
        if (network) {
          await removeTrustedNetwork(network.addedAt);
        }
        return { ok: true };
      }
      case "MARK_NETWORK_TRUSTED": {
        const { addTrustedNetwork } = await import('./sync-storage');
        const { getCurrentNetwork } = await import('./sync-network');
        const network = await getCurrentNetwork();
        await addTrustedNetwork({
          ssid: network.ssid,
          bssid: network.bssid,
          isTailscale: network.isTailscale,
          addedAt: Date.now(),
        });
        return { ok: true };
      }
      default:
        return { ok: false, error: "unknown message type" };
    }
  } catch (err) {
    return {
      ok: false,
      error: err instanceof Error ? err.message : "operation failed",
    };
  }
}

/** Notify all content scripts that the vault just became available. */
function broadcastToContentScripts(message: { type: string }): void {
  chrome.tabs.query({}, (tabs) => {
    for (const tab of tabs) {
      if (tab.id !== undefined) {
        chrome.tabs.sendMessage(tab.id, message).catch(() => {
          // Tab may not have a content script — silently ignore
        });
      }
    }
  });
}

function shouldRefreshAutoLock(type: MessageType): boolean {
  return (
    type !== "HAS_VAULT" &&
    type !== "IS_UNLOCKED" &&
    type !== "LOCK" &&
    type !== "UPDATE_SETTINGS" &&
    type !== "GET_PENDING_PASSKEY" &&
    type !== "GET_PASSKEY_PROVIDER_STATE"
  );
}

function isAllowedCanaryWebhookUrl(value: string): boolean {
  try {
    const url = new URL(value);
    if (url.protocol !== "https:") return false;

    const hostname = url.hostname.toLowerCase();
    if (
      hostname === "localhost" ||
      hostname.endsWith(".localhost") ||
      isPrivateIpv4(hostname) ||
      isPrivateIpv6(hostname)
    ) {
      return false;
    }

    return true;
  } catch {
    return false;
  }
}

function isPrivateIpv4(hostname: string): boolean {
  if (PRIVATE_IPV4_RANGES.some((range) => hostname.startsWith(range.prefix))) {
    return true;
  }

  const octets = hostname.split(".").map((part) => Number(part));
  if (octets.length !== 4 || octets.some((octet) => !Number.isInteger(octet) || octet < 0 || octet > 255)) {
    return false;
  }

  return octets[0] === 172 && octets[1] >= 16 && octets[1] <= 31;
}

function isPrivateIpv6(hostname: string): boolean {
  const normalized = hostname.replace(/^\[|\]$/g, "").toLowerCase();
  return (
    normalized === "::1" ||
    normalized.startsWith("fc") ||
    normalized.startsWith("fd") ||
    normalized.startsWith("fe80:")
  );
}