yt_metadata/src/youtubeSearch.js

A package to fetch YouTube content details, either by URL or search query.

const { getNextApiKey } = require('./apiKeyRotation');

/*
  * Searches for a YouTube media using the YouTube API.
  * @param {string} query - The search query.
  * @return {Promise<[string, string]|null>} A tuple of media ID and type, or null on failure.
*/
async function searchYoutubeMedia(query) {
	try {
		const apiKey = getNextApiKey();
		const searchURL = `https://www.googleapis.com/youtube/v3/search?part=snippet&maxResults=1&q=${encodeURIComponent(query)}&key=${apiKey}`;
		const response = await fetch(searchURL);
		const data = await response.json();

		if (!response.ok) {
			throw new Error(`YouTube API responded with ${response.status}`);
		}

		return extractMediaDetails(data);
	} catch (error) {
		console.error(`Error in searchYoutubeMedia: ${error.message}`);
		return null;
	}
}

/**
 * Extracts media details from the YouTube API response.
 * @param {object} data - The API response data.
 * @returns {[string, string]|null} - A tuple of media ID and type, or null on failure.
 */
function extractMediaDetails(data) {
	if (data.items && data.items.length > 0) {
		const { videoId, playlistId, channelId } = data.items[0].id;
		const mediatype = videoId ? 'video' : playlistId ? 'playlist' : channelId ? 'channel' : null;

		return [videoId || playlistId || channelId, mediatype];
	} else {
		return null;
	}
}

module.exports = {
	searchYoutubeMedia,
};