feat: added support rt news

FOSWLY · Jan 7, 2025 · 7f996fb · 7f996fb
1 parent fe54e5c
commit 7f996fb
Show file tree

Hide file tree

Showing 5 changed files with 90 additions and 0 deletions.
diff --git a/packages/core/src/types/service.ts b/packages/core/src/types/service.ts
@@ -57,6 +57,7 @@ export enum VideoService {
   dzen = "dzen",
   cloudflarestream = "cloudflarestream",
   loom = "loom",
+  rtnews = "rtnews",
 }
 
 export type ServiceConf<T = VideoService> = {

diff --git a/packages/ext/src/helpers/index.ts b/packages/ext/src/helpers/index.ts
@@ -52,6 +52,7 @@ import CloudflareStreamHelper from "./cloudflarestream";
 import DouyinHelper from "./douyin";
 import LoomHelper from "./loom";
 import ArtstationHelper from "./artstation";
+import RtNewsHelper from "./rtnews";
 
 export * as MailRuHelper from "./mailru";
 export * as WeverseHelper from "./weverse";
@@ -102,6 +103,7 @@ export * as CloudflareStreamHelper from "./cloudflarestream";
 export * as DouyinHelper from "./douyin";
 export * as LoomHelper from "./loom";
 export * as ArtstationHelper from "./artstation";
+export * as RtNewsHelper from "./rtnews";
 
 export const availableHelpers = {
   [CoreVideoService.mailru]: MailRuHelper,
@@ -154,6 +156,7 @@ export const availableHelpers = {
   [CoreVideoService.dzen]: DzenHelper,
   [CoreVideoService.cloudflarestream]: CloudflareStreamHelper,
   [CoreVideoService.loom]: LoomHelper,
+  [CoreVideoService.rtnews]: RtNewsHelper,
   [ExtVideoService.udemy]: UdemyHelper,
   [ExtVideoService.coursera]: CourseraHelper,
   [ExtVideoService.douyin]: DouyinHelper,

diff --git a/packages/ext/src/helpers/rtnews.ts b/packages/ext/src/helpers/rtnews.ts
@@ -0,0 +1,34 @@
+import { BaseHelper } from "./base";
+import { MinimalVideoData } from "../types/client";
+
+import { proxyMedia } from "@vot.js/shared/utils/utils";
+
+export default class RtNewsHelper extends BaseHelper {
+  // eslint-disable-next-line @typescript-eslint/require-await
+  async getVideoData(videoId: string): Promise<MinimalVideoData | undefined> {
+    const videoEl = document.querySelector(".jw-video, .media__video_noscript");
+    if (!videoEl) {
+      return undefined;
+    }
+
+    let videoSrc = videoEl.getAttribute("src");
+    if (!videoSrc) {
+      return undefined;
+    }
+
+    // yandex has case sensitive check of video format
+    if (videoSrc.endsWith(".MP4")) {
+      videoSrc = proxyMedia(videoSrc);
+    }
+
+    return {
+      videoId,
+      url: videoSrc,
+    };
+  }
+
+  // eslint-disable-next-line @typescript-eslint/require-await
+  async getVideoId(url: URL) {
+    return url.pathname.slice(1);
+  }
+}
diff --git a/packages/node/src/helpers/index.ts b/packages/node/src/helpers/index.ts
@@ -48,6 +48,7 @@ import YoutubeHelper from "./youtube";
 import DzenHelper from "./dzen";
 import CloudflareStreamHelper from "./cloudflarestream";
 import LoomHelper from "./loom";
+import RtNewsHelper from "./rtnews";
 
 export * as MailRuHelper from "./mailru";
 export * as WeverseHelper from "./weverse";
@@ -94,6 +95,7 @@ export * as YoutubeHelper from "./youtube";
 export * as DzenHelper from "./dzen";
 export * as CloudflareStreamHelper from "./cloudflarestream";
 export * as LoomHelper from "./loom";
+export * as RtNewsHelper from "./rtnews";
 
 export const availableHelpers = {
   [CoreVideoService.mailru]: MailRuHelper,
@@ -146,6 +148,7 @@ export const availableHelpers = {
   [CoreVideoService.dzen]: DzenHelper,
   [CoreVideoService.cloudflarestream]: CloudflareStreamHelper,
   [CoreVideoService.loom]: LoomHelper,
+  [CoreVideoService.rtnews]: RtNewsHelper,
 };
 
 export type AvailableVideoHelpers = typeof availableHelpers;

diff --git a/packages/node/src/helpers/rtnews.ts b/packages/node/src/helpers/rtnews.ts
@@ -0,0 +1,49 @@
+import { parseFromString } from "dom-parser";
+
+import { BaseHelper } from "./base";
+import { MinimalVideoData } from "../types/client";
+
+import Logger from "@vot.js/shared/utils/logger";
+import { proxyMedia } from "@vot.js/shared/utils/utils";
+
+export default class RtNewsHelper extends BaseHelper {
+  async getVideoData(videoId: string): Promise<MinimalVideoData | undefined> {
+    try {
+      const res = await this.fetch(this.service?.url + videoId);
+      const content = await res.text();
+
+      const doc = parseFromString(content.replace(/<!DOCTYPE html>/i, ""));
+      const videoEl = doc.getElementsByClassName("media__video_noscript")?.[0];
+      if (!videoEl) {
+        return undefined;
+      }
+
+      let videoSrc = videoEl.getAttribute("src");
+      if (!videoSrc) {
+        return undefined;
+      }
+
+      // yandex has case sensitive check of video format
+      if (videoSrc.endsWith(".MP4")) {
+        videoSrc = proxyMedia(videoSrc);
+      }
+
+      return {
+        videoId,
+        url: videoSrc,
+      };
+    } catch (err) {
+      Logger.error(
+        `Failed to get rt news video data by video ID: ${videoId}, because: ${
+          (err as Error).message
+        }`,
+      );
+      return undefined;
+    }
+  }
+
+  // eslint-disable-next-line @typescript-eslint/require-await
+  async getVideoId(url: URL) {
+    return url.pathname.slice(1);
+  }
+}