discord-player-play-dl/src/Structures/ExtractorModel.ts

import { ExtractorModelData } from '../types/types';

class ExtractorModel {
    name: string;
    private _raw: any;

    /**
     * Model for raw Discord Player extractors
     * @param {String} extractorName Name of the extractor
     * @param {Object} data Extractor object
     */
    constructor(extractorName: string, data: any) {
        /**
         * The extractor name
         * @type {String}
         */
        this.name = extractorName;

        Object.defineProperty(this, '_raw', { value: data, configurable: false, writable: false, enumerable: false });
    }

    /**
     * Method to handle requests from `Player.play()`
     * @param {String} query Query to handle
     * @returns {Promise<ExtractorModelData>}
     */
    async handle(query: string): Promise<ExtractorModelData> {
        const data = await this._raw.getInfo(query);
        if (!data) return null;

        return {
            title: data.title,
            duration: data.duration,
            thumbnail: data.thumbnail,
            engine: data.engine,
            views: data.views,
            author: data.author,
            description: data.description,
            url: data.url
        };
    }

    /**
     * Method used by Discord Player to validate query with this extractor
     * @param {String} query The query to validate
     * @returns {Boolean}
     */
    validate(query: string): boolean {
        return Boolean(this._raw.validate(query));
    }

    /**
     * The extractor version
     * @type {String}
     */
    get version(): string {
        return this._raw.version ?? '0.0.0';
    }

    /**
     * If player should mark this extractor as important
     * @type {Boolean}
     */
    get important(): boolean {
        return Boolean(this._raw.important);
    }
}

export default ExtractorModel;
export { ExtractorModel };