style: format media music implementation
This commit is contained in:
@@ -2,13 +2,13 @@ import { AppError } from "../errors";
|
||||
import { discordPlayer } from "../player";
|
||||
import { MediaQueue } from "./mediaQueue";
|
||||
import { resolveMediaSource } from "./mediaResolver";
|
||||
import { createMusicPlayer } from "./musicPlayer";
|
||||
import type {
|
||||
MediaState,
|
||||
MusicPlayback,
|
||||
MusicPlayer,
|
||||
ResolvedMediaSource,
|
||||
} from "./mediaTypes";
|
||||
import { createMusicPlayer } from "./musicPlayer";
|
||||
|
||||
export interface MediaControllerDependencies {
|
||||
isVoiceConnected?: () => boolean;
|
||||
@@ -39,9 +39,9 @@ export class MediaController {
|
||||
|
||||
async queue(source: string): Promise<MediaState> {
|
||||
this.assertCanStart();
|
||||
const resolved = await (this.dependencies.resolveMediaSource ?? resolveMediaSource)(
|
||||
source,
|
||||
);
|
||||
const resolved = await (
|
||||
this.dependencies.resolveMediaSource ?? resolveMediaSource
|
||||
)(source);
|
||||
this.queueStore.add(resolved);
|
||||
this.startNextIfIdle();
|
||||
return this.emitState();
|
||||
@@ -49,7 +49,11 @@ export class MediaController {
|
||||
|
||||
async skip(): Promise<MediaState> {
|
||||
if (this.skipInProgress) {
|
||||
throw new AppError("Skip already in progress", "MEDIA_SKIP_IN_PROGRESS", 409);
|
||||
throw new AppError(
|
||||
"Skip already in progress",
|
||||
"MEDIA_SKIP_IN_PROGRESS",
|
||||
409,
|
||||
);
|
||||
}
|
||||
|
||||
this.skipInProgress = true;
|
||||
@@ -74,8 +78,8 @@ export class MediaController {
|
||||
}
|
||||
|
||||
private assertCanStart(): void {
|
||||
const isVoiceConnected = this.dependencies.isVoiceConnected ??
|
||||
(() => discordPlayer.isConnected());
|
||||
const isVoiceConnected =
|
||||
this.dependencies.isVoiceConnected ?? (() => discordPlayer.isConnected());
|
||||
if (!isVoiceConnected()) {
|
||||
throw new AppError(
|
||||
"Connect to a voice channel before playing media",
|
||||
|
||||
@@ -56,4 +56,4 @@ export class MediaQueue {
|
||||
queue: this.items.map((item) => ({ ...item })),
|
||||
};
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -37,4 +37,4 @@ export interface DiscordAudioPlayer {
|
||||
isConnected(): boolean;
|
||||
playStream(stream: Readable): void;
|
||||
stop(): void;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -1,5 +1,5 @@
|
||||
import { spawn as nodeSpawn } from "node:child_process";
|
||||
import type { ChildProcessWithoutNullStreams } from "node:child_process";
|
||||
import { spawn as nodeSpawn } from "node:child_process";
|
||||
import { discordPlayer } from "../player";
|
||||
import type {
|
||||
DiscordAudioPlayer,
|
||||
|
||||
@@ -79,4 +79,4 @@ export function createBroadcaster() {
|
||||
};
|
||||
}
|
||||
|
||||
export type ModerationBroadcaster = ReturnType<typeof createBroadcaster>;
|
||||
export type ModerationBroadcaster = ReturnType<typeof createBroadcaster>;
|
||||
|
||||
@@ -128,4 +128,4 @@ export interface AnalysisQueueStatus {
|
||||
queuedConversations: number;
|
||||
activeRequests: number;
|
||||
lastError: string | null;
|
||||
}
|
||||
}
|
||||
|
||||
@@ -23,7 +23,11 @@ export function createMediaRoutes(controller: MediaRouteController): Router {
|
||||
try {
|
||||
const { source } = req.body as { source?: string };
|
||||
if (!source) {
|
||||
throw new AppError("Media source is required", "MISSING_MEDIA_SOURCE", 400);
|
||||
throw new AppError(
|
||||
"Media source is required",
|
||||
"MISSING_MEDIA_SOURCE",
|
||||
400,
|
||||
);
|
||||
}
|
||||
res.json(await controller.queue(source));
|
||||
} catch (error) {
|
||||
@@ -48,4 +52,4 @@ export function createMediaRoutes(controller: MediaRouteController): Router {
|
||||
});
|
||||
|
||||
return router;
|
||||
}
|
||||
}
|
||||
|
||||
+7
-2
@@ -8,6 +8,7 @@ import * as prism from "prism-media";
|
||||
import { WebSocketServer } from "ws";
|
||||
import { AppError } from "./errors";
|
||||
import { createChildLogger, logger } from "./logger";
|
||||
import { MediaController } from "./media/mediaController";
|
||||
import { getMetrics, uptimeGauge } from "./metrics";
|
||||
import { createBroadcaster } from "./moderation/broadcaster";
|
||||
import type { ModerationBroadcaster } from "./moderation/types";
|
||||
@@ -19,7 +20,6 @@ import { createMessageRoutes } from "./routes/messageRoutes";
|
||||
import { createSyncRoutes } from "./routes/syncRoutes";
|
||||
import { createUIStateRoutes } from "./routes/uiStateRoutes";
|
||||
import { createVoiceRoutes } from "./routes/voiceRoutes";
|
||||
import { MediaController } from "./media/mediaController";
|
||||
import type { VoiceController } from "./voiceController";
|
||||
|
||||
const wsLogger = createChildLogger("webserver");
|
||||
@@ -378,7 +378,12 @@ export async function startWebserver(
|
||||
}),
|
||||
);
|
||||
ws.send(JSON.stringify({ type: "ui_state", state: getSharedUIState() }));
|
||||
ws.send(JSON.stringify({ type: "media_state", state: mediaController.getState() }));
|
||||
ws.send(
|
||||
JSON.stringify({
|
||||
type: "media_state",
|
||||
state: mediaController.getState(),
|
||||
}),
|
||||
);
|
||||
|
||||
ws.on("message", (data: Buffer | ArrayBuffer | Buffer[]) => {
|
||||
if (!Buffer.isBuffer(data)) return;
|
||||
|
||||
@@ -1,7 +1,11 @@
|
||||
import { describe, expect, it, vi } from "vitest";
|
||||
import { AppError } from "../../src/errors";
|
||||
import { MediaController } from "../../src/media/mediaController";
|
||||
import type { MusicPlayback, MusicPlayer, ResolvedMediaSource } from "../../src/media/mediaTypes";
|
||||
import type {
|
||||
MusicPlayback,
|
||||
MusicPlayer,
|
||||
ResolvedMediaSource,
|
||||
} from "../../src/media/mediaTypes";
|
||||
|
||||
function deferred() {
|
||||
let resolve!: () => void;
|
||||
@@ -26,7 +30,9 @@ describe("MediaController", () => {
|
||||
musicPlayer: { play: vi.fn() },
|
||||
});
|
||||
|
||||
await expect(controller.queue("https://example.com/song.mp3")).rejects.toMatchObject({
|
||||
await expect(
|
||||
controller.queue("https://example.com/song.mp3"),
|
||||
).rejects.toMatchObject({
|
||||
code: "VOICE_NOT_CONNECTED",
|
||||
statusCode: 409,
|
||||
} satisfies Partial<AppError>);
|
||||
@@ -40,7 +46,9 @@ describe("MediaController", () => {
|
||||
musicPlayer: { play: vi.fn() },
|
||||
});
|
||||
|
||||
await expect(controller.queue("https://example.com/song.mp3")).rejects.toMatchObject({
|
||||
await expect(
|
||||
controller.queue("https://example.com/song.mp3"),
|
||||
).rejects.toMatchObject({
|
||||
code: "BROWSER_STREAM_ACTIVE",
|
||||
statusCode: 409,
|
||||
} satisfies Partial<AppError>);
|
||||
@@ -94,7 +102,10 @@ describe("MediaController", () => {
|
||||
const musicPlayer: MusicPlayer = {
|
||||
play: vi
|
||||
.fn()
|
||||
.mockReturnValueOnce({ done: new Promise<void>(() => {}), stop: currentStop })
|
||||
.mockReturnValueOnce({
|
||||
done: new Promise<void>(() => {}),
|
||||
stop: currentStop,
|
||||
})
|
||||
.mockReturnValueOnce({ done: nextPlayback.promise, stop: vi.fn() }),
|
||||
};
|
||||
const controller = new MediaController({
|
||||
@@ -170,7 +181,9 @@ describe("MediaController", () => {
|
||||
isVoiceConnected: () => true,
|
||||
isBrowserStreaming: () => false,
|
||||
resolveMediaSource: async (input) => source(input),
|
||||
musicPlayer: { play: vi.fn(() => ({ done: new Promise<void>(() => {}), stop })) },
|
||||
musicPlayer: {
|
||||
play: vi.fn(() => ({ done: new Promise<void>(() => {}), stop })),
|
||||
},
|
||||
});
|
||||
await controller.queue("https://example.com/song.mp3");
|
||||
|
||||
@@ -186,7 +199,9 @@ describe("MediaController", () => {
|
||||
isVoiceConnected: () => true,
|
||||
isBrowserStreaming: () => false,
|
||||
resolveMediaSource: async (input) => source(input),
|
||||
musicPlayer: { play: vi.fn(() => ({ done: new Promise(() => {}), stop: vi.fn() })) },
|
||||
musicPlayer: {
|
||||
play: vi.fn(() => ({ done: new Promise(() => {}), stop: vi.fn() })),
|
||||
},
|
||||
onStateChange,
|
||||
});
|
||||
|
||||
|
||||
@@ -2,7 +2,9 @@ import { describe, expect, it } from "vitest";
|
||||
import { MediaQueue } from "../../src/media/mediaQueue";
|
||||
import type { ResolvedMediaSource } from "../../src/media/mediaTypes";
|
||||
|
||||
function source(overrides: Partial<ResolvedMediaSource> = {}): ResolvedMediaSource {
|
||||
function source(
|
||||
overrides: Partial<ResolvedMediaSource> = {},
|
||||
): ResolvedMediaSource {
|
||||
return {
|
||||
source: "https://example.com/audio.ogg",
|
||||
title: "audio.ogg",
|
||||
@@ -13,7 +15,10 @@ function source(overrides: Partial<ResolvedMediaSource> = {}): ResolvedMediaSour
|
||||
|
||||
describe("MediaQueue", () => {
|
||||
it("adds items with stable queue metadata", () => {
|
||||
const queue = new MediaQueue(() => "item-1", () => 1700000000000);
|
||||
const queue = new MediaQueue(
|
||||
() => "item-1",
|
||||
() => 1700000000000,
|
||||
);
|
||||
|
||||
const item = queue.add(source(), "tester");
|
||||
|
||||
@@ -31,7 +36,10 @@ describe("MediaQueue", () => {
|
||||
});
|
||||
|
||||
it("marks the next queued item as playing", () => {
|
||||
const queue = new MediaQueue(() => "item-1", () => 1700000000000);
|
||||
const queue = new MediaQueue(
|
||||
() => "item-1",
|
||||
() => 1700000000000,
|
||||
);
|
||||
const item = queue.add(source(), "tester");
|
||||
|
||||
expect(queue.startNext()).toEqual({ ...item, status: "playing" });
|
||||
@@ -43,7 +51,10 @@ describe("MediaQueue", () => {
|
||||
|
||||
it("removes current item and starts following item", () => {
|
||||
let id = 0;
|
||||
const queue = new MediaQueue(() => `item-${++id}`, () => 1700000000000);
|
||||
const queue = new MediaQueue(
|
||||
() => `item-${++id}`,
|
||||
() => 1700000000000,
|
||||
);
|
||||
queue.add(source({ title: "first" }), "tester");
|
||||
queue.add(source({ title: "second" }), "tester");
|
||||
queue.startNext();
|
||||
@@ -56,7 +67,10 @@ describe("MediaQueue", () => {
|
||||
});
|
||||
|
||||
it("returns the failed current item", () => {
|
||||
const queue = new MediaQueue(() => "item-1", () => 1700000000000);
|
||||
const queue = new MediaQueue(
|
||||
() => "item-1",
|
||||
() => 1700000000000,
|
||||
);
|
||||
const item = queue.add(source(), "tester");
|
||||
queue.startNext();
|
||||
|
||||
@@ -65,7 +79,10 @@ describe("MediaQueue", () => {
|
||||
});
|
||||
|
||||
it("clears current and queued items", () => {
|
||||
const queue = new MediaQueue(() => "item-1", () => 1700000000000);
|
||||
const queue = new MediaQueue(
|
||||
() => "item-1",
|
||||
() => 1700000000000,
|
||||
);
|
||||
queue.add(source(), "tester");
|
||||
queue.startNext();
|
||||
|
||||
|
||||
@@ -7,7 +7,9 @@ import { resolveMediaSource } from "../../src/media/mediaResolver";
|
||||
|
||||
describe("resolveMediaSource", () => {
|
||||
it("accepts http URLs", async () => {
|
||||
await expect(resolveMediaSource("https://example.com/music.mp3")).resolves.toEqual({
|
||||
await expect(
|
||||
resolveMediaSource("https://example.com/music.mp3"),
|
||||
).resolves.toEqual({
|
||||
source: "https://example.com/music.mp3",
|
||||
title: "music.mp3",
|
||||
kind: "url",
|
||||
@@ -43,14 +45,18 @@ describe("resolveMediaSource", () => {
|
||||
});
|
||||
|
||||
it("rejects unsupported sources", async () => {
|
||||
await expect(resolveMediaSource("not a url or file")).rejects.toMatchObject({
|
||||
code: "UNSUPPORTED_MEDIA_SOURCE",
|
||||
statusCode: 400,
|
||||
} satisfies Partial<AppError>);
|
||||
await expect(resolveMediaSource("not a url or file")).rejects.toMatchObject(
|
||||
{
|
||||
code: "UNSUPPORTED_MEDIA_SOURCE",
|
||||
statusCode: 400,
|
||||
} satisfies Partial<AppError>,
|
||||
);
|
||||
});
|
||||
|
||||
it("rejects non-http URL sources", async () => {
|
||||
await expect(resolveMediaSource("file:///tmp/song.mp3")).rejects.toMatchObject({
|
||||
await expect(
|
||||
resolveMediaSource("file:///tmp/song.mp3"),
|
||||
).rejects.toMatchObject({
|
||||
code: "UNSUPPORTED_MEDIA_SOURCE",
|
||||
statusCode: 400,
|
||||
} satisfies Partial<AppError>);
|
||||
|
||||
@@ -1,8 +1,8 @@
|
||||
import { EventEmitter } from "node:events";
|
||||
import { PassThrough } from "node:stream";
|
||||
import { describe, expect, it, vi } from "vitest";
|
||||
import { createMusicPlayer } from "../../src/media/musicPlayer";
|
||||
import type { DiscordAudioPlayer } from "../../src/media/mediaTypes";
|
||||
import { createMusicPlayer } from "../../src/media/musicPlayer";
|
||||
|
||||
class FakeProcess extends EventEmitter {
|
||||
stdout = new PassThrough();
|
||||
@@ -34,23 +34,27 @@ describe("createMusicPlayer", () => {
|
||||
proc.emit("close", 0);
|
||||
await playback.done;
|
||||
|
||||
expect(spawn).toHaveBeenCalledWith("ffmpeg", [
|
||||
"-hide_banner",
|
||||
"-loglevel",
|
||||
"warning",
|
||||
"-i",
|
||||
"https://example.com/song.mp3",
|
||||
"-vn",
|
||||
"-acodec",
|
||||
"libopus",
|
||||
"-ar",
|
||||
"48000",
|
||||
"-ac",
|
||||
"2",
|
||||
"-f",
|
||||
"ogg",
|
||||
"pipe:1",
|
||||
], { stdio: ["ignore", "pipe", "pipe"] });
|
||||
expect(spawn).toHaveBeenCalledWith(
|
||||
"ffmpeg",
|
||||
[
|
||||
"-hide_banner",
|
||||
"-loglevel",
|
||||
"warning",
|
||||
"-i",
|
||||
"https://example.com/song.mp3",
|
||||
"-vn",
|
||||
"-acodec",
|
||||
"libopus",
|
||||
"-ar",
|
||||
"48000",
|
||||
"-ac",
|
||||
"2",
|
||||
"-f",
|
||||
"ogg",
|
||||
"pipe:1",
|
||||
],
|
||||
{ stdio: ["ignore", "pipe", "pipe"] },
|
||||
);
|
||||
expect(discordPlayer.playStream).toHaveBeenCalledWith(proc.stdout);
|
||||
});
|
||||
|
||||
@@ -64,7 +68,11 @@ describe("createMusicPlayer", () => {
|
||||
const player = createMusicPlayer({ spawn, discordPlayer });
|
||||
|
||||
expect(() =>
|
||||
player.play({ source: "/tmp/song.ogg", title: "song.ogg", kind: "local" }),
|
||||
player.play({
|
||||
source: "/tmp/song.ogg",
|
||||
title: "song.ogg",
|
||||
kind: "local",
|
||||
}),
|
||||
).toThrow("Discord audio player is not connected");
|
||||
expect(spawn).not.toHaveBeenCalled();
|
||||
});
|
||||
@@ -76,9 +84,16 @@ describe("createMusicPlayer", () => {
|
||||
playStream: vi.fn(),
|
||||
stop: vi.fn(),
|
||||
};
|
||||
const player = createMusicPlayer({ spawn: vi.fn(() => proc), discordPlayer });
|
||||
const player = createMusicPlayer({
|
||||
spawn: vi.fn(() => proc),
|
||||
discordPlayer,
|
||||
});
|
||||
|
||||
const playback = player.play({ source: "/tmp/song.ogg", title: "song.ogg", kind: "local" });
|
||||
const playback = player.play({
|
||||
source: "/tmp/song.ogg",
|
||||
title: "song.ogg",
|
||||
kind: "local",
|
||||
});
|
||||
playback.stop();
|
||||
playback.stop();
|
||||
|
||||
@@ -86,4 +101,4 @@ describe("createMusicPlayer", () => {
|
||||
expect(proc.kill).toHaveBeenCalledWith("SIGTERM");
|
||||
expect(discordPlayer.stop).toHaveBeenCalledTimes(1);
|
||||
});
|
||||
});
|
||||
});
|
||||
|
||||
@@ -2,7 +2,11 @@ import type { Request, Response } from "express";
|
||||
import { describe, expect, it, vi } from "vitest";
|
||||
import { createMediaRoutes } from "../../src/routes/mediaRoutes";
|
||||
|
||||
function getHandler(router: ReturnType<typeof createMediaRoutes>, path: string, method: string) {
|
||||
function getHandler(
|
||||
router: ReturnType<typeof createMediaRoutes>,
|
||||
path: string,
|
||||
method: string,
|
||||
) {
|
||||
const layer = router.stack.find((item) => item.route?.path === path);
|
||||
return layer?.route?.stack.find((item) => item.method === method)?.handle;
|
||||
}
|
||||
@@ -15,12 +19,20 @@ describe("createMediaRoutes", () => {
|
||||
skip: vi.fn(),
|
||||
stop: vi.fn(),
|
||||
};
|
||||
const handler = getHandler(createMediaRoutes(controller), "/media/status", "get");
|
||||
const handler = getHandler(
|
||||
createMediaRoutes(controller),
|
||||
"/media/status",
|
||||
"get",
|
||||
);
|
||||
const json = vi.fn();
|
||||
|
||||
await handler?.({} as Request, { json } as unknown as Response, vi.fn());
|
||||
|
||||
expect(json).toHaveBeenCalledWith({ playing: false, current: null, queue: [] });
|
||||
expect(json).toHaveBeenCalledWith({
|
||||
playing: false,
|
||||
current: null,
|
||||
queue: [],
|
||||
});
|
||||
});
|
||||
|
||||
it("queues a source", async () => {
|
||||
@@ -31,7 +43,11 @@ describe("createMediaRoutes", () => {
|
||||
skip: vi.fn(),
|
||||
stop: vi.fn(),
|
||||
};
|
||||
const handler = getHandler(createMediaRoutes(controller), "/media/queue", "post");
|
||||
const handler = getHandler(
|
||||
createMediaRoutes(controller),
|
||||
"/media/queue",
|
||||
"post",
|
||||
);
|
||||
const json = vi.fn();
|
||||
|
||||
await handler?.(
|
||||
@@ -40,7 +56,9 @@ describe("createMediaRoutes", () => {
|
||||
vi.fn(),
|
||||
);
|
||||
|
||||
expect(controller.queue).toHaveBeenCalledWith("https://example.com/song.mp3");
|
||||
expect(controller.queue).toHaveBeenCalledWith(
|
||||
"https://example.com/song.mp3",
|
||||
);
|
||||
expect(json).toHaveBeenCalledWith(state);
|
||||
});
|
||||
|
||||
@@ -51,7 +69,11 @@ describe("createMediaRoutes", () => {
|
||||
skip: vi.fn(),
|
||||
stop: vi.fn(),
|
||||
};
|
||||
const handler = getHandler(createMediaRoutes(controller), "/media/queue", "post");
|
||||
const handler = getHandler(
|
||||
createMediaRoutes(controller),
|
||||
"/media/queue",
|
||||
"post",
|
||||
);
|
||||
const next = vi.fn();
|
||||
|
||||
await handler?.(
|
||||
|
||||
Reference in New Issue
Block a user