import { merge } from "../core/utils.ts";
import type { Site } from "../core.ts";
export interface PaginateOptions { size: number;
url: (page: number) => string;}
export type Paginator = ( results: unknown[], userOptions?: Partial<PaginateOptions>,) => Generator<PaginateResult, void, unknown>;
export interface PaginateResult { url: string;
results: unknown[];
pagination: { page: number;
totalPages: number;
totalResults: number;
previous: string | null;
next: string | null; };}
export interface Options { name: string;
options: PaginateOptions;}
export const defaults: Options = { name: "paginate", options: { size: 10, url: () => "", },};
export default function (userOptions?: Partial<Options>) { const options = merge(defaults, userOptions);
return (site: Site) => { if (!userOptions?.options?.url) { const ext = site.options.prettyUrls ? "/index.html" : ".html"; options.options.url = (page: number) => `./page-${page}${ext}`; }
site.data(options.name, createPaginator(options.options)); };}
export function createPaginator(defaults: PaginateOptions): Paginator { return function* paginate( results: unknown[], userOptions: Partial<PaginateOptions> = {}, ) { const options = merge(defaults, userOptions); const totalResults = results.length; const totalPages = Math.ceil(results.length / options.size);
let page = 1; let data = createPageData(page);
for (const result of results) { data.results.push(result);
if (data.results.length >= options.size) { yield data;
data = createPageData(++page); } }
if (data.results.length) { yield data; }
function createPageData(page: number): PaginateResult { return { url: options.url(page), results: [], pagination: { page, totalPages, totalResults, previous: page > 1 ? options.url(page - 1) : null, next: totalPages > page ? options.url(page + 1) : null, }, }; } };}