import { describe, expect, it, beforeEach, vi } from "vitest";
import type { PairedDevice } from "../shared/sync-types";
import { upsertPairedDevice, findPairedDeviceById } from "./paired-devices";
import {
  triggerSyncIfReachable,
  setSyncRunner,
  _resetSyncTriggerForTest,
  getSyncState,
  hydratePersistedSyncState,
} from "./sync-trigger";

function makeStorageStub() {
  const sync = new Map<string, unknown>();
  const local = new Map<string, unknown>();
  function area(store: Map<string, unknown>) {
    return {
      async get(key?: string | string[] | null) {
        if (key === null || key === undefined) {
          const all: Record<string, unknown> = {};
          for (const [k, v] of store) all[k] = v;
          return all;
        }
        const keys = Array.isArray(key) ? key : [key];
        const out: Record<string, unknown> = {};
        for (const k of keys) if (store.has(k)) out[k] = store.get(k);
        return out;
      },
      async set(values: Record<string, unknown>) {
        for (const [k, v] of Object.entries(values)) store.set(k, v);
      },
      async remove(key: string | string[]) {
        const keys = Array.isArray(key) ? key : [key];
        for (const k of keys) store.delete(k);
      },
    };
  }
  (globalThis as unknown as { chrome: unknown }).chrome = {
    storage: { sync: area(sync), local: area(local) },
    runtime: { sendMessage: () => Promise.resolve() },
  };
}

const ANDROID: PairedDevice = {
  deviceId: "AAA-android",
  deviceName: "Pixel",
  deviceType: "android",
  longTermPublicKeyHex: "a".repeat(64),
  trustSource: "direct",
  pairedAt: 1700000000000,
};

const IOS: PairedDevice = {
  deviceId: "BBB-ios",
  deviceName: "iPhone",
  deviceType: "ios",
  longTermPublicKeyHex: "b".repeat(64),
  trustSource: "direct",
  pairedAt: 1700000000000,
};

const EXT: PairedDevice = {
  deviceId: "CCC-ext",
  deviceName: "Other Chrome",
  deviceType: "extension",
  longTermPublicKeyHex: "c".repeat(64),
  pairedAt: 1700000000000,
};

describe("triggerSyncIfReachable", () => {
  beforeEach(() => {
    makeStorageStub();
    _resetSyncTriggerForTest();
  });

  it("does not call the runner when no mobile hosts are paired", async () => {
    await upsertPairedDevice(EXT);
    const runner = vi.fn(async () => ({ ok: true as const, entriesReceived: 0, entriesSent: 0 }));
    setSyncRunner(runner);
    await triggerSyncIfReachable({ reason: "manual" });
    expect(runner).not.toHaveBeenCalled();
  });

  it("calls the runner once per mobile host on a single trigger", async () => {
    await upsertPairedDevice(ANDROID);
    await upsertPairedDevice(IOS);
    await upsertPairedDevice(EXT);
    const seen: string[] = [];
    setSyncRunner(async (device) => {
      seen.push(device.deviceId);
      return { ok: true, entriesReceived: 1, entriesSent: 0 };
    });
    await triggerSyncIfReachable({ reason: "manual" });
    expect(seen.sort()).toEqual([ANDROID.deviceId, IOS.deviceId]);
  });

  it("marks lastSyncedAt on each host that completes successfully", async () => {
    await upsertPairedDevice(ANDROID);
    await upsertPairedDevice(IOS);
    setSyncRunner(async () => ({ ok: true, entriesReceived: 0, entriesSent: 0 }));
    await triggerSyncIfReachable({ reason: "manual" });
    const a = await findPairedDeviceById(ANDROID.deviceId);
    const b = await findPairedDeviceById(IOS.deviceId);
    expect(typeof a?.lastSyncedAt).toBe("number");
    expect(typeof b?.lastSyncedAt).toBe("number");
  });

  it("does not start a second sync if one is already in flight (non-force)", async () => {
    await upsertPairedDevice(ANDROID);
    const runner = vi.fn(async () => {
      await new Promise<void>((r) => setTimeout(r, 50));
      return { ok: true as const, entriesReceived: 0, entriesSent: 0 };
    });
    setSyncRunner(runner);
    const a = triggerSyncIfReachable({ reason: "vault_change" });
    const b = triggerSyncIfReachable({ reason: "vault_change" });
    await Promise.all([a, b]);
    expect(runner).toHaveBeenCalledTimes(1);
  });

  it("force:true bypasses the syncInFlight guard for manual triggers", async () => {
    await upsertPairedDevice(ANDROID);
    const runner = vi.fn(async () => {
      await new Promise<void>((r) => setTimeout(r, 30));
      return { ok: true as const, entriesReceived: 0, entriesSent: 0 };
    });
    setSyncRunner(runner);
    const a = triggerSyncIfReachable({ reason: "vault_change" });
    const b = triggerSyncIfReachable({ reason: "manual", force: true });
    await Promise.all([a, b]);
    expect(runner).toHaveBeenCalledTimes(2);
  });

  it("transitions to error state when no host succeeds", async () => {
    await upsertPairedDevice(ANDROID);
    setSyncRunner(async () => {
      throw new Error("unreachable");
    });
    await triggerSyncIfReachable({ reason: "manual" });
    expect(getSyncState().status).toBe("error");
    expect(getSyncState().error).toBe("unreachable");
  });

  it("returns to idle state on success with aggregated entry counts", async () => {
    await upsertPairedDevice(ANDROID);
    await upsertPairedDevice(IOS);
    setSyncRunner(async () => ({ ok: true, entriesReceived: 3, entriesSent: 2 }));
    await triggerSyncIfReachable({ reason: "manual" });
    const s = getSyncState();
    expect(s.status).toBe("idle");
    expect(s.entriesReceived).toBe(6);
    expect(s.entriesSent).toBe(4);
    expect(typeof s.lastSyncAt).toBe("number");
  });

  it("syncInFlight is cleared in finally even when runner throws", async () => {
    await upsertPairedDevice(ANDROID);
    setSyncRunner(async () => { throw new Error("runner exploded"); });
    await triggerSyncIfReachable({ reason: "manual", force: true });
    // After the trigger resolves, a second non-force trigger must be able to run.
    let ran = false;
    setSyncRunner(async () => { ran = true; return { ok: true as const, entriesReceived: 0, entriesSent: 0 }; });
    await triggerSyncIfReachable({ reason: "vault_change" });
    expect(ran).toBe(true);
  });

  it("does not call runner when only vouched devices are in registry", async () => {
    const vouched: PairedDevice = {
      ...ANDROID,
      deviceId: "VVV-vouched",
      trustSource: "vouched",
    };
    await upsertPairedDevice(vouched);
    let ran = false;
    setSyncRunner(async () => { ran = true; return { ok: true as const, entriesReceived: 0, entriesSent: 0 }; });
    await triggerSyncIfReachable({ reason: "foreground" });
    expect(ran).toBe(false);
  });

  it("persists peach_last_sync_result to storage after a successful sync", async () => {
    await upsertPairedDevice(ANDROID);
    setSyncRunner(async () => ({ ok: true, entriesReceived: 3, entriesSent: 1 }));
    await triggerSyncIfReachable({ reason: "manual" });
    // Read the raw storage value to confirm persistence.
    const raw = await (globalThis as unknown as { chrome: { storage: { local: { get(k: string): Promise<Record<string, unknown>> } } } })
      .chrome.storage.local.get("peach_last_sync_result");
    expect(raw.peach_last_sync_result).toEqual({ entriesReceived: 3, entriesSent: 1 });
  });

  it("hydratePersistedSyncState restores lastSyncAt across SW restart simulation", async () => {
    // Write a value directly to storage, then hydrate — simulates SW restart.
    await (globalThis as unknown as { chrome: { storage: { local: { set(v: Record<string, unknown>): Promise<void> } } } })
      .chrome.storage.local.set({ peach_last_sync_at: 1_700_005_000_000 });
    _resetSyncTriggerForTest();
    expect(getSyncState().lastSyncAt).toBeUndefined();
    await hydratePersistedSyncState();
    expect(getSyncState().lastSyncAt).toBe(1_700_005_000_000);
  });
});