import type { Loader, LoaderContext } from "astro/loaders";
import { VideoSnippetSchema } from "./schema";
import { google } from "googleapis";

export function youtubeLoader({
  channelId,
  apiKey,
  maxResults = 5,
}: {
  channelId: string;
  apiKey: string;
  maxResults?: number;
}): Loader {
  return {
    name: "youtube-loader",
    load: async (context: LoaderContext): Promise<void> => {
      const youtube = google.youtube("v3");
      const mostRecentVideoId = context.meta.get("mostRecentVideoId") || 0;
      let first;

      const params = {
        key: apiKey,
        channelId: channelId,
        order: "date",
        part: ["snippet"],
        type: ["video"],
        maxResults: maxResults,
      };

      const result = await youtube.search.list(params);

      if (result?.data?.items === undefined) {
        context.logger.error("Could not find any videos on YouTube.");
        return;
      }

      for (const video of result?.data?.items) {
        if (video?.id?.kind === "youtube#video" && video?.id?.videoId) {
          if (mostRecentVideoId && mostRecentVideoId === video.id.videoId) {
            break;
          }

          if (!first) {
            first = video.id.videoId;
          }

          const data = await context.parseData({
            id: video.id.videoId,
            data: JSON.parse(JSON.stringify(video.snippet)),
          });

          const digest = context.generateDigest(data);

          context.store.set({
            id: video.id.videoId,
            data,
            digest,
            rendered: {
              html: video.id.videoId,
            },
          });
        }
      }

      if (first) {
        context.meta.set("mostRecentVideoId", first);
      }

      context.logger.info(`Fetched ${maxResults} videos from YouTube.`);
    },
    schema: VideoSnippetSchema,
  };
}