Last updated 2021-06-16 09:45:53

source/script/7f3c8323580b2d79dfe42a315656355a.js

@@ -1,84 +1,69 @@
+const enabled_live = false; // 开启直播推荐，默认关闭
+
 try {
   let body = $response.body.replace(/\"room_id\":(\d{2,})/g,'"room_id":"$1"');
   let obj = JSON.parse(body);
   if (obj.data) obj.data = filter_data(obj.data);
-  if (obj.aweme_list) obj.aweme_list = filter_list(obj.aweme_list);
-  if (obj.aweme_detail) obj.aweme_detail = filter_detail(obj.aweme_detail);
-  if (obj.aweme_details) obj.aweme_details = filter_details(obj.aweme_details);
+  if (obj.aweme_list) obj.aweme_list = filter_aweme_list(obj.aweme_list);
+  if (obj.aweme_detail) obj.aweme_detail = filter_aweme_detail(obj.aweme_detail);
   $done({ body: JSON.stringify(obj) });
 } catch (error) {
-  console.log("脚本处理时遇到些问题，响应不会被修改\n" + error);
+  console.log("脚本运行错误，跳过处理。\n" + error);
   $done({});
 }
 
 function filter_data(data) {
-  for (var i = data.length - 1; i >= 0; i--) {
-    if (data[i].aweme_info) {
-      if (data[i].aweme_info.is_ads === true) {
-        data.splice(i, 1);
-      } else if (data[i].aweme_info.video) {
-        data[i].aweme_info.status.reviewed = 1;
-        data[i].aweme_info.video_control.prevent_download_type = 0;
-        data[i].aweme_info.video_control.allow_download = true;
-        delete data[i].aweme_info.video.misc_download_addrs;
-        let play = data[i].aweme_info.video.play_addr.url_list;
-        data[i].aweme_info.video.download_addr.url_list = play;
-        let download = data[i].aweme_info.video.download_addr;
-        data[i].aweme_info.video.download_suffix_logo_addr = download;
-      }
-    }
-    if (data[i].aweme) {
-      data[i].aweme.video_control.allow_download = true;
-      data[i].aweme.video_control.prevent_download_type = 0;
-      data[i].aweme.status.reviewed = 1;
-      delete data[i].aweme.video.misc_download_addrs;
-      let play = data[i].aweme.video.play_addr.url_list;
-      data[i].aweme.video.download_addr.url_list = play;
-      let download = data[i].aweme.video.download_addr;
-      data[i].aweme.video.download_suffix_logo_addr = download;
+  if (data && data.length > 0) {
+    let i = data.length;
+    while (i--) {
+      let element = data[i].aweme;
+      if (element.images) filter_images(element.images);
+      if (element.video) filter_videos(element);
     }
   }
   return data;
 }
 
-function filter_list(list) {
-  for (var i = list.length - 1; i >= 0; i--) {
-    if (list[i].video && list[i].is_ads != true) {
-      list[i].video_control.allow_download = true;
-      list[i].video_control.prevent_download_type = 0;
-      list[i].status.reviewed = 1;
-      delete list[i].video.misc_download_addrs;
-      let play = list[i].video.play_addr.url_list;
-      list[i].video.download_addr.url_list = play;
-      let download = list[i].video.download_addr;
-      list[i].video.download_suffix_logo_addr = download;
-    } else {
-      list.splice(i, 1);
+function filter_aweme_list(aweme_list) {
+  if (aweme_list && aweme_list.length > 0) {
+    let i = aweme_list.length;
+    while (i--) {
+      let element = aweme_list[i];
+      if (element.is_ads == true) {
+        aweme_list.splice(i, 1);
+      } else if (element.images) {
+        filter_images(element.images);
+      } else if (element.video) {
+        filter_videos(element);
+      } else {
+        if (!enabled_live) aweme_list.splice(i, 1);
+      }
     }
   }
-  return list;
+  return aweme_list;
 }
 
-function filter_detail(detail) {
-  detail.status.reviewed = 1;
-  detail.video_control.allow_download = true;
-  detail.video_control.prevent_download_type = 0;
-  let play = detail.video.play_addr.url_list;
-  detail.video.download_addr.url_list = play;
-  let download = detail.video.download_addr;
-  detail.video.download_suffix_logo_addr = download;
-  return detail;
+function filter_aweme_detail(aweme_detail) {
+  if (aweme_detail.images) filter_images(aweme_detail.images);
+  if (aweme_detail.video) filter_videos(aweme_detail);
+  return aweme_detail;
 }
 
-function filter_details(details) {
-  for (var i = details.length - 1; i >= 0; i--) {
-    details[i].status.reviewed = 1;
-    details[i].video_control.allow_download = true;
-    details[i].video_control.prevent_download_type = 0;
-    let play = details[i].video.play_addr.url_list;
-    details[i].video.download_addr.url_list = play;
-    let download = details[i].video.download_addr;
-    details[i].video.download_suffix_logo_addr = download;
+function filter_images(images) {
+  let j = images.length;
+  while (j--) {
+    images[j].download_url_list = images[j].url_list;
   }
-  return details;
+  return images;
+}
+
+function filter_videos(videos) {
+  videos.status.reviewed = 1;
+  videos.video_control.allow_download = true;
+  videos.video_control.prevent_download_type = 0;
+  delete videos.video.misc_download_addrs;
+  const play_url = videos.video.play_addr;
+  videos.video.download_addr = play_url;
+  videos.video.download_suffix_logo_addr = play_url;
+  return videos;
 }