hope dis works (#116)

* hope dis works * exposemap * update doc * doc * yea * more fixes
sern-handler · Jul 24, 2024 · c9c419f · c9c419f
1 parent e85e5fd
commit c9c419f
Showing 1 changed file with 41 additions and 24 deletions.
diff --git a/plugins/cooldown.ts b/plugins/cooldown.ts
@@ -1,18 +1,19 @@
-// @ts-nocheck
 /**
  * @plugin
  * Allows you to set cooldowns (or "ratelimits") for commands, limits user/channel/guild actions.
+ * An extra function cooldown2 is exported if you want your cooldown to be local to the command.
  * @author @trueharuu [<@504698587221852172>]
  * @version 1.0.0
  * @example
  * ```ts
- * import { cooldown } from "../plugins/cooldown";
+ * import { cooldown, cooldown2 } from "../plugins/cooldown";
  * import { commandModule } from "@sern/handler";
+ * //IF you want this cooldown to be local to this command: 
+ * const localCooldown = cooldown2()
  * export default commandModule({
- *  plugins: [cooldown.add( [ ['channel', '1/4'] ] )], // limit to 1 action every 4 seconds per channel
- *  execute: (ctx) => {
- * 		//your code here
- *  }
+ *  plugins: [cooldown.add([['channel', '1/4']]), // limit to 1 action every 4 seconds per channel
+ *            localCooldown.add([["user", "1/10"]])], // limit to 1 action every 10 seconds, local to command
+ *  execute: (ctx) => { //your code here }
  * })
  * ```
  * @end
@@ -60,7 +61,6 @@ export class ExpiryMap<K, V> extends Map<K, V> {
 	}
 }
 
-export const map = new ExpiryMap<string, number>();
 
 function parseCooldown(
 	location: CooldownLocation,
@@ -107,17 +107,16 @@ export interface RecievedCooldown {
 }
 type CooldownResponse = (cooldown: RecievedCooldown) => any;
 
-function add(
-	items: Array<
-		| [CooldownLocation | keyof typeof CooldownLocation, CooldownString]
-		| Cooldown
-	>,
-	message?: CooldownResponse,
-) {
+function __add(map : ExpiryMap<string, number>,
+               items: Array<| [CooldownLocation 
+                            | keyof typeof CooldownLocation, CooldownString]
+                            | Cooldown>,
+                message?: CooldownResponse) {
 	const raw = items.map((c) => {
 		if (!Array.isArray(c)) return c;
 		return parseCooldown(c[0] as CooldownLocation, c[1]);
 	}) as Array<Cooldown>;
+
 	return CommandControlPlugin<CommandType.Both>(async (context, args) => {
 		for (const { location, actions, seconds } of raw) {
 			const id = getPropertyForLocation(context, location);
@@ -147,17 +146,35 @@ function add(
 	});
 }
 
-type Location = (value: CooldownString) => ReturnType<typeof add>;
+type Location = (value: CooldownString) => ReturnType<typeof __add>;
+
+const locationsFn  = (map: ExpiryMap<string, number>)=>  ({
+	[CooldownLocation.channel]: (value) => __add(map, [[CooldownLocation.channel, value]]),
+	[CooldownLocation.user]: (value) => __add(map, [[CooldownLocation.user, value]]),
+	[CooldownLocation.guild]: (value) => __add(map, [[CooldownLocation.guild, value]]),
+} as Record<CooldownLocation, Location>);
+
+
 
-const locations: Record<CooldownLocation, Location> = {
-	[CooldownLocation.channel]: (value) =>
-		add([[CooldownLocation.channel, value]]),
-	[CooldownLocation.user]: (value) => add([[CooldownLocation.user, value]]),
-	[CooldownLocation.guild]: (value) => add([[CooldownLocation.guild, value]]),
+export const cooldown2 = () => {
+    const cooldownMap = new ExpiryMap<string, number>();
+    return {
+        add:(items: Array<| [CooldownLocation 
+                            | keyof typeof CooldownLocation, CooldownString]
+                            | Cooldown>,
+             message?: CooldownResponse) =>  __add(cooldownMap, items, message),
+        locations: locationsFn(cooldownMap),
+        map: cooldownMap
+    }
 };
 
+const map = new ExpiryMap<string, number>();
 export const cooldown = {
-	add,
-	locations,
-	map,
-};
+    map,
+    locations: locationsFn(map),
+    add: (items: Array<| [CooldownLocation 
+                       | keyof typeof CooldownLocation, CooldownString]
+                       | Cooldown>,
+          message?: CooldownResponse) => __add(map, items, message)
+}
+