feat: Rework on streaming to use WebM format

This commit is contained in:
2024-06-21 00:30:21 +07:00
parent 237a041301
commit a8062b5fe2
14 changed files with 1553 additions and 400 deletions

91
src/utils/Request.ts Normal file
View File

@@ -0,0 +1,91 @@
import axios, { AxiosRequestConfig, AxiosResponse } from "axios";
export async function makeStreamRequest(
url: string,
options: AxiosRequestConfig = {},
body?: any,
): Promise<AxiosResponse> {
const { headers = {}, method = "GET" } = options;
let config: AxiosRequestConfig = {
url,
method,
headers,
data: body,
responseType: "stream",
};
// Override / Add config
config = Object.assign(config, options);
try {
const response = await axios(config);
return response;
} catch (err) {
throw err;
}
}
export async function getStream(
url: string,
options: AxiosRequestConfig = { method: "GET" },
): Promise<AxiosResponse<any, any>> {
try {
let response = await makeStreamRequest(url, options);
const visitedUrls = new Set<string>();
// Handle redirection and detect redirection loop
while (
response.status >= 300 &&
response.status < 400 &&
response.headers.location
) {
const redirectUrl = response.headers.location;
if (visitedUrls.has(redirectUrl)) {
throw new Error("Redirection loop detected");
}
visitedUrls.add(redirectUrl);
response = await makeStreamRequest(redirectUrl, options);
}
return response;
} catch (error) {
throw error;
}
}
export async function getYouTubeFormats(id: string) {
const new_url = `https://www.youtube.com/watch?v=${id}&has_verified=1`;
const response = await axios.get(new_url, {
headers: { "accept-language": "en-US,en;q=0.9" },
withCredentials: true,
});
const body = response.data;
const match = body.match(
/var ytInitialPlayerResponse = (.*?)(?=;\s*<\/script>)/,
);
const data = match ? match[1] : null;
if (!data) throw new Error("Failed to get YouTube formats");
try {
let formats = JSON.parse(data).streamingData.adaptiveFormats;
if (!formats) throw new Error("Failed to parse YouTube formats");
// Filters only audio formats that are webm
formats = formats.filter((format: any) =>
format.mimeType.startsWith("audio/webm;"),
);
// Sort the quality of the formats
formats = formats.sort((a: any, b: any) => {
const aQuality = a.audioQuality === "AUDIO_QUALITY_MEDIUM" ? 0 : 1;
const bQuality = b.audioQuality === "AUDIO_QUALITY_MEDIUM" ? 0 : 1;
return aQuality - bQuality;
});
return formats;
} catch (err) {
throw new Error("Failed to parse YouTube formats");
}
}