cobalt/src/modules/services/vk.js

48 lines
2.5 KiB
JavaScript
Raw Normal View History

2022-07-08 18:17:56 +00:00
import got from "got";
import { xml2json } from "xml-js";
import loc from "../../localization/manager.js";
2022-07-08 18:17:56 +00:00
import { genericUserAgent, maxVideoDuration, services } from "../config.js";
2022-07-17 11:21:51 +00:00
import selectQuality from "../stream/selectQuality.js";
2022-07-08 18:17:56 +00:00
export default async function(obj) {
try {
let html = await got.get(`https://vk.com/video-${obj.userId}_${obj.videoId}`, { headers: { "user-agent": genericUserAgent } });
html.on('error', (err) => {
return false;
});
html = html.body;
if (html.includes(`{"lang":`)) {
let js = JSON.parse('{"lang":' + html.split(`{"lang":`)[1].split(']);')[0]);
if (js["mvData"]["is_active_live"] == '0') {
if (js["mvData"]["duration"] <= maxVideoDuration / 1000) {
let mpd = JSON.parse(xml2json(js["player"]["params"][0]["manifest"], { compact: true, spaces: 4 }));
let repr = mpd["MPD"]["Period"]["AdaptationSet"]["Representation"];
if (!mpd["MPD"]["Period"]["AdaptationSet"]["Representation"]) {
repr = mpd["MPD"]["Period"]["AdaptationSet"][0]["Representation"];
}
let attr = repr[repr.length - 1]["_attributes"];
let selectedQuality = `url${attr["height"]}`;
let maxQuality = js["player"]["params"][0][selectedQuality].split('type=')[1].slice(0, 1)
2022-08-16 07:14:19 +00:00
let userQuality = selectQuality('vk', obj.quality, Object.entries(services.vk.quality_match).reduce((r, [k, v]) => { r[v] = k; return r; })[maxQuality])
let id = js["player"]["params"][0][selectedQuality].split("id=")[1]
2022-07-08 18:17:56 +00:00
if (selectedQuality in js["player"]["params"][0]) {
return { urls: js["player"]["params"][0][selectedQuality].replace(`type=${maxQuality}`, `type=${services.vk.quality_match[userQuality]}`), filename: `vk_${id}_${attr['width']}x${attr['height']}.mp4`, audioFilename: loc(obj.lang, 'ErrorEmptyDownload') };
2022-07-08 18:17:56 +00:00
} else {
return { error: loc(obj.lang, 'ErrorEmptyDownload') };
2022-07-08 18:17:56 +00:00
}
} else {
return { error: loc(obj.lang, 'ErrorLengthLimit', maxVideoDuration / 60000) };
2022-07-08 18:17:56 +00:00
}
} else {
return { error: loc(obj.lang, 'ErrorLiveVideo') };
2022-07-08 18:17:56 +00:00
}
} else {
return { error: loc(obj.lang, 'ErrorEmptyDownload') };
2022-07-08 18:17:56 +00:00
}
} catch (err) {
return { error: loc(obj.lang, 'ErrorBadFetch') };
2022-07-08 18:17:56 +00:00
}
}