zyno-bot-addons/src/classes/structures/voiceState.js

Create easily addons for Zyno Bot

const Save = require('../save.js');
const { validatePermission, getAddonPermission } = require('../../utils/functions.js');
const GuildMemberManager = require('../managers/guildMemberManager.js');
const VoiceChannel = require('./channel/voiceChannel.js');
const StageChannel = require('./channel/stageChannel.js');
const VoiceStateManager = require('../managers/voiceStateManager.js');
const scopes = require('../../bitfields/scopes.js');
const guildManager = require('../managers/guildManager.js');
const channelManager = require('../managers/channelManager.js');

class VoiceState{
    constructor(voiceState, addon, structureHandler, cache){
        this.addon = addon;
        if(typeof voiceState.member === 'object' && !Array.isArray(voiceState.member) && voiceState.member !== null){
            this.id = voiceState.member.id;
        } else {
            this.id = null;
        }
        this.connected = typeof voiceState.channelId === 'string';
        this.selfMute = voiceState.selfMute;
        this.selfDeaf = voiceState.selfDeaf;
        this.selfVideo = voiceState.selfVideo;
        this.streaming = voiceState.streaming;
        this.suppress = voiceState.suppress;
        this.serverMute = voiceState.serverMute;
        this.serverDeaf = voiceState.serverDeaf;
        this.mute = this.selfMute || this.serverMute || false;
        this.deaf = this.selfDeaf || this.serverDeaf || false;
        this.guildId = voiceState?.guild?.id ?? (voiceState.member ?? voiceState.channel)?.guild?.id;
        this.channelId = voiceState.channelId || null;
        if(cache){
            const addonVoiceStateManager = VoiceStateManager.get(addon.name) || new Save();
            const guildVoiceStateManager = addonVoiceStateManager.get(this.guildId) || new Save();
            guildVoiceStateManager.set(this.id, this);
            addonVoiceStateManager.set(this.guildId, guildVoiceStateManager);
            VoiceStateManager.set(addon.name, addonVoiceStateManager);
        }
        this.disconnect = function(reason){
            return new Promise((resolve, reject) => {
                if(!validatePermission(getAddonPermission(addon.name), scopes.bitfield.MEMBERS)) return reject(`Missing members scope in bitfield`);
                if(typeof reason !== 'string') reason = undefined;
                voiceState.disconnect(reason).then(() => {
                    resolve();
                }).catch(reject);
            });
        }
        this.setDeaf = function(deaf, reason){
            return new Promise((resolve, reject) => {
                if(!validatePermission(getAddonPermission(addon.name), scopes.bitfield.MEMBERS)) return reject(`Missing members scope in bitfield`);
                if(typeof deaf !== 'string' && typeof deaf !== 'boolean') deaf = !this.serverDeaf;
                if(typeof reason !== 'string') reason = undefined;
                voiceState.setDeaf(deaf, reason).then(m => {
                    resolve();
                }).catch(reject);
            });
        }
        this.setMute = function(mute, reason){
            return new Promise((resolve, reject) => {
                if(!validatePermission(getAddonPermission(addon.name), scopes.bitfield.MEMBERS)) return reject(`Missing members scope in bitfield`);
                if(typeof mute !== 'string' && typeof mute !== 'boolean') mute = !this.serverMute;
                if(typeof reason !== 'string') reason = undefined;
                voiceState.setMute(mute, reason).then(m => {
                    resolve();
                }).catch(reject);
            });
        }
        this.setChannel = function(channel, reason){
            return new Promise((resolve, reject) => {
                if(!validatePermission(getAddonPermission(addon.name), scopes.bitfield.MEMBERS)) return reject(`Missing members scope in bitfield`);
                if(typeof channel !== 'string' && !(channel instanceof VoiceChannel) && !(channel instanceof StageChannel)) return reject(`The channel is not a resolvable channel`);
                if(typeof reason !== 'string') reason = undefined;
                if(typeof channel !== 'string') channel = channel.id;
                voiceState.setChannel(channel, reason).then(m => {
                    resolve();
                }).catch(reject);
            })
        }
    }
    get member(){
        return ((GuildMemberManager.get(this.addon.name) || new Save()).get(this.guildId) || new Save()).get(this.id);
    }
    get channel(){
        if(this.channelId === null) return null;
        const addonChannelManager = channelManager.get(this.addon.name) || new Save();
        const guildChannelManager = addonChannelManager.get(this.guildId) || new Save();
        return guildChannelManager.get(this.channelId);
    }
    get guild(){
        const addonGuildManager = guildManager.get(this.addon.name) || new Save();
        return addonGuildManager.get(this.guildId);
    }
}

module.exports = VoiceState;