mirror of
https://github.com/ZHANGTIANYAO1/teamspeak-music-bot.git
synced 2026-10-02 13:02:49 +08:00
Compare commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
faf6ac09ec |
No files matched your search
@@ -364,7 +364,6 @@ sudo systemctl start tsmusicbot
|
|||||||
| `!mode <seq\|loop\|random\|rloop>` | 切换播放模式 |
|
| `!mode <seq\|loop\|random\|rloop>` | 切换播放模式 |
|
||||||
| `!playlist <歌单名或ID>` | 加载歌单(支持名称模糊搜索和 ID;Jellyfin 歌单 GUID 也可直接粘贴) |
|
| `!playlist <歌单名或ID>` | 加载歌单(支持名称模糊搜索和 ID;Jellyfin 歌单 GUID 也可直接粘贴) |
|
||||||
| `!playlist -q <歌单名>` | 从 QQ 音乐搜索并加载歌单 |
|
| `!playlist -q <歌单名>` | 从 QQ 音乐搜索并加载歌单 |
|
||||||
| `!playlist <歌单链接>` | 直接粘贴网易云 / QQ 音乐 / YouTube 歌单链接加载,平台由链接自动识别,无需加 `-q` 等标志;也可直接粘贴 App 的分享文案或短链(`163cn.tv`、`c6.y.qq.com`) |
|
|
||||||
| `!album <专辑名或ID>` | 加载专辑(支持名称搜索 / 数字 ID / Jellyfin GUID) |
|
| `!album <专辑名或ID>` | 加载专辑(支持名称搜索 / 数字 ID / Jellyfin GUID) |
|
||||||
| `!artist <歌手名>` | 按歌手循环播放(支持 `-j`/`-n`/`-q`/`-k`/`-b`/`-y`) |
|
| `!artist <歌手名>` | 按歌手循环播放(支持 `-j`/`-n`/`-q`/`-k`/`-b`/`-y`) |
|
||||||
| `!fm` | 私人 FM(默认网易云,自动续播) |
|
| `!fm` | 私人 FM(默认网易云,自动续播) |
|
||||||
|
|||||||
@@ -1590,70 +1590,3 @@ describe("BotInstance Bilibili multi-P resolution", () => {
|
|||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|
||||||
|
|
||||||
describe("cmdPlaylist with a playlist link (#160)", () => {
|
|
||||||
const cmdPlaylist = (BotInstance.prototype as any).cmdPlaylist as (
|
|
||||||
this: unknown, cmd: { name: string; args: string; rawArgs: string[]; flags: Set<string> },
|
|
||||||
) => Promise<string>;
|
|
||||||
|
|
||||||
function makeCtx() {
|
|
||||||
const song = { id: "s1", name: "Song", artist: "A", album: "B", duration: 1, coverUrl: "" };
|
|
||||||
const makeProvider = (platform: string) => ({
|
|
||||||
platform,
|
|
||||||
search: vi.fn().mockResolvedValue({ songs: [], playlists: [] }),
|
|
||||||
getPlaylistSongs: vi.fn().mockResolvedValue([song]),
|
|
||||||
});
|
|
||||||
const providers: Record<string, any> = {
|
|
||||||
netease: makeProvider("netease"),
|
|
||||||
qq: makeProvider("qq"),
|
|
||||||
youtube: makeProvider("youtube"),
|
|
||||||
};
|
|
||||||
const queued: any[] = [];
|
|
||||||
return {
|
|
||||||
providers,
|
|
||||||
queued,
|
|
||||||
getProvider: vi.fn(() => providers.netease),
|
|
||||||
getProviderFor: vi.fn((p: string) => providers[p]),
|
|
||||||
assertProviderEnabled: vi.fn(),
|
|
||||||
extractId: (BotInstance.prototype as any).extractId,
|
|
||||||
looksLikeCollectionId: (BotInstance.prototype as any).looksLikeCollectionId,
|
|
||||||
player: { stop: vi.fn() },
|
|
||||||
queue: { clear: vi.fn(), add: (s: any) => queued.push(s), play: () => queued[0] },
|
|
||||||
disableFmMode: vi.fn(),
|
|
||||||
withRequester: (s: any) => s,
|
|
||||||
resolveAndPlay: vi.fn(async () => true),
|
|
||||||
sweepLocalAudio: vi.fn(),
|
|
||||||
emit: vi.fn(),
|
|
||||||
};
|
|
||||||
}
|
|
||||||
const cmd = (args: string, flags: string[] = []) =>
|
|
||||||
({ name: "playlist", args, rawArgs: args.split(" "), flags: new Set(flags) });
|
|
||||||
|
|
||||||
it("routes a QQ playlist link to QQ even without -q (default is NetEase)", async () => {
|
|
||||||
const ctx = makeCtx();
|
|
||||||
const reply = await cmdPlaylist.call(ctx, cmd("[URL]https://y.qq.com/n/ryqq/playlist/8052190267[/URL]"));
|
|
||||||
expect(ctx.providers.qq.getPlaylistSongs).toHaveBeenCalledWith("8052190267");
|
|
||||||
expect(ctx.providers.netease.getPlaylistSongs).not.toHaveBeenCalled();
|
|
||||||
expect(ctx.queued[0].platform).toBe("qq");
|
|
||||||
expect(reply).toMatch(/^Loaded 1 songs/);
|
|
||||||
});
|
|
||||||
|
|
||||||
it("loads a YouTube playlist link by its list id instead of name-searching the URL", async () => {
|
|
||||||
const ctx = makeCtx();
|
|
||||||
await cmdPlaylist.call(ctx, cmd("https://www.youtube.com/playlist?list=PLabc123"));
|
|
||||||
expect(ctx.providers.youtube.getPlaylistSongs).toHaveBeenCalledWith("PLabc123");
|
|
||||||
expect(ctx.providers.netease.search).not.toHaveBeenCalled();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("checks the link's platform is enabled", async () => {
|
|
||||||
const ctx = makeCtx();
|
|
||||||
ctx.assertProviderEnabled.mockImplementation(() => { throw new Error("音源未启用:qq"); });
|
|
||||||
await expect(cmdPlaylist.call(ctx, cmd("https://y.qq.com/n/ryqq/playlist/1"))).rejects.toThrow("音源未启用");
|
|
||||||
});
|
|
||||||
|
|
||||||
it("keeps the old behavior for a bare id", async () => {
|
|
||||||
const ctx = makeCtx();
|
|
||||||
await cmdPlaylist.call(ctx, cmd("2829883282"));
|
|
||||||
expect(ctx.providers.netease.getPlaylistSongs).toHaveBeenCalledWith("2829883282");
|
|
||||||
});
|
|
||||||
});
|
|
||||||
+10
-26
@@ -13,13 +13,7 @@ import {
|
|||||||
canRunCommand,
|
canRunCommand,
|
||||||
type ParsedCommand,
|
type ParsedCommand,
|
||||||
} from "./commands.js";
|
} from "./commands.js";
|
||||||
import {
|
import { parseSongRef, parseSelectionIndex } from "./song-ref.js";
|
||||||
parseSongRef,
|
|
||||||
parseSelectionIndex,
|
|
||||||
parsePlaylistRef,
|
|
||||||
findShareShortLink,
|
|
||||||
resolveShareLink,
|
|
||||||
} from "./song-ref.js";
|
|
||||||
import { splitTextIntoChunks } from "./text-chunk.js";
|
import { splitTextIntoChunks } from "./text-chunk.js";
|
||||||
import type { Logger } from "../logger.js";
|
import type { Logger } from "../logger.js";
|
||||||
import { SHARED_QUEUE_OWNER, type BotDatabase, type ProfileConfig, type StoredSong } from "../data/database.js";
|
import { SHARED_QUEUE_OWNER, type BotDatabase, type ProfileConfig, type StoredSong } from "../data/database.js";
|
||||||
@@ -462,11 +456,16 @@ export class BotInstance extends EventEmitter {
|
|||||||
this.voiceDucking.removeSpeaker(event.id);
|
this.voiceDucking.removeSpeaker(event.id);
|
||||||
void this.refreshOccupancy();
|
void this.refreshOccupancy();
|
||||||
});
|
});
|
||||||
this.tsClient.on("clientMoved", (event: { id: number }) => {
|
this.tsClient.on("clientMoved", (event: { id: number; targetChannelID: bigint }) => {
|
||||||
if (event.id === this.tsClient.getClientId()) {
|
if (event.id === this.tsClient.getClientId()) {
|
||||||
// Moving the bot invalidates every activity deadline from its old
|
// Moving the bot invalidates every activity deadline from its old
|
||||||
// channel even if no individual leave events arrive.
|
// channel even if no individual leave events arrive.
|
||||||
this.voiceDucking.reset(false);
|
this.voiceDucking.reset(false);
|
||||||
|
// Carry the now-playing channel description over to the new
|
||||||
|
// channel instead of leaving it stale in the old one (#159).
|
||||||
|
this.profileManager.onChannelMoved(event.targetChannelID).catch((err) => {
|
||||||
|
this.logger.warn({ err }, "Channel description move update failed");
|
||||||
|
});
|
||||||
} else {
|
} else {
|
||||||
this.voiceDucking.removeSpeaker(event.id);
|
this.voiceDucking.removeSpeaker(event.id);
|
||||||
}
|
}
|
||||||
@@ -1519,21 +1518,8 @@ export class BotInstance extends EventEmitter {
|
|||||||
}
|
}
|
||||||
|
|
||||||
private async cmdPlaylist(cmd: ParsedCommand, requesterName?: string): Promise<string> {
|
private async cmdPlaylist(cmd: ParsedCommand, requesterName?: string): Promise<string> {
|
||||||
if (!cmd.args) return "Usage: !playlist <playlist name, ID or link>";
|
if (!cmd.args) return "Usage: !playlist <playlist name or ID>";
|
||||||
|
const provider = this.getProvider(cmd.flags);
|
||||||
// A playlist link (#160) names its own platform, so it wins over flags.
|
|
||||||
// App share short links are followed one hop to the real URL first.
|
|
||||||
let ref = parsePlaylistRef(cmd.args);
|
|
||||||
if (!ref) {
|
|
||||||
const shortLink = findShareShortLink(cmd.args);
|
|
||||||
if (shortLink) {
|
|
||||||
const target = await resolveShareLink(shortLink);
|
|
||||||
ref = target ? parsePlaylistRef(target) : null;
|
|
||||||
if (!ref) return "Could not open that share link — paste the full playlist link or its ID instead";
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if (ref) this.assertProviderEnabled(ref.platform);
|
|
||||||
const provider = ref ? this.getProviderFor(ref.platform) : this.getProvider(cmd.flags);
|
|
||||||
|
|
||||||
// Determine if input is a direct ID (numeric / Jellyfin GUID) or a name search
|
// Determine if input is a direct ID (numeric / Jellyfin GUID) or a name search
|
||||||
const id = this.extractId(cmd.args);
|
const id = this.extractId(cmd.args);
|
||||||
@@ -1541,9 +1527,7 @@ export class BotInstance extends EventEmitter {
|
|||||||
|
|
||||||
let playlistId: string;
|
let playlistId: string;
|
||||||
|
|
||||||
if (ref) {
|
if (isDirectId || id !== cmd.args) {
|
||||||
playlistId = ref.id;
|
|
||||||
} else if (isDirectId || id !== cmd.args) {
|
|
||||||
// Input is a direct ID or URL containing an ID — use existing logic
|
// Input is a direct ID or URL containing an ID — use existing logic
|
||||||
playlistId = id;
|
playlistId = id;
|
||||||
} else {
|
} else {
|
||||||
|
|||||||
@@ -202,3 +202,65 @@ describe("BotProfileManager loadCustomAvatar (pre-connect load, #148)", () => {
|
|||||||
expect(ts.uploadCalls[0].equals(Buffer.from([2, 2]))).toBe(true);
|
expect(ts.uploadCalls[0].equals(Buffer.from([2, 2]))).toBe(true);
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|
||||||
|
describe("BotProfileManager channel description follows the bot (#159)", () => {
|
||||||
|
const cfgChannelDesc = { ...cfgOff, channelDescEnabled: true };
|
||||||
|
let ts: ReturnType<typeof makeMockTs> & { cid: bigint };
|
||||||
|
let channelEdits: () => string[];
|
||||||
|
|
||||||
|
beforeEach(() => {
|
||||||
|
ts = makeMockTs() as any;
|
||||||
|
ts.cid = 5n;
|
||||||
|
(ts as any).getChannelId = () => ts.cid;
|
||||||
|
channelEdits = () =>
|
||||||
|
(ts.sendCommandNoWait as any).mock.calls
|
||||||
|
.map((c: any[]) => c[0] as string)
|
||||||
|
.filter((cmd: string) => cmd.startsWith("channeledit"));
|
||||||
|
});
|
||||||
|
|
||||||
|
it("clears the old channel and fills the new one when moved while playing", async () => {
|
||||||
|
const pm = new BotProfileManager(ts as any, noopLogger, cfgChannelDesc, "Bot");
|
||||||
|
await pm.onSongChange(fakeSong);
|
||||||
|
expect(channelEdits()).toEqual([
|
||||||
|
expect.stringMatching(/^channeledit cid=5 channel_description=\S+/),
|
||||||
|
]);
|
||||||
|
|
||||||
|
ts.cid = 9n;
|
||||||
|
await pm.onChannelMoved(9n);
|
||||||
|
|
||||||
|
const edits = channelEdits();
|
||||||
|
expect(edits[1]).toBe("channeledit cid=5 channel_description=");
|
||||||
|
expect(edits[2]).toMatch(/^channeledit cid=9 channel_description=\S+/);
|
||||||
|
});
|
||||||
|
|
||||||
|
it("stopping after a move clears the channel the bot is in now, not the old one", async () => {
|
||||||
|
const pm = new BotProfileManager(ts as any, noopLogger, cfgChannelDesc, "Bot");
|
||||||
|
await pm.onSongChange(fakeSong);
|
||||||
|
ts.cid = 9n;
|
||||||
|
await pm.onChannelMoved(9n);
|
||||||
|
await pm.onSongChange(null);
|
||||||
|
expect(channelEdits().at(-1)).toBe("channeledit cid=9 channel_description=");
|
||||||
|
});
|
||||||
|
|
||||||
|
it("a move while idle touches no channel description", async () => {
|
||||||
|
const pm = new BotProfileManager(ts as any, noopLogger, cfgChannelDesc, "Bot");
|
||||||
|
ts.cid = 9n;
|
||||||
|
await pm.onChannelMoved(9n);
|
||||||
|
expect(channelEdits()).toEqual([]);
|
||||||
|
});
|
||||||
|
|
||||||
|
it("a move is ignored when the channel description feature is off", async () => {
|
||||||
|
const pm = new BotProfileManager(ts as any, noopLogger, cfgOff, "Bot");
|
||||||
|
await pm.onSongChange(fakeSong);
|
||||||
|
ts.cid = 9n;
|
||||||
|
await pm.onChannelMoved(9n);
|
||||||
|
expect(channelEdits()).toEqual([]);
|
||||||
|
});
|
||||||
|
|
||||||
|
it("an event for the channel the description is already in is a no-op", async () => {
|
||||||
|
const pm = new BotProfileManager(ts as any, noopLogger, cfgChannelDesc, "Bot");
|
||||||
|
await pm.onSongChange(fakeSong);
|
||||||
|
await pm.onChannelMoved(5n);
|
||||||
|
expect(channelEdits()).toHaveLength(1);
|
||||||
|
});
|
||||||
|
});
|
||||||
+47
-5
@@ -32,6 +32,13 @@ export class BotProfileManager {
|
|||||||
* pushed immediately (idle) or wait for the next stop event (playing).
|
* pushed immediately (idle) or wait for the next stop event (playing).
|
||||||
*/
|
*/
|
||||||
private currentSong: QueuedSong | null = null;
|
private currentSong: QueuedSong | null = null;
|
||||||
|
/**
|
||||||
|
* Channel whose description currently holds our now-playing text, or null
|
||||||
|
* if we have not written one. Remembered so that when the bot is moved we
|
||||||
|
* can still clean up the channel it was taken out of (#159) — by then
|
||||||
|
* getChannelId() already reports the new channel.
|
||||||
|
*/
|
||||||
|
private channelDescCid: bigint | null = null;
|
||||||
|
|
||||||
/** Per-feature permission-denied flags. Reset on reconnect. */
|
/** Per-feature permission-denied flags. Reset on reconnect. */
|
||||||
private permDenied = {
|
private permDenied = {
|
||||||
@@ -134,6 +141,8 @@ export class BotProfileManager {
|
|||||||
onConnect(): void {
|
onConnect(): void {
|
||||||
this.generation++;
|
this.generation++;
|
||||||
this.currentSong = null;
|
this.currentSong = null;
|
||||||
|
// Channel ids are per-server; never carry one across a (re)connect.
|
||||||
|
this.channelDescCid = null;
|
||||||
this.permDenied = {
|
this.permDenied = {
|
||||||
avatar: false,
|
avatar: false,
|
||||||
description: false,
|
description: false,
|
||||||
@@ -150,6 +159,31 @@ export class BotProfileManager {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Called when the bot itself has been moved to another channel (#159).
|
||||||
|
* Clears the now-playing text from the channel it left and, if a song is
|
||||||
|
* playing, writes it to the channel it is in now.
|
||||||
|
*/
|
||||||
|
async onChannelMoved(newChannelId: bigint): Promise<void> {
|
||||||
|
if (!this.config.channelDescEnabled || this.permDenied.channelDesc) return;
|
||||||
|
const oldChannelId = this.channelDescCid;
|
||||||
|
if (oldChannelId === newChannelId) return;
|
||||||
|
try {
|
||||||
|
if (oldChannelId !== null) {
|
||||||
|
await this.tsClient.sendCommandNoWait(
|
||||||
|
`channeledit cid=${oldChannelId} channel_description=`,
|
||||||
|
);
|
||||||
|
this.channelDescCid = null;
|
||||||
|
}
|
||||||
|
} catch (err) {
|
||||||
|
this.handleFeatureError("channelDesc", err);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
if (this.currentSong) {
|
||||||
|
await this.updateChannelDescription(this.currentSong, newChannelId);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
getConfig(): ProfileConfig {
|
getConfig(): ProfileConfig {
|
||||||
return { ...this.config };
|
return { ...this.config };
|
||||||
}
|
}
|
||||||
@@ -400,18 +434,25 @@ export class BotProfileManager {
|
|||||||
return str.slice(0, end) + ellipsis;
|
return str.slice(0, end) + ellipsis;
|
||||||
}
|
}
|
||||||
|
|
||||||
private async updateChannelDescription(song: QueuedSong | null): Promise<void> {
|
private async updateChannelDescription(
|
||||||
|
song: QueuedSong | null,
|
||||||
|
targetChannelId?: bigint,
|
||||||
|
): Promise<void> {
|
||||||
if (!this.config.channelDescEnabled || this.permDenied.channelDesc) return;
|
if (!this.config.channelDescEnabled || this.permDenied.channelDesc) return;
|
||||||
try {
|
try {
|
||||||
const channelId = this.tsClient.getChannelId();
|
const channelId = targetChannelId ?? this.tsClient.getChannelId();
|
||||||
if (channelId === 0n) return; // unknown channel
|
|
||||||
|
|
||||||
if (!song) {
|
if (!song) {
|
||||||
|
// Prefer the channel we actually wrote to, in case a move event
|
||||||
|
// was missed and the bot is somewhere else now.
|
||||||
|
const target = this.channelDescCid ?? channelId;
|
||||||
|
if (target === 0n) return; // unknown channel
|
||||||
await this.tsClient.sendCommandNoWait(
|
await this.tsClient.sendCommandNoWait(
|
||||||
`channeledit cid=${channelId} channel_description=`,
|
`channeledit cid=${target} channel_description=`,
|
||||||
);
|
);
|
||||||
|
this.channelDescCid = null;
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
if (channelId === 0n) return; // unknown channel
|
||||||
|
|
||||||
const lines = [
|
const lines = [
|
||||||
`\u266A \u6B63\u5728\u64AD\u653E: ${song.name} - ${song.artist}`, // ♪ 正在播放:
|
`\u266A \u6B63\u5728\u64AD\u653E: ${song.name} - ${song.artist}`, // ♪ 正在播放:
|
||||||
@@ -422,6 +463,7 @@ export class BotProfileManager {
|
|||||||
await this.tsClient.sendCommandNoWait(
|
await this.tsClient.sendCommandNoWait(
|
||||||
`channeledit cid=${channelId} channel_description=${escapeTS3(desc)}`,
|
`channeledit cid=${channelId} channel_description=${escapeTS3(desc)}`,
|
||||||
);
|
);
|
||||||
|
this.channelDescCid = channelId;
|
||||||
} catch (err) {
|
} catch (err) {
|
||||||
this.handleFeatureError("channelDesc", err);
|
this.handleFeatureError("channelDesc", err);
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
import { describe, it, expect } from "vitest";
|
import { describe, it, expect } from "vitest";
|
||||||
import { parseSongRef, parseSelectionIndex, parsePlaylistRef, findShareShortLink, resolveShareLink } from "./song-ref.js";
|
import { parseSongRef, parseSelectionIndex } from "./song-ref.js";
|
||||||
|
|
||||||
describe("parseSongRef (#90 exact-song selection)", () => {
|
describe("parseSongRef (#90 exact-song selection)", () => {
|
||||||
it("returns null for a plain search term", () => {
|
it("returns null for a plain search term", () => {
|
||||||
@@ -120,65 +120,3 @@ describe("parseSelectionIndex (#90 pick from last search)", () => {
|
|||||||
expect(parseSelectionIndex("")).toBeNull();
|
expect(parseSelectionIndex("")).toBeNull();
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
|
|
||||||
describe("parsePlaylistRef (#160 play a playlist from its link)", () => {
|
|
||||||
it("returns null for a playlist name or a bare id (caller keeps its old logic)", () => {
|
|
||||||
expect(parsePlaylistRef("华语经典")).toBeNull();
|
|
||||||
expect(parsePlaylistRef("2829883282")).toBeNull();
|
|
||||||
expect(parsePlaylistRef("")).toBeNull();
|
|
||||||
});
|
|
||||||
|
|
||||||
it("parses NetEase playlist URLs (web, hash route, mobile share)", () => {
|
|
||||||
expect(parsePlaylistRef("https://music.163.com/playlist?id=2829883282")).toEqual({ id: "2829883282", platform: "netease" });
|
|
||||||
expect(parsePlaylistRef("https://music.163.com/#/playlist?id=2829883282")).toEqual({ id: "2829883282", platform: "netease" });
|
|
||||||
expect(parsePlaylistRef("https://y.music.163.com/m/playlist?id=2829883282&userid=77&creatorId=77")).toEqual({ id: "2829883282", platform: "netease" });
|
|
||||||
expect(parsePlaylistRef("https://music.163.com/playlist/2829883282")).toEqual({ id: "2829883282", platform: "netease" });
|
|
||||||
});
|
|
||||||
|
|
||||||
it("does not mistake a NetEase userid= for the playlist id", () => {
|
|
||||||
expect(parsePlaylistRef("https://music.163.com/playlist?userid=77&id=123")).toEqual({ id: "123", platform: "netease" });
|
|
||||||
});
|
|
||||||
|
|
||||||
it("parses QQ Music playlist URLs", () => {
|
|
||||||
expect(parsePlaylistRef("https://y.qq.com/n/ryqq/playlist/8052190267")).toEqual({ id: "8052190267", platform: "qq" });
|
|
||||||
expect(parsePlaylistRef("https://i.y.qq.com/n2/m/share/details/taoge.html?platform=11&appshare=android_qq&hosteuin=abc&id=8052190267&appversion=13")).toEqual({ id: "8052190267", platform: "qq" });
|
|
||||||
});
|
|
||||||
|
|
||||||
it("parses YouTube playlist URLs by their list= id", () => {
|
|
||||||
expect(parsePlaylistRef("https://www.youtube.com/playlist?list=PLx0sYbCqOb8TBPRdmBHs5Iftvv9TPboYG")).toEqual({ id: "PLx0sYbCqOb8TBPRdmBHs5Iftvv9TPboYG", platform: "youtube" });
|
|
||||||
expect(parsePlaylistRef("https://youtu.be/abc?list=PLabc-_1")).toEqual({ id: "PLabc-_1", platform: "youtube" });
|
|
||||||
});
|
|
||||||
|
|
||||||
it("unwraps the [URL] BBCode the TeamSpeak client adds to pasted links", () => {
|
|
||||||
expect(parsePlaylistRef("[URL]https://y.qq.com/n/ryqq/playlist/8052190267[/URL]")).toEqual({ id: "8052190267", platform: "qq" });
|
|
||||||
});
|
|
||||||
|
|
||||||
it("finds the link inside an app's share text", () => {
|
|
||||||
expect(parsePlaylistRef("分享某人创建的歌单「深夜」: https://y.music.163.com/m/playlist?id=123&userid=77 (来自@网易云音乐)")).toEqual({ id: "123", platform: "netease" });
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("findShareShortLink (#160)", () => {
|
|
||||||
it("finds NetEase and QQ app short links, even inside share text or BBCode", () => {
|
|
||||||
expect(findShareShortLink("歌单「深夜」: https://163cn.tv/Abc123 (来自@网易云音乐)")).toBe("https://163cn.tv/Abc123");
|
|
||||||
expect(findShareShortLink("[URL]https://c6.y.qq.com/base/fcgi-bin/u?__=AbCd12[/URL]")).toBe("https://c6.y.qq.com/base/fcgi-bin/u?__=AbCd12");
|
|
||||||
});
|
|
||||||
|
|
||||||
it("ignores every other host, so we never fetch arbitrary user-supplied URLs", () => {
|
|
||||||
expect(findShareShortLink("https://evil.example/163cn.tv/Abc")).toBeNull();
|
|
||||||
expect(findShareShortLink("http://127.0.0.1:8080/x")).toBeNull();
|
|
||||||
expect(findShareShortLink("华语经典")).toBeNull();
|
|
||||||
});
|
|
||||||
});
|
|
||||||
|
|
||||||
describe("resolveShareLink (#160)", () => {
|
|
||||||
it("returns the redirect target", async () => {
|
|
||||||
const get = async () => ({ status: 302, location: "https://music.163.com/playlist?id=123" });
|
|
||||||
expect(await resolveShareLink("https://163cn.tv/Abc", get)).toBe("https://music.163.com/playlist?id=123");
|
|
||||||
});
|
|
||||||
|
|
||||||
it("returns null when there is no redirect or the request fails", async () => {
|
|
||||||
expect(await resolveShareLink("https://163cn.tv/Abc", async () => ({ status: 200, location: undefined }))).toBeNull();
|
|
||||||
expect(await resolveShareLink("https://163cn.tv/Abc", async () => { throw new Error("boom"); })).toBeNull();
|
|
||||||
});
|
|
||||||
});
|
|
||||||
@@ -1,5 +1,3 @@
|
|||||||
import axios from "axios";
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Parsing helpers for picking an EXACT song in a !play / !add / !playnext query,
|
* Parsing helpers for picking an EXACT song in a !play / !add / !playnext query,
|
||||||
* so same-name songs can be disambiguated instead of always getting the single
|
* so same-name songs can be disambiguated instead of always getting the single
|
||||||
@@ -100,82 +98,3 @@ export function parseSelectionIndex(raw: string): number | null {
|
|||||||
const n = parseInt(m[1], 10);
|
const n = parseInt(m[1], 10);
|
||||||
return Number.isFinite(n) && n > 0 ? n : null;
|
return Number.isFinite(n) && n > 0 ? n : null;
|
||||||
}
|
}
|
||||||
|
|
||||||
export interface PlaylistRef {
|
|
||||||
id: string;
|
|
||||||
platform: "netease" | "qq" | "youtube";
|
|
||||||
}
|
|
||||||
|
|
||||||
/** Drop the [URL]…[/URL] BBCode the TeamSpeak client wraps around pasted links. */
|
|
||||||
function stripUrlBBCode(text: string): string {
|
|
||||||
return text.replace(/\[\/?url(?:=[^\]]*)?\]/gi, " ");
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Detect a playlist URL (#160) — a web link, or the full link inside an app's
|
|
||||||
* share text. The platform comes from the URL, so a QQ link works without
|
|
||||||
* `-q`. Returns `null` for anything else (a playlist name or bare id), which
|
|
||||||
* the caller handles as before.
|
|
||||||
*/
|
|
||||||
export function parsePlaylistRef(raw: string): PlaylistRef | null {
|
|
||||||
const q = stripUrlBBCode(raw ?? "").trim();
|
|
||||||
if (!q) return null;
|
|
||||||
|
|
||||||
if (/music\.163\.com/i.test(q)) {
|
|
||||||
const m = /[?&#/]id=(\d+)/.exec(q) ?? /\/playlist\/(\d+)/.exec(q);
|
|
||||||
if (m) return { id: m[1], platform: "netease" };
|
|
||||||
}
|
|
||||||
|
|
||||||
if (/y\.qq\.com/i.test(q)) {
|
|
||||||
const m = /\/playlist\/(\d+)/.exec(q) ?? /[?&](?:id|disstid)=(\d+)/.exec(q);
|
|
||||||
if (m) return { id: m[1], platform: "qq" };
|
|
||||||
}
|
|
||||||
|
|
||||||
if (/youtube\.com|youtu\.be/i.test(q)) {
|
|
||||||
const m = /[?&]list=([\w-]+)/.exec(q);
|
|
||||||
if (m) return { id: m[1], platform: "youtube" };
|
|
||||||
}
|
|
||||||
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
|
|
||||||
/**
|
|
||||||
* Find a NetEase (163cn.tv) or QQ Music (c6.y.qq.com/base/fcgi-bin/u) share
|
|
||||||
* short link — what the phone apps copy. Only these hosts are recognized so
|
|
||||||
* the bot never fetches an arbitrary user-supplied URL.
|
|
||||||
*/
|
|
||||||
export function findShareShortLink(raw: string): string | null {
|
|
||||||
const q = stripUrlBBCode(raw ?? "");
|
|
||||||
const m =
|
|
||||||
/https?:\/\/163cn\.(?:tv|link)\/[0-9A-Za-z]+/i.exec(q) ??
|
|
||||||
/https?:\/\/c\d*\.y\.qq\.com\/base\/fcgi-bin\/u\?__=[0-9A-Za-z]+/i.exec(q);
|
|
||||||
return m ? m[0] : null;
|
|
||||||
}
|
|
||||||
|
|
||||||
type RedirectGet = (url: string) => Promise<{ status: number; location: string | undefined }>;
|
|
||||||
|
|
||||||
const redirectGet: RedirectGet = async (url) => {
|
|
||||||
const res = await axios.get(url, {
|
|
||||||
maxRedirects: 0,
|
|
||||||
timeout: 5000,
|
|
||||||
validateStatus: () => true,
|
|
||||||
responseType: "stream",
|
|
||||||
});
|
|
||||||
res.data?.destroy?.();
|
|
||||||
const location = res.headers.location;
|
|
||||||
return { status: res.status, location: typeof location === "string" ? location : undefined };
|
|
||||||
};
|
|
||||||
|
|
||||||
/** Follow a share short link one hop. Returns the target URL, or null. */
|
|
||||||
export async function resolveShareLink(
|
|
||||||
url: string,
|
|
||||||
get: RedirectGet = redirectGet,
|
|
||||||
): Promise<string | null> {
|
|
||||||
try {
|
|
||||||
const { status, location } = await get(url);
|
|
||||||
if (status < 300 || status >= 400 || !location) return null;
|
|
||||||
return new URL(location, url).toString();
|
|
||||||
} catch {
|
|
||||||
return null;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Reference in new issue
Block a user