Reorganize project. Add folder for shared code between backend/dashboard. Switch from jest to ava for tests.

backend/src/plugins/PingableRolesPlugin.ts

@@ -0,0 +1,150 @@
+import { decorators as d, IPluginOptions } from "knub";
+import { Message, Role, TextableChannel } from "eris";
+import { GuildPingableRoles } from "../data/GuildPingableRoles";
+import { PingableRole } from "../data/entities/PingableRole";
+import { errorMessage, successMessage } from "../utils";
+import { ZeppelinPlugin } from "./ZeppelinPlugin";
+import * as t from "io-ts";
+
+const ConfigSchema = t.type({
+  can_manage: t.boolean,
+});
+type TConfigSchema = t.TypeOf<typeof ConfigSchema>;
+
+const TIMEOUT = 10 * 1000;
+
+export class PingableRolesPlugin extends ZeppelinPlugin<TConfigSchema> {
+  public static pluginName = "pingable_roles";
+  public static configSchema = ConfigSchema;
+
+  public static pluginInfo = {
+    prettyName: "Pingable roles",
+  };
+
+  protected pingableRoles: GuildPingableRoles;
+  protected cache: Map<string, PingableRole[]>;
+  protected timeouts: Map<string, any>;
+
+  public static getStaticDefaultOptions(): IPluginOptions<TConfigSchema> {
+    return {
+      config: {
+        can_manage: false,
+      },
+
+      overrides: [
+        {
+          level: ">=100",
+          config: {
+            can_manage: true,
+          },
+        },
+      ],
+    };
+  }
+
+  onLoad() {
+    this.pingableRoles = GuildPingableRoles.getGuildInstance(this.guildId);
+
+    this.cache = new Map();
+    this.timeouts = new Map();
+  }
+
+  protected async getPingableRolesForChannel(channelId: string): Promise<PingableRole[]> {
+    if (!this.cache.has(channelId)) {
+      this.cache.set(channelId, await this.pingableRoles.getForChannel(channelId));
+    }
+
+    return this.cache.get(channelId);
+  }
+
+  @d.command("pingable_role disable", "<channelId:channelId> <role:role>")
+  @d.permission("can_manage")
+  async disablePingableRoleCmd(msg: Message, args: { channelId: string; role: Role }) {
+    const pingableRole = await this.pingableRoles.getByChannelAndRoleId(args.channelId, args.role.id);
+    if (!pingableRole) {
+      msg.channel.createMessage(errorMessage(`**${args.role.name}** is not set as pingable in <#${args.channelId}>`));
+      return;
+    }
+
+    await this.pingableRoles.delete(args.channelId, args.role.id);
+    this.cache.delete(args.channelId);
+
+    msg.channel.createMessage(
+      successMessage(`**${args.role.name}** is no longer set as pingable in <#${args.channelId}>`),
+    );
+  }
+
+  @d.command("pingable_role", "<channelId:channelId> <role:role>")
+  @d.permission("can_manage")
+  async setPingableRoleCmd(msg: Message, args: { channelId: string; role: Role }) {
+    const existingPingableRole = await this.pingableRoles.getByChannelAndRoleId(args.channelId, args.role.id);
+    if (existingPingableRole) {
+      msg.channel.createMessage(
+        errorMessage(`**${args.role.name}** is already set as pingable in <#${args.channelId}>`),
+      );
+      return;
+    }
+
+    await this.pingableRoles.add(args.channelId, args.role.id);
+    this.cache.delete(args.channelId);
+
+    msg.channel.createMessage(successMessage(`**${args.role.name}** has been set as pingable in <#${args.channelId}>`));
+  }
+
+  @d.event("typingStart")
+  async onTypingStart(channel: TextableChannel) {
+    const pingableRoles = await this.getPingableRolesForChannel(channel.id);
+    if (pingableRoles.length === 0) return;
+
+    if (this.timeouts.has(channel.id)) {
+      clearTimeout(this.timeouts.get(channel.id));
+    }
+
+    this.enablePingableRoles(pingableRoles);
+
+    const timeout = setTimeout(() => {
+      this.disablePingableRoles(pingableRoles);
+    }, TIMEOUT);
+    this.timeouts.set(channel.id, timeout);
+  }
+
+  @d.event("messageCreate")
+  async onMessageCreate(msg: Message) {
+    const pingableRoles = await this.getPingableRolesForChannel(msg.channel.id);
+    if (pingableRoles.length === 0) return;
+
+    if (this.timeouts.has(msg.channel.id)) {
+      clearTimeout(this.timeouts.get(msg.channel.id));
+    }
+
+    this.disablePingableRoles(pingableRoles);
+  }
+
+  protected enablePingableRoles(pingableRoles: PingableRole[]) {
+    for (const pingableRole of pingableRoles) {
+      const role = this.guild.roles.get(pingableRole.role_id);
+      if (!role) continue;
+
+      role.edit(
+        {
+          mentionable: true,
+        },
+        "Enable pingable role",
+      );
+    }
+  }
+
+  protected disablePingableRoles(pingableRoles: PingableRole[]) {
+    for (const pingableRole of pingableRoles) {
+      const role = this.guild.roles.get(pingableRole.role_id);
+      if (!role) continue;
+
+      role.edit(
+        {
+          mentionable: false,
+        },
+        "Disable pingable role",
+      );
+    }
+  }
+}