import { ok } from "@atcute/client";
import type { Cid, ResourceUri } from "@atcute/lexicons/syntax";
import { ClientResponseError, type BobbinContext } from "$lib/api/client";
import { orNull } from "$lib/api/load";
import { mainSchema as awaitRecordSchema } from "$lib/api/lexicons/types/sh/tangled/bobbin/awaitRecord";

export interface RecordRevision {
	uri: ResourceUri;
	cid?: Cid;
}

export const awaitRecord = async (ctx: BobbinContext, revision: RecordRevision) => {
	return ok(
		ctx.xrpc.call(awaitRecordSchema, {
			params: {
				uri: revision.uri,
				...(revision.cid ? { cid: revision.cid } : {})
			}
		})
	);
};

export const awaitIndexedRecord = async (
	ctx: BobbinContext,
	revision: RecordRevision
): Promise<void> => {
	const result = await awaitRecord(ctx, revision);
	if (result.status !== "indexed") {
		throw new Error(result.detail || result.reason || "Bobbin rejected the record.");
	}
};

export const awaitIndexedOrSaved = async (
	ctx: BobbinContext,
	revision: { uri: string; cid?: string }
): Promise<void> => {
	try {
		await awaitIndexedRecord(ctx, revision as RecordRevision);
	} catch (cause) {
		if (cause instanceof ClientResponseError) {
			throw new Error(`Saved, but the appview has not caught up yet (${cause.message}).`, {
				cause
			});
		}
		throw cause;
	}
};

const DELETION_BUDGET_MS = 5000;
const DELETION_POLL_MS = 150;

export const awaitDeletedRecord = async (
	ctx: BobbinContext,
	uri: ResourceUri,
	{ budgetMs = DELETION_BUDGET_MS, pollMs = DELETION_POLL_MS } = {}
): Promise<boolean> => {
	const deadline = Date.now() + budgetMs;
	for (;;) {
		const result = await orNull(awaitRecord(ctx, { uri }));
		if (result === null) return false;
		if (result.status !== "indexed") return true;
		if (Date.now() >= deadline) return false;
		await new Promise((done) => setTimeout(done, pollMs));
	}
};