/**
 * High-level sync flows over a RelayPool, mirroring src/nostr/snapshot.py:
 * publish a chunked snapshot + manifest, fetch the latest snapshot with
 * chunk-hash verification, publish and replay deltas.
 */

import { sha256 } from "@noble/hashes/sha2.js";
import { base64 } from "@scure/base";
import { bytesToHex } from "../util/bytes.js";
import {
  buildChunkEvent,
  buildDeltaEvent,
  buildManifestEvent,
  signEvent,
  signerPublicKeyHex,
  type Filter,
  type NostrEvent,
} from "./events.js";
import {
  KIND_DELTA,
  KIND_MANIFEST,
  KIND_SNAPSHOT_CHUNK,
  newManifestId,
  manifestOrderMs,
  parseManifest,
  prepareSnapshot,
  reassembleSnapshot,
  type Manifest,
} from "./snapshot.js";
import type { RelayPool } from "./relay.js";

function nowUnix(): number {
  return Math.floor(Date.now() / 1000);
}

/**
 * Strictly increasing publication timestamp in milliseconds.
 *
 * A wall-clock millisecond is not enough on its own: two snapshots published
 * back to back can land in the same millisecond, and then the ordering falls
 * through to the event-id tie-break — arbitrary with respect to time, which
 * is the whole bug this field exists to fix, just in a narrower window.
 * Forcing each value above the last one this process produced makes it a
 * monotonic sequence that also happens to be a timestamp.
 *
 * Across processes or machines the clock is still the only shared reference,
 * so this narrows the tie window rather than closing it everywhere. A vault
 * synced from two machines in the same millisecond remains ordered by event
 * id, and is unchanged by this.
 */
let lastPublishedMs = 0;
function nextPublishedMs(): number {
  lastPublishedMs = Math.max(Date.now(), lastPublishedMs + 1);
  return lastPublishedMs;
}

export interface PublishedSnapshot {
  manifest: Manifest;
  manifestId: string;
  manifestEventId: string;
  chunkEventIds: string[];
}

/** Compress, chunk, sign, and publish a snapshot plus its manifest. */
export async function publishSnapshot(
  pool: RelayPool,
  privateKeyHex: string,
  keyIndex: Uint8Array,
  encryptedBytes: Uint8Array,
  options: { limit?: number; createdAt?: number; publishedMs?: number } = {},
): Promise<PublishedSnapshot> {
  const createdAt = options.createdAt ?? nowUnix();
  // Sub-second publication time, signed as part of the manifest. Two syncs in
  // the same second are otherwise indistinguishable by time, and the restore
  // side then has to guess which is newer.
  const publishedMs = options.publishedMs ?? nextPublishedMs();
  const { manifest, chunks } = await prepareSnapshot(encryptedBytes, options.limit ?? 50_000);

  const chunkEventIds: string[] = [];
  for (let i = 0; i < chunks.length; i++) {
    const meta = manifest.chunks[i]!;
    const event = signEvent(
      privateKeyHex,
      buildChunkEvent(meta.id, base64.encode(chunks[i]!), createdAt),
    );
    const results = await pool.publish(event);
    if (!results.some((r) => r.ok)) {
      throw new Error(`no relay accepted chunk ${meta.id}`);
    }
    meta.event_id = event.id;
    chunkEventIds.push(event.id);
  }

  const { id: manifestId, nonce } = newManifestId(keyIndex);
  manifest.nonce = base64.encode(nonce);
  manifest.delta_since = createdAt;
  manifest.published_ms = publishedMs;

  const manifestJson = JSON.stringify({
    ver: manifest.ver,
    algo: manifest.algo,
    chunks: manifest.chunks,
    delta_since: manifest.delta_since,
    nonce: manifest.nonce,
    index0: manifest.index0,
    published_ms: manifest.published_ms,
  });
  const manifestEvent = signEvent(
    privateKeyHex,
    buildManifestEvent(manifestId, manifestJson, createdAt),
  );
  const results = await pool.publish(manifestEvent);
  if (!results.some((r) => r.ok)) throw new Error("no relay accepted the manifest");

  return { manifest, manifestId, manifestEventId: manifestEvent.id, chunkEventIds };
}

export interface FetchedSnapshot {
  manifest: Manifest;
  manifestEvent: NostrEvent;
  /** Decompressed, still-encrypted vault payload. */
  encrypted: Uint8Array;
  /**
   * Manifests newer than the one returned that could not be assembled.
   * Non-empty means the client was walked backwards — usually because a
   * relay withheld a chunk — and the caller should say so out loud.
   */
  skippedNewerManifests: string[];
}

/** Fetch the newest manifest and all its chunks, verifying every hash. */
export async function fetchLatestSnapshot(
  pool: RelayPool,
  privateKeyHex: string,
): Promise<FetchedSnapshot | null> {
  const pubkey = signerPublicKeyHex(privateKeyHex);
  const manifests = await pool.fetch({ authors: [pubkey], kinds: [KIND_MANIFEST] });
  if (manifests.length === 0) return null;
  // Newest first. `created_at` alone is whole seconds, so two snapshots
  // published in the same second tie; the manifest's signed `published_ms`
  // breaks that tie by actual publication time rather than arbitrarily.
  // Getting this wrong is not a tie-break detail: the loser of the
  // comparison IS the vault the user gets back, so an arbitrary tie-break
  // silently restored a stale vault about half the time.
  //
  // Event id remains the final tie-break, for manifests that agree to the
  // millisecond. Arrival order is never used — it is relay-controlled, which
  // would let a relay choose the winner. Event id is a hash of the event's
  // own contents, so it is a stable total order no party can steer.
  const orderOf = new Map<string, number>();
  for (const ev of manifests) {
    let ms = ev.created_at * 1000;
    try {
      ms = manifestOrderMs(parseManifest(ev.content), ev.created_at);
    } catch {
      // Unparseable manifests are skipped further down; order them by
      // created_at so they do not jump the queue on the way there.
    }
    orderOf.set(ev.id, ms);
  }
  manifests.sort(
    (a, b) =>
      orderOf.get(b.id)! - orderOf.get(a.id)! ||
      (a.id < b.id ? 1 : a.id > b.id ? -1 : 0),
  );

  // Falling back to an older manifest is a downgrade: a relay that withholds
  // one chunk of the newest snapshot can walk a client backwards through its
  // history. Record what was skipped so callers can surface it.
  const skipped: string[] = [];
  for (const manifestEvent of manifests) {
    let manifest: Manifest;
    try {
      manifest = parseManifest(manifestEvent.content);
    } catch {
      skipped.push(manifestEvent.id);
      continue;
    }
    const chunks: Uint8Array[] = [];
    let complete = true;
    for (const meta of manifest.chunks) {
      const filter = meta.event_id
        ? { ids: [meta.event_id], authors: [pubkey], kinds: [KIND_SNAPSHOT_CHUNK] }
        : { authors: [pubkey], kinds: [KIND_SNAPSHOT_CHUNK], "#d": [meta.id], limit: 1 };
      const events = await pool.fetch(filter);
      // Decode defensively: base64.decode throws on malformed input, and a
      // relay can include any event it likes in the response. An unguarded
      // decode here let one extra frame abort the entire restore.
      const match = events.find((ev) => {
        try {
          return bytesToHex(sha256(base64.decode(ev.content))) === meta.hash;
        } catch {
          return false;
        }
      });
      if (!match) {
        complete = false;
        break;
      }
      chunks.push(base64.decode(match.content));
    }
    if (!complete) {
      skipped.push(manifestEvent.id);
      continue;
    }
    const encrypted = await reassembleSnapshot(manifest, chunks);
    return { manifest, manifestEvent, encrypted, skippedNewerManifests: skipped };
  }
  return null;
}

/** Publish an encrypted delta referencing the manifest event. */
export async function publishDelta(
  pool: RelayPool,
  privateKeyHex: string,
  manifestEventId: string,
  deltaBytes: Uint8Array,
  options: { createdAt?: number } = {},
): Promise<string> {
  const event = signEvent(
    privateKeyHex,
    buildDeltaEvent(manifestEventId, base64.encode(deltaBytes), options.createdAt ?? nowUnix()),
  );
  const results = await pool.publish(event);
  if (!results.some((r) => r.ok)) throw new Error("no relay accepted the delta");
  return event.id;
}

/**
 * Fetch deltas since a timestamp, sorted by created_at (replay order).
 *
 * `manifestEventId` binds the result to one snapshot lineage. Deltas carry
 * an `e` tag naming the manifest they extend; without checking it a relay
 * could replay a validly-signed delta from a different (older) lineage into
 * this restore. Callers restoring a specific snapshot must pass it.
 */
export async function fetchDeltasSince(
  pool: RelayPool,
  privateKeyHex: string,
  since: number,
  manifestEventId?: string,
): Promise<Uint8Array[]> {
  const pubkey = signerPublicKeyHex(privateKeyHex);
  const filter: Filter = { authors: [pubkey], kinds: [KIND_DELTA], since };
  if (manifestEventId) filter["#e"] = [manifestEventId];
  const events = await pool.fetch(filter);
  const bound = manifestEventId
    ? events.filter((ev) =>
        ev.tags.some((t) => t[0] === "e" && t[1] === manifestEventId),
      )
    : events;
  bound.sort((a, b) => a.created_at - b.created_at);
  return bound.map((ev) => base64.decode(ev.content));
}