mirrors/cobalt
1
0
Fork 0
mirror of https://github.com/imputnet/cobalt.git synced 2025-07-18 19:28:29 +00:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

improvement: isAudioMuted support in nicovideo

Browse Source
This commit is contained in:
mikhail 2024-05-22 23:27:57 +05:00
parent d39fd06387
commit 9a5ff35eb7
2 changed files with 24 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -191,6 +191,7 @@ export default async function(host, patternMatch, lang, obj) {
id: patternMatch.id, id: patternMatch.id,
quality: obj.vQuality, quality: obj.vQuality,
isAudioOnly: isAudioOnly, isAudioOnly: isAudioOnly,
isAudioMuted: obj.isAudioMuted,
}); });
break; break;
default: default:

34
src/modules/processing/services/nicovideo.js
View File

@ -18,8 +18,9 @@ const NICOVIDEO_HLS_API_URL =
"https://nvapi.nicovideo.jp/v1/watch/%s/access-rights/hls?actionTrackId=%s"; "https://nvapi.nicovideo.jp/v1/watch/%s/access-rights/hls?actionTrackId=%s";
class CobaltError extends Error { class CobaltError extends Error {
constructor(localizatedMessage) { constructor(locMessage) {
this.localizatedMessage = localizatedMessage; super(); // gdsfkgjsoiredgjhredszdfpijgkertoindsuf
this.locMessage = locMessage;
} }
} }
@ -113,7 +114,7 @@ async function fetchContentURL(id, actionTrackId, accessRightKey, outputs) {
return data.data.contentUrl; return data.data.contentUrl;
} }
async function getHLSContent(contentURL, quality, isAudioOnly) { async function getHLSContent(contentURL, quality, isAudioOnly, isAudioMuted) {
const hls = await fetch(contentURL) const hls = await fetch(contentURL)
.then((response) => response.text()) .then((response) => response.text())
.then((response) => HLS.parse(response)); .then((response) => HLS.parse(response));
@ -137,31 +138,41 @@ async function getHLSContent(contentURL, quality, isAudioOnly) {
? { resolution: null, urls: audioUrl, type: "audio" } ? { resolution: null, urls: audioUrl, type: "audio" }
: { : {
resolution: hlsContent.resolution, resolution: hlsContent.resolution,
urls: [hlsContent.uri, audioUrl], urls: isAudioMuted ? hlsContent.uri : [hlsContent.uri, audioUrl],
type: "video", type: "video",
}; };
} }
export default async function nicovideo({ id, quality, isAudioOnly }) { export default async function nicovideo({
id,
quality,
isAudioOnly,
isAudioMuted,
}) {
try { try {
const { actionTrackId, title, author, lengthInSeconds } = await getBasicVideoInformation(id); const { actionTrackId, title, author, lengthInSeconds } =
await getBasicVideoInformation(id);
if (lengthInSeconds > env.durationLimit) { if (lengthInSeconds > env.durationLimit) {
throw new CobaltError(['ErrorLengthLimit', env.durationLimit / 60]); throw new CobaltError(["ErrorLengthLimit", env.durationLimit / 60]);
} }
const { resolution, urls, type } = await fetchGuestData(id, actionTrackId) const { resolution, urls, type } = await fetchGuestData(id, actionTrackId)
.then(({ accessRightKey, outputs }) => .then(({ accessRightKey, outputs }) =>
fetchContentURL(id, actionTrackId, accessRightKey, outputs) fetchContentURL(id, actionTrackId, accessRightKey, outputs)
) )
.then((contentURL) => getHLSContent(contentURL, quality, isAudioOnly)); .then((contentURL) =>
getHLSContent(contentURL, quality, isAudioOnly, isAudioMuted)
);
return { return {
urls, urls,
isAudioOnly: type === "audio", isAudioOnly: type === "audio",
fileMetadata: { fileMetadata: {
title: cleanString(title.trim()), title: cleanString(title.trim()),
artist: cleanString(author.nickname.trim()), artist: author.nickname
? cleanString(author.nickname.trim())
: undefined,
}, },
// bible accurate object concatenation // bible accurate object concatenation
filenameAttributes: { filenameAttributes: {
@ -177,9 +188,10 @@ export default async function nicovideo({ id, quality, isAudioOnly }) {
} }
: {}), : {}),
}, },
...(type === "audio" ? { isM3U8: true, bestAudio: "mp3" } : {}), ...(type === "audio" || typeof urls === "string" ? { isM3U8: true } : {}),
...(type === "audio" ? { bestAudio: "mp3" } : {}),
}; };
} catch (error) { } catch (error) {
return { error: error.localizatedMessage ?? "ErrorSomethingWentWrong" }; return { error: error.locMessage ?? "ErrorSomethingWentWrong" };
} }
} }