fix: add dynamic rate limits (#2892)

2026-06-22 04:12:06 +10:00 · 2026-05-31 00:34:28 +10:00
parent 22ceff43e3
commit 61138cdd81
54 changed files with 3306 additions and 109 deletions
@@ -0,0 +1,86 @@
+import { mailer } from '@documenso/email/mailer';
+import OrganisationLimitExceededEmailTemplate from '@documenso/email/templates/organisation-limit-exceeded';
+import { prisma } from '@documenso/prisma';
+import { msg } from '@lingui/core/macro';
+import { createElement } from 'react';
+
+import { getI18nInstance } from '../../../client-only/providers/i18n-server';
+import { NEXT_PUBLIC_WEBAPP_URL } from '../../../constants/app';
+import { ORGANISATION_MEMBER_ROLE_PERMISSIONS_MAP } from '../../../constants/organisations';
+import { getEmailContext } from '../../../server-only/email/get-email-context';
+import { renderEmailWithI18N } from '../../../utils/render-email-with-i18n';
+import type { JobRunIO } from '../../client/_internal/job';
+import type { TSendOrganisationLimitExceededEmailJobDefinition } from './send-organisation-limit-exceeded-email';
+
+export const run = async ({
+  payload,
+  io,
+}: {
+  payload: TSendOrganisationLimitExceededEmailJobDefinition;
+  io: JobRunIO;
+}) => {
+  const organisation = await prisma.organisation.findFirstOrThrow({
+    where: {
+      id: payload.organisationId,
+    },
+    include: {
+      members: {
+        where: {
+          organisationGroupMembers: {
+            some: {
+              group: {
+                organisationRole: {
+                  in: ORGANISATION_MEMBER_ROLE_PERMISSIONS_MAP['MANAGE_ORGANISATION'],
+                },
+              },
+            },
+          },
+        },
+        include: {
+          user: {
+            select: {
+              id: true,
+              email: true,
+              name: true,
+            },
+          },
+        },
+      },
+    },
+  });
+
+  const { branding, emailLanguage, senderEmail } = await getEmailContext({
+    emailType: 'INTERNAL',
+    source: {
+      type: 'organisation',
+      organisationId: organisation.id,
+    },
+  });
+
+  for (const member of organisation.members) {
+    await io.runTask(`send-organisation-limit-exceeded-email-${member.id}`, async () => {
+      const emailContent = createElement(OrganisationLimitExceededEmailTemplate, {
+        assetBaseUrl: NEXT_PUBLIC_WEBAPP_URL(),
+        organisationName: organisation.name,
+        counter: payload.counter,
+        kind: payload.kind,
+        period: payload.period,
+      });
+
+      const [html, text] = await Promise.all([
+        renderEmailWithI18N(emailContent, { lang: emailLanguage, branding }),
+        renderEmailWithI18N(emailContent, { lang: emailLanguage, branding, plainText: true }),
+      ]);
+
+      const i18n = await getI18nInstance(emailLanguage);
+
+      await mailer.sendMail({
+        to: member.user.email,
+        from: senderEmail,
+        subject: i18n._(msg`Organisation Review Required`),
+        html,
+        text,
+      });
+    });
+  }
+};