import { resolveViewUrl } from "./document";

const STALE_OFFSET_HOURS = 24;

export async function ingestDocument(
	db: D1Database,
	queue: Queue,
	params: {
		did: string;
		rkey: string;
		collection: string;
		cid?: string;
		record?: Record<string, unknown>;
	},
): Promise<void> {
	const { did, rkey, collection, cid, record } = params;

	// Upsert repo_records
	await db
		.prepare(
			`INSERT INTO repo_records (did, rkey, collection, cid, synced_at)
       VALUES (?, ?, ?, ?, datetime('now'))
       ON CONFLICT(did, collection, rkey) DO UPDATE SET
         cid = ?,
         synced_at = datetime('now')`,
		)
		.bind(did, rkey, collection, cid || null, cid || null)
		.run();

	// If we have the full record, upsert resolved_documents with initial data
	if (record) {
		const uri = `at://${did}/${collection}/${rkey}`;
		const doc = record as {
			title?: string;
			path?: string;
			site?: string;
			content?: unknown;
			textContent?: string;
			publishedAt?: string;
			coverImage?: unknown;
			bskyPostRef?: { uri: string; cid: string };
			tags?: string[];
		};

		let viewUrl: string | null = null;
		if (doc.site && doc.path) {
			viewUrl = await resolveViewUrl(db, doc.site, doc.path);
		}

		await db
			.prepare(
				`INSERT INTO resolved_documents (uri, did, rkey, title, path, site, content, text_content, published_at, view_url, resolved_at, stale_at)
         VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, datetime('now'), datetime('now', '+${STALE_OFFSET_HOURS} hours'))
         ON CONFLICT(uri) DO UPDATE SET
           title = ?, path = ?, site = ?, content = ?, text_content = ?, published_at = ?, view_url = ?, resolved_at = datetime('now'), stale_at = datetime('now', '+${STALE_OFFSET_HOURS} hours')`,
			)
			.bind(
				uri,
				did,
				rkey,
				doc.title || null,
				doc.path || null,
				doc.site || null,
				doc.content ? JSON.stringify(doc.content) : null,
				doc.textContent || null,
				doc.publishedAt || null,
				viewUrl,
				doc.title || null,
				doc.path || null,
				doc.site || null,
				doc.content ? JSON.stringify(doc.content) : null,
				doc.textContent || null,
				doc.publishedAt || null,
				viewUrl,
			)
			.run();
	}

	// Queue for full resolution (verification, publication lookup, etc.)
	await queue.send({ did, collection, rkey });
}

export async function deleteDocument(
	db: D1Database,
	params: { did: string; collection: string; rkey: string },
): Promise<void> {
	const { did, collection, rkey } = params;

	await db
		.prepare(
			"DELETE FROM repo_records WHERE did = ? AND collection = ? AND rkey = ?",
		)
		.bind(did, collection, rkey)
		.run();

	const uri = `at://${did}/${collection}/${rkey}`;
	await db
		.prepare("DELETE FROM resolved_documents WHERE uri = ?")
		.bind(uri)
		.run();
}