1import type { Adapter } from "../types";
2import { collectMediaFromDocument } from "../../media/markdown-media";
3import { extractYouTubeTranscriptDocument } from "./transcript";
4import { isYouTubeHost, parseYouTubeVideoId } from "./utils";
5 
6export const youtubeAdapter: Adapter = {
7  name: "youtube",
8  match(input) {
9    return isYouTubeHost(input.url.hostname);
10  },
11  async process(context) {
12    const videoId = parseYouTubeVideoId(context.input.url);
13    if (!videoId) {
14      return {
15        status: "no_document",
16      };
17    }
18 
19    context.log.info(`Loading ${context.input.url.toString()} with youtube adapter`);
20    const document = await extractYouTubeTranscriptDocument(context, videoId);
21    if (!document) {
22      return {
23        status: "no_document",
24      };
25    }
26 
27    return {
28      status: "ok",
29      document,
30      media: collectMediaFromDocument(document),
31    };
32  },
33};
34