feat: support h265 for video embeds (#10)

thanks @Valafi

package.json

@@ -5,6 +5,7 @@
     "dev": "wrangler dev src/index.ts --env=local",
     "build": "wrangler build src/index.ts",
     "deploy": "wrangler deploy --minify src/index.ts",
+    "deploy:staging": "wrangler deploy --env=staging --minify src/index.ts",
     "test": "jest",
     "prettier": "prettier . --write"
   },

src/index.ts

@@ -210,25 +210,28 @@ app.get('/generate/alternate', (c) => {
 
 app.get('/generate/video/:videoId', async (c) => {
   const { videoId } = c.req.param()
+  const forceLow = c.req.query('h264') === 'true' || c.req.query('encoder') === 'h264' || c.req.query('quality') === 'h264'
 
   try {
+    // To ensure the video is valid, decrease load on TikWM by checking the video data first
     const data = await scrapeVideoData(videoId)
 
-    /*
-        if (!(data instanceof Error)) {
-            if(data.video.playAddr) {
-                return c.redirect(data.video.playAddr)
-            } else {
-                return new Response('No video found', { status: 404,
-                    headers: {
-                        'Cache-Control': 'no-cache, no-store, must-revalidate',
-                    }
-                })
-            }
+    if (data instanceof Error) {
+      return new Response((data as Error).message, {
+        status: 500,
+        headers: {
+          'Cache-Control': 'no-cache, no-store, must-revalidate'
         }
-    */
+      })
+    }
+
+    const highAvailable = data.video.bitrateInfo.find((bitrate) => bitrate.CodecType.includes('h265'))
 
-    return c.redirect(`https://tikwm.com/video/media/play/${videoId}.mp4`)
+    if (highAvailable && !forceLow) {
+      return c.redirect(`https://tikwm.com/video/media/hdplay/${videoId}.mp4`)
+    } else {
+      return c.redirect(`https://tikwm.com/video/media/play/${videoId}.mp4`)
+    }
   } catch (e) {
     return new Response((e as Error).message, {
       status: 500,

src/types/Web.ts

@@ -2916,7 +2916,18 @@ export interface Video {
   downloadAddr: string
   shareCover: string[]
   reflowCover: string
+  bitrate: number
+  encodedType: string
+  format: string
+  videoQuality: string
+  encodeUserTag: string
+  codecType: string
+  definition: string
+  subtitleInfos: any[]
   zoomCover: ZoomCover
+  volumeInfo: VolumeInfo
+  bitrateInfo: BitrateInfo[]
+  VQScore: string
 }
 
 export interface ZoomCover {
@@ -2926,6 +2937,29 @@ export interface ZoomCover {
   '960': string
 }
 
+export interface VolumeInfo {
+  Loudness: number
+  Peak: number
+}
+
+export interface BitrateInfo {
+  Bitrate: number
+  QualityType: number
+  GearName: string
+  PlayAddr: PlayAddr
+  CodecType: string
+  MVMAF: string
+}
+
+export interface PlayAddr {
+  DataSize: string
+  Uri: string
+  UrlList: string[]
+  UrlKey: string
+  FileHash: string
+  FileCs: string
+}
+
 export interface Author {
   id: string
   shortId: string

wrangler.toml

@@ -2,24 +2,11 @@ name = "fxtiktok-rewrite"
 compatibility_date = "2023-01-01"
 main = "src/index.ts"
 
-vars = { WORKER_ENV = "production" }
+vars = { WORKER_ENV = "production", BASE_URL = "https://fxtiktok-rewrite.dargy.workers.dev" }
 
-[env.local]
-# these will be used only when --env=local
-vars = { WORKER_ENV = "local" }
-
-# [vars]
-# MY_VARIABLE = "production_value"
-
-# [[kv_namespaces]]
-# binding = "MY_KV_NAMESPACE"
-# id = "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"
+[env.staging]
+name = "fxtiktok-rewrite-dev"
+vars = { WORKER_ENV = "staging", BASE_URL = "https://fxtiktok-rewrite-dev.dargy.workers.dev" }
 
-# [[r2_buckets]]
-# binding = "MY_BUCKET"
-# bucket_name = "my-bucket"
-
-# [[d1_databases]]
-# binding = "DB"
-# database_name = "my-database"
-# database_id = ""
+[env.local]
+vars = { WORKER_ENV = "local", BASE_URL = "http://127.0.0.1:8787" }