import { z } from "zod";
const YOUTUBE_URL_REGEX =
/^https?:\/\/(?:www\.)?(?:youtube\.com\/(?:watch\?v=|shorts\/)|youtu\.be\/)[\w-]+/;
export const YouTubeUrlSchema = z
.string()
.trim()
.refine((url) => YOUTUBE_URL_REGEX.test(url), {
message:
"Invalid YouTube URL. Expected format: youtube.com/watch?v=ID, youtu.be/ID, or youtube.com/shorts/ID",
});
export const DetailLevelSchema = z
.enum(["brief", "medium", "detailed"])
.default("medium");
export const SummarizeInputSchema = z.object({
youtube_url: YouTubeUrlSchema,
detail_level: DetailLevelSchema,
});
export const AskInputSchema = z.object({
youtube_url: YouTubeUrlSchema,
question: z.string().min(1, "Question cannot be empty"),
});
export type SummarizeInput = z.infer<typeof SummarizeInputSchema>;
export type AskInput = z.infer<typeof AskInputSchema>;
export type DetailLevel = z.infer<typeof DetailLevelSchema>;
export function validateYouTubeUrl(url: string): string {
return YouTubeUrlSchema.parse(url);
}
export function extractVideoId(url: string): string {
const validUrl = validateYouTubeUrl(url);
const urlObj = new URL(validUrl);
if (urlObj.hostname === "youtu.be") {
return urlObj.pathname.slice(1);
}
if (
urlObj.hostname === "www.youtube.com" ||
urlObj.hostname === "youtube.com"
) {
if (urlObj.pathname === "/watch") {
const videoId = urlObj.searchParams.get("v");
if (videoId) return videoId;
}
if (urlObj.pathname.startsWith("/shorts/")) {
return urlObj.pathname.split("/")[2];
}
}
throw new Error(`Cannot extract video ID from: ${url}`);
}
