feat: getChannels for ChannelOnly inhibitor

Garlic-Team · Jan 21, 2022 · d325351 · d325351
1 parent 75d8273
commit d325351
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/src/inhibitors/ChannelOnly.ts b/src/inhibitors/ChannelOnly.ts
@@ -4,18 +4,25 @@ import { ComponentContext } from '../lib/structures/contexts/ComponentContext';
 import { Inhibitor, InhibitorOptions } from './Inhibitor';
 
 export interface ChannelOnlyOptions extends InhibitorOptions {
-	ids: Array<Snowflake>;
+	ids?: Array<Snowflake>;
+	getChannels?(ctx: CommandContext | ComponentContext): Array<Snowflake>;
 }
 
 export class ChannelOnly extends Inhibitor {
-	public readonly ids: Array<Snowflake>;
+	public ids: Array<Snowflake>;
+	public getChannels?(ctx: CommandContext | ComponentContext): Array<Snowflake>;
 
 	constructor(options: ChannelOnlyOptions) {
 		super(options);
+
 		this.ids = options.ids;
+		this.getChannels = options.getChannels;
 	}
 
 	run(ctx: CommandContext | ComponentContext): boolean | any {
+		const dynamicChannels = this.getChannels?.(ctx);
+		if (dynamicChannels) this.ids = dynamicChannels;
+
 		if (!this.ids.includes(ctx.channelId))
 			return ctx.reply({
 				content: this.resolveMessage(ctx) || 'This command can not be used in this channel',

diff --git a/src/inhibitors/index.ts b/src/inhibitors/index.ts
@@ -20,7 +20,6 @@ export * from './MemberRoles';
  * @deprecated
  */
 export const UserPermissions = MemberPermissions;
-//export * as UserPermissions from './MemberPermissions';
 
 /**
  * @description Use MemberRoles instead of UserRoles