technovangelist.videoprojects/2024-05-31-agentframeworks/videoSearchTool.ts

import * as readline from "readline/promises";

type VideoSearchResults = {
  id: string;
  title: string;
  channelId: string;
  channelTitle: string;
  daysSincePublished: number;
}

type VideoDetails = {
  videoString: string;
  title: string;
  viewCount: number;
  likeCount: number;
  commentCount: number;
  subscriberCount: number;
  score: number;
}

const getInput = async (question: string): Promise<string> => {
  let output = "";
  const rl = readline.createInterface({
    input: process.stdin,
    output: process.stdout,
  });
  output = await rl.question(question);
  rl.close();
  return output;
}

export const getTopic = async (): Promise<string> => {
  const topic = await getInput("Enter a topic to search for: ");
  return topic;
}

export const getDetails = async (topic: string): Promise<string> => {
  const details = await getInput(`Enter a description for the video on ${topic}: `);
  return details;
}

export const videoSearchTool = async (topic: string, maxResults: number = 35): Promise<VideoSearchResults[]> => {

  const results: VideoSearchResults[] = [];
  topic = encodeURIComponent(topic);
  const apiKey = Bun.env.YouTubeAPIKey;
  const url = `https://www.googleapis.com/youtube/v3/search?part=snippet&maxResults=${maxResults}&q=${topic}&type=video&key=${apiKey}`;

  const response = await fetch(url);
  const data = await response.json();

  if (data.items) {
    for (const item of data.items) {
      const id = item.id.videoId;
      const title = item.snippet.title;
      const channelId = item.snippet.channelId;
      const channelTitle = item.snippet.channelTitle;
      const daysSincePublished = Math.floor((new Date().getTime() - new Date(item.snippet.publishedAt.slice(0, 10)).getTime())/(1000*60*60*24)); 

      results.push({
        id,
        title,
        channelId,
        channelTitle,
        daysSincePublished,
      });
    }
  }

  return results;
}

export const videoDetailsTool = async (video: VideoSearchResults): Promise<VideoDetails> => {

}

  const apiKey = Bun.env.YouTubeAPIKey;
  const videoUrl = `https://www.googleapis.com/youtube/v3/videos?part=snippet,statistics&id=${video.id}&key=${apiKey}`;
  const channelUrl = `https://www.googleapis.com/youtube/v3/channels?part=statistics&id=${video.channelId}&key=${apiKey}`;

  const videoResponse = await fetch(videoUrl);
  const channelResponse = await fetch(channelUrl);

  const videoData = await videoResponse.json();
  const channelData = await channelResponse.json();

  const title = videoData.items[0].snippet.title;
  const viewCount = (videoData.items[0].statistics.viewCount as number);
  const likeCount = videoData.items[0].statistics.likeCount;
  const commentCount = videoData.items[0].statistics.commentCount;
  const subscriberCount = channelData.items[0].statistics.subscriberCount;

  const videoString = `   - Title: ${title}\n   - Channel: ${video.channelTitle}\n   - View Count: ${Number(viewCount).toLocaleString()}\n   - Days Since Published: ${video.daysSincePublished}\n   - Likes: ${Number(likeCount).toLocaleString()}\n   - Subscriber Count: ${Number(subscriberCount).toLocaleString('en-US')}\n   - Video URL: https://www.youtube.com/watch?v=${video.id}\n`;

  // const score = (viewCount / video.daysSincePublished / subscriberCount) + (likeCount * 10) + (commentCount * 100);
  const score = (((viewCount / subscriberCount) * 0.4) + ((likeCount / viewCount) * 0.3) + ((commentCount / viewCount) * 0.2) * (1/video.daysSincePublished)) ;

  return {
    videoString, 
    title,
    viewCount,
    likeCount,
    commentCount,
    subscriberCount,
    score
  };

}