91 lines
3.0 KiB
TypeScript
91 lines
3.0 KiB
TypeScript
import { REDTUBE_BASE_URL } from "@/constants/urls";
|
|
import { FetchParams, GalleryData, VideoData, VideoSourceItem } from "@/meta/data";
|
|
import { getHeaders } from "../common/headers";
|
|
import { getDataFromRedis, storeDataIntoRedis } from "@/redis/client";
|
|
import { DEFAULT_REDTUBE_GALLERY_EXPIRY, DEFAULT_REDTUBE_VIDEO_EXPIRY, DEFAULT_RELATED_VIDEO_KEY_PATH } from "@/constants/redis";
|
|
|
|
import * as cheerio from "cheerio";
|
|
|
|
import axios, { AxiosError } from "axios";
|
|
import { createSessionCookie, findGetMediaUrlInTagblock } from "../common/mindgeek";
|
|
import { Platforms } from "@/meta/settings";
|
|
import { getRedTubeMediaUrlList } from "./url";
|
|
|
|
export const fetchRedTubeVideoData = async (videoId: string, params?: FetchParams): Promise<[VideoData, GalleryData[]]> => {
|
|
|
|
let data: VideoData = {
|
|
hlsUrl: '',
|
|
srcSet: []
|
|
}
|
|
|
|
let relatedData: GalleryData[] = [];
|
|
|
|
let mediaUrl, sessionCookie, convertedData: VideoSourceItem[]
|
|
|
|
let reqHeaders = getHeaders(REDTUBE_BASE_URL)
|
|
|
|
const queryUrl = `${REDTUBE_BASE_URL}/${videoId.replace(/\//g, '')}`
|
|
|
|
const cachedVideoData = await getDataFromRedis(queryUrl)
|
|
const cachedRelatedData = await getDataFromRedis(queryUrl + DEFAULT_RELATED_VIDEO_KEY_PATH)
|
|
|
|
if (cachedVideoData) {
|
|
return [cachedVideoData as VideoData, cachedRelatedData as GalleryData[] ?? []]
|
|
}
|
|
|
|
await axios.get(queryUrl, reqHeaders)
|
|
|
|
.then(async response => {
|
|
|
|
sessionCookie = response?.headers["set-cookie"] ? createSessionCookie(response?.headers["set-cookie"]) : '';
|
|
|
|
const html = response.data;
|
|
|
|
const $ = cheerio.load(html);
|
|
|
|
const scriptTags = $("script");
|
|
|
|
scriptTags.map((idx, elem) => {
|
|
const getMediaUrl = findGetMediaUrlInTagblock($(elem).toString().replace(/\\/g, ''), 'media/mp4') ?? null
|
|
|
|
if (getMediaUrl) {
|
|
mediaUrl = `${REDTUBE_BASE_URL}${getMediaUrl}`
|
|
}
|
|
|
|
})
|
|
|
|
const wrapperId = "#related_videos_center li.tm_video_block"
|
|
|
|
const thumbs = $(wrapperId);
|
|
|
|
thumbs.map((key, thumb) => {
|
|
|
|
const videoUrl = $(thumb).find("a.tm_video_link").attr("href")?.split('/')[1];
|
|
const imgUrl = $(thumb).find("img.js_thumbImageTag").attr("data-src")
|
|
const text = $(thumb).find("a.tm_video_title").attr("title");
|
|
|
|
videoUrl && imgUrl && text && relatedData.push({
|
|
videoUrl,
|
|
imgUrl,
|
|
text,
|
|
platform: Platforms.redtube
|
|
})
|
|
})
|
|
|
|
}).catch((error: AxiosError) => {
|
|
// error handling goes here
|
|
});
|
|
|
|
if (sessionCookie && mediaUrl) {
|
|
convertedData = await getRedTubeMediaUrlList(mediaUrl, sessionCookie)
|
|
data.srcSet = convertedData.reverse()
|
|
|
|
await storeDataIntoRedis(queryUrl, data, DEFAULT_REDTUBE_VIDEO_EXPIRY);
|
|
}
|
|
|
|
if (relatedData.length > 0) {
|
|
await storeDataIntoRedis(queryUrl + DEFAULT_RELATED_VIDEO_KEY_PATH, relatedData, DEFAULT_REDTUBE_GALLERY_EXPIRY);
|
|
}
|
|
|
|
return [ data, relatedData ]
|
|
} |