feat(Stream): support playing local files

skick1234 · Jun 5, 2024 · 8101244 · 8101244
1 parent dcd8a45
commit 8101244
Show file tree

Hide file tree

Showing 4 changed files with 15 additions and 9 deletions.
diff --git a/src/DisTube.ts b/src/DisTube.ts
@@ -224,7 +224,7 @@ export class DisTube extends TypedEmitter<TypedDisTubeEvents> {
     const queue = this.getQueue(voiceChannel) || (await this.queues.create(voiceChannel, textChannel));
     await queue._taskQueue.queuing();
     try {
-      this.debug(`[${queue.id}] Resolving song: ${typeof song === "string" ? song : JSON.stringify(song)}`);
+      this.debug(`[${queue.id}] Playing input: ${song}`);
       const resolved = await this.handler.resolve(song, { member, metadata });
       const isNsfw = isNsfwChannel(queue?.textChannel || textChannel);
       if (resolved instanceof Playlist) {

diff --git a/src/core/DisTubeHandler.ts b/src/core/DisTubeHandler.ts
@@ -33,7 +33,7 @@ export class DisTubeHandler extends DisTubeBase {
         const plugin =
           (await this._getPluginFromURL(input)) || (await this._getPluginFromURL(await this.followRedirectLink(input)));
         if (!plugin) throw new DisTubeError("NOT_SUPPORTED_URL");
-        this.debug(`[${plugin.constructor.name}] Resolving song from url: ${input}`);
+        this.debug(`[${plugin.constructor.name}] Resolving from url: ${input}`);
         return plugin.resolve(input, options);
       }
       try {

diff --git a/src/core/DisTubeStream.ts b/src/core/DisTubeStream.ts
@@ -1,6 +1,6 @@
 import { Transform } from "stream";
+import { DisTubeError, Events } from "..";
 import { spawn, spawnSync } from "child_process";
-import { DisTubeError, Events, isURL } from "..";
 import { TypedEmitter } from "tiny-typed-emitter";
 import { StreamType, createAudioResource } from "@discordjs/voice";
 import type { TransformCallback } from "stream";
@@ -63,12 +63,6 @@ export class DisTubeStream extends TypedEmitter<{
    * @param options - Stream options
    */
   constructor(url: string, options: StreamOptions) {
-    if (typeof url !== "string" || !isURL(url)) {
-      throw new DisTubeError("INVALID_TYPE", "an URL", url);
-    }
-    if (!options || typeof options !== "object" || Array.isArray(options)) {
-      throw new DisTubeError("INVALID_TYPE", "object", options, "options");
-    }
     super();
     const { ffmpeg, seek } = options;
     const opts: FFmpegArg = {
@@ -88,6 +82,14 @@ export class DisTubeStream extends TypedEmitter<{
 
     if (typeof seek === "number" && seek > 0) opts.ss = seek.toString();
 
+    const fileUrl = new URL(url);
+    if (fileUrl.protocol === "file:") {
+      opts.reconnect = null;
+      opts.reconnect_streamed = null;
+      opts.reconnect_delay_max = null;
+      opts.i = fileUrl.hostname + fileUrl.pathname;
+    }
+
     this.#ffmpegPath = ffmpeg.path;
     this.#opts = [
       ...Object.entries(opts)

diff --git a/src/struct/Playlist.ts b/src/struct/Playlist.ts
@@ -96,4 +96,8 @@ export class Playlist<T = unknown> implements PlaylistInfo {
     this.#metadata = metadata;
     this.songs.forEach(s => (s.metadata = metadata));
   }
+
+  toString() {
+    return `${this.name} (${this.songs.length} songs)`;
+  }
 }