import type { Pipeline, PipelineStageError } from "corvos";
import type { StandardSchemaV1, StandardTypedV1 } from "corvos/deps/standard_schema.ts";

import { Page } from "corvos";
import { replaceExtension } from "corvos/core/utils/path.ts";

import type { Node } from "./strapi/schema.ts";

import { blocksToTSX } from "./strapi/blocks.tsx";
import { StrapiLoader } from "./strapi/loader.ts";

export type { Node } from "./strapi/schema.ts";
export { nodeSchema, fileAttributeSchema, imageAttributeSchema } from "./strapi/schema.ts";
export { Blocks } from "./strapi/blocks.tsx";

export interface StrapiContent<Type extends "single" | "collection" = "single" | "collection"> {
	type: Type;
	schema: StandardSchemaV1;
	query?: Record<string, string>;
}

type ContentSingle<T extends Record<string, StrapiContent>> = {
	[Key in keyof T]: T[Key]["type"] extends "single" ? Key : never;
}[keyof T];

type ContentCollection<T extends Record<string, StrapiContent>> = {
	[Key in keyof T]: T[Key]["type"] extends "collection" ? Key : never;
}[keyof T];

export interface Options<Content extends Record<string, StrapiContent>> {
	url: string;
	token: string;
	content: Content;
}

export function strapi<Content extends Record<string, StrapiContent>>(options: Options<Content>) {
	const loader = new StrapiLoader(options.url, options.token);

	function load<const Key extends ContentSingle<Content> & string>(
		key: Key,
		query?: Record<string, string>,
	): Promise<
		{
			documentId: string;
			publishedAt: Date | null;
			updatedAt: Date;
		} & StandardTypedV1.InferOutput<Content[Key]["schema"]>
	>;
	function load<const Key extends ContentCollection<Content> & string>(
		key: Key,
		query?: Record<string, string>,
	): AsyncGenerator<
		| PipelineStageError
		| ({
				documentId: string;
				publishedAt: Date | null;
				updatedAt: Date;
		  } & StandardTypedV1.InferOutput<Content[Key]["schema"]>),
		void,
		unknown
	>;
	function load<const Key extends Content & string>(
		key: Key,
		query?: Record<string, string>,
	):
		| Promise<
				{
					documentId: string;
					publishedAt: Date | null;
					updatedAt: Date;
				} & StandardTypedV1.InferOutput<Content[Key]["schema"]>
		  >
		| AsyncGenerator<
				| PipelineStageError
				| ({
						documentId: string;
						publishedAt: Date | null;
						updatedAt: Date;
				  } & StandardTypedV1.InferOutput<Content[Key]["schema"]>),
				void,
				unknown
		  > {
		if (options.content[key]!.type === "single") {
			return loader.loadSingle(key, options.content[key]!.schema, {
				...options.content[key]!.query,
				...query,
			});
		} else {
			return loader.loadCollection(key, options.content[key]!.schema, {
				...options.content[key]!.query,
				...query,
			});
		}
	}

	return <Loaders extends string, Out extends { ".strapi": [Node[], {}] }, Helpers extends {}>(
		pipeline: Pipeline<Loaders, Out, Helpers>,
	) =>
		pipeline
			.process("strapi", ".strapi", function* (page) {
				yield new Page(
					".jsxelement",
					blocksToTSX(page.content),
					page.data as Out[".strapi"][1],
					replaceExtension(page.url, ".jsx"),
					page,
				);
			})
			.helper("strapi", load);
}