Rate limiting

Author: Tibowl

pages/api/sendmessage.ts

@@ -2,6 +2,7 @@ import { NextApiRequest, NextApiResponse } from "next"
 import fetch from "node-fetch"
 import { config } from "../../utils/config"
 import { parseUser } from "../../utils/parse-user"
+import { DiscordUser } from "../../utils/types"
 
 
 export default async function api(req: NextApiRequest, res: NextApiResponse) {
@@ -17,10 +18,25 @@ export default async function api(req: NextApiRequest, res: NextApiResponse) {
     }
   }
 
+  if (ratelimit(user))
+    return res.status(429).send("Please wait before clicking again")
+
   const fetched = await fetch(`${config.discordUri}/testmessage/${user.id}`, {
     headers: { Authorization: `${config.discordSecret}`, "Content-Type": "application/json" },
     method: "POST"
   })
 
   res.status(fetched.status).send(await fetched.text())
 }
+
+const map = new Map<string, number>()
+function ratelimit(user: DiscordUser): boolean {
+  if (map.get(user.id) ?? 0 > Date.now())
+    return true
+
+  map.set(user.id, Date.now() + 3000)
+  setTimeout(() => {
+    map.delete(user.id)
+  }, 3000)
+  return false
+}

pages/tools/reminders.tsx

@@ -52,12 +52,13 @@ const erorrModalStyle: ReactModal.Styles = {
   }
 }
 
-export default class Reminders extends Component<Props, { reminders: Reminder[], createReminderOpen: boolean }> {
+export default class Reminders extends Component<Props, { reminders: Reminder[], createReminderOpen: boolean, disabledTest: boolean }> {
   constructor(props: Props) {
     super(props)
     this.state = {
       reminders: this.props.reminders,
-      createReminderOpen: false
+      createReminderOpen: false,
+      disabledTest: false
     }
   }
 
@@ -86,8 +87,31 @@ export default class Reminders extends Component<Props, { reminders: Reminder[],
           })}
         />)}
         <div className="flex flex-wrap gap-2">
-          <span className="bg-blue-600 text-slate-50 w-fit px-3 py-1 text-center rounded-lg mt-2 cursor-pointer" onClick={() => sendTest()}>Send test message</span>
-          <span className="bg-green-600 text-slate-50 w-fit px-3 py-1 text-center rounded-lg mt-2 cursor-pointer" onClick={() => this.setState({ createReminderOpen: true })}>Create reminder</span>
+          <button
+            className="bg-blue-600 disabled:bg-gray-900 text-slate-50 disabled:text-slate-400 w-fit px-3 py-1 text-center rounded-lg mt-2 cursor-pointer"
+            onClick={() => {
+              this.setState({
+                disabledTest: true
+              })
+
+              sendTest()
+
+              setTimeout(() => {
+                this.setState({
+                  disabledTest: false
+                })
+              }, 5000)
+            }}
+            disabled={this.state.disabledTest}
+          >
+            Send test message
+          </button>
+          <button
+            className="bg-green-600 text-slate-50 w-fit px-3 py-1 text-center rounded-lg mt-2 cursor-pointer"
+            onClick={() => this.setState({ createReminderOpen: true })}
+          >
+            Create reminder
+          </button>
           <CreateReminder
             isOpen={this.state.createReminderOpen}
             requestClose={() => this.setState({ createReminderOpen: false })}