ziplayer 0.2.6 → 0.2.7-dev.1

package/dist/structures/PlayerManager.js

@@ -5,6 +5,7 @@ exports.getInstance = getInstance;
 const events_1 = require("events");
 const Player_1 = require("./Player");
 const timeout_1 = require("../utils/timeout");
+const PersistenceManager_1 = require("../persistence/PersistenceManager");
 const GLOBAL_MANAGER_KEY = Symbol.for("ziplayer.PlayerManager.instance");
 const getGlobalManager = () => {
     try {
@@ -46,7 +47,9 @@ const setGlobalManager = (instance) => {
  *       nodes: [{ host: "localhost", port: 2333, password: "youshallnotpass" }]
  *     })
  *   ],
- *   extractorTimeout: 10000
+ *   extractorTimeout: 10000,
+ *   autoCleanup: true,
+ *   cleanupInterval: 60000
  * });
  *
  * // Create a player for a guild
@@ -72,7 +75,7 @@ class PlayerManager extends events_1.EventEmitter {
     }
     debug(message, ...optionalParams) {
         if (this.listenerCount("debug") > 0) {
-            this.emit("debug", message, ...optionalParams);
+            this.emit("debug", `[PlayerManager] ${message}`, ...optionalParams);
             if (!this.B_debug) {
                 this.B_debug = true;
             }
@@ -81,9 +84,18 @@ class PlayerManager extends events_1.EventEmitter {
     constructor(options = {}) {
         super();
         this.players = new Map();
+        this.SEARCH_CACHE_TTL = 60 * 1000; // 1 minute
+        this.MAX_CACHE_SIZE = 100;
+        this.cleanupInterval = null;
+        this.statsInterval = null;
         this.B_debug = false;
         this.extractorTimeout = 10000;
+        this.autoCleanup = true;
+        this.cleanupTimeout = 60000; // 1 minute
+        this.enableSearchCache = true;
         this.plugins = [];
+        this.searchCache = new Map();
+        // Initialize plugins
         const provided = options.plugins || [];
         for (const p of provided) {
             try {
@@ -94,13 +106,30 @@ class PlayerManager extends events_1.EventEmitter {
                     const instance = new p();
                     this.plugins.push(instance);
                 }
+                this.debug(`Registered plugin: ${p.name || "unnamed"}`);
             }
             catch (e) {
-                this.debug(`[PlayerManager] Failed to init plugin:`, e);
+                this.debug(`Failed to init plugin:`, e);
             }
         }
         this.extensions = options.extensions || [];
+        this.extractorTimeout = options.extractorTimeout ?? 10000;
+        this.autoCleanup = options.autoCleanup ?? true;
+        this.cleanupTimeout = options.cleanupInterval ?? 60000;
+        this.enableSearchCache = options.enableSearchCache ?? true;
+        if (options.persistence) {
+            this.initPersistence(options.persistence);
+        }
+        // Setup auto cleanup
+        if (this.autoCleanup) {
+            this.startAutoCleanup();
+        }
+        // Setup stats collection (optional)
+        if (options.enableStatsCollection) {
+            this.startStatsCollection();
+        }
         setGlobalManager(this);
+        this.debug(`Initialized with ${this.plugins.length} plugins, ${this.extensions.length} extensions`);
     }
     withTimeout(promise, message) {
         const timeout = this.extractorTimeout;
@@ -113,48 +142,108 @@ class PlayerManager extends events_1.EventEmitter {
             return guildOrId.id;
         throw new Error("Invalid guild or guildId provided.");
     }
+    getSearchCacheKey(query) {
+        return query.toLowerCase().trim();
+    }
+    getCachedSearch(query) {
+        if (!this.enableSearchCache)
+            return null;
+        const key = this.getSearchCacheKey(query);
+        const cached = this.searchCache.get(key);
+        if (cached && Date.now() < cached.expiresAt) {
+            this.debug(`[Cache] Search hit for: ${query}`);
+            return cached.data;
+        }
+        if (cached) {
+            this.searchCache.delete(key);
+        }
+        return null;
+    }
+    setCachedSearch(query, result) {
+        if (!this.enableSearchCache)
+            return;
+        // Clean up old entries if cache is too large
+        if (this.searchCache.size >= this.MAX_CACHE_SIZE) {
+            const oldest = Array.from(this.searchCache.entries()).sort((a, b) => a[1].timestamp - b[1].timestamp)[0];
+            if (oldest)
+                this.searchCache.delete(oldest[0]);
+        }
+        const key = this.getSearchCacheKey(query);
+        this.searchCache.set(key, {
+            data: result,
+            timestamp: Date.now(),
+            expiresAt: Date.now() + this.SEARCH_CACHE_TTL,
+        });
+        this.debug(`[Cache] Search stored for: ${query}`);
+    }
+    clearExpiredCache() {
+        const now = Date.now();
+        let expiredCount = 0;
+        for (const [key, entry] of this.searchCache) {
+            if (now >= entry.expiresAt) {
+                this.searchCache.delete(key);
+                expiredCount++;
+            }
+        }
+        if (expiredCount > 0) {
+            this.debug(`[Cache] Cleared ${expiredCount} expired search entries`);
+        }
+    }
+    startAutoCleanup() {
+        if (this.cleanupInterval) {
+            clearInterval(this.cleanupInterval);
+        }
+        this.cleanupInterval = setInterval(() => {
+            this.cleanupInactivePlayers();
+            this.clearExpiredCache();
+        }, this.cleanupTimeout);
+        this.debug(`Auto-cleanup started with interval: ${this.cleanupTimeout}ms`);
+    }
+    startStatsCollection() {
+        if (this.statsInterval) {
+            clearInterval(this.statsInterval);
+        }
+        this.statsInterval = setInterval(() => {
+            const stats = this.getStats();
+            this.emit("stats", stats);
+        }, 30000); // Every 30 seconds
+    }
+    cleanupInactivePlayers() {
+        let cleanedCount = 0;
+        for (const [guildId, player] of this.players) {
+            // Clean up players that are not playing and not connected
+            if (!player.isPlaying && !player.connection && player.queue.isEmpty) {
+                const idleTime = Date.now() - player._lastActivity || Date.now();
+                if (idleTime > this.cleanupTimeout) {
+                    this.debug(`Cleaning up inactive player for guild: ${guildId}`);
+                    player.destroy();
+                    this.players.delete(guildId);
+                    cleanedCount++;
+                }
+            }
+        }
+        if (cleanedCount > 0) {
+            this.debug(`Cleaned up ${cleanedCount} inactive players`);
+        }
+    }
     /**
      * Create a new player for a guild
      *
      * @param {string | {id: string}} guildOrId - Guild ID or guild object
      * @param {PlayerOptions} options - Player configuration options
      * @returns {Promise<Player>} The created player instance
-     *
-     * @example
-     * // Create player with basic options
-     * const player = await manager.create(guildId, {
-     *   tts: { interrupt: true, volume: 1 },
-     *   leaveOnEnd: true,
-     *   leaveTimeout: 30000
-     * });
-     *
-     * // Create player with advanced options
-     * const advancedPlayer = await manager.create(guild, {
-     *   volume: 0.8,
-     *   quality: "high",
-     *   selfDeaf: false,
-     *   selfMute: false,
-     *   tts: {
-     *     createPlayer: true,
-     *     interrupt: true,
-     *     volume: 1.0,
-     *     Max_Time_TTS: 30000
-     *   },
-     *   userdata: { customData: "example" }
-     * });
-     *
-     * // Connect and play immediately
-     * await player.connect(voiceChannel);
-     * await player.play("Never Gonna Give You Up", userId);
      */
     async create(guildOrId, options) {
         const guildId = this.resolveGuildId(guildOrId);
         if (this.players.has(guildId)) {
+            this.debug(`Player already exists for guildId: ${guildId}, returning existing`);
             return this.players.get(guildId);
         }
-        this.debug(`[PlayerManager] Creating player for guildId: ${guildId}`);
+        this.debug(`Creating player for guildId: ${guildId}`);
         const player = new Player_1.Player(guildId, options, this);
+        // Add all registered plugins
         this.plugins.forEach((plugin) => player.addPlugin(plugin));
+        // Activate extensions
         let extsToActivate = [];
         const optExts = options?.extensions;
         if (Array.isArray(optExts)) {
@@ -172,6 +261,10 @@ class PlayerManager extends events_1.EventEmitter {
                 extsToActivate = optExts;
             }
         }
+        else {
+            // Use all extensions by default
+            extsToActivate = this.extensions;
+        }
         for (const ext of extsToActivate) {
             let instance = ext;
             if (typeof ext === "function") {
@@ -179,7 +272,7 @@ class PlayerManager extends events_1.EventEmitter {
                     instance = new ext(player);
                 }
                 catch (e) {
-                    this.debug(`[PlayerManager] Extension constructor error:`, e);
+                    this.debug(`Extension constructor error for ${ext.name}:`, e);
                     continue;
                 }
             }
@@ -192,11 +285,11 @@ class PlayerManager extends events_1.EventEmitter {
                     let activated = true;
                     try {
                         activated = await (0, timeout_1.withTimeout)(Promise.resolve(extInstance.active({ manager: this, player })), player.options.extractorTimeout ?? 15000, `Extension ${extInstance?.name} activation timed out`);
-                        this.debug(`[PlayerManager] Extension ${extInstance?.name} active`);
+                        this.debug(`Extension ${extInstance?.name} active check returned: ${activated}`);
                     }
                     catch (e) {
                         activated = false;
-                        this.debug(`[PlayerManager] Extension activation error:`, e);
+                        this.debug(`Extension activation error for ${extInstance?.name}:`, e);
                     }
                     if (activated === false) {
                         player.detachExtension(extInstance);
@@ -205,14 +298,25 @@ class PlayerManager extends events_1.EventEmitter {
                 }
             }
         }
-        // Forward all player events
+        // Forward all player events to manager
+        this.setupEventForwarding(player, guildId);
+        // Mark last activity
+        player._lastActivity = Date.now();
+        this.players.set(guildId, player);
+        this.debug(`Player created for guildId: ${guildId}`);
+        return player;
+    }
+    setupEventForwarding(player, guildId) {
         player.on("willPlay", (track, tracks) => this.emit("willPlay", player, track, tracks));
-        player.on("trackStart", (track) => this.emit("trackStart", player, track));
+        player.on("trackStart", (track) => {
+            player._lastActivity = Date.now();
+            this.emit("trackStart", player, track);
+        });
         player.on("trackEnd", (track) => this.emit("trackEnd", player, track));
         player.on("queueEnd", () => this.emit("queueEnd", player));
         player.on("playerError", (error, track) => this.emit("playerError", player, error, track));
         player.on("connectionError", (error) => this.emit("connectionError", player, error));
-        player.on("volumeChange", (old, volume) => this.emit("volumeChange", player, old, volume));
+        player.on("volumeChange", (oldVol, newVol) => this.emit("volumeChange", player, oldVol, newVol));
         player.on("queueAdd", (track) => this.emit("queueAdd", player, track));
         player.on("queueAddList", (tracks) => this.emit("queueAddList", player, tracks));
         player.on("queueRemove", (track, index) => this.emit("queueRemove", player, track, index));
@@ -222,6 +326,7 @@ class PlayerManager extends events_1.EventEmitter {
         player.on("playerDestroy", () => {
             this.emit("playerDestroy", player);
             this.players.delete(guildId);
+            this.debug(`Player destroyed for guildId: ${guildId}`);
         });
         player.on("ttsStart", (payload) => this.emit("ttsStart", player, payload));
         player.on("ttsEnd", () => this.emit("ttsEnd", player));
@@ -230,133 +335,193 @@ class PlayerManager extends events_1.EventEmitter {
                 this.emit("debug", ...args);
             }
         });
-        this.players.set(guildId, player);
-        return player;
     }
     /**
      * Get an existing player for a guild
      *
      * @param {string | {id: string}} guildOrId - Guild ID or guild object
      * @returns {Player | undefined} The player instance or undefined if not found
-     * @example
-     * // Get player by guild ID
-     * const player = manager.get(guildId);
-     * if (player) {
-     *   await player.play("Never Gonna Give You Up", userId);
-     * } else {
-     *   console.log("No player found for this guild");
-     * }
-     *
-     * // Get player by guild object
-     * const playerFromGuild = manager.get(guild);
-     * if (playerFromGuild) {
-     *   playerFromGuild.setVolume(0.5);
-     * }
-     *
-     * // Check if player exists before using
-     * const existingPlayer = manager.get(guildId);
-     * if (existingPlayer && existingPlayer.playing) {
-     *   existingPlayer.pause();
-     * }
      */
     get(guildOrId) {
         const guildId = this.resolveGuildId(guildOrId);
-        return this.players.get(guildId);
+        const player = this.players.get(guildId);
+        if (player) {
+            player._lastActivity = Date.now();
+        }
+        return player;
     }
     /**
-     * Get an existing player for a guild
-     *
-     * @param {string | {id: string}} guildOrId - Guild ID or guild object
-     * @returns {Player | undefined} The player instance or undefined
-     * @example
-     * const player = manager.get(guildId);
-     * if (player) {
-     *   await player.play("song name", userId);
-     * }
+     * Get an existing player for a guild (alias for get)
      */
     getPlayer(guildOrId) {
-        const guildId = this.resolveGuildId(guildOrId);
-        return this.players.get(guildId);
+        return this.get(guildOrId);
     }
     /**
      * Get all players
      *
      * @returns {Player[]} All player instances
-     * @example
-     * const players = manager.getall();
-     * console.log(`Players: ${players.length}`);
      */
-    getall() {
+    getAll() {
         return Array.from(this.players.values());
     }
+    /**
+     * Alias for getAll
+     */
+    getall() {
+        return this.getAll();
+    }
+    /**
+     * Get players by filter
+     *
+     * @param {(player: Player) => boolean} filter - Filter function
+     * @returns {Player[]} Filtered player instances
+     */
+    getPlayersByFilter(filter) {
+        return this.getAll().filter(filter);
+    }
+    /**
+     * Get players in a voice channel
+     *
+     * @param {string} channelId - Voice channel ID
+     * @returns {Player[]} Players in the channel
+     */
+    getPlayersInChannel(channelId) {
+        return this.getAll().filter((p) => p.connection?.joinConfig.channelId === channelId);
+    }
     /**
      * Destroy a player and clean up resources
      *
      * @param {string | {id: string}} guildOrId - Guild ID or guild object
      * @returns {boolean} True if player was destroyed, false if not found
-     * @example
-     * // Destroy player by guild ID
-     * const destroyed = manager.delete(guildId);
-     * if (destroyed) {
-     *   console.log("Player destroyed successfully");
-     * } else {
-     *   console.log("No player found to destroy");
-     * }
-     *
-     * // Destroy player by guild object
-     * const destroyedFromGuild = manager.delete(guild);
-     * console.log(`Player destroyed: ${destroyedFromGuild}`);
-     *
-     * // Clean up all players
-     * for (const [guildId, player] of manager.players) {
-     *   const destroyed = manager.delete(guildId);
-     *   console.log(`Destroyed player for ${guildId}: ${destroyed}`);
-     * }
      */
     delete(guildOrId) {
         const guildId = this.resolveGuildId(guildOrId);
         const player = this.players.get(guildId);
         if (player) {
-            this.debug(`[PlayerManager] Deleting player for guildId: ${guildId}`);
+            this.debug(`Deleting player for guildId: ${guildId}`);
             player.destroy();
-            return this.players.delete(guildId);
+            return true;
         }
         return false;
     }
+    /**
+     * Destroy multiple players by filter
+     *
+     * @param {(player: Player) => boolean} filter - Filter function
+     * @returns {number} Number of players destroyed
+     */
+    deleteWhere(filter) {
+        const toDelete = this.getPlayersByFilter(filter);
+        let count = 0;
+        for (const player of toDelete) {
+            const guildId = player.guildId;
+            player.destroy();
+            this.players.delete(guildId);
+            count++;
+        }
+        if (count > 0) {
+            this.debug(`Deleted ${count} players by filter`);
+        }
+        return count;
+    }
     /**
      * Check if a player exists for a guild
      *
      * @param {string | {id: string}} guildOrId - Guild ID or guild object
-     * @returns {boolean} True if player exists, false if not
-     * @example
-     * const exists = manager.has(guildId);
-     * console.log(`Player exists: ${exists}`);
+     * @returns {boolean} True if player exists
      */
     has(guildOrId) {
         const guildId = this.resolveGuildId(guildOrId);
         return this.players.has(guildId);
     }
+    /**
+     * Get number of players
+     */
     get size() {
         return this.players.size;
     }
+    /**
+     * Check if debug is enabled
+     */
     get debugEnabled() {
         return this.B_debug;
     }
     /**
-     * Destroy all players
+     * Get manager statistics
+     *
+     * @returns {PlayerStats} Statistics about players
+     */
+    getStats() {
+        let activePlayers = 0;
+        let pausedPlayers = 0;
+        let connectedPlayers = 0;
+        let totalTracksInQueue = 0;
+        for (const player of this.players.values()) {
+            if (player.isPlaying)
+                activePlayers++;
+            if (player.isPaused)
+                pausedPlayers++;
+            if (player.connection)
+                connectedPlayers++;
+            totalTracksInQueue += player.queueSize;
+        }
+        return {
+            totalPlayers: this.players.size,
+            activePlayers,
+            pausedPlayers,
+            connectedPlayers,
+            totalTracksInQueue,
+        };
+    }
+    /**
+     * Broadcast an action to all players
      *
-     * @returns {void}
+     * @param {string} action - Action to perform
+     * @param {...any[]} args - Arguments for the action
      * @example
-     * manager.destroy();
-     * console.log(`All players destroyed`);
+     * manager.broadcast("setVolume", 50);
+     * manager.broadcast("pause");
+     */
+    broadcast(action, ...args) {
+        for (const player of this.players.values()) {
+            if (typeof player[action] === "function") {
+                try {
+                    player[action](...args);
+                }
+                catch (error) {
+                    this.debug(`Error broadcasting ${action} to ${player.guildId}:`, error);
+                }
+            }
+        }
+    }
+    /**
+     * Destroy all players and clean up
      */
     destroy() {
-        this.debug(`[PlayerManager] Destroying all players`);
+        this.debug(`Destroying all players`);
+        if (this.persistenceManager) {
+            this.persistenceManager.saveAll().catch((err) => {
+                this.debug("Failed to save players before destroy:", err);
+            });
+            this.persistenceManager.shutdown().catch(console.error);
+        }
+        // Stop cleanup intervals
+        if (this.cleanupInterval) {
+            clearInterval(this.cleanupInterval);
+            this.cleanupInterval = null;
+        }
+        if (this.statsInterval) {
+            clearInterval(this.statsInterval);
+            this.statsInterval = null;
+        }
+        // Destroy all players
         for (const player of this.players.values()) {
             player.destroy();
         }
         this.players.clear();
+        this.searchCache.clear();
         this.removeAllListeners();
+        this.debug(`PlayerManager destroyed`);
     }
     /**
      * Search using registered plugins without creating a Player.
@@ -364,28 +529,158 @@ class PlayerManager extends events_1.EventEmitter {
      * @param {string} query - The query to search for
      * @param {string} requestedBy - The user ID who requested the search
      * @returns {Promise<SearchResult>} The search result
-     * @example
-     * const result = await manager.search("Never Gonna Give You Up", userId);
-     * console.log(`Search result: ${result.tracks.length} tracks`);
      */
     async search(query, requestedBy) {
-        this.debug(`[PlayerManager] Search called with query: ${query}, requestedBy: ${requestedBy}`);
+        this.debug(`Search called with query: ${query}, requestedBy: ${requestedBy}`);
+        // Check cache first
+        const cached = this.getCachedSearch(query);
+        if (cached) {
+            return cached;
+        }
         const plugin = this.plugins.find((p) => p.canHandle(query));
         if (!plugin) {
-            this.debug(`[PlayerManager] No plugin found to handle: ${query}`);
+            this.debug(`No plugin found to handle: ${query}`);
             throw new Error(`No plugin found to handle: ${query}`);
         }
         try {
-            return await this.withTimeout(plugin.search(query, requestedBy), "Search operation timed out");
+            const result = await this.withTimeout(plugin.search(query, requestedBy), "Search operation timed out");
+            this.setCachedSearch(query, result);
+            return result;
         }
         catch (error) {
-            this.debug(`[PlayerManager] Search error:`, error);
+            this.debug(`Search error:`, error);
             throw error;
         }
     }
+    /**
+     * Clear search cache
+     */
+    clearSearchCache() {
+        const size = this.searchCache.size;
+        this.searchCache.clear();
+        this.debug(`Cleared ${size} search cache entries`);
+    }
+    /**
+     * Register a plugin after initialization
+     *
+     * @param {SourcePlugin} plugin - Plugin to register
+     */
+    registerPlugin(plugin) {
+        this.plugins.push(plugin);
+        this.debug(`Registered plugin: ${plugin.name}`);
+        // Register plugin with all existing players
+        for (const player of this.players.values()) {
+            player.addPlugin(plugin);
+        }
+    }
+    /**
+     * Unregister a plugin
+     *
+     * @param {string} name - Plugin name to unregister
+     * @returns {boolean} True if plugin was unregistered
+     */
+    unregisterPlugin(name) {
+        const index = this.plugins.findIndex((p) => p.name === name);
+        if (index === -1)
+            return false;
+        this.plugins.splice(index, 1);
+        this.debug(`Unregistered plugin: ${name}`);
+        // Note: Cannot easily remove plugins from existing players
+        return true;
+    }
+    /**
+     * Get all registered plugins
+     */
+    getPlugins() {
+        return [...this.plugins];
+    }
+    /**
+     * Register an extension after initialization
+     *
+     * @param {BaseExtension} extension - Extension to register
+     */
+    registerExtension(extension) {
+        this.extensions.push(extension);
+        this.debug(`Registered extension: ${extension.name}`);
+        // Register extension with all existing players
+        for (const player of this.players.values()) {
+            player.attachExtension(extension);
+        }
+    }
+    /**
+     * Get manager configuration
+     */
+    getConfig() {
+        return {
+            extractorTimeout: this.extractorTimeout,
+            autoCleanup: this.autoCleanup,
+            cleanupTimeout: this.cleanupTimeout,
+            enableSearchCache: this.enableSearchCache,
+            pluginsCount: this.plugins.length,
+            extensionsCount: this.extensions.length,
+            playersCount: this.players.size,
+        };
+    }
+    initPersistence(persistenceOptions) {
+        this.persistenceManager = new PersistenceManager_1.PersistenceManager(this, persistenceOptions);
+        // Forward persistence events
+        this.persistenceManager.on("playerSaved", (guildId) => {
+            this.emit("playerSaved", guildId);
+        });
+        this.persistenceManager.on("playerLoaded", (guildId, data) => {
+            this.emit("playerLoaded", guildId, data);
+        });
+        this.persistenceManager.on("savedAll", (results) => {
+            this.emit("savedAll", results);
+        });
+        this.persistenceManager.on("loadedAll", (results) => {
+            this.emit("loadedAll", results);
+        });
+        this.debug("Persistence manager initialized");
+    }
+    /**
+     * Get persistence manager
+     */
+    getPersistence() {
+        return this.persistenceManager;
+    }
+    /**
+     * Save all players
+     */
+    async saveAllPlayers() {
+        if (!this.persistenceManager) {
+            throw new Error("Persistence not enabled");
+        }
+        return await this.persistenceManager.saveAll();
+    }
+    /**
+     * Load all players
+     */
+    async loadAllPlayers(restorePosition = true) {
+        if (!this.persistenceManager) {
+            throw new Error("Persistence not enabled");
+        }
+        return await this.persistenceManager.loadAll(restorePosition);
+    }
+    /**
+     * Save a specific player
+     */
+    async savePlayer(guildId) {
+        if (!this.persistenceManager)
+            return false;
+        const player = this.get(guildId);
+        if (!player)
+            return false;
+        return await this.persistenceManager.savePlayer(player);
+    }
 }
 exports.PlayerManager = PlayerManager;
 PlayerManager.instance = null;
+/**
+ * Get the global PlayerManager instance
+ *
+ * @returns {PlayerManager | null} Global instance or null
+ */
 function getInstance() {
     const globalInst = (0, exports.getGlobalManager)();
     if (!globalInst) {