From 4a6f28b17b6c1da3824ea65becafebacc0381eb6 Mon Sep 17 00:00:00 2001
From: Dragory <2606411+Dragory@users.noreply.github.com>
Date: Wed, 29 Jul 2020 00:48:51 +0300
Subject: [PATCH] Fix error on cooldown check in SelfGrantableRolesPlugin

---
 .../SelfGrantableRoles/SelfGrantableRolesPlugin.ts        | 8 ++++++--
 backend/src/plugins/SelfGrantableRoles/types.ts           | 5 ++++-
 2 files changed, 10 insertions(+), 3 deletions(-)

diff --git a/backend/src/plugins/SelfGrantableRoles/SelfGrantableRolesPlugin.ts b/backend/src/plugins/SelfGrantableRoles/SelfGrantableRolesPlugin.ts
index 09c1d796..1482bde1 100644
--- a/backend/src/plugins/SelfGrantableRoles/SelfGrantableRolesPlugin.ts
+++ b/backend/src/plugins/SelfGrantableRoles/SelfGrantableRolesPlugin.ts
@@ -1,4 +1,4 @@
-import { PluginOptions } from "knub";
+import { CooldownManager, PluginOptions } from "knub";
 import { SelfGrantableRolesPluginType, ConfigSchema, defaultSelfGrantableRoleEntry } from "./types";
 import { zeppelinPlugin } from "../ZeppelinPluginBlueprint";
 import { trimPluginDescription } from "src/utils";
@@ -91,5 +91,9 @@ export const SelfGrantableRolesPlugin = zeppelinPlugin<SelfGrantableRolesPluginT
     RoleHelpCmd,
     RoleRemoveCmd,
     RoleAddCmd,
-  ]
+  ],
+
+  onLoad(pluginData) {
+    pluginData.state.cooldowns = new CooldownManager();
+  },
 });
diff --git a/backend/src/plugins/SelfGrantableRoles/types.ts b/backend/src/plugins/SelfGrantableRoles/types.ts
index 724f697d..825b1498 100644
--- a/backend/src/plugins/SelfGrantableRoles/types.ts
+++ b/backend/src/plugins/SelfGrantableRoles/types.ts
@@ -1,5 +1,5 @@
 import * as t from "io-ts";
-import { BasePluginType, command } from "knub";
+import { BasePluginType, command, CooldownManager } from "knub";
 
 const RoleMap = t.record(t.string, t.array(t.string));
 
@@ -26,6 +26,9 @@ export const defaultSelfGrantableRoleEntry: t.TypeOf<typeof PartialRoleEntry> =
 
 export interface SelfGrantableRolesPluginType extends BasePluginType {
   config: TConfigSchema;
+  state: {
+    cooldowns: CooldownManager;
+  };
 }
 
 export const selfGrantableRolesCmd = command<SelfGrantableRolesPluginType>();