zeppelin/src/plugins/CompanionChannels.ts

import { decorators as d, IPluginOptions, logger } from "knub";
import { ZeppelinPlugin } from "./ZeppelinPlugin";
import { Member, Channel, GuildChannel, PermissionOverwrite, Permission, Message } from "eris";
import * as t from "io-ts";

// Permissions using these numbers: https://abal.moe/Eris/docs/reference (add all allowed/denied ones up)
const CompanionChannel = t.type({
  channelIds: t.array(t.string),
  permissions: t.number,
});
type TCompanionChannel = t.TypeOf<typeof CompanionChannel>;

const ConfigSchema = t.type({
  channels: t.record(t.string, CompanionChannel),
});
type TConfigSchema = t.TypeOf<typeof ConfigSchema>;

interface ICompanionChannelMap {
  [channelId: string]: TCompanionChannel;
}

export class CompanionChannelPlugin extends ZeppelinPlugin<TConfigSchema> {
  public static pluginName = "companion_channels";
  protected static configSchema = ConfigSchema;

  companionChannels: Map<string, TCompanionChannel> = new Map();

  protected static getStaticDefaultOptions(): IPluginOptions<TConfigSchema> {
    return {
      config: {
        channels: {},
      },
    };
  }

  onLoad() {
    const tempCompanionChannels: ICompanionChannelMap = this.getConfig().channels;

    for (const [channelId, opts] of Object.entries(tempCompanionChannels)) {
      this.companionChannels.set(channelId, opts);
    }
  }

  onUnload() {
    this.companionChannels.clear();
  }

  async handleCompanionPermissions(userID: string, voiceChannelId: string, remove?: boolean) {
    if (this.companionChannels.has(voiceChannelId)) {
      const compChannels = this.companionChannels.get(voiceChannelId);
      compChannels.channelIds.forEach(textChannelId => {
        const textChannel = <GuildChannel>this.bot.getChannel(textChannelId);

        if (remove) {
          textChannel.deletePermission(userID, `Companion Channel for ${voiceChannelId} | User Left`);
        } else {
          textChannel.editPermission(
            userID,
            compChannels.permissions,
            0,
            "member",
            `Companion Channel for ${voiceChannelId} | User Joined`,
          );
        }
      });
    }
  }

  @d.event("voiceChannelJoin")
  onVoiceChannelJoin(member: Member, voiceChannel: Channel) {
    this.handleCompanionPermissions(member.id, voiceChannel.id);
  }

  @d.event("voiceChannelSwitch")
  onVoiceChannelSwitch(member: Member, newChannel: Channel, oldChannel: Channel) {
    this.handleCompanionPermissions(member.id, oldChannel.id, true);
    this.handleCompanionPermissions(member.id, newChannel.id);
  }

  @d.event("voiceChannelLeave")
  onVoiceChannelLeave(member: Member, voiceChannel: Channel) {
    this.handleCompanionPermissions(member.id, voiceChannel.id, true);
  }
}