admin/cobalt
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

api: move request functions to separate file

Browse Source 
- request status id is no longer a cryptic number
- descriptive function names
This commit is contained in:
wukko
2024-05-15 21:39:44 +06:00
parent c10012130b
commit cc6345ff63
6 changed files with 183 additions and 131 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
src/modules/processing/match.js
View File

@@ -1,6 +1,6 @@
import { strict as assert } from "node:assert";
import { apiJSON } from "../sub/utils.js";
import { createResponse } from "../processing/request.js";
import { errorUnsupported, genericError, brokenLink } from "../sub/errors.js";
import loc from "../../localization/manager.js";
@@ -45,8 +45,8 @@ export default async function(host, patternMatch, lang, obj) {
try {
let r, isAudioOnly = !!obj.isAudioOnly, disableMetadata = !!obj.disableMetadata;
if (!testers[host]) return apiJSON(0, { t: errorUnsupported(lang) });
if (!(testers[host](patternMatch))) return apiJSON(0, { t: brokenLink(lang, host) });
if (!testers[host]) return createResponse("error", { t: errorUnsupported(lang) });
if (!(testers[host](patternMatch))) return createResponse("error", { t: brokenLink(lang, host) });
switch (host) {
case "twitter":
@@ -177,17 +177,17 @@ export default async function(host, patternMatch, lang, obj) {
r = await dailymotion(patternMatch);
break;
default:
return apiJSON(0, { t: errorUnsupported(lang) });
return createResponse("error", { t: errorUnsupported(lang) });
}
if (r.isAudioOnly) isAudioOnly = true;
let isAudioMuted = isAudioOnly ? false : obj.isAudioMuted;
if (r.error && r.critical)
return apiJSON(6, { t: loc(lang, r.error) })
return createResponse("critical", { t: loc(lang, r.error) })
if (r.error)
return apiJSON(0, {
return createResponse("error", {
t: Array.isArray(r.error)
? loc(lang, r.error[0], r.error[1])
: loc(lang, r.error)
@@ -199,7 +199,7 @@ export default async function(host, patternMatch, lang, obj) {
obj.filenamePattern, obj.twitterGif,
requestIP
)
} catch (e) {
return apiJSON(0, { t: genericError(lang, host) })
} catch {
return createResponse("error", { t: genericError(lang, host) })
}
}

26
src/modules/processing/matchActionDecider.js
View File

@@ -1,11 +1,11 @@
import { audioIgnore, services, supportedAudio } from "../config.js";
import { apiJSON } from "../sub/utils.js";
import { createResponse } from "../processing/request.js";
import loc from "../../localization/manager.js";
import createFilename from "./createFilename.js";
export default function(r, host, userFormat, isAudioOnly, lang, isAudioMuted, disableMetadata, filenamePattern, toGif, requestIP) {
let action,
responseType = 2,
responseType = "stream",
defaultParams = {
u: r.urls,
service: host,
@@ -36,10 +36,10 @@ export default function(r, host, userFormat, isAudioOnly, lang, isAudioMuted, di
switch (action) {
default:
return apiJSON(0, { t: loc(lang, 'ErrorEmptyDownload') });
return createResponse("error", { t: loc(lang, 'ErrorEmptyDownload') });
case "photo":
responseType = 1;
responseType = "redirect";
break;
case "gif":
@@ -56,11 +56,12 @@ export default function(r, host, userFormat, isAudioOnly, lang, isAudioMuted, di
u: Array.isArray(r.urls) ? r.urls[0] : r.urls,
mute: true
}
if (host === "reddit" && r.typeId === 1) responseType = 1;
if (host === "reddit" && r.typeId === "redirect")
responseType = "redirect";
break;
case "picker":
responseType = 5;
responseType = "picker";
switch (host) {
case "instagram":
case "twitter":
@@ -98,7 +99,7 @@ export default function(r, host, userFormat, isAudioOnly, lang, isAudioMuted, di
if (Array.isArray(r.urls)) {
params = { type: "render" }
} else {
responseType = 1;
responseType = "redirect";
}
break;
@@ -106,7 +107,7 @@ export default function(r, host, userFormat, isAudioOnly, lang, isAudioMuted, di
if (r.type === "remux") {
params = { type: r.type };
} else {
responseType = 1;
responseType = "redirect";
}
break;
@@ -121,14 +122,15 @@ export default function(r, host, userFormat, isAudioOnly, lang, isAudioMuted, di
case "tumblr":
case "pinterest":
case "streamable":
responseType = 1;
responseType = "redirect";
break;
}
break;
case "audio":
if ((host === "reddit" && r.typeId === 1) || audioIgnore.includes(host)) {
return apiJSON(0, { t: loc(lang, 'ErrorEmptyDownload') })
if (audioIgnore.includes(host)
|| (host === "reddit" && r.typeId === "redirect")) {
return createResponse("error", { t: loc(lang, 'ErrorEmptyDownload') })
}
let processType = "render",
@@ -178,5 +180,5 @@ export default function(r, host, userFormat, isAudioOnly, lang, isAudioMuted, di
break;
}
return apiJSON(responseType, {...defaultParams, ...params})
return createResponse(responseType, {...defaultParams, ...params})
}

154
src/modules/processing/request.js Normal file
View File

@@ -0,0 +1,154 @@
import ipaddr from "ipaddr.js";
import { normalizeURL } from "../processing/url.js";
import { createStream } from "../stream/manage.js";
const apiVar = {
allowed: {
vCodec: ["h264", "av1", "vp9"],
vQuality: ["max", "4320", "2160", "1440", "1080", "720", "480", "360", "240", "144"],
aFormat: ["best", "mp3", "ogg", "wav", "opus"],
filenamePattern: ["classic", "pretty", "basic", "nerdy"]
},
booleanOnly: [
"isAudioOnly",
"isTTFullAudio",
"isAudioMuted",
"dubLang",
"disableMetadata",
"twitterGif",
"tiktokH265"
]
}
export function createResponse(responseType, responseData) {
try {
let status = 200,
response = {};
switch(responseType) {
case "error":
status = 400;
break;
case "rate-limit":
status = 429;
break;
}
switch (responseType) {
case "error":
case "success":
case "rate-limit":
response = {
text: responseData.t
}
break;
case "redirect":
response = {
url: responseData.u
}
break;
case "stream":
response = {
url: createStream(responseData)
}
break;
case "picker":
let pickerType = "various",
audio = false;
if (responseData.service === "tiktok") {
audio = responseData.u
pickerType = "images"
}
response = {
pickerType: pickerType,
picker: responseData.picker,
audio: audio
}
break;
default:
throw "unreachable"
}
return {
status,
body: {
status: responseType,
...response
}
}
} catch {
return {
status: 500,
body: {
status: "error",
text: "Internal Server Error"
}
}
}
}
export function verifyRequest(request) {
try {
let template = {
url: normalizeURL(decodeURIComponent(request.url)),
vCodec: "h264",
vQuality: "720",
aFormat: "mp3",
filenamePattern: "classic",
isAudioOnly: false,
isTTFullAudio: false,
isAudioMuted: false,
disableMetadata: false,
dubLang: false,
twitterGif: false,
tiktokH265: false
}
const requestKeys = Object.keys(request);
const templateKeys = Object.keys(template);
if (requestKeys.length > templateKeys.length + 1 || !request.url) {
return false;
}
for (const i in requestKeys) {
const key = requestKeys[i];
const item = request[key];
if (String(key) !== "url" && templateKeys.includes(key)) {
if (apiVar.booleanOnly.includes(key)) {
template[key] = !!item;
} else if (apiVar.allowed[key] && apiVar.allowed[key].includes(item)) {
template[key] = String(item)
}
}
}
if (template.dubLang)
template.dubLang = verifyLanguageCode(request.dubLang);
return template
} catch {
return false
}
}
export function getIP(req) {
const strippedIP = req.ip.replace(/^::ffff:/, '');
const ip = ipaddr.parse(strippedIP);
if (ip.kind() === 'ipv4') {
return strippedIP;
}
const prefix = 56;
const v6Bytes = ip.toByteArray();
v6Bytes.fill(0, prefix / 8);
return ipaddr.fromByteArray(v6Bytes).toString();
}

6
src/modules/processing/services/reddit.js
View File

@@ -68,7 +68,7 @@ export default async function(obj) {
data = data[0]?.data?.children[0]?.data;
if (data?.url?.endsWith('.gif')) return {
typeId: 1,
typeId: "redirect",
urls: data.url
}
@@ -106,12 +106,12 @@ export default async function(obj) {
let id = video.split('/')[3];
if (!audio) return {
typeId: 1,
typeId: "redirect",
urls: video
}
return {
typeId: 2,
typeId: "stream",
type: "render",
urls: [video, audioFileLink],
audioFilename: `reddit_${id}_audio`,