api: flatten code directories, better filenames, remove old files

api/src/processing/match-action.js

@@ -0,0 +1,204 @@
+import { audioIgnore, services, supportedAudio } from "../config.js";
+import { createResponse } from "./request.js";
+import createFilename from "./create-filename.js";
+import { createStream } from "../stream/manage.js";
+
+export default function(r, host, userFormat, isAudioOnly, lang, isAudioMuted, disableMetadata, filenamePattern, toGif, requestIP) {
+    let action,
+        responseType = "stream",
+        defaultParams = {
+            u: r.urls,
+            headers: r.headers,
+            service: host,
+            filename: r.filenameAttributes ?
+                    createFilename(r.filenameAttributes, filenamePattern, isAudioOnly, isAudioMuted) : r.filename,
+            fileMetadata: !disableMetadata ? r.fileMetadata : false,
+            requestIP
+        },
+        params = {},
+        audioFormat = String(userFormat);
+
+    if (r.isPhoto) action = "photo";
+    else if (r.picker) action = "picker"
+    else if (r.isGif && toGif) action = "gif";
+    else if (isAudioMuted) action = "muteVideo";
+    else if (isAudioOnly) action = "audio";
+    else if (r.isM3U8) action = "m3u8";
+    else action = "video";
+
+    if (action === "picker" || action === "audio") {
+        if (!r.filenameAttributes) defaultParams.filename = r.audioFilename;
+        defaultParams.isAudioOnly = true;
+        defaultParams.audioFormat = audioFormat;
+    }
+    if (isAudioMuted && !r.filenameAttributes) {
+        defaultParams.filename = r.filename.replace('.', '_mute.')
+    }
+
+    switch (action) {
+        default:
+            return createResponse("error", {
+                code: "ErrorEmptyDownload"
+            });
+
+        case "photo":
+            responseType = "redirect";
+            break;
+
+        case "gif":
+            params = { type: "gif" }
+            break;
+
+        case "m3u8":
+            params = {
+                type: Array.isArray(r.urls) ? "render" : "remux"
+            }
+            break;
+
+        case "muteVideo":
+            let muteType = "mute";
+            if (Array.isArray(r.urls) && !r.isM3U8) {
+                muteType = "bridge";
+            }
+            params = {
+                type: muteType,
+                u: Array.isArray(r.urls) ? r.urls[0] : r.urls,
+                mute: true
+            }
+            if (host === "reddit" && r.typeId === "redirect")
+                responseType = "redirect";
+            break;
+
+        case "picker":
+            responseType = "picker";
+            switch (host) {
+                case "instagram":
+                case "twitter":
+                case "snapchat":
+                    params = { picker: r.picker };
+                    break;
+                case "tiktok":
+                    let audioStreamType = "render";
+                    if (r.bestAudio === "mp3" && (audioFormat === "mp3" || audioFormat === "best")) {
+                        audioFormat = "mp3";
+                        audioStreamType = "bridge"
+                    }
+                    params = {
+                        picker: r.picker,
+                        u: createStream({
+                            service: "tiktok",
+                            type: audioStreamType,
+                            u: r.urls,
+                            headers: r.headers,
+                            filename: r.audioFilename,
+                            isAudioOnly: true,
+                            audioFormat,
+                        }),
+                        copy: audioFormat === "best"
+                    }
+            }
+            break;
+
+        case "video":
+            switch (host) {
+                case "bilibili":
+                    params = { type: "render" };
+                    break;
+                case "youtube":
+                    params = { type: r.type };
+                    break;
+                case "reddit":
+                    responseType = r.typeId;
+                    params = { type: r.type };
+                    break;
+                case "vimeo":
+                    if (Array.isArray(r.urls)) {
+                        params = { type: "render" }
+                    } else {
+                        responseType = "redirect";
+                    }
+                    break;
+
+                case "twitter":
+                    if (r.type === "remux") {
+                        params = { type: r.type };
+                    } else {
+                        responseType = "redirect";
+                    }
+                    break;
+
+                case "vk":
+                case "tiktok":
+                    params = { type: "bridge" };
+                    break;
+
+                case "facebook":
+                case "vine":
+                case "instagram":
+                case "tumblr":
+                case "pinterest":
+                case "streamable":
+                case "snapchat":
+                case "loom":
+                    responseType = "redirect";
+                    break;
+            }
+            break;
+
+        case "audio":
+            if (audioIgnore.includes(host) || (host === "reddit" && r.typeId === "redirect")) {
+                return createResponse("error", {
+                    code: "ErrorEmptyDownload"
+                })
+            }
+
+            let processType = "render",
+                copy = false;
+
+            if (!supportedAudio.includes(audioFormat)) {
+                audioFormat = "best"
+            }
+
+            const serviceBestAudio = r.bestAudio || services[host]["bestAudio"];
+            const isBestAudio = audioFormat === "best";
+            const isBestOrMp3 = isBestAudio || audioFormat === "mp3";
+            const isBestAudioDefined = isBestAudio && serviceBestAudio;
+            const isBestHostAudio = serviceBestAudio && (audioFormat === serviceBestAudio);
+
+            const isTumblrAudio = host === "tumblr" && !r.filename;
+            const isSoundCloud = host === "soundcloud";
+            const isTiktok = host === "tiktok";
+
+            if (isBestAudioDefined || isBestHostAudio) {
+                audioFormat = serviceBestAudio;
+                processType = "bridge";
+                if (isSoundCloud || (isTiktok && audioFormat === "m4a")) {
+                    processType = "render"
+                    copy = true
+                }
+            } else if (isBestAudio && !isSoundCloud) {
+                audioFormat = "m4a";
+                copy = true
+            }
+
+            if (isTumblrAudio && isBestOrMp3) {
+                audioFormat = "mp3";
+                processType = "bridge"
+            }
+
+            if (r.isM3U8 || host === "vimeo") {
+                copy = false;
+                processType = "render"
+            }
+
+            params = {
+                type: processType,
+                u: Array.isArray(r.urls) ? r.urls[1] : r.urls,
+                audioFormat: audioFormat,
+                copy: copy
+            }
+            break;
+    }
+
+    return createResponse(responseType, {...defaultParams, ...params})
+}