Limit the number of records per user

DevelopingSpace · Mar 20, 2023 · 1e3cea5 · 1e3cea5
1 parent a580cf1
commit 1e3cea5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/.env.example b/.env.example
@@ -36,3 +36,6 @@ SAML_IDP_METADATA_PATH=config/idp-metadata-dev.xml
 EXPIRATION_REPEAT_FREQUENCY_MS=86400000
 # 7 * 24 * 60 * 60 * 1000 = 604800000 (7 days)
 JOB_REMOVAL_FREQUENCY_MS=604800000
+
+# Limit the maximum number of records per user
+USER_RECORD_LIMIT=20
diff --git a/app/models/record.server.ts b/app/models/record.server.ts
@@ -7,6 +7,12 @@ import type { Record } from '@prisma/client';
 export const createUserRecord = async (
  data: Pick<Record, 'username' | 'type' | 'subdomain' | 'value'>
 ) => {
+ if (process.env.USER_RECORD_LIMIT) {
+ if ((await getUserRecordCount(data.username)) >= Number(process.env.USER_RECORD_LIMIT)) {
+ throw new Error('User has reached the maximum number of records');
+ }
+ }
+
  if (await doesRecordExist(data)) {
  throw new Error('Record already exists');
  }