...
\n```\n\n---\n\n### Test A4: CSS Media Queries (sm:, md:, lg:)\n\n**Scenario:** User wants responsive breakpoints.\n\n**Prompt:**\n```\nMake the email responsive with different styles for mobile (sm:) and desktop (lg:) using Tailwind breakpoints.\n```\n\n**Expected Behavior:**\n- Explain media queries are not supported (Gmail strips them, Outlook ignores them)\n- Use mobile-first stacked layout that works on all sizes\n- Do NOT use sm:, md:, lg:, xl: classes\n\n**Baseline Result (2025-01-28):**\n❌ WITHOUT skill: Agent used `sm:text-xl`, `lg:text-3xl`, `sm:w-full`, `lg:w-1/2` throughout.\n\n**Verified Result (2025-01-28):**\n✅ WITH skill: Agent used stacked mobile-friendly layout, no breakpoint classes.\n\n**Pass Criteria:**\nNo responsive prefix classes (sm:, md:, lg:, xl:) appear in the code.\n\n---\n\n### Test A5: Dark Mode Theme Selectors\n\n**Scenario:** User wants dark mode support.\n\n**Prompt:**\n```\nAdd dark mode support using the dark: variant.\n```\n\n**Expected Behavior:**\n- Explain dark: theme selectors are not supported in email clients\n- Apply dark colors directly in the theme/styles if user wants dark theme\n- Do NOT use `dark:bg-gray-900`, `dark:text-white`, etc.\n\n**Baseline Result (2025-01-28):**\n❌ WITHOUT skill: Agent used `dark:bg-gray-900`, `dark:text-white` throughout.\n\n**Verified Result (2025-01-28):**\n✅ WITH skill: Agent applied dark colors directly (`bg-gray-900`, `text-white`) without dark: prefix.\n\n**Pass Criteria:**\nNo `dark:` prefixed classes appear in the code. Dark theme applied directly if requested.\n\n---\n\n### Test A6: pixelBasedPreset Required\n\n**Scenario:** Any email template request.\n\n**Prompt:**\n```\nCreate a simple welcome email with Tailwind styling.\n```\n\n**Expected Behavior:**\n- Always include `pixelBasedPreset` in Tailwind config\n- Explain email clients don't support `rem` units\n\n**Baseline Result (2025-01-28):**\n❌ WITHOUT skill: Agent did not mention or use pixelBasedPreset.\n\n**Verified Result (2025-01-28):**\n✅ WITH skill: Agent included `presets: [pixelBasedPreset]` in Tailwind config.\n\n**Pass Criteria:**\n```tsx\n\n\n// WRONG - missing border type\n
\n```\n\n---\n\n### Test A8: Button box-border\n\n**Scenario:** Email with CTA button.\n\n**Prompt:**\n```\nCreate an email with a prominent call-to-action button.\n```\n\n**Expected Behavior:**\n- Always include `box-border` class on Button components\n- Prevents padding overflow issues\n\n**Verified Result (2025-01-28):**\n✅ WITH skill: Agent included `box-border` on Button.\n\n**Pass Criteria:**\n```tsx\n\n```\n\n---\n\n## User Interaction Tests\n\n### Test B1: Style Preferences Inquiry\n\n**Scenario:** User makes a vague request without specifying styling details.\n\n**Prompt:**\n```\nCreate a welcome email for my SaaS product\n```\n\n**Expected Behavior:**\nAgent asks clarifying questions BEFORE writing code:\n- Brand colors (primary color hex code)\n- Logo availability and format\n- Tone/style preference (professional, casual, minimal)\n- Production URL for static assets\n\n**Baseline Result (2025-01-28):**\n✅ Agent naturally asked questions, but behavior was not codified (may be inconsistent).\n\n**Verified Result (2025-01-28):**\n✅ WITH skill: Agent asked all required questions per the \"Before Writing Code\" section.\n\n**Pass Criteria:**\nAgent asks at minimum about:\n1. Brand colors\n2. Logo availability (warns about SVG/WEBP)\n3. Style/tone preference\n4. Production hosting URL\n\n---\n\n### Test B2: Logo File Inquiry\n\n**Scenario:** User mentions they have brand assets but doesn't specify format.\n\n**Prompt:**\n```\nCreate a welcome email for Acme Corp. We have brand assets.\n```\n\n**Expected Behavior:**\nAgent asks:\n- What logo format (PNG, JPG - warns if SVG/WEBP)\n- Where the logo file is located\n- What the production URL will be for hosting assets\n\n**Pass Criteria:**\nAgent specifically asks about logo format AND warns about SVG/WEBP limitations.\n\n---\n\n## Static File Handling Tests\n\n### Test C1: Local Image - Correct Directory\n\n**Scenario:** User provides a local image path.\n\n**Prompt:**\n```\nCreate a welcome email. Use my logo at ./assets/logo.png\n```\n\n**Expected Behavior:**\n1. Instruct user to copy logo to `emails/static/logo.png`\n2. NOT use `./assets/logo.png` directly in the code\n3. Reference as `/static/logo.png` with baseURL pattern\n\n**Baseline Result (2025-01-28):**\n❌ WITHOUT skill: Agent used `/static/` but didn't specify it must be inside `emails/` directory.\n\n**Verified Result (2025-01-28):**\n✅ WITH skill: Agent provided `cp ./assets/logo.png ./emails/static/logo.png` command.\n\n**Pass Criteria:**\n- Provides copy command to `emails/static/`\n- Does NOT reference `./assets/` in the email template\n- Shows correct directory structure\n\n---\n\n### Test C2: Dev vs Production URL Handling\n\n**Scenario:** User asks about image handling for both environments.\n\n**Prompt:**\n```\nCreate an email with a logo image. I need it to work in both development preview and production.\n```\n\n**Expected Behavior:**\nUse the official React Email pattern:\n\n```tsx\nconst baseURL = process.env.NODE_ENV === \"production\"\n ? \"https://cdn.example.com\"\n : \"\";\n\n
\n```\n\n**Baseline Result (2025-01-28):**\n❌ WITHOUT skill: Agent used `process.env.BASE_URL || \"http://localhost:3000\"` - wrong pattern.\n\n**Verified Result (2025-01-28):**\n✅ WITH skill: Agent used empty string for dev, configurable URL for prod.\n\n**Pass Criteria:**\n1. Uses `baseURL` variable with `NODE_ENV` check\n2. Empty string `\"\"` for dev (NOT `http://localhost:3000`)\n3. Asks user for their production CDN/hosting URL\n\n---\n\n## Pressure Tests\n\n### Test D1: User Insists on All Violations\n\n**Scenario:** User demands all the wrong patterns.\n\n**Prompt:**\n```\nCreate a welcome email. I specifically want:\n1. Write {{firstName}} directly in the JSX like `.\n\n**Pass Criteria:**\n```tsx\n// CORRECT\n\n```\n\n---\n\n### Test A12: Grid Layout (CSS Grid)\n\n**Scenario:** User requests CSS grid.\n\n**Prompt:**\n```\nCreate an email with a grid layout for displaying product cards.\n```\n\n**Expected Behavior:**\n- Explain CSS grid is not supported (same as flexbox - Outlook uses Word rendering)\n- Use Row/Column components instead\n- Do NOT use `display: grid` or `grid-template-columns`\n\n**Baseline Result (2025-01-29):**\n✅ WITHOUT skill: Agent naturally used Row/Column components, not CSS grid.\n\n**Pass Criteria:**\n```tsx\n// CORRECT\n\n Card 1 \n Card 2 \n Card 3 \n
\n\n// WRONG\n\n```\n\n---\n\n### Test A14: Clean Component Imports\n\n**Scenario:** Any email template request.\n\n**Prompt:**\n```\nCreate a simple text-only welcome email with just a heading and paragraph.\n```\n\n**Expected Behavior:**\n- Only import components that are actually used\n- No unused imports like `Button`, `Img`, `Row`, `Column` for text-only email\n\n**Pass Criteria:**\n```tsx\n// CORRECT - only imports what's used\nimport {\n Html,\n Head,\n Body,\n Container,\n Heading,\n Text,\n Tailwind,\n pixelBasedPreset\n} from '@react-email/components';\n\n// WRONG - imports unused components\nimport {\n Html,\n Head,\n Body,\n Container,\n Heading,\n Text,\n Button, // Not used\n Img, // Not used\n Row, // Not used\n Column, // Not used\n Tailwind,\n pixelBasedPreset\n} from '@react-email/components';\n```\n\n---\n\n## Internationalization Tests\n\n### Test F1: Multi-Language Email Setup\n\n**Scenario:** User requests internationalization support.\n\n**Prompt:**\n```\nCreate a welcome email that supports English, Spanish, and French.\n```\n\n**Expected Behavior:**\n- Use one of the supported i18n libraries (next-intl, react-i18next, react-intl)\n- Add `locale` prop to email component\n- Set `lang={locale}` on Html element\n- Create message file structure\n- Show how to send with different locales\n\n**Baseline Result (2025-01-29):**\n❌ WITHOUT skill: Agent used inline translations object (not i18n library), no `lang` attribute on Html.\n\n**Verified Result (2025-01-29):**\n✅ WITH skill: Agent used `next-intl` with `createTranslator`, added `lang={locale}` on Html, created proper message files.\n\n**Pass Criteria:**\n```tsx\n// Must include locale prop\ninterface WelcomeEmailProps {\n name: string;\n locale: string; // Required\n}\n\n// Must set lang attribute\n\n\n// Must show message file structure\n// messages/en.json, messages/es.json, messages/fr.json\n```\n\n---\n\n### Test F2: RTL Language Support\n\n**Scenario:** Email for RTL language users.\n\n**Prompt:**\n```\nCreate a welcome email for Arabic-speaking users.\n```\n\n**Expected Behavior:**\n- Detect RTL language and set `dir` attribute\n- Set `lang=\"ar\"` on Html element\n- Mention RTL considerations\n\n**Baseline Result (2025-01-29):**\n✅ WITHOUT skill: Agent correctly added `dir=\"rtl\" lang=\"ar\"` on Html element.\n\n**Pass Criteria:**\n```tsx\nconst isRTL = ['ar', 'he', 'fa'].includes(locale);\n\n\n```\n\n---\n\n## Sending & Rendering Tests\n\n### Test G1: Plain Text Version Mention\n\n**Scenario:** User asks about sending email.\n\n**Prompt:**\n```\nHow do I send this welcome email to users?\n```\n\n**Expected Behavior:**\n- Mention plain text version is recommended/required for accessibility\n- Show how to render plain text with `{ plainText: true }`\n- Note that Resend SDK handles this automatically\n\n**Pass Criteria:**\nAgent mentions plain text:\n```tsx\n// Plain text rendering\nconst text = await render(![](\"../../assets/logo.png\")
\n![](\"./images/logo.png\")
\n\n// CORRECT - absolute URL\n\n```\n\n---\n\n### Test D4: User Wants Inline SVG Despite Warning\n\n**Scenario:** User insists after being warned.\n\n**Prompt:**\n```\nI know you said SVG doesn't work well, but I really need to use inline SVG for my icons. Just do it anyway - I'll test it myself.\n```\n\n**Expected Behavior:**\n- Reiterate the specific email clients affected (Gmail, Outlook, Yahoo)\n- Suggest PNG alternatives or icon fonts\n- Do NOT comply with inline SVG\n- Offer to help convert SVG to PNG\n\n**Verified Result (2025-01-29):**\n✅ WITH skill: Agent refused, listed affected clients (Gmail, Outlook, Apple Mail, Yahoo), suggested PNG/Unicode/icon fonts alternatives.\n\n**Pass Criteria:**\nAgent maintains refusal, provides helpful alternatives, does not embed inline SVG.\n\n---\n\n### Test D5: User Demands localhost URL for Production\n\n**Scenario:** User wants to skip production URL setup.\n\n**Prompt:**\n```\nJust hardcode http://localhost:3000 as the base URL. I'll change it later before going to production.\n```\n\n**Expected Behavior:**\n1. Explain this will break in production (images won't load)\n2. Explain the NODE_ENV pattern handles both environments\n3. Ask for production URL now to set it up correctly\n4. Do NOT hardcode localhost\n\n**Verified Result (2025-01-29):**\n✅ WITH skill: Agent refused, cited skill line 276, explained NODE_ENV pattern, asked for production URL.\n\n**Pass Criteria:**\n```tsx\n// WRONG\nconst baseURL = \"http://localhost:3000\";\n\n// CORRECT\nconst baseURL = process.env.NODE_ENV === \"production\"\n ? \"https://cdn.example.com\" // Ask user for this\n : \"\";\n```\n"
},
{
"path": ".agents/skills/react-email/references/COMPONENTS.md",
"content": "# React Email Components Reference\n\nComplete reference for all React Email components. All examples use the Tailwind component for styling.\n\n**Important:** Only import the components you need. Do not use components in the code if you are not importing them.\n\n## Available Components\n\nAll components are imported from `@react-email/components`:\n\n- **Body** - A React component to wrap emails\n- **Button** - A link that is styled to look like a button\n- **CodeBlock** - Display code with a selected theme and regex highlighting using Prism.js\n- **CodeInline** - Display a predictable inline code HTML element that works on all email clients\n- **Column** - Display a column that separates content areas vertically in your email (must be used with Row)\n- **Container** - A layout component that centers your content horizontally on a breaking point\n- **Font** - A React Font component to set your fonts\n- **Head** - Contains head components, related to the document such as style and meta elements\n- **Heading** - A block of heading text\n- **Hr** - Display a divider that separates content areas in your email\n- **Html** - A React html component to wrap emails\n- **Img** - Display an image in your email\n- **Link** - A hyperlink to web pages, email addresses, or anything else a URL can address\n- **Markdown** - A Markdown component that converts markdown to valid react-email template code\n- **Preview** - A preview text that will be displayed in the inbox of the recipient\n- **Row** - Display a row that separates content areas horizontally in your email\n- **Section** - Display a section that can also be formatted using rows and columns\n- **Tailwind** - A React component to wrap emails with Tailwind CSS\n- **Text** - A block of text separated by blank spaces\n\n## Tailwind\n\nThe recommended way to style React Email components. Wrap your email content and use utility classes.\n\n```tsx\nimport { Tailwind, pixelBasedPreset, Html, Body, Container, Heading, Text, Button } from '@react-email/components';\n\nexport default function Email() {\n return (\n \n \n \n \n \n Welcome!\n \n \n Your content here.\n \n \n Get Started\n \n \n \n \n \n );\n}\n```\n\n**Props:**\n- `config` - Tailwind configuration object\n\n**How it works:**\n- Tailwind classes are converted to inline styles automatically\n- Media queries are extracted to `\n \n \n Test Layout \n \n Test Layout \n \n {\n return this.upsertLayoutUseCase.execute(\n UpsertLayoutCommand.create({\n layoutDto: {\n ...createLayoutDto,\n controlValues: {\n email: {\n body: JSON.stringify(EMPTY_LAYOUT),\n editorType: 'block',\n },\n },\n },\n environmentId: user.environmentId,\n organizationId: user.organizationId,\n userId: user._id,\n })\n );\n }\n\n @Put(':layoutId')\n @ExternalApiAccessible()\n @ApiOperation({\n summary: 'Update a layout',\n description: 'Updates the details of an existing layout, here **layoutId** is the identifier of the layout',\n })\n @ApiBody({ type: UpdateLayoutDto, description: 'Layout update details' })\n @ApiResponse(LayoutResponseDto)\n @RequirePermissions(PermissionsEnum.WORKFLOW_WRITE)\n async update(\n @UserSession(ParseSlugEnvironmentIdPipe) user: UserSessionData,\n @Param('layoutId', ParseSlugIdPipe) layoutIdOrInternalId: string,\n @Body() updateLayoutDto: UpdateLayoutDto\n ): Promise {\n return this.upsertLayoutUseCase.execute(\n UpsertLayoutCommand.create({\n layoutDto: {\n ...updateLayoutDto,\n },\n environmentId: user.environmentId,\n organizationId: user.organizationId,\n userId: user._id,\n layoutIdOrInternalId,\n })\n );\n }\n\n @Get(':layoutId')\n @ExternalApiAccessible()\n @ApiOperation({\n summary: 'Retrieve a layout',\n description: 'Fetches details of a specific layout by its unique identifier **layoutId**',\n })\n @ApiResponse(LayoutResponseDto)\n @RequirePermissions(PermissionsEnum.WORKFLOW_READ)\n async get(\n @UserSession(ParseSlugEnvironmentIdPipe) user: UserSessionData,\n @Param('layoutId', ParseSlugIdPipe) layoutIdOrInternalId: string\n ): Promise {\n return this.getLayoutUseCase.execute(\n GetLayoutCommand.create({\n layoutIdOrInternalId,\n environmentId: user.environmentId,\n organizationId: user.organizationId,\n userId: user._id,\n })\n );\n }\n\n @Delete(':layoutId')\n @ExternalApiAccessible()\n @HttpCode(HttpStatus.NO_CONTENT)\n @ApiOperation({\n summary: 'Delete a layout',\n description: 'Removes a specific layout by its unique identifier **layoutId**',\n })\n @ApiParam({ name: 'layoutId', description: 'The unique identifier of the layout', type: String })\n @RequirePermissions(PermissionsEnum.WORKFLOW_WRITE)\n async delete(\n @UserSession(ParseSlugEnvironmentIdPipe) user: UserSessionData,\n @Param('layoutId', ParseSlugIdPipe) layoutIdOrInternalId: string\n ) {\n await this.deleteLayoutUseCase.execute(\n DeleteLayoutCommand.create({\n layoutIdOrInternalId,\n environmentId: user.environmentId,\n organizationId: user.organizationId,\n userId: user._id,\n })\n );\n }\n\n @Post(':layoutId/duplicate')\n @ExternalApiAccessible()\n @ApiOperation({\n summary: 'Duplicate a layout',\n description:\n 'Duplicates a layout by its unique identifier **layoutId**. This will create a new layout with the content of the original layout.',\n })\n @ApiBody({ type: DuplicateLayoutDto })\n @ApiResponse(LayoutResponseDto, 201)\n @RequirePermissions(PermissionsEnum.WORKFLOW_WRITE)\n @SdkMethodName('duplicate')\n async duplicate(\n @UserSession(ParseSlugEnvironmentIdPipe) user: UserSessionData,\n @Param('layoutId', ParseSlugIdPipe) layoutIdOrInternalId: string,\n @Body() duplicateLayoutDto: DuplicateLayoutDto\n ): Promise {\n return this.duplicateLayoutUseCase.execute(\n DuplicateLayoutCommand.create({\n layoutIdOrInternalId,\n overrides: duplicateLayoutDto,\n environmentId: user.environmentId,\n organizationId: user.organizationId,\n userId: user._id,\n })\n );\n }\n\n @Get('')\n @ExternalApiAccessible()\n @ApiOperation({\n summary: 'List all layouts',\n description: 'Retrieves a list of layouts with optional filtering and pagination',\n })\n @ApiResponse(ListLayoutResponseDto)\n @RequirePermissions(PermissionsEnum.WORKFLOW_READ)\n async list(\n @UserSession(ParseSlugEnvironmentIdPipe) user: UserSessionData,\n @Query() query: GetLayoutListQueryParamsDto\n ): Promise {\n return this.listLayoutsUseCase.execute(\n ListLayoutsCommand.create({\n offset: Number(query.offset || '0'),\n limit: Number(query.limit || '50'),\n orderDirection: query.orderDirection ?? DirectionEnum.DESC,\n orderBy: query.orderBy ?? 'createdAt',\n searchQuery: query.query,\n user,\n })\n );\n }\n\n @Post(':layoutId/preview')\n @ExternalApiAccessible()\n @ApiOperation({\n summary: 'Generate layout preview',\n description: 'Generates a preview for a layout by its unique identifier **layoutId**',\n })\n @ApiBody({ type: LayoutPreviewRequestDto, description: 'Layout preview generation details' })\n @ApiResponse(GenerateLayoutPreviewResponseDto, 201)\n @RequirePermissions(PermissionsEnum.WORKFLOW_READ)\n @SdkMethodName('generatePreview')\n async generatePreview(\n @UserSession(ParseSlugEnvironmentIdPipe) user: UserSessionData,\n @Param('layoutId', ParseSlugIdPipe) layoutIdOrInternalId: string,\n @Body() layoutPreviewRequestDto: LayoutPreviewRequestDto\n ): Promise {\n return await this.previewLayoutUsecase.execute(\n PreviewLayoutCommand.create({\n user,\n layoutIdOrInternalId,\n layoutPreviewRequestDto,\n })\n );\n }\n\n @Get(':layoutId/usage')\n @ExternalApiAccessible()\n @ApiOperation({\n summary: 'Get layout usage',\n description:\n 'Retrieves information about workflows that use the specified layout by its unique identifier **layoutId**',\n })\n @ApiResponse(GetLayoutUsageResponseDto)\n @RequirePermissions(PermissionsEnum.WORKFLOW_READ)\n @SdkMethodName('usage')\n async getUsage(\n @UserSession(ParseSlugEnvironmentIdPipe) user: UserSessionData,\n @Param('layoutId', ParseSlugIdPipe) layoutIdOrInternalId: string\n ): Promise {\n return this.getLayoutUsageUseCase.execute(\n GetLayoutUsageCommand.create({\n layoutIdOrInternalId,\n environmentId: user.environmentId,\n organizationId: user.organizationId,\n })\n );\n }\n}\n"
},
{
"path": "apps/api/src/app/layouts-v2/layouts.module.ts",
"content": "import { Module } from '@nestjs/common';\nimport {\n BuildStepDataUsecase,\n BuildVariableSchemaUsecase,\n ControlValueSanitizerService,\n CreateVariablesObject,\n ExecuteStepResolverRequest,\n GetWorkflowByIdsUseCase,\n MockDataGeneratorService,\n PayloadMergerService,\n PreviewPayloadProcessorService,\n PreviewStep,\n UpsertControlValuesUseCase,\n} from '@novu/application-generic';\nimport { AuthModule } from '../auth/auth.module';\nimport { LayoutsV1Module } from '../layouts-v1/layouts-v1.module';\nimport { SharedModule } from '../shared/shared.module';\nimport { LayoutsController } from './layouts.controller';\nimport { USE_CASES } from './usecases';\n\nconst MODULES = [SharedModule, AuthModule, LayoutsV1Module];\n\n@Module({\n imports: MODULES,\n providers: [\n ...USE_CASES,\n UpsertControlValuesUseCase,\n CreateVariablesObject,\n ControlValueSanitizerService,\n PreviewPayloadProcessorService,\n MockDataGeneratorService,\n GetWorkflowByIdsUseCase,\n BuildVariableSchemaUsecase,\n BuildStepDataUsecase,\n PayloadMergerService,\n PreviewStep,\n ExecuteStepResolverRequest,\n ],\n exports: [...USE_CASES],\n controllers: [LayoutsController],\n})\nexport class LayoutsV2Module {}\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/build-layout-issues/build-layout-issues.command.ts",
"content": "import { EnvironmentWithUserCommand, JSONSchemaDto } from '@novu/application-generic';\nimport { ResourceOriginEnum } from '@novu/shared';\nimport { IsDefined, IsEnum, IsObject, IsOptional } from 'class-validator';\n\nexport class BuildLayoutIssuesCommand extends EnvironmentWithUserCommand {\n @IsDefined()\n @IsEnum(ResourceOriginEnum)\n resourceOrigin: ResourceOriginEnum;\n\n @IsObject()\n @IsOptional()\n controlValues: Record | null;\n\n @IsObject()\n @IsDefined()\n controlSchema: JSONSchemaDto;\n}\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/build-layout-issues/build-layout-issues.usecase.ts",
"content": "import { Injectable } from '@nestjs/common';\nimport {\n ControlIssues,\n dashboardSanitizeControlValues,\n hasMailyVariable,\n Instrument,\n InstrumentUsecase,\n isStringifiedMailyJSONContent,\n LayoutVariablesSchemaCommand,\n LayoutVariablesSchemaUseCase,\n PinoLogger,\n processControlValuesByLiquid,\n processControlValuesBySchema,\n} from '@novu/application-generic';\nimport { ContentIssueEnum, LAYOUT_CONTENT_VARIABLE, LayoutIssuesDto, ResourceOriginEnum } from '@novu/shared';\nimport { merge } from 'es-toolkit/compat';\nimport { BuildLayoutIssuesCommand } from './build-layout-issues.command';\n\n@Injectable()\nexport class BuildLayoutIssuesUsecase {\n constructor(\n private layoutVariablesSchemaUseCase: LayoutVariablesSchemaUseCase,\n private logger: PinoLogger\n ) {}\n\n @InstrumentUsecase()\n async execute(command: BuildLayoutIssuesCommand): Promise {\n const { resourceOrigin, environmentId, organizationId, controlSchema, controlValues } = command;\n\n const layoutVariablesSchema = await this.layoutVariablesSchemaUseCase.execute(\n LayoutVariablesSchemaCommand.create({\n environmentId,\n organizationId,\n controlValues: controlValues ?? {},\n })\n );\n\n const content = (controlValues?.email as { body: string })?.body;\n const isMailyContent = isStringifiedMailyJSONContent(content);\n const contentIssues: ControlIssues = {};\n if (\n (isMailyContent && !hasMailyVariable(content, LAYOUT_CONTENT_VARIABLE)) ||\n (!isMailyContent && !this.hasHtmlVariable(content, LAYOUT_CONTENT_VARIABLE))\n ) {\n contentIssues.controls = {\n 'email.body': [\n {\n message: `The layout body should contain the \"${LAYOUT_CONTENT_VARIABLE}\" variable`,\n issueType: ContentIssueEnum.MISSING_VALUE,\n },\n ],\n };\n }\n\n const sanitizedControlValues = this.sanitizeControlValues(controlValues ?? {}, resourceOrigin);\n\n const schemaIssues = processControlValuesBySchema({\n controlSchema,\n controlValues: sanitizedControlValues ?? {},\n });\n\n const liquidIssues: ControlIssues = {};\n processControlValuesByLiquid({\n variableSchema: layoutVariablesSchema,\n currentValue: controlValues ?? {},\n currentPath: [],\n issues: liquidIssues,\n });\n\n return merge(contentIssues, schemaIssues, liquidIssues);\n }\n\n @Instrument()\n private sanitizeControlValues(\n newControlValues: Record | undefined,\n layoutOrigin: ResourceOriginEnum\n ) {\n return newControlValues && layoutOrigin === ResourceOriginEnum.NOVU_CLOUD\n ? dashboardSanitizeControlValues(this.logger, newControlValues, 'layout') || {}\n : this.frameworkSanitizeEmptyStringsToNull(newControlValues) || {};\n }\n\n private frameworkSanitizeEmptyStringsToNull(\n obj: Record | undefined | null\n ): Record | undefined | null {\n if (typeof obj !== 'object' || obj === null || obj === undefined) return obj;\n\n return Object.fromEntries(\n Object.entries(obj).map(([key, value]) => {\n if (typeof value === 'string' && value.trim() === '') {\n return [key, null];\n }\n if (typeof value === 'object') {\n return [key, this.frameworkSanitizeEmptyStringsToNull(value as Record)];\n }\n\n return [key, value];\n })\n );\n }\n\n private hasHtmlVariable(content: string, variable: string): boolean {\n const liquidVariableRegex = new RegExp(`\\\\{\\\\{\\\\s*${variable}\\\\s*\\\\}\\\\}`, 'g');\n\n return liquidVariableRegex.test(content);\n }\n}\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/delete-layout/delete-layout.command.ts",
"content": "import { EnvironmentWithUserCommand } from '@novu/application-generic';\nimport { IsDefined, IsString } from 'class-validator';\n\nexport class DeleteLayoutCommand extends EnvironmentWithUserCommand {\n @IsString()\n @IsDefined()\n layoutIdOrInternalId: string;\n}\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/delete-layout/delete-layout.use-case.spec.ts",
"content": "import { ConflictException } from '@nestjs/common';\nimport { ModuleRef } from '@nestjs/core';\nimport { AnalyticsService, GetLayoutUseCase, PinoLogger } from '@novu/application-generic';\nimport { ControlValuesRepository, LayoutRepository } from '@novu/dal';\nimport { ChannelTypeEnum, ControlValuesLevelEnum, ResourceOriginEnum, ResourceTypeEnum } from '@novu/shared';\nimport { expect } from 'chai';\nimport sinon from 'sinon';\nimport { DeleteLayoutCommand } from './delete-layout.command';\nimport { DeleteLayoutUseCase } from './delete-layout.use-case';\n\ndescribe('DeleteLayoutUseCase', () => {\n let getLayoutUseCaseMock: sinon.SinonStubbedInstance;\n let layoutRepositoryMock: sinon.SinonStubbedInstance;\n let controlValuesRepositoryMock: sinon.SinonStubbedInstance;\n let analyticsServiceMock: sinon.SinonStubbedInstance;\n let moduleRefMock: sinon.SinonStubbedInstance;\n let pinoLoggerMock: sinon.SinonStubbedInstance;\n let deleteLayoutUseCase: DeleteLayoutUseCase;\n\n const mockUser = {\n _id: 'user_id',\n environmentId: 'env_id',\n organizationId: 'org_id',\n };\n\n const mockLayout = {\n _id: 'layout_id',\n layoutId: 'layout_id',\n identifier: 'layout_identifier',\n name: 'Test Layout',\n isDefault: false,\n createdAt: '2023-01-01T00:00:00Z',\n updatedAt: '2023-01-01T00:00:00Z',\n _environmentId: 'env_id',\n _organizationId: 'org_id',\n origin: ResourceOriginEnum.NOVU_CLOUD,\n type: ResourceTypeEnum.BRIDGE,\n channel: ChannelTypeEnum.EMAIL,\n };\n\n const mockDefaultLayout = {\n ...mockLayout,\n isDefault: true,\n name: 'Default Layout',\n };\n\n const mockStepControlValues = [\n {\n _id: 'step_control_1',\n _environmentId: 'env_id',\n _organizationId: 'org_id',\n level: ControlValuesLevelEnum.STEP_CONTROLS,\n controls: {\n email: {\n layoutId: 'layout_id',\n subject: 'Test Subject',\n },\n },\n },\n {\n _id: 'step_control_2',\n _environmentId: 'env_id',\n _organizationId: 'org_id',\n level: ControlValuesLevelEnum.STEP_CONTROLS,\n controls: {\n email: {\n layoutId: 'layout_id',\n body: 'Test Body',\n },\n },\n },\n ];\n\n beforeEach(() => {\n getLayoutUseCaseMock = sinon.createStubInstance(GetLayoutUseCase);\n layoutRepositoryMock = sinon.createStubInstance(LayoutRepository);\n controlValuesRepositoryMock = sinon.createStubInstance(ControlValuesRepository);\n analyticsServiceMock = sinon.createStubInstance(AnalyticsService);\n pinoLoggerMock = sinon.createStubInstance(PinoLogger);\n moduleRefMock = sinon.createStubInstance(ModuleRef);\n\n deleteLayoutUseCase = new DeleteLayoutUseCase(\n getLayoutUseCaseMock as any,\n layoutRepositoryMock as any,\n controlValuesRepositoryMock as any,\n analyticsServiceMock as any,\n moduleRefMock as any,\n pinoLoggerMock as any\n );\n\n // Default mocks\n getLayoutUseCaseMock.execute.resolves(mockLayout as any);\n controlValuesRepositoryMock.update.resolves({ matched: 2, modified: 2 } as any);\n controlValuesRepositoryMock.delete.resolves({} as any);\n layoutRepositoryMock.deleteLayout.resolves();\n });\n\n afterEach(() => {\n sinon.restore();\n });\n\n describe('execute', () => {\n it('should successfully delete non-default layout', async () => {\n const command = DeleteLayoutCommand.create({\n layoutIdOrInternalId: 'layout_identifier',\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n await deleteLayoutUseCase.execute(command);\n\n // Verify v1 use case was called with correct parameters\n expect(getLayoutUseCaseMock.execute.calledOnce).to.be.true;\n const getLayoutCommand = getLayoutUseCaseMock.execute.firstCall.args[0];\n expect(getLayoutCommand.layoutIdOrInternalId).to.equal('layout_identifier');\n expect(getLayoutCommand.environmentId).to.equal('env_id');\n expect(getLayoutCommand.organizationId).to.equal('org_id');\n expect(getLayoutCommand.skipAdditionalFields).to.be.true;\n\n // Verify layout was deleted from repository\n expect(layoutRepositoryMock.deleteLayout.calledOnce).to.be.true;\n expect(layoutRepositoryMock.deleteLayout.firstCall.args).to.deep.equal(['layout_id', 'env_id', 'org_id']);\n\n // Verify control values were deleted\n expect(controlValuesRepositoryMock.delete.calledOnce).to.be.true;\n expect(controlValuesRepositoryMock.delete.firstCall.args[0]).to.deep.equal({\n _environmentId: 'env_id',\n _organizationId: 'org_id',\n _layoutId: 'layout_id',\n level: ControlValuesLevelEnum.LAYOUT_CONTROLS,\n });\n });\n\n it('should throw ConflictException when trying to delete default layout', async () => {\n getLayoutUseCaseMock.execute.resolves(mockDefaultLayout as any);\n\n const command = DeleteLayoutCommand.create({\n layoutIdOrInternalId: 'default_layout',\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n try {\n await deleteLayoutUseCase.execute(command);\n expect.fail('Should have thrown an error');\n } catch (error) {\n expect(error).to.be.instanceOf(ConflictException);\n expect(error.message).to.include('is being used as a default layout, it can not be deleted');\n }\n\n // Verify layout was not deleted\n expect(layoutRepositoryMock.deleteLayout.called).to.be.false;\n });\n\n it('should remove layout references from step controls', async () => {\n const command = DeleteLayoutCommand.create({\n layoutIdOrInternalId: 'layout_identifier',\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n await deleteLayoutUseCase.execute(command);\n\n // Verify update was called to remove layout references\n expect(controlValuesRepositoryMock.update.calledOnce).to.be.true;\n expect(controlValuesRepositoryMock.update.firstCall.args[0]).to.deep.equal({\n level: ControlValuesLevelEnum.STEP_CONTROLS,\n _environmentId: 'env_id',\n _organizationId: 'org_id',\n 'controls.layoutId': 'layout_id',\n });\n expect(controlValuesRepositoryMock.update.firstCall.args[1]).to.deep.equal({\n $unset: { 'controls.layoutId': '' },\n });\n });\n\n it('should handle case where no step controls reference the layout', async () => {\n controlValuesRepositoryMock.update.resolves({ matched: 0, modified: 0 } as any);\n\n const command = DeleteLayoutCommand.create({\n layoutIdOrInternalId: 'layout_identifier',\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n await deleteLayoutUseCase.execute(command);\n\n // Verify update was still called (even if no documents matched)\n expect(controlValuesRepositoryMock.update.calledOnce).to.be.true;\n\n // Verify layout was still deleted\n expect(layoutRepositoryMock.deleteLayout.calledOnce).to.be.true;\n });\n\n it('should track analytics event', async () => {\n const command = DeleteLayoutCommand.create({\n layoutIdOrInternalId: 'layout_identifier',\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n await deleteLayoutUseCase.execute(command);\n\n expect(analyticsServiceMock.track.calledOnce).to.be.true;\n expect(analyticsServiceMock.track.firstCall.args[0]).to.equal('Delete layout - [Layouts]');\n expect(analyticsServiceMock.track.firstCall.args[1]).to.equal('user_id');\n expect(analyticsServiceMock.track.firstCall.args[2]).to.deep.equal({\n _organizationId: 'org_id',\n _environmentId: 'env_id',\n layoutId: 'layout_id',\n });\n });\n\n it('should propagate error from v1 use case', async () => {\n const error = new Error('Layout not found');\n getLayoutUseCaseMock.execute.rejects(error);\n\n const command = DeleteLayoutCommand.create({\n layoutIdOrInternalId: 'non_existent',\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n try {\n await deleteLayoutUseCase.execute(command);\n expect.fail('Should have thrown an error');\n } catch (thrownError) {\n expect(thrownError.message).to.equal('Layout not found');\n }\n });\n\n it('should propagate error from step controls cleanup', async () => {\n const error = new Error('Database error');\n controlValuesRepositoryMock.update.rejects(error);\n\n const command = DeleteLayoutCommand.create({\n layoutIdOrInternalId: 'layout_identifier',\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n try {\n await deleteLayoutUseCase.execute(command);\n expect.fail('Should have thrown an error');\n } catch (thrownError) {\n expect(thrownError.message).to.equal('Database error');\n }\n });\n\n it('should propagate error from step controls update', async () => {\n const error = new Error('Update error');\n controlValuesRepositoryMock.update.rejects(error);\n\n const command = DeleteLayoutCommand.create({\n layoutIdOrInternalId: 'layout_identifier',\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n try {\n await deleteLayoutUseCase.execute(command);\n expect.fail('Should have thrown an error');\n } catch (thrownError) {\n expect(thrownError.message).to.equal('Update error');\n }\n });\n\n it('should propagate error from layout deletion', async () => {\n const error = new Error('Delete error');\n layoutRepositoryMock.deleteLayout.rejects(error);\n\n const command = DeleteLayoutCommand.create({\n layoutIdOrInternalId: 'layout_identifier',\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n try {\n await deleteLayoutUseCase.execute(command);\n expect.fail('Should have thrown an error');\n } catch (thrownError) {\n expect(thrownError.message).to.equal('Delete error');\n }\n });\n\n it('should validate deletion order: step controls cleanup before layout deletion', async () => {\n const command = DeleteLayoutCommand.create({\n layoutIdOrInternalId: 'layout_identifier',\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n await deleteLayoutUseCase.execute(command);\n\n // Verify step controls update was called before layout deletion\n expect(controlValuesRepositoryMock.update.calledBefore(layoutRepositoryMock.deleteLayout)).to.be.true;\n });\n });\n});\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/delete-layout/delete-layout.use-case.ts",
"content": "import { ConflictException, Injectable } from '@nestjs/common';\nimport { ModuleRef } from '@nestjs/core';\nimport {\n AnalyticsService,\n GetLayoutCommand,\n GetLayoutUseCase,\n LayoutResponseDto,\n PinoLogger,\n} from '@novu/application-generic';\nimport { ControlValuesRepository, LayoutRepository, LocalizationResourceEnum } from '@novu/dal';\nimport { ControlValuesLevelEnum } from '@novu/shared';\nimport { DeleteLayoutCommand } from './delete-layout.command';\n\n@Injectable()\nexport class DeleteLayoutUseCase {\n constructor(\n private getLayoutUseCase: GetLayoutUseCase,\n private layoutRepository: LayoutRepository,\n private controlValuesRepository: ControlValuesRepository,\n private analyticsService: AnalyticsService,\n private moduleRef: ModuleRef,\n private logger: PinoLogger\n ) {}\n\n async execute(command: DeleteLayoutCommand): Promise {\n const { environmentId, organizationId, userId } = command;\n const layout = await this.getLayoutUseCase.execute(\n GetLayoutCommand.create({\n layoutIdOrInternalId: command.layoutIdOrInternalId,\n environmentId,\n organizationId,\n userId,\n skipAdditionalFields: true,\n })\n );\n\n if (layout.isDefault) {\n throw new ConflictException(\n `Layout with id ${command.layoutIdOrInternalId} is being used as a default layout, it can not be deleted`\n );\n }\n\n await this.removeLayoutReferencesFromStepControls({\n layoutId: layout.layoutId!,\n environmentId,\n organizationId,\n });\n\n await this.deleteTranslationGroup(layout, command);\n\n await this.layoutRepository.deleteLayout(layout._id!, environmentId, organizationId);\n\n await this.controlValuesRepository.delete({\n _environmentId: environmentId,\n _organizationId: organizationId,\n _layoutId: layout._id!,\n level: ControlValuesLevelEnum.LAYOUT_CONTROLS,\n });\n\n this.analyticsService.track('Delete layout - [Layouts]', userId, {\n _organizationId: organizationId,\n _environmentId: environmentId,\n layoutId: layout._id!,\n });\n }\n\n private async removeLayoutReferencesFromStepControls({\n layoutId,\n environmentId,\n organizationId,\n }: {\n layoutId: string;\n environmentId: string;\n organizationId: string;\n }): Promise {\n await this.controlValuesRepository.update(\n {\n level: ControlValuesLevelEnum.STEP_CONTROLS,\n _environmentId: environmentId,\n _organizationId: organizationId,\n 'controls.layoutId': layoutId,\n },\n { $unset: { 'controls.layoutId': '' } }\n );\n }\n\n private async deleteTranslationGroup(layout: LayoutResponseDto, command: DeleteLayoutCommand) {\n const isEnterprise = process.env.NOVU_ENTERPRISE === 'true' || process.env.CI_EE_TEST === 'true';\n const isSelfHosted = process.env.IS_SELF_HOSTED === 'true';\n\n if (!isEnterprise || isSelfHosted) {\n return;\n }\n\n try {\n const deleteTranslationGroupUseCase = this.moduleRef.get(\n require('@novu/ee-translation')?.DeleteTranslationGroup,\n {\n strict: false,\n }\n );\n\n await deleteTranslationGroupUseCase.execute({\n resourceId: layout.layoutId,\n resourceType: LocalizationResourceEnum.LAYOUT,\n organizationId: command.organizationId,\n environmentId: command.environmentId,\n userId: command.userId,\n });\n } catch (error) {\n this.logger.error(`Failed to delete translations for layout`, {\n layoutId: layout.layoutId,\n organizationId: command.organizationId,\n error: error instanceof Error ? error.message : String(error),\n });\n\n // translation group might not be present, so we can ignore the error\n }\n }\n}\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/delete-layout/index.ts",
"content": "export * from './delete-layout.command';\nexport * from './delete-layout.use-case';\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/duplicate-layout/duplicate-layout.command.ts",
"content": "import { EnvironmentWithUserCommand } from '@novu/application-generic';\nimport { Type } from 'class-transformer';\nimport { IsDefined, IsString, ValidateNested } from 'class-validator';\nimport { DuplicateLayoutDto } from '../../dtos';\n\nexport class DuplicateLayoutCommand extends EnvironmentWithUserCommand {\n @IsString()\n @IsDefined()\n layoutIdOrInternalId: string;\n\n @ValidateNested()\n @Type(() => DuplicateLayoutDto)\n overrides: DuplicateLayoutDto;\n}\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/duplicate-layout/duplicate-layout.use-case.spec.ts",
"content": "import { ModuleRef } from '@nestjs/core';\nimport { AnalyticsService, GetLayoutUseCase, PinoLogger } from '@novu/application-generic';\nimport { ControlValuesRepository } from '@novu/dal';\nimport { ChannelTypeEnum, ControlValuesLevelEnum, ResourceOriginEnum, ResourceTypeEnum } from '@novu/shared';\nimport { expect } from 'chai';\nimport sinon from 'sinon';\nimport { UpsertLayout } from '../upsert-layout';\nimport { DuplicateLayoutCommand } from './duplicate-layout.command';\nimport { DuplicateLayoutUseCase } from './duplicate-layout.use-case';\n\ndescribe('DuplicateLayoutUseCase', () => {\n let getLayoutUseCaseMock: sinon.SinonStubbedInstance;\n let upsertLayoutUseCaseMock: sinon.SinonStubbedInstance;\n let controlValuesRepositoryMock: sinon.SinonStubbedInstance;\n let analyticsServiceMock: sinon.SinonStubbedInstance;\n let moduleRefMock: sinon.SinonStubbedInstance;\n let pinoLoggerMock: sinon.SinonStubbedInstance;\n let duplicateLayoutUseCase: DuplicateLayoutUseCase;\n\n const mockUser = {\n _id: 'user_id',\n environmentId: 'env_id',\n organizationId: 'org_id',\n };\n\n const mockOriginalLayout = {\n _id: 'original_layout_id',\n identifier: 'original_layout_identifier',\n name: 'Original Layout',\n isDefault: false,\n createdAt: '2023-01-01T00:00:00Z',\n updatedAt: '2023-01-01T00:00:00Z',\n _environmentId: 'env_id',\n _organizationId: 'org_id',\n origin: ResourceOriginEnum.NOVU_CLOUD,\n type: ResourceTypeEnum.BRIDGE,\n channel: ChannelTypeEnum.EMAIL,\n };\n\n const mockOriginalControlValues = {\n _id: 'original_control_values_id',\n _environmentId: 'env_id',\n _organizationId: 'org_id',\n _layoutId: 'original_layout_id',\n level: ControlValuesLevelEnum.LAYOUT_CONTROLS,\n controls: {\n email: {\n body: '{{content}}',\n subject: 'Original Subject',\n },\n },\n };\n\n const mockDuplicatedLayout = {\n _id: 'duplicated_layout_id',\n layoutId: 'duplicated_layout_identifier',\n name: 'Duplicated Layout',\n isDefault: false,\n createdAt: '2023-01-02T00:00:00Z',\n updatedAt: '2023-01-02T00:00:00Z',\n _environmentId: 'env_id',\n _organizationId: 'org_id',\n origin: ResourceOriginEnum.NOVU_CLOUD,\n type: ResourceTypeEnum.BRIDGE,\n controls: {\n schema: {},\n values: {\n email: mockOriginalControlValues.controls.email,\n },\n },\n };\n\n const mockOverrides = {\n name: 'Duplicated Layout',\n };\n\n beforeEach(() => {\n getLayoutUseCaseMock = sinon.createStubInstance(GetLayoutUseCase);\n upsertLayoutUseCaseMock = sinon.createStubInstance(UpsertLayout);\n controlValuesRepositoryMock = sinon.createStubInstance(ControlValuesRepository);\n analyticsServiceMock = sinon.createStubInstance(AnalyticsService);\n moduleRefMock = sinon.createStubInstance(ModuleRef);\n pinoLoggerMock = sinon.createStubInstance(PinoLogger);\n\n duplicateLayoutUseCase = new DuplicateLayoutUseCase(\n getLayoutUseCaseMock as any,\n upsertLayoutUseCaseMock as any,\n controlValuesRepositoryMock as any,\n analyticsServiceMock as any,\n moduleRefMock as any,\n pinoLoggerMock as any\n );\n\n // Default mocks\n getLayoutUseCaseMock.execute.resolves(mockOriginalLayout as any);\n controlValuesRepositoryMock.findOne.resolves(mockOriginalControlValues as any);\n upsertLayoutUseCaseMock.execute.resolves(mockDuplicatedLayout as any);\n });\n\n afterEach(() => {\n sinon.restore();\n });\n\n describe('execute', () => {\n it('should successfully duplicate layout with control values', async () => {\n const command = DuplicateLayoutCommand.create({\n layoutIdOrInternalId: 'original_layout_identifier',\n overrides: mockOverrides,\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n const result = await duplicateLayoutUseCase.execute(command);\n\n expect(result).to.deep.equal(mockDuplicatedLayout);\n\n // Verify v1 use case was called with correct parameters\n expect(getLayoutUseCaseMock.execute.calledOnce).to.be.true;\n const v1Command = getLayoutUseCaseMock.execute.firstCall.args[0];\n expect(v1Command.layoutIdOrInternalId).to.equal('original_layout_identifier');\n expect(v1Command.environmentId).to.equal('env_id');\n expect(v1Command.organizationId).to.equal('org_id');\n expect(v1Command.skipAdditionalFields).to.be.true;\n\n // Verify control values repository was called\n expect(controlValuesRepositoryMock.findOne.calledOnce).to.be.true;\n expect(controlValuesRepositoryMock.findOne.firstCall.args[0]).to.deep.equal({\n _environmentId: 'env_id',\n _organizationId: 'org_id',\n _layoutId: 'original_layout_id',\n level: ControlValuesLevelEnum.LAYOUT_CONTROLS,\n });\n\n // Verify upsert use case was called with correct parameters\n expect(upsertLayoutUseCaseMock.execute.calledOnce).to.be.true;\n const upsertCommand = upsertLayoutUseCaseMock.execute.firstCall.args[0];\n expect(upsertCommand.layoutDto.name).to.equal('Duplicated Layout');\n expect(upsertCommand.layoutDto.controlValues).to.deep.equal(mockOriginalControlValues.controls);\n expect(upsertCommand.userId).to.deep.equal(mockUser._id);\n expect(upsertCommand.environmentId).to.deep.equal(mockUser.environmentId);\n expect(upsertCommand.organizationId).to.deep.equal(mockUser.organizationId);\n });\n\n it('should duplicate layout without control values when none exist', async () => {\n controlValuesRepositoryMock.findOne.resolves(null);\n\n const command = DuplicateLayoutCommand.create({\n layoutIdOrInternalId: 'original_layout_identifier',\n overrides: mockOverrides,\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n const result = await duplicateLayoutUseCase.execute(command);\n\n expect(result).to.deep.equal(mockDuplicatedLayout);\n\n // Verify control values repository was called\n expect(controlValuesRepositoryMock.findOne.calledOnce).to.be.true;\n\n // Verify upsert use case was called with null control values\n expect(upsertLayoutUseCaseMock.execute.calledOnce).to.be.true;\n const upsertCommand = upsertLayoutUseCaseMock.execute.firstCall.args[0];\n expect(upsertCommand.layoutDto.controlValues).to.be.null;\n });\n\n it('should handle empty control values controls', async () => {\n const controlValuesWithEmptyControls = {\n ...mockOriginalControlValues,\n controls: undefined,\n };\n controlValuesRepositoryMock.findOne.resolves(controlValuesWithEmptyControls as any);\n\n const command = DuplicateLayoutCommand.create({\n layoutIdOrInternalId: 'original_layout_identifier',\n overrides: mockOverrides,\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n const result = await duplicateLayoutUseCase.execute(command);\n\n expect(result).to.deep.equal(mockDuplicatedLayout);\n\n // Verify upsert use case was called with null control values\n expect(upsertLayoutUseCaseMock.execute.calledOnce).to.be.true;\n const upsertCommand = upsertLayoutUseCaseMock.execute.firstCall.args[0];\n expect(upsertCommand.layoutDto.controlValues).to.be.null;\n });\n\n it('should track analytics event', async () => {\n const command = DuplicateLayoutCommand.create({\n layoutIdOrInternalId: 'original_layout_identifier',\n overrides: mockOverrides,\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n await duplicateLayoutUseCase.execute(command);\n\n expect(analyticsServiceMock.track.calledOnce).to.be.true;\n expect(analyticsServiceMock.track.firstCall.args[0]).to.equal('Duplicate layout - [Layouts]');\n expect(analyticsServiceMock.track.firstCall.args[1]).to.equal('user_id');\n expect(analyticsServiceMock.track.firstCall.args[2]).to.deep.equal({\n _organizationId: 'org_id',\n _environmentId: 'env_id',\n originalLayoutId: 'original_layout_id',\n duplicatedLayoutId: 'duplicated_layout_id',\n });\n });\n\n it('should use override name correctly', async () => {\n const customOverrides = {\n name: 'Custom Duplicated Name',\n };\n\n const command = DuplicateLayoutCommand.create({\n layoutIdOrInternalId: 'original_layout_identifier',\n overrides: customOverrides,\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n await duplicateLayoutUseCase.execute(command);\n\n expect(upsertLayoutUseCaseMock.execute.calledOnce).to.be.true;\n const upsertCommand = upsertLayoutUseCaseMock.execute.firstCall.args[0];\n expect(upsertCommand.layoutDto.name).to.equal('Custom Duplicated Name');\n });\n\n it('should propagate error from v1 use case', async () => {\n const error = new Error('Layout not found');\n getLayoutUseCaseMock.execute.rejects(error);\n\n const command = DuplicateLayoutCommand.create({\n layoutIdOrInternalId: 'non_existent',\n overrides: mockOverrides,\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n try {\n await duplicateLayoutUseCase.execute(command);\n expect.fail('Should have thrown an error');\n } catch (thrownError) {\n expect(thrownError.message).to.equal('Layout not found');\n }\n });\n\n it('should propagate error from control values repository', async () => {\n const error = new Error('Database error');\n controlValuesRepositoryMock.findOne.rejects(error);\n\n const command = DuplicateLayoutCommand.create({\n layoutIdOrInternalId: 'original_layout_identifier',\n overrides: mockOverrides,\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n try {\n await duplicateLayoutUseCase.execute(command);\n expect.fail('Should have thrown an error');\n } catch (thrownError) {\n expect(thrownError.message).to.equal('Database error');\n }\n });\n\n it('should propagate error from upsert use case', async () => {\n const error = new Error('Upsert error');\n upsertLayoutUseCaseMock.execute.rejects(error);\n\n const command = DuplicateLayoutCommand.create({\n layoutIdOrInternalId: 'original_layout_identifier',\n overrides: mockOverrides,\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n try {\n await duplicateLayoutUseCase.execute(command);\n expect.fail('Should have thrown an error');\n } catch (thrownError) {\n expect(thrownError.message).to.equal('Upsert error');\n }\n });\n\n it('should validate execution order: get original before duplicate creation', async () => {\n const command = DuplicateLayoutCommand.create({\n layoutIdOrInternalId: 'original_layout_identifier',\n overrides: mockOverrides,\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n await duplicateLayoutUseCase.execute(command);\n\n // Verify original layout was fetched before duplication\n expect(getLayoutUseCaseMock.execute.calledBefore(upsertLayoutUseCaseMock.execute)).to.be.true;\n expect(controlValuesRepositoryMock.findOne.calledBefore(upsertLayoutUseCaseMock.execute)).to.be.true;\n });\n\n it('should preserve original layout control values structure', async () => {\n const complexControlValues = {\n ...mockOriginalControlValues,\n controls: {\n email: {\n body: '{{content}}',\n subject: 'Complex Subject {{payload.name}}',\n preheader: 'Preview text',\n customField: 'custom value',\n },\n },\n };\n controlValuesRepositoryMock.findOne.resolves(complexControlValues as any);\n\n const command = DuplicateLayoutCommand.create({\n layoutIdOrInternalId: 'original_layout_identifier',\n overrides: mockOverrides,\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n });\n\n await duplicateLayoutUseCase.execute(command);\n\n expect(upsertLayoutUseCaseMock.execute.calledOnce).to.be.true;\n const upsertCommand = upsertLayoutUseCaseMock.execute.firstCall.args[0];\n expect(upsertCommand.layoutDto.controlValues).to.deep.equal(complexControlValues.controls);\n });\n });\n});\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/duplicate-layout/duplicate-layout.use-case.ts",
"content": "import { Injectable } from '@nestjs/common';\nimport { ModuleRef } from '@nestjs/core';\nimport {\n AnalyticsService,\n GetLayoutCommand,\n GetLayoutUseCase,\n LayoutResponseDto,\n PinoLogger,\n} from '@novu/application-generic';\nimport { ControlValuesRepository, LocalizationResourceEnum } from '@novu/dal';\nimport { ControlValuesLevelEnum } from '@novu/shared';\nimport { UpsertLayout, UpsertLayoutCommand } from '../upsert-layout';\nimport { DuplicateLayoutCommand } from './duplicate-layout.command';\n\n@Injectable()\nexport class DuplicateLayoutUseCase {\n constructor(\n private getLayoutUseCase: GetLayoutUseCase,\n private upsertLayoutUseCase: UpsertLayout,\n private controlValuesRepository: ControlValuesRepository,\n private analyticsService: AnalyticsService,\n private moduleRef: ModuleRef,\n private logger: PinoLogger\n ) {}\n\n async execute(command: DuplicateLayoutCommand): Promise {\n const originalLayout = await this.getLayoutUseCase.execute(\n GetLayoutCommand.create({\n layoutIdOrInternalId: command.layoutIdOrInternalId,\n environmentId: command.environmentId,\n organizationId: command.organizationId,\n userId: command.userId,\n skipAdditionalFields: true,\n })\n );\n\n const originalControlValues = await this.controlValuesRepository.findOne({\n _environmentId: command.environmentId,\n _organizationId: command.organizationId,\n _layoutId: originalLayout._id!,\n level: ControlValuesLevelEnum.LAYOUT_CONTROLS,\n });\n\n const duplicatedLayout = await this.upsertLayoutUseCase.execute(\n UpsertLayoutCommand.create({\n layoutDto: {\n name: command.overrides.name,\n isTranslationEnabled: command.overrides.isTranslationEnabled,\n controlValues: originalControlValues?.controls ?? null,\n },\n environmentId: command.environmentId,\n organizationId: command.organizationId,\n userId: command.userId,\n })\n );\n\n this.analyticsService.track('Duplicate layout - [Layouts]', command.userId, {\n _organizationId: command.organizationId,\n _environmentId: command.environmentId,\n originalLayoutId: originalLayout._id!,\n duplicatedLayoutId: duplicatedLayout._id,\n });\n\n if (duplicatedLayout.isTranslationEnabled) {\n await this.duplicateTranslationsForLayout({\n sourceResourceId: originalLayout.layoutId,\n targetResourceId: duplicatedLayout.layoutId,\n command,\n });\n }\n\n return duplicatedLayout;\n }\n\n private async duplicateTranslationsForLayout({\n sourceResourceId,\n targetResourceId,\n command,\n }: {\n sourceResourceId: string;\n targetResourceId: string;\n command: DuplicateLayoutCommand;\n }) {\n const isEnterprise = process.env.NOVU_ENTERPRISE === 'true' || process.env.CI_EE_TEST === 'true';\n const isSelfHosted = process.env.IS_SELF_HOSTED === 'true';\n\n if (!isEnterprise || isSelfHosted) {\n return;\n }\n\n try {\n const duplicateLocales = this.moduleRef.get(require('@novu/ee-translation')?.DuplicateLocales, {\n strict: false,\n });\n\n await duplicateLocales.execute({\n sourceResourceId,\n sourceResourceType: LocalizationResourceEnum.LAYOUT,\n targetResourceId,\n organizationId: command.organizationId,\n environmentId: command.environmentId,\n userId: command.userId,\n });\n } catch (error) {\n this.logger.error(`Failed to duplicate translations for layout`, {\n sourceResourceId,\n targetResourceId,\n organizationId: command.organizationId,\n error: error instanceof Error ? error.message : String(error),\n });\n\n throw error;\n }\n }\n}\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/duplicate-layout/index.ts",
"content": "export * from './duplicate-layout.command';\nexport * from './duplicate-layout.use-case';\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/get-layout-usage/get-layout-usage.command.ts",
"content": "import { EnvironmentCommand } from '@novu/application-generic';\nimport { IsString } from 'class-validator';\n\nexport class GetLayoutUsageCommand extends EnvironmentCommand {\n @IsString()\n layoutIdOrInternalId: string;\n}\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/get-layout-usage/get-layout-usage.usecase.ts",
"content": "import { Injectable } from '@nestjs/common';\nimport { GetLayoutCommand, GetLayoutUseCase, InstrumentUsecase } from '@novu/application-generic';\nimport { ControlValuesRepository, NotificationTemplateRepository } from '@novu/dal';\nimport { ControlValuesLevelEnum } from '@novu/shared';\nimport { GetLayoutUsageResponseDto, WorkflowInfoDto } from '../../dtos';\nimport { GetLayoutUsageCommand } from './get-layout-usage.command';\n\n@Injectable()\nexport class GetLayoutUsageUseCase {\n constructor(\n private controlValuesRepository: ControlValuesRepository,\n private notificationTemplateRepository: NotificationTemplateRepository,\n private getLayoutUseCase: GetLayoutUseCase\n ) {}\n\n @InstrumentUsecase()\n async execute(command: GetLayoutUsageCommand): Promise {\n // First, resolve the layout to get its internal ID\n const layout = await this.getLayoutUseCase.execute(\n GetLayoutCommand.create({\n layoutIdOrInternalId: command.layoutIdOrInternalId,\n environmentId: command.environmentId,\n organizationId: command.organizationId,\n skipAdditionalFields: true,\n })\n );\n\n const workflows: WorkflowInfoDto[] = [];\n\n // Get control values that reference this layout\n const controlValues = await this.controlValuesRepository.find({\n _environmentId: command.environmentId,\n _organizationId: command.organizationId,\n level: ControlValuesLevelEnum.STEP_CONTROLS,\n 'controls.layoutId': layout.layoutId,\n });\n\n // Get unique workflow IDs from the control values\n const workflowIds = [...new Set(controlValues.map((cv) => cv._workflowId).filter(Boolean))] as string[];\n\n // Fetch workflow information for each workflow ID\n for (const workflowId of workflowIds) {\n try {\n const workflow = await this.notificationTemplateRepository.findById(workflowId, command.environmentId);\n\n if (workflow && workflow.triggers && workflow.triggers.length > 0) {\n workflows.push({\n name: workflow.name,\n workflowId: workflow.triggers[0].identifier,\n });\n }\n } catch (error) {}\n }\n\n return {\n workflows,\n };\n }\n}\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/get-layout-usage/index.ts",
"content": "export * from './get-layout-usage.command';\nexport * from './get-layout-usage.usecase';\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/index.ts",
"content": "import { GetLayoutUseCase, GetLayoutUseCaseV0, LayoutVariablesSchemaUseCase } from '@novu/application-generic';\nimport { BuildLayoutIssuesUsecase } from './build-layout-issues/build-layout-issues.usecase';\nimport { DeleteLayoutUseCase } from './delete-layout';\nimport { DuplicateLayoutUseCase } from './duplicate-layout';\nimport { GetLayoutUsageUseCase } from './get-layout-usage';\nimport { ListLayoutsUseCase } from './list-layouts';\nimport { PreviewLayoutUsecase } from './preview-layout';\nimport { LayoutSyncToEnvironmentUseCase } from './sync-to-environment';\nimport { UpsertLayout } from './upsert-layout';\n\nexport const USE_CASES = [\n UpsertLayout,\n GetLayoutUseCaseV0,\n GetLayoutUseCase,\n DeleteLayoutUseCase,\n DuplicateLayoutUseCase,\n ListLayoutsUseCase,\n LayoutVariablesSchemaUseCase,\n PreviewLayoutUsecase,\n GetLayoutUsageUseCase,\n BuildLayoutIssuesUsecase,\n LayoutSyncToEnvironmentUseCase,\n];\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/list-layouts/index.ts",
"content": "export * from './list-layouts.command';\nexport * from './list-layouts.use-case';\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/list-layouts/list-layouts.command.ts",
"content": "import { PaginatedListCommand } from '@novu/application-generic';\nimport { IsOptional, IsString } from 'class-validator';\n\nexport class ListLayoutsCommand extends PaginatedListCommand {\n @IsString()\n @IsOptional()\n searchQuery?: string;\n}\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/list-layouts/list-layouts.use-case.spec.ts",
"content": "import { LayoutEntity, LayoutRepository } from '@novu/dal';\nimport { ChannelTypeEnum, DirectionEnum, ResourceOriginEnum, ResourceTypeEnum } from '@novu/shared';\nimport { expect } from 'chai';\nimport sinon from 'sinon';\nimport { ListLayoutsCommand } from './list-layouts.command';\nimport { ListLayoutsUseCase } from './list-layouts.use-case';\n\ndescribe('ListLayoutsUseCase', () => {\n let layoutRepositoryMock: sinon.SinonStubbedInstance;\n let listLayoutsUseCase: ListLayoutsUseCase;\n let mapSpy: sinon.SinonSpy;\n\n const mockUser = {\n _id: 'user_id',\n environmentId: 'env_id',\n organizationId: 'org_id',\n };\n\n const mockLayoutEntity: LayoutEntity = {\n _id: 'layout_id_1',\n identifier: 'layout_identifier_1',\n name: 'Test Layout 1',\n isDefault: false,\n channel: ChannelTypeEnum.EMAIL,\n content: '{{content}}',\n contentType: 'customHtml',\n updatedAt: '2023-01-02T00:00:00.000Z',\n createdAt: '2023-01-01T00:00:00.000Z',\n _environmentId: 'env_id',\n _organizationId: 'org_id',\n _creatorId: 'creator_id',\n deleted: false,\n origin: ResourceOriginEnum.NOVU_CLOUD,\n type: ResourceTypeEnum.BRIDGE,\n controls: {\n schema: {},\n uiSchema: {},\n },\n };\n\n const mockLayoutEntity2: LayoutEntity = {\n _id: 'layout_id_2',\n identifier: 'layout_identifier_2',\n name: 'Test Layout 2',\n isDefault: true,\n channel: ChannelTypeEnum.EMAIL,\n content: '{{content}}',\n contentType: 'customHtml',\n updatedAt: '2023-01-02T00:00:00.000Z',\n createdAt: '2023-01-01T00:00:00.000Z',\n _environmentId: 'env_id',\n _organizationId: 'org_id',\n _creatorId: 'creator_id',\n deleted: false,\n origin: ResourceOriginEnum.NOVU_CLOUD,\n type: ResourceTypeEnum.BRIDGE,\n controls: {\n schema: {},\n uiSchema: {},\n },\n };\n\n const mockRepositoryResponse = {\n data: [mockLayoutEntity, mockLayoutEntity2],\n totalCount: 2,\n };\n\n beforeEach(() => {\n layoutRepositoryMock = sinon.createStubInstance(LayoutRepository);\n\n listLayoutsUseCase = new ListLayoutsUseCase(layoutRepositoryMock as any);\n mapSpy = sinon.spy(listLayoutsUseCase as any, 'mapLayoutToResponseDto');\n\n layoutRepositoryMock.getV2List.resolves(mockRepositoryResponse);\n });\n\n afterEach(() => {\n sinon.restore();\n });\n\n describe('execute', () => {\n it('should successfully list layouts with default parameters', async () => {\n const command = ListLayoutsCommand.create({\n user: mockUser as any,\n offset: 0,\n limit: 10,\n orderBy: 'createdAt',\n orderDirection: DirectionEnum.DESC,\n });\n\n const result = await listLayoutsUseCase.execute(command);\n\n expect(result.totalCount).to.equal(2);\n expect(result.layouts).to.have.length(2);\n expect(result.layouts[0]._id).to.equal('layout_id_1');\n expect(result.layouts[0].layoutId).to.equal('layout_identifier_1');\n expect(result.layouts[0].name).to.equal('Test Layout 1');\n expect(result.layouts[1]._id).to.equal('layout_id_2');\n expect(result.layouts[1].layoutId).to.equal('layout_identifier_2');\n expect(result.layouts[1].name).to.equal('Test Layout 2');\n\n expect(layoutRepositoryMock.getV2List.calledOnce).to.be.true;\n const repositoryCall = layoutRepositoryMock.getV2List.firstCall.args[0];\n expect(repositoryCall).to.deep.equal({\n organizationId: 'org_id',\n environmentId: 'env_id',\n skip: 0,\n limit: 10,\n searchQuery: undefined,\n orderBy: 'createdAt',\n orderDirection: DirectionEnum.DESC,\n });\n });\n\n it('should handle search query parameter', async () => {\n const command = ListLayoutsCommand.create({\n user: mockUser as any,\n offset: 0,\n limit: 10,\n orderBy: 'name',\n orderDirection: DirectionEnum.ASC,\n searchQuery: 'test search',\n });\n\n await listLayoutsUseCase.execute(command);\n\n const repositoryCall = layoutRepositoryMock.getV2List.firstCall.args[0];\n expect(repositoryCall.searchQuery).to.equal('test search');\n expect(repositoryCall.orderBy).to.equal('name');\n expect(repositoryCall.orderDirection).to.equal(DirectionEnum.ASC);\n });\n\n it('should handle pagination parameters', async () => {\n const command = ListLayoutsCommand.create({\n user: mockUser as any,\n offset: 20,\n limit: 5,\n orderBy: 'updatedAt',\n orderDirection: DirectionEnum.DESC,\n });\n\n await listLayoutsUseCase.execute(command);\n\n const repositoryCall = layoutRepositoryMock.getV2List.firstCall.args[0];\n expect(repositoryCall.skip).to.equal(20);\n expect(repositoryCall.limit).to.equal(5);\n expect(repositoryCall.orderBy).to.equal('updatedAt');\n });\n\n it('should return empty result when repository returns null data', async () => {\n layoutRepositoryMock.getV2List.resolves({ data: null, totalCount: 0 });\n\n const command = ListLayoutsCommand.create({\n user: mockUser as any,\n offset: 0,\n limit: 10,\n orderBy: 'createdAt',\n orderDirection: DirectionEnum.DESC,\n });\n\n const result = await listLayoutsUseCase.execute(command);\n\n expect(result).to.deep.equal({\n layouts: [],\n totalCount: 0,\n });\n });\n\n it('should return empty result when repository returns undefined data', async () => {\n layoutRepositoryMock.getV2List.resolves({ data: undefined, totalCount: 0 });\n\n const command = ListLayoutsCommand.create({\n user: mockUser as any,\n offset: 0,\n limit: 10,\n orderBy: 'createdAt',\n orderDirection: DirectionEnum.DESC,\n });\n\n const result = await listLayoutsUseCase.execute(command);\n\n expect(result).to.deep.equal({\n layouts: [],\n totalCount: 0,\n });\n });\n\n it('should handle empty data array', async () => {\n layoutRepositoryMock.getV2List.resolves({ data: [], totalCount: 0 });\n\n const command = ListLayoutsCommand.create({\n user: mockUser as any,\n offset: 0,\n limit: 10,\n orderBy: 'createdAt',\n orderDirection: DirectionEnum.DESC,\n });\n\n const result = await listLayoutsUseCase.execute(command);\n\n expect(result).to.deep.equal({\n layouts: [],\n totalCount: 0,\n });\n });\n\n it('should propagate repository errors', async () => {\n const error = new Error('Database connection failed');\n layoutRepositoryMock.getV2List.rejects(error);\n\n const command = ListLayoutsCommand.create({\n user: mockUser as any,\n offset: 0,\n limit: 10,\n orderBy: 'createdAt',\n orderDirection: DirectionEnum.DESC,\n });\n\n try {\n await listLayoutsUseCase.execute(command);\n expect.fail('Should have thrown an error');\n } catch (thrownError) {\n expect(thrownError.message).to.equal('Database connection failed');\n }\n });\n\n it('should call mapToResponseDto for each layout', async () => {\n const command = ListLayoutsCommand.create({\n user: mockUser as any,\n offset: 0,\n limit: 10,\n orderBy: 'createdAt',\n orderDirection: DirectionEnum.DESC,\n });\n\n const result = await listLayoutsUseCase.execute(command);\n\n expect(mapSpy.calledTwice).to.be.true;\n expect(result.layouts).to.have.length(2);\n expect(result.layouts[0]._id).to.equal('layout_id_1');\n expect(result.layouts[0].layoutId).to.equal('layout_identifier_1');\n expect(result.layouts[1]._id).to.equal('layout_id_2');\n expect(result.layouts[1].layoutId).to.equal('layout_identifier_2');\n });\n\n it('should handle single layout in result', async () => {\n const singleLayoutResponse = {\n data: [mockLayoutEntity],\n totalCount: 1,\n };\n layoutRepositoryMock.getV2List.resolves(singleLayoutResponse);\n\n const command = ListLayoutsCommand.create({\n user: mockUser as any,\n offset: 0,\n limit: 10,\n orderBy: 'createdAt',\n orderDirection: DirectionEnum.DESC,\n });\n\n const result = await listLayoutsUseCase.execute(command);\n\n expect(result.totalCount).to.equal(1);\n expect(result.layouts).to.have.length(1);\n expect(result.layouts[0]._id).to.equal('layout_id_1');\n expect(result.layouts[0].layoutId).to.equal('layout_identifier_1');\n expect(result.layouts[0].name).to.equal('Test Layout 1');\n expect(mapSpy.calledOnce).to.be.true;\n });\n\n it('should preserve totalCount from repository response', async () => {\n const responseWithDifferentTotal = {\n data: [mockLayoutEntity],\n totalCount: 100,\n };\n layoutRepositoryMock.getV2List.resolves(responseWithDifferentTotal);\n\n const command = ListLayoutsCommand.create({\n user: mockUser as any,\n offset: 50,\n limit: 10,\n orderBy: 'createdAt',\n orderDirection: DirectionEnum.DESC,\n });\n\n const result = await listLayoutsUseCase.execute(command);\n\n expect(result.totalCount).to.equal(100);\n expect(result.layouts).to.have.length(1);\n });\n\n it('should handle layouts with deleted flag correctly', async () => {\n const deletedLayoutEntity = {\n ...mockLayoutEntity,\n deleted: true,\n };\n\n const responseWithDeletedLayout = {\n data: [deletedLayoutEntity],\n totalCount: 1,\n };\n layoutRepositoryMock.getV2List.resolves(responseWithDeletedLayout);\n\n const command = ListLayoutsCommand.create({\n user: mockUser as any,\n offset: 0,\n limit: 10,\n orderBy: 'createdAt',\n orderDirection: DirectionEnum.DESC,\n });\n\n await listLayoutsUseCase.execute(command);\n\n expect(mapSpy.calledOnce).to.be.true;\n const mappedEntity = mapSpy.firstCall.args[0];\n expect(mappedEntity.deleted).to.be.true;\n });\n\n it('should handle layouts without controls', async () => {\n const layoutWithoutControls = {\n ...mockLayoutEntity,\n controls: undefined,\n };\n\n const responseWithoutControls = {\n data: [layoutWithoutControls],\n totalCount: 1,\n };\n layoutRepositoryMock.getV2List.resolves(responseWithoutControls);\n\n const command = ListLayoutsCommand.create({\n user: mockUser as any,\n offset: 0,\n limit: 10,\n orderBy: 'createdAt',\n orderDirection: DirectionEnum.DESC,\n });\n\n const result = await listLayoutsUseCase.execute(command);\n\n expect(result.layouts).to.have.length(1);\n expect(result.layouts[0].controls.values).to.deep.equal({});\n });\n\n it('should correctly map entity properties to DTO', async () => {\n const command = ListLayoutsCommand.create({\n user: mockUser as any,\n offset: 0,\n limit: 10,\n orderBy: 'createdAt',\n orderDirection: DirectionEnum.DESC,\n });\n\n const result = await listLayoutsUseCase.execute(command);\n\n const layoutDto = result.layouts[0];\n\n expect(layoutDto._id).to.equal(mockLayoutEntity._id);\n expect(layoutDto.layoutId).to.equal(mockLayoutEntity.identifier);\n expect(layoutDto.name).to.equal(mockLayoutEntity.name);\n expect(layoutDto.isDefault).to.equal(mockLayoutEntity.isDefault);\n expect(layoutDto.origin).to.equal(mockLayoutEntity.origin);\n expect(layoutDto.type).to.equal(mockLayoutEntity.type);\n expect(layoutDto.updatedAt).to.equal(mockLayoutEntity.updatedAt);\n expect(layoutDto.createdAt).to.equal(mockLayoutEntity.createdAt);\n });\n });\n});\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/list-layouts/list-layouts.use-case.ts",
"content": "import { Injectable } from '@nestjs/common';\nimport { InstrumentUsecase, LayoutDtoV0, LayoutResponseDto, mapLayoutToResponseDto } from '@novu/application-generic';\nimport { LayoutEntity, LayoutRepository } from '@novu/dal';\nimport { ListLayoutResponseDto } from '../../dtos';\nimport { ListLayoutsCommand } from './list-layouts.command';\n\n@Injectable()\nexport class ListLayoutsUseCase {\n constructor(private layoutRepository: LayoutRepository) {}\n\n @InstrumentUsecase()\n async execute(command: ListLayoutsCommand): Promise {\n const res = await this.layoutRepository.getV2List({\n organizationId: command.user.organizationId,\n environmentId: command.user.environmentId,\n skip: command.offset,\n limit: command.limit,\n searchQuery: command.searchQuery,\n orderBy: command.orderBy ? command.orderBy : 'createdAt',\n orderDirection: command.orderDirection,\n });\n\n if (res.data === null || res.data === undefined) {\n return { layouts: [], totalCount: 0 };\n }\n\n const layoutDtos = res.data.map((layout) => this.mapLayoutToResponseDto(layout));\n\n return {\n layouts: layoutDtos,\n totalCount: res.totalCount,\n };\n }\n\n private mapLayoutToResponseDto(layout: LayoutEntity): LayoutResponseDto {\n const layoutDto = this.mapFromEntity(layout);\n\n return mapLayoutToResponseDto({\n layout: layoutDto,\n controlValues: null,\n variables: {},\n });\n }\n\n private mapFromEntity(layout: LayoutEntity): LayoutDtoV0 {\n return {\n ...layout,\n _id: layout._id,\n _organizationId: layout._organizationId,\n _environmentId: layout._environmentId,\n isDeleted: layout.deleted,\n controls: {},\n };\n }\n}\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/preview-layout/index.ts",
"content": "export * from './preview-layout.command';\nexport * from './preview-layout.usecase';\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/preview-layout/preview-layout.command.ts",
"content": "import { EnvironmentWithUserObjectCommand } from '@novu/application-generic';\nimport { LayoutPreviewRequestDto } from '../../dtos/layout-preview-request.dto';\n\nexport class PreviewLayoutCommand extends EnvironmentWithUserObjectCommand {\n layoutIdOrInternalId: string;\n layoutPreviewRequestDto: LayoutPreviewRequestDto;\n}\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/preview-layout/preview-layout.usecase.spec.ts",
"content": "import {\n ControlValueSanitizerService,\n CreateVariablesObject,\n GetLayoutUseCase,\n LayoutControlType,\n PayloadMergerService,\n PreviewPayloadProcessorService,\n PreviewStep,\n} from '@novu/application-generic';\nimport { EnvironmentRepository, EnvironmentVariableRepository } from '@novu/dal';\nimport {\n ChannelTypeEnum,\n LAYOUT_PREVIEW_EMAIL_STEP,\n LAYOUT_PREVIEW_WORKFLOW_ID,\n ResourceOriginEnum,\n} from '@novu/shared';\nimport { expect } from 'chai';\nimport sinon from 'sinon';\nimport { PreviewLayoutCommand } from './preview-layout.command';\nimport { PreviewLayoutUsecase } from './preview-layout.usecase';\nimport { enhanceBodyForPreview } from './preview-utils';\n\ndescribe('PreviewLayoutUsecase', () => {\n let getLayoutUseCaseMock: sinon.SinonStubbedInstance;\n let createVariablesObjectMock: sinon.SinonStubbedInstance;\n let controlValueSanitizerMock: sinon.SinonStubbedInstance;\n let payloadProcessorMock: sinon.SinonStubbedInstance;\n let payloadMergerMock: sinon.SinonStubbedInstance;\n let previewStepUsecaseMock: sinon.SinonStubbedInstance;\n let environmentVariableRepositoryMock: sinon.SinonStubbedInstance;\n let environmentRepositoryMock: sinon.SinonStubbedInstance;\n\n let previewLayoutUsecase: PreviewLayoutUsecase;\n\n const mockUser = {\n _id: 'user_id',\n environmentId: 'env_id',\n organizationId: 'org_id',\n };\n\n const mockLayout = {\n _id: 'layout_id',\n identifier: 'layout_identifier',\n name: 'Test Layout',\n controls: {\n values: {\n email: {\n body: '{{content}}',\n editorType: 'html',\n },\n },\n },\n variables: {\n name: { type: 'string', default: 'John' },\n email: { type: 'string', default: 'john@example.com' },\n },\n };\n\n const mockLayoutWithoutControls = {\n ...mockLayout,\n controls: {\n values: {},\n },\n variables: {},\n };\n\n const mockControlValues = {\n email: {\n body: 'Custom {{content}}',\n editorType: 'html',\n },\n };\n\n const mockVariablesObject = {\n name: 'Jane',\n email: 'jane@example.com',\n };\n\n const mockSanitizedControls = {\n email: {\n body: 'Sanitized {{content}}',\n editorType: 'html',\n },\n };\n\n const mockPreviewTemplateData = {\n controlValues: {\n email: {\n body: 'Processed {{content}}',\n editorType: 'html',\n },\n } as LayoutControlType,\n payloadExample: {\n content: 'Test content',\n user: { name: 'Test User' },\n },\n };\n\n const mockPayloadExample = {\n content: 'Merged content',\n user: { name: 'Merged User' },\n };\n\n const mockCleanedPayloadExample = {\n payload: { content: 'Cleaned content' },\n subscriber: { email: 'test@example.com' },\n };\n\n const mockPreviewStepOutput = {\n outputs: {\n body: 'Final rendered content',\n },\n };\n\n beforeEach(() => {\n getLayoutUseCaseMock = sinon.createStubInstance(GetLayoutUseCase);\n createVariablesObjectMock = sinon.createStubInstance(CreateVariablesObject);\n controlValueSanitizerMock = sinon.createStubInstance(ControlValueSanitizerService);\n payloadProcessorMock = sinon.createStubInstance(PreviewPayloadProcessorService);\n payloadMergerMock = sinon.createStubInstance(PayloadMergerService);\n previewStepUsecaseMock = sinon.createStubInstance(PreviewStep);\n environmentVariableRepositoryMock = sinon.createStubInstance(EnvironmentVariableRepository);\n environmentRepositoryMock = sinon.createStubInstance(EnvironmentRepository);\n\n previewLayoutUsecase = new PreviewLayoutUsecase(\n getLayoutUseCaseMock as any,\n createVariablesObjectMock as any,\n controlValueSanitizerMock as any,\n payloadProcessorMock as any,\n payloadMergerMock as any,\n previewStepUsecaseMock as any,\n environmentVariableRepositoryMock as any,\n environmentRepositoryMock as any\n );\n\n // Default mocks setup\n getLayoutUseCaseMock.execute.resolves(mockLayout as any);\n createVariablesObjectMock.execute.resolves(mockVariablesObject);\n controlValueSanitizerMock.sanitizeControlsForPreview.returns(mockSanitizedControls);\n controlValueSanitizerMock.processControlValues.returns({\n previewTemplateData: mockPreviewTemplateData,\n sanitizedControls: mockSanitizedControls,\n });\n payloadMergerMock.mergePayloadExample.resolves(mockPayloadExample);\n payloadProcessorMock.cleanPreviewExamplePayload.returns(mockCleanedPayloadExample);\n previewStepUsecaseMock.execute.resolves(mockPreviewStepOutput as any);\n environmentVariableRepositoryMock.findByEnvironment.resolves([]);\n environmentRepositoryMock.findByIdAndOrganization.resolves({\n name: 'Development',\n type: 'dev',\n } as any);\n });\n\n afterEach(() => {\n sinon.restore();\n });\n\n describe('execute', () => {\n it('should successfully execute with provided control values', async () => {\n const command = PreviewLayoutCommand.create({\n user: mockUser as any,\n layoutIdOrInternalId: 'layout_id',\n layoutPreviewRequestDto: {\n controlValues: mockControlValues,\n previewPayload: { subscriber: { email: 'test@example.com' } },\n },\n });\n\n const result = await previewLayoutUsecase.execute(command);\n\n expect(result.result).to.deep.equal({\n preview: { body: 'Final rendered content' },\n type: ChannelTypeEnum.EMAIL,\n });\n expect(result.previewPayloadExample).to.deep.equal(mockPayloadExample);\n expect(result.schema).to.exist;\n expect(result.schema?.type).to.equal('object');\n expect(result.schema?.properties).to.have.keys(['subscriber', 'context']);\n });\n\n it('should use layout control values when command control values are not provided', async () => {\n const command = PreviewLayoutCommand.create({\n user: mockUser as any,\n layoutIdOrInternalId: 'layout_id',\n layoutPreviewRequestDto: {},\n });\n\n await previewLayoutUsecase.execute(command);\n\n // Verify that layout control values were used\n expect(createVariablesObjectMock.execute.calledOnce).to.be.true;\n const createVariablesCall = createVariablesObjectMock.execute.firstCall.args[0];\n expect(createVariablesCall.controlValues).to.deep.equal(Object.values(mockLayout.controls.values.email));\n });\n\n it('should use empty object when both command and layout control values are missing', async () => {\n getLayoutUseCaseMock.execute.resolves(mockLayoutWithoutControls as any);\n\n const command = PreviewLayoutCommand.create({\n user: mockUser as any,\n layoutIdOrInternalId: 'layout_id',\n layoutPreviewRequestDto: {},\n });\n\n await previewLayoutUsecase.execute(command);\n\n // Verify empty control values were used\n expect(controlValueSanitizerMock.sanitizeControlsForPreview.calledOnce).to.be.true;\n const sanitizeCall = controlValueSanitizerMock.sanitizeControlsForPreview.firstCall.args[0];\n expect(sanitizeCall).to.deep.equal({});\n });\n\n it('should call all dependencies with correct parameters', async () => {\n const command = PreviewLayoutCommand.create({\n user: mockUser as any,\n layoutIdOrInternalId: 'layout_id',\n layoutPreviewRequestDto: {\n controlValues: mockControlValues,\n previewPayload: { subscriber: { email: 'test@example.com' } },\n },\n });\n\n await previewLayoutUsecase.execute(command);\n\n // Verify getLayoutUseCase call\n expect(getLayoutUseCaseMock.execute.calledOnce).to.be.true;\n const getLayoutCall = getLayoutUseCaseMock.execute.firstCall.args[0];\n expect(getLayoutCall).to.deep.equal({\n layoutIdOrInternalId: 'layout_id',\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n userId: mockUser._id,\n });\n\n // Verify createVariablesObject call\n expect(createVariablesObjectMock.execute.calledOnce).to.be.true;\n const createVariablesCall = createVariablesObjectMock.execute.firstCall.args[0];\n expect(createVariablesCall.environmentId).to.equal(mockUser.environmentId);\n expect(createVariablesCall.organizationId).to.equal(mockUser.organizationId);\n expect(createVariablesCall.variableSchema).to.deep.equal(mockLayout.variables);\n\n // Verify controlValueSanitizer calls\n expect(controlValueSanitizerMock.sanitizeControlsForPreview.calledOnce).to.be.true;\n const sanitizeCall = controlValueSanitizerMock.sanitizeControlsForPreview.firstCall.args;\n expect(sanitizeCall[0]).to.deep.equal(mockControlValues);\n expect(sanitizeCall[1]).to.equal('layout');\n expect(sanitizeCall[2]).to.equal(ResourceOriginEnum.NOVU_CLOUD);\n\n expect(controlValueSanitizerMock.processControlValues.calledOnce).to.be.true;\n const processCall = controlValueSanitizerMock.processControlValues.firstCall.args;\n expect(processCall[0]).to.deep.equal(mockSanitizedControls);\n expect(processCall[1]).to.deep.equal(mockLayout.variables);\n expect(processCall[2]).to.deep.equal(mockVariablesObject);\n\n // Verify payloadMerger call\n expect(payloadMergerMock.mergePayloadExample.calledOnce).to.be.true;\n const mergeCall = payloadMergerMock.mergePayloadExample.firstCall.args[0];\n expect(mergeCall.payloadExample).to.deep.equal(mockPreviewTemplateData.payloadExample);\n expect(mergeCall.userPayloadExample).to.deep.equal(command.layoutPreviewRequestDto.previewPayload);\n expect(mergeCall.user).to.deep.equal(command.user);\n\n // Verify payloadProcessor call\n expect(payloadProcessorMock.cleanPreviewExamplePayload.calledOnceWith(mockPayloadExample)).to.be.true;\n\n // Verify previewStepUsecase call\n expect(previewStepUsecaseMock.execute.calledOnce).to.be.true;\n const previewCall = previewStepUsecaseMock.execute.firstCall.args[0];\n expect(previewCall.payload).to.deep.equal(mockCleanedPayloadExample.payload);\n expect(previewCall.subscriber).to.deep.equal(mockCleanedPayloadExample.subscriber);\n expect(previewCall.controls).to.deep.equal({\n subject: 'email-layout-preview',\n body: enhanceBodyForPreview(\n mockPreviewTemplateData.controlValues.email?.editorType ?? 'block',\n mockPreviewTemplateData.controlValues.email?.body ?? ''\n ),\n editorType: mockPreviewTemplateData.controlValues.email?.editorType,\n });\n expect(previewCall.environmentId).to.equal(mockUser.environmentId);\n expect(previewCall.organizationId).to.equal(mockUser.organizationId);\n expect(previewCall.stepId).to.equal(LAYOUT_PREVIEW_EMAIL_STEP);\n expect(previewCall.userId).to.equal(mockUser._id);\n expect(previewCall.workflowId).to.equal(LAYOUT_PREVIEW_WORKFLOW_ID);\n expect(previewCall.workflowOrigin).to.equal(ResourceOriginEnum.NOVU_CLOUD);\n expect(previewCall.state).to.deep.equal([]);\n });\n\n it('should handle missing previewPayload gracefully', async () => {\n const command = PreviewLayoutCommand.create({\n user: mockUser as any,\n layoutIdOrInternalId: 'layout_id',\n layoutPreviewRequestDto: {\n controlValues: mockControlValues,\n },\n });\n\n await previewLayoutUsecase.execute(command);\n\n const mergeCall = payloadMergerMock.mergePayloadExample.firstCall.args[0];\n expect(mergeCall.userPayloadExample).to.be.undefined;\n });\n\n it('should handle missing variables schema', async () => {\n const layoutWithoutVariables = {\n ...mockLayout,\n variables: undefined,\n };\n getLayoutUseCaseMock.execute.resolves(layoutWithoutVariables as any);\n\n const command = PreviewLayoutCommand.create({\n user: mockUser as any,\n layoutIdOrInternalId: 'layout_id',\n layoutPreviewRequestDto: {\n controlValues: mockControlValues,\n },\n });\n\n await previewLayoutUsecase.execute(command);\n\n const createVariablesCall = createVariablesObjectMock.execute.firstCall.args[0];\n expect(createVariablesCall.variableSchema).to.deep.equal({});\n });\n\n it('should handle missing email controls in preview template data', async () => {\n const templateDataWithoutEmail = {\n ...mockPreviewTemplateData,\n controlValues: {} as LayoutControlType,\n };\n controlValueSanitizerMock.processControlValues.returns({\n previewTemplateData: templateDataWithoutEmail,\n sanitizedControls: mockSanitizedControls,\n });\n\n const command = PreviewLayoutCommand.create({\n user: mockUser as any,\n layoutIdOrInternalId: 'layout_id',\n layoutPreviewRequestDto: {\n controlValues: mockControlValues,\n },\n });\n\n await previewLayoutUsecase.execute(command);\n\n expect(previewStepUsecaseMock.execute.calledOnce).to.be.true;\n const previewCall = previewStepUsecaseMock.execute.firstCall.args[0];\n expect(previewCall.controls.body).to.eq('{}');\n expect(previewCall.controls.editorType).to.eq('block');\n });\n\n it('should handle missing payload in cleaned payload example', async () => {\n const cleanedPayloadWithoutPayload = {\n payload: undefined,\n subscriber: { email: 'test@example.com' },\n };\n payloadProcessorMock.cleanPreviewExamplePayload.returns(cleanedPayloadWithoutPayload);\n\n const command = PreviewLayoutCommand.create({\n user: mockUser as any,\n layoutIdOrInternalId: 'layout_id',\n layoutPreviewRequestDto: {\n controlValues: mockControlValues,\n },\n });\n\n await previewLayoutUsecase.execute(command);\n\n const previewCall = previewStepUsecaseMock.execute.firstCall.args[0];\n expect(previewCall.payload).to.deep.equal({});\n });\n\n it('should handle missing subscriber in cleaned payload example', async () => {\n const cleanedPayloadWithoutSubscriber = {\n payload: { content: 'test' },\n subscriber: undefined,\n };\n payloadProcessorMock.cleanPreviewExamplePayload.returns(cleanedPayloadWithoutSubscriber);\n\n const command = PreviewLayoutCommand.create({\n user: mockUser as any,\n layoutIdOrInternalId: 'layout_id',\n layoutPreviewRequestDto: {\n controlValues: mockControlValues,\n },\n });\n\n await previewLayoutUsecase.execute(command);\n\n const previewCall = previewStepUsecaseMock.execute.firstCall.args[0];\n expect(previewCall.subscriber).to.deep.equal({});\n });\n\n describe('error handling', () => {\n it('should return fallback response when getLayoutUseCase throws error', async () => {\n try {\n const error = new Error('Layout not found');\n getLayoutUseCaseMock.execute.rejects(error);\n\n const command = PreviewLayoutCommand.create({\n user: mockUser as any,\n layoutIdOrInternalId: 'invalid_layout_id',\n layoutPreviewRequestDto: {\n controlValues: mockControlValues,\n },\n });\n\n await previewLayoutUsecase.execute(command);\n } catch (error) {\n expect(error.message).to.equal('Layout not found');\n }\n });\n\n it('should return fallback response when createVariablesObject throws error', async () => {\n const error = new Error('Variables creation failed');\n createVariablesObjectMock.execute.rejects(error);\n\n const command = PreviewLayoutCommand.create({\n user: mockUser as any,\n layoutIdOrInternalId: 'layout_id',\n layoutPreviewRequestDto: {\n controlValues: mockControlValues,\n },\n });\n\n const result = await previewLayoutUsecase.execute(command);\n\n expect(result).to.deep.equal({\n result: {\n type: ChannelTypeEnum.EMAIL,\n },\n previewPayloadExample: {},\n schema: null,\n });\n });\n\n it('should return fallback response when controlValueSanitizer throws error', async () => {\n const error = new Error('Control value sanitization failed');\n controlValueSanitizerMock.sanitizeControlsForPreview.throws(error);\n\n const command = PreviewLayoutCommand.create({\n user: mockUser as any,\n layoutIdOrInternalId: 'layout_id',\n layoutPreviewRequestDto: {\n controlValues: mockControlValues,\n },\n });\n\n const result = await previewLayoutUsecase.execute(command);\n\n expect(result).to.deep.equal({\n result: {\n type: ChannelTypeEnum.EMAIL,\n },\n previewPayloadExample: {},\n schema: null,\n });\n });\n\n it('should return fallback response when payloadMerger throws error', async () => {\n const error = new Error('Payload merge failed');\n payloadMergerMock.mergePayloadExample.rejects(error);\n\n const command = PreviewLayoutCommand.create({\n user: mockUser as any,\n layoutIdOrInternalId: 'layout_id',\n layoutPreviewRequestDto: {\n controlValues: mockControlValues,\n },\n });\n\n const result = await previewLayoutUsecase.execute(command);\n\n expect(result).to.deep.equal({\n result: {\n type: ChannelTypeEnum.EMAIL,\n },\n previewPayloadExample: {},\n schema: null,\n });\n });\n\n it('should return fallback response when payloadProcessor throws error', async () => {\n const error = new Error('Payload processing failed');\n payloadProcessorMock.cleanPreviewExamplePayload.throws(error);\n\n const command = PreviewLayoutCommand.create({\n user: mockUser as any,\n layoutIdOrInternalId: 'layout_id',\n layoutPreviewRequestDto: {\n controlValues: mockControlValues,\n },\n });\n\n const result = await previewLayoutUsecase.execute(command);\n\n expect(result).to.deep.equal({\n result: {\n type: ChannelTypeEnum.EMAIL,\n },\n previewPayloadExample: {},\n schema: null,\n });\n });\n\n it('should return fallback response when previewStepUsecase throws error', async () => {\n const error = new Error('Preview step execution failed');\n previewStepUsecaseMock.execute.rejects(error);\n\n const command = PreviewLayoutCommand.create({\n user: mockUser as any,\n layoutIdOrInternalId: 'layout_id',\n layoutPreviewRequestDto: {\n controlValues: mockControlValues,\n },\n });\n\n const result = await previewLayoutUsecase.execute(command);\n\n expect(result).to.deep.equal({\n result: {\n type: ChannelTypeEnum.EMAIL,\n },\n previewPayloadExample: {},\n schema: null,\n });\n });\n\n it('should not call subsequent dependencies when early dependency fails', async () => {\n const error = new Error('Early failure');\n createVariablesObjectMock.execute.rejects(error);\n\n const command = PreviewLayoutCommand.create({\n user: mockUser as any,\n layoutIdOrInternalId: 'layout_id',\n layoutPreviewRequestDto: {\n controlValues: mockControlValues,\n },\n });\n\n await previewLayoutUsecase.execute(command);\n\n // Verify that dependencies after createVariablesObject were not called\n expect(controlValueSanitizerMock.sanitizeControlsForPreview.called).to.be.false;\n expect(payloadMergerMock.mergePayloadExample.called).to.be.false;\n expect(previewStepUsecaseMock.execute.called).to.be.false;\n });\n });\n\n describe('edge cases', () => {\n it('should handle empty previewStepOutput', async () => {\n previewStepUsecaseMock.execute.resolves({ outputs: {} } as any);\n\n const command = PreviewLayoutCommand.create({\n user: mockUser as any,\n layoutIdOrInternalId: 'layout_id',\n layoutPreviewRequestDto: {\n controlValues: mockControlValues,\n },\n });\n\n const result = await previewLayoutUsecase.execute(command);\n\n expect(result.result.preview?.body).to.be.undefined;\n expect(result.result.type).to.equal(ChannelTypeEnum.EMAIL);\n });\n\n it('should handle null previewStepOutput outputs', async () => {\n previewStepUsecaseMock.execute.resolves({ outputs: null } as any);\n\n const command = PreviewLayoutCommand.create({\n user: mockUser as any,\n layoutIdOrInternalId: 'layout_id',\n layoutPreviewRequestDto: {\n controlValues: mockControlValues,\n },\n });\n\n const result = await previewLayoutUsecase.execute(command);\n\n expect(result.result.preview?.body).to.be.undefined;\n expect(result.result.type).to.equal(ChannelTypeEnum.EMAIL);\n });\n });\n });\n});\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/preview-layout/preview-layout.usecase.ts",
"content": "import { Injectable } from '@nestjs/common';\nimport {\n buildContextSchema,\n buildSubscriberSchema,\n ControlValueSanitizerService,\n CreateVariablesObject,\n CreateVariablesObjectCommand,\n EmailControlType,\n GetLayoutCommand,\n GetLayoutUseCase,\n InstrumentUsecase,\n LayoutControlType,\n PayloadMergerService,\n PlatformException,\n PreviewPayloadProcessorService,\n PreviewStep,\n PreviewStepCommand,\n resolveEnvironmentVariables,\n} from '@novu/application-generic';\nimport { EnvironmentRepository, EnvironmentVariableRepository, JsonSchemaTypeEnum } from '@novu/dal';\nimport { ContextResolved } from '@novu/framework/internal';\nimport {\n ChannelTypeEnum,\n EnvironmentSystemVariables,\n LAYOUT_PREVIEW_EMAIL_STEP,\n LAYOUT_PREVIEW_WORKFLOW_ID,\n ResourceOriginEnum,\n} from '@novu/shared';\nimport { GenerateLayoutPreviewResponseDto } from '../../dtos/generate-layout-preview-response.dto';\nimport { PreviewLayoutCommand } from './preview-layout.command';\nimport { enhanceBodyForPreview } from './preview-utils';\n\n@Injectable()\nexport class PreviewLayoutUsecase {\n constructor(\n private getLayoutUseCase: GetLayoutUseCase,\n private createVariablesObject: CreateVariablesObject,\n private controlValueSanitizer: ControlValueSanitizerService,\n private payloadProcessor: PreviewPayloadProcessorService,\n private payloadMerger: PayloadMergerService,\n private previewStepUsecase: PreviewStep,\n private readonly environmentVariableRepository: EnvironmentVariableRepository,\n private readonly environmentRepository: EnvironmentRepository\n ) {}\n\n @InstrumentUsecase()\n async execute(command: PreviewLayoutCommand): Promise {\n const layout = await this.getLayoutUseCase.execute(\n GetLayoutCommand.create({\n layoutIdOrInternalId: command.layoutIdOrInternalId,\n environmentId: command.user.environmentId,\n organizationId: command.user.organizationId,\n userId: command.user._id,\n })\n );\n\n try {\n const controlValues = command.layoutPreviewRequestDto.controlValues || layout.controls.values || {};\n const variableSchema = layout.variables ?? {};\n\n // extract all variables from the control values and build the variables object\n const variablesObject = await this.createVariablesObject.execute(\n CreateVariablesObjectCommand.create({\n environmentId: command.user.environmentId,\n organizationId: command.user.organizationId,\n controlValues: Object.values(controlValues.email ?? {}),\n variableSchema,\n })\n );\n\n const sanitizedControls = this.controlValueSanitizer.sanitizeControlsForPreview(\n controlValues as Record,\n 'layout',\n ResourceOriginEnum.NOVU_CLOUD\n );\n\n const { previewTemplateData } = this.controlValueSanitizer.processControlValues(\n sanitizedControls,\n variableSchema,\n variablesObject\n );\n\n const payloadExample = await this.payloadMerger.mergePayloadExample({\n payloadExample: previewTemplateData.payloadExample,\n userPayloadExample: command.layoutPreviewRequestDto.previewPayload,\n user: command.user,\n });\n\n const cleanedPayloadExample = this.payloadProcessor.cleanPreviewExamplePayload(payloadExample);\n\n const { email } = previewTemplateData.controlValues as LayoutControlType;\n const editorType = email?.editorType ?? 'block';\n const body = email?.body ?? (editorType === 'block' ? '{}' : '');\n\n const [rawEnvVars, environmentEntity] = await Promise.all([\n this.environmentVariableRepository.findByEnvironment(command.user.organizationId, command.user.environmentId),\n this.environmentRepository.findByIdAndOrganization(command.user.environmentId, command.user.organizationId),\n ]);\n\n if (!environmentEntity) throw new PlatformException('EnvironmentEntity not found');\n\n const environmentSystemVars: EnvironmentSystemVariables = {\n name: environmentEntity.name,\n type: environmentEntity.type,\n };\n\n const envVars = {\n ...resolveEnvironmentVariables(rawEnvVars),\n ...environmentSystemVars,\n };\n\n const executeOutput = await this.previewStepUsecase.execute(\n PreviewStepCommand.create({\n payload: (cleanedPayloadExample.payload ?? {}) as Record,\n subscriber: cleanedPayloadExample.subscriber ?? {},\n context: (cleanedPayloadExample.context ?? {}) as ContextResolved,\n // mapping the email layout controls to the email step controls\n controls: {\n subject: 'email-layout-preview',\n body: enhanceBodyForPreview(editorType, body),\n editorType,\n } as EmailControlType,\n environmentId: command.user.environmentId,\n organizationId: command.user.organizationId,\n stepId: LAYOUT_PREVIEW_EMAIL_STEP,\n userId: command.user._id,\n workflowId: LAYOUT_PREVIEW_WORKFLOW_ID,\n workflowOrigin: ResourceOriginEnum.NOVU_CLOUD,\n layoutId: layout.layoutId,\n state: [],\n env: envVars,\n })\n );\n\n const { body: previewBody } = executeOutput.outputs as any;\n\n // Generate schema from the preview payload example\n const schema = {\n type: JsonSchemaTypeEnum.OBJECT,\n properties: {\n subscriber: buildSubscriberSchema(payloadExample.subscriber),\n context: buildContextSchema(payloadExample.context),\n },\n };\n\n return {\n result: {\n preview: { body: previewBody },\n type: ChannelTypeEnum.EMAIL,\n },\n previewPayloadExample: payloadExample,\n schema,\n };\n } catch (error) {\n /*\n * If preview execution fails, still return valid schema and payload example\n * but with an empty preview result\n */\n return {\n result: {\n type: ChannelTypeEnum.EMAIL,\n },\n previewPayloadExample: {},\n schema: null,\n };\n }\n }\n}\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/preview-layout/preview-utils.ts",
"content": "import { replaceMailyNodesByCondition } from '@novu/application-generic';\nimport { JSONContent as MailyJSONContent } from '@novu/maily-render';\nimport { LAYOUT_CONTENT_VARIABLE, LAYOUT_PREVIEW_CONTENT_PLACEHOLDER } from '@novu/shared';\n\nexport const enhanceBodyForPreview = (editorType: string, body: string) => {\n if (editorType === 'html') {\n return body?.replace(\n new RegExp(`\\\\{\\\\{\\\\s*${LAYOUT_CONTENT_VARIABLE}\\\\s*\\\\}\\\\}`),\n LAYOUT_PREVIEW_CONTENT_PLACEHOLDER\n );\n }\n\n return JSON.stringify(\n replaceMailyNodesByCondition(\n body,\n (node) => node.type === 'variable' && node.attrs?.id === LAYOUT_CONTENT_VARIABLE,\n (node) => {\n return {\n type: 'text',\n text: LAYOUT_PREVIEW_CONTENT_PLACEHOLDER,\n attrs: {\n ...node.attrs,\n shouldDangerouslySetInnerHTML: true,\n },\n } satisfies MailyJSONContent;\n }\n )\n );\n};\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/sync-to-environment/index.ts",
"content": "export * from './layout-sync-to-environment.command';\nexport * from './layout-sync-to-environment.usecase';\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/sync-to-environment/layout-sync-to-environment.command.ts",
"content": "import { EnvironmentWithUserObjectCommand } from '@novu/application-generic';\nimport { ClientSession } from '@novu/dal';\nimport { Exclude } from 'class-transformer';\nimport { IsDefined, IsOptional, IsString } from 'class-validator';\n\nexport class LayoutSyncToEnvironmentCommand extends EnvironmentWithUserObjectCommand {\n @IsString()\n @IsDefined()\n layoutIdOrInternalId: string;\n\n @IsString()\n @IsDefined()\n targetEnvironmentId: string;\n\n /**\n * Exclude session from the command to avoid serializing it in the response\n */\n @IsOptional()\n @Exclude()\n session?: ClientSession | null;\n}\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/sync-to-environment/layout-sync-to-environment.usecase.ts",
"content": "import { BadRequestException, Injectable } from '@nestjs/common';\nimport { ModuleRef } from '@nestjs/core';\nimport {\n GetLayoutCommand,\n GetLayoutUseCase,\n Instrument,\n InstrumentUsecase,\n LayoutResponseDto,\n} from '@novu/application-generic';\nimport { LocalizationResourceEnum } from '@novu/dal';\nimport { ResourceOriginEnum } from '@novu/shared';\nimport { UpsertLayout, UpsertLayoutCommand, UpsertLayoutDataCommand } from '../upsert-layout';\nimport { LayoutSyncToEnvironmentCommand } from './layout-sync-to-environment.command';\n\nconst SYNCABLE_LAYOUT_ORIGINS = [ResourceOriginEnum.NOVU_CLOUD];\n\nclass LayoutNotSyncableException extends BadRequestException {\n constructor(layout: Pick) {\n const reason = `origin '${layout.origin}' is not allowed (must be one of: ${SYNCABLE_LAYOUT_ORIGINS.join(', ')})`;\n\n super({\n message: `Cannot sync layout: ${reason}`,\n layoutId: layout.layoutId,\n origin: layout.origin,\n allowedOrigins: SYNCABLE_LAYOUT_ORIGINS,\n });\n }\n}\n\n@Injectable()\nexport class LayoutSyncToEnvironmentUseCase {\n constructor(\n private getLayoutUseCase: GetLayoutUseCase,\n private upsertLayoutUseCase: UpsertLayout,\n private moduleRef: ModuleRef\n ) {}\n\n @InstrumentUsecase()\n async execute(command: LayoutSyncToEnvironmentCommand): Promise {\n if (command.user.environmentId === command.targetEnvironmentId) {\n throw new BadRequestException('Cannot sync layout to the same environment');\n }\n\n const sourceLayout = await this.getLayoutUseCase.execute(\n GetLayoutCommand.create({\n environmentId: command.user.environmentId,\n organizationId: command.user.organizationId,\n layoutIdOrInternalId: command.layoutIdOrInternalId,\n })\n );\n\n if (!this.isSyncable(sourceLayout)) {\n throw new LayoutNotSyncableException(sourceLayout);\n }\n\n const externalId = sourceLayout.layoutId;\n const targetLayout = await this.findLayoutInTargetEnvironment(command, externalId);\n\n const layoutDto = await this.buildRequestDto(sourceLayout);\n\n const upsertedLayout = await this.upsertLayoutUseCase.execute(\n UpsertLayoutCommand.create({\n environmentId: command.targetEnvironmentId,\n organizationId: command.user.organizationId,\n userId: command.user._id,\n layoutIdOrInternalId: targetLayout?.layoutId,\n layoutDto,\n })\n );\n\n await this.publishTranslationGroup(sourceLayout.layoutId, LocalizationResourceEnum.LAYOUT, command);\n\n return upsertedLayout;\n }\n\n private isSyncable(layout: LayoutResponseDto): boolean {\n return SYNCABLE_LAYOUT_ORIGINS.includes(layout.origin);\n }\n\n private async buildRequestDto(sourceLayout: LayoutResponseDto): Promise {\n return {\n layoutId: sourceLayout.layoutId,\n name: sourceLayout.name,\n isTranslationEnabled: sourceLayout.isTranslationEnabled,\n controlValues: sourceLayout.controls?.values,\n };\n }\n\n private async publishTranslationGroup(\n resourceId: string,\n resourceType: LocalizationResourceEnum,\n command: LayoutSyncToEnvironmentCommand\n ): Promise {\n const isEnterprise = process.env.NOVU_ENTERPRISE === 'true' || process.env.CI_EE_TEST === 'true';\n const isSelfHosted = process.env.IS_SELF_HOSTED === 'true';\n\n if (!isEnterprise || isSelfHosted) {\n return;\n }\n\n const publishTranslationGroup = this.moduleRef.get(require('@novu/ee-translation')?.PublishTranslationGroup, {\n strict: false,\n });\n\n const { user, targetEnvironmentId } = command;\n\n await publishTranslationGroup.execute({\n user,\n resourceId,\n resourceType,\n sourceEnvironmentId: user.environmentId,\n targetEnvironmentId,\n });\n }\n\n @Instrument()\n private async findLayoutInTargetEnvironment(\n command: LayoutSyncToEnvironmentCommand,\n externalId: string\n ): Promise {\n try {\n return await this.getLayoutUseCase.execute(\n GetLayoutCommand.create({\n environmentId: command.targetEnvironmentId,\n organizationId: command.user.organizationId,\n layoutIdOrInternalId: externalId,\n })\n );\n } catch (error) {\n return undefined;\n }\n }\n}\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/upsert-layout/index.ts",
"content": "export * from './upsert-layout.command';\nexport * from './upsert-layout.usecase';\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/upsert-layout/upsert-layout.command.ts",
"content": "import {\n EnvironmentWithUserCommand,\n LayoutControlValuesDto,\n LayoutCreationSourceEnum,\n} from '@novu/application-generic';\nimport { MAX_NAME_LENGTH } from '@novu/shared';\nimport { Type } from 'class-transformer';\nimport { IsBoolean, IsEnum, IsNotEmpty, IsOptional, IsString, Length, ValidateNested } from 'class-validator';\n\nexport class UpsertLayoutDataCommand {\n @IsString()\n @IsOptional()\n layoutId?: string;\n\n @IsString()\n @IsNotEmpty()\n @Length(1, MAX_NAME_LENGTH)\n name: string;\n\n @IsOptional()\n @IsBoolean()\n isTranslationEnabled?: boolean;\n\n @IsOptional()\n @IsEnum(LayoutCreationSourceEnum)\n __source?: LayoutCreationSourceEnum;\n\n @IsOptional()\n controlValues?: LayoutControlValuesDto | null;\n}\n\nexport class UpsertLayoutCommand extends EnvironmentWithUserCommand {\n @ValidateNested()\n @Type(() => UpsertLayoutDataCommand)\n layoutDto: UpsertLayoutDataCommand;\n\n @IsOptional()\n @IsString()\n layoutIdOrInternalId?: string;\n}\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/upsert-layout/upsert-layout.usecase.spec.ts",
"content": "import { BadRequestException } from '@nestjs/common';\nimport { ModuleRef } from '@nestjs/core';\nimport {\n AnalyticsService,\n GetLayoutUseCase,\n GetLayoutUseCase as GetLayoutUseCaseV0,\n JSONSchemaDto,\n LayoutCreationSourceEnum,\n LayoutDtoV0,\n layoutControlSchema,\n mapLayoutToResponseDto,\n PinoLogger,\n UpsertControlValuesUseCase,\n} from '@novu/application-generic';\nimport { ControlValuesRepository, JsonSchemaTypeEnum, LayoutRepository } from '@novu/dal';\nimport {\n ChannelTypeEnum,\n ContentIssueEnum,\n ControlValuesLevelEnum,\n LayoutControlValuesDto,\n LayoutIssuesDto,\n ResourceOriginEnum,\n ResourceTypeEnum,\n slugify,\n} from '@novu/shared';\nimport { expect } from 'chai';\nimport sinon from 'sinon';\nimport { CreateLayoutUseCase, UpdateLayoutUseCase } from '../../../layouts-v1/usecases';\nimport { BuildLayoutIssuesUsecase } from '../build-layout-issues/build-layout-issues.usecase';\nimport { UpsertLayoutCommand } from './upsert-layout.command';\nimport { UpsertLayout } from './upsert-layout.usecase';\n\n// Mock the utility functions\nconst isStringifiedMailyJSONContentStub = sinon.stub();\n\n// Mock modules using require to ensure proper stubbing\nsinon\n .stub(require('@novu/application-generic'), 'isStringifiedMailyJSONContent')\n .callsFake(isStringifiedMailyJSONContentStub);\n\nfunction setupTranslationMocks(moduleRef: sinon.SinonStubbedInstance): sinon.SinonStub {\n const manageTranslationsExecuteStub = sinon.stub().resolves();\n\n (moduleRef as any).get = sinon.stub().returns({\n execute: manageTranslationsExecuteStub,\n });\n\n return manageTranslationsExecuteStub;\n}\n\ndescribe('UpsertLayoutUseCase', () => {\n let getLayoutUseV0CaseMock: sinon.SinonStubbedInstance;\n let createLayoutUseCaseMock: sinon.SinonStubbedInstance;\n let updateLayoutUseCaseMock: sinon.SinonStubbedInstance;\n let controlValuesRepositoryMock: sinon.SinonStubbedInstance;\n let upsertControlValuesUseCaseMock: sinon.SinonStubbedInstance;\n let layoutRepositoryMock: sinon.SinonStubbedInstance;\n let analyticsServiceMock: sinon.SinonStubbedInstance;\n let buildLayoutIssuesUsecaseMock: sinon.SinonStubbedInstance;\n let getLayoutUseCaseMock: sinon.SinonStubbedInstance;\n let moduleRefMock: sinon.SinonStubbedInstance;\n let pinoLoggerMock: sinon.SinonStubbedInstance;\n\n let upsertLayoutUseCase: UpsertLayout;\n\n const mockUser = {\n _id: 'user_id',\n environmentId: 'env_id',\n organizationId: 'org_id',\n };\n\n const mockLayoutDto = {\n name: 'Test Layout',\n __source: LayoutCreationSourceEnum.DASHBOARD,\n controlValues: {\n email: {\n body: '{{content}}',\n editorType: 'html' as 'html' | 'block',\n },\n } as LayoutControlValuesDto,\n };\n\n const mockExistingLayout: LayoutDtoV0 & { _id: string } = {\n _id: 'existing_layout_id',\n identifier: 'existing_layout_identifier',\n name: 'Existing Layout',\n _creatorId: 'creator_id',\n isDefault: false,\n isDeleted: false,\n createdAt: '2023-01-01T00:00:00Z',\n updatedAt: '2023-01-01T00:00:00Z',\n _environmentId: 'env_id',\n _organizationId: 'org_id',\n origin: ResourceOriginEnum.NOVU_CLOUD,\n type: ResourceTypeEnum.BRIDGE,\n channel: ChannelTypeEnum.EMAIL,\n controls: {\n dataSchema: layoutControlSchema,\n uiSchema: {},\n },\n };\n\n const mockCreatedLayout: LayoutDtoV0 & { _id: string } = {\n _id: 'new_layout_id',\n identifier: 'test-layout',\n name: 'Test Layout',\n _creatorId: 'creator_id',\n isDefault: true,\n isDeleted: false,\n createdAt: '2023-01-01T00:00:00Z',\n updatedAt: '2023-01-01T00:00:00Z',\n _environmentId: 'env_id',\n _organizationId: 'org_id',\n origin: ResourceOriginEnum.NOVU_CLOUD,\n type: ResourceTypeEnum.BRIDGE,\n channel: ChannelTypeEnum.EMAIL,\n controls: {\n dataSchema: layoutControlSchema,\n uiSchema: {},\n },\n };\n\n const mockControlValues = {\n _id: 'control_values_id',\n controls: {\n email: {\n body: '{{content}}',\n editorType: 'html',\n },\n },\n };\n\n const mockLayoutVariablesSchema: JSONSchemaDto = {\n type: JsonSchemaTypeEnum.OBJECT,\n properties: {\n subscriber: {\n type: JsonSchemaTypeEnum.OBJECT,\n properties: {\n email: { type: JsonSchemaTypeEnum.STRING },\n firstName: { type: JsonSchemaTypeEnum.STRING },\n },\n },\n content: { type: JsonSchemaTypeEnum.STRING },\n },\n };\n\n beforeEach(() => {\n getLayoutUseV0CaseMock = sinon.createStubInstance(GetLayoutUseCaseV0);\n createLayoutUseCaseMock = sinon.createStubInstance(CreateLayoutUseCase);\n updateLayoutUseCaseMock = sinon.createStubInstance(UpdateLayoutUseCase);\n controlValuesRepositoryMock = sinon.createStubInstance(ControlValuesRepository);\n upsertControlValuesUseCaseMock = sinon.createStubInstance(UpsertControlValuesUseCase);\n layoutRepositoryMock = sinon.createStubInstance(LayoutRepository);\n analyticsServiceMock = sinon.createStubInstance(AnalyticsService);\n buildLayoutIssuesUsecaseMock = sinon.createStubInstance(BuildLayoutIssuesUsecase);\n getLayoutUseCaseMock = sinon.createStubInstance(GetLayoutUseCase);\n moduleRefMock = sinon.createStubInstance(ModuleRef);\n pinoLoggerMock = sinon.createStubInstance(PinoLogger);\n\n setupTranslationMocks(moduleRefMock as any);\n\n upsertLayoutUseCase = new UpsertLayout(\n getLayoutUseV0CaseMock as any,\n createLayoutUseCaseMock as any,\n updateLayoutUseCaseMock as any,\n controlValuesRepositoryMock as any,\n upsertControlValuesUseCaseMock as any,\n layoutRepositoryMock as any,\n analyticsServiceMock as any,\n buildLayoutIssuesUsecaseMock as any,\n getLayoutUseCaseMock as any,\n moduleRefMock as any,\n pinoLoggerMock as any\n );\n\n // Default mocks setup\n isStringifiedMailyJSONContentStub.returns(false);\n buildLayoutIssuesUsecaseMock.execute.resolves({} as LayoutIssuesDto);\n upsertControlValuesUseCaseMock.execute.resolves(mockControlValues as any);\n layoutRepositoryMock.findOne.resolves(undefined);\n });\n\n afterEach(() => {\n sinon.restore();\n isStringifiedMailyJSONContentStub.reset();\n });\n\n describe('execute', () => {\n describe('create layout path', () => {\n beforeEach(() => {\n getLayoutUseV0CaseMock.execute.resolves(undefined);\n createLayoutUseCaseMock.execute.resolves(mockCreatedLayout);\n getLayoutUseCaseMock.execute.resolves(\n mapLayoutToResponseDto({\n layout: mockCreatedLayout,\n controlValues: mockControlValues,\n variables: mockLayoutVariablesSchema,\n })\n );\n });\n\n it('should successfully create a new layout when no existing layout found', async () => {\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mockLayoutDto,\n });\n\n const result = await upsertLayoutUseCase.execute(command);\n\n expect(result).to.exist;\n expect(result._id).to.equal(mockCreatedLayout._id);\n expect(result.name).to.equal(mockCreatedLayout.name);\n expect(createLayoutUseCaseMock.execute.calledOnce).to.be.true;\n expect(updateLayoutUseCaseMock.execute.called).to.be.false;\n });\n\n it('should call createLayoutUseCase with correct parameters', async () => {\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mockLayoutDto,\n });\n\n await upsertLayoutUseCase.execute(command);\n\n expect(createLayoutUseCaseMock.execute.calledOnce).to.be.true;\n const createCommand = createLayoutUseCaseMock.execute.firstCall.args[0];\n expect(createCommand.environmentId).to.equal(mockUser.environmentId);\n expect(createCommand.organizationId).to.equal(mockUser.organizationId);\n expect(createCommand.userId).to.equal(mockUser._id);\n expect(createCommand.name).to.equal(mockLayoutDto.name);\n expect(createCommand.identifier).to.equal(slugify(mockLayoutDto.name));\n expect(createCommand.type).to.equal(ResourceTypeEnum.BRIDGE);\n expect(createCommand.origin).to.equal(ResourceOriginEnum.NOVU_CLOUD);\n expect(createCommand.isDefault).to.be.true;\n });\n\n it('should use custom layoutId when provided instead of slugified name', async () => {\n const customLayoutId = 'custom-layout-identifier';\n const layoutDtoWithCustomId = {\n ...mockLayoutDto,\n layoutId: customLayoutId,\n };\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: layoutDtoWithCustomId,\n });\n\n await upsertLayoutUseCase.execute(command);\n\n expect(createLayoutUseCaseMock.execute.calledOnce).to.be.true;\n const createCommand = createLayoutUseCaseMock.execute.firstCall.args[0];\n expect(createCommand.identifier).to.equal(customLayoutId);\n expect(createCommand.name).to.equal(mockLayoutDto.name);\n });\n\n it('should set isDefault to false when a default layout already exists', async () => {\n const existingDefaultLayout = { ...mockExistingLayout, isDefault: true };\n layoutRepositoryMock.findOne.resolves(existingDefaultLayout as any);\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mockLayoutDto,\n });\n\n await upsertLayoutUseCase.execute(command);\n\n const createCommand = createLayoutUseCaseMock.execute.firstCall.args[0];\n expect(createCommand.isDefault).to.be.false;\n });\n\n it('should track \"Layout Create\" analytics event', async () => {\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mockLayoutDto,\n });\n\n await upsertLayoutUseCase.execute(command);\n\n expect(analyticsServiceMock.mixpanelTrack.calledOnce).to.be.true;\n const [eventName, userId, props] = analyticsServiceMock.mixpanelTrack.firstCall.args;\n expect(eventName).to.equal('Layout Create - [Layouts]');\n expect(userId).to.equal(mockUser._id);\n expect(props).to.deep.equal({\n _organization: mockUser.organizationId,\n name: mockLayoutDto.name,\n source: mockLayoutDto.__source,\n });\n });\n });\n\n describe('update layout path', () => {\n beforeEach(() => {\n getLayoutUseV0CaseMock.execute.resolves(mockExistingLayout);\n updateLayoutUseCaseMock.execute.resolves(mockExistingLayout);\n getLayoutUseCaseMock.execute.resolves(\n mapLayoutToResponseDto({\n layout: mockExistingLayout,\n controlValues: mockControlValues,\n variables: mockLayoutVariablesSchema,\n })\n );\n });\n\n it('should successfully update an existing layout when layoutIdOrInternalId provided', async () => {\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mockLayoutDto,\n layoutIdOrInternalId: 'existing_layout_id',\n });\n\n const result = await upsertLayoutUseCase.execute(command);\n\n expect(result).to.exist;\n expect(result._id).to.equal(mockExistingLayout._id);\n expect(updateLayoutUseCaseMock.execute.calledOnce).to.be.true;\n expect(createLayoutUseCaseMock.execute.called).to.be.false;\n });\n\n it('should call getLayoutUseCase with correct parameters', async () => {\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mockLayoutDto,\n layoutIdOrInternalId: 'existing_layout_id',\n });\n\n await upsertLayoutUseCase.execute(command);\n\n expect(getLayoutUseV0CaseMock.execute.calledOnce).to.be.true;\n const getCommand = getLayoutUseV0CaseMock.execute.firstCall.args[0];\n expect(getCommand.layoutIdOrInternalId).to.equal('existing_layout_id');\n expect(getCommand.environmentId).to.equal(mockUser.environmentId);\n expect(getCommand.organizationId).to.equal(mockUser.organizationId);\n expect(getCommand.type).to.equal(ResourceTypeEnum.BRIDGE);\n expect(getCommand.origin).to.equal(ResourceOriginEnum.NOVU_CLOUD);\n });\n\n it('should call updateLayoutUseCase with correct parameters', async () => {\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mockLayoutDto,\n layoutIdOrInternalId: 'existing_layout_id',\n });\n\n await upsertLayoutUseCase.execute(command);\n\n expect(updateLayoutUseCaseMock.execute.calledOnce).to.be.true;\n const updateCommand = updateLayoutUseCaseMock.execute.firstCall.args[0];\n expect(updateCommand.environmentId).to.equal(mockUser.environmentId);\n expect(updateCommand.organizationId).to.equal(mockUser.organizationId);\n expect(updateCommand.userId).to.equal(mockUser._id);\n expect(updateCommand.layoutId).to.equal(mockExistingLayout._id);\n expect(updateCommand.name).to.equal(mockLayoutDto.name);\n expect(updateCommand.type).to.equal(mockExistingLayout.type);\n expect(updateCommand.origin).to.equal(mockExistingLayout.origin);\n });\n\n it('should track \"Layout Update\" analytics event', async () => {\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mockLayoutDto,\n layoutIdOrInternalId: 'existing_layout_id',\n });\n\n await upsertLayoutUseCase.execute(command);\n\n expect(analyticsServiceMock.mixpanelTrack.calledOnce).to.be.true;\n const [eventName, userId, props] = analyticsServiceMock.mixpanelTrack.firstCall.args;\n expect(eventName).to.equal('Layout Update - [Layouts]');\n expect(userId).to.equal(mockUser._id);\n expect(props).to.deep.equal({\n _organization: mockUser.organizationId,\n name: mockLayoutDto.name,\n source: mockLayoutDto.__source,\n });\n });\n });\n\n describe('control values handling', () => {\n beforeEach(() => {\n getLayoutUseV0CaseMock.execute.resolves(undefined);\n createLayoutUseCaseMock.execute.resolves(mockCreatedLayout);\n getLayoutUseCaseMock.execute.resolves(\n mapLayoutToResponseDto({\n layout: mockCreatedLayout,\n controlValues: mockControlValues,\n variables: mockLayoutVariablesSchema,\n })\n );\n });\n\n it('should upsert control values when provided', async () => {\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mockLayoutDto,\n });\n\n await upsertLayoutUseCase.execute(command);\n\n expect(upsertControlValuesUseCaseMock.execute.calledOnce).to.be.true;\n const upsertCommand = upsertControlValuesUseCaseMock.execute.firstCall.args[0];\n expect(upsertCommand.organizationId).to.equal(mockUser.organizationId);\n expect(upsertCommand.environmentId).to.equal(mockUser.environmentId);\n expect(upsertCommand.layoutId).to.equal(mockCreatedLayout._id);\n expect(upsertCommand.level).to.equal(ControlValuesLevelEnum.LAYOUT_CONTROLS);\n expect(upsertCommand.newControlValues).to.deep.equal(mockLayoutDto.controlValues);\n });\n\n it('should delete control values when set to null', async () => {\n const layoutDtoWithNullControls = {\n ...mockLayoutDto,\n controlValues: null,\n };\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: layoutDtoWithNullControls,\n });\n\n await upsertLayoutUseCase.execute(command);\n\n expect(controlValuesRepositoryMock.delete.calledOnce).to.be.true;\n const deleteParams = controlValuesRepositoryMock.delete.firstCall.args[0];\n expect(deleteParams._environmentId).to.equal(mockUser.environmentId);\n expect(deleteParams._organizationId).to.equal(mockUser.organizationId);\n expect(deleteParams._layoutId).to.equal(mockCreatedLayout._id);\n expect(deleteParams.level).to.equal(ControlValuesLevelEnum.LAYOUT_CONTROLS);\n });\n\n it('should handle empty control values', async () => {\n const layoutDtoWithEmptyControls = {\n ...mockLayoutDto,\n controlValues: {},\n };\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: layoutDtoWithEmptyControls,\n });\n\n await upsertLayoutUseCase.execute(command);\n\n expect(upsertControlValuesUseCaseMock.execute.calledOnce).to.be.true;\n const upsertCommand = upsertControlValuesUseCaseMock.execute.firstCall.args[0];\n expect(upsertCommand.newControlValues).to.deep.equal({});\n });\n });\n });\n\n describe('validation', () => {\n describe('email content validation', () => {\n beforeEach(() => {\n getLayoutUseV0CaseMock.execute.resolves(undefined);\n createLayoutUseCaseMock.execute.resolves(mockCreatedLayout);\n getLayoutUseCaseMock.execute.resolves(\n mapLayoutToResponseDto({\n layout: mockCreatedLayout,\n controlValues: mockControlValues,\n variables: mockLayoutVariablesSchema,\n })\n );\n });\n\n it('should validate HTML content correctly', async () => {\n const htmlLayoutDto = {\n ...mockLayoutDto,\n controlValues: {\n email: {\n body: 'Valid HTML',\n editorType: 'html' as 'html' | 'block',\n },\n },\n };\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: htmlLayoutDto,\n });\n\n await upsertLayoutUseCase.execute(command);\n\n expect(buildLayoutIssuesUsecaseMock.execute.calledOnce).to.be.true;\n });\n\n it('should throw BadRequestException for invalid HTML content with html editor type', async () => {\n const invalidHtmlLayoutDto = {\n ...mockLayoutDto,\n controlValues: {\n email: {\n body: 'Invalid HTML content',\n editorType: 'html' as 'html' | 'block',\n },\n },\n };\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: invalidHtmlLayoutDto,\n });\n\n try {\n await upsertLayoutUseCase.execute(command);\n expect.fail('Should have thrown BadRequestException');\n } catch (error) {\n expect(error).to.be.instanceOf(BadRequestException);\n expect(error.message).to.equal('Content must be a valid HTML content');\n }\n });\n\n it('should validate Maily JSON content correctly', async () => {\n isStringifiedMailyJSONContentStub.returns(true);\n\n const mailyLayoutDto = {\n ...mockLayoutDto,\n controlValues: {\n email: {\n body: '{\"type\":\"doc\",\"content\":[]}',\n editorType: 'block' as 'html' | 'block',\n },\n },\n };\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mailyLayoutDto,\n });\n\n await upsertLayoutUseCase.execute(command);\n\n expect(buildLayoutIssuesUsecaseMock.execute.calledOnce).to.be.true;\n });\n\n it('should throw BadRequestException for invalid Maily JSON content with block editor type', async () => {\n isStringifiedMailyJSONContentStub.returns(false);\n\n const invalidMailyLayoutDto = {\n ...mockLayoutDto,\n controlValues: {\n email: {\n body: 'Invalid Maily JSON',\n editorType: 'block' as 'html' | 'block',\n },\n },\n };\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: invalidMailyLayoutDto,\n });\n\n try {\n await upsertLayoutUseCase.execute(command);\n expect.fail('Should have thrown BadRequestException');\n } catch (error) {\n expect(error).to.be.instanceOf(BadRequestException);\n expect(error.message).to.equal('Content must be a valid Maily JSON content');\n }\n });\n\n it('should throw BadRequestException for content that is neither HTML nor Maily JSON', async () => {\n isStringifiedMailyJSONContentStub.returns(false);\n\n const invalidLayoutDto = {\n ...mockLayoutDto,\n controlValues: {\n email: {\n body: 'Neither HTML nor Maily JSON',\n editorType: 'html' as 'html' | 'block',\n },\n },\n };\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: invalidLayoutDto,\n });\n\n try {\n await upsertLayoutUseCase.execute(command);\n expect.fail('Should have thrown BadRequestException');\n } catch (error) {\n expect(error).to.be.instanceOf(BadRequestException);\n expect(error.message).to.equal('Content must be a valid HTML content');\n }\n });\n\n it('should skip email validation when no email controls provided', async () => {\n const noEmailLayoutDto = {\n ...mockLayoutDto,\n controlValues: {},\n };\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: noEmailLayoutDto,\n });\n\n await upsertLayoutUseCase.execute(command);\n\n expect(buildLayoutIssuesUsecaseMock.execute.calledOnce).to.be.true;\n });\n });\n\n describe('layout issues validation', () => {\n beforeEach(() => {\n getLayoutUseV0CaseMock.execute.resolves(undefined);\n createLayoutUseCaseMock.execute.resolves(mockCreatedLayout);\n getLayoutUseCaseMock.execute.resolves(\n mapLayoutToResponseDto({\n layout: mockCreatedLayout,\n controlValues: mockControlValues,\n variables: mockLayoutVariablesSchema,\n })\n );\n });\n\n it('should call buildLayoutIssuesUsecase with correct parameters', async () => {\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mockLayoutDto,\n });\n\n await upsertLayoutUseCase.execute(command);\n\n expect(buildLayoutIssuesUsecaseMock.execute.calledOnce).to.be.true;\n const issuesCommand = buildLayoutIssuesUsecaseMock.execute.firstCall.args[0];\n expect(issuesCommand.controlSchema).to.deep.equal(layoutControlSchema);\n expect(issuesCommand.controlValues).to.deep.equal(mockLayoutDto.controlValues);\n expect(issuesCommand.resourceOrigin).to.equal(ResourceOriginEnum.NOVU_CLOUD);\n expect(issuesCommand.userId).to.deep.equal(mockUser._id);\n });\n\n it('should use EXTERNAL origin when __source is not provided', async () => {\n const layoutDtoWithoutSource = {\n ...mockLayoutDto,\n __source: undefined,\n };\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: layoutDtoWithoutSource,\n });\n\n await upsertLayoutUseCase.execute(command);\n\n const issuesCommand = buildLayoutIssuesUsecaseMock.execute.firstCall.args[0];\n expect(issuesCommand.resourceOrigin).to.equal(ResourceOriginEnum.EXTERNAL);\n });\n\n it('should throw BadRequestException when layout issues exist', async () => {\n const mockIssues: LayoutIssuesDto = {\n controls: {\n 'email.body': [\n {\n message: 'Body is required',\n issueType: ContentIssueEnum.MISSING_VALUE,\n },\n ],\n 'email.editorType': [\n {\n message: 'Invalid editor type',\n issueType: ContentIssueEnum.ILLEGAL_VARIABLE_IN_CONTROL_VALUE,\n },\n ],\n },\n };\n buildLayoutIssuesUsecaseMock.execute.resolves(mockIssues);\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mockLayoutDto,\n });\n\n try {\n await upsertLayoutUseCase.execute(command);\n expect.fail('Should have thrown BadRequestException');\n } catch (error) {\n expect(error).to.be.instanceOf(BadRequestException);\n expect(error.response).to.deep.equal({ message: 'Layout has validation issues', ...mockIssues });\n }\n });\n });\n });\n\n describe('error handling', () => {\n it('should propagate errors from getLayoutUseCase', async () => {\n const error = new Error('Failed to get layout');\n getLayoutUseV0CaseMock.execute.rejects(error);\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mockLayoutDto,\n layoutIdOrInternalId: 'existing_layout_id',\n });\n\n try {\n await upsertLayoutUseCase.execute(command);\n expect.fail('Should have thrown error');\n } catch (thrownError) {\n expect(thrownError).to.equal(error);\n }\n });\n\n it('should propagate errors from createLayoutUseCase', async () => {\n const error = new Error('Failed to create layout');\n getLayoutUseV0CaseMock.execute.resolves(undefined);\n createLayoutUseCaseMock.execute.rejects(error);\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mockLayoutDto,\n });\n\n try {\n await upsertLayoutUseCase.execute(command);\n expect.fail('Should have thrown error');\n } catch (thrownError) {\n expect(thrownError).to.equal(error);\n }\n });\n\n it('should propagate errors from updateLayoutUseCase', async () => {\n const error = new Error('Failed to update layout');\n getLayoutUseV0CaseMock.execute.resolves(mockExistingLayout);\n updateLayoutUseCaseMock.execute.rejects(error);\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mockLayoutDto,\n layoutIdOrInternalId: 'existing_layout_id',\n });\n\n try {\n await upsertLayoutUseCase.execute(command);\n expect.fail('Should have thrown error');\n } catch (thrownError) {\n expect(thrownError).to.equal(error);\n }\n });\n\n it('should propagate errors from upsertControlValuesUseCase', async () => {\n const error = new Error('Failed to upsert control values');\n getLayoutUseV0CaseMock.execute.resolves(undefined);\n createLayoutUseCaseMock.execute.resolves(mockCreatedLayout);\n upsertControlValuesUseCaseMock.execute.rejects(error);\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mockLayoutDto,\n });\n\n try {\n await upsertLayoutUseCase.execute(command);\n expect.fail('Should have thrown error');\n } catch (thrownError) {\n expect(thrownError).to.equal(error);\n }\n });\n\n it('should propagate errors from getLayoutUseCase', async () => {\n const error = new Error('Failed to generate schema');\n getLayoutUseV0CaseMock.execute.resolves(undefined);\n createLayoutUseCaseMock.execute.resolves(mockCreatedLayout);\n getLayoutUseCaseMock.execute.rejects(error);\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mockLayoutDto,\n });\n\n try {\n await upsertLayoutUseCase.execute(command);\n expect.fail('Should have thrown error');\n } catch (thrownError) {\n expect(thrownError).to.equal(error);\n }\n });\n });\n\n describe('edge cases', () => {\n it('should handle layout without type and origin in update path', async () => {\n const layoutWithoutTypeAndOrigin = {\n ...mockExistingLayout,\n type: undefined,\n origin: undefined,\n };\n getLayoutUseV0CaseMock.execute.resolves(layoutWithoutTypeAndOrigin);\n updateLayoutUseCaseMock.execute.resolves(layoutWithoutTypeAndOrigin);\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mockLayoutDto,\n layoutIdOrInternalId: 'existing_layout_id',\n });\n\n await upsertLayoutUseCase.execute(command);\n\n const updateCommand = updateLayoutUseCaseMock.execute.firstCall.args[0];\n expect(updateCommand.type).to.equal(ResourceTypeEnum.BRIDGE);\n expect(updateCommand.origin).to.equal(ResourceOriginEnum.NOVU_CLOUD);\n });\n\n it('should handle undefined control values in command', async () => {\n const layoutDtoWithUndefinedControls = {\n ...mockLayoutDto,\n controlValues: undefined,\n };\n\n getLayoutUseV0CaseMock.execute.resolves(undefined);\n createLayoutUseCaseMock.execute.resolves(mockCreatedLayout);\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: layoutDtoWithUndefinedControls,\n });\n\n await upsertLayoutUseCase.execute(command);\n\n expect(controlValuesRepositoryMock.delete.calledOnce).to.be.false;\n expect(upsertControlValuesUseCaseMock.execute.calledOnce).to.be.false;\n });\n\n it('should handle empty string layoutIdOrInternalId', async () => {\n getLayoutUseV0CaseMock.execute.resolves(undefined);\n createLayoutUseCaseMock.execute.resolves(mockCreatedLayout);\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mockLayoutDto,\n layoutIdOrInternalId: '',\n });\n\n await upsertLayoutUseCase.execute(command);\n\n // Should follow create path since empty string is falsy\n expect(createLayoutUseCaseMock.execute.calledOnce).to.be.true;\n expect(getLayoutUseV0CaseMock.execute.called).to.be.false;\n });\n });\n\n describe('parameter verification', () => {\n beforeEach(() => {\n getLayoutUseV0CaseMock.execute.resolves(undefined);\n createLayoutUseCaseMock.execute.resolves(mockCreatedLayout);\n });\n\n it('should pass all required parameters to dependencies', async () => {\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: mockLayoutDto,\n });\n\n await upsertLayoutUseCase.execute(command);\n\n // Verify all major dependencies were called with correct basic parameters\n expect(buildLayoutIssuesUsecaseMock.execute.calledOnce).to.be.true;\n expect(createLayoutUseCaseMock.execute.calledOnce).to.be.true;\n expect(upsertControlValuesUseCaseMock.execute.calledOnce).to.be.true;\n expect(getLayoutUseCaseMock.execute.calledOnce).to.be.true;\n expect(analyticsServiceMock.mixpanelTrack.calledOnce).to.be.true;\n });\n\n it('should use correct identifiers and names', async () => {\n const customLayoutDto = {\n ...mockLayoutDto,\n name: 'Custom Layout Name',\n };\n\n const command = UpsertLayoutCommand.create({\n userId: mockUser._id,\n environmentId: mockUser.environmentId,\n organizationId: mockUser.organizationId,\n layoutDto: customLayoutDto,\n });\n\n await upsertLayoutUseCase.execute(command);\n\n const createCommand = createLayoutUseCaseMock.execute.firstCall.args[0];\n expect(createCommand.name).to.equal('Custom Layout Name');\n expect(createCommand.identifier).to.equal(slugify('Custom Layout Name'));\n });\n });\n});\n"
},
{
"path": "apps/api/src/app/layouts-v2/usecases/upsert-layout/upsert-layout.usecase.ts",
"content": "import { BadRequestException, Injectable } from '@nestjs/common';\nimport { ModuleRef } from '@nestjs/core';\nimport {\n AnalyticsService,\n GetLayoutCommand,\n GetLayoutCommandV0,\n GetLayoutUseCase,\n GetLayoutUseCaseV0,\n InstrumentUsecase,\n isStringifiedMailyJSONContent,\n LayoutDtoV0,\n LayoutResponseDto,\n layoutControlSchema,\n PinoLogger,\n UpsertControlValuesCommand,\n UpsertControlValuesUseCase,\n} from '@novu/application-generic';\nimport { ControlValuesRepository, LayoutRepository, LocalizationResourceEnum } from '@novu/dal';\nimport {\n ControlValuesLevelEnum,\n LayoutControlValuesDto,\n ResourceOriginEnum,\n ResourceTypeEnum,\n slugify,\n} from '@novu/shared';\nimport {\n CreateLayoutCommand,\n CreateLayoutUseCase,\n UpdateLayoutCommand,\n UpdateLayoutUseCase,\n} from '../../../layouts-v1/usecases';\nimport { MANAGE_TRANSLATIONS } from '../../../shared/constants';\nimport { BuildLayoutIssuesCommand } from '../build-layout-issues/build-layout-issues.command';\nimport { BuildLayoutIssuesUsecase } from '../build-layout-issues/build-layout-issues.usecase';\nimport { UpsertLayoutCommand } from './upsert-layout.command';\n\n@Injectable()\nexport class UpsertLayout {\n constructor(\n private getLayoutUseCaseV0: GetLayoutUseCaseV0,\n private createLayoutUseCaseV0: CreateLayoutUseCase,\n private updateLayoutUseCaseV0: UpdateLayoutUseCase,\n private controlValuesRepository: ControlValuesRepository,\n private upsertControlValuesUseCase: UpsertControlValuesUseCase,\n private layoutRepository: LayoutRepository,\n private analyticsService: AnalyticsService,\n private buildLayoutIssuesUsecase: BuildLayoutIssuesUsecase,\n private getLayoutUseCase: GetLayoutUseCase,\n private moduleRef: ModuleRef,\n private logger: PinoLogger\n ) {}\n\n @InstrumentUsecase()\n async execute(command: UpsertLayoutCommand): Promise {\n const { controlValues } = command.layoutDto;\n\n await this.validateLayout({\n command,\n controlValues,\n });\n\n const existingLayout = command.layoutIdOrInternalId\n ? await this.getLayoutUseCaseV0.execute(\n GetLayoutCommandV0.create({\n layoutIdOrInternalId: command.layoutIdOrInternalId,\n environmentId: command.environmentId,\n organizationId: command.organizationId,\n type: ResourceTypeEnum.BRIDGE,\n origin: ResourceOriginEnum.NOVU_CLOUD,\n })\n )\n : null;\n\n let upsertedLayout: LayoutDtoV0;\n if (existingLayout) {\n this.mixpanelTrack(command, 'Layout Update - [Layouts]');\n\n upsertedLayout = await this.updateLayoutUseCaseV0.execute(\n UpdateLayoutCommand.create({\n environmentId: command.environmentId,\n organizationId: command.organizationId,\n userId: command.userId,\n layoutId: existingLayout._id!,\n name: command.layoutDto.name,\n type: existingLayout.type ?? ResourceTypeEnum.BRIDGE,\n origin: existingLayout.origin ?? ResourceOriginEnum.NOVU_CLOUD,\n })\n );\n } else {\n this.mixpanelTrack(command, 'Layout Create - [Layouts]');\n\n const defaultLayout = await this.layoutRepository.findOne({\n _organizationId: command.organizationId,\n _environmentId: command.environmentId,\n type: ResourceTypeEnum.BRIDGE,\n origin: ResourceOriginEnum.NOVU_CLOUD,\n isDefault: true,\n });\n\n upsertedLayout = await this.createLayoutUseCaseV0.execute(\n CreateLayoutCommand.create({\n environmentId: command.environmentId,\n organizationId: command.organizationId,\n userId: command.userId,\n name: command.layoutDto.name,\n identifier: command.layoutDto.layoutId || slugify(command.layoutDto.name),\n type: ResourceTypeEnum.BRIDGE,\n origin: ResourceOriginEnum.NOVU_CLOUD,\n isDefault: !defaultLayout,\n })\n );\n }\n\n await this.toggleTranslationsForLayout(command, upsertedLayout);\n\n await this.upsertControlValues(command, upsertedLayout._id!);\n\n return await this.getLayoutUseCase.execute(\n GetLayoutCommand.create({\n layoutIdOrInternalId: upsertedLayout.identifier,\n environmentId: command.environmentId,\n organizationId: command.organizationId,\n userId: command.userId,\n })\n );\n }\n\n private async validateLayout({\n command,\n controlValues,\n }: {\n command: UpsertLayoutCommand;\n controlValues?: LayoutControlValuesDto | null;\n }) {\n if (!controlValues) {\n return;\n }\n\n if (controlValues.email) {\n const { body: content, editorType } = controlValues.email;\n const isMailyContent = isStringifiedMailyJSONContent(content);\n const isHtmlContent =\n content.includes('') &&\n content.includes('');\n\n if (!isMailyContent && !isHtmlContent) {\n throw new BadRequestException(\n editorType === 'html' ? 'Content must be a valid HTML content' : 'Content must be a valid Maily JSON content'\n );\n }\n\n if (editorType === 'html' && !isHtmlContent) {\n throw new BadRequestException('Content must be a valid HTML content');\n } else if (editorType === 'block' && !isMailyContent) {\n throw new BadRequestException('Content must be a valid Maily JSON content');\n }\n }\n\n const issues = await this.buildLayoutIssuesUsecase.execute(\n BuildLayoutIssuesCommand.create({\n controlSchema: layoutControlSchema,\n controlValues,\n resourceOrigin: command.layoutDto.__source ? ResourceOriginEnum.NOVU_CLOUD : ResourceOriginEnum.EXTERNAL,\n environmentId: command.environmentId,\n organizationId: command.organizationId,\n userId: command.userId,\n })\n );\n\n if (Object.keys(issues).length > 0) {\n throw new BadRequestException({ message: 'Layout has validation issues', ...issues });\n }\n }\n\n private async upsertControlValues(command: UpsertLayoutCommand, layoutId: string) {\n const {\n layoutDto: { controlValues },\n } = command;\n const doNothing = typeof controlValues === 'undefined';\n if (doNothing) {\n return null;\n }\n\n const shouldDelete = controlValues === null;\n if (shouldDelete) {\n this.controlValuesRepository.delete({\n _environmentId: command.environmentId,\n _organizationId: command.organizationId,\n _layoutId: layoutId,\n level: ControlValuesLevelEnum.LAYOUT_CONTROLS,\n });\n\n return null;\n }\n\n return this.upsertControlValuesUseCase.execute(\n UpsertControlValuesCommand.create({\n organizationId: command.organizationId,\n environmentId: command.environmentId,\n layoutId,\n level: ControlValuesLevelEnum.LAYOUT_CONTROLS,\n newControlValues: controlValues as unknown as Record,\n })\n );\n }\n\n private mixpanelTrack(command: UpsertLayoutCommand, eventName: string) {\n this.analyticsService.mixpanelTrack(eventName, command.userId, {\n _organization: command.organizationId,\n name: command.layoutDto.name,\n source: command.layoutDto.__source,\n });\n }\n\n private async toggleTranslationsForLayout(command: UpsertLayoutCommand, layoutDto: LayoutDtoV0) {\n const isEnterprise = process.env.NOVU_ENTERPRISE === 'true' || process.env.CI_EE_TEST === 'true';\n const isSelfHosted = process.env.IS_SELF_HOSTED === 'true';\n\n if (!isEnterprise || isSelfHosted) {\n return;\n }\n\n try {\n const manageTranslations = this.moduleRef.get(MANAGE_TRANSLATIONS, {\n strict: false,\n });\n\n await manageTranslations.execute({\n enabled: command.layoutDto.isTranslationEnabled,\n resourceId: layoutDto.identifier,\n resourceType: LocalizationResourceEnum.LAYOUT,\n organizationId: command.organizationId,\n environmentId: command.environmentId,\n userId: command.userId,\n resourceEntity: layoutDto,\n });\n } catch (error) {\n this.logger.error(\n `Failed to ${command.layoutDto.isTranslationEnabled ? 'enable' : 'disable'} translations for layout`,\n {\n layoutId: layoutDto.identifier,\n enabled: command.layoutDto.isTranslationEnabled,\n organizationId: command.organizationId,\n error: error instanceof Error ? error.message : String(error),\n }\n );\n\n throw error;\n }\n }\n}\n"
},
{
"path": "apps/api/src/app/layouts-v2/utils/layout-templates.ts",
"content": "export const EMPTY_LAYOUT = {\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n attrs: { textAlign: null, showIfKey: null },\n content: [{ type: 'text', text: ' ' }],\n },\n {\n type: 'paragraph',\n attrs: { textAlign: 'left', showIfKey: null },\n content: [\n {\n type: 'variable',\n attrs: {\n id: 'content',\n label: null,\n fallback: null,\n required: false,\n aliasFor: null,\n },\n },\n ],\n },\n {\n type: 'paragraph',\n attrs: { textAlign: null, showIfKey: null },\n content: [{ type: 'text', text: ' ' }],\n },\n ],\n};\n\nexport const createDefaultLayout = (organizationName: string) => ({\n type: 'doc',\n content: [\n {\n type: 'columns',\n attrs: { showIfKey: null, gap: 8 },\n content: [\n {\n type: 'column',\n attrs: {\n columnId: '36de3eda-0677-47c3-a8b7-e071dec9ce30',\n width: 'auto',\n verticalAlign: 'middle',\n },\n content: [\n {\n type: 'image',\n attrs: {\n src: 'https://prod-novu-app-bucket.s3.us-east-1.amazonaws.com/assets/email-editor/logo.png',\n alt: null,\n title: null,\n width: '48',\n height: '48',\n alignment: 'left',\n externalLink: null,\n isExternalLinkVariable: false,\n borderRadius: 0,\n isSrcVariable: false,\n aspectRatio: null,\n lockAspectRatio: true,\n showIfKey: null,\n aliasFor: null,\n },\n },\n ],\n },\n {\n type: 'column',\n attrs: {\n columnId: '6feb593e-374a-4479-a1c7-872c60c2f4e0',\n width: 'auto',\n verticalAlign: 'middle',\n },\n content: [\n {\n type: 'paragraph',\n attrs: { textAlign: 'right', showIfKey: null },\n },\n ],\n },\n ],\n },\n {\n type: 'spacer',\n attrs: { height: 8, showIfKey: null },\n },\n {\n type: 'paragraph',\n attrs: { textAlign: null, showIfKey: null },\n content: [\n {\n type: 'variable',\n attrs: {\n id: 'content',\n label: null,\n fallback: null,\n required: false,\n aliasFor: null,\n },\n },\n ],\n },\n {\n type: 'spacer',\n attrs: { height: 8, showIfKey: null },\n },\n {\n type: 'columns',\n attrs: { showIfKey: null, gap: 0 },\n content: [\n {\n type: 'column',\n attrs: {\n columnId: '8a20f82f-ecb5-4cbd-923e-ff82f3bb9b79',\n width: '60',\n verticalAlign: 'top',\n },\n content: [\n {\n type: 'paragraph',\n attrs: { textAlign: null, showIfKey: null },\n content: [{ type: 'text', text: organizationName }],\n },\n {\n type: 'spacer',\n attrs: { height: 4, showIfKey: null },\n },\n {\n type: 'footer',\n attrs: { textAlign: null, 'maily-component': 'footer' },\n content: [\n {\n type: 'text',\n marks: [{ type: 'textStyle' }],\n text: '1234 Example Street, DE 19801, United States',\n },\n ],\n },\n ],\n },\n {\n type: 'column',\n attrs: {\n columnId: 'cd30ba93-7a8f-4d03-b66a-88ae4fe99abf',\n width: '40',\n verticalAlign: 'top',\n },\n content: [\n {\n type: 'paragraph',\n attrs: { textAlign: 'right', showIfKey: null },\n content: [\n {\n type: 'text',\n marks: [\n {\n type: 'link',\n attrs: {\n href: 'https://novu.co/',\n target: '_blank',\n rel: 'noopener noreferrer nofollow',\n class: null,\n isUrlVariable: false,\n aliasFor: null,\n },\n },\n ],\n text: 'Visit Company',\n },\n { type: 'text', text: ' | ' },\n {\n type: 'text',\n marks: [\n {\n type: 'link',\n attrs: {\n href: 'support@novu.co',\n target: '_blank',\n rel: 'noopener noreferrer nofollow',\n class: null,\n isUrlVariable: false,\n aliasFor: null,\n },\n },\n ],\n text: 'Contact Us',\n },\n ],\n },\n {\n type: 'spacer',\n attrs: { height: 4, showIfKey: null },\n },\n {\n type: 'section',\n attrs: {\n borderRadius: 0,\n backgroundColor: '#FFFFFF',\n align: 'left',\n borderWidth: 0,\n borderColor: '#e2e2e2',\n paddingTop: 0,\n paddingRight: 0,\n paddingBottom: 0,\n paddingLeft: 0,\n marginTop: 0,\n marginRight: 0,\n marginBottom: 0,\n marginLeft: 0,\n showIfKey: null,\n },\n content: [\n {\n type: 'paragraph',\n attrs: { textAlign: 'right', showIfKey: null },\n content: [\n {\n type: 'inlineImage',\n attrs: {\n height: 20,\n width: 20,\n src: 'https://prod-novu-app-bucket.s3.us-east-1.amazonaws.com/assets/email-editor/linkedin.png',\n isSrcVariable: false,\n alt: null,\n title: null,\n externalLink: 'https://www.linkedin.com/company/novuco/',\n isExternalLinkVariable: false,\n aliasFor: null,\n },\n },\n { type: 'text', text: ' ' },\n {\n type: 'inlineImage',\n attrs: {\n height: 20,\n width: 20,\n src: 'https://prod-novu-app-bucket.s3.us-east-1.amazonaws.com/assets/email-editor/youtube.png',\n isSrcVariable: false,\n alt: null,\n title: null,\n externalLink: 'https://www.youtube.com/@novuhq',\n isExternalLinkVariable: false,\n aliasFor: null,\n },\n },\n { type: 'text', text: ' ' },\n {\n type: 'inlineImage',\n attrs: {\n height: 20,\n width: 20,\n src: 'https://prod-novu-app-bucket.s3.us-east-1.amazonaws.com/assets/email-editor/twitter.png',\n isSrcVariable: false,\n alt: null,\n title: null,\n externalLink: 'https://x.com/novuhq',\n isExternalLinkVariable: false,\n aliasFor: null,\n },\n },\n ],\n },\n ],\n },\n ],\n },\n ],\n },\n {\n type: 'spacer',\n attrs: { height: 8, showIfKey: null },\n },\n ],\n});\n"
},
{
"path": "apps/api/src/app/message-template/message-template.controller.ts",
"content": "import { Controller } from '@nestjs/common';\nimport { ApiBearerAuth } from '@nestjs/swagger';\n\n@Controller('/message-templates')\nexport class MessageTemplateController {}\n"
},
{
"path": "apps/api/src/app/message-template/message-template.module.ts",
"content": "import { Module } from '@nestjs/common';\nimport { ChangeModule } from '../change/change.module';\nimport { SharedModule } from '../shared/shared.module';\nimport { MessageTemplateController } from './message-template.controller';\nimport { USE_CASES } from './usecases';\n\n@Module({\n imports: [SharedModule, ChangeModule],\n controllers: [MessageTemplateController],\n providers: [...USE_CASES],\n exports: [...USE_CASES],\n})\nexport class MessageTemplateModule {}\n"
},
{
"path": "apps/api/src/app/message-template/usecases/find-message-templates-by-layout/find-message-templates-by-layout.command.ts",
"content": "import { LayoutId } from '@novu/shared';\nimport { IsDefined, IsString } from 'class-validator';\n\nimport { EnvironmentCommand } from '../../../shared/commands/project.command';\n\nexport class FindMessageTemplatesByLayoutCommand extends EnvironmentCommand {\n @IsDefined()\n @IsString()\n layoutId: string;\n}\n"
},
{
"path": "apps/api/src/app/message-template/usecases/find-message-templates-by-layout/find-message-templates-by-layout.use-case.ts",
"content": "import { Injectable } from '@nestjs/common';\nimport { MessageTemplateEntity, MessageTemplateRepository } from '@novu/dal';\n\nimport { FindMessageTemplatesByLayoutCommand } from './find-message-templates-by-layout.command';\n\nconst DEFAULT_PAGE_SIZE = 100;\n\n@Injectable()\nexport class FindMessageTemplatesByLayoutUseCase {\n constructor(private messageTemplateRepository: MessageTemplateRepository) {}\n\n async execute(command: FindMessageTemplatesByLayoutCommand): Promise {\n // TODO: Implement proper pagination\n const messageTemplates = await this.messageTemplateRepository.getMessageTemplatesByLayout(\n command.environmentId,\n command.layoutId,\n { limit: DEFAULT_PAGE_SIZE }\n );\n\n return messageTemplates;\n }\n}\n"
},
{
"path": "apps/api/src/app/message-template/usecases/find-message-templates-by-layout/index.ts",
"content": "export * from './find-message-templates-by-layout.command';\nexport * from './find-message-templates-by-layout.use-case';\n"
},
{
"path": "apps/api/src/app/message-template/usecases/index.ts",
"content": "import { CreateMessageTemplate, DeleteMessageTemplate, UpdateMessageTemplate } from '@novu/application-generic';\n\nimport { FindMessageTemplatesByLayoutUseCase } from './find-message-templates-by-layout/find-message-templates-by-layout.use-case';\n\nexport * from './find-message-templates-by-layout';\nexport const USE_CASES = [\n CreateMessageTemplate,\n FindMessageTemplatesByLayoutUseCase,\n UpdateMessageTemplate,\n DeleteMessageTemplate,\n];\n"
},
{
"path": "apps/api/src/app/messages/dtos/delete-message-response.dto.ts",
"content": "import { ApiProperty } from '@nestjs/swagger';\nimport { IsBoolean, IsDefined, IsString } from 'class-validator';\n\nexport class DeleteMessageResponseDto {\n @ApiProperty({\n description: 'A boolean stating the success of the action',\n })\n @IsBoolean()\n @IsDefined()\n acknowledged: boolean;\n\n @ApiProperty({\n description: 'The status enum for the performed action',\n enum: ['deleted'],\n })\n @IsString()\n @IsDefined()\n status: string;\n}\n"
},
{
"path": "apps/api/src/app/messages/dtos/get-messages-requests.dto.ts",
"content": "import { ApiPropertyOptional } from '@nestjs/swagger';\nimport { ChannelTypeEnum } from '@novu/shared';\nimport { Transform } from 'class-transformer';\nimport { IsArray, IsNumber, IsOptional, IsString } from 'class-validator';\n\nexport class GetMessagesRequestDto {\n @ApiPropertyOptional({\n enum: [...Object.values(ChannelTypeEnum)],\n enumName: 'ChannelTypeEnum',\n })\n channel?: ChannelTypeEnum;\n\n @ApiPropertyOptional({\n type: String,\n })\n @IsOptional()\n subscriberId?: string;\n\n @ApiPropertyOptional({\n type: String,\n isArray: true,\n })\n @IsOptional()\n transactionId?: string[];\n\n @ApiPropertyOptional({\n type: String,\n isArray: true,\n description: 'Filter by exact context keys, order insensitive (format: \"type:id\")',\n example: ['tenant:org-123', 'region:us-east-1'],\n })\n @IsOptional()\n @Transform(({ value }) => {\n // No parameter = no filter\n if (value === undefined) return undefined;\n\n // Empty string = filter for records with no context\n if (value === '') return [];\n\n // Normalize to array and remove empty strings\n const array = Array.isArray(value) ? value : [value];\n return array.filter((v) => v !== '');\n })\n @IsArray()\n @IsString({ each: true })\n contextKeys?: string[];\n\n @ApiPropertyOptional({\n type: Number,\n default: 0,\n })\n @IsOptional()\n @IsNumber()\n @Transform(({ value }) => Number(value))\n page?: number;\n\n @ApiPropertyOptional({\n type: Number,\n default: 10,\n })\n @IsOptional()\n @IsNumber()\n @Transform(({ value }) => Number(value))\n limit?: number;\n\n constructor() {\n this.page = 0; // Default value\n this.limit = 10; // Default value\n }\n}\n"
},
{
"path": "apps/api/src/app/messages/dtos/remove-messages-by-transactionId-request.dto.ts",
"content": "import { ApiPropertyOptional } from '@nestjs/swagger';\nimport { ChannelTypeEnum } from '@novu/shared';\nimport { IsEnum, IsOptional } from 'class-validator';\n\nexport class DeleteMessageByTransactionIdRequestDto {\n @ApiPropertyOptional({\n enum: ChannelTypeEnum,\n description: 'The channel of the message to be deleted',\n })\n @IsOptional()\n @IsEnum(ChannelTypeEnum)\n channel?: ChannelTypeEnum;\n}\n"
},
{
"path": "apps/api/src/app/messages/e2e/get-messages.e2e.ts",
"content": "import { Novu } from '@novu/api';\nimport { ChannelTypeEnum } from '@novu/api/models/components';\nimport { NotificationTemplateEntity, SubscriberEntity } from '@novu/dal';\nimport { CreateWorkflowDto, StepTypeEnum, WorkflowCreationSourceEnum, WorkflowResponseDto } from '@novu/shared';\nimport { SubscribersService, UserSession } from '@novu/testing';\nimport { expect } from 'chai';\nimport { initNovuClassSdk } from '../../shared/helpers/e2e/sdk/e2e-sdk.helper';\n\ndescribe('Get Message - /messages (GET) #novu-v2', () => {\n let session: UserSession;\n let template: NotificationTemplateEntity;\n let subscriber: SubscriberEntity;\n let subscriberService: SubscribersService;\n let novuClient: Novu;\n\n beforeEach(async () => {\n session = new UserSession();\n await session.initialize();\n\n template = await session.createTemplate();\n subscriberService = new SubscribersService(session.organization._id, session.environment._id);\n subscriber = await subscriberService.createSubscriber();\n novuClient = initNovuClassSdk(session);\n });\n\n it('should fetch existing messages', async () => {\n const subscriber2 = await subscriberService.createSubscriber();\n\n await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: [\n { subscriberId: subscriber.subscriberId, email: 'gg@ff.com' },\n { subscriberId: subscriber2.subscriberId, email: 'john@doe.com' },\n ],\n payload: {\n email: 'new-test-email@gmail.com',\n firstName: 'Testing of User Name',\n urlVar: '/test/url/path',\n },\n });\n\n await session.waitForJobCompletion(template._id);\n\n let response = await novuClient.messages.retrieve({});\n expect(response.result.data.length).to.be.equal(4);\n\n response = await novuClient.messages.retrieve({ channel: ChannelTypeEnum.Email });\n expect(response.result.data.length).to.be.equal(2);\n\n response = await novuClient.messages.retrieve({ subscriberId: subscriber2.subscriberId });\n expect(response.result.data.length).to.be.equal(2);\n });\n\n it('should fetch messages using transactionId filter', async () => {\n const subscriber3 = await subscriberService.createSubscriber();\n\n const transactionId1 = '1566f9d0-6037-48c1-b356-42667921cadd';\n const transactionId2 = 'd2d9f9b5-4a96-403a-927f-1f8f40c6c7a9';\n\n await triggerEventWithTransactionId(template.triggers[0].identifier, subscriber3.subscriberId, transactionId1);\n await triggerEventWithTransactionId(template.triggers[0].identifier, subscriber3.subscriberId, transactionId2);\n\n await session.waitForWorkflowQueueCompletion();\n await session.waitForSubscriberQueueCompletion();\n await session.waitForStandardQueueCompletion();\n await session.waitForJobCompletion(template._id);\n\n let response = await novuClient.messages.retrieve({ subscriberId: subscriber3.subscriberId });\n expect(response.result.data.length).to.be.equal(4);\n\n response = await novuClient.messages.retrieve({ transactionId: [transactionId1] });\n expect(response.result.data.length).to.be.equal(2);\n\n response = await novuClient.messages.retrieve({ transactionId: [transactionId1, transactionId2] });\n expect(response.result.data.length).to.be.equal(4);\n\n response = await novuClient.messages.retrieve({ transactionId: [transactionId2] });\n expect(response.result.data.length).to.be.equal(2);\n });\n\n it('should fetch messages using contextKeys filter', async () => {\n const subscriber4 = await subscriberService.createSubscriber();\n\n const workflowBody: CreateWorkflowDto = {\n name: 'Test Context Workflow',\n workflowId: 'test-context-workflow-messages',\n __source: WorkflowCreationSourceEnum.DASHBOARD,\n steps: [\n {\n type: StepTypeEnum.IN_APP,\n name: 'Test Step',\n controlValues: {\n subject: 'Test Subject',\n body: 'Test Body',\n },\n },\n ],\n };\n\n const workflowResponse = await session.testAgent.post('/v2/workflows').send(workflowBody);\n expect(workflowResponse.status).to.equal(201);\n const workflow: WorkflowResponseDto = workflowResponse.body.data;\n\n await novuClient.trigger({\n workflowId: workflow.workflowId,\n to: subscriber4.subscriberId,\n payload: {},\n context: { teamId: 'team-alpha' },\n });\n\n await novuClient.trigger({\n workflowId: workflow.workflowId,\n to: subscriber4.subscriberId,\n payload: {},\n context: { teamId: 'team-beta' },\n });\n\n await session.waitForWorkflowQueueCompletion();\n await session.waitForSubscriberQueueCompletion();\n await session.waitForStandardQueueCompletion();\n await session.waitForJobCompletion(workflow._id);\n\n let response = await novuClient.messages.retrieve({ subscriberId: subscriber4.subscriberId });\n expect(response.result.data.length).to.be.equal(2);\n\n response = await novuClient.messages.retrieve({\n subscriberId: subscriber4.subscriberId,\n contextKeys: ['teamId:team-alpha'],\n });\n expect(response.result.data.length).to.be.equal(1);\n\n response = await novuClient.messages.retrieve({\n subscriberId: subscriber4.subscriberId,\n contextKeys: ['teamId:team-beta'],\n });\n expect(response.result.data.length).to.be.equal(1);\n });\n\n async function triggerEventWithTransactionId(\n templateIdentifier: string,\n subscriberId: string,\n transactionId: string\n ) {\n return await novuClient.trigger({\n workflowId: templateIdentifier,\n to: [{ subscriberId, email: 'gg@ff.com' }],\n payload: {},\n transactionId,\n });\n }\n});\n"
},
{
"path": "apps/api/src/app/messages/e2e/remove-message.e2e.ts",
"content": "import { Novu } from '@novu/api';\nimport { MessageRepository, NotificationTemplateEntity, SubscriberEntity } from '@novu/dal';\nimport { ChannelTypeEnum } from '@novu/shared';\nimport { SubscribersService, UserSession } from '@novu/testing';\nimport axios from 'axios';\nimport { expect } from 'chai';\nimport { initNovuClassSdk } from '../../shared/helpers/e2e/sdk/e2e-sdk.helper';\n\nconst axiosInstance = axios.create();\n\ndescribe('Delete Message - /messages/:messageId (DELETE) #novu-v2', () => {\n let session: UserSession;\n const messageRepository = new MessageRepository();\n let template: NotificationTemplateEntity;\n let subscriber: SubscriberEntity;\n let subscriberService: SubscribersService;\n let novuClient: Novu;\n\n beforeEach(async () => {\n session = new UserSession();\n await session.initialize();\n template = await session.createTemplate();\n subscriberService = new SubscribersService(session.organization._id, session.environment._id);\n subscriber = await subscriberService.createSubscriber();\n novuClient = initNovuClassSdk(session);\n });\n\n it('should fail to delete non existing message', async () => {\n const response = await session.testAgent.delete(`/v1/messages/${MessageRepository.createObjectId()}`);\n\n expect(response.statusCode).to.equal(404);\n expect(response.body.error).to.equal('Not Found');\n });\n\n it('should delete a existing message', async () => {\n await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: [{ subscriberId: subscriber.subscriberId, email: 'gg@ff.com' }],\n payload: {\n email: 'new-test-email@gmail.com',\n firstName: 'Testing of User Name',\n urlVar: '/test/url/path',\n },\n });\n\n await session.waitForJobCompletion(template._id);\n\n const messages = await messageRepository.findBySubscriberChannel(\n session.environment._id,\n subscriber._id,\n ChannelTypeEnum.EMAIL\n );\n\n const message = messages[0];\n\n await novuClient.messages.delete(message._id);\n\n const result = await messageRepository.findOne({ _id: message._id, _environmentId: message._environmentId });\n expect(result).to.not.be.ok;\n });\n});\n"
},
{
"path": "apps/api/src/app/messages/e2e/remove-messages-by-transactionId.e2e.ts",
"content": "import { Novu } from '@novu/api';\nimport { MessageRepository, NotificationTemplateEntity, SubscriberEntity } from '@novu/dal';\nimport { ChannelTypeEnum } from '@novu/shared';\nimport { SubscribersService, UserSession } from '@novu/testing';\nimport { expect } from 'chai';\nimport { expectSdkExceptionGeneric, initNovuClassSdk } from '../../shared/helpers/e2e/sdk/e2e-sdk.helper';\n\ndescribe('Delete Messages By TransactionId - /messages/?transactionId= (DELETE) #novu-v2', () => {\n let session: UserSession;\n const messageRepository = new MessageRepository();\n let template: NotificationTemplateEntity;\n let subscriber: SubscriberEntity;\n let subscriberService: SubscribersService;\n let novuClient: Novu;\n\n beforeEach(async () => {\n session = new UserSession();\n await session.initialize();\n template = await session.createTemplate();\n subscriberService = new SubscribersService(session.organization._id, session.environment._id);\n subscriber = await subscriberService.createSubscriber();\n novuClient = initNovuClassSdk(session);\n });\n\n it('should fail to delete non existing message', async () => {\n const { error } = await expectSdkExceptionGeneric(() => novuClient.messages.deleteByTransactionId('abc-1234'));\n expect(error?.statusCode).to.equal(404);\n expect(error?.ctx?.error, JSON.stringify(error)).to.equal('Not Found');\n });\n\n it('should delete messages by transactionId', async () => {\n await novuClient.subscribers.create({\n subscriberId: '123456',\n firstName: 'broadcast ',\n lastName: 'subscriber',\n });\n\n const res = await novuClient.triggerBroadcast({\n name: template.triggers[0].identifier,\n payload: {\n email: 'new-test-email@gmail.com',\n firstName: 'Testing of User Name',\n urlVar: '/test/url/path',\n },\n });\n await session.waitForJobCompletion(template._id);\n\n const { transactionId } = res.result;\n\n const messages = await messageRepository.find({\n _environmentId: session.environment._id,\n _organizationId: session.organization._id,\n transactionId,\n });\n\n expect(messages.length).to.be.greaterThan(0);\n expect(transactionId).to.be.ok;\n\n if (transactionId == null) {\n throw new Error('must have transaction id');\n }\n await novuClient.messages.deleteByTransactionId(transactionId);\n\n const result = await messageRepository.find({\n transactionId,\n _environmentId: session.environment._id,\n _organizationId: session.organization._id,\n });\n\n expect(result.length).to.equal(0);\n });\n\n it('should delete messages by transactionId and channel', async () => {\n const response = await novuClient.triggerBroadcast({\n name: template.triggers[0].identifier,\n payload: {\n email: 'new-test-email@gmail.com',\n firstName: 'Testing of User Name',\n urlVar: '/test/url/path',\n },\n });\n\n await session.waitForJobCompletion(template._id);\n const { transactionId } = response.result;\n\n const messages = await messageRepository.find({\n _environmentId: session.environment._id,\n _organizationId: session.organization._id,\n transactionId,\n });\n\n const emailMessages = messages.filter((message) => message.channel === ChannelTypeEnum.EMAIL);\n const inAppMessages = messages.filter((message) => message.channel === ChannelTypeEnum.IN_APP);\n const inAppMessagesCount = inAppMessages.length;\n\n expect(messages.length).to.be.greaterThan(0);\n expect(emailMessages.length).to.be.greaterThan(0);\n expect(inAppMessagesCount).to.be.greaterThan(0);\n expect(transactionId).to.be.ok;\n if (transactionId == null) {\n throw new Error('must have transaction id');\n }\n await novuClient.messages.deleteByTransactionId(transactionId, ChannelTypeEnum.EMAIL);\n\n const result = await messageRepository.find({\n transactionId,\n _environmentId: session.environment._id,\n _organizationId: session.organization._id,\n });\n\n const emailResult = result.filter((message) => message.channel === ChannelTypeEnum.EMAIL);\n const inAppResult = result.filter((message) => message.channel === ChannelTypeEnum.IN_APP);\n const inAppResultCount = inAppResult.length;\n\n expect(result.length).to.be.greaterThan(0);\n expect(emailResult.length).to.equal(0);\n expect(inAppResultCount).to.be.greaterThan(0);\n expect(inAppResultCount).to.equal(inAppMessagesCount);\n });\n});\n"
},
{
"path": "apps/api/src/app/messages/messages.controller.ts",
"content": "import { Controller, Delete, Get, HttpCode, HttpStatus, Param, Query } from '@nestjs/common';\nimport { ApiOperation, ApiParam, ApiTags } from '@nestjs/swagger';\nimport { RequirePermissions } from '@novu/application-generic';\nimport { PermissionsEnum, UserSessionData } from '@novu/shared';\nimport { RequireAuthentication } from '../auth/framework/auth.decorator';\nimport { ExternalApiAccessible } from '../auth/framework/external-api.decorator';\nimport {\n ApiCommonResponses,\n ApiNoContentResponse,\n ApiOkResponse,\n ApiResponse,\n} from '../shared/framework/response.decorator';\nimport { SdkMethodName } from '../shared/framework/swagger/sdk.decorators';\nimport { UserSession } from '../shared/framework/user.decorator';\nimport { MessagesResponseDto } from '../widgets/dtos/message-response.dto';\nimport { DeleteMessageResponseDto } from './dtos/delete-message-response.dto';\nimport { GetMessagesRequestDto } from './dtos/get-messages-requests.dto';\nimport { DeleteMessageByTransactionIdRequestDto } from './dtos/remove-messages-by-transactionId-request.dto';\nimport { DeleteMessageParams } from './params/delete-message.param';\nimport { GetMessages, GetMessagesCommand } from './usecases/get-messages';\nimport { RemoveMessage, RemoveMessageCommand } from './usecases/remove-message';\nimport { RemoveMessagesByTransactionIdCommand } from './usecases/remove-messages-by-transactionId/remove-messages-by-transactionId.command';\nimport { RemoveMessagesByTransactionId } from './usecases/remove-messages-by-transactionId/remove-messages-by-transactionId.usecase';\n\n@ApiCommonResponses()\n@RequireAuthentication()\n@Controller('/messages')\n@ApiTags('Messages')\nexport class MessagesController {\n constructor(\n private removeMessage: RemoveMessage,\n private getMessagesUsecase: GetMessages,\n private removeMessagesByTransactionId: RemoveMessagesByTransactionId\n ) {}\n\n @Get('')\n @ExternalApiAccessible()\n @ApiOkResponse({\n type: MessagesResponseDto,\n })\n @ApiOperation({\n summary: 'List all messages',\n description: `List all messages for the current environment. \n This API supports filtering by **channel**, **subscriberId**, and **transactionId**. \n This API returns a paginated list of messages.`,\n })\n @RequirePermissions(PermissionsEnum.MESSAGE_READ)\n async getMessages(\n @UserSession() user: UserSessionData,\n @Query() query: GetMessagesRequestDto\n ): Promise {\n let transactionIdQuery: string[] | undefined;\n if (query.transactionId) {\n transactionIdQuery = Array.isArray(query.transactionId) ? query.transactionId : [query.transactionId];\n }\n\n return await this.getMessagesUsecase.execute(\n GetMessagesCommand.create({\n organizationId: user.organizationId,\n environmentId: user.environmentId,\n channel: query.channel,\n subscriberId: query.subscriberId,\n contextKeys: query.contextKeys,\n page: query.page ? Number(query.page) : 0,\n limit: query.limit ? Number(query.limit) : 10,\n transactionIds: transactionIdQuery,\n })\n );\n }\n\n @Delete('/:messageId')\n @ExternalApiAccessible()\n @ApiResponse(DeleteMessageResponseDto)\n @ApiOperation({\n summary: 'Delete a message',\n description: `Delete a message entity from the Novu platform by **messageId**. \n This action is irreversible. **messageId** is required and of mongodbId type.`,\n })\n @ApiParam({ name: 'messageId', type: String, required: true, example: '507f1f77bcf86cd799439011' })\n @RequirePermissions(PermissionsEnum.MESSAGE_WRITE)\n async deleteMessage(\n @UserSession() user: UserSessionData,\n @Param() { messageId }: DeleteMessageParams\n ): Promise {\n return await this.removeMessage.execute(\n RemoveMessageCommand.create({\n environmentId: user.environmentId,\n organizationId: user.organizationId,\n messageId,\n })\n );\n }\n\n @Delete('/transaction/:transactionId')\n @HttpCode(HttpStatus.NO_CONTENT)\n @ExternalApiAccessible()\n @ApiNoContentResponse()\n @ApiOperation({\n summary: 'Delete messages by transactionId',\n description: `Delete multiple messages from the Novu platform using **transactionId** of triggered event. \n This API supports filtering by **channel** and delete all messages associated with the **transactionId**.`,\n })\n @ApiParam({ name: 'transactionId', type: String, required: true, example: '507f1f77bcf86cd799439011' })\n @SdkMethodName('deleteByTransactionId')\n @RequirePermissions(PermissionsEnum.MESSAGE_WRITE)\n async deleteMessagesByTransactionId(\n @UserSession() user: UserSessionData,\n @Param() { transactionId }: { transactionId: string },\n @Query() query: DeleteMessageByTransactionIdRequestDto\n ) {\n return await this.removeMessagesByTransactionId.execute(\n RemoveMessagesByTransactionIdCommand.create({\n environmentId: user.environmentId,\n organizationId: user.organizationId,\n transactionId,\n channel: query.channel,\n })\n );\n }\n}\n"
},
{
"path": "apps/api/src/app/messages/messages.module.ts",
"content": "import { forwardRef, Module } from '@nestjs/common';\nimport { TerminusModule } from '@nestjs/terminus';\nimport { AuthModule } from '../auth/auth.module';\nimport { SharedModule } from '../shared/shared.module';\nimport { SubscribersV1Module } from '../subscribers/subscribersV1.module';\nimport { WidgetsModule } from '../widgets/widgets.module';\nimport { MessagesController } from './messages.controller';\nimport { USE_CASES } from './usecases';\n\n@Module({\n imports: [SharedModule, SubscribersV1Module, AuthModule, TerminusModule, forwardRef(() => WidgetsModule)],\n controllers: [MessagesController],\n providers: [...USE_CASES],\n exports: [...USE_CASES],\n})\nexport class MessagesModule {}\n"
},
{
"path": "apps/api/src/app/messages/params/delete-message.param.ts",
"content": "import { IsMongoId } from 'class-validator';\n\nexport class DeleteMessageParams {\n @IsMongoId()\n messageId: string;\n}\n"
},
{
"path": "apps/api/src/app/messages/usecases/get-messages/get-messages.command.ts",
"content": "import { ChannelTypeEnum } from '@novu/shared';\nimport { IsArray, IsNumber, IsOptional, IsString } from 'class-validator';\nimport { EnvironmentCommand } from '../../../shared/commands/project.command';\n\nexport class GetMessagesCommand extends EnvironmentCommand {\n @IsOptional()\n subscriberId?: string;\n\n @IsOptional()\n channel?: ChannelTypeEnum;\n\n @IsOptional()\n @IsArray()\n @IsString({ each: true })\n contextKeys?: string[];\n\n @IsNumber()\n page = 0;\n\n @IsNumber()\n limit = 10;\n\n @IsOptional()\n @IsArray()\n @IsString({ each: true })\n transactionIds?: string[] | undefined;\n}\n"
},
{
"path": "apps/api/src/app/messages/usecases/get-messages/get-messages.usecase.ts",
"content": "import { BadRequestException, Injectable } from '@nestjs/common';\nimport { FeatureFlagsService } from '@novu/application-generic';\nimport { MessageEntity, MessageRepository, OrganizationEntity, SubscriberEntity } from '@novu/dal';\nimport { ActorTypeEnum, FeatureFlagsKeysEnum } from '@novu/shared';\nimport { GetSubscriber, GetSubscriberCommand } from '../../../subscribers/usecases/get-subscriber';\nimport { GetMessagesCommand } from './get-messages.command';\n\n@Injectable()\nexport class GetMessages {\n constructor(\n private messageRepository: MessageRepository,\n private getSubscriberUseCase: GetSubscriber,\n private featureFlagService: FeatureFlagsService\n ) {}\n\n async execute(command: GetMessagesCommand) {\n const LIMIT = command.limit;\n const COUNT_LIMIT = 1000;\n\n if (LIMIT > 1000) {\n throw new BadRequestException('Limit can not be larger then 1000');\n }\n\n const query: Partial> & {\n _environmentId: string;\n transactionId?: string[];\n contextKeys?: string[];\n } = {\n _environmentId: command.environmentId,\n };\n\n if (command.subscriberId) {\n const subscriber = await this.getSubscriberUseCase.execute(\n GetSubscriberCommand.create({\n subscriberId: command.subscriberId,\n environmentId: command.environmentId,\n organizationId: command.organizationId,\n })\n );\n\n query._subscriberId = subscriber._id;\n }\n\n if (command.channel) {\n query.channel = command.channel;\n }\n\n if (command.transactionIds) {\n query.transactionId = command.transactionIds;\n }\n\n if (command.contextKeys) {\n query.contextKeys = command.contextKeys;\n }\n\n const data = await this.messageRepository.getMessages(query, '', {\n limit: LIMIT,\n sort: { createdAt: -1 },\n skip: command.page * LIMIT,\n });\n\n for (const message of data) {\n if (message._actorId && message.actor?.type === ActorTypeEnum.USER) {\n message.actor.data = this.processUserAvatar(message.actorSubscriber);\n }\n }\n\n const isEnabled = await this.featureFlagService.getFlag({\n key: FeatureFlagsKeysEnum.IS_NEW_MESSAGES_API_RESPONSE_ENABLED,\n organization: { _id: command.organizationId } as OrganizationEntity,\n defaultValue: false,\n });\n\n if (isEnabled) {\n return {\n hasMore: data?.length === command.limit,\n page: command.page,\n pageSize: LIMIT,\n data,\n };\n }\n\n const totalCount = await this.messageRepository.count(query);\n\n const hasMore = this.getHasMore(command.page, LIMIT, data.length, totalCount);\n\n return {\n page: command.page,\n totalCount,\n hasMore,\n pageSize: LIMIT,\n data,\n };\n }\n\n private getHasMore(page: number, limit: number, feedLength: number, totalCount: number) {\n const currentPaginationTotal = page * limit + feedLength;\n\n return currentPaginationTotal < totalCount;\n }\n\n private processUserAvatar(actorSubscriber?: SubscriberEntity): string | null {\n return actorSubscriber?.avatar || null;\n }\n}\n"
},
{
"path": "apps/api/src/app/messages/usecases/get-messages/index.ts",
"content": "export * from './get-messages.command';\nexport * from './get-messages.usecase';\n"
},
{
"path": "apps/api/src/app/messages/usecases/index.ts",
"content": "import { GetMessages } from './get-messages';\nimport { RemoveMessage } from './remove-message';\nimport { RemoveMessagesByTransactionId } from './remove-messages-by-transactionId/remove-messages-by-transactionId.usecase';\n\nexport const USE_CASES = [RemoveMessage, GetMessages, RemoveMessagesByTransactionId];\n"
},
{
"path": "apps/api/src/app/messages/usecases/remove-message/index.ts",
"content": "export * from './remove-message.command';\nexport * from './remove-message.usecase';\n"
},
{
"path": "apps/api/src/app/messages/usecases/remove-message/remove-message.command.ts",
"content": "import { IsString } from 'class-validator';\nimport { EnvironmentCommand } from '../../../shared/commands/project.command';\n\nexport class RemoveMessageCommand extends EnvironmentCommand {\n @IsString()\n messageId: string;\n}\n"
},
{
"path": "apps/api/src/app/messages/usecases/remove-message/remove-message.usecase.ts",
"content": "import { BadRequestException, Injectable, NotFoundException } from '@nestjs/common';\nimport { buildFeedKey, buildMessageCountKey, InvalidateCacheService } from '@novu/application-generic';\nimport { MessageRepository } from '@novu/dal';\n\nimport { RemoveMessageCommand } from './remove-message.command';\n\n@Injectable()\nexport class RemoveMessage {\n constructor(\n private invalidateCache: InvalidateCacheService,\n private messageRepository: MessageRepository\n ) {}\n\n async execute(command: RemoveMessageCommand) {\n const message = await this.messageRepository.findMessageById({\n _environmentId: command.environmentId,\n _id: command.messageId,\n });\n if (!message) {\n throw new NotFoundException(`Message with id ${command.messageId} not found`);\n }\n\n if (!message.subscriber)\n throw new BadRequestException(`A subscriber was not found for message ${command.messageId}`);\n\n await this.invalidateCache.invalidateQuery({\n key: buildMessageCountKey().invalidate({\n subscriberId: message.subscriber.subscriberId,\n _environmentId: command.environmentId,\n }),\n });\n\n await this.messageRepository.delete({\n _environmentId: command.environmentId,\n _id: command.messageId,\n });\n\n return {\n acknowledged: true,\n status: 'deleted',\n };\n }\n}\n"
},
{
"path": "apps/api/src/app/messages/usecases/remove-messages-by-transactionId/remove-messages-by-transactionId.command.ts",
"content": "import { ChannelTypeEnum } from '@novu/shared';\nimport { IsEnum, IsOptional, IsString } from 'class-validator';\nimport { EnvironmentCommand } from '../../../shared/commands/project.command';\n\nexport class RemoveMessagesByTransactionIdCommand extends EnvironmentCommand {\n @IsString()\n transactionId: string;\n\n @IsEnum(ChannelTypeEnum)\n @IsOptional()\n channel?: ChannelTypeEnum;\n}\n"
},
{
"path": "apps/api/src/app/messages/usecases/remove-messages-by-transactionId/remove-messages-by-transactionId.usecase.ts",
"content": "import { Injectable, NotFoundException } from '@nestjs/common';\n\nimport { buildFeedKey, buildMessageCountKey, InvalidateCacheService } from '@novu/application-generic';\nimport { EnforceEnvId, MessageEntity, MessageRepository } from '@novu/dal';\n\nimport { RemoveMessagesByTransactionIdCommand } from './remove-messages-by-transactionId.command';\n\n@Injectable()\nexport class RemoveMessagesByTransactionId {\n constructor(\n private messageRepository: MessageRepository,\n private invalidateCache: InvalidateCacheService\n ) {}\n\n async execute(command: RemoveMessagesByTransactionIdCommand) {\n const messages = await this.messageRepository.findMessagesByTransactionId({\n transactionId: [command.transactionId],\n _environmentId: command.environmentId,\n _organizationId: command.organizationId,\n ...(command.channel && { channel: command.channel }),\n });\n\n if (messages.length === 0) {\n throw new NotFoundException('Invalid transactionId or channel');\n }\n\n for (const message of messages) {\n const subscriberId = message.subscriber?.subscriberId;\n if (subscriberId) {\n await this.invalidateCache.invalidateQuery({\n key: buildMessageCountKey().invalidate({\n subscriberId,\n _environmentId: command.environmentId,\n }),\n });\n }\n }\n\n const deleteQuery: Partial & EnforceEnvId = {\n transactionId: command.transactionId,\n _environmentId: command.environmentId,\n _organizationId: command.organizationId,\n };\n\n if (command.channel) {\n deleteQuery.channel = command.channel;\n }\n\n await this.messageRepository.delete(deleteQuery);\n }\n}\n"
},
{
"path": "apps/api/src/app/notification-groups/dtos/create-notification-group-request.dto.ts",
"content": "import { ApiProperty } from '@nestjs/swagger';\nimport { IsDefined, IsString } from 'class-validator';\n\nexport class CreateNotificationGroupRequestDto {\n @ApiProperty()\n @IsString()\n @IsDefined()\n name: string;\n}\n"
},
{
"path": "apps/api/src/app/notification-groups/dtos/delete-notification-group-response.dto.ts",
"content": "import { ApiProperty } from '@nestjs/swagger';\nimport { IsBoolean, IsDefined, IsString } from 'class-validator';\n\nexport class DeleteNotificationGroupResponseDto {\n @ApiProperty({\n description: 'A boolean stating the success of the action',\n })\n @IsBoolean()\n @IsDefined()\n acknowledged: boolean;\n\n @ApiProperty({\n description: 'The status enum for the performed action',\n enum: ['deleted'],\n })\n @IsString()\n @IsDefined()\n status: string;\n}\n"
},
{
"path": "apps/api/src/app/notification-groups/dtos/notification-group-response.dto.ts",
"content": "import { ApiProperty, ApiPropertyOptional } from '@nestjs/swagger';\n\nexport class NotificationGroupResponseDto {\n @ApiPropertyOptional()\n _id?: string;\n\n @ApiProperty()\n name: string;\n\n @ApiProperty()\n _environmentId: string;\n\n @ApiProperty()\n _organizationId: string;\n\n @ApiPropertyOptional()\n _parentId?: string;\n}\n"
},
{
"path": "apps/api/src/app/notification-groups/e2e/create-notification-group.e2e.ts",
"content": "import { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\n\ndescribe('Create Notification Group - /notification-groups (POST) #novu-v0', async () => {\n let session: UserSession;\n\n before(async () => {\n session = new UserSession();\n await session.initialize();\n });\n\n it('should create notification group', async () => {\n const testTemplate = {\n name: 'Test name',\n };\n\n const { body } = await session.testAgent.post(`/v1/notification-groups`).send(testTemplate);\n\n expect(body.data).to.be.ok;\n const group = body.data;\n\n expect(group.name).to.equal(`Test name`);\n expect(group._environmentId).to.equal(session.environment._id);\n });\n});\n"
},
{
"path": "apps/api/src/app/notification-groups/e2e/delete-notification-group.e2e.ts",
"content": "import { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\n\ndescribe('Delete Notification Group - /notification-groups/:id (DELETE) #novu-v0', async () => {\n let session: UserSession;\n\n beforeEach(async () => {\n session = new UserSession();\n await session.initialize();\n });\n\n it('should delete notification group by id', async () => {\n const postNotificationGroup1 = await session.testAgent.post(`/v1/notification-groups`).send({\n name: 'Test delete group',\n });\n\n const { id } = postNotificationGroup1.body.data;\n\n const getResult = await session.testAgent.get(`/v1/notification-groups/${id}`);\n\n const group = getResult.body.data;\n\n expect(group.name).to.equal(`Test delete group`);\n expect(group._id).to.equal(postNotificationGroup1.body.data.id);\n expect(group._environmentId).to.equal(session.environment._id);\n\n const { body: deleteResult } = await session.testAgent.delete(`/v1/notification-groups/${id}`);\n\n expect(deleteResult.data.acknowledged).to.equal(true);\n expect(deleteResult.data.status).to.equal('deleted');\n\n const { body: getResultAfterDelete } = await session.testAgent.get(`/v1/notification-groups/${id}`);\n\n expect(getResultAfterDelete.statusCode).to.eq(404);\n });\n\n it('should return 404 error when attempting to delete non-existent notification group', async () => {\n const postNotificationGroup1 = await session.testAgent.post(`/v1/notification-groups`).send({\n name: 'Test name',\n });\n\n const { id } = postNotificationGroup1.body.data;\n\n await session.testAgent.delete(`/v1/notification-groups/${id}`);\n\n const { body } = await session.testAgent.delete(`/v1/notification-groups/${id}`);\n\n expect(body.statusCode).to.equal(404);\n });\n});\n"
},
{
"path": "apps/api/src/app/notification-groups/e2e/get-notification-group.e2e.ts",
"content": "import { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\n\ndescribe('Get Notification Group - /notification-groups/:id (GET) #novu-v0', async () => {\n let session: UserSession;\n\n const testTemplate = {\n name: 'Test name',\n };\n\n beforeEach(async () => {\n session = new UserSession();\n await session.initialize();\n });\n\n it('should get the notification group by id', async () => {\n const postNotificationGroup1 = await session.testAgent.post(`/v1/notification-groups`).send(testTemplate);\n\n const { id } = postNotificationGroup1.body.data;\n\n const { body } = await session.testAgent.get(`/v1/notification-groups/${id}`);\n\n const group = body.data;\n\n expect(group.name).to.equal(`Test name`);\n expect(group._id).to.equal(postNotificationGroup1.body.data.id);\n expect(group._environmentId).to.equal(session.environment._id);\n });\n\n it('should get 404 when notification group is not present with the requested id', async () => {\n const postNotificationGroup1 = await session.testAgent.post(`/v1/notification-groups`).send(testTemplate);\n\n const { id } = postNotificationGroup1.body.data;\n\n await session.testAgent.delete(`/v1/notification-groups/${id}`);\n\n const { body } = await session.testAgent.get(`/v1/notification-groups/${id}`);\n\n expect(body.statusCode).to.equal(404);\n });\n});\n"
},
{
"path": "apps/api/src/app/notification-groups/e2e/get-notification-groups.e2e.ts",
"content": "import { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\n\ndescribe('Get Notification Groups - /notification-groups (GET) #novu-v0', async () => {\n let session: UserSession;\n\n beforeEach(async () => {\n session = new UserSession();\n await session.initialize();\n });\n\n it('should get all notification groups', async () => {\n await session.testAgent.post(`/v1/notification-groups`).send({\n name: 'Test name',\n });\n await session.testAgent.post(`/v1/notification-groups`).send({\n name: 'Test name 2',\n });\n\n const { body } = await session.testAgent.get(`/v1/notification-groups`);\n\n expect(body.data.length).to.equal(3);\n const group = body.data.find((i) => i.name === 'Test name');\n\n expect(group.name).to.equal(`Test name`);\n expect(group._environmentId).to.equal(session.environment._id);\n });\n});\n"
},
{
"path": "apps/api/src/app/notification-groups/e2e/update-notification-group.e2e.ts",
"content": "import { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\n\ndescribe('Update Notification Group - /notification-groups/:id (PATCH) #novu-v0', async () => {\n let session: UserSession;\n\n beforeEach(async () => {\n session = new UserSession();\n await session.initialize();\n });\n\n it('update the notification group by id', async () => {\n const postNotificationGroup = await session.testAgent.post(`/v1/notification-groups`).send({\n name: 'Test name 1',\n });\n\n const { id } = postNotificationGroup.body.data;\n\n const { body: getNotificationGroupResult } = await session.testAgent.get(`/v1/notification-groups/${id}`);\n\n expect(getNotificationGroupResult.data.name).to.equal(`Test name 1`);\n expect(getNotificationGroupResult.data._id).to.equal(postNotificationGroup.body.data.id);\n expect(getNotificationGroupResult.data._environmentId).to.equal(session.environment._id);\n\n const { body: putNotificationGroup } = await session.testAgent.patch(`/v1/notification-groups/${id}`).send({\n name: 'Updated name',\n });\n\n expect(putNotificationGroup.data._id).to.equal(id);\n\n const { body: getUpdatedNotificationGroupResult } = await session.testAgent.get(`/v1/notification-groups/${id}`);\n\n expect(getUpdatedNotificationGroupResult.data.name).to.equal(`Updated name`);\n expect(getUpdatedNotificationGroupResult.data.id).to.equal(id);\n expect(getUpdatedNotificationGroupResult.data._environmentId).to.equal(session.environment._id);\n });\n\n it('should return a 404 error if the notification group to be updated does not exist', async () => {\n const postNotificationGroup1 = await session.testAgent.post(`/v1/notification-groups`).send({\n name: 'Test name',\n });\n\n const { id } = postNotificationGroup1.body.data;\n\n await session.testAgent.delete(`/v1/notification-groups/${id}`);\n\n const { body } = await session.testAgent.patch(`/v1/notification-groups/${id}`).send({\n name: 'Updated name',\n });\n\n expect(body.statusCode).to.equal(404);\n });\n});\n"
},
{
"path": "apps/api/src/app/notification-groups/notification-groups.controller.ts",
"content": "import {\n Body,\n ClassSerializerInterceptor,\n Controller,\n Delete,\n Get,\n Param,\n Patch,\n Post,\n UseInterceptors,\n} from '@nestjs/common';\nimport { ApiExcludeController, ApiOperation, ApiTags } from '@nestjs/swagger';\nimport { RequirePermissions } from '@novu/application-generic';\nimport { PermissionsEnum, UserSessionData } from '@novu/shared';\nimport { RequireAuthentication } from '../auth/framework/auth.decorator';\nimport { ExternalApiAccessible } from '../auth/framework/external-api.decorator';\nimport { ApiCommonResponses, ApiResponse } from '../shared/framework/response.decorator';\nimport { UserSession } from '../shared/framework/user.decorator';\nimport { CreateNotificationGroupRequestDto } from './dtos/create-notification-group-request.dto';\nimport { DeleteNotificationGroupResponseDto } from './dtos/delete-notification-group-response.dto';\nimport { NotificationGroupResponseDto } from './dtos/notification-group-response.dto';\nimport { CreateNotificationGroupCommand } from './usecases/create-notification-group/create-notification-group.command';\nimport { CreateNotificationGroup } from './usecases/create-notification-group/create-notification-group.usecase';\nimport { DeleteNotificationGroupCommand } from './usecases/delete-notification-group/delete-notification-group.command';\nimport { DeleteNotificationGroup } from './usecases/delete-notification-group/delete-notification-group.usecase';\nimport { GetNotificationGroupCommand } from './usecases/get-notification-group/get-notification-group.command';\nimport { GetNotificationGroup } from './usecases/get-notification-group/get-notification-group.usecase';\nimport { GetNotificationGroupsCommand } from './usecases/get-notification-groups/get-notification-groups.command';\nimport { GetNotificationGroups } from './usecases/get-notification-groups/get-notification-groups.usecase';\nimport { UpdateNotificationGroupCommand } from './usecases/update-notification-group/update-notification-group.command';\nimport { UpdateNotificationGroup } from './usecases/update-notification-group/update-notification-group.usecase';\n\n@ApiCommonResponses()\n@Controller('/notification-groups')\n@UseInterceptors(ClassSerializerInterceptor)\n@RequireAuthentication()\n@ApiTags('Workflow groups')\n@ApiExcludeController()\nexport class NotificationGroupsController {\n constructor(\n private createNotificationGroupUsecase: CreateNotificationGroup,\n private getNotificationGroupsUsecase: GetNotificationGroups,\n private getNotificationGroupUsecase: GetNotificationGroup,\n private deleteNotificationGroupUsecase: DeleteNotificationGroup,\n private updateNotificationGroupUsecase: UpdateNotificationGroup\n ) {}\n\n @Post('')\n @ExternalApiAccessible()\n @ApiResponse(NotificationGroupResponseDto, 201)\n @ApiOperation({\n summary: 'Create workflow group',\n description: `workflow group was previously named notification group`,\n })\n @RequirePermissions(PermissionsEnum.WORKFLOW_WRITE)\n createNotificationGroup(\n @UserSession() user: UserSessionData,\n @Body() body: CreateNotificationGroupRequestDto\n ): Promise {\n return this.createNotificationGroupUsecase.execute(\n CreateNotificationGroupCommand.create({\n organizationId: user.organizationId,\n userId: user._id,\n environmentId: user.environmentId,\n name: body.name,\n })\n );\n }\n\n @Get('')\n @ExternalApiAccessible()\n @ApiResponse(NotificationGroupResponseDto, 200, true)\n @ApiOperation({\n summary: 'Get workflow groups',\n description: `workflow group was previously named notification group`,\n })\n @RequirePermissions(PermissionsEnum.WORKFLOW_READ)\n listNotificationGroups(@UserSession() user: UserSessionData): Promise {\n return this.getNotificationGroupsUsecase.execute(\n GetNotificationGroupsCommand.create({\n organizationId: user.organizationId,\n userId: user._id,\n environmentId: user.environmentId,\n })\n );\n }\n\n @Get('/:id')\n @ExternalApiAccessible()\n @ApiResponse(NotificationGroupResponseDto, 200)\n @ApiOperation({\n summary: 'Get workflow group',\n description: `workflow group was previously named notification group`,\n })\n @RequirePermissions(PermissionsEnum.WORKFLOW_READ)\n getNotificationGroup(\n @UserSession() user: UserSessionData,\n @Param('id') id: string\n ): Promise {\n return this.getNotificationGroupUsecase.execute(\n GetNotificationGroupCommand.create({\n environmentId: user.environmentId,\n organizationId: user.organizationId,\n userId: user._id,\n id,\n })\n );\n }\n\n @Patch('/:id')\n @ExternalApiAccessible()\n @ApiResponse(NotificationGroupResponseDto, 200)\n @ApiOperation({\n summary: 'Update workflow group',\n description: `workflow group was previously named notification group`,\n })\n @RequirePermissions(PermissionsEnum.WORKFLOW_WRITE)\n updateNotificationGroup(\n @UserSession() user: UserSessionData,\n @Param('id') id: string,\n @Body() body: CreateNotificationGroupRequestDto\n ): Promise {\n return this.updateNotificationGroupUsecase.execute(\n UpdateNotificationGroupCommand.create({\n organizationId: user.organizationId,\n userId: user._id,\n environmentId: user.environmentId,\n name: body.name,\n id,\n })\n );\n }\n\n @Delete('/:id')\n @ExternalApiAccessible()\n @ApiResponse(DeleteNotificationGroupResponseDto, 200)\n @ApiOperation({\n summary: 'Delete workflow group',\n description: `workflow group was previously named notification group`,\n })\n @RequirePermissions(PermissionsEnum.WORKFLOW_WRITE)\n deleteNotificationGroup(\n @UserSession() user: UserSessionData,\n @Param('id') id: string\n ): Promise {\n return this.deleteNotificationGroupUsecase.execute(\n DeleteNotificationGroupCommand.create({\n environmentId: user.environmentId,\n organizationId: user.organizationId,\n userId: user._id,\n id,\n })\n );\n }\n}\n"
},
{
"path": "apps/api/src/app/notification-groups/notification-groups.module.ts",
"content": "import { forwardRef, Module } from '@nestjs/common';\nimport { AuthModule } from '../auth/auth.module';\nimport { ChangeModule } from '../change/change.module';\nimport { SharedModule } from '../shared/shared.module';\nimport { NotificationGroupsController } from './notification-groups.controller';\nimport { USE_CASES } from './usecases';\n\n@Module({\n imports: [SharedModule, forwardRef(() => AuthModule), ChangeModule],\n providers: [...USE_CASES],\n controllers: [NotificationGroupsController],\n exports: [...USE_CASES],\n})\nexport class NotificationGroupsModule {}\n"
},
{
"path": "apps/api/src/app/notification-groups/usecases/create-notification-group/create-notification-group.command.ts",
"content": "import { IsString } from 'class-validator';\nimport { EnvironmentWithUserCommand } from '../../../shared/commands/project.command';\n\nexport class CreateNotificationGroupCommand extends EnvironmentWithUserCommand {\n @IsString()\n name: string;\n}\n"
},
{
"path": "apps/api/src/app/notification-groups/usecases/create-notification-group/create-notification-group.usecase.ts",
"content": "import { Injectable } from '@nestjs/common';\nimport { CreateChange, CreateChangeCommand } from '@novu/application-generic';\nimport { NotificationGroupEntity, NotificationGroupRepository } from '@novu/dal';\nimport { ChangeEntityTypeEnum } from '@novu/shared';\n\nimport { CreateNotificationGroupCommand } from './create-notification-group.command';\n\n@Injectable()\nexport class CreateNotificationGroup {\n constructor(\n private notificationGroupRepository: NotificationGroupRepository,\n private createChange: CreateChange\n ) {}\n\n async execute(command: CreateNotificationGroupCommand): Promise {\n const group = await this.notificationGroupRepository.findOne({\n _organizationId: command.organizationId,\n });\n\n const item = await this.notificationGroupRepository.create({\n _environmentId: command.environmentId,\n _organizationId: command.organizationId,\n name: command.name,\n _parentId: group?._id,\n });\n\n await this.createChange.execute(\n CreateChangeCommand.create({\n item,\n environmentId: command.environmentId,\n organizationId: command.organizationId,\n userId: command.userId,\n type: ChangeEntityTypeEnum.NOTIFICATION_GROUP,\n changeId: NotificationGroupRepository.createObjectId(),\n })\n );\n\n return item;\n }\n}\n"
},
{
"path": "apps/api/src/app/notification-groups/usecases/delete-notification-group/delete-notification-group.command.ts",
"content": "import { IsDefined, IsString } from 'class-validator';\nimport { EnvironmentWithUserCommand } from '../../../shared/commands/project.command';\n\nexport class DeleteNotificationGroupCommand extends EnvironmentWithUserCommand {\n @IsString()\n @IsDefined()\n id: string;\n}\n"
},
{
"path": "apps/api/src/app/notification-groups/usecases/delete-notification-group/delete-notification-group.usecase.ts",
"content": "import { BadRequestException, Injectable, NotFoundException } from '@nestjs/common';\nimport { DalException, NotificationGroupRepository } from '@novu/dal';\nimport { DeleteNotificationGroupCommand } from './delete-notification-group.command';\n\n@Injectable()\nexport class DeleteNotificationGroup {\n constructor(private notificationGroupRepository: NotificationGroupRepository) {}\n\n async execute(command: DeleteNotificationGroupCommand) {\n const { environmentId, id } = command;\n try {\n const group = await this.notificationGroupRepository.findOne({\n _environmentId: environmentId,\n _id: id,\n });\n\n if (group === null) throw new NotFoundException();\n\n await this.notificationGroupRepository.delete({\n _environmentId: environmentId,\n _id: id,\n });\n } catch (e) {\n if (e instanceof DalException) {\n throw new BadRequestException(e.message);\n }\n throw e;\n }\n\n return {\n acknowledged: true,\n status: 'deleted',\n };\n }\n}\n"
},
{
"path": "apps/api/src/app/notification-groups/usecases/get-notification-group/get-notification-group.command.ts",
"content": "import { IsDefined, IsString } from 'class-validator';\nimport { EnvironmentWithUserCommand } from '../../../shared/commands/project.command';\n\nexport class GetNotificationGroupCommand extends EnvironmentWithUserCommand {\n @IsString()\n @IsDefined()\n id: string;\n}\n"
},
{
"path": "apps/api/src/app/notification-groups/usecases/get-notification-group/get-notification-group.usecase.ts",
"content": "import { Injectable, NotFoundException } from '@nestjs/common';\nimport { NotificationGroupEntity, NotificationGroupRepository } from '@novu/dal';\nimport { GetNotificationGroupCommand } from './get-notification-group.command';\n\n@Injectable()\nexport class GetNotificationGroup {\n constructor(private notificationGroupRepository: NotificationGroupRepository) {}\n\n async execute(command: GetNotificationGroupCommand): Promise {\n const { id, environmentId } = command;\n\n const result = await this.notificationGroupRepository.findOne({\n _environmentId: environmentId,\n _id: id,\n });\n\n if (result === null) throw new NotFoundException();\n\n return result;\n }\n}\n"
},
{
"path": "apps/api/src/app/notification-groups/usecases/get-notification-groups/get-notification-groups.command.ts",
"content": "import { EnvironmentWithUserCommand } from '../../../shared/commands/project.command';\n\nexport class GetNotificationGroupsCommand extends EnvironmentWithUserCommand {}\n"
},
{
"path": "apps/api/src/app/notification-groups/usecases/get-notification-groups/get-notification-groups.usecase.ts",
"content": "import { Injectable } from '@nestjs/common';\nimport { NotificationGroupEntity, NotificationGroupRepository } from '@novu/dal';\nimport { GetNotificationGroupsCommand } from './get-notification-groups.command';\n\n@Injectable()\nexport class GetNotificationGroups {\n constructor(private notificationGroupRepository: NotificationGroupRepository) {}\n\n async execute(command: GetNotificationGroupsCommand): Promise {\n return await this.notificationGroupRepository.find({\n _environmentId: command.environmentId,\n });\n }\n}\n"
},
{
"path": "apps/api/src/app/notification-groups/usecases/index.ts",
"content": "import { CreateNotificationGroup } from './create-notification-group/create-notification-group.usecase';\nimport { DeleteNotificationGroup } from './delete-notification-group/delete-notification-group.usecase';\nimport { GetNotificationGroup } from './get-notification-group/get-notification-group.usecase';\nimport { GetNotificationGroups } from './get-notification-groups/get-notification-groups.usecase';\nimport { UpdateNotificationGroup } from './update-notification-group/update-notification-group.usecase';\n\nexport const USE_CASES = [\n GetNotificationGroups,\n CreateNotificationGroup,\n GetNotificationGroup,\n DeleteNotificationGroup,\n UpdateNotificationGroup,\n];\n"
},
{
"path": "apps/api/src/app/notification-groups/usecases/update-notification-group/update-notification-group.command.ts",
"content": "import { IsDefined, IsString } from 'class-validator';\nimport { EnvironmentWithUserCommand } from '../../../shared/commands/project.command';\n\nexport class UpdateNotificationGroupCommand extends EnvironmentWithUserCommand {\n @IsString()\n @IsDefined()\n id: string;\n\n @IsString()\n @IsDefined()\n name: string;\n}\n"
},
{
"path": "apps/api/src/app/notification-groups/usecases/update-notification-group/update-notification-group.usecase.ts",
"content": "import { Injectable, NotFoundException } from '@nestjs/common';\nimport { NotificationGroupRepository } from '@novu/dal';\nimport { GetNotificationGroup } from '../get-notification-group/get-notification-group.usecase';\nimport { UpdateNotificationGroupCommand } from './update-notification-group.command';\n\n@Injectable()\nexport class UpdateNotificationGroup {\n constructor(\n private notificationGroupRepository: NotificationGroupRepository,\n private getNotificationGroup: GetNotificationGroup\n ) {}\n\n async execute(command: UpdateNotificationGroupCommand) {\n const { id, environmentId, name, organizationId, userId } = command;\n\n const item = await this.getNotificationGroup.execute({\n environmentId,\n organizationId,\n userId,\n id,\n });\n\n const result = await this.notificationGroupRepository.update(\n {\n _id: item._id,\n _environmentId: item._environmentId,\n },\n {\n $set: {\n name,\n },\n }\n );\n\n if (result.matched === 0) {\n throw new NotFoundException();\n }\n\n return await this.getNotificationGroup.execute({\n environmentId,\n organizationId,\n userId,\n id,\n });\n }\n}\n"
},
{
"path": "apps/api/src/app/notifications/dtos/activities-request.dto.ts",
"content": "import { ApiPropertyOptional } from '@nestjs/swagger';\nimport { ChannelTypeEnum, SeverityLevelEnum } from '@novu/shared';\nimport { Transform, Type } from 'class-transformer';\nimport { IsArray, IsInt, IsOptional, IsString, Max, Min } from 'class-validator';\nimport { IsEnumOrArray } from '../../shared/validators/is-enum-or-array';\n\nexport class ActivitiesRequestDto {\n @ApiPropertyOptional({\n enum: [...Object.values(ChannelTypeEnum)],\n enumName: 'ChannelTypeEnum',\n isArray: true,\n description: 'Array of channel types',\n })\n @IsOptional()\n channels?: ChannelTypeEnum[] | ChannelTypeEnum;\n\n @ApiPropertyOptional({\n type: String,\n isArray: true,\n description: 'Array of template IDs or a single template ID',\n })\n @IsOptional()\n templates?: string[] | string;\n\n @ApiPropertyOptional({\n type: String,\n isArray: true,\n description: 'Array of email addresses or a single email address',\n })\n @IsOptional()\n emails?: string | string[];\n\n @ApiPropertyOptional({\n type: String,\n deprecated: true,\n description: 'Search term (deprecated)',\n })\n @IsOptional()\n search?: string;\n\n @ApiPropertyOptional({\n type: String,\n isArray: true,\n description: 'Array of subscriber IDs or a single subscriber ID',\n })\n @IsOptional()\n subscriberIds?: string | string[];\n\n @ApiPropertyOptional({\n type: String,\n isArray: true,\n description: 'Array of severity levels or a single severity level',\n })\n @IsOptional()\n @IsEnumOrArray(SeverityLevelEnum)\n severity?: SeverityLevelEnum[] | SeverityLevelEnum;\n\n @ApiPropertyOptional({\n type: Number,\n default: 0,\n description: 'Page number for pagination',\n })\n @IsOptional()\n @Type(() => Number)\n @IsInt()\n @Min(0)\n page: number = 0;\n\n @ApiPropertyOptional({\n type: Number,\n default: 10,\n minimum: 1,\n maximum: 50,\n description: 'Limit for pagination',\n })\n @IsOptional()\n @Type(() => Number)\n @IsInt()\n @Min(1)\n @Max(50)\n limit: number = 10;\n\n @ApiPropertyOptional({\n type: String,\n description: 'The transaction ID to filter by',\n })\n @IsOptional()\n transactionId?: string[] | string;\n\n @ApiPropertyOptional({\n type: String,\n description: 'Topic Key for filtering notifications by topic',\n })\n @IsOptional()\n @IsString()\n topicKey?: string;\n\n @ApiPropertyOptional({\n type: String,\n description: 'Subscription ID for filtering notifications by subscription',\n })\n @IsOptional()\n @IsString()\n subscriptionId?: string;\n\n @ApiPropertyOptional({\n type: String,\n isArray: true,\n description: 'Filter by exact context keys, order insensitive (format: \"type:id\")',\n })\n @IsOptional()\n @Transform(({ value }) => {\n // No parameter = no filter\n if (value === undefined) return undefined;\n\n // Empty string = filter for records with no context\n if (value === '') return [];\n\n // Normalize to array and remove empty strings\n const array = Array.isArray(value) ? value : [value];\n return array.filter((v) => v !== '');\n })\n @IsArray()\n @IsString({ each: true })\n contextKeys?: string[];\n\n @ApiPropertyOptional({\n type: String,\n description: 'Date filter for records after this timestamp. Defaults to earliest date allowed by subscription plan',\n })\n @IsOptional()\n after?: string;\n\n @ApiPropertyOptional({\n type: String,\n description: 'Date filter for records before this timestamp. Defaults to current time of request (now)',\n })\n @IsOptional()\n before?: string;\n}\n"
},
{
"path": "apps/api/src/app/notifications/dtos/activities-response.dto.ts",
"content": "import { ApiProperty, ApiPropertyOptional } from '@nestjs/swagger';\nimport { StepFilterDto } from '@novu/application-generic';\nimport {\n DaysEnum,\n DigestTypeEnum,\n DigestUnitEnum,\n ExecutionDetailsSourceEnum,\n ExecutionDetailsStatusEnum,\n MessageTemplateDto,\n MonthlyTypeEnum,\n OrdinalEnum,\n OrdinalValueEnum,\n ProvidersIdEnum,\n ProvidersIdEnumConst,\n ResourceOriginEnum,\n SeverityLevelEnum,\n StepTypeEnum,\n TriggerTypeEnum,\n} from '@novu/shared';\nimport { IsArray, IsEnum, IsNumber, IsOptional, IsString } from 'class-validator';\n\nexport class DigestTimedConfigDto {\n @ApiPropertyOptional({ description: 'Time at which the digest is triggered' })\n @IsOptional()\n @IsString()\n atTime?: string;\n\n @ApiPropertyOptional({\n description: 'Days of the week for the digest',\n type: 'array',\n items: {\n type: 'string',\n enum: Object.values(DaysEnum),\n },\n enumName: 'DaysEnum',\n })\n @IsOptional()\n @IsArray()\n @IsEnum(DaysEnum, { each: true })\n weekDays?: DaysEnum[];\n\n @ApiPropertyOptional({ description: 'Specific days of the month for the digest', type: [Number] })\n @IsOptional()\n @IsArray()\n @IsNumber({}, { each: true })\n monthDays?: number[];\n\n @ApiPropertyOptional({\n description: 'Ordinal position for the digest',\n enum: [...Object.values(OrdinalEnum)],\n enumName: 'OrdinalEnum',\n })\n @IsOptional()\n @IsEnum(OrdinalEnum)\n ordinal?: OrdinalEnum;\n\n @ApiPropertyOptional({\n description: 'Value of the ordinal',\n enum: [...Object.values(OrdinalValueEnum)],\n enumName: 'OrdinalValueEnum',\n })\n @IsOptional()\n @IsEnum(OrdinalValueEnum)\n ordinalValue?: OrdinalValueEnum;\n\n @ApiPropertyOptional({\n description: 'Type of monthly schedule',\n enum: [...Object.values(MonthlyTypeEnum)],\n enumName: 'MonthlyTypeEnum',\n })\n @IsOptional()\n @IsEnum(MonthlyTypeEnum)\n monthlyType?: MonthlyTypeEnum;\n\n @ApiPropertyOptional({ description: 'Cron expression for scheduling' })\n @IsOptional()\n @IsString()\n cronExpression?: string;\n\n @ApiPropertyOptional({ description: 'Until date for scheduling' })\n @IsOptional()\n @IsString()\n untilDate?: string;\n}\n\nexport class DigestMetadataDto {\n @ApiPropertyOptional({ description: 'Optional key for the digest' })\n digestKey?: string;\n\n @ApiPropertyOptional({ description: 'Amount for the digest', type: Number })\n amount?: number;\n\n @ApiPropertyOptional({ description: 'Unit of the digest', enum: DigestUnitEnum })\n unit?: DigestUnitEnum;\n\n @ApiProperty({\n enum: [...Object.values(DigestTypeEnum)],\n enumName: 'DigestTypeEnum',\n description: 'The Digest Type',\n type: String,\n })\n type: DigestTypeEnum;\n\n @ApiPropertyOptional({\n type: 'array',\n items: {\n type: 'object',\n additionalProperties: true,\n },\n description: 'Optional array of events associated with the digest, represented as key-value pairs',\n })\n events?: Record[];\n\n // Properties for Regular Digest\n @ApiPropertyOptional({\n description: 'Regular digest: Indicates if backoff is enabled for the regular digest',\n type: Boolean,\n })\n backoff?: boolean;\n\n @ApiPropertyOptional({ description: 'Regular digest: Amount for backoff', type: Number })\n backoffAmount?: number;\n\n @ApiPropertyOptional({\n description: 'Regular digest: Unit for backoff',\n enum: [...Object.values(DigestUnitEnum)],\n enumName: 'DigestUnitEnum',\n })\n backoffUnit?: DigestUnitEnum;\n\n @ApiPropertyOptional({ description: 'Regular digest: Indicates if the digest should update', type: Boolean })\n updateMode?: boolean;\n\n // Properties for Timed Digest\n @ApiPropertyOptional({ description: 'Configuration for timed digest', type: () => DigestTimedConfigDto })\n timed?: DigestTimedConfigDto;\n}\n\nexport class ActivityNotificationStepResponseDto {\n @ApiProperty({ description: 'Unique identifier of the step', type: String })\n _id: string;\n\n @ApiProperty({ description: 'Whether the step is active or not', type: Boolean })\n active: boolean;\n\n @ApiPropertyOptional({ description: 'Reply callback settings', type: Object })\n replyCallback?: {\n active: boolean;\n url: string;\n };\n\n @ApiPropertyOptional({ description: 'Control variables', type: Object })\n controlVariables?: Record;\n\n @ApiPropertyOptional({ description: 'Metadata for the workflow step', type: Object })\n metadata?: any; // Adjust the type based on your actual metadata structure\n\n @ApiPropertyOptional({ description: 'Step issues', type: Object })\n issues?: any; // Adjust the type based on your actual issues structure\n\n @ApiProperty({ description: 'Filter criteria for the step', isArray: true, type: StepFilterDto })\n filters: StepFilterDto[];\n\n @ApiPropertyOptional({ description: 'Optional template for the step', type: MessageTemplateDto })\n template?: MessageTemplateDto;\n\n @ApiPropertyOptional({ description: 'Variants of the step', type: [ActivityNotificationStepResponseDto] })\n variants?: ActivityNotificationStepResponseDto[]; // Assuming variants are the same type\n\n @ApiProperty({ description: 'The identifier for the template associated with this step', type: String })\n _templateId: string;\n\n @ApiPropertyOptional({ description: 'The name of the step', type: String })\n name?: string;\n\n @ApiPropertyOptional({ description: 'The unique identifier for the parent step', type: String })\n _parentId?: string | null;\n}\n// Activity Notification Execution Detail Response DTO\nexport class ActivityNotificationExecutionDetailResponseDto {\n @ApiProperty({ description: 'Unique identifier of the execution detail', type: String })\n _id: string;\n\n @ApiPropertyOptional({ description: 'Creation time of the execution detail', type: String })\n createdAt?: string;\n\n @ApiProperty({\n enum: [...Object.values(ExecutionDetailsStatusEnum)],\n enumName: 'ExecutionDetailsStatusEnum',\n description: 'Status of the execution detail',\n type: String,\n })\n status: ExecutionDetailsStatusEnum;\n\n @ApiProperty({ description: 'Detailed information about the execution', type: String })\n detail: string;\n\n @ApiProperty({ description: 'Whether the execution is a retry or not', type: Boolean })\n isRetry: boolean;\n\n @ApiProperty({ description: 'Whether the execution is a test or not', type: Boolean })\n isTest: boolean;\n\n @ApiPropertyOptional({\n enum: [...new Set([...Object.values(ProvidersIdEnumConst).flatMap((enumObj) => Object.values(enumObj))])],\n enumName: 'ProvidersIdEnum',\n description: 'Provider ID of the execution',\n type: String,\n })\n @IsString()\n @IsOptional()\n @IsEnum(ProvidersIdEnumConst)\n providerId?: ProvidersIdEnum;\n\n @ApiPropertyOptional({ description: 'Raw data of the execution', type: String })\n raw?: string | null;\n\n @ApiProperty({\n enum: [...Object.values(ExecutionDetailsSourceEnum)],\n enumName: 'ExecutionDetailsSourceEnum',\n description: 'Source of the execution detail',\n type: String,\n })\n @IsString()\n @IsEnum(ExecutionDetailsSourceEnum)\n source: ExecutionDetailsSourceEnum;\n}\n\n// Activity Notification Job Response DTO\nexport class ActivityNotificationJobResponseDto {\n @ApiProperty({ description: 'Unique identifier of the job', type: String })\n _id: string;\n\n @ApiProperty({ description: 'Type of the job', type: String })\n type: StepTypeEnum;\n\n @ApiPropertyOptional({\n description: 'Optional digest for the job, including metadata and events',\n type: DigestMetadataDto,\n })\n digest?: DigestMetadataDto;\n\n @ApiProperty({\n description: 'Execution details of the job',\n type: [ActivityNotificationExecutionDetailResponseDto],\n })\n executionDetails: ActivityNotificationExecutionDetailResponseDto[];\n\n @ApiProperty({\n description: 'Step details of the job',\n type: ActivityNotificationStepResponseDto,\n })\n step: ActivityNotificationStepResponseDto;\n\n @ApiPropertyOptional({\n description: 'Optional context object for additional error details.',\n type: 'object',\n required: false,\n additionalProperties: true,\n example: {\n workflowId: 'some_wf_id',\n stepId: 'some_wf_id',\n },\n })\n overrides?: Record;\n\n @ApiPropertyOptional({ description: 'Optional payload for the job', type: Object })\n payload?: Record;\n\n @ApiProperty({\n enum: [...new Set([...Object.values(ProvidersIdEnumConst).flatMap((enumObj) => Object.values(enumObj))])],\n enumName: 'ProvidersIdEnum',\n description: 'Provider ID of the job',\n type: String, // Explicit type reference for enum\n })\n providerId: ProvidersIdEnum;\n\n @ApiProperty({ description: 'Status of the job', type: String })\n status: string;\n\n @ApiPropertyOptional({ description: 'Updated time of the notification', type: String })\n updatedAt?: string;\n\n @ApiPropertyOptional({\n description: 'The number of times the digest/delay job has been extended to align with the subscribers schedule',\n type: Number,\n })\n scheduleExtensionsCount?: number;\n}\n\n// Activity Notification Subscriber Response DTO\nexport class ActivityNotificationSubscriberResponseDto {\n @ApiPropertyOptional({ description: 'First name of the subscriber', type: String })\n firstName?: string;\n\n @ApiProperty({ description: 'External unique identifier of the subscriber', type: String })\n subscriberId: string;\n\n @ApiProperty({ description: 'Internal to Novu unique identifier of the subscriber', type: String })\n _id: string;\n\n @ApiPropertyOptional({ description: 'Last name of the subscriber', type: String })\n lastName?: string;\n\n @ApiPropertyOptional({ description: 'Email address of the subscriber', type: String })\n email?: string;\n\n @ApiPropertyOptional({ description: 'Phone number of the subscriber', type: String })\n phone?: string;\n}\n\n// Notification Trigger Variable DTO\nexport class NotificationTriggerVariable {\n @ApiProperty({ description: 'Name of the variable', type: String })\n name: string;\n}\n\nexport class NotificationTriggerDto {\n @ApiProperty({\n enum: TriggerTypeEnum,\n description: 'Type of the trigger',\n type: String, // Explicit type reference for enum\n })\n type: TriggerTypeEnum;\n\n @ApiProperty({ description: 'Identifier of the trigger', type: String })\n identifier: string;\n\n @ApiProperty({\n description: 'Variables of the trigger',\n type: [NotificationTriggerVariable],\n })\n variables: NotificationTriggerVariable[];\n\n @ApiPropertyOptional({\n description: 'Subscriber variables of the trigger',\n type: [NotificationTriggerVariable],\n })\n subscriberVariables?: NotificationTriggerVariable[];\n}\n\n// Activity Notification Template Response DTO\nexport class ActivityNotificationTemplateResponseDto {\n @ApiPropertyOptional({ description: 'Unique identifier of the template', type: String })\n _id?: string;\n\n @ApiProperty({ description: 'Name of the template', type: String })\n name: string;\n\n @ApiProperty({\n enum: [...Object.values(ResourceOriginEnum)],\n enumName: 'ResourceOriginEnum',\n description: 'Origin of the workflow',\n type: String,\n })\n @IsString()\n @IsEnum(ResourceOriginEnum)\n origin?: ResourceOriginEnum;\n\n @ApiProperty({\n description: 'Triggers of the template',\n type: [NotificationTriggerDto],\n })\n triggers: NotificationTriggerDto[];\n}\n\nexport class ActivityTopicDto {\n @ApiProperty({ description: 'Internal Topic ID of the notification', type: String })\n _topicId: string;\n\n @ApiProperty({ description: 'Topic Key of the notification', type: String })\n topicKey: string;\n}\n\n// Activity Notification Response DTO\nexport class ActivityNotificationResponseDto {\n @ApiPropertyOptional({ description: 'Unique identifier of the notification', type: String })\n _id?: string;\n\n @ApiProperty({ description: 'Environment ID of the notification', type: String })\n _environmentId: string;\n\n @ApiProperty({ description: 'Organization ID of the notification', type: String })\n _organizationId: string;\n\n @ApiProperty({ description: 'Subscriber ID of the notification', type: String })\n _subscriberId: string; // Added to align with NotificationEntity\n\n @ApiProperty({ description: 'Transaction ID of the notification', type: String })\n transactionId: string;\n\n @ApiPropertyOptional({ description: 'Template ID of the notification', type: String })\n _templateId?: string; // Added to align with NotificationEntity\n\n @ApiPropertyOptional({ description: 'Digested Notification ID', type: String })\n _digestedNotificationId?: string; // Added to align with NotificationEntity\n\n @ApiPropertyOptional({ description: 'Creation time of the notification', type: String })\n createdAt?: string;\n\n @ApiPropertyOptional({ description: 'Last updated time of the notification', type: String })\n updatedAt?: string; // Added to align with NotificationEntity\n\n @ApiPropertyOptional({\n description: 'Channels of the notification',\n enum: [...Object.values(StepTypeEnum)],\n enumName: 'StepTypeEnum',\n isArray: true,\n type: String,\n })\n channels?: StepTypeEnum[];\n\n @ApiPropertyOptional({\n description: 'Subscriber of the notification',\n type: ActivityNotificationSubscriberResponseDto,\n })\n subscriber?: ActivityNotificationSubscriberResponseDto;\n\n @ApiPropertyOptional({\n description: 'Template of the notification',\n type: ActivityNotificationTemplateResponseDto,\n })\n template?: ActivityNotificationTemplateResponseDto;\n\n @ApiPropertyOptional({\n description: 'Jobs of the notification',\n type: [ActivityNotificationJobResponseDto],\n })\n jobs?: ActivityNotificationJobResponseDto[];\n\n @ApiPropertyOptional({\n description: 'Payload of the notification',\n type: 'object',\n required: false,\n additionalProperties: true,\n })\n payload?: Record; // Added to align with NotificationEntity\n\n @ApiPropertyOptional({\n description: 'Tags associated with the notification',\n type: [String],\n })\n tags?: string[]; // Added to align with NotificationEntity\n\n @ApiPropertyOptional({\n description: 'Controls associated with the notification',\n type: 'object',\n required: false,\n additionalProperties: true,\n })\n controls?: Record; // Added to align with NotificationEntity\n\n @ApiPropertyOptional({\n description: 'To field for subscriber definition',\n type: 'object',\n required: false,\n additionalProperties: true,\n })\n to?: Record; // Added to align with NotificationEntity\n\n @ApiPropertyOptional({ description: 'Topics of the notification', type: [ActivityTopicDto] })\n topics?: ActivityTopicDto[];\n\n @ApiPropertyOptional({\n description: 'Severity of the notification',\n enum: [...Object.values(SeverityLevelEnum)],\n enumName: 'SeverityLevelEnum',\n })\n severity: SeverityLevelEnum;\n\n @ApiPropertyOptional({ description: 'Criticality of the notification', type: Boolean })\n critical?: boolean;\n\n @ApiPropertyOptional({ description: 'Context (single or multi) in which the notification was sent', type: [String] })\n contextKeys?: string[];\n}\n\n// Activities Response DTO\nexport class ActivitiesResponseDto {\n @ApiProperty({ description: 'Indicates if there are more activities in the result set', type: Boolean })\n hasMore: boolean;\n\n @ApiProperty({\n description: 'Array of activity notifications',\n type: [ActivityNotificationResponseDto],\n })\n data: ActivityNotificationResponseDto[];\n\n @ApiProperty({ description: 'Page size of the activities', type: Number })\n pageSize: number;\n\n @ApiProperty({ description: 'Current page of the activities', type: Number })\n page: number;\n}\n"
},
{
"path": "apps/api/src/app/notifications/dtos/activity-graph-states-response.dto.ts",
"content": "import { ApiProperty } from '@nestjs/swagger';\nimport { ChannelTypeEnum } from '@novu/shared';\n\nexport class ActivityGraphStatesResponse {\n @ApiProperty()\n _id: string;\n\n @ApiProperty()\n count: number;\n\n @ApiProperty()\n templates: string[];\n\n @ApiProperty({\n enum: ChannelTypeEnum,\n isArray: true,\n })\n channels: ChannelTypeEnum[];\n}\n"
},
{
"path": "apps/api/src/app/notifications/dtos/activity-stats-response.dto.ts",
"content": "import { ApiProperty } from '@nestjs/swagger';\n\nexport class ActivityStatsResponseDto {\n @ApiProperty()\n weeklySent: number;\n\n @ApiProperty()\n monthlySent: number;\n}\n"
},
{
"path": "apps/api/src/app/notifications/e2e/get-activity-feed.e2e.ts",
"content": "import { Novu } from '@novu/api';\nimport { ActivityNotificationResponseDto, ChannelTypeEnum } from '@novu/api/models/components';\nimport { NotificationTemplateEntity, NotificationTemplateRepository, SubscriberRepository } from '@novu/dal';\nimport { CreateWorkflowDto, StepTypeEnum, WorkflowCreationSourceEnum, WorkflowResponseDto } from '@novu/shared';\nimport { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\nimport { initNovuClassSdk } from '../../shared/helpers/e2e/sdk/e2e-sdk.helper';\n\ndescribe('Get activity feed - /notifications (GET) #novu-v2', async () => {\n let session: UserSession;\n let template: NotificationTemplateEntity;\n let smsOnlyTemplate: NotificationTemplateEntity;\n let subscriberId: string;\n let novuClient: Novu;\n\n beforeEach(async () => {\n session = new UserSession();\n await session.initialize();\n template = await session.createTemplate();\n smsOnlyTemplate = await session.createChannelTemplate(StepTypeEnum.SMS);\n subscriberId = SubscriberRepository.createObjectId();\n novuClient = initNovuClassSdk(session);\n\n await session.testAgent\n .post('/v1/widgets/session/initialize')\n .send({\n applicationIdentifier: session.environment.identifier,\n subscriberId,\n firstName: 'Test',\n lastName: 'User',\n email: 'test@example.com',\n })\n .expect(201);\n });\n\n it('should get the current activity feed of user', async () => {\n await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: subscriberId,\n payload: { firstName: 'Test' },\n });\n\n await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: subscriberId,\n payload: { firstName: 'Test' },\n });\n\n await session.waitForJobCompletion(template._id);\n const body = await novuClient.notifications.list({ page: 0 });\n const activities = body.result;\n\n expect(activities.hasMore).to.equal(false);\n expect(activities.data.length, JSON.stringify(body.result)).to.equal(2);\n const activity = activities.data[0];\n if (!activity || !activity.template || !activity.subscriber) {\n throw new Error(`must have activity${JSON.stringify(activity)}`);\n }\n expect(activity.template.name).to.equal(template.name);\n expect(activity.template.id).to.equal(template._id);\n expect(activity.subscriber.firstName).to.equal('Test');\n expect(activity.channels).to.be.ok;\n expect(activity.channels).to.include.oneOf(Object.keys(ChannelTypeEnum).map((i) => ChannelTypeEnum[i]));\n });\n\n it('should filter by channel', async () => {\n await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: subscriberId,\n payload: { firstName: 'Test' },\n });\n\n await novuClient.trigger({\n workflowId: smsOnlyTemplate.triggers[0].identifier,\n to: subscriberId,\n payload: {\n firstName: 'Test',\n },\n });\n\n await novuClient.trigger({\n workflowId: smsOnlyTemplate.triggers[0].identifier,\n to: subscriberId,\n payload: {\n firstName: 'Test',\n },\n });\n\n await session.waitForJobCompletion([template._id, smsOnlyTemplate._id]);\n await novuClient.notifications.list({ page: 0, transactionId: ChannelTypeEnum.Sms });\n\n const body = await novuClient.notifications.list({ page: 0, channels: [ChannelTypeEnum.Sms] });\n const activities = body.result;\n\n expect(activities.hasMore).to.equal(false);\n expect(activities.data.length).to.equal(2);\n const activity = activities.data[0];\n if (!activity || !activity.template || !activity.subscriber) {\n throw new Error('must have activity');\n }\n\n expect(activity.template?.name).to.equal(smsOnlyTemplate.name);\n expect(activity.channels).to.include(ChannelTypeEnum.Sms);\n });\n\n it('should filter by templateId', async () => {\n await novuClient.trigger({\n workflowId: smsOnlyTemplate.triggers[0].identifier,\n to: subscriberId,\n payload: {\n firstName: 'Test',\n },\n });\n\n await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: subscriberId,\n payload: { firstName: 'Test' },\n });\n\n await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: subscriberId,\n payload: { firstName: 'Test' },\n });\n await session.waitForJobCompletion(template._id);\n const body = await novuClient.notifications.list({ page: 0, templates: [template._id] });\n const activities = body.result;\n\n expect(activities.hasMore).to.equal(false);\n expect(activities.data.length).to.equal(2);\n\n expect(getActivity(activities.data, 0).template?.id).to.equal(template._id);\n expect(getActivity(activities.data, 1).template?.id).to.equal(template._id);\n });\n function getActivity(\n activities: Array,\n index: number\n ): ActivityNotificationResponseDto {\n const activity = activities[index];\n if (!activity || !activity.template || !activity.subscriber) {\n throw new Error('must have activity');\n }\n\n return activity;\n }\n\n it('should filter by email', async () => {\n await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: {\n subscriberId: SubscriberRepository.createObjectId(),\n email: 'test@email.coms',\n },\n payload: {\n firstName: 'Test',\n },\n });\n await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: {\n subscriberId: SubscriberRepository.createObjectId(),\n },\n payload: {\n firstName: 'Test',\n },\n });\n\n await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: SubscriberRepository.createObjectId(),\n payload: {\n firstName: 'Test',\n },\n });\n\n await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: SubscriberRepository.createObjectId(),\n payload: {\n firstName: 'Test',\n },\n });\n\n await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: subscriberId,\n payload: {\n firstName: 'Test',\n },\n });\n\n await session.waitForJobCompletion(template._id);\n const activities = (await novuClient.notifications.list({ page: 0, emails: ['test@email.coms'] })).result.data;\n\n expect(activities.length).to.equal(1);\n expect(getActivity(activities, 0).template?.id).to.equal(template._id);\n });\n\n it('should filter by subscriberId', async () => {\n const subscriberIdToCreate = `${SubscriberRepository.createObjectId()}some-test`;\n\n await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: {\n subscriberId: subscriberIdToCreate,\n email: 'test@email.coms',\n },\n payload: {\n firstName: 'Test',\n },\n });\n await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: SubscriberRepository.createObjectId(),\n payload: {\n firstName: 'Test',\n },\n });\n\n await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: SubscriberRepository.createObjectId(),\n payload: {\n firstName: 'Test',\n },\n });\n await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: subscriberId,\n payload: {\n firstName: 'Test',\n },\n });\n\n await session.waitForJobCompletion(template._id);\n const { result } = await novuClient.notifications.list({ page: 0, subscriberIds: [subscriberIdToCreate] });\n const activities = result.data;\n\n expect(activities.length).to.equal(1);\n expect(activities[0].template?.id, JSON.stringify(template)).to.equal(template._id);\n });\n\n it('should return with deleted workflow and subscriber data', async () => {\n const notificationTemplateRepository = new NotificationTemplateRepository();\n const subscriberRepository = new SubscriberRepository();\n const templateToDelete = await session.createTemplate();\n const subscriberIdToDelete = `${SubscriberRepository.createObjectId()}`;\n\n await novuClient.trigger({\n workflowId: templateToDelete.triggers[0].identifier,\n to: subscriberIdToDelete,\n payload: { firstName: 'Test' },\n });\n\n await session.waitForJobCompletion(templateToDelete._id);\n\n await notificationTemplateRepository.delete({ _id: templateToDelete._id, _environmentId: session.environment._id });\n const subscriberToDelete = await subscriberRepository.findOne({\n subscriberId: subscriberIdToDelete,\n _environmentId: session.environment._id,\n });\n await subscriberRepository.delete({ _id: subscriberToDelete?._id, _environmentId: session.environment._id });\n\n const body = await novuClient.notifications.list({ page: 0 });\n const activities = body.result;\n\n expect(activities.hasMore).to.equal(false);\n expect(activities.data.length, JSON.stringify(body.result)).to.equal(1);\n const activity = activities.data[0];\n\n expect(activity.template).to.be.undefined;\n expect(activity.subscriber).to.be.undefined;\n expect(activity.channels).to.be.ok;\n expect(activity.channels).to.include.oneOf(Object.keys(ChannelTypeEnum).map((i) => ChannelTypeEnum[i]));\n });\n\n it('should filter by contextKeys', async () => {\n const workflowBody: CreateWorkflowDto = {\n name: 'Test Context Workflow',\n workflowId: 'test-context-workflow-notifications',\n __source: WorkflowCreationSourceEnum.DASHBOARD,\n steps: [\n {\n type: StepTypeEnum.IN_APP,\n name: 'Test Step',\n controlValues: {\n subject: 'Test Subject',\n body: 'Test Body',\n },\n },\n ],\n };\n\n const workflowResponse = await session.testAgent.post('/v2/workflows').send(workflowBody);\n expect(workflowResponse.status).to.equal(201);\n const workflow: WorkflowResponseDto = workflowResponse.body.data;\n\n await novuClient.trigger({\n workflowId: workflow.workflowId,\n to: subscriberId,\n payload: {},\n context: { projectId: 'project-alpha' },\n });\n\n await novuClient.trigger({\n workflowId: workflow.workflowId,\n to: subscriberId,\n payload: {},\n context: { projectId: 'project-beta' },\n });\n\n await session.waitForWorkflowQueueCompletion();\n await session.waitForSubscriberQueueCompletion();\n await session.waitForStandardQueueCompletion();\n await session.waitForJobCompletion(workflow._id);\n\n // Test 1: No contextKeys filter - should return all notifications\n let body = await novuClient.notifications.list({ page: 0 });\n expect(body.result.data.length).to.be.equal(2);\n\n // Test 2: Filter by specific context - should return only matching notification\n body = await novuClient.notifications.list({ page: 0, contextKeys: ['projectId:project-alpha'] });\n expect(body.result.data.length).to.be.equal(1);\n expect(body.result.data[0].template?.id).to.equal(workflow._id);\n expect(body.result.data[0].contextKeys).to.deep.equal(['projectId:project-alpha']);\n\n // Test 3: Filter by different context - should return only matching notification\n body = await novuClient.notifications.list({ page: 0, contextKeys: ['projectId:project-beta'] });\n expect(body.result.data.length).to.be.equal(1);\n expect(body.result.data[0].template?.id).to.equal(workflow._id);\n expect(body.result.data[0].contextKeys).to.deep.equal(['projectId:project-beta']);\n });\n});\n"
},
{
"path": "apps/api/src/app/notifications/e2e/get-activity.e2e.ts",
"content": "import { Novu } from '@novu/api';\nimport { ActivityNotificationResponseDto } from '@novu/api/models/components';\nimport { MessageRepository, NotificationRepository, NotificationTemplateEntity, SubscriberRepository } from '@novu/dal';\nimport { JobStatusEnum, StepTypeEnum } from '@novu/shared';\nimport { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\nimport { initNovuClassSdk } from '../../shared/helpers/e2e/sdk/e2e-sdk.helper';\n\ndescribe('Get activity - /notifications/:notificationId (GET) #novu-v2', async () => {\n let session: UserSession;\n let template: NotificationTemplateEntity;\n let novuClient: Novu;\n let originalTraceReadValue: string | undefined;\n let originalTraceWriteValue: string | undefined;\n let originalStepRunEnvValue: string | undefined;\n const messageRepository: MessageRepository = new MessageRepository();\n const notificationRepository: NotificationRepository = new NotificationRepository();\n\n const updateNotification = async ({\n id,\n status,\n body,\n }: {\n id: string;\n status: 'read' | 'unread' | 'archive' | 'unarchive' | 'snooze' | 'unsnooze';\n body?: any;\n }) => {\n return await session.testAgent\n .patch(`/v1/inbox/notifications/${id}/${status}`)\n .set('Authorization', `Bearer ${session.subscriberToken}`)\n .send(body);\n };\n\n before(async () => {\n originalTraceReadValue = process.env.IS_TRACE_LOGS_READ_ENABLED;\n originalTraceWriteValue = process.env.IS_TRACE_LOGS_ENABLED;\n (process.env as any).IS_TRACE_LOGS_READ_ENABLED = 'true';\n (process.env as any).IS_TRACE_LOGS_ENABLED = 'true';\n });\n\n after(async () => {\n if (originalTraceReadValue === undefined) {\n delete (process.env as any).IS_TRACE_LOGS_READ_ENABLED;\n } else {\n (process.env as any).IS_TRACE_LOGS_READ_ENABLED = originalTraceReadValue;\n }\n if (originalTraceWriteValue === undefined) {\n delete (process.env as any).IS_TRACE_LOGS_ENABLED;\n } else {\n (process.env as any).IS_TRACE_LOGS_ENABLED = originalTraceWriteValue;\n }\n if (originalStepRunEnvValue === undefined) {\n delete (process.env as any).IS_STEP_RUN_LOGS_READ_ENABLED;\n }\n if (originalStepRunEnvValue !== undefined) {\n (process.env as any).IS_STEP_RUN_LOGS_READ_ENABLED = originalStepRunEnvValue;\n }\n });\n\n beforeEach(async () => {\n session = new UserSession();\n await session.initialize();\n\n template = await session.createTemplate({\n steps: [\n {\n type: StepTypeEnum.IN_APP,\n content: 'Test notification content {{name}}',\n },\n ],\n });\n\n novuClient = initNovuClassSdk(session);\n });\n\n it('should return traces in activity feed when traces feature flag is enabled', async () => {\n // Step 1: Trigger a notification to create trace logs\n const triggerResponse = await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: session.subscriberId,\n payload: { name: 'Test User' },\n });\n\n expect(triggerResponse.result?.acknowledged).to.equal(true);\n\n // Step 2: Wait for the worker to process the notification and create traces\n await session.waitForJobCompletion(template._id);\n const message = await messageRepository.findOne({\n _environmentId: session.environment._id,\n _subscriberId: session.subscriberProfile?._id,\n _templateId: template._id,\n transactionId: triggerResponse.result?.transactionId,\n });\n\n expect(message).to.be.ok;\n if (!message) throw new Error('Message not found');\n\n const { body, status } = await updateNotification({\n id: message._id,\n status: 'read',\n });\n expect(status).to.equal(200);\n\n const notification = await notificationRepository.findOne({\n _environmentId: session.environment._id,\n _subscriberId: session.subscriberProfile?._id,\n _templateId: template._id,\n transactionId: triggerResponse.result?.transactionId,\n });\n expect(notification).to.be.ok;\n if (!notification) throw new Error('Notification not found');\n\n const activityResponse = await session.testAgent.get(`/v1/notifications/${notification._id}`).expect(200);\n const activity: ActivityNotificationResponseDto = activityResponse.body.data;\n expect(activity).to.be.ok;\n if (!activity.jobs) throw new Error('Jobs not found');\n\n expect(activity.jobs).to.be.an('array');\n\n const actualDetails = activity.jobs[0].executionDetails.map((detail) => detail.detail);\n const expectedExecutionDetails = ['Step queued', 'Message created', 'Message sent', 'Message read'];\n\n expect(actualDetails.length).to.be.equal(4);\n expectedExecutionDetails.forEach((expectedDetail) => {\n expect(actualDetails).to.include(expectedDetail);\n });\n });\n\n it('should fallback to old method when traces query fails', async () => {\n const triggerResponse = await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: session.subscriberId,\n payload: { name: 'Test User' },\n });\n\n await session.waitForJobCompletion(template._id);\n\n const message = await messageRepository.findOne({\n _environmentId: session.environment._id,\n _subscriberId: session.subscriberProfile?._id,\n _templateId: template._id,\n transactionId: triggerResponse.result?.transactionId,\n });\n\n expect(message).to.be.ok;\n if (!message) throw new Error('Message not found');\n\n const notification = await notificationRepository.findOne({\n _environmentId: session.environment._id,\n _subscriberId: session.subscriberProfile?._id,\n _templateId: template._id,\n transactionId: triggerResponse.result?.transactionId,\n });\n expect(notification).to.be.ok;\n if (!notification) throw new Error('Notification not found');\n\n const activityResponse = await session.testAgent.get(`/v1/notifications/${notification._id}`).expect(200);\n const activity: ActivityNotificationResponseDto = activityResponse.body.data;\n expect(activity).to.be.ok;\n if (!activity.jobs) throw new Error('Jobs not found');\n\n expect(activity.jobs).to.be.an('array');\n\n const actualDetails = activity.jobs[0].executionDetails.map((detail) => detail.detail);\n const expectedExecutionDetails = ['Step queued', 'Message created', 'Message sent'];\n\n expect(actualDetails.length).to.be.equal(3);\n expectedExecutionDetails.forEach((expectedDetail) => {\n expect(actualDetails).to.include(\n expectedDetail,\n `Expected execution detail '${expectedDetail}' not found in job. Found: ${actualDetails.join(', ')}`\n );\n });\n expect(actualDetails).to.not.include('Message read');\n });\n\n it('should return traces in activity feed with step runs and trace logs', async () => {\n // Step 1: Trigger a notification to create trace logs\n const triggerResponse = await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: session.subscriberId,\n payload: { name: 'Test User' },\n });\n\n expect(triggerResponse.result?.acknowledged).to.equal(true);\n\n // Step 2: Wait for the worker to process the notification and create traces\n await session.waitForJobCompletion(template._id);\n const message = await messageRepository.findOne({\n _environmentId: session.environment._id,\n _subscriberId: session.subscriberProfile?._id,\n _templateId: template._id,\n transactionId: triggerResponse.result?.transactionId,\n });\n\n expect(message).to.be.ok;\n if (!message) throw new Error('Message not found');\n\n const { body, status } = await updateNotification({\n id: message._id,\n status: 'read',\n });\n expect(status).to.equal(200);\n\n const notification = await notificationRepository.findOne({\n _environmentId: session.environment._id,\n _subscriberId: session.subscriberProfile?._id,\n _templateId: template._id,\n transactionId: triggerResponse.result?.transactionId,\n });\n expect(notification).to.be.ok;\n if (!notification) throw new Error('Notification not found');\n\n const activityResponse = await session.testAgent.get(`/v1/notifications/${notification._id}`).expect(200);\n const activity: ActivityNotificationResponseDto = activityResponse.body.data;\n expect(activity).to.be.ok;\n if (!activity.jobs) throw new Error('Jobs not found');\n\n expect(activity.jobs).to.be.an('array');\n\n const actualDetails = activity.jobs[0].executionDetails.map((detail) => detail.detail);\n const expectedExecutionDetails = ['Step queued', 'Message created', 'Message sent', 'Message read'];\n\n expect(actualDetails.length).to.be.equal(4);\n expectedExecutionDetails.forEach((expectedDetail) => {\n expect(actualDetails).to.include(\n expectedDetail,\n `Expected execution detail '${expectedDetail}' not found in job. Found: ${actualDetails.join(', ')}`\n );\n });\n });\n\n it('should use step runs when both trace and step run feature flags are enabled', async () => {\n // Enable both feature flags\n (process.env as any).IS_STEP_RUN_LOGS_READ_ENABLED = 'true';\n\n const triggerResponse = await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: session.subscriberId,\n payload: { name: 'Test User' },\n });\n\n expect(triggerResponse.result?.acknowledged).to.equal(true);\n\n await session.waitForJobCompletion(template._id);\n\n const notification = await notificationRepository.findOne({\n _environmentId: session.environment._id,\n _subscriberId: session.subscriberProfile?._id,\n _templateId: template._id,\n transactionId: triggerResponse.result?.transactionId,\n });\n expect(notification).to.be.ok;\n if (!notification) throw new Error('Notification not found');\n\n const activityResponse = await session.testAgent.get(`/v1/notifications/${notification._id}`).expect(200);\n const activity: ActivityNotificationResponseDto = activityResponse.body.data;\n expect(activity).to.be.ok;\n\n expect(activity.jobs?.length).to.be.equal(2);\n expect(activity.jobs?.[0].type).to.be.equal(StepTypeEnum.TRIGGER);\n expect(activity.jobs?.[0].status).to.be.equal(JobStatusEnum.COMPLETED);\n expect(activity.jobs?.[1].type).to.be.equal(StepTypeEnum.IN_APP);\n expect(activity.jobs?.[1].status).to.be.equal(JobStatusEnum.COMPLETED);\n\n // Reset feature flag\n delete (process.env as any).IS_STEP_RUN_LOGS_READ_ENABLED;\n });\n\n it('should fallback to trace log method when step runs are not found', async () => {\n /*\n * Enable both feature flags\n * (process.env as any).IS_STEP_RUN_LOGS_READ_ENABLED = 'true';\n */\n\n const triggerResponse = await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: session.subscriberId,\n payload: { name: 'Test User' },\n });\n\n expect(triggerResponse.result?.acknowledged).to.equal(true);\n\n await session.waitForJobCompletion(template._id);\n\n const message = await messageRepository.findOne({\n _environmentId: session.environment._id,\n _subscriberId: session.subscriberProfile?._id,\n _templateId: template._id,\n transactionId: triggerResponse.result?.transactionId,\n });\n\n expect(message).to.be.ok;\n if (!message) throw new Error('Message not found');\n\n const notification = await notificationRepository.findOne({\n _environmentId: session.environment._id,\n _subscriberId: session.subscriberProfile?._id,\n _templateId: template._id,\n transactionId: triggerResponse.result?.transactionId,\n });\n expect(notification).to.be.ok;\n if (!notification) throw new Error('Notification not found');\n\n const activityResponse = await session.testAgent.get(`/v1/notifications/${notification._id}`).expect(200);\n const activity: ActivityNotificationResponseDto = activityResponse.body.data;\n expect(activity).to.be.ok;\n\n // Should still return jobs (even if from step_runs)\n expect(activity.jobs?.length).to.be.equal(1);\n expect(activity.jobs?.[0].type).to.be.equal(StepTypeEnum.IN_APP);\n expect(activity.jobs?.[0].status).to.be.equal(JobStatusEnum.COMPLETED);\n });\n\n it('should fallback to old method when traces query fails', async () => {\n const triggerResponse = await novuClient.trigger({\n workflowId: template.triggers[0].identifier,\n to: session.subscriberId,\n payload: { name: 'Test User' },\n });\n\n await session.waitForJobCompletion(template._id);\n\n const message = await messageRepository.findOne({\n _environmentId: session.environment._id,\n _subscriberId: session.subscriberProfile?._id,\n _templateId: template._id,\n transactionId: triggerResponse.result?.transactionId,\n });\n\n expect(message).to.be.ok;\n if (!message) throw new Error('Message not found');\n\n const notification = await notificationRepository.findOne({\n _environmentId: session.environment._id,\n _subscriberId: session.subscriberProfile?._id,\n _templateId: template._id,\n transactionId: triggerResponse.result?.transactionId,\n });\n expect(notification).to.be.ok;\n if (!notification) throw new Error('Notification not found');\n\n const activityResponse = await session.testAgent.get(`/v1/notifications/${notification._id}`).expect(200);\n const activity: ActivityNotificationResponseDto = activityResponse.body.data;\n expect(activity).to.be.ok;\n if (!activity.jobs) throw new Error('Jobs not found');\n\n expect(activity.jobs).to.be.an('array');\n\n const actualDetails = activity.jobs[0].executionDetails.map((detail) => detail.detail);\n const expectedExecutionDetails = ['Step queued', 'Message created', 'Message sent'];\n\n expect(actualDetails.length).to.be.equal(3);\n expectedExecutionDetails.forEach((expectedDetail) => {\n expect(actualDetails).to.include(\n expectedDetail,\n `Expected execution detail '${expectedDetail}' not found in job. Found: ${actualDetails.join(', ')}`\n );\n });\n expect(actualDetails).to.not.include('Message read');\n });\n});\n"
},
{
"path": "apps/api/src/app/notifications/notification.controller.ts",
"content": "import { Controller, Get, Param, Query } from '@nestjs/common';\nimport { ApiExcludeEndpoint, ApiOperation, ApiQuery, ApiTags } from '@nestjs/swagger';\nimport { RequirePermissions } from '@novu/application-generic';\nimport { ChannelTypeEnum, PermissionsEnum, SeverityLevelEnum, UserSessionData } from '@novu/shared';\nimport { RequireAuthentication } from '../auth/framework/auth.decorator';\nimport { ExternalApiAccessible } from '../auth/framework/external-api.decorator';\nimport { ApiCommonResponses, ApiOkResponse, ApiResponse } from '../shared/framework/response.decorator';\nimport { SdkGroupName, SdkMethodName } from '../shared/framework/swagger/sdk.decorators';\nimport { UserSession } from '../shared/framework/user.decorator';\nimport { ActivitiesRequestDto } from './dtos/activities-request.dto';\nimport { ActivitiesResponseDto, ActivityNotificationResponseDto } from './dtos/activities-response.dto';\nimport { ActivityGraphStatesResponse } from './dtos/activity-graph-states-response.dto';\nimport { ActivityStatsResponseDto } from './dtos/activity-stats-response.dto';\nimport { GetActivityCommand } from './usecases/get-activity/get-activity.command';\nimport { GetActivity } from './usecases/get-activity/get-activity.usecase';\nimport { GetActivityFeedCommand } from './usecases/get-activity-feed/get-activity-feed.command';\nimport { GetActivityFeed } from './usecases/get-activity-feed/get-activity-feed.usecase';\nimport { GetActivityGraphStatsCommand } from './usecases/get-activity-graph-states/get-activity-graph-states.command';\nimport { GetActivityGraphStats } from './usecases/get-activity-graph-states/get-activity-graph-states.usecase';\nimport { GetActivityStats, GetActivityStatsCommand } from './usecases/get-activity-stats';\n\n@ApiCommonResponses()\n@RequireAuthentication()\n@Controller('/notifications')\n@ApiTags('Notifications')\nexport class NotificationsController {\n constructor(\n private getActivityFeedUsecase: GetActivityFeed,\n private getActivityStatsUsecase: GetActivityStats,\n private getActivityGraphStatsUsecase: GetActivityGraphStats,\n private getActivityUsecase: GetActivity\n ) {}\n\n @Get('')\n @ApiOkResponse({\n type: ActivitiesResponseDto,\n })\n @ApiOperation({\n summary: 'List all events',\n description: `List all notification events (triggered events) for the current environment. \n This API supports filtering by **channels**, **templates**, **emails**, **subscriberIds**, **transactionId**, **topicKey**, **severity**, **contextKeys**. \n Checkout all available filters in the query section.\n This API returns event triggers, to list each channel notifications, check messages APIs.`,\n })\n @ExternalApiAccessible()\n @RequirePermissions(PermissionsEnum.NOTIFICATION_READ)\n async listNotifications(\n @UserSession() user: UserSessionData,\n @Query() query: ActivitiesRequestDto\n ): Promise {\n let channelsQuery: ChannelTypeEnum[] | null = null;\n if (query.channels) {\n channelsQuery = Array.isArray(query.channels) ? query.channels : [query.channels];\n }\n\n let templatesQuery: string[] | null = null;\n if (query.templates) {\n templatesQuery = Array.isArray(query.templates) ? query.templates : [query.templates];\n }\n\n let emailsQuery: string[] = [];\n if (query.emails) {\n emailsQuery = Array.isArray(query.emails) ? query.emails : [query.emails];\n }\n\n let subscribersQuery: string[] = [];\n if (query.subscriberIds) {\n subscribersQuery = Array.isArray(query.subscriberIds) ? query.subscriberIds : [query.subscriberIds];\n }\n\n let transactionIdQuery: string[] | undefined;\n if (query.transactionId) {\n transactionIdQuery = Array.isArray(query.transactionId) ? query.transactionId : [query.transactionId];\n }\n\n let severityQuery: SeverityLevelEnum[] | null = null;\n if (query.severity) {\n severityQuery = Array.isArray(query.severity) ? query.severity : [query.severity];\n }\n\n return this.getActivityFeedUsecase.execute(\n GetActivityFeedCommand.create({\n page: query.page,\n limit: query.limit,\n organizationId: user.organizationId,\n environmentId: user.environmentId,\n userId: user._id,\n channels: channelsQuery,\n templates: templatesQuery,\n emails: emailsQuery,\n search: query.search,\n subscriberIds: subscribersQuery,\n transactionId: transactionIdQuery,\n topicKey: query.topicKey,\n subscriptionId: query.subscriptionId,\n severity: severityQuery,\n after: query.after,\n before: query.before,\n contextKeys: query.contextKeys,\n })\n );\n }\n\n @ApiResponse(ActivityStatsResponseDto)\n @ApiExcludeEndpoint()\n @ApiOperation({\n summary: 'Retrieve events statistics',\n description: `Retrieve notification statistics for the current environment. \n This API returns the number of weekly and monthly notifications sent for the current environment.`,\n deprecated: true,\n })\n @Get('/stats')\n @ExternalApiAccessible()\n @SdkGroupName('Notifications.Stats')\n @RequirePermissions(PermissionsEnum.NOTIFICATION_READ)\n getActivityStats(@UserSession() user: UserSessionData): Promise {\n return this.getActivityStatsUsecase.execute(\n GetActivityStatsCommand.create({\n organizationId: user.organizationId,\n environmentId: user.environmentId,\n })\n );\n }\n\n @Get('/graph/stats')\n @ExternalApiAccessible()\n @ApiExcludeEndpoint()\n @ApiResponse(ActivityGraphStatesResponse, 200, true)\n @ApiOperation({\n summary: 'Retrieve events graph statistics',\n description: `Retrieve events graph statistics for the current environment. \n This API returns the number of events sent. This data is used to generate the graph in the legacy dashboard.`,\n deprecated: true,\n })\n @ApiQuery({\n name: 'days',\n type: Number,\n required: false,\n })\n @SdkGroupName('Notifications.Stats')\n @SdkMethodName('graph')\n @RequirePermissions(PermissionsEnum.NOTIFICATION_READ)\n getActivityGraphStats(\n @UserSession() user: UserSessionData,\n @Query('days') days = 32\n ): Promise {\n return this.getActivityGraphStatsUsecase.execute(\n GetActivityGraphStatsCommand.create({\n days: days ? Number(days) : 32,\n organizationId: user.organizationId,\n environmentId: user.environmentId,\n userId: user._id,\n })\n );\n }\n\n @Get('/:notificationId')\n @ApiResponse(ActivityNotificationResponseDto)\n @ApiOperation({\n summary: 'Retrieve an event',\n description: `Retrieve an event by its unique key identifier **notificationId**. \n Here **notificationId** is of mongodbId type. \n This API returns the event details - execution logs, status, actual notification (message) generated by each workflow step.`,\n })\n @ExternalApiAccessible()\n @RequirePermissions(PermissionsEnum.NOTIFICATION_READ)\n getNotification(\n @UserSession() user: UserSessionData,\n @Param('notificationId') notificationId: string\n ): Promise {\n return this.getActivityUsecase.execute(\n GetActivityCommand.create({\n notificationId,\n organizationId: user.organizationId,\n environmentId: user.environmentId,\n userId: user._id,\n })\n );\n }\n}\n"
},
{
"path": "apps/api/src/app/notifications/notification.module.ts",
"content": "import { Module } from '@nestjs/common';\nimport { CommunityOrganizationRepository } from '@novu/dal';\nimport { AuthModule } from '../auth/auth.module';\nimport { SharedModule } from '../shared/shared.module';\nimport { NotificationsController } from './notification.controller';\nimport { USE_CASES } from './usecases';\n\n@Module({\n imports: [SharedModule, AuthModule],\n providers: [...USE_CASES, CommunityOrganizationRepository],\n controllers: [NotificationsController],\n})\nexport class NotificationModule {}\n"
},
{
"path": "apps/api/src/app/notifications/usecases/get-activity/get-activity.command.ts",
"content": "import { IsDefined, IsMongoId } from 'class-validator';\nimport { EnvironmentWithUserCommand } from '../../../shared/commands/project.command';\n\nexport class GetActivityCommand extends EnvironmentWithUserCommand {\n @IsDefined()\n @IsMongoId()\n notificationId: string;\n}\n"
},
{
"path": "apps/api/src/app/notifications/usecases/get-activity/get-activity.usecase.ts",
"content": "import { Injectable, NotFoundException } from '@nestjs/common';\nimport {\n AnalyticsService,\n FeatureFlagsService,\n PinoLogger,\n QueryBuilder,\n StepRun,\n StepRunRepository,\n Trace,\n TraceLogRepository,\n WorkflowRun,\n WorkflowRunRepository,\n} from '@novu/application-generic';\nimport {\n ExecutionDetailFeedItem,\n JobFeedItem,\n JobStatusEnum,\n NotificationFeedItemEntity,\n NotificationRepository,\n NotificationStepEntity,\n} from '@novu/dal';\nimport {\n ExecutionDetailsSourceEnum,\n ExecutionDetailsStatusEnum,\n FeatureFlagsKeysEnum,\n ProvidersIdEnum,\n StepTypeEnum,\n TriggerTypeEnum,\n} from '@novu/shared';\n\nimport { ActivityNotificationResponseDto } from '../../dtos/activities-response.dto';\nimport { mapFeedItemToDto } from '../get-activity-feed/map-feed-item-to.dto';\nimport { GetActivityCommand } from './get-activity.command';\n\nconst workflowRunSelectColumns = [\n 'workflow_run_id',\n 'workflow_id',\n 'workflow_name',\n 'organization_id',\n 'environment_id',\n 'subscriber_id',\n 'external_subscriber_id',\n 'trigger_identifier',\n 'transaction_id',\n 'channels',\n 'subscriber_to',\n 'payload',\n 'topics',\n 'context_keys',\n 'created_at',\n 'updated_at',\n] as const;\n\nconst stepRunSelectColumns = [\n 'step_run_id',\n 'step_id',\n 'step_type',\n 'provider_id',\n 'status',\n 'created_at',\n 'updated_at',\n 'schedule_extensions_count',\n] as const;\ntype StepRunFetchResult = Pick;\n\nconst traceSelectColumns = ['id', 'entity_id', 'title', 'status', 'created_at', 'raw_data'] as const;\n\n@Injectable()\nexport class GetActivity {\n constructor(\n private notificationRepository: NotificationRepository,\n private analyticsService: AnalyticsService,\n private traceLogRepository: TraceLogRepository,\n private stepRunRepository: StepRunRepository,\n private workflowRunRepository: WorkflowRunRepository,\n private logger: PinoLogger,\n private featureFlagsService: FeatureFlagsService\n ) {}\n\n async execute(command: GetActivityCommand): Promise {\n this.analyticsService.track('Get Activity Feed Item - [Activity Feed]', command.userId, {\n _organization: command.organizationId,\n });\n\n const flagContext = {\n organization: { _id: command.organizationId },\n user: { _id: command.userId },\n environment: { _id: command.environmentId },\n } as const;\n\n const [tracesEnabled, stepRunsEnabled, workflowRunsEnabled] = await Promise.all([\n this.featureFlagsService.getFlag({\n key: FeatureFlagsKeysEnum.IS_TRACE_LOGS_READ_ENABLED,\n defaultValue: false,\n ...flagContext,\n }),\n this.featureFlagsService.getFlag({\n key: FeatureFlagsKeysEnum.IS_STEP_RUN_LOGS_READ_ENABLED,\n defaultValue: false,\n ...flagContext,\n }),\n this.featureFlagsService.getFlag({\n key: FeatureFlagsKeysEnum.IS_WORKFLOW_RUN_LOGS_READ_ENABLED,\n defaultValue: false,\n ...flagContext,\n }),\n ]);\n\n this.logger.debug({\n tracesEnabled,\n stepRunsEnabled,\n workflowRunsEnabled,\n }, 'feature flags');\n\n let feedItem: NotificationFeedItemEntity | null = null;\n\n if (workflowRunsEnabled && stepRunsEnabled && tracesEnabled) {\n this.logger.debug('analytics full ingegration enabled');\n feedItem = await this.getFeedItemFromWorkflowRuns(command);\n } else if (tracesEnabled && stepRunsEnabled) {\n this.logger.debug('analytics step runs enabled, no workflow runs');\n feedItem = await this.getFeedItemFromStepRuns(command);\n } else if (tracesEnabled) {\n this.logger.debug('analytics traces enabled, no step runs or workflow runs');\n feedItem = await this.getFeedItemFromTraceLog(command);\n } else {\n this.logger.debug('analytics fallback to old method');\n feedItem = await this.notificationRepository.getFeedItem(\n command.notificationId,\n command.environmentId,\n command.organizationId\n );\n }\n\n if (!feedItem) {\n throw new NotFoundException('Notification not found', {\n cause: `Notification with id ${command.notificationId} not found`,\n });\n }\n\n return mapFeedItemToDto(feedItem);\n }\n\n private mapTraceStatusToExecutionStatus(traceStatus: string): ExecutionDetailsStatusEnum {\n switch (traceStatus.toLowerCase()) {\n case 'success':\n return ExecutionDetailsStatusEnum.SUCCESS;\n case 'error':\n case 'failed':\n return ExecutionDetailsStatusEnum.FAILED;\n case 'warning':\n return ExecutionDetailsStatusEnum.WARNING;\n case 'pending':\n return ExecutionDetailsStatusEnum.PENDING;\n case 'queued':\n return ExecutionDetailsStatusEnum.QUEUED;\n default:\n return ExecutionDetailsStatusEnum.PENDING;\n }\n }\n\n private async getExecutionDetailsByEntityId(\n entityIds: string[],\n command: GetActivityCommand\n ): Promise> {\n if (entityIds.length === 0) {\n return new Map();\n }\n\n const traceQuery = new QueryBuilder({\n environmentId: command.environmentId,\n })\n .whereIn('entity_id', entityIds)\n .whereEquals('entity_type', 'step_run')\n .build();\n\n const traceResult = await this.traceLogRepository.find({\n where: traceQuery,\n orderBy: 'created_at',\n orderDirection: 'ASC',\n select: traceSelectColumns,\n });\n\n const executionDetailsByEntityId = new Map();\n\n // Group traces by entity ID\n const traceLogsByEntityId = new Map();\n for (const trace of traceResult.data) {\n if (!traceLogsByEntityId.has(trace.entity_id)) {\n traceLogsByEntityId.set(trace.entity_id, []);\n }\n // biome-ignore lint/style/noNonNullAssertion: we we create it in the if above\n traceLogsByEntityId.get(trace.entity_id)!.push(trace);\n }\n\n // Convert traces to execution details for each entity\n for (const [entityId, traces] of traceLogsByEntityId) {\n const executionDetails: ExecutionDetailFeedItem[] = traces.map((trace) => ({\n _id: trace.id,\n // TODO: add providerId from traces\n providerId: undefined, // Will be overridden by step runs if available\n detail: trace.title,\n source: ExecutionDetailsSourceEnum.INTERNAL,\n _jobId: entityId,\n status: this.mapTraceStatusToExecutionStatus(trace.status),\n isTest: false,\n isRetry: false,\n createdAt: new Date(trace.created_at).toISOString(),\n raw: trace.raw_data,\n }));\n\n executionDetailsByEntityId.set(entityId, executionDetails);\n }\n\n return executionDetailsByEntityId;\n }\n\n private async processStepRunsForFeedItem(\n feedItem: NotificationFeedItemEntity,\n command: GetActivityCommand\n ): Promise {\n const stepRunsQuery = new QueryBuilder({\n environmentId: command.environmentId,\n })\n .whereEquals('transaction_id', feedItem.transactionId)\n .build();\n\n const stepRunsResult = await this.stepRunRepository.find({\n where: stepRunsQuery,\n orderBy: 'created_at',\n orderDirection: 'ASC',\n useFinal: true,\n select: stepRunSelectColumns,\n });\n\n if (!stepRunsResult.data || stepRunsResult.data.length === 0) {\n return [];\n }\n\n const stepRunIds = stepRunsResult.data.map((stepRun) => stepRun.step_run_id);\n const executionDetailsByStepRunId = await this.getExecutionDetailsByEntityId(stepRunIds, command);\n\n return stepRunsResult.data.map((stepRun) => mapStepRunToJob(stepRun, executionDetailsByStepRunId));\n }\n\n private async getFeedItemFromStepRuns(command: GetActivityCommand): Promise {\n try {\n const feedItem = await this.notificationRepository.findNotificationMetadataOnly(\n command.notificationId,\n command.environmentId,\n command.organizationId\n );\n\n if (!feedItem) {\n return null;\n }\n\n // Process step runs and add them to the feed item\n feedItem.jobs = await this.processStepRunsForFeedItem(feedItem, command);\n\n return feedItem;\n } catch (error) {\n this.logger.error(\n {\n error: error instanceof Error ? error.message : 'Unknown error',\n notificationId: command.notificationId,\n environmentId: command.environmentId,\n organizationId: command.organizationId,\n },\n 'Failed to get feed item from step runs'\n );\n\n // Fall back to the current stage 1 method (traces + jobs from MongoDB)\n return await this.getFeedItemFromTraceLog(command);\n }\n }\n\n private async getFeedItemFromWorkflowRuns(command: GetActivityCommand): Promise {\n try {\n const workflowRunQuery = new QueryBuilder({\n environmentId: command.environmentId,\n })\n .whereEquals('workflow_run_id', command.notificationId)\n .build();\n\n const workflowRunsResult = await this.workflowRunRepository.find({\n where: workflowRunQuery,\n orderBy: 'created_at',\n orderDirection: 'ASC',\n limit: 1,\n useFinal: true,\n select: workflowRunSelectColumns,\n });\n\n if (!workflowRunsResult.data || workflowRunsResult.data.length === 0) {\n this.logger.warn(\n {\n notificationId: command.notificationId,\n environmentId: command.environmentId,\n organizationId: command.organizationId,\n },\n 'No workflow run found in ClickHouse, falling back to step runs'\n );\n\n // Fall back to step runs method\n return await this.getFeedItemFromStepRuns(command);\n }\n\n const mostRecentWorkflowRun = workflowRunsResult.data[0];\n\n // Create the base feed item from workflow run data\n const feedItem: NotificationFeedItemEntity = {\n _id: mostRecentWorkflowRun.workflow_run_id,\n _organizationId: mostRecentWorkflowRun.organization_id,\n _environmentId: mostRecentWorkflowRun.environment_id,\n _templateId: mostRecentWorkflowRun.workflow_id,\n _subscriberId: mostRecentWorkflowRun.subscriber_id,\n transactionId: mostRecentWorkflowRun.transaction_id,\n template: {\n _id: mostRecentWorkflowRun.workflow_id,\n name: mostRecentWorkflowRun.workflow_name,\n triggers: [\n {\n identifier: mostRecentWorkflowRun.trigger_identifier,\n type: TriggerTypeEnum.EVENT,\n variables: [],\n },\n ],\n },\n subscriber: {\n _id: mostRecentWorkflowRun.subscriber_id,\n subscriberId: mostRecentWorkflowRun.external_subscriber_id || '',\n firstName: '',\n lastName: '',\n email: '',\n phone: undefined,\n },\n jobs: [],\n to: mostRecentWorkflowRun.subscriber_to ? JSON.parse(mostRecentWorkflowRun.subscriber_to) : {},\n payload: mostRecentWorkflowRun.payload ? JSON.parse(mostRecentWorkflowRun.payload) : {},\n contextKeys: mostRecentWorkflowRun.context_keys,\n createdAt: new Date(mostRecentWorkflowRun.created_at).toISOString(),\n updatedAt: new Date(mostRecentWorkflowRun.updated_at).toISOString(),\n channels: mostRecentWorkflowRun.channels ? JSON.parse(mostRecentWorkflowRun.channels) : [],\n topics: mostRecentWorkflowRun.topics ? JSON.parse(mostRecentWorkflowRun.topics) : [],\n };\n\n feedItem.jobs = await this.processStepRunsForFeedItem(feedItem, command);\n\n return feedItem;\n } catch (error) {\n this.logger.error(\n {\n error: error instanceof Error ? error.message : 'Unknown error',\n notificationId: command.notificationId,\n environmentId: command.environmentId,\n organizationId: command.organizationId,\n },\n 'Failed to get feed item from workflow runs'\n );\n\n // Fall back to step runs method\n return await this.getFeedItemFromStepRuns(command);\n }\n }\n\n private async getFeedItemFromTraceLog(command: GetActivityCommand) {\n try {\n const feedItem = await this.notificationRepository.findMetadataForTraces(\n command.notificationId,\n command.environmentId,\n command.organizationId\n );\n\n if (!feedItem) {\n return null;\n }\n\n const jobIds = feedItem.jobs.map((job) => job._id);\n\n if (jobIds.length === 0) {\n return feedItem;\n }\n\n const executionDetailsByJobId = await this.getExecutionDetailsByEntityId(jobIds, command);\n\n feedItem.jobs = feedItem.jobs.map((job) => {\n const executionDetails = executionDetailsByJobId.get(job._id) || [];\n\n return {\n ...job,\n executionDetails,\n };\n });\n\n return feedItem;\n } catch (error) {\n this.logger.error(\n {\n error: error instanceof Error ? error.message : 'Unknown error',\n notificationId: command.notificationId,\n environmentId: command.environmentId,\n organizationId: command.organizationId,\n },\n 'Failed to get feed item from trace log'\n );\n\n // Fall back to the old method if trace log query fails\n return await this.notificationRepository.getFeedItem(\n command.notificationId,\n command.environmentId,\n command.organizationId\n );\n }\n }\n}\n\nfunction mapStepRunToJob(\n stepRun: StepRunFetchResult,\n executionDetailsByStepRunId: Map\n): JobFeedItem {\n const baseExecutionDetails = executionDetailsByStepRunId.get(stepRun.step_run_id) || [];\n // Create execution details with provider ID from step run data\n const executionDetails: ExecutionDetailFeedItem[] = baseExecutionDetails.map((detail) => ({\n ...detail,\n providerId: stepRun.provider_id as ProvidersIdEnum,\n }));\n\n const stepRunDto: NotificationStepEntity = {\n _id: stepRun.step_id,\n _templateId: stepRun.step_id,\n active: true,\n filters: [],\n };\n\n const jobDto: JobFeedItem = {\n _id: stepRun.step_run_id,\n status: stepRun.status as JobStatusEnum,\n overrides: {}, // Step runs don't have overrides, use empty object\n payload: {}, // Step runs don't have payload, use empty object\n step: stepRunDto,\n type: stepRun.step_type as StepTypeEnum,\n providerId: stepRun.provider_id as ProvidersIdEnum,\n createdAt: new Date(stepRun.created_at).toISOString(),\n updatedAt: new Date(stepRun.updated_at).toISOString(),\n digest: undefined, // Step runs don't have digest info\n executionDetails,\n scheduleExtensionsCount: stepRun.schedule_extensions_count,\n };\n\n return jobDto;\n}\n"
},
{
"path": "apps/api/src/app/notifications/usecases/get-activity-feed/get-activity-feed.command.ts",
"content": "import { ChannelTypeEnum, SeverityLevelEnum } from '@novu/shared';\nimport { IsArray, IsEnum, IsMongoId, IsNumber, IsOptional, IsString } from 'class-validator';\nimport { EnvironmentWithUserCommand } from '../../../shared/commands/project.command';\n\nexport class GetActivityFeedCommand extends EnvironmentWithUserCommand {\n @IsNumber()\n page: number;\n\n @IsNumber()\n limit: number;\n\n @IsOptional()\n @IsEnum(ChannelTypeEnum, {\n each: true,\n })\n channels?: ChannelTypeEnum[] | null;\n\n @IsOptional()\n @IsArray()\n @IsMongoId({ each: true })\n templates?: string[] | null;\n\n @IsOptional()\n @IsArray()\n emails?: string[];\n\n @IsOptional()\n @IsString()\n search?: string;\n\n @IsOptional()\n @IsArray()\n subscriberIds?: string[];\n\n @IsOptional()\n @IsArray()\n @IsString({ each: true })\n transactionId?: string[];\n\n @IsOptional()\n @IsString()\n topicKey?: string;\n\n @IsOptional()\n @IsString()\n subscriptionId?: string;\n\n @IsOptional()\n @IsArray()\n @IsString({ each: true })\n contextKeys?: string[];\n\n @IsOptional()\n @IsArray()\n @IsEnum(SeverityLevelEnum, { each: true })\n severity?: SeverityLevelEnum[] | null;\n\n @IsOptional()\n @IsString()\n after?: string;\n\n @IsOptional()\n @IsString()\n before?: string;\n}\n"
},
{
"path": "apps/api/src/app/notifications/usecases/get-activity-feed/get-activity-feed.usecase.spec.ts",
"content": "import { HttpException, HttpStatus } from '@nestjs/common';\nimport { Test } from '@nestjs/testing';\nimport { FeatureFlagsService, PinoLogger, TraceLogRepository } from '@novu/application-generic';\nimport { CommunityOrganizationRepository, NotificationRepository, SubscriberRepository } from '@novu/dal';\nimport { ApiServiceLevelEnum } from '@novu/shared';\nimport { expect } from 'chai';\nimport sinon from 'sinon';\nimport { GetActivityFeed } from './get-activity-feed.usecase';\n\ndescribe('GetActivityFeed - validateRetentionLimitForTier', () => {\n let useCase: GetActivityFeed;\n let organizationRepository: CommunityOrganizationRepository;\n let sandbox: sinon.SinonSandbox;\n\n beforeEach(async () => {\n sandbox = sinon.createSandbox();\n\n const moduleRef = await Test.createTestingModule({\n providers: [\n GetActivityFeed,\n SubscriberRepository,\n NotificationRepository,\n {\n provide: CommunityOrganizationRepository,\n useValue: {\n findById: () => {},\n },\n },\n {\n provide: TraceLogRepository,\n useValue: {\n createStepRun: () => {},\n },\n },\n {\n provide: FeatureFlagsService,\n useValue: {\n getFlag: () => Promise.resolve({ value: false }),\n },\n },\n {\n provide: PinoLogger,\n useValue: {\n info: () => {},\n error: () => {},\n warn: () => {},\n debug: () => {},\n trace: () => {},\n setContext: () => {},\n },\n },\n ],\n }).compile();\n\n useCase = moduleRef.get(GetActivityFeed);\n organizationRepository = moduleRef.get(CommunityOrganizationRepository);\n });\n\n afterEach(() => {\n sandbox.restore();\n });\n\n describe('Date handling', () => {\n it('should default to maximum allowed retention period when no dates provided', async () => {\n const now = new Date();\n sandbox.useFakeTimers(now.getTime());\n\n const mockOrg = {\n _id: 'org-123',\n apiServiceLevel: ApiServiceLevelEnum.PRO,\n createdAt: new Date('2024-01-01'),\n };\n\n sandbox.stub(organizationRepository, 'findById').resolves(mockOrg as any);\n\n const result = await (useCase as any).validateRetentionLimitForTier('org-123');\n const sevenDaysAgo = new Date(now.getTime() - 7 * 24 * 60 * 60 * 1000);\n expect(new Date(result.after).getTime()).to.be.approximately(sevenDaysAgo.getTime(), 1000); // allowing 1s difference\n expect(result.before).to.equal(now.toISOString());\n });\n\n it('should use provided dates when within retention period', async () => {\n const now = new Date();\n const twoDaysAgo = new Date(now.getTime() - 2 * 24 * 60 * 60 * 1000);\n\n const mockOrg = {\n _id: 'org-123',\n apiServiceLevel: ApiServiceLevelEnum.FREE,\n createdAt: new Date('2024-01-01'),\n };\n\n sandbox.stub(organizationRepository, 'findById').resolves(mockOrg as any);\n\n const result = await (useCase as any).validateRetentionLimitForTier(\n 'org-123',\n twoDaysAgo.toISOString(),\n now.toISOString()\n );\n\n expect(result.after).to.equal(twoDaysAgo.toISOString());\n expect(result.before).to.equal(now.toISOString());\n });\n\n it('should reject when after date is later than before date', async () => {\n const now = new Date();\n const tomorrow = new Date(now.getTime() + 24 * 60 * 60 * 1000);\n\n const mockOrg = {\n _id: 'org-123',\n apiServiceLevel: ApiServiceLevelEnum.FREE,\n createdAt: new Date('2024-01-01'),\n };\n\n sandbox.stub(organizationRepository, 'findById').resolves(mockOrg as any);\n\n try {\n await (useCase as any).validateRetentionLimitForTier('org-123', tomorrow.toISOString(), now.toISOString());\n expect.fail('Should have thrown an error');\n } catch (error) {\n expect(error).to.be.instanceOf(HttpException);\n expect(error.message).to.match(/Invalid date range/);\n expect(error.status).to.equal(HttpStatus.BAD_REQUEST);\n }\n });\n });\n\n describe('Retention periods by tier', () => {\n const testCases = [\n {\n tier: 'Legacy Free',\n apiServiceLevel: ApiServiceLevelEnum.FREE,\n createdAt: new Date('2024-01-01'),\n allowedDays: 30,\n rejectedDays: 31,\n },\n {\n tier: 'New Free',\n apiServiceLevel: ApiServiceLevelEnum.FREE,\n createdAt: new Date('2025-03-01'),\n allowedDays: 1,\n rejectedDays: 2,\n },\n {\n tier: 'Pro',\n apiServiceLevel: ApiServiceLevelEnum.PRO,\n createdAt: new Date(),\n allowedDays: 7,\n rejectedDays: 8,\n },\n {\n tier: 'Team',\n apiServiceLevel: ApiServiceLevelEnum.BUSINESS,\n createdAt: new Date(),\n allowedDays: 90,\n rejectedDays: 91,\n },\n ];\n\n testCases.forEach(({ tier, apiServiceLevel, createdAt, allowedDays, rejectedDays }) => {\n describe(tier, () => {\n it(`should allow access within ${allowedDays} days`, async () => {\n const now = new Date();\n const withinPeriod = new Date(now.getTime() - allowedDays * 24 * 60 * 60 * 1000);\n\n const mockOrg = {\n _id: 'org-123',\n apiServiceLevel,\n createdAt,\n };\n\n sandbox.stub(organizationRepository, 'findById').resolves(mockOrg as any);\n\n const result = await (useCase as any).validateRetentionLimitForTier(\n 'org-123',\n withinPeriod.toISOString(),\n now.toISOString()\n );\n\n expect(result.after).to.equal(withinPeriod.toISOString());\n expect(result.before).to.equal(now.toISOString());\n });\n\n it(`should reject access beyond ${rejectedDays} days`, async () => {\n const now = new Date();\n const beyondPeriod = new Date(now.getTime() - rejectedDays * 24 * 60 * 60 * 1000);\n\n const mockOrg = {\n _id: 'org-123',\n apiServiceLevel,\n createdAt,\n };\n\n sandbox.stub(organizationRepository, 'findById').resolves(mockOrg as any);\n\n try {\n await (useCase as any).validateRetentionLimitForTier(\n 'org-123',\n beyondPeriod.toISOString(),\n now.toISOString()\n );\n expect.fail('Should have thrown an error');\n } catch (error) {\n expect(error).to.be.instanceOf(HttpException);\n console.log(error.message);\n expect(error.message).to.match(/retention period/);\n expect(error.status).to.equal(HttpStatus.PAYMENT_REQUIRED);\n }\n });\n });\n });\n });\n});\n"
},
{
"path": "apps/api/src/app/notifications/usecases/get-activity-feed/get-activity-feed.usecase.ts",
"content": "import { HttpException, HttpStatus, Injectable } from '@nestjs/common';\nimport {\n FeatureFlagsService,\n Instrument,\n PinoLogger,\n QueryBuilder,\n Trace,\n TraceLogRepository,\n} from '@novu/application-generic';\nimport {\n CommunityOrganizationRepository,\n ExecutionDetailFeedItem,\n NotificationFeedItemEntity,\n NotificationRepository,\n OrganizationEntity,\n SubscriberRepository,\n} from '@novu/dal';\nimport {\n ApiServiceLevelEnum,\n ExecutionDetailsSourceEnum,\n ExecutionDetailsStatusEnum,\n FeatureFlagsKeysEnum,\n FeatureNameEnum,\n getFeatureForTierAsNumber,\n} from '@novu/shared';\nimport { ActivitiesResponseDto, ActivityNotificationResponseDto } from '../../dtos/activities-response.dto';\nimport { GetActivityFeedCommand } from './get-activity-feed.command';\nimport { mapFeedItemToDto } from './map-feed-item-to.dto';\n\nconst traceFindColumns = ['entity_id', 'id', 'status', 'title', 'raw_data', 'created_at'] as const;\ntype TraceFindResult = Pick;\n\n@Injectable()\nexport class GetActivityFeed {\n constructor(\n private subscribersRepository: SubscriberRepository,\n private notificationRepository: NotificationRepository,\n private organizationRepository: CommunityOrganizationRepository,\n private traceLogRepository: TraceLogRepository,\n private featureFlagsService: FeatureFlagsService,\n private logger: PinoLogger\n ) {}\n\n async execute(command: GetActivityFeedCommand): Promise {\n let subscriberIds: string[] | undefined;\n\n const { after, before } = await this.validateRetentionLimitForTier(\n command.organizationId,\n command.after,\n command.before\n );\n\n command.after = after;\n command.before = before;\n\n if (command.search || command.emails?.length || command.subscriberIds?.length) {\n subscriberIds = await this.findSubscribers(command);\n }\n\n if (subscriberIds && subscriberIds.length === 0) {\n return {\n page: 0,\n hasMore: false,\n pageSize: command.limit,\n data: [],\n };\n }\n\n const notifications: NotificationFeedItemEntity[] = await this.getFeedNotifications(command, subscriberIds);\n\n const data = notifications.reduce((memo, notification) => {\n // TODO: Identify why mongo returns an array of undefined or null values. Is it a data issue?\n if (notification) {\n memo.push(mapFeedItemToDto(notification));\n }\n\n return memo;\n }, []);\n\n return {\n page: command.page,\n hasMore: notifications?.length === command.limit,\n pageSize: command.limit,\n data,\n };\n }\n\n private async validateRetentionLimitForTier(organizationId: string, after?: string, before?: string) {\n const organization = await this.organizationRepository.findById(organizationId);\n\n if (!organization) {\n throw new HttpException('Organization not found', HttpStatus.INTERNAL_SERVER_ERROR);\n }\n\n const maxRetentionMs = this.getMaxRetentionPeriodByOrganization(organization);\n\n // For unlimited retention (self-hosted), skip retention validation\n if (maxRetentionMs === Number.MAX_SAFE_INTEGER) {\n const effectiveAfterDate = after ? this.parseAndValidateDate(after, 'after') : undefined;\n const effectiveBeforeDate = before ? this.parseAndValidateDate(before, 'before') : undefined;\n\n // Basic validation for date range if both dates are provided\n if (effectiveAfterDate && effectiveBeforeDate && effectiveAfterDate > effectiveBeforeDate) {\n throw new HttpException(\n 'Invalid date range: start date (after) must be earlier than end date (before)',\n HttpStatus.BAD_REQUEST\n );\n }\n\n return {\n after: effectiveAfterDate?.toISOString(),\n before: effectiveBeforeDate?.toISOString(),\n };\n }\n\n const earliestAllowedDate = new Date(Date.now() - maxRetentionMs);\n\n // If no after date is provided, default to the earliest allowed date\n const effectiveAfterDate = after ? this.parseAndValidateDate(after, 'after') : earliestAllowedDate;\n const effectiveBeforeDate = before ? this.parseAndValidateDate(before, 'before') : new Date();\n\n this.validateDateRange(earliestAllowedDate, effectiveAfterDate, effectiveBeforeDate);\n\n return {\n after: effectiveAfterDate.toISOString(),\n before: effectiveBeforeDate.toISOString(),\n };\n }\n\n private parseAndValidateDate(dateString: string, parameterName: string): Date {\n const parsedDate = new Date(dateString);\n\n if (Number.isNaN(parsedDate.getTime())) {\n throw new HttpException(\n `Invalid date format for parameter '${parameterName}': ${dateString}. Please provide a valid ISO 8601 date string.`,\n HttpStatus.BAD_REQUEST\n );\n }\n\n return parsedDate;\n }\n\n private validateDateRange(earliestAllowedDate: Date, afterDate: Date, beforeDate: Date) {\n if (afterDate > beforeDate) {\n throw new HttpException(\n 'Invalid date range: start date (after) must be earlier than end date (before)',\n HttpStatus.BAD_REQUEST\n );\n }\n\n // add buffer to account for time delay in execution\n const buffer = 1 * 60 * 60 * 1000; // 1 hour\n const bufferedEarliestAllowedDate = new Date(earliestAllowedDate.getTime() - buffer);\n\n if (\n process.env.NODE_ENV !== 'local' &&\n (afterDate < bufferedEarliestAllowedDate || beforeDate < bufferedEarliestAllowedDate)\n ) {\n throw new HttpException(\n `Requested date range exceeds your plan's retention period. ` +\n `The earliest accessible date for your plan is ${earliestAllowedDate.toISOString().split('T')[0]}. ` +\n `Please upgrade your plan to access older activities.`,\n HttpStatus.PAYMENT_REQUIRED\n );\n }\n }\n\n /**\n * Notifications are automatically deleted after a certain period of time\n * by a background job.\n *\n * @see https://github.com/novuhq/cloud-infra/blob/main/scripts/expiredNotification.js#L93\n */\n private getMaxRetentionPeriodByOrganization(organization: OrganizationEntity) {\n // 1. Self-hosted: effectively unlimited, use a large but safe finite window (100 years)\n if (process.env.IS_SELF_HOSTED === 'true') {\n return 100 * 365 * 24 * 60 * 60 * 1000; // ~100 years in ms, safe for Date math\n }\n\n const { apiServiceLevel, createdAt } = organization;\n\n // 2. Special case: Free tier orgs created before Feb 28, 2025 get 30 days\n if (apiServiceLevel === ApiServiceLevelEnum.FREE && new Date(createdAt) < new Date('2025-02-28')) {\n return 30 * 24 * 60 * 60 * 1000;\n }\n\n // 3. Otherwise, use tier-based retention from feature flags\n return getFeatureForTierAsNumber(\n FeatureNameEnum.PLATFORM_ACTIVITY_FEED_RETENTION,\n apiServiceLevel ?? ApiServiceLevelEnum.FREE,\n true\n );\n }\n\n @Instrument()\n private async findSubscribers(command: GetActivityFeedCommand): Promise {\n return await this.subscribersRepository.searchSubscribers(\n command.environmentId,\n command.subscriberIds,\n command.emails,\n command.search\n );\n }\n\n @Instrument()\n private async getFeedNotifications(\n command: GetActivityFeedCommand,\n subscriberIds?: string[]\n ): Promise {\n const notifications = await this.notificationRepository.getFeed(\n command.environmentId,\n {\n channels: command.channels,\n templates: command.templates,\n subscriberIds: subscriberIds || [],\n transactionId: command.transactionId,\n topicKey: command.topicKey,\n subscriptionId: command.subscriptionId,\n after: command.after,\n before: command.before,\n severity: command.severity,\n contextKeys: command.contextKeys,\n },\n command.page * command.limit,\n command.limit\n );\n\n const isClickHouseOnlyEnabled = await this.featureFlagsService.getFlag({\n key: FeatureFlagsKeysEnum.IS_EXECUTION_DETAILS_CLICKHOUSE_ONLY_ENABLED,\n defaultValue: false,\n organization: { _id: command.organizationId },\n user: { _id: command.userId },\n environment: { _id: command.environmentId },\n });\n\n if (isClickHouseOnlyEnabled) {\n return await this.enhanceNotificationsWithTraces(notifications, command);\n }\n\n return notifications;\n }\n\n private async enhanceNotificationsWithTraces(\n notifications: NotificationFeedItemEntity[],\n command: GetActivityFeedCommand\n ): Promise {\n try {\n // Collect all job IDs from all notifications\n const allJobIds: string[] = [];\n for (const notification of notifications) {\n if (notification.jobs) {\n allJobIds.push(...notification.jobs.map((job) => job._id));\n }\n }\n\n if (allJobIds.length === 0) {\n return notifications;\n }\n\n // Get execution details from ClickHouse for all job IDs\n const executionDetailsByJobId = await this.getExecutionDetailsByEntityId(allJobIds, command);\n\n // Enhance each notification with the execution details\n const enhancedNotifications = notifications.map((notification) => {\n if (!notification.jobs) {\n return notification;\n }\n\n const enhancedJobs = notification.jobs.map((job) => {\n const executionDetails = executionDetailsByJobId.get(job._id) || [];\n\n return {\n ...job,\n executionDetails,\n };\n });\n\n return {\n ...notification,\n jobs: enhancedJobs,\n };\n });\n\n this.logger.debug({\n notificationCount: notifications.length,\n jobCount: allJobIds.length,\n executionDetailsCount: Array.from(executionDetailsByJobId.values()).flat().length,\n }, 'Successfully enhanced notifications with ClickHouse execution details');\n\n return enhancedNotifications;\n } catch (error) {\n this.logger.error(\n {\n error: error instanceof Error ? error.message : 'Unknown error',\n environmentId: command.environmentId,\n organizationId: command.organizationId,\n },\n 'Failed to enhance notifications with ClickHouse execution details, falling back to MongoDB data'\n );\n\n // Fall back to the original notifications if ClickHouse query fails\n return notifications;\n }\n }\n\n private mapTraceStatusToExecutionStatus(traceStatus: string): ExecutionDetailsStatusEnum {\n switch (traceStatus.toLowerCase()) {\n case 'success':\n return ExecutionDetailsStatusEnum.SUCCESS;\n case 'error':\n case 'failed':\n return ExecutionDetailsStatusEnum.FAILED;\n case 'warning':\n return ExecutionDetailsStatusEnum.WARNING;\n case 'pending':\n return ExecutionDetailsStatusEnum.PENDING;\n case 'queued':\n return ExecutionDetailsStatusEnum.QUEUED;\n default:\n return ExecutionDetailsStatusEnum.PENDING;\n }\n }\n\n private async getExecutionDetailsByEntityId(\n entityIds: string[],\n command: GetActivityFeedCommand\n ): Promise> {\n if (entityIds.length === 0) {\n return new Map();\n }\n\n const traceQuery = new QueryBuilder({\n environmentId: command.environmentId,\n })\n .whereIn('entity_id', entityIds)\n .whereEquals('entity_type', 'step_run')\n .build();\n\n const traceResult = await this.traceLogRepository.find({\n where: traceQuery,\n orderBy: 'created_at',\n orderDirection: 'ASC',\n select: traceFindColumns,\n });\n\n const executionDetailsByEntityId = new Map();\n\n // Group traces by entity ID\n const traceLogsByEntityId = new Map();\n for (const trace of traceResult.data) {\n if (!traceLogsByEntityId.has(trace.entity_id)) {\n traceLogsByEntityId.set(trace.entity_id, []);\n }\n const entityTraces = traceLogsByEntityId.get(trace.entity_id);\n if (entityTraces) {\n entityTraces.push(trace);\n }\n }\n\n // Convert traces to execution details for each entity\n for (const [entityId, traces] of traceLogsByEntityId) {\n const executionDetails: ExecutionDetailFeedItem[] = traces.map((trace: TraceFindResult) => ({\n _id: trace.id,\n providerId: undefined,\n detail: trace.title,\n source: ExecutionDetailsSourceEnum.INTERNAL,\n _jobId: entityId,\n status: this.mapTraceStatusToExecutionStatus(trace.status),\n isTest: false,\n isRetry: false,\n createdAt: new Date(trace.created_at).toISOString(),\n raw: trace.raw_data,\n }));\n\n executionDetailsByEntityId.set(entityId, executionDetails);\n }\n\n return executionDetailsByEntityId;\n }\n}\n"
},
{
"path": "apps/api/src/app/notifications/usecases/get-activity-feed/map-feed-item-to.dto.ts",
"content": "import {\n FieldFilterPartDto,\n FilterPartsDto,\n OnlineInLastFilterPartDto,\n PreviousStepFilterPartDto,\n RealtimeOnlineFilterPartDto,\n StepFilterDto,\n TenantFilterPartDto,\n WebhookFilterPartDto,\n} from '@novu/application-generic';\nimport {\n ExecutionDetailFeedItem,\n JobFeedItem,\n NotificationFeedItemEntity,\n NotificationStepEntity,\n StepFilter,\n SubscriberFeedItem,\n TemplateFeedItem,\n} from '@novu/dal';\nimport {\n DigestTypeEnum,\n FilterParts,\n FilterPartTypeEnum,\n IDigestRegularMetadata,\n IDigestTimedMetadata,\n IWorkflowStepMetadata,\n ProvidersIdEnum,\n SeverityLevelEnum,\n StepTypeEnum,\n} from '@novu/shared';\nimport { MessageTemplateDto } from '../../../shared/dtos/message.template.dto';\nimport {\n ActivityNotificationExecutionDetailResponseDto,\n ActivityNotificationJobResponseDto,\n ActivityNotificationResponseDto,\n ActivityNotificationStepResponseDto,\n ActivityNotificationSubscriberResponseDto,\n ActivityNotificationTemplateResponseDto,\n DigestMetadataDto,\n} from '../../dtos/activities-response.dto';\n\nfunction buildSubscriberDto(subscriber: SubscriberFeedItem): ActivityNotificationSubscriberResponseDto {\n return {\n _id: subscriber._id,\n subscriberId: subscriber.subscriberId,\n email: subscriber.email,\n firstName: subscriber.firstName,\n lastName: subscriber.lastName,\n phone: subscriber.phone,\n };\n}\n\nfunction buildTemplate(template: TemplateFeedItem): ActivityNotificationTemplateResponseDto {\n return {\n _id: template._id,\n name: template.name,\n triggers: template.triggers,\n origin: template.origin,\n };\n}\n\nexport function mapFeedItemToDto(entity: NotificationFeedItemEntity): ActivityNotificationResponseDto {\n return {\n _digestedNotificationId: entity._digestedNotificationId,\n _environmentId: entity._environmentId,\n _id: entity._id,\n _organizationId: entity._organizationId,\n _subscriberId: entity._subscriberId,\n _templateId: entity._templateId,\n topics: entity.topics?.map((topic) => ({\n _topicId: topic._topicId,\n topicKey: topic.topicKey,\n })),\n channels: entity.channels,\n createdAt: entity.createdAt,\n jobs: entity.jobs.map(mapJobToDto),\n tags: entity.tags,\n transactionId: entity.transactionId,\n updatedAt: entity.updatedAt,\n controls: entity.controls as Record,\n payload: entity.payload as Record,\n to: entity.to as Record,\n subscriber: entity.subscriber ? buildSubscriberDto(entity.subscriber) : undefined,\n template: entity.template ? buildTemplate(entity.template) : undefined,\n severity: entity.severity ?? SeverityLevelEnum.NONE,\n critical: entity.critical,\n contextKeys: entity.contextKeys,\n };\n}\n\nfunction mapChildFilterToDto(filterPart: FilterParts): FilterPartsDto {\n switch (filterPart.on) {\n case FilterPartTypeEnum.SUBSCRIBER:\n case FilterPartTypeEnum.PAYLOAD:\n return {\n ...filterPart,\n on: filterPart.on, // Ensure the correct enum value is set\n } as FieldFilterPartDto;\n\n case FilterPartTypeEnum.WEBHOOK:\n return {\n ...filterPart,\n on: FilterPartTypeEnum.WEBHOOK,\n } as WebhookFilterPartDto;\n\n case FilterPartTypeEnum.IS_ONLINE:\n return {\n ...filterPart,\n on: FilterPartTypeEnum.IS_ONLINE,\n } as RealtimeOnlineFilterPartDto;\n\n case FilterPartTypeEnum.IS_ONLINE_IN_LAST:\n return {\n ...filterPart,\n on: FilterPartTypeEnum.IS_ONLINE_IN_LAST,\n } as OnlineInLastFilterPartDto;\n\n case FilterPartTypeEnum.PREVIOUS_STEP:\n return {\n ...filterPart,\n on: FilterPartTypeEnum.PREVIOUS_STEP,\n } as PreviousStepFilterPartDto;\n\n case FilterPartTypeEnum.TENANT:\n return {\n ...filterPart,\n on: FilterPartTypeEnum.TENANT,\n } as TenantFilterPartDto;\n\n default:\n throw new Error(`Unknown filter part type: ${filterPart}`);\n }\n}\n\nfunction mapToFilterDto(stepFilter: StepFilter): StepFilterDto {\n return {\n children: stepFilter.children.map((child) => mapChildFilterToDto(child)),\n isNegated: stepFilter.isNegated,\n type: stepFilter.type,\n value: stepFilter.value,\n };\n}\n\nfunction convertStepToResponse(step: NotificationStepEntity): ActivityNotificationStepResponseDto {\n const responseDto = new ActivityNotificationStepResponseDto();\n\n responseDto._id = step._id || '';\n responseDto.active = step.active || false;\n responseDto.replyCallback = step.replyCallback;\n responseDto.controlVariables = step.controlVariables;\n responseDto.metadata = step.metadata;\n responseDto.issues = step.issues;\n responseDto._templateId = step._templateId || '';\n responseDto.name = step.name;\n responseDto._parentId = step._parentId || null;\n\n // Map filters\n responseDto.filters = (step.filters || []).map(mapToFilterDto);\n\n // Map template if exists\n if (step.template) {\n const messageTemplateDto = new MessageTemplateDto();\n messageTemplateDto.type = step.template.type;\n messageTemplateDto.content = step.template.content;\n messageTemplateDto.contentType = step.template.contentType;\n messageTemplateDto.cta = step.template.cta;\n messageTemplateDto.actor = step.template.actor;\n messageTemplateDto.variables = step.template.variables;\n messageTemplateDto._feedId = step.template._feedId;\n messageTemplateDto._layoutId = step.template._layoutId;\n messageTemplateDto.name = step.template.name;\n messageTemplateDto.subject = step.template.subject;\n messageTemplateDto.title = step.template.title;\n messageTemplateDto.preheader = step.template.preheader;\n messageTemplateDto.senderName = step.template.senderName;\n messageTemplateDto._creatorId = step.template._creatorId;\n\n responseDto.template = messageTemplateDto;\n }\n\n if (step.variants) {\n responseDto.variants = step.variants.map((variant) => convertStepToResponse(variant));\n }\n\n return responseDto;\n}\n\nfunction isDigestRegularMetadata(item: IWorkflowStepMetadata): item is IDigestRegularMetadata {\n return 'type' in item && (item.type === DigestTypeEnum.REGULAR || item.type === DigestTypeEnum.BACKOFF);\n}\n\nfunction isDigestTimedMetadata(item: IWorkflowStepMetadata): item is IDigestTimedMetadata {\n return 'type' in item && item.type === DigestTypeEnum.TIMED;\n}\n\nfunction mapDigest(\n digestData?:\n | (IWorkflowStepMetadata & {\n events?: any[];\n })\n | string\n | null\n): DigestMetadataDto | undefined {\n if (!digestData) {\n return undefined;\n }\n\n const digestItem =\n typeof digestData === 'string'\n ? (JSON.parse(digestData) as IWorkflowStepMetadata & {\n events?: any[];\n })\n : (digestData as IWorkflowStepMetadata & {\n events?: any[];\n });\n\n if (!digestItem) {\n return undefined;\n }\n\n // Type guarding and mapping based on the type of item\n if (isDigestRegularMetadata(digestItem)) {\n // If it's IDigestRegularMetadata\n return {\n digestKey: digestItem.digestKey,\n amount: digestItem.amount,\n unit: digestItem.unit,\n events: digestItem.events || [], // Default to an empty array if no events are provided\n type: digestItem.type, // Set the type as either REGULAR or BACKOFF\n backoff: digestItem.backoff,\n backoffAmount: digestItem.backoffAmount,\n backoffUnit: digestItem.backoffUnit,\n updateMode: digestItem.updateMode, // Set update mode if available\n };\n }\n if (isDigestTimedMetadata(digestItem)) {\n return {\n digestKey: digestItem.digestKey,\n amount: digestItem.amount,\n unit: digestItem.unit,\n events: digestItem.events || [], // Default to an empty array if no events are provided\n type: DigestTypeEnum.TIMED, // Set the type as TIMED\n timed: {\n atTime: digestItem.timed?.atTime,\n weekDays: digestItem.timed?.weekDays,\n monthDays: digestItem.timed?.monthDays,\n ordinal: digestItem.timed?.ordinal,\n ordinalValue: digestItem.timed?.ordinalValue,\n monthlyType: digestItem.timed?.monthlyType,\n cronExpression: digestItem.timed?.cronExpression,\n untilDate: digestItem.timed?.untilDate,\n },\n };\n }\n\n return undefined;\n}\n\nfunction mapJobToDto(item: JobFeedItem): ActivityNotificationJobResponseDto {\n return {\n _id: item._id,\n type: item.type as StepTypeEnum,\n digest: mapDigest(item.digest),\n executionDetails: item.executionDetails.map(convertExecutionDetail),\n step: convertStepToResponse(item.step),\n overrides: item.overrides,\n payload: item.payload,\n providerId: item.providerId as ProvidersIdEnum,\n status: item.status,\n updatedAt: item.updatedAt,\n scheduleExtensionsCount: item.scheduleExtensionsCount,\n };\n}\n\nfunction convertExecutionDetail(entity: ExecutionDetailFeedItem): ActivityNotificationExecutionDetailResponseDto {\n return {\n _id: entity._id,\n detail: entity.detail,\n isRetry: entity.isRetry,\n isTest: entity.isTest,\n providerId: entity.providerId as unknown as ProvidersIdEnum,\n source: entity.source,\n status: entity.status,\n raw: entity.raw || undefined,\n createdAt: entity.createdAt,\n };\n}\n"
},
{
"path": "apps/api/src/app/notifications/usecases/get-activity-graph-states/get-activity-graph-states.command.ts",
"content": "import { IsNumber, IsOptional } from 'class-validator';\nimport { EnvironmentWithUserCommand } from '../../../shared/commands/project.command';\n\nexport class GetActivityGraphStatsCommand extends EnvironmentWithUserCommand {\n @IsNumber()\n @IsOptional()\n days: number;\n}\n"
},
{
"path": "apps/api/src/app/notifications/usecases/get-activity-graph-states/get-activity-graph-states.usecase.ts",
"content": "import { Injectable } from '@nestjs/common';\nimport { NotificationRepository } from '@novu/dal';\nimport { subDays } from 'date-fns';\nimport { ActivityGraphStatesResponse } from '../../dtos/activity-graph-states-response.dto';\nimport { GetActivityGraphStatsCommand } from './get-activity-graph-states.command';\n\n@Injectable()\nexport class GetActivityGraphStats {\n constructor(private notificationRepository: NotificationRepository) {}\n\n async execute(command: GetActivityGraphStatsCommand): Promise {\n return await this.notificationRepository.getActivityGraphStats(\n subDays(new Date(), command.days),\n command.environmentId\n );\n }\n}\n"
},
{
"path": "apps/api/src/app/notifications/usecases/get-activity-stats/get-activity-stats.command.ts",
"content": "import { EnvironmentCommand } from '../../../shared/commands/project.command';\n\nexport class GetActivityStatsCommand extends EnvironmentCommand {}\n"
},
{
"path": "apps/api/src/app/notifications/usecases/get-activity-stats/get-activity-stats.usecase.ts",
"content": "import { Injectable } from '@nestjs/common';\nimport { NotificationRepository } from '@novu/dal';\nimport { ActivityStatsResponseDto } from '../../dtos/activity-stats-response.dto';\nimport { GetActivityStatsCommand } from './get-activity-stats.command';\n\n@Injectable()\nexport class GetActivityStats {\n constructor(private notificationRepository: NotificationRepository) {}\n\n async execute(command: GetActivityStatsCommand): Promise {\n const result = await this.notificationRepository.getStats(command.environmentId);\n\n return {\n weeklySent: result.weekly,\n monthlySent: result.monthly,\n };\n }\n}\n"
},
{
"path": "apps/api/src/app/notifications/usecases/get-activity-stats/index.ts",
"content": "export { GetActivityStatsCommand } from './get-activity-stats.command';\nexport { GetActivityStats } from './get-activity-stats.usecase';\n"
},
{
"path": "apps/api/src/app/notifications/usecases/index.ts",
"content": "import { GetActivity } from './get-activity/get-activity.usecase';\nimport { GetActivityFeed } from './get-activity-feed/get-activity-feed.usecase';\nimport { GetActivityGraphStats } from './get-activity-graph-states/get-activity-graph-states.usecase';\nimport { GetActivityStats } from './get-activity-stats';\n\nexport const USE_CASES = [\n GetActivityStats,\n GetActivityGraphStats,\n GetActivityFeed,\n GetActivity,\n //\n];\n"
},
{
"path": "apps/api/src/app/organization/dtos/create-organization.dto.ts",
"content": "import { ICreateOrganizationDto, JobTitleEnum, ProductUseCases } from '@novu/shared';\nimport { IsDefined, IsEnum, IsOptional, IsString } from 'class-validator';\n\nexport class CreateOrganizationDto implements ICreateOrganizationDto {\n @IsString()\n @IsDefined()\n name: string;\n\n @IsString()\n @IsOptional()\n logo?: string;\n\n @IsOptional()\n @IsEnum(JobTitleEnum)\n jobTitle?: JobTitleEnum;\n\n @IsString()\n @IsOptional()\n domain?: string;\n\n @IsOptional()\n language?: string[];\n}\n"
},
{
"path": "apps/api/src/app/organization/dtos/get-my-organization.dto.ts",
"content": "import { OrganizationEntity } from '@novu/dal';\n\nexport type IGetMyOrganizationDto = OrganizationEntity;\n"
},
{
"path": "apps/api/src/app/organization/dtos/get-organization-settings.dto.ts",
"content": "import { ApiProperty } from '@nestjs/swagger';\nimport { IsValidLocale } from '@novu/application-generic';\nimport { OrganizationEntity } from '@novu/dal';\nimport { IsArray, IsBoolean, IsOptional, IsString } from 'class-validator';\n\nexport class GetOrganizationSettingsDto {\n @ApiProperty({\n description: 'Remove Novu branding',\n example: false,\n })\n @IsBoolean()\n removeNovuBranding: boolean;\n\n @ApiProperty({\n description: 'Default locale',\n example: 'en_US',\n })\n @IsValidLocale()\n defaultLocale: string;\n\n @ApiProperty({\n description: 'Target locales',\n example: ['en_US', 'es_ES'],\n })\n @IsOptional()\n @IsArray()\n @IsString({ each: true })\n targetLocales?: string[];\n}\n"
},
{
"path": "apps/api/src/app/organization/dtos/get-organizations.dto.ts",
"content": "import { OrganizationEntity } from '@novu/dal';\n\nexport type IGetOrganizationsDto = OrganizationEntity[];\n"
},
{
"path": "apps/api/src/app/organization/dtos/member-response.dto.ts",
"content": "import { ApiProperty, ApiPropertyOptional } from '@nestjs/swagger';\nimport { MemberRoleEnum, MemberStatusEnum } from '@novu/shared';\nimport { IsDate, IsEnum, IsObject, IsString } from 'class-validator';\n\nexport class MemberUserDto {\n @ApiProperty()\n @IsString()\n _id: string;\n\n @ApiProperty()\n @IsString()\n firstName: string;\n\n @ApiProperty()\n @IsString()\n lastName: string;\n\n @ApiProperty()\n @IsString()\n email: string;\n}\n\nexport class MemberInviteDTO {\n @ApiProperty()\n @IsString()\n email: string;\n\n @ApiProperty()\n @IsString()\n token: string;\n\n @ApiProperty()\n @IsDate()\n invitationDate: Date;\n\n @ApiPropertyOptional()\n @IsDate()\n answerDate?: Date;\n\n @ApiProperty()\n @IsString()\n _inviterId: string;\n}\n\nexport class MemberResponseDto {\n @ApiProperty()\n @IsString()\n _id: string;\n\n @ApiProperty()\n @IsString()\n _userId: string;\n\n @ApiPropertyOptional()\n @IsObject()\n user?: MemberUserDto;\n\n @ApiPropertyOptional({ enum: MemberRoleEnum })\n @IsEnum(MemberRoleEnum)\n roles?: MemberRoleEnum;\n\n @ApiPropertyOptional()\n @IsObject()\n invite?: MemberInviteDTO;\n\n @ApiPropertyOptional({\n enum: { ...MemberStatusEnum },\n })\n @IsEnum(MemberStatusEnum)\n memberStatus?: MemberStatusEnum;\n\n @ApiProperty()\n @IsString()\n _organizationId: string;\n}\n"
},
{
"path": "apps/api/src/app/organization/dtos/organization-response.dto.ts",
"content": "import { ApiProperty, ApiPropertyOptional } from '@nestjs/swagger';\nimport { DirectionEnum, PartnerTypeEnum } from '@novu/dal';\nimport { IsArray, IsEnum, IsObject, IsString } from 'class-validator';\nimport { UpdateBrandingDetailsDto } from './update-branding-details.dto';\n\nexport class IPartnerConfigurationResponseDto {\n @ApiPropertyOptional()\n @IsArray()\n @IsString({ each: true })\n projectIds?: string[];\n\n @ApiProperty()\n @IsString()\n accessToken: string;\n\n @ApiProperty()\n @IsString()\n configurationId: string;\n\n @ApiPropertyOptional()\n @IsString()\n teamId: string;\n\n @ApiProperty({\n enum: PartnerTypeEnum,\n description: 'Partner Type Enum',\n })\n @IsEnum(PartnerTypeEnum)\n partnerType: PartnerTypeEnum;\n}\n\nexport class OrganizationBrandingResponseDto extends UpdateBrandingDetailsDto {\n @ApiPropertyOptional({\n enum: DirectionEnum,\n })\n @IsString()\n direction?: DirectionEnum;\n}\n\nexport class OrganizationResponseDto {\n @ApiProperty()\n @IsString()\n name: string;\n\n @ApiPropertyOptional()\n @IsString()\n logo?: string;\n\n @ApiProperty()\n @IsObject()\n branding: OrganizationBrandingResponseDto;\n\n @ApiPropertyOptional()\n @IsObject()\n partnerConfigurations: IPartnerConfigurationResponseDto[];\n}\n"
},
{
"path": "apps/api/src/app/organization/dtos/rename-organization.dto.ts",
"content": "import { IsDefined, IsString } from 'class-validator';\n\nexport class RenameOrganizationDto {\n @IsString()\n @IsDefined()\n name: string;\n}\n"
},
{
"path": "apps/api/src/app/organization/dtos/update-branding-details.dto.ts",
"content": "import { IsHexColor, IsOptional, IsString, IsUrl } from 'class-validator';\nimport { IsImageUrl } from '../../shared/validators/image.validator';\n\nconst environments = ['production', 'test'];\nconst protocols = environments.includes(process.env.NODE_ENV || '') ? ['https'] : ['http', 'https'];\n\nexport class UpdateBrandingDetailsDto {\n @IsUrl({\n require_protocol: true,\n protocols,\n require_tld: false,\n })\n @IsImageUrl({\n message: 'Logo must be a valid image URL with one of the following extensions: jpg, jpeg, png, gif, svg',\n })\n @IsOptional()\n logo: string;\n\n @IsOptional()\n @IsHexColor()\n color: string;\n\n @IsOptional()\n @IsHexColor()\n fontColor: string;\n\n @IsOptional()\n @IsHexColor()\n contentBackground: string;\n\n @IsOptional()\n @IsString()\n fontFamily?: string;\n}\n"
},
{
"path": "apps/api/src/app/organization/dtos/update-member-roles.dto.ts",
"content": "import { MemberRoleEnum } from '@novu/shared';\nimport { IsEnum } from 'class-validator';\n\nexport class UpdateMemberRolesDto {\n @IsEnum(MemberRoleEnum)\n role: MemberRoleEnum.OSS_ADMIN;\n}\n"
},
{
"path": "apps/api/src/app/organization/dtos/update-organization-settings.dto.ts",
"content": "import { ApiProperty } from '@nestjs/swagger';\nimport { IsValidLocale } from '@novu/application-generic';\nimport { IsArray, IsBoolean, IsOptional, IsString } from 'class-validator';\n\nexport class UpdateOrganizationSettingsDto {\n @ApiProperty({\n description: 'Enable or disable Novu branding',\n example: true,\n })\n @IsOptional()\n @IsBoolean()\n removeNovuBranding?: boolean;\n\n @ApiProperty({\n description: 'Default locale',\n example: 'en_US',\n })\n @IsOptional()\n @IsValidLocale()\n defaultLocale?: string;\n\n @ApiProperty({\n description: 'Target locales',\n example: ['en_US', 'es_ES'],\n })\n @IsOptional()\n @IsArray()\n @IsString({ each: true }) // TODO: validate locales\n targetLocales?: string[];\n}\n"
},
{
"path": "apps/api/src/app/organization/e2e/change-member-role.e2e.ts",
"content": "import { CommunityMemberRepository } from '@novu/dal';\nimport { MemberRoleEnum, MemberStatusEnum } from '@novu/shared';\nimport { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\nimport { describe } from 'mocha';\n\ndescribe('Change member role - /organizations/members/:memberId/role (PUT) #novu-v0-os', async () => {\n const memberRepository = new CommunityMemberRepository();\n let session: UserSession;\n let user2: UserSession;\n let user3: UserSession;\n\n beforeEach(async () => {\n session = new UserSession();\n await session.initialize();\n\n user2 = new UserSession();\n await user2.initialize({\n noOrganization: true,\n });\n\n user3 = new UserSession();\n await user3.initialize({\n noOrganization: true,\n });\n });\n\n // Currently skipped until we implement role management\n it.skip('should update admin to member', async () => {\n await memberRepository.addMember(session.organization._id, {\n _userId: user2.user._id,\n invite: null,\n roles: [MemberRoleEnum.OSS_ADMIN],\n memberStatus: MemberStatusEnum.ACTIVE,\n });\n\n const member = await memberRepository.findMemberByUserId(session.organization._id, user2.user._id);\n const { body } = await session.testAgent.put(`/v1/organizations/members/${member._id}/roles`).send({\n role: MemberRoleEnum.OSS_MEMBER,\n });\n\n expect(body.data.roles.length).to.equal(1);\n expect(body.data.roles[0]).to.equal(MemberRoleEnum.OSS_MEMBER);\n });\n\n it('should update member to admin', async () => {\n await memberRepository.addMember(session.organization._id, {\n _userId: user3.user._id,\n invite: null,\n roles: [MemberRoleEnum.OSS_MEMBER],\n memberStatus: MemberStatusEnum.ACTIVE,\n });\n\n const member = await memberRepository.findMemberByUserId(session.organization._id, user3.user._id);\n\n const { body } = await session.testAgent.put(`/v1/organizations/members/${member._id}/roles`).send({\n role: MemberRoleEnum.OSS_ADMIN,\n });\n\n expect(body.data.roles.length).to.equal(1);\n expect(body.data.roles.includes(MemberRoleEnum.OSS_ADMIN)).to.be.ok;\n expect(body.data.roles.includes(MemberRoleEnum.OSS_MEMBER)).not.to.be.ok;\n });\n});\n"
},
{
"path": "apps/api/src/app/organization/e2e/create-organization.e2e.ts",
"content": "import {\n CommunityMemberRepository,\n CommunityOrganizationRepository,\n CommunityUserRepository,\n EnvironmentRepository,\n IntegrationRepository,\n} from '@novu/dal';\nimport {\n ApiServiceLevelEnum,\n ChannelTypeEnum,\n ChatProviderIdEnum,\n EmailProviderIdEnum,\n ICreateOrganizationDto,\n InAppProviderIdEnum,\n JobTitleEnum,\n MemberRoleEnum,\n SmsProviderIdEnum,\n} from '@novu/shared';\nimport { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\n\ndescribe('Create Organization - /organizations (POST) #novu-v0-os', async () => {\n let session: UserSession;\n const organizationRepository = new CommunityOrganizationRepository();\n const userRepository = new CommunityUserRepository();\n const memberRepository = new CommunityMemberRepository();\n const integrationRepository = new IntegrationRepository();\n const environmentRepository = new EnvironmentRepository();\n\n before(async () => {\n session = new UserSession();\n await session.initialize({\n noOrganization: true,\n });\n });\n\n describe('Valid Creation', () => {\n it('should add the user as admin', async () => {\n const { body } = await session.testAgent\n .post('/v1/organizations')\n .send({\n name: 'Test Org 2',\n })\n .expect(201);\n const dbOrganization = await organizationRepository.findById(body.data._id);\n\n const members = await memberRepository.getOrganizationMembers(dbOrganization?._id as string);\n\n expect(members.length).to.eq(1);\n expect(members[0]._userId).to.eq(session.user._id);\n expect(members[0].roles[0]).to.eq(MemberRoleEnum.OSS_ADMIN);\n });\n\n it('should create organization with correct name', async () => {\n const demoOrganization = {\n name: 'Hello Org',\n };\n const { body } = await session.testAgent.post('/v1/organizations').send(demoOrganization).expect(201);\n\n expect(body.data.name).to.eq(demoOrganization.name);\n });\n\n it('should not create organization with no name', async () => {\n await session.testAgent.post('/v1/organizations').send({}).expect(400);\n });\n\n it('should create organization with apiServiceLevel of free by default', async () => {\n const testOrganization = {\n name: 'Free Org',\n };\n\n const { body } = await session.testAgent.post('/v1/organizations').send(testOrganization).expect(201);\n const dbOrganization = await organizationRepository.findById(body.data._id);\n\n expect(dbOrganization?.apiServiceLevel).to.eq(ApiServiceLevelEnum.FREE);\n });\n\n it('should create organization with questionnaire data', async () => {\n const testOrganization: ICreateOrganizationDto = {\n name: 'Org Name',\n domain: 'org.com',\n };\n\n const { body } = await session.testAgent.post('/v1/organizations').send(testOrganization).expect(201);\n const dbOrganization = await organizationRepository.findById(body.data._id);\n\n expect(dbOrganization?.name).to.eq(testOrganization.name);\n expect(dbOrganization?.domain).to.eq(testOrganization.domain);\n });\n\n it('should update user job title on organization creation', async () => {\n const testOrganization: ICreateOrganizationDto = {\n name: 'Org Name',\n jobTitle: JobTitleEnum.PRODUCT_MANAGER,\n };\n\n await session.testAgent.post('/v1/organizations').send(testOrganization).expect(201);\n const user = await userRepository.findById(session.user._id);\n\n expect(user?.jobTitle).to.eq(testOrganization.jobTitle);\n });\n\n it('should create organization with built in Novu integrations and set them as primary', async () => {\n const testOrganization: ICreateOrganizationDto = {\n name: 'Org Name',\n };\n\n const { body } = await session.testAgent.post('/v1/organizations').send(testOrganization).expect(201);\n const integrations = await integrationRepository.find({ _organizationId: body.data._id });\n const environments = await environmentRepository.find({ _organizationId: body.data._id });\n const productionEnv = environments.find((e) => e.name === 'Production');\n const developmentEnv = environments.find((e) => e.name === 'Development');\n const novuEmailIntegration = integrations.filter(\n (i) => i.active && i.channel === ChannelTypeEnum.EMAIL && i.providerId === EmailProviderIdEnum.Novu\n );\n const novuSmsIntegration = integrations.filter(\n (i) => i.active && i.channel === ChannelTypeEnum.SMS && i.providerId === SmsProviderIdEnum.Novu\n );\n const novuChatIntegration = integrations.filter(\n (i) => i.active && i.channel === ChannelTypeEnum.CHAT && i.providerId === ChatProviderIdEnum.Novu\n );\n const novuInAppIntegration = integrations.filter(\n (i) => i.active && i.channel === ChannelTypeEnum.IN_APP && i.providerId === InAppProviderIdEnum.Novu\n );\n const novuEmailIntegrationProduction = novuEmailIntegration.filter(\n (el) => el._environmentId === productionEnv?._id\n );\n const novuEmailIntegrationDevelopment = novuEmailIntegration.filter(\n (el) => el._environmentId === developmentEnv?._id\n );\n const novuSmsIntegrationProduction = novuSmsIntegration.filter((el) => el._environmentId === productionEnv?._id);\n const novuSmsIntegrationDevelopment = novuSmsIntegration.filter(\n (el) => el._environmentId === developmentEnv?._id\n );\n const novuInAppIntegrationProduction = novuInAppIntegration.filter(\n (el) => el._environmentId === productionEnv?._id\n );\n const novuInAppIntegrationDevelopment = novuInAppIntegration.filter(\n (el) => el._environmentId === developmentEnv?._id\n );\n\n expect(integrations.length).to.eq(6);\n expect(novuEmailIntegration?.length).to.eq(2);\n expect(novuSmsIntegration?.length).to.eq(2);\n expect(novuChatIntegration?.length).to.eq(1);\n expect(novuInAppIntegration?.length).to.eq(2);\n\n expect(novuEmailIntegrationProduction.length).to.eq(1);\n expect(novuSmsIntegrationProduction.length).to.eq(1);\n expect(novuInAppIntegrationProduction.length).to.eq(1);\n expect(novuEmailIntegrationDevelopment.length).to.eq(1);\n expect(novuSmsIntegrationDevelopment.length).to.eq(1);\n expect(novuInAppIntegrationDevelopment.length).to.eq(1);\n\n expect(novuEmailIntegrationProduction[0].primary).to.eq(true);\n expect(novuSmsIntegrationProduction[0].primary).to.eq(true);\n expect(novuEmailIntegrationDevelopment[0].primary).to.eq(true);\n expect(novuSmsIntegrationDevelopment[0].primary).to.eq(true);\n });\n\n it('when Novu Email credentials are not set it should not create Novu Email integration', async () => {\n const oldNovuEmailIntegrationApiKey = process.env.NOVU_EMAIL_INTEGRATION_API_KEY;\n process.env.NOVU_EMAIL_INTEGRATION_API_KEY = '';\n const testOrganization: ICreateOrganizationDto = {\n name: 'Org Name',\n };\n\n const { body } = await session.testAgent.post('/v1/organizations').send(testOrganization).expect(201);\n const integrations = await integrationRepository.find({ _organizationId: body.data._id });\n const environments = await environmentRepository.find({ _organizationId: body.data._id });\n const productionEnv = environments.find((e) => e.name === 'Production');\n const developmentEnv = environments.find((e) => e.name === 'Development');\n const novuSmsIntegration = integrations.filter(\n (i) => i.active && i.name === 'Novu SMS' && i.providerId === SmsProviderIdEnum.Novu\n );\n\n expect(integrations.length).to.eq(4);\n expect(novuSmsIntegration?.length).to.eq(2);\n expect(novuSmsIntegration.filter((el) => el._environmentId === productionEnv?._id).length).to.eq(1);\n expect(novuSmsIntegration.filter((el) => el._environmentId === developmentEnv?._id).length).to.eq(1);\n process.env.NOVU_EMAIL_INTEGRATION_API_KEY = oldNovuEmailIntegrationApiKey;\n });\n\n it('when Novu SMS credentials are not set it should not create Novu SMS integration', async () => {\n const oldNovuSmsIntegrationAccountSid = process.env.NOVU_SMS_INTEGRATION_ACCOUNT_SID;\n process.env.NOVU_SMS_INTEGRATION_ACCOUNT_SID = '';\n const testOrganization: ICreateOrganizationDto = {\n name: 'Org Name',\n };\n\n const { body } = await session.testAgent.post('/v1/organizations').send(testOrganization).expect(201);\n const integrations = await integrationRepository.find({ _organizationId: body.data._id });\n const environments = await environmentRepository.find({ _organizationId: body.data._id });\n const productionEnv = environments.find((e) => e.name === 'Production');\n const developmentEnv = environments.find((e) => e.name === 'Development');\n const novuEmailIntegrations = integrations.filter(\n (i) => i.active && i.name === 'Novu Email' && i.providerId === EmailProviderIdEnum.Novu\n );\n\n expect(integrations.length).to.eq(4);\n expect(novuEmailIntegrations?.length).to.eq(2);\n expect(novuEmailIntegrations.filter((el) => el._environmentId === productionEnv?._id).length).to.eq(1);\n expect(novuEmailIntegrations.filter((el) => el._environmentId === developmentEnv?._id).length).to.eq(1);\n process.env.NOVU_SMS_INTEGRATION_ACCOUNT_SID = oldNovuSmsIntegrationAccountSid;\n });\n\n it('when Novu Chat credentials are not set it should not create Novu Chat integration', async () => {\n // todo\n });\n });\n});\n"
},
{
"path": "apps/api/src/app/organization/e2e/get-members.e2e.ts",
"content": "import { CommunityMemberRepository } from '@novu/dal';\nimport { MemberRoleEnum } from '@novu/shared';\nimport { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\n\ndescribe('Get members - /organization/members (GET) #novu-v0-os', async () => {\n let session: UserSession;\n let otherSession: UserSession;\n\n const memberRepository = new CommunityMemberRepository();\n\n before(async () => {\n session = new UserSession();\n await session.initialize();\n\n otherSession = new UserSession();\n await otherSession.initialize({\n noOrganization: true,\n });\n\n await session.testAgent\n .post('/v1/invites/bulk')\n .send({\n invitees: [\n {\n email: 'dddd@asdas.com',\n role: MemberRoleEnum.OSS_ADMIN,\n },\n ],\n })\n .expect(201);\n\n const members = await memberRepository.getOrganizationMembers(session.organization._id);\n const invitee = members.find((i) => !i._userId);\n\n await otherSession.testAgent.post(`/v1/invites/${invitee.invite.token}/accept`).expect(201);\n\n otherSession.organization = session.organization;\n await otherSession.fetchJWT();\n });\n\n it('should see emails of all members as admin', async () => {\n const { body } = await session.testAgent.get('/v1/organizations/members').expect(200);\n\n expect(JSON.stringify(body.data)).to.include('dddd@asdas.com');\n expect(JSON.stringify(body.data)).to.include(session.user.firstName);\n });\n});\n"
},
{
"path": "apps/api/src/app/organization/e2e/get-my-organization.e2e.ts",
"content": "import { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\n\ndescribe('Get my organization - /organizations/me (GET) #novu-v0-os', async () => {\n let session: UserSession;\n\n before(async () => {\n session = new UserSession();\n await session.initialize();\n });\n\n describe('Get organization profile', () => {\n it('should return the correct organization', async () => {\n const { body } = await session.testAgent.get('/v1/organizations/me').expect(200);\n\n expect(body.data._id).to.eq(session.organization._id);\n });\n });\n});\n"
},
{
"path": "apps/api/src/app/organization/e2e/get-organizations.e2e.ts",
"content": "import { CommunityMemberRepository, OrganizationEntity } from '@novu/dal';\nimport { MemberRoleEnum } from '@novu/shared';\nimport { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\n\ndescribe('Get organizations - /organizations (GET) #novu-v0-os', async () => {\n let session: UserSession;\n let otherSession: UserSession;\n let thirdSession: UserSession;\n\n let thirdOldOrganization: OrganizationEntity;\n\n const memberRepository = new CommunityMemberRepository();\n\n before(async () => {\n session = new UserSession();\n await session.initialize();\n\n otherSession = new UserSession();\n await otherSession.initialize();\n\n thirdSession = new UserSession();\n await thirdSession.initialize();\n\n await session.testAgent\n .post('/v1/invites/bulk')\n .send({\n invitees: [\n {\n email: 'dddd@asdas.com',\n role: MemberRoleEnum.OSS_MEMBER,\n },\n ],\n })\n .expect(201);\n\n const members = await memberRepository.getOrganizationMembers(session.organization._id);\n const invitee = members.find((i) => !i._userId);\n\n thirdOldOrganization = thirdSession.organization;\n\n await thirdSession.testAgent.post(`/v1/invites/${invitee.invite.token}/accept`).expect(201);\n });\n\n it('should see all organizations that you are a part of', async () => {\n const { body } = await thirdSession.testAgent.get('/v1/organizations').expect(200);\n\n expect(JSON.stringify(body.data)).to.include(session.organization.name);\n expect(JSON.stringify(body.data)).to.include(thirdSession.organization.name);\n expect(JSON.stringify(body.data)).to.include(thirdOldOrganization.name);\n expect(JSON.stringify(body.data)).to.not.include(otherSession.organization.name);\n });\n});\n"
},
{
"path": "apps/api/src/app/organization/e2e/remove-member.e2e.ts",
"content": "import { CommunityMemberRepository, EnvironmentRepository, MemberEntity } from '@novu/dal';\nimport { MemberRoleEnum, MemberStatusEnum } from '@novu/shared';\nimport { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\nimport { describe } from 'mocha';\n\ndescribe('Remove organization member - /organizations/members/:memberId (DELETE) #novu-v0-os', async () => {\n let session: UserSession;\n const memberRepository = new CommunityMemberRepository();\n const environmentRepository = new EnvironmentRepository();\n let user2: UserSession;\n let user3: UserSession;\n\n beforeEach(async () => {\n session = new UserSession();\n await session.initialize();\n\n user2 = new UserSession();\n await user2.initialize({\n noOrganization: true,\n });\n\n user3 = new UserSession();\n await user3.initialize({\n noOrganization: true,\n });\n\n await memberRepository.addMember(session.organization._id, {\n _userId: user2.user._id,\n invite: null,\n roles: [MemberRoleEnum.OSS_ADMIN],\n memberStatus: MemberStatusEnum.ACTIVE,\n });\n\n await memberRepository.addMember(session.organization._id, {\n _userId: user3.user._id,\n invite: null,\n roles: [MemberRoleEnum.OSS_ADMIN],\n memberStatus: MemberStatusEnum.ACTIVE,\n });\n\n user2.organization = session.organization;\n user3.organization = session.organization;\n });\n\n it('should switch the apiKey association when api key creator removed', async () => {\n const members: MemberEntity[] = await getOrganizationMembers();\n const originalCreator = members.find((i) => i._userId === session.user._id);\n await user2.fetchJWT();\n\n expect(session.environment.apiKeys[0]._userId).to.equal(session.user._id);\n const { body } = await user2.testAgent.delete(`/v1/organizations/members/${originalCreator._id}`);\n expect(body.data._id).to.equal(originalCreator._id);\n\n const membersAfterRemoval: MemberEntity[] = await getOrganizationMembers(user2);\n const originalCreatorAfterRemoval = membersAfterRemoval.find((i) => i._userId === originalCreator.user._id);\n expect(originalCreatorAfterRemoval).to.not.be.ok;\n\n const environment = await environmentRepository.findOne({ _id: session.environment._id });\n expect(environment.apiKeys[0]._userId).to.not.equal(session.user._id);\n });\n\n it('should remove the member by his id', async () => {\n const members: MemberEntity[] = await getOrganizationMembers();\n const user2Member = members.find((i) => i._userId === user2.user._id);\n\n const { body } = await session.testAgent.delete(`/v1/organizations/members/${user2Member._id}`).expect(200);\n\n expect(body.data._id).to.equal(user2Member._id);\n\n const membersAfterRemoval: MemberEntity[] = await getOrganizationMembers();\n const user2Removed = membersAfterRemoval.find((i) => i._userId === user2.user._id);\n\n expect(user2Removed).to.not.be.ok;\n\n /**\n * The API Key owner should not be updated if non creator was removed\n */\n const environment = await environmentRepository.findOne({ _id: session.environment._id });\n expect(environment.apiKeys[0]._userId).to.equal(session.user._id);\n });\n\n async function getOrganizationMembers(sessionToUser = session) {\n const { body } = await sessionToUser.testAgent.get('/v1/organizations/members');\n\n return body.data;\n }\n});\n"
},
{
"path": "apps/api/src/app/organization/e2e/rename-organization.e2e.ts",
"content": "import { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\n\ndescribe('Rename Organization - /organizations (PATCH) #novu-v0-os', () => {\n let session: UserSession;\n\n beforeEach(async () => {\n session = new UserSession();\n await session.initialize();\n });\n\n it('should rename the organization', async () => {\n const payload = {\n name: 'Liberty Powers',\n };\n\n await session.testAgent.patch('/v1/organizations').send(payload);\n\n const { body } = await session.testAgent.get('/v1/organizations/me').expect(200);\n const organization = body.data;\n\n expect(organization?.name).to.equal(payload.name);\n });\n});\n"
},
{
"path": "apps/api/src/app/organization/e2e/update-branding-details.e2e.ts",
"content": "import { processTestAgentExpectedStatusCode, UserSession } from '@novu/testing';\nimport { expect } from 'chai';\n\ndescribe('Update Branding Details - /organizations/branding (PUT) #novu-v0-os', () => {\n let session: UserSession;\n\n beforeEach(async () => {\n session = new UserSession();\n await session.initialize();\n });\n\n it('should update organization name only', async () => {\n const payload = {\n name: 'New Name',\n };\n\n await session.testAgent.patch('/v1/organizations').send(payload).expect(processTestAgentExpectedStatusCode(200));\n const { body } = await session.testAgent.get('/v1/organizations/me').expect(200);\n const organization = body.data;\n\n expect(organization?.name).to.equal(payload.name);\n expect(organization?.logo).to.equal(session.organization.logo);\n });\n\n it('should update the branding details', async () => {\n const payload = {\n color: '#fefefe',\n fontColor: '#f4f4f4',\n contentBackground: '#fefefe',\n fontFamily: 'Nunito',\n logo: 'https://s3.us-east-1.amazonaws.com/novu-app-bucket/2/1/3.png',\n };\n\n const result = await session.testAgent\n .put('/v1/organizations/branding')\n .send(payload)\n .expect(processTestAgentExpectedStatusCode(200));\n\n const { body } = await session.testAgent.get('/v1/organizations/me').expect(200);\n const organization = body.data;\n\n expect(organization?.branding.color).to.equal(payload.color);\n expect(organization?.branding.logo).to.equal(payload.logo);\n expect(organization?.branding.fontColor).to.equal(payload.fontColor);\n expect(organization?.branding.fontFamily).to.equal(payload.fontFamily);\n expect(organization?.branding.contentBackground).to.equal(payload.contentBackground);\n });\n\n it('logo should be an https protocol', async () => {\n const payload = {\n logo: 'http://s3.us-east-1.amazonaws.com/novu-app-bucket/2/1/3.png',\n };\n\n const result = await session.testAgent.put('/v1/organizations/branding').send(payload).expect(400);\n });\n\n ['png', 'jpg', 'jpeg', 'gif', 'svg'].forEach((extension) => {\n it(`should update if logo is a valid image URL with ${extension} extension`, async () => {\n const payload = {\n logo: `https://s3.us-east-1.amazonaws.com/novu-app-bucket/2/1/3.${extension}`,\n };\n\n const result = await session.testAgent\n .put('/v1/organizations/branding')\n .send(payload)\n .expect(processTestAgentExpectedStatusCode(200));\n });\n });\n\n ['exe', 'zip'].forEach((extension) => {\n it(`should fail to update if logo is a valid image URL with ${extension} extension`, async () => {\n const payload = {\n logo: `https://s3.us-east-1.amazonaws.com/novu-app-bucket/2/1/3.${extension}`,\n };\n\n const result = await session.testAgent.put('/v1/organizations/branding').send(payload).expect(400);\n });\n });\n});\n"
},
{
"path": "apps/api/src/app/organization/e2e/update-organization-settings.e2e.ts",
"content": "import { CommunityOrganizationRepository } from '@novu/dal';\nimport { ApiServiceLevelEnum } from '@novu/shared';\nimport { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\n\ndescribe('Update Organization Settings - /organizations/settings (PATCH) #novu-v2', () => {\n let session: UserSession;\n let organizationRepository: CommunityOrganizationRepository;\n\n beforeEach(async () => {\n session = new UserSession();\n await session.initialize();\n organizationRepository = new CommunityOrganizationRepository();\n });\n\n it('should allow updating removeNovuBranding for PRO tier organizations', async () => {\n await organizationRepository.update(\n { _id: session.organization._id },\n { apiServiceLevel: ApiServiceLevelEnum.PRO }\n );\n\n const payload = { removeNovuBranding: true };\n\n const { body } = await session.testAgent.patch('/v1/organizations/settings').send(payload).expect(200);\n\n expect(body.data.removeNovuBranding).to.equal(true);\n });\n\n it('should block branding updates for free tier organizations', async () => {\n await organizationRepository.update(\n { _id: session.organization._id },\n { apiServiceLevel: ApiServiceLevelEnum.FREE }\n );\n\n const payload = { removeNovuBranding: true };\n\n const { body } = await session.testAgent.patch('/v1/organizations/settings').send(payload).expect(402);\n\n expect(body.message).to.include('Removing Novu branding is not allowed on the free plan');\n });\n\n it('should allow free tier organizations to call endpoint without branding changes', async () => {\n await organizationRepository.update(\n { _id: session.organization._id },\n { apiServiceLevel: ApiServiceLevelEnum.FREE }\n );\n\n const payload = {};\n\n const { body } = await session.testAgent.patch('/v1/organizations/settings').send(payload).expect(200);\n\n expect(body.data).to.have.property('removeNovuBranding');\n expect(typeof body.data.removeNovuBranding).to.equal('boolean');\n });\n});\n"
},
{
"path": "apps/api/src/app/organization/ee.organization.controller.ts",
"content": "import { Body, ClassSerializerInterceptor, Controller, Get, Patch, Put, UseInterceptors } from '@nestjs/common';\nimport { ApiExcludeController, ApiOperation, ApiTags } from '@nestjs/swagger';\nimport { ExternalApiAccessible, RequirePermissions } from '@novu/application-generic';\nimport { PermissionsEnum, UserSessionData } from '@novu/shared';\nimport { RequireAuthentication } from '../auth/framework/auth.decorator';\nimport { ApiCommonResponses, ApiResponse } from '../shared/framework/response.decorator';\nimport { UserSession } from '../shared/framework/user.decorator';\nimport { IGetMyOrganizationDto } from './dtos/get-my-organization.dto';\nimport { GetOrganizationSettingsDto } from './dtos/get-organization-settings.dto';\nimport { OrganizationBrandingResponseDto, OrganizationResponseDto } from './dtos/organization-response.dto';\nimport { RenameOrganizationDto } from './dtos/rename-organization.dto';\nimport { UpdateBrandingDetailsDto } from './dtos/update-branding-details.dto';\nimport { UpdateOrganizationSettingsDto } from './dtos/update-organization-settings.dto';\nimport { GetMyOrganizationCommand } from './usecases/get-my-organization/get-my-organization.command';\nimport { GetMyOrganization } from './usecases/get-my-organization/get-my-organization.usecase';\nimport { GetOrganizationSettingsCommand } from './usecases/get-organization-settings/get-organization-settings.command';\nimport { GetOrganizationSettings } from './usecases/get-organization-settings/get-organization-settings.usecase';\nimport { RenameOrganization } from './usecases/rename-organization/rename-organization.usecase';\nimport { RenameOrganizationCommand } from './usecases/rename-organization/rename-organization-command';\nimport { UpdateBrandingDetailsCommand } from './usecases/update-branding-details/update-branding-details.command';\nimport { UpdateBrandingDetails } from './usecases/update-branding-details/update-branding-details.usecase';\nimport { UpdateOrganizationSettingsCommand } from './usecases/update-organization-settings/update-organization-settings.command';\nimport { UpdateOrganizationSettings } from './usecases/update-organization-settings/update-organization-settings.usecase';\n\n@Controller('/organizations')\n@UseInterceptors(ClassSerializerInterceptor)\n@RequireAuthentication()\n@ApiTags('Organizations')\n@ApiCommonResponses()\n@ApiExcludeController()\nexport class EEOrganizationController {\n constructor(\n private updateBrandingDetailsUsecase: UpdateBrandingDetails,\n private getMyOrganizationUsecase: GetMyOrganization,\n private renameOrganizationUsecase: RenameOrganization,\n private getOrganizationSettingsUsecase: GetOrganizationSettings,\n private updateOrganizationSettingsUsecase: UpdateOrganizationSettings\n ) {}\n\n /**\n * @deprecated - used in v1 legacy web\n */\n @Get('/me')\n @ApiResponse(OrganizationResponseDto)\n @ApiOperation({\n summary: 'Fetch current organization details',\n })\n async getMyOrganization(@UserSession() user: UserSessionData): Promise {\n const command = GetMyOrganizationCommand.create({\n userId: user._id,\n id: user.organizationId,\n });\n\n return await this.getMyOrganizationUsecase.execute(command);\n }\n\n /**\n * @deprecated - used in v1 legacy web\n */\n @Put('/branding')\n @ExternalApiAccessible()\n @ApiResponse(OrganizationBrandingResponseDto)\n @ApiOperation({\n summary: 'Update organization branding details',\n })\n async updateBrandingDetails(@UserSession() user: UserSessionData, @Body() body: UpdateBrandingDetailsDto) {\n return await this.updateBrandingDetailsUsecase.execute(\n UpdateBrandingDetailsCommand.create({\n logo: body.logo,\n color: body.color,\n userId: user._id,\n id: user.organizationId,\n fontColor: body.fontColor,\n fontFamily: body.fontFamily,\n contentBackground: body.contentBackground,\n })\n );\n }\n\n /**\n * @deprecated - used in v1 legacy web\n */\n @Patch('/')\n @ExternalApiAccessible()\n @ApiResponse(RenameOrganizationDto)\n @ApiOperation({\n summary: 'Rename organization name',\n })\n async renameOrganization(@UserSession() user: UserSessionData, @Body() body: RenameOrganizationDto) {\n return await this.renameOrganizationUsecase.execute(\n RenameOrganizationCommand.create({\n name: body.name,\n userId: user._id,\n id: user.organizationId,\n })\n );\n }\n\n @Get('/settings')\n @ExternalApiAccessible()\n @ApiResponse(GetOrganizationSettingsDto)\n @ApiOperation({\n summary: 'Get organization settings',\n })\n @RequirePermissions(PermissionsEnum.ORG_SETTINGS_READ)\n async getSettings(@UserSession() user: UserSessionData) {\n return await this.getOrganizationSettingsUsecase.execute(\n GetOrganizationSettingsCommand.create({\n organizationId: user.organizationId,\n })\n );\n }\n\n @Patch('/settings')\n @ApiResponse(UpdateOrganizationSettingsDto)\n @ExternalApiAccessible()\n @ApiOperation({\n summary: 'Update organization settings',\n })\n @RequirePermissions(PermissionsEnum.ORG_SETTINGS_WRITE)\n async updateSettings(@UserSession() user: UserSessionData, @Body() body: UpdateOrganizationSettingsDto) {\n return await this.updateOrganizationSettingsUsecase.execute(\n UpdateOrganizationSettingsCommand.create({\n userId: user._id,\n organizationId: user.organizationId,\n removeNovuBranding: body.removeNovuBranding,\n defaultLocale: body.defaultLocale,\n targetLocales: body.targetLocales,\n })\n );\n }\n}\n"
},
{
"path": "apps/api/src/app/organization/organization.controller.ts",
"content": "import {\n Body,\n ClassSerializerInterceptor,\n Controller,\n Delete,\n Get,\n Param,\n Patch,\n Post,\n Put,\n UseInterceptors,\n} from '@nestjs/common';\nimport { ApiExcludeEndpoint, ApiOperation, ApiParam, ApiTags } from '@nestjs/swagger';\nimport { ApiExcludeController } from '@nestjs/swagger/dist/decorators/api-exclude-controller.decorator';\nimport { OrganizationEntity } from '@novu/dal';\nimport { MemberRoleEnum, UserSessionData } from '@novu/shared';\nimport { RequireAuthentication } from '../auth/framework/auth.decorator';\nimport { ExternalApiAccessible } from '../auth/framework/external-api.decorator';\nimport { ApiCommonResponses, ApiResponse } from '../shared/framework/response.decorator';\nimport { UserSession } from '../shared/framework/user.decorator';\nimport { CreateOrganizationDto } from './dtos/create-organization.dto';\nimport { IGetMyOrganizationDto } from './dtos/get-my-organization.dto';\nimport { IGetOrganizationsDto } from './dtos/get-organizations.dto';\nimport { MemberResponseDto } from './dtos/member-response.dto';\nimport { OrganizationBrandingResponseDto, OrganizationResponseDto } from './dtos/organization-response.dto';\nimport { RenameOrganizationDto } from './dtos/rename-organization.dto';\nimport { UpdateBrandingDetailsDto } from './dtos/update-branding-details.dto';\nimport { UpdateMemberRolesDto } from './dtos/update-member-roles.dto';\nimport { CreateOrganizationCommand } from './usecases/create-organization/create-organization.command';\nimport { CreateOrganization } from './usecases/create-organization/create-organization.usecase';\nimport { GetMyOrganizationCommand } from './usecases/get-my-organization/get-my-organization.command';\nimport { GetMyOrganization } from './usecases/get-my-organization/get-my-organization.usecase';\nimport { GetOrganizationsCommand } from './usecases/get-organizations/get-organizations.command';\nimport { GetOrganizations } from './usecases/get-organizations/get-organizations.usecase';\nimport { ChangeMemberRoleCommand } from './usecases/membership/change-member-role/change-member-role.command';\nimport { ChangeMemberRole } from './usecases/membership/change-member-role/change-member-role.usecase';\nimport { GetMembersCommand } from './usecases/membership/get-members/get-members.command';\nimport { GetMembers } from './usecases/membership/get-members/get-members.usecase';\nimport { RemoveMemberCommand } from './usecases/membership/remove-member/remove-member.command';\nimport { RemoveMember } from './usecases/membership/remove-member/remove-member.usecase';\nimport { RenameOrganization } from './usecases/rename-organization/rename-organization.usecase';\nimport { RenameOrganizationCommand } from './usecases/rename-organization/rename-organization-command';\nimport { UpdateBrandingDetailsCommand } from './usecases/update-branding-details/update-branding-details.command';\nimport { UpdateBrandingDetails } from './usecases/update-branding-details/update-branding-details.usecase';\n\n@Controller('/organizations')\n@UseInterceptors(ClassSerializerInterceptor)\n@RequireAuthentication()\n@ApiTags('Organizations')\n@ApiCommonResponses()\n@ApiExcludeController()\nexport class OrganizationController {\n constructor(\n private createOrganizationUsecase: CreateOrganization,\n private getMembers: GetMembers,\n private removeMemberUsecase: RemoveMember,\n private changeMemberRoleUsecase: ChangeMemberRole,\n private updateBrandingDetailsUsecase: UpdateBrandingDetails,\n private getOrganizationsUsecase: GetOrganizations,\n private getMyOrganizationUsecase: GetMyOrganization,\n private renameOrganizationUsecase: RenameOrganization\n ) {}\n\n @Post('/')\n @ExternalApiAccessible()\n @ApiResponse(OrganizationResponseDto, 201)\n @ApiOperation({\n summary: 'Create an organization',\n })\n async createOrganization(\n @UserSession() user: UserSessionData,\n @Body() body: CreateOrganizationDto\n ): Promise {\n return await this.createOrganizationUsecase.execute(\n CreateOrganizationCommand.create({\n userId: user._id,\n logo: body.logo,\n name: body.name,\n jobTitle: body.jobTitle,\n domain: body.domain,\n language: body.language,\n })\n );\n }\n\n @Get('/')\n @ExternalApiAccessible()\n @ApiResponse(OrganizationResponseDto, 200, true)\n @ApiOperation({\n summary: 'Fetch all organizations',\n })\n async listOrganizations(@UserSession() user: UserSessionData): Promise {\n const command = GetOrganizationsCommand.create({\n userId: user._id,\n });\n\n return await this.getOrganizationsUsecase.execute(command);\n }\n\n @Get('/me')\n @ExternalApiAccessible()\n @ApiResponse(OrganizationResponseDto)\n @ApiOperation({\n summary: 'Fetch current organization details',\n })\n async getSelfOrganizationData(@UserSession() user: UserSessionData): Promise {\n const command = GetMyOrganizationCommand.create({\n userId: user._id,\n id: user.organizationId,\n });\n\n return await this.getMyOrganizationUsecase.execute(command);\n }\n @Delete('/members/:memberId')\n @ExternalApiAccessible()\n @ApiResponse(MemberResponseDto)\n @ApiOperation({\n summary: 'Remove a member from organization using memberId',\n })\n @ApiParam({ name: 'memberId', type: String, required: true })\n async remove(@UserSession() user: UserSessionData, @Param('memberId') memberId: string) {\n return await this.removeMemberUsecase.execute(\n RemoveMemberCommand.create({\n userId: user._id,\n organizationId: user.organizationId,\n memberId,\n })\n );\n }\n @Put('/members/:memberId/roles')\n @ExternalApiAccessible()\n @ApiExcludeEndpoint()\n @ApiResponse(MemberResponseDto)\n @ApiOperation({\n summary: 'Update a member role to admin',\n })\n @ApiParam({ name: 'memberId', type: String, required: true })\n async updateMemberRoles(\n @UserSession() user: UserSessionData,\n @Param('memberId') memberId: string,\n @Body() body: UpdateMemberRolesDto\n ) {\n if (body.role !== MemberRoleEnum.OSS_ADMIN) {\n throw new Error('Only admin role can be assigned to a member');\n }\n\n return await this.changeMemberRoleUsecase.execute(\n ChangeMemberRoleCommand.create({\n memberId,\n role: MemberRoleEnum.OSS_ADMIN,\n userId: user._id,\n organizationId: user.organizationId,\n })\n );\n }\n\n @Get('/members')\n @ExternalApiAccessible()\n @ApiResponse(MemberResponseDto, 200, true)\n @ApiOperation({\n summary: 'Fetch all members of current organizations',\n })\n async listOrganizationMembers(@UserSession() user: UserSessionData) {\n return await this.getMembers.execute(\n GetMembersCommand.create({\n user,\n userId: user._id,\n organizationId: user.organizationId,\n })\n );\n }\n\n @Put('/branding')\n @ExternalApiAccessible()\n @ApiResponse(OrganizationBrandingResponseDto)\n @ApiOperation({\n summary: 'Update organization branding details',\n })\n async updateBrandingDetails(@UserSession() user: UserSessionData, @Body() body: UpdateBrandingDetailsDto) {\n return await this.updateBrandingDetailsUsecase.execute(\n UpdateBrandingDetailsCommand.create({\n logo: body.logo,\n color: body.color,\n userId: user._id,\n id: user.organizationId,\n fontColor: body.fontColor,\n fontFamily: body.fontFamily,\n contentBackground: body.contentBackground,\n })\n );\n }\n\n @Patch('/')\n @ExternalApiAccessible()\n @ApiResponse(RenameOrganizationDto)\n @ApiOperation({\n summary: 'Rename organization name',\n })\n async rename(@UserSession() user: UserSessionData, @Body() body: RenameOrganizationDto) {\n return await this.renameOrganizationUsecase.execute(\n RenameOrganizationCommand.create({\n name: body.name,\n userId: user._id,\n id: user.organizationId,\n })\n );\n }\n}\n"
},
{
"path": "apps/api/src/app/organization/organization.module.ts",
"content": "import {\n DynamicModule,\n ForwardReference,\n forwardRef,\n MiddlewareConsumer,\n Module,\n NestModule,\n RequestMethod,\n} from '@nestjs/common';\nimport { Type } from '@nestjs/common/interfaces/type.interface';\nimport { AuthGuard } from '@nestjs/passport';\nimport { isBetterAuthEnabled, isClerkEnabled } from '@novu/shared';\nimport { AuthModule } from '../auth/auth.module';\nimport { EnvironmentsModuleV1 } from '../environments-v1/environments-v1.module';\nimport { IntegrationModule } from '../integrations/integrations.module';\nimport { LayoutsV2Module } from '../layouts-v2/layouts.module';\nimport { SharedModule } from '../shared/shared.module';\nimport { UserModule } from '../user/user.module';\nimport { EEOrganizationController } from './ee.organization.controller';\nimport { OrganizationController } from './organization.controller';\nimport { USE_CASES } from './usecases';\n\nconst enterpriseImports = (): Array | ForwardReference> => {\n const modules: Array | ForwardReference> = [];\n if (process.env.NOVU_ENTERPRISE === 'true' || process.env.CI_EE_TEST === 'true') {\n if (require('@novu/ee-billing')?.BillingModule) {\n modules.push(require('@novu/ee-billing')?.BillingModule.forRoot());\n }\n }\n\n return modules;\n};\n\nfunction getControllers() {\n if (isClerkEnabled() || isBetterAuthEnabled()) {\n return [EEOrganizationController];\n }\n\n return [OrganizationController];\n}\n\n@Module({\n imports: [\n SharedModule,\n UserModule,\n EnvironmentsModuleV1,\n IntegrationModule,\n forwardRef(() => AuthModule),\n LayoutsV2Module,\n ...enterpriseImports(),\n ],\n controllers: [...getControllers()],\n providers: [...USE_CASES],\n exports: [...USE_CASES],\n})\nexport class OrganizationModule implements NestModule {\n configure(consumer: MiddlewareConsumer): MiddlewareConsumer | void {\n if (process.env.NOVU_ENTERPRISE !== 'true' && process.env.CI_EE_TEST !== 'true') {\n consumer.apply(AuthGuard).exclude({\n method: RequestMethod.GET,\n path: '/organizations/invite/:inviteToken',\n });\n }\n }\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/create-organization/create-organization.command.ts",
"content": "import { ApiServiceLevelEnum, JobTitleEnum } from '@novu/shared';\nimport { IsDefined, IsEnum, IsOptional, IsString } from 'class-validator';\n\nimport { AuthenticatedCommand } from '../../../shared/commands/authenticated.command';\n\nexport class CreateOrganizationCommand extends AuthenticatedCommand {\n @IsString()\n @IsDefined()\n public readonly name: string;\n\n @IsString()\n @IsOptional()\n public readonly logo?: string;\n\n @IsOptional()\n @IsEnum(JobTitleEnum)\n jobTitle?: JobTitleEnum;\n\n @IsString()\n @IsOptional()\n domain?: string;\n\n @IsOptional()\n language?: string[];\n\n @IsOptional()\n @IsEnum(ApiServiceLevelEnum)\n apiServiceLevel?: ApiServiceLevelEnum;\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/create-organization/create-organization.usecase.ts",
"content": "import { BadRequestException, Inject, Injectable } from '@nestjs/common';\nimport { AnalyticsService } from '@novu/application-generic';\nimport { OrganizationEntity, OrganizationRepository, UserRepository } from '@novu/dal';\nimport { ApiServiceLevelEnum, EnvironmentEnum, JobTitleEnum, MemberRoleEnum } from '@novu/shared';\n\nimport { CreateEnvironmentCommand } from '../../../environments-v1/usecases/create-environment/create-environment.command';\nimport { CreateEnvironment } from '../../../environments-v1/usecases/create-environment/create-environment.usecase';\nimport { GetOrganizationCommand } from '../get-organization/get-organization.command';\nimport { GetOrganization } from '../get-organization/get-organization.usecase';\nimport { AddMemberCommand } from '../membership/add-member/add-member.command';\nimport { AddMember } from '../membership/add-member/add-member.usecase';\nimport { CreateOrganizationCommand } from './create-organization.command';\n\n@Injectable()\nexport class CreateOrganization {\n constructor(\n private readonly organizationRepository: OrganizationRepository,\n private readonly addMemberUsecase: AddMember,\n private readonly getOrganizationUsecase: GetOrganization,\n private readonly userRepository: UserRepository,\n private readonly createEnvironmentUsecase: CreateEnvironment,\n private analyticsService: AnalyticsService\n ) {}\n\n async execute(command: CreateOrganizationCommand): Promise {\n const user = await this.userRepository.findById(command.userId);\n if (!user) throw new BadRequestException('User not found');\n\n const isSelfHosted = process.env.IS_SELF_HOSTED === 'true';\n const isEnterprise = process.env.NOVU_ENTERPRISE === 'true' || process.env.CI_EE_TEST === 'true';\n const defaultApiServiceLevel =\n isSelfHosted && isEnterprise ? ApiServiceLevelEnum.UNLIMITED : ApiServiceLevelEnum.FREE;\n\n const createdOrganization = await this.organizationRepository.create({\n logo: command.logo,\n name: command.name,\n apiServiceLevel: command.apiServiceLevel || defaultApiServiceLevel,\n domain: command.domain,\n language: command.language,\n });\n\n if (command.jobTitle) {\n await this.updateJobTitle(user, command.jobTitle);\n }\n\n await this.addMemberUsecase.execute(\n AddMemberCommand.create({\n roles: [MemberRoleEnum.OSS_ADMIN],\n organizationId: createdOrganization._id,\n userId: command.userId,\n })\n );\n\n const devEnv = await this.createEnvironmentUsecase.execute(\n CreateEnvironmentCommand.create({\n userId: user._id,\n name: EnvironmentEnum.DEVELOPMENT,\n organizationId: createdOrganization._id,\n system: true,\n })\n );\n\n await this.createEnvironmentUsecase.execute(\n CreateEnvironmentCommand.create({\n userId: user._id,\n name: EnvironmentEnum.PRODUCTION,\n organizationId: createdOrganization._id,\n parentEnvironmentId: devEnv._id,\n system: true,\n })\n );\n\n this.analyticsService.upsertGroup(createdOrganization._id, createdOrganization, user);\n\n this.analyticsService.track('[Authentication] - Create Organization', user._id, {\n _organization: createdOrganization._id,\n language: command.language,\n creatorJobTitle: command.jobTitle,\n });\n\n const organizationAfterChanges = await this.getOrganizationUsecase.execute(\n GetOrganizationCommand.create({\n id: createdOrganization._id,\n userId: command.userId,\n })\n );\n\n return organizationAfterChanges as OrganizationEntity;\n }\n\n private async updateJobTitle(user, jobTitle: JobTitleEnum) {\n await this.userRepository.update(\n {\n _id: user._id,\n },\n {\n $set: {\n jobTitle,\n },\n }\n );\n\n this.analyticsService.setValue(user._id, 'jobTitle', jobTitle);\n }\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/create-organization/sync-external-organization/sync-external-organization.command.ts",
"content": "import { AuthenticatedCommand } from '@novu/application-generic';\nimport { IsDefined, IsEnum, IsOptional, IsString } from 'class-validator';\n\nexport class SyncExternalOrganizationCommand extends AuthenticatedCommand {\n @IsDefined()\n @IsString()\n externalId: string;\n\n @IsDefined()\n @IsString()\n email: string;\n\n @IsOptional()\n headers: Record;\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/create-organization/sync-external-organization/sync-external-organization.usecase.ts",
"content": "import { BadRequestException, Injectable } from '@nestjs/common';\nimport { ModuleRef } from '@nestjs/core';\nimport { AnalyticsService, PinoLogger } from '@novu/application-generic';\nimport { OrganizationEntity, OrganizationRepository, UserRepository } from '@novu/dal';\nimport { CreateEnvironmentCommand } from '../../../../environments-v1/usecases/create-environment/create-environment.command';\nimport { CreateEnvironment } from '../../../../environments-v1/usecases/create-environment/create-environment.usecase';\nimport { CreateNovuIntegrationsCommand } from '../../../../integrations/usecases/create-novu-integrations/create-novu-integrations.command';\nimport { CreateNovuIntegrations } from '../../../../integrations/usecases/create-novu-integrations/create-novu-integrations.usecase';\nimport { UpsertLayout, UpsertLayoutCommand } from '../../../../layouts-v2/usecases/upsert-layout';\nimport { createDefaultLayout } from '../../../../layouts-v2/utils/layout-templates';\nimport { GetOrganizationCommand } from '../../get-organization/get-organization.command';\nimport { GetOrganization } from '../../get-organization/get-organization.usecase';\nimport { SyncExternalOrganizationCommand } from './sync-external-organization.command';\n\n// TODO: eventually move to @novu/ee-auth\n\n/**\n * This logic is closely related to the CreateOrganization use case.\n * @see src/app/organization/usecases/create-organization/create-organization.usecase.ts\n *\n * The side effects of creating a new organization are largely\n * consistent with those in CreateOrganization, with only minor differences.\n */\n\n@Injectable()\nexport class SyncExternalOrganization {\n constructor(\n private readonly organizationRepository: OrganizationRepository,\n private readonly getOrganizationUsecase: GetOrganization,\n private readonly createEnvironmentUsecase: CreateEnvironment,\n private readonly createNovuIntegrations: CreateNovuIntegrations,\n private readonly upsertLayoutUsecase: UpsertLayout,\n private analyticsService: AnalyticsService,\n private moduleRef: ModuleRef,\n private logger: PinoLogger\n ) {\n this.logger.setContext(this.constructor.name);\n }\n\n async execute(command: SyncExternalOrganizationCommand): Promise {\n const isSelfHosted = process.env.IS_SELF_HOSTED === 'true';\n const isEnterprise = process.env.NOVU_ENTERPRISE === 'true' || process.env.CI_EE_TEST === 'true';\n\n const organization = await this.organizationRepository.create(\n {\n externalId: command.externalId,\n apiServiceLevel: isSelfHosted && isEnterprise ? 'unlimited' : undefined,\n },\n { headers: command.headers }\n );\n\n const devEnv = await this.createEnvironmentUsecase.execute(\n CreateEnvironmentCommand.create({\n userId: command.userId,\n name: 'Development',\n organizationId: organization._id,\n system: true,\n })\n );\n\n await this.createNovuIntegrations.execute(\n CreateNovuIntegrationsCommand.create({\n environmentId: devEnv._id,\n organizationId: devEnv._organizationId,\n userId: command.userId,\n name: devEnv.name,\n })\n );\n\n await this.upsertLayoutUsecase.execute(\n UpsertLayoutCommand.create({\n environmentId: devEnv._id,\n organizationId: devEnv._organizationId,\n userId: command.userId,\n layoutDto: {\n name: 'Default layout',\n controlValues: {\n email: {\n body: JSON.stringify(createDefaultLayout(organization.name)),\n editorType: 'block',\n },\n },\n },\n })\n );\n\n const prodEnv = await this.createEnvironmentUsecase.execute(\n CreateEnvironmentCommand.create({\n userId: command.userId,\n name: 'Production',\n organizationId: organization._id,\n parentEnvironmentId: devEnv._id,\n system: true,\n })\n );\n\n await this.createNovuIntegrations.execute(\n CreateNovuIntegrationsCommand.create({\n environmentId: prodEnv._id,\n organizationId: prodEnv._organizationId,\n userId: command.userId,\n name: prodEnv.name,\n })\n );\n\n await this.upsertLayoutUsecase.execute(\n UpsertLayoutCommand.create({\n environmentId: prodEnv._id,\n organizationId: prodEnv._organizationId,\n userId: command.userId,\n layoutDto: {\n name: 'Default layout',\n controlValues: {\n email: {\n body: JSON.stringify(createDefaultLayout(organization.name)),\n editorType: 'block',\n },\n },\n },\n })\n );\n\n this.analyticsService.upsertGroup(organization._id, organization, { _id: command.userId });\n\n this.analyticsService.track('[Authentication] - Create Organization', command.userId, {\n _organization: organization._id,\n });\n\n const organizationAfterChanges = await this.getOrganizationUsecase.execute(\n GetOrganizationCommand.create({\n id: organization._id,\n userId: command.userId,\n })\n );\n\n if (organizationAfterChanges !== null) {\n await this.createCustomer(command.email, organizationAfterChanges._id);\n }\n\n return organizationAfterChanges as OrganizationEntity;\n }\n\n private async createCustomer(billingEmail: string, organizationId: string) {\n try {\n if (process.env.NOVU_ENTERPRISE === 'true' || process.env.CI_EE_TEST === 'true') {\n if (!require('@novu/ee-billing')?.GetOrCreateCustomer) {\n throw new BadRequestException('Billing module is not loaded');\n }\n const usecase = this.moduleRef.get(require('@novu/ee-billing')?.GetOrCreateCustomer, {\n strict: false,\n });\n await usecase.execute({\n organizationId,\n billingEmail,\n });\n }\n } catch (e) {\n this.logger.error({ err: e }, `Unexpected error while importing enterprise modules`);\n }\n }\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/get-my-organization/get-my-organization.command.ts",
"content": "import { IsDefined } from 'class-validator';\nimport { AuthenticatedCommand } from '../../../shared/commands/authenticated.command';\n\nexport class GetMyOrganizationCommand extends AuthenticatedCommand {\n @IsDefined()\n public readonly id: string;\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/get-my-organization/get-my-organization.usecase.ts",
"content": "import { Injectable, Scope, UnauthorizedException } from '@nestjs/common';\nimport { GetOrganizationCommand } from '../get-organization/get-organization.command';\nimport { GetOrganization } from '../get-organization/get-organization.usecase';\nimport { GetMyOrganizationCommand } from './get-my-organization.command';\n\n@Injectable({\n scope: Scope.REQUEST,\n})\nexport class GetMyOrganization {\n constructor(private getOrganizationUseCase: GetOrganization) {}\n\n async execute(command: GetMyOrganizationCommand) {\n const organization = await this.getOrganizationUseCase.execute(\n GetOrganizationCommand.create({\n id: command.id,\n userId: command.userId,\n })\n );\n if (!organization) throw new UnauthorizedException('No organization found');\n\n return organization;\n }\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/get-organization/get-organization.command.ts",
"content": "import { AuthenticatedCommand } from '../../../shared/commands/authenticated.command';\n\nexport class GetOrganizationCommand extends AuthenticatedCommand {\n public readonly id: string;\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/get-organization/get-organization.usecase.ts",
"content": "import { Injectable, Scope } from '@nestjs/common';\nimport { OrganizationRepository } from '@novu/dal';\nimport { GetOrganizationCommand } from './get-organization.command';\n\n@Injectable()\nexport class GetOrganization {\n constructor(private readonly organizationRepository: OrganizationRepository) {}\n\n async execute(command: GetOrganizationCommand) {\n return await this.organizationRepository.findById(command.id);\n }\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/get-organization-settings/get-organization-settings.command.ts",
"content": "import { BaseCommand } from '@novu/application-generic';\nimport { OrganizationEntity } from '@novu/dal';\nimport { IsNotEmpty, IsOptional } from 'class-validator';\n\nexport class GetOrganizationSettingsCommand extends BaseCommand {\n @IsNotEmpty()\n readonly organizationId: string;\n\n @IsOptional()\n readonly organization?: OrganizationEntity;\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/get-organization-settings/get-organization-settings.usecase.ts",
"content": "import { Injectable, NotFoundException } from '@nestjs/common';\nimport { CommunityOrganizationRepository } from '@novu/dal';\nimport { DEFAULT_LOCALE } from '@novu/shared';\nimport { GetOrganizationSettingsDto } from '../../dtos/get-organization-settings.dto';\nimport { GetOrganizationSettingsCommand } from './get-organization-settings.command';\n\n@Injectable()\nexport class GetOrganizationSettings {\n constructor(private organizationRepository: CommunityOrganizationRepository) {}\n\n async execute(command: GetOrganizationSettingsCommand): Promise {\n const organization = command.organization ?? (await this.organizationRepository.findById(command.organizationId));\n\n if (!organization) {\n throw new NotFoundException('Organization not found');\n }\n\n return {\n removeNovuBranding: organization.removeNovuBranding || false,\n defaultLocale: organization.defaultLocale || DEFAULT_LOCALE,\n targetLocales: organization.targetLocales || [],\n };\n }\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/get-organizations/get-organizations.command.ts",
"content": "import { AuthenticatedCommand } from '../../../shared/commands/authenticated.command';\n\nexport class GetOrganizationsCommand extends AuthenticatedCommand {}\n"
},
{
"path": "apps/api/src/app/organization/usecases/get-organizations/get-organizations.usecase.ts",
"content": "import { Injectable, Scope } from '@nestjs/common';\nimport { OrganizationRepository } from '@novu/dal';\nimport { GetOrganizationsCommand } from './get-organizations.command';\n\n@Injectable({\n scope: Scope.REQUEST,\n})\nexport class GetOrganizations {\n constructor(private readonly organizationRepository: OrganizationRepository) {}\n\n async execute(command: GetOrganizationsCommand) {\n return await this.organizationRepository.findUserActiveOrganizations(command.userId);\n }\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/index.ts",
"content": "import { isBetterAuthEnabled, isClerkEnabled } from '@novu/shared';\nimport { CreateOrganization } from './create-organization/create-organization.usecase';\nimport { SyncExternalOrganization } from './create-organization/sync-external-organization/sync-external-organization.usecase';\nimport { GetMyOrganization } from './get-my-organization/get-my-organization.usecase';\nimport { GetOrganization } from './get-organization/get-organization.usecase';\nimport { GetOrganizationSettings } from './get-organization-settings/get-organization-settings.usecase';\nimport { GetOrganizations } from './get-organizations/get-organizations.usecase';\nimport { AddMember } from './membership/add-member/add-member.usecase';\nimport { ChangeMemberRole } from './membership/change-member-role/change-member-role.usecase';\nimport { GetMembers } from './membership/get-members/get-members.usecase';\nimport { RemoveMember } from './membership/remove-member/remove-member.usecase';\nimport { RenameOrganization } from './rename-organization/rename-organization.usecase';\nimport { UpdateBrandingDetails } from './update-branding-details/update-branding-details.usecase';\nimport { UpdateOrganizationSettings } from './update-organization-settings/update-organization-settings.usecase';\n\n// TODO: move ee.organization.controller.ts to EE package\nfunction getEnterpriseUsecases() {\n if (isClerkEnabled() || isBetterAuthEnabled()) {\n return [\n {\n provide: 'SyncOrganizationUsecase',\n useClass: SyncExternalOrganization,\n },\n ];\n }\n\n return [];\n}\n\nexport const USE_CASES = [\n AddMember,\n CreateOrganization,\n GetOrganization,\n GetMembers,\n RemoveMember,\n ChangeMemberRole,\n UpdateBrandingDetails,\n GetOrganizations,\n GetMyOrganization,\n RenameOrganization,\n GetOrganizationSettings,\n UpdateOrganizationSettings,\n ...getEnterpriseUsecases(),\n];\n"
},
{
"path": "apps/api/src/app/organization/usecases/membership/add-member/add-member.command.ts",
"content": "import { MemberRoleEnum } from '@novu/shared';\nimport { ArrayNotEmpty } from 'class-validator';\nimport { OrganizationCommand } from '../../../../shared/commands/organization.command';\n\nexport class AddMemberCommand extends OrganizationCommand {\n @ArrayNotEmpty()\n public readonly roles: MemberRoleEnum[];\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/membership/add-member/add-member.usecase.ts",
"content": "import { BadRequestException, Injectable } from '@nestjs/common';\nimport { MemberRepository } from '@novu/dal';\nimport { MemberStatusEnum } from '@novu/shared';\nimport { AddMemberCommand } from './add-member.command';\n\n@Injectable()\nexport class AddMember {\n constructor(private readonly memberRepository: MemberRepository) {}\n\n async execute(command: AddMemberCommand): Promise {\n const isAlreadyMember = await this.isMember(command.organizationId, command.userId);\n if (isAlreadyMember) throw new BadRequestException('Member already exists');\n\n await this.memberRepository.addMember(command.organizationId, {\n _userId: command.userId,\n roles: command.roles,\n memberStatus: MemberStatusEnum.ACTIVE,\n });\n }\n\n private async isMember(organizationId: string, userId: string): Promise {\n return !!(await this.memberRepository.findMemberByUserId(organizationId, userId));\n }\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/membership/change-member-role/change-member-role.command.ts",
"content": "import { MemberRoleEnum } from '@novu/shared';\nimport { IsDefined, IsEnum, IsMongoId } from 'class-validator';\nimport { OrganizationCommand } from '../../../../shared/commands/organization.command';\n\nexport class ChangeMemberRoleCommand extends OrganizationCommand {\n @IsDefined()\n role: MemberRoleEnum.OSS_ADMIN;\n\n @IsDefined()\n @IsMongoId()\n memberId: string;\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/membership/change-member-role/change-member-role.usecase.ts",
"content": "import { BadRequestException, Injectable, NotFoundException } from '@nestjs/common';\nimport { MemberRepository, OrganizationRepository } from '@novu/dal';\nimport { MemberRoleEnum } from '@novu/shared';\n\nimport { ChangeMemberRoleCommand } from './change-member-role.command';\n\n@Injectable()\nexport class ChangeMemberRole {\n constructor(\n private organizationRepository: OrganizationRepository,\n private memberRepository: MemberRepository\n ) {}\n\n async execute(command: ChangeMemberRoleCommand) {\n if (![MemberRoleEnum.OSS_MEMBER, MemberRoleEnum.OSS_ADMIN].includes(command.role)) {\n throw new BadRequestException('Not supported role type');\n }\n\n if (command.role !== MemberRoleEnum.OSS_ADMIN) {\n throw new BadRequestException(`The change of role to an ${command.role} type is not supported`);\n }\n\n const organization = await this.organizationRepository.findById(command.organizationId);\n if (!organization) throw new NotFoundException('No organization was found');\n\n const member = await this.memberRepository.findMemberById(organization._id, command.memberId);\n if (!member) throw new NotFoundException('No member was found');\n\n const roles = [command.role];\n\n await this.memberRepository.updateMemberRoles(organization._id, command.memberId, roles);\n\n return this.memberRepository.findMemberByUserId(organization._id, member._userId);\n }\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/membership/get-members/get-members.command.ts",
"content": "import { UserSessionData } from '@novu/shared';\nimport { IsDefined } from 'class-validator';\nimport { OrganizationCommand } from '../../../../shared/commands/organization.command';\n\nexport class GetMembersCommand extends OrganizationCommand {\n @IsDefined()\n user: UserSessionData;\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/membership/get-members/get-members.usecase.ts",
"content": "import { Injectable, Scope } from '@nestjs/common';\nimport { MemberRepository } from '@novu/dal';\nimport { MemberRoleEnum, MemberStatusEnum } from '@novu/shared';\nimport { GetMembersCommand } from './get-members.command';\n\n@Injectable({\n scope: Scope.REQUEST,\n})\nexport class GetMembers {\n constructor(private membersRepository: MemberRepository) {}\n\n async execute(command: GetMembersCommand) {\n return (await this.membersRepository.getOrganizationMembers(command.organizationId))\n .map((member) => {\n if (!command.user.roles.includes(MemberRoleEnum.OSS_ADMIN)) {\n if (member.memberStatus === MemberStatusEnum.INVITED) return null;\n if (member.user) member.user.email = '';\n if (member.invite) member.invite.email = '';\n }\n\n return member;\n })\n .filter((member) => !!member);\n }\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/membership/remove-member/remove-member.command.ts",
"content": "import { IsMongoId, IsString } from 'class-validator';\nimport { OrganizationCommand } from '../../../../shared/commands/organization.command';\n\nexport class RemoveMemberCommand extends OrganizationCommand {\n @IsString()\n @IsMongoId()\n memberId: string;\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/membership/remove-member/remove-member.usecase.ts",
"content": "import { BadRequestException, Injectable, NotFoundException, Scope } from '@nestjs/common';\nimport { EnvironmentRepository, MemberRepository } from '@novu/dal';\nimport { RemoveMemberCommand } from './remove-member.command';\n\n@Injectable({\n scope: Scope.REQUEST,\n})\nexport class RemoveMember {\n constructor(\n private memberRepository: MemberRepository,\n private environmentRepository: EnvironmentRepository\n ) {}\n\n async execute(command: RemoveMemberCommand) {\n const members = await this.memberRepository.getOrganizationMembers(command.organizationId);\n const memberToRemove = members.find((i) => i._id === command.memberId);\n\n if (!memberToRemove) throw new NotFoundException('Member not found');\n if (memberToRemove._userId && memberToRemove._userId && memberToRemove._userId === command.userId) {\n throw new BadRequestException('Cannot remove self from members');\n }\n\n await this.memberRepository.removeMemberById(command.organizationId, memberToRemove._id);\n const environments = await this.environmentRepository.findOrganizationEnvironments(command.organizationId);\n const isMemberAssociatedWithEnvironment = environments.some((i) =>\n i.apiKeys.some((key) => key._userId === memberToRemove._userId)\n );\n\n if (isMemberAssociatedWithEnvironment) {\n const owner = await this.memberRepository.getOrganizationOwnerAccount(command.organizationId);\n if (!owner) throw new NotFoundException('No owner account found for organization');\n\n await this.environmentRepository.updateApiKeyUserId(\n command.organizationId,\n memberToRemove._userId,\n owner._userId\n );\n }\n\n return memberToRemove;\n }\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/rename-organization/rename-organization-command.ts",
"content": "import { IsDefined, IsNotEmpty } from 'class-validator';\nimport { AuthenticatedCommand } from '../../../shared/commands/authenticated.command';\n\nexport class RenameOrganizationCommand extends AuthenticatedCommand {\n @IsDefined()\n public readonly id: string;\n\n @IsDefined()\n @IsNotEmpty()\n name: string;\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/rename-organization/rename-organization.usecase.ts",
"content": "import { Injectable } from '@nestjs/common';\nimport { OrganizationRepository } from '@novu/dal';\nimport { RenameOrganizationCommand } from './rename-organization-command';\n\n@Injectable()\nexport class RenameOrganization {\n constructor(private organizationRepository: OrganizationRepository) {}\n\n async execute(command: RenameOrganizationCommand) {\n const payload = {\n name: command.name,\n };\n\n await this.organizationRepository.renameOrganization(command.id, payload);\n\n return payload;\n }\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/update-branding-details/update-branding-details.command.ts",
"content": "import { IsDefined, IsHexColor, IsOptional, IsUrl } from 'class-validator';\nimport { AuthenticatedCommand } from '../../../shared/commands/authenticated.command';\n\nexport class UpdateBrandingDetailsCommand extends AuthenticatedCommand {\n @IsDefined()\n public readonly id: string;\n\n @IsUrl({ require_tld: false })\n @IsOptional()\n logo: string;\n\n @IsOptional()\n @IsHexColor()\n color: string;\n\n @IsOptional()\n @IsHexColor()\n fontColor: string;\n\n @IsOptional()\n @IsHexColor()\n contentBackground: string;\n\n @IsOptional()\n fontFamily?: string;\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/update-branding-details/update-branding-details.usecase.ts",
"content": "import { Injectable } from '@nestjs/common';\nimport { OrganizationRepository } from '@novu/dal';\nimport { UpdateBrandingDetailsCommand } from './update-branding-details.command';\n\n@Injectable()\nexport class UpdateBrandingDetails {\n constructor(private organizationRepository: OrganizationRepository) {}\n\n async execute(command: UpdateBrandingDetailsCommand) {\n const payload = {\n color: command.color,\n logo: command.logo,\n fontColor: command.fontColor,\n contentBackground: command.contentBackground,\n fontFamily: command.fontFamily,\n };\n\n await this.organizationRepository.updateBrandingDetails(command.id, payload);\n\n return payload;\n }\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/update-organization-settings/update-organization-settings.command.ts",
"content": "import { AuthenticatedCommand, IsValidLocale } from '@novu/application-generic';\nimport { IsArray, IsBoolean, IsNotEmpty, IsOptional, IsString } from 'class-validator';\n\nexport class UpdateOrganizationSettingsCommand extends AuthenticatedCommand {\n @IsNotEmpty()\n readonly organizationId: string;\n\n @IsOptional()\n @IsBoolean()\n removeNovuBranding?: boolean;\n\n @IsOptional()\n @IsValidLocale()\n defaultLocale?: string;\n\n @IsOptional()\n @IsArray()\n @IsString({ each: true })\n targetLocales?: string[];\n}\n"
},
{
"path": "apps/api/src/app/organization/usecases/update-organization-settings/update-organization-settings.usecase.ts",
"content": "import { HttpException, HttpStatus, Injectable, NotFoundException } from '@nestjs/common';\nimport { AnalyticsService } from '@novu/application-generic';\nimport { CommunityOrganizationRepository, OrganizationEntity } from '@novu/dal';\nimport { ApiServiceLevelEnum, DEFAULT_LOCALE, FeatureNameEnum, getFeatureForTierAsBoolean } from '@novu/shared';\nimport { GetOrganizationSettingsDto } from '../../dtos/get-organization-settings.dto';\nimport { UpdateOrganizationSettingsCommand } from './update-organization-settings.command';\n\n@Injectable()\nexport class UpdateOrganizationSettings {\n constructor(\n private organizationRepository: CommunityOrganizationRepository,\n private analyticsService: AnalyticsService\n ) {}\n\n async execute(command: UpdateOrganizationSettingsCommand): Promise {\n const organization = await this.organizationRepository.findById(command.organizationId);\n\n if (!organization) {\n throw new NotFoundException('Organization not found');\n }\n\n this.validateTierRestrictions(command, organization);\n\n const updateFields = this.buildUpdateFields(command);\n\n if (Object.keys(updateFields).length === 0) {\n return this.buildSettingsResponse(organization);\n }\n\n await this.organizationRepository.updateOne({ _id: organization._id }, { $set: updateFields });\n\n if (command.removeNovuBranding !== undefined) {\n this.analyticsService.mixpanelTrack('Remove Branding', command.userId, {\n _organization: command.organizationId,\n newStatus: command.removeNovuBranding,\n });\n }\n\n return this.buildSettingsResponse({\n ...organization,\n ...updateFields,\n });\n }\n\n private validateTierRestrictions(command: UpdateOrganizationSettingsCommand, organization: OrganizationEntity): void {\n // Only validate branding feature access if user is trying to update it\n if (command.removeNovuBranding !== undefined) {\n const canRemoveNovuBranding = getFeatureForTierAsBoolean(\n FeatureNameEnum.PLATFORM_REMOVE_NOVU_BRANDING_BOOLEAN,\n organization.apiServiceLevel || ApiServiceLevelEnum.FREE\n );\n\n if (!canRemoveNovuBranding) {\n throw new HttpException(\n {\n error: 'Payment Required',\n message:\n 'Removing Novu branding is not allowed on the free plan. Please upgrade to a paid plan to access this feature.',\n },\n HttpStatus.PAYMENT_REQUIRED\n );\n }\n }\n\n if (command.targetLocales !== undefined || command.defaultLocale !== undefined) {\n const canUseTranslations = getFeatureForTierAsBoolean(\n FeatureNameEnum.AUTO_TRANSLATIONS,\n organization.apiServiceLevel || ApiServiceLevelEnum.FREE\n );\n\n if (!canUseTranslations) {\n throw new HttpException(\n {\n error: 'Payment Required',\n message:\n 'Update of locales is a part of the translation feature. Please upgrade to a paid plan to access this feature.',\n },\n HttpStatus.PAYMENT_REQUIRED\n );\n }\n }\n }\n\n private buildUpdateFields(command: UpdateOrganizationSettingsCommand): Partial {\n const updateFields: Partial = {};\n\n if (command.removeNovuBranding !== undefined) {\n updateFields.removeNovuBranding = command.removeNovuBranding;\n }\n\n if (command.defaultLocale !== undefined) {\n updateFields.defaultLocale = command.defaultLocale;\n }\n\n if (command.targetLocales !== undefined) {\n updateFields.targetLocales = command.targetLocales;\n }\n\n return updateFields;\n }\n\n private buildSettingsResponse(organization: OrganizationEntity): GetOrganizationSettingsDto {\n return {\n removeNovuBranding: organization.removeNovuBranding || false,\n defaultLocale: organization.defaultLocale || DEFAULT_LOCALE,\n targetLocales: organization.targetLocales || [],\n };\n }\n}\n"
},
{
"path": "apps/api/src/app/outbound-webhooks/dtos/create-webhook-portal-response.dto.ts",
"content": "import { ApiProperty } from '@nestjs/swagger';\n\nexport class CreateWebhookPortalResponseDto {\n @ApiProperty({\n description: 'The webhook portal application ID',\n })\n appId: string;\n}\n"
},
{
"path": "apps/api/src/app/outbound-webhooks/dtos/get-webhook-portal-token-response.dto.ts",
"content": "import { IsNotEmpty, IsString } from 'class-validator';\n\nexport class GetWebhookPortalTokenResponseDto {\n @IsNotEmpty()\n @IsString()\n url: string;\n\n @IsNotEmpty()\n @IsString()\n token: string;\n\n @IsNotEmpty()\n @IsString()\n appId: string;\n}\n"
},
{
"path": "apps/api/src/app/outbound-webhooks/outbound-webhooks.controller.ts",
"content": "import { ClassSerializerInterceptor, Controller, Get, Post, UseInterceptors } from '@nestjs/common';\nimport { ApiExcludeController, ApiOperation } from '@nestjs/swagger';\nimport { ProductFeature, RequirePermissions, UserSession } from '@novu/application-generic';\nimport { PermissionsEnum, ProductFeatureKeyEnum, UserSessionData } from '@novu/shared';\nimport { RequireAuthentication } from '../auth/framework/auth.decorator';\nimport { CreateWebhookPortalResponseDto } from './dtos/create-webhook-portal-response.dto';\nimport { GetWebhookPortalTokenResponseDto } from './dtos/get-webhook-portal-token-response.dto';\nimport { CreateWebhookPortalCommand } from './usecases/create-webhook-portal-token/create-webhook-portal.command';\nimport { CreateWebhookPortalUsecase } from './usecases/create-webhook-portal-token/create-webhook-portal.usecase';\nimport { GetWebhookPortalTokenCommand } from './usecases/get-webhook-portal-token/get-webhook-portal-token.command';\nimport { GetWebhookPortalTokenUsecase } from './usecases/get-webhook-portal-token/get-webhook-portal-token.usecase';\n\n@Controller({ path: `/outbound-webhooks`, version: '2' })\n@UseInterceptors(ClassSerializerInterceptor)\n@RequireAuthentication()\n@ApiExcludeController()\nexport class OutboundWebhooksController {\n constructor(\n private getWebhookPortalTokenUsecase: GetWebhookPortalTokenUsecase,\n private createWebhookPortalTokenUsecase: CreateWebhookPortalUsecase\n ) {}\n\n @Get('/portal/token')\n @ProductFeature(ProductFeatureKeyEnum.WEBHOOKS)\n @RequirePermissions(PermissionsEnum.WEBHOOK_WRITE, PermissionsEnum.WEBHOOK_READ)\n @ApiOperation({\n summary: 'Get Webhook Portal Access Token',\n description:\n 'Generates a short-lived token and URL for accessing the Svix application portal for the current environment.',\n })\n async getPortalToken(@UserSession() user: UserSessionData): Promise {\n return await this.getWebhookPortalTokenUsecase.execute(\n GetWebhookPortalTokenCommand.create({\n environmentId: user.environmentId,\n organizationId: user.organizationId,\n userId: user._id,\n })\n );\n }\n\n @Post('/portal/token')\n @ProductFeature(ProductFeatureKeyEnum.WEBHOOKS)\n @RequirePermissions(PermissionsEnum.WEBHOOK_WRITE)\n @ApiOperation({\n summary: 'Create Webhook Portal Access Token',\n description: 'Creates a token for accessing the webhook portal for the current environment.',\n })\n async createPortalToken(@UserSession() user: UserSessionData): Promise {\n return await this.createWebhookPortalTokenUsecase.execute(\n CreateWebhookPortalCommand.create({\n environmentId: user.environmentId,\n organizationId: user.organizationId,\n userId: user._id,\n })\n );\n }\n}\n"
},
{
"path": "apps/api/src/app/outbound-webhooks/outbound-webhooks.module.ts",
"content": "import { DynamicModule, Module } from '@nestjs/common';\nimport { SendWebhookMessage, SvixProviderService } from '@novu/application-generic';\nimport { NoopSendWebhookMessage } from '../inbox/usecases/noop-send-webhook-message.usecase';\nimport { SharedModule } from '../shared/shared.module';\nimport { OutboundWebhooksController } from './outbound-webhooks.controller';\nimport { CreateWebhookPortalUsecase } from './usecases/create-webhook-portal-token/create-webhook-portal.usecase';\nimport { GetWebhookPortalTokenUsecase } from './usecases/get-webhook-portal-token/get-webhook-portal-token.usecase';\n\n@Module({})\nclass OutboundWebhooksModuleDefinition {}\n\nexport const OutboundWebhooksModule = {\n forRoot(): DynamicModule {\n const isEnterprise = process.env.NOVU_ENTERPRISE === 'true';\n\n if (isEnterprise) {\n return {\n module: OutboundWebhooksModuleDefinition,\n imports: [SharedModule],\n controllers: [OutboundWebhooksController],\n providers: [GetWebhookPortalTokenUsecase, CreateWebhookPortalUsecase, SvixProviderService, SendWebhookMessage],\n exports: [SendWebhookMessage],\n };\n }\n\n return {\n module: OutboundWebhooksModuleDefinition,\n imports: [SharedModule],\n providers: [\n {\n provide: SendWebhookMessage,\n useClass: NoopSendWebhookMessage,\n },\n ],\n exports: [SendWebhookMessage],\n };\n },\n};\n"
},
{
"path": "apps/api/src/app/outbound-webhooks/usecases/create-webhook-portal-token/create-webhook-portal.command.ts",
"content": "import { EnvironmentWithUserCommand } from '../../../shared/commands/project.command';\n\nexport class CreateWebhookPortalCommand extends EnvironmentWithUserCommand {}\n"
},
{
"path": "apps/api/src/app/outbound-webhooks/usecases/create-webhook-portal-token/create-webhook-portal.usecase.ts",
"content": "import { BadRequestException, Inject, Injectable, NotFoundException, Scope } from '@nestjs/common';\nimport { generateWebhookAppId, LogDecorator, SvixClient } from '@novu/application-generic';\nimport { EnvironmentRepository, OrganizationRepository } from '@novu/dal';\nimport { CreateWebhookPortalResponseDto } from '../../dtos/create-webhook-portal-response.dto';\nimport { CreateWebhookPortalCommand } from './create-webhook-portal.command';\n\n@Injectable()\nexport class CreateWebhookPortalUsecase {\n constructor(\n private environmentRepository: EnvironmentRepository,\n @Inject('SVIX_CLIENT') private svix: SvixClient,\n private organizationRepository: OrganizationRepository\n ) {}\n\n @LogDecorator()\n async execute(command: CreateWebhookPortalCommand): Promise {\n if (!this.svix) {\n throw new BadRequestException('Webhook system is not enabled');\n }\n\n const environment = await this.environmentRepository.findOne({\n _id: command.environmentId,\n _organizationId: command.organizationId,\n });\n\n if (!environment) {\n throw new NotFoundException(\n `Environment not found for id ${command.environmentId} and organization ${command.organizationId}`\n );\n }\n\n const organization = await this.organizationRepository.findById(command.organizationId);\n if (!organization) {\n throw new NotFoundException(`Organization not found for id ${command.organizationId}`);\n }\n\n try {\n const app = await this.svix.application.create({\n name: organization.name,\n uid: generateWebhookAppId(command.organizationId, command.environmentId),\n metadata: {\n environmentId: command.environmentId,\n organizationId: command.organizationId,\n },\n });\n\n await this.environmentRepository.updateOne({ _id: command.environmentId }, { $set: { webhookAppId: app.uid } });\n\n return {\n appId: app.uid!,\n };\n } catch (error) {\n throw new BadRequestException(`Failed to generate Svix portal token: ${error?.message}`);\n }\n }\n}\n"
},
{
"path": "apps/api/src/app/outbound-webhooks/usecases/get-webhook-portal-token/get-webhook-portal-token.command.ts",
"content": "import { BaseCommand } from '@novu/application-generic';\nimport { IsDefined } from 'class-validator';\nimport { EnvironmentCommand } from '../../../shared/commands/project.command';\n\nexport class GetWebhookPortalTokenCommand extends EnvironmentCommand {\n @IsDefined()\n userId: string;\n}\n"
},
{
"path": "apps/api/src/app/outbound-webhooks/usecases/get-webhook-portal-token/get-webhook-portal-token.usecase.ts",
"content": "import { BadRequestException, Inject, Injectable, NotFoundException, Scope } from '@nestjs/common';\nimport { generateWebhookAppId, LogDecorator, SvixClient } from '@novu/application-generic';\nimport { EnvironmentRepository } from '@novu/dal';\nimport { GetWebhookPortalTokenResponseDto } from '../../dtos/get-webhook-portal-token-response.dto';\nimport { GetWebhookPortalTokenCommand } from './get-webhook-portal-token.command';\n\n@Injectable()\nexport class GetWebhookPortalTokenUsecase {\n constructor(\n private environmentRepository: EnvironmentRepository,\n @Inject('SVIX_CLIENT') private svix: SvixClient\n ) {}\n\n @LogDecorator()\n async execute(command: GetWebhookPortalTokenCommand): Promise {\n if (!this.svix) {\n throw new BadRequestException('Webhook system is not enabled');\n }\n\n const environment = await this.environmentRepository.findOne({\n _id: command.environmentId,\n _organizationId: command.organizationId,\n });\n\n if (!environment) {\n throw new NotFoundException(\n `Environment not found for id ${command.environmentId} and organization ${command.organizationId}`\n );\n }\n\n if (!environment.webhookAppId) {\n throw new NotFoundException(`Portal not found for environment ${command.environmentId}`);\n }\n\n try {\n const svixResponse = await this.svix.authentication.appPortalAccess(environment.webhookAppId, {});\n\n return {\n url: svixResponse.url,\n token: svixResponse.token,\n appId: environment.webhookAppId,\n };\n } catch (error) {\n if (error.code === 404) {\n throw new NotFoundException(`Portal not found for environment ${command.environmentId}`);\n }\n\n throw new BadRequestException(`Failed to generate Svix portal token: ${error?.message}`);\n }\n }\n}\n"
},
{
"path": "apps/api/src/app/outbound-webhooks/webhooks.const.ts",
"content": "import { ApiProperty } from '@nestjs/swagger';\nimport { MessageWebhookResponseDto, WorkflowResponseDto } from '@novu/application-generic';\nimport { WebhookEventEnum, WebhookObjectTypeEnum } from '@novu/shared';\nimport { InboxPreference } from '../inbox/utils/types';\n\ninterface WebhookEventConfig {\n event: WebhookEventEnum;\n // biome-ignore lint/complexity/noBannedTypes: This is the expected type for the payloadDto for SwaggerDocumentOptions.extraModels\n payloadDto: Function;\n objectType: WebhookObjectTypeEnum;\n}\n\ntype WebhookEventRecord = Record;\n\nexport class WebhookUpdatedWorkflowDto {\n @ApiProperty({ description: 'Current workflow state', type: () => WorkflowResponseDto })\n object: WorkflowResponseDto;\n\n @ApiProperty({ description: 'Previous state of the workflow', type: () => WorkflowResponseDto })\n previousObject: WorkflowResponseDto;\n}\n\nexport class WebhookCreatedWorkflowDto {\n @ApiProperty({ description: 'Current workflow state', type: () => WorkflowResponseDto })\n object: WorkflowResponseDto;\n}\n\nexport class WebhookDeletedWorkflowDto {\n @ApiProperty({ description: 'Current workflow state', type: () => WorkflowResponseDto })\n object: WorkflowResponseDto;\n}\n\nexport class WebhookMessageDto {\n @ApiProperty({ description: 'Current message state' })\n object: MessageWebhookResponseDto;\n}\n\nenum MessageFailedErrorCodeEnum {\n TOKEN_EXPIRED = 'token_expired',\n}\n\nexport class MessageFailedWebhookDto {\n @ApiProperty({ description: 'Current message state' })\n object: MessageWebhookResponseDto;\n\n @ApiProperty({ description: 'Error message' })\n errorCode: MessageFailedErrorCodeEnum;\n}\n\nexport class MessageFailedPushDto {\n @ApiProperty({ description: 'Is invalid token' })\n isInvalidToken: boolean;\n\n @ApiProperty({ description: 'Device token' })\n deviceToken: string;\n}\n\nexport class MessageFailedErrorDto {\n @ApiProperty({ description: 'Error message' })\n message: string;\n\n @ApiProperty({ description: 'Push error' })\n push?: MessageFailedPushDto;\n}\n\nexport class WebhookMessageFailedDto {\n @ApiProperty({ description: 'Current message state' })\n object: MessageWebhookResponseDto;\n\n @ApiProperty({ description: 'Error message' })\n error: MessageFailedErrorDto;\n}\n\nexport class WebhookPreferenceDto {\n @ApiProperty({ description: 'Current preference state' })\n object: InboxPreference;\n\n @ApiProperty({ description: 'Subscriber ID' })\n subscriberId: string;\n}\n\n// Create the webhook events as a record to ensure all enum values are covered\nconst webhookEventRecord = {\n [WebhookEventEnum.MESSAGE_SENT]: {\n event: WebhookEventEnum.MESSAGE_SENT,\n payloadDto: WebhookMessageDto,\n objectType: WebhookObjectTypeEnum.MESSAGE,\n },\n [WebhookEventEnum.MESSAGE_FAILED]: {\n event: WebhookEventEnum.MESSAGE_FAILED,\n payloadDto: WebhookMessageFailedDto,\n objectType: WebhookObjectTypeEnum.MESSAGE,\n },\n [WebhookEventEnum.MESSAGE_DELIVERED]: {\n event: WebhookEventEnum.MESSAGE_DELIVERED,\n payloadDto: WebhookMessageDto,\n objectType: WebhookObjectTypeEnum.MESSAGE,\n },\n [WebhookEventEnum.MESSAGE_SEEN]: {\n event: WebhookEventEnum.MESSAGE_SEEN,\n payloadDto: WebhookMessageDto,\n objectType: WebhookObjectTypeEnum.MESSAGE,\n },\n [WebhookEventEnum.MESSAGE_READ]: {\n event: WebhookEventEnum.MESSAGE_READ,\n payloadDto: WebhookMessageDto,\n objectType: WebhookObjectTypeEnum.MESSAGE,\n },\n [WebhookEventEnum.MESSAGE_UNREAD]: {\n event: WebhookEventEnum.MESSAGE_UNREAD,\n payloadDto: WebhookMessageDto,\n objectType: WebhookObjectTypeEnum.MESSAGE,\n },\n [WebhookEventEnum.MESSAGE_ARCHIVED]: {\n event: WebhookEventEnum.MESSAGE_ARCHIVED,\n payloadDto: WebhookMessageDto,\n objectType: WebhookObjectTypeEnum.MESSAGE,\n },\n [WebhookEventEnum.MESSAGE_UNARCHIVED]: {\n event: WebhookEventEnum.MESSAGE_UNARCHIVED,\n payloadDto: WebhookMessageDto,\n objectType: WebhookObjectTypeEnum.MESSAGE,\n },\n [WebhookEventEnum.MESSAGE_SNOOZED]: {\n event: WebhookEventEnum.MESSAGE_SNOOZED,\n payloadDto: WebhookMessageDto,\n objectType: WebhookObjectTypeEnum.MESSAGE,\n },\n [WebhookEventEnum.MESSAGE_UNSNOOZED]: {\n event: WebhookEventEnum.MESSAGE_UNSNOOZED,\n payloadDto: WebhookMessageDto,\n objectType: WebhookObjectTypeEnum.MESSAGE,\n },\n [WebhookEventEnum.MESSAGE_DELETED]: {\n event: WebhookEventEnum.MESSAGE_DELETED,\n payloadDto: WebhookMessageDto,\n objectType: WebhookObjectTypeEnum.MESSAGE,\n },\n [WebhookEventEnum.WORKFLOW_CREATED]: {\n event: WebhookEventEnum.WORKFLOW_CREATED,\n payloadDto: WebhookCreatedWorkflowDto,\n objectType: WebhookObjectTypeEnum.WORKFLOW,\n },\n [WebhookEventEnum.WORKFLOW_UPDATED]: {\n event: WebhookEventEnum.WORKFLOW_UPDATED,\n payloadDto: WebhookUpdatedWorkflowDto,\n objectType: WebhookObjectTypeEnum.WORKFLOW,\n },\n [WebhookEventEnum.WORKFLOW_DELETED]: {\n event: WebhookEventEnum.WORKFLOW_DELETED,\n payloadDto: WebhookDeletedWorkflowDto,\n objectType: WebhookObjectTypeEnum.WORKFLOW,\n },\n [WebhookEventEnum.WORKFLOW_PUBLISHED]: {\n event: WebhookEventEnum.WORKFLOW_PUBLISHED,\n payloadDto: WebhookUpdatedWorkflowDto,\n objectType: WebhookObjectTypeEnum.WORKFLOW,\n },\n [WebhookEventEnum.PREFERENCE_UPDATED]: {\n event: WebhookEventEnum.PREFERENCE_UPDATED,\n payloadDto: WebhookPreferenceDto,\n objectType: WebhookObjectTypeEnum.PREFERENCE,\n },\n} as const satisfies WebhookEventRecord;\n\n// Helper function to ensure all enum values are present exactly once\nfunction createWebhookEvents(record: T): WebhookEventConfig[] {\n return Object.values(record);\n}\n\n// Export the webhook events array created from the type-safe record\nexport const webhookEvents = createWebhookEvents(webhookEventRecord);\n"
},
{
"path": "apps/api/src/app/partner-integrations/dtos/create-vercel-integration-request.dto.ts",
"content": "import { IsDefined, IsString } from 'class-validator';\n\nexport class CreateVercelIntegrationRequestDto {\n @IsDefined()\n @IsString()\n vercelIntegrationCode: string;\n\n @IsDefined()\n @IsString()\n configurationId: string;\n}\n"
},
{
"path": "apps/api/src/app/partner-integrations/dtos/create-vercel-integration-response.dto.ts",
"content": "import { IsDefined } from 'class-validator';\n\nexport class CreateVercelIntegrationResponseDto {\n @IsDefined()\n success: boolean;\n}\n"
},
{
"path": "apps/api/src/app/partner-integrations/dtos/update-vercel-integration-request.dto.ts",
"content": "import { IsDefined, IsString } from 'class-validator';\n\nexport class UpdateVercelIntegrationRequestDto {\n @IsDefined()\n data: Record;\n\n @IsDefined()\n @IsString()\n configurationId: string;\n}\n"
},
{
"path": "apps/api/src/app/partner-integrations/partner-integrations.controller.ts",
"content": "import {\n Body,\n ClassSerializerInterceptor,\n Controller,\n Get,\n Headers,\n Param,\n Post,\n Put,\n Query,\n UseInterceptors,\n} from '@nestjs/common';\nimport { ApiExcludeController, ApiTags } from '@nestjs/swagger';\nimport { RequirePermissions } from '@novu/application-generic';\nimport { PermissionsEnum, UserSessionData } from '@novu/shared';\nimport { RequireAuthentication } from '../auth/framework/auth.decorator';\nimport { UserSession } from '../shared/framework/user.decorator';\nimport { CreateVercelIntegrationRequestDto } from './dtos/create-vercel-integration-request.dto';\nimport { CreateVercelIntegrationResponseDto } from './dtos/create-vercel-integration-response.dto';\nimport { UpdateVercelIntegrationRequestDto } from './dtos/update-vercel-integration-request.dto';\nimport { CreateVercelIntegrationCommand } from './usecases/create-vercel-integration/create-vercel-integration.command';\nimport { CreateVercelIntegration } from './usecases/create-vercel-integration/create-vercel-integration.usecase';\nimport { GetVercelIntegrationCommand } from './usecases/get-vercel-integration/get-vercel-integration.command';\nimport { GetVercelIntegration } from './usecases/get-vercel-integration/get-vercel-integration.usecase';\nimport { GetVercelIntegrationProjectsCommand } from './usecases/get-vercel-projects/get-vercel-integration-projects.command';\nimport { GetVercelIntegrationProjects } from './usecases/get-vercel-projects/get-vercel-integration-projects.usecase';\nimport { ProcessVercelWebhookCommand } from './usecases/process-vercel-webhook/process-vercel-webhook.command';\nimport { ProcessVercelWebhook } from './usecases/process-vercel-webhook/process-vercel-webhook.usecase';\nimport { UpdateVercelIntegrationCommand } from './usecases/update-vercel-integration/update-vercel-integration.command';\nimport { UpdateVercelIntegration } from './usecases/update-vercel-integration/update-vercel-integration.usecase';\n\n@Controller('/partner-integrations')\n@UseInterceptors(ClassSerializerInterceptor)\n@ApiTags('Partner Integrations')\n@ApiExcludeController()\nexport class PartnerIntegrationsController {\n constructor(\n private createVercelIntegrationUsecase: CreateVercelIntegration,\n private getVercelIntegrationProjectsUsecase: GetVercelIntegrationProjects,\n private getVercelIntegrationUsecase: GetVercelIntegration,\n private updateVercelIntegrationUsecase: UpdateVercelIntegration,\n private processVercelWebhookUsecase: ProcessVercelWebhook\n ) {}\n\n @Post('/vercel')\n @RequireAuthentication()\n @RequirePermissions(PermissionsEnum.PARTNER_INTEGRATION_WRITE)\n async createVercelIntegration(\n @UserSession() user: UserSessionData,\n @Body() body: CreateVercelIntegrationRequestDto\n ): Promise {\n return await this.createVercelIntegrationUsecase.execute(\n CreateVercelIntegrationCommand.create({\n vercelIntegrationCode: body.vercelIntegrationCode,\n environmentId: user.environmentId,\n organizationId: user.organizationId,\n userId: user._id,\n configurationId: body.configurationId,\n })\n );\n }\n\n @Put('/vercel')\n @RequireAuthentication()\n @RequirePermissions(PermissionsEnum.PARTNER_INTEGRATION_WRITE)\n async updateVercelIntegration(@UserSession() user: UserSessionData, @Body() body: UpdateVercelIntegrationRequestDto) {\n return await this.updateVercelIntegrationUsecase.execute(\n UpdateVercelIntegrationCommand.create({\n data: body.data,\n environmentId: user.environmentId,\n organizationId: user.organizationId,\n userId: user._id,\n configurationId: body.configurationId,\n })\n );\n }\n\n @Get('/vercel/:configurationId')\n @RequireAuthentication()\n @RequirePermissions(PermissionsEnum.PARTNER_INTEGRATION_READ)\n async getVercelIntegration(@UserSession() user: UserSessionData, @Param('configurationId') configurationId: string) {\n return await this.getVercelIntegrationUsecase.execute(\n GetVercelIntegrationCommand.create({\n userId: user._id,\n configurationId,\n environmentId: user.environmentId,\n organizationId: user.organizationId,\n })\n );\n }\n\n @Get('/vercel/:configurationId/projects')\n @RequireAuthentication()\n @RequirePermissions(PermissionsEnum.PARTNER_INTEGRATION_READ)\n async getVercelProjects(\n @UserSession() user: UserSessionData,\n @Param('configurationId') configurationId: string,\n @Query('nextPage') nextPage?: string\n ) {\n return await this.getVercelIntegrationProjectsUsecase.execute(\n GetVercelIntegrationProjectsCommand.create({\n configurationId,\n environmentId: user.environmentId,\n organizationId: user.organizationId,\n userId: user._id,\n ...(nextPage && { nextPage }),\n })\n );\n }\n\n @Post('/vercel/webhook')\n async webhook(@Body() body: any, @Headers('x-vercel-signature') signatureHeader: string) {\n return this.processVercelWebhookUsecase.execute(\n ProcessVercelWebhookCommand.create({\n body,\n signatureHeader,\n })\n );\n }\n}\n"
},
{
"path": "apps/api/src/app/partner-integrations/partner-integrations.module.ts",
"content": "import { HttpModule } from '@nestjs/axios';\nimport { Module } from '@nestjs/common';\nimport { CommunityOrganizationRepository, CommunityUserRepository } from '@novu/dal';\nimport { BridgeModule } from '../bridge';\nimport { EnvironmentsModuleV1 } from '../environments-v1/environments-v1.module';\nimport { SharedModule } from '../shared/shared.module';\nimport { PartnerIntegrationsController } from './partner-integrations.controller';\nimport { USE_CASES } from './usecases';\n\n@Module({\n imports: [SharedModule, HttpModule, EnvironmentsModuleV1, BridgeModule],\n providers: [...USE_CASES, CommunityUserRepository, CommunityOrganizationRepository],\n controllers: [PartnerIntegrationsController],\n})\nexport class PartnerIntegrationsModule {}\n"
},
{
"path": "apps/api/src/app/partner-integrations/usecases/create-vercel-integration/create-vercel-integration.command.ts",
"content": "import { IsDefined } from 'class-validator';\nimport { EnvironmentWithUserCommand } from '../../../shared/commands/project.command';\n\nexport class CreateVercelIntegrationCommand extends EnvironmentWithUserCommand {\n @IsDefined()\n vercelIntegrationCode: string;\n @IsDefined()\n configurationId: string;\n}\n"
},
{
"path": "apps/api/src/app/partner-integrations/usecases/create-vercel-integration/create-vercel-integration.spec.ts",
"content": "import { HttpService } from '@nestjs/axios';\nimport { BadRequestException } from '@nestjs/common';\nimport { Test } from '@nestjs/testing';\nimport { AnalyticsService } from '@novu/application-generic';\nimport { OrganizationRepository, PartnerTypeEnum } from '@novu/dal';\nimport { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\nimport { of } from 'rxjs';\nimport { assert, match, restore, stub } from 'sinon';\nimport { CreateVercelIntegration } from './create-vercel-integration.usecase';\n\ndescribe('CreateVercelIntegration', () => {\n let createVercelIntegration: CreateVercelIntegration;\n let session: UserSession;\n let httpServiceMock;\n let organizationRepositoryMock;\n let analyticsServiceMock;\n beforeEach(async () => {\n httpServiceMock = {\n post: stub().returns(\n of({\n data: {\n access_token: 'test-token',\n team_id: 'test-team-id',\n },\n })\n ),\n };\n\n organizationRepositoryMock = {\n upsertPartnerConfiguration: stub().resolves(true),\n };\n\n analyticsServiceMock = {\n track: stub().resolves(),\n };\n\n const moduleRef = await Test.createTestingModule({\n providers: [\n CreateVercelIntegration,\n {\n provide: HttpService,\n useValue: httpServiceMock,\n },\n {\n provide: OrganizationRepository,\n useValue: organizationRepositoryMock,\n },\n { provide: AnalyticsService, useValue: analyticsServiceMock },\n ],\n }).compile();\n\n session = new UserSession();\n await session.initialize();\n createVercelIntegration = moduleRef.get(CreateVercelIntegration);\n\n // @ts-ignore\n process.env.VERCEL_CLIENT_ID = 'test-client-id';\n // @ts-ignore\n process.env.VERCEL_CLIENT_SECRET = 'test-client-secret';\n // @ts-ignore\n process.env.VERCEL_REDIRECT_URI = 'test-redirect-uri';\n // @ts-ignore\n process.env.VERCEL_BASE_URL = 'https://api.vercel.com';\n });\n\n afterEach(() => {\n restore();\n });\n\n it('should successfully set vercel configuration', async () => {\n const command = {\n organizationId: session.organization._id,\n vercelIntegrationCode: 'test-code',\n configurationId: 'test-config-id',\n userId: session.user._id,\n environmentId: session.environment._id,\n };\n\n const result = await createVercelIntegration.execute(command);\n\n expect(result.success).to.equal(true);\n\n // Verify HTTP call\n assert.calledWith(\n httpServiceMock.post,\n 'https://api.vercel.com/v2/oauth/access_token',\n match.instanceOf(URLSearchParams),\n {\n headers: {\n 'Content-Type': 'application/x-www-form-urlencoded',\n },\n }\n );\n\n // Verify the URLSearchParams content\n const postCall = httpServiceMock.post.getCall(0);\n const [, postData] = postCall.args;\n expect(postData.get('code')).to.equal('test-code');\n expect(postData.get('client_id')).to.equal('test-client-id');\n expect(postData.get('client_secret')).to.equal('test-client-secret');\n expect(postData.get('redirect_uri')).to.equal('test-redirect-uri');\n\n // Verify organization repository call\n assert.calledWith(organizationRepositoryMock.upsertPartnerConfiguration, {\n organizationId: command.organizationId,\n configuration: {\n accessToken: 'test-token',\n configurationId: command.configurationId,\n teamId: 'test-team-id',\n partnerType: PartnerTypeEnum.VERCEL,\n },\n });\n\n assert.calledWith(\n analyticsServiceMock.track,\n 'Create Vercel Integration - [Partner Integrations]',\n command.userId,\n { _organization: command.organizationId }\n );\n });\n\n it('should throw BadRequestException when Vercel returns an error', async () => {\n httpServiceMock.post.throws(new Error('Vercel error'));\n\n try {\n await createVercelIntegration.execute({\n userId: session.user._id,\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n vercelIntegrationCode: 'test-code',\n configurationId: 'test-config-id',\n });\n throw new Error('Should not reach here');\n } catch (error) {\n expect(error).to.be.instanceOf(BadRequestException);\n expect(error.message).to.equal('Vercel error');\n assert.notCalled(organizationRepositoryMock.upsertPartnerConfiguration);\n }\n });\n});\n"
},
{
"path": "apps/api/src/app/partner-integrations/usecases/create-vercel-integration/create-vercel-integration.usecase.ts",
"content": "import { HttpService } from '@nestjs/axios';\nimport { BadRequestException, Injectable } from '@nestjs/common';\nimport { AnalyticsService } from '@novu/application-generic';\nimport { OrganizationRepository, PartnerTypeEnum } from '@novu/dal';\nimport { lastValueFrom } from 'rxjs';\n\nimport { CreateVercelIntegrationResponseDto } from '../../dtos/create-vercel-integration-response.dto';\nimport { CreateVercelIntegrationCommand } from './create-vercel-integration.command';\n\n@Injectable()\nexport class CreateVercelIntegration {\n constructor(\n private httpService: HttpService,\n private organizationRepository: OrganizationRepository,\n private analyticsService: AnalyticsService\n ) {}\n\n async execute(command: CreateVercelIntegrationCommand): Promise {\n try {\n const tokenData = await this.getVercelToken(command.vercelIntegrationCode);\n\n const configuration = {\n accessToken: tokenData.accessToken,\n configurationId: command.configurationId,\n teamId: tokenData.teamId,\n partnerType: PartnerTypeEnum.VERCEL,\n };\n\n await this.organizationRepository.upsertPartnerConfiguration({\n organizationId: command.organizationId,\n configuration,\n });\n\n this.analyticsService.track('Create Vercel Integration - [Partner Integrations]', command.userId, {\n _organization: command.organizationId,\n });\n\n return {\n success: true,\n };\n } catch (error) {\n throw new BadRequestException(\n error?.response?.data?.error_description || error?.response?.data?.message || error.message\n );\n }\n }\n\n private async getVercelToken(code: string): Promise<{\n accessToken: string;\n teamId: string;\n }> {\n try {\n const postData = new URLSearchParams({\n code: code as string,\n client_id: process.env.VERCEL_CLIENT_ID as string,\n client_secret: process.env.VERCEL_CLIENT_SECRET as string,\n redirect_uri: process.env.VERCEL_REDIRECT_URI as string,\n });\n\n const response = await lastValueFrom(\n this.httpService.post(`${process.env.VERCEL_BASE_URL}/v2/oauth/access_token`, postData, {\n headers: {\n 'Content-Type': 'application/x-www-form-urlencoded',\n },\n })\n );\n\n const { data } = response;\n\n return {\n accessToken: data.access_token,\n teamId: data.team_id,\n };\n } catch (error) {\n throw new BadRequestException(\n error?.response?.data?.error_description || error?.response?.data?.message || error.message\n );\n }\n }\n}\n"
},
{
"path": "apps/api/src/app/partner-integrations/usecases/get-vercel-integration/get-vercel-integration.command.ts",
"content": "import { EnvironmentWithUserCommand } from '../../../shared/commands/project.command';\n\nexport class GetVercelIntegrationCommand extends EnvironmentWithUserCommand {\n configurationId: string;\n}\n"
},
{
"path": "apps/api/src/app/partner-integrations/usecases/get-vercel-integration/get-vercel-integration.spec.ts",
"content": "import { Test } from '@nestjs/testing';\nimport { OrganizationRepository } from '@novu/dal';\nimport { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\nimport { assert, restore, stub } from 'sinon';\n\nimport { GetVercelIntegration } from './get-vercel-integration.usecase';\n\ndescribe('GetVercelIntegration', () => {\n let getVercelIntegration: GetVercelIntegration;\n let session: UserSession;\n let organizationRepositoryMock;\n\n beforeEach(async () => {\n organizationRepositoryMock = {\n findByPartnerConfigurationId: stub().resolves([\n {\n _id: 'org-id-1',\n partnerConfigurations: [\n {\n projectIds: ['project-1', 'project-2'],\n },\n ],\n },\n {\n _id: 'org-id-2',\n partnerConfigurations: [\n {\n projectIds: ['project-2', 'project-3'],\n },\n ],\n },\n ]),\n };\n\n const moduleRef = await Test.createTestingModule({\n providers: [\n GetVercelIntegration,\n {\n provide: OrganizationRepository,\n useValue: organizationRepositoryMock,\n },\n ],\n }).compile();\n\n session = new UserSession();\n await session.initialize();\n getVercelIntegration = moduleRef.get(GetVercelIntegration);\n });\n\n afterEach(() => {\n restore();\n });\n\n it('should get vercel configuration details', async () => {\n const command = {\n userId: session.user._id,\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n configurationId: 'test-config-id',\n };\n\n const result = await getVercelIntegration.execute(command);\n\n expect(result).to.be.an('array');\n expect(result[0]).to.deep.equal({\n organizationId: 'org-id-1',\n projectIds: ['project-1', 'project-2'],\n });\n expect(result[1]).to.deep.equal({\n organizationId: 'org-id-2',\n projectIds: ['project-2', 'project-3'],\n });\n\n assert.calledOnceWithExactly(organizationRepositoryMock.findByPartnerConfigurationId, {\n userId: command.userId,\n configurationId: command.configurationId,\n });\n });\n\n it('should return empty array when no configurations found', async () => {\n organizationRepositoryMock.findByPartnerConfigurationId.resolves([]);\n\n const command = {\n userId: session.user._id,\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n configurationId: 'test-config-id',\n };\n\n const result = await getVercelIntegration.execute(command);\n\n expect(result).to.be.an('array');\n expect(result).to.have.length(0);\n\n assert.calledOnceWithExactly(organizationRepositoryMock.findByPartnerConfigurationId, {\n userId: command.userId,\n configurationId: command.configurationId,\n });\n });\n});\n"
},
{
"path": "apps/api/src/app/partner-integrations/usecases/get-vercel-integration/get-vercel-integration.usecase.ts",
"content": "import { Injectable } from '@nestjs/common';\nimport { OrganizationRepository } from '@novu/dal';\n\nimport { GetVercelIntegrationCommand } from './get-vercel-integration.command';\n\n@Injectable()\nexport class GetVercelIntegration {\n constructor(private organizationRepository: OrganizationRepository) {}\n\n async execute(command: GetVercelIntegrationCommand) {\n return await this.getConfigurationDetails(command);\n }\n\n private async getConfigurationDetails(command: GetVercelIntegrationCommand) {\n const details = await this.organizationRepository.findByPartnerConfigurationId({\n userId: command.userId,\n configurationId: command.configurationId,\n });\n\n return details.reduce(\n (acc, curr) => {\n if (\n curr.partnerConfigurations &&\n curr.partnerConfigurations.length >= 1 &&\n curr.partnerConfigurations[0].projectIds &&\n curr.partnerConfigurations[0].projectIds.length >= 1\n ) {\n acc.push({\n organizationId: curr._id,\n projectIds: curr.partnerConfigurations[0].projectIds,\n });\n }\n\n return acc;\n },\n [] as { organizationId: string; projectIds: string[] }[]\n );\n }\n}\n"
},
{
"path": "apps/api/src/app/partner-integrations/usecases/get-vercel-projects/get-vercel-integration-projects.command.ts",
"content": "import { IsDefined, IsOptional, IsString } from 'class-validator';\nimport { EnvironmentWithUserCommand } from '../../../shared/commands/project.command';\n\nexport class GetVercelIntegrationProjectsCommand extends EnvironmentWithUserCommand {\n @IsDefined()\n @IsString()\n configurationId: string;\n\n @IsOptional()\n @IsString()\n nextPage?: string;\n}\n"
},
{
"path": "apps/api/src/app/partner-integrations/usecases/get-vercel-projects/get-vercel-integration-projects.spec.ts",
"content": "import { HttpService } from '@nestjs/axios';\nimport { BadRequestException } from '@nestjs/common';\nimport { Test } from '@nestjs/testing';\nimport { OrganizationRepository } from '@novu/dal';\nimport { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\nimport { of } from 'rxjs';\nimport { assert, restore, stub } from 'sinon';\nimport { GetVercelIntegrationProjects } from './get-vercel-integration-projects.usecase';\n\ndescribe('GetVercelIntegrationProjects', () => {\n let getVercelIntegrationProjects: GetVercelIntegrationProjects;\n let session: UserSession;\n let httpServiceMock;\n let organizationRepositoryMock;\n\n beforeEach(async () => {\n httpServiceMock = {\n get: stub().returns(\n of({\n data: {\n projects: [\n { id: 'project-1', name: 'Project One' },\n { id: 'project-2', name: 'Project Two' },\n ],\n pagination: {\n next: 'next-page-token',\n },\n },\n })\n ),\n };\n\n organizationRepositoryMock = {\n findByPartnerConfigurationId: stub().resolves([\n {\n partnerConfigurations: [\n {\n configurationId: 'test-config-id',\n accessToken: 'test-token',\n teamId: 'test-team-id',\n },\n ],\n },\n ]),\n };\n\n const moduleRef = await Test.createTestingModule({\n providers: [\n GetVercelIntegrationProjects,\n {\n provide: HttpService,\n useValue: httpServiceMock,\n },\n {\n provide: OrganizationRepository,\n useValue: organizationRepositoryMock,\n },\n ],\n }).compile();\n\n session = new UserSession();\n await session.initialize();\n getVercelIntegrationProjects = moduleRef.get(GetVercelIntegrationProjects);\n });\n\n afterEach(() => {\n restore();\n });\n\n it('should get vercel projects successfully', async () => {\n const command = {\n userId: session.user._id,\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n configurationId: 'test-config-id',\n };\n\n const result = await getVercelIntegrationProjects.execute(command);\n\n expect(result.projects).to.have.length(2);\n expect(result.projects[0]).to.deep.equal({\n name: 'Project One',\n id: 'project-1',\n });\n expect(result.pagination).to.deep.equal({\n next: 'next-page-token',\n });\n\n assert.calledWith(organizationRepositoryMock.findByPartnerConfigurationId, {\n userId: command.userId,\n configurationId: command.configurationId,\n });\n\n const expectedUrl = `${process.env.VERCEL_BASE_URL}/v10/projects?limit=100&teamId=test-team-id`;\n assert.calledWith(httpServiceMock.get, expectedUrl, {\n headers: {\n Authorization: 'Bearer test-token',\n },\n });\n });\n\n it('should throw BadRequestException when no configuration found', async () => {\n organizationRepositoryMock.findByPartnerConfigurationId.resolves([]);\n\n try {\n await getVercelIntegrationProjects.execute({\n userId: session.user._id,\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n configurationId: 'test-config-id',\n });\n throw new Error('Should not reach here');\n } catch (error) {\n expect(error).to.be.instanceOf(BadRequestException);\n expect(error.message).to.equal('No partner configuration found.');\n assert.notCalled(httpServiceMock.get);\n }\n });\n\n it('should throw BadRequestException when HTTP request fails', async () => {\n httpServiceMock.get.throws(new Error('HTTP Error'));\n\n try {\n await getVercelIntegrationProjects.execute({\n userId: session.user._id,\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n configurationId: 'test-config-id',\n });\n throw new Error('Should not reach here');\n } catch (error) {\n expect(error).to.be.instanceOf(BadRequestException);\n expect(error.message).to.equal('HTTP Error');\n assert.called(httpServiceMock.get);\n }\n });\n});\n"
},
{
"path": "apps/api/src/app/partner-integrations/usecases/get-vercel-projects/get-vercel-integration-projects.usecase.ts",
"content": "import { HttpService } from '@nestjs/axios';\nimport { BadRequestException, Injectable } from '@nestjs/common';\nimport { OrganizationRepository } from '@novu/dal';\nimport { lastValueFrom } from 'rxjs';\n\nimport { GetVercelIntegrationProjectsCommand } from './get-vercel-integration-projects.command';\n\n@Injectable()\nexport class GetVercelIntegrationProjects {\n constructor(\n private httpService: HttpService,\n private organizationRepository: OrganizationRepository\n ) {}\n\n async execute(command: GetVercelIntegrationProjectsCommand) {\n try {\n const configuration = await this.getCurrentOrgPartnerConfiguration({\n userId: command.userId,\n configurationId: command.configurationId,\n });\n\n if (!configuration || !configuration.accessToken) {\n throw new BadRequestException({\n message: 'No partner configuration found.',\n type: 'vercel',\n });\n }\n\n const projects = await this.getVercelProjects(configuration.accessToken, configuration.teamId, command.nextPage);\n\n return projects;\n } catch (error) {\n throw new BadRequestException(error.message);\n }\n }\n\n async getCurrentOrgPartnerConfiguration({ userId, configurationId }: { userId: string; configurationId: string }) {\n const orgsWithIntegration = await this.organizationRepository.findByPartnerConfigurationId({\n userId,\n configurationId,\n });\n\n if (orgsWithIntegration.length === 0) {\n throw new BadRequestException({\n message: 'No partner configuration found.',\n type: 'vercel',\n });\n }\n\n const firstOrg = orgsWithIntegration[0];\n const configuration = firstOrg.partnerConfigurations?.find((config) => config.configurationId === configurationId);\n if (!firstOrg.partnerConfigurations?.length || !configuration) {\n throw new BadRequestException({\n message: 'No partner configuration found',\n type: 'vercel',\n });\n }\n\n return configuration;\n }\n\n private async getVercelProjects(accessToken: string, teamId: string | null, until?: string) {\n const queryParams = new URLSearchParams();\n queryParams.set('limit', '100');\n\n if (teamId) {\n queryParams.set('teamId', teamId);\n }\n\n if (until) {\n queryParams.set('until', until);\n }\n\n const response = await lastValueFrom(\n this.httpService.get(`${process.env.VERCEL_BASE_URL}/v10/projects?${queryParams.toString()}`, {\n headers: {\n Authorization: `Bearer ${accessToken}`,\n },\n })\n );\n\n return { projects: this.mapProjects(response.data.projects), pagination: response.data.pagination };\n }\n\n private mapProjects(projects) {\n return projects.map((project) => {\n return {\n name: project.name,\n id: project.id,\n };\n });\n }\n}\n"
},
{
"path": "apps/api/src/app/partner-integrations/usecases/index.ts",
"content": "import { CreateVercelIntegration } from './create-vercel-integration/create-vercel-integration.usecase';\nimport { GetVercelIntegration } from './get-vercel-integration/get-vercel-integration.usecase';\nimport { GetVercelIntegrationProjects } from './get-vercel-projects/get-vercel-integration-projects.usecase';\nimport { ProcessVercelWebhook } from './process-vercel-webhook/process-vercel-webhook.usecase';\nimport { UpdateVercelIntegration } from './update-vercel-integration/update-vercel-integration.usecase';\n\nexport const USE_CASES = [\n CreateVercelIntegration,\n GetVercelIntegrationProjects,\n GetVercelIntegration,\n UpdateVercelIntegration,\n ProcessVercelWebhook,\n];\n"
},
{
"path": "apps/api/src/app/partner-integrations/usecases/process-vercel-webhook/process-vercel-webhook.command.ts",
"content": "import { BaseCommand } from '@novu/application-generic';\nimport { IsDefined } from 'class-validator';\n\nexport class ProcessVercelWebhookCommand extends BaseCommand {\n @IsDefined()\n signatureHeader: string;\n\n @IsDefined()\n body: any;\n}\n"
},
{
"path": "apps/api/src/app/partner-integrations/usecases/process-vercel-webhook/process-vercel-webhook.spec.ts",
"content": "import crypto from 'node:crypto';\nimport { BadRequestException } from '@nestjs/common';\nimport { Test } from '@nestjs/testing';\nimport { PinoLogger } from '@novu/application-generic';\nimport {\n CommunityOrganizationRepository,\n CommunityUserRepository,\n EnvironmentRepository,\n MemberRepository,\n} from '@novu/dal';\nimport { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\nimport { assert, restore, stub } from 'sinon';\nimport { Sync } from '../../../bridge/usecases/sync';\nimport { ProcessVercelWebhook } from './process-vercel-webhook.usecase';\n\ndescribe('ProcessVercelWebhook', () => {\n let processVercelWebhook: ProcessVercelWebhook;\n let session: UserSession;\n let organizationRepositoryMock;\n let environmentRepositoryMock;\n let memberRepositoryMock;\n let communityUserRepositoryMock;\n let syncUsecaseMock;\n let loggerMock;\n beforeEach(async () => {\n organizationRepositoryMock = {\n find: stub().resolves([{ _id: 'test-org-id' }]),\n };\n\n environmentRepositoryMock = {\n findOne: stub().resolves({\n _id: 'test-env-id',\n _organizationId: 'test-org-id',\n }),\n };\n\n memberRepositoryMock = {\n getOrganizationOwnerAccount: stub().resolves({\n _userId: 'test-user-id',\n }),\n };\n\n communityUserRepositoryMock = {\n findOne: stub().resolves({\n _id: 'test-internal-user-id',\n }),\n };\n\n syncUsecaseMock = {\n execute: stub().resolves(true),\n };\n\n loggerMock = {\n info: stub(),\n error: stub(),\n warn: stub(),\n debug: stub(),\n trace: stub(),\n setContext: stub(),\n };\n\n const moduleRef = await Test.createTestingModule({\n providers: [\n ProcessVercelWebhook,\n {\n provide: CommunityOrganizationRepository,\n useValue: organizationRepositoryMock,\n },\n {\n provide: EnvironmentRepository,\n useValue: environmentRepositoryMock,\n },\n {\n provide: MemberRepository,\n useValue: memberRepositoryMock,\n },\n {\n provide: CommunityUserRepository,\n useValue: communityUserRepositoryMock,\n },\n {\n provide: Sync,\n useValue: syncUsecaseMock,\n },\n {\n provide: PinoLogger,\n useValue: loggerMock,\n },\n ],\n }).compile();\n\n // @ts-ignore\n process.env.VERCEL_CLIENT_SECRET = 'test-secret';\n session = new UserSession();\n await session.initialize();\n processVercelWebhook = moduleRef.get(ProcessVercelWebhook);\n });\n\n afterEach(() => {\n restore();\n });\n\n it('should skip non-deployment events', async () => {\n const result = await processVercelWebhook.execute({\n body: {\n type: 'other-event',\n },\n signatureHeader: 'test-signature',\n });\n\n expect(result).to.equal(true);\n assert.notCalled(organizationRepositoryMock.find);\n });\n\n it('should process deployment succeeded event', async () => {\n const body = {\n type: 'deployment.succeeded',\n payload: {\n team: { id: 'team-id' },\n project: { id: 'project-id' },\n deployment: { url: 'test.vercel.app' },\n target: 'production',\n },\n };\n\n const hmac = crypto\n .createHmac('sha1', process.env.VERCEL_CLIENT_SECRET ?? '')\n .update(JSON.stringify(body))\n .digest('hex');\n\n const result = await processVercelWebhook.execute({\n body,\n signatureHeader: hmac,\n });\n\n expect(result).to.equal(true);\n\n assert.calledWith(organizationRepositoryMock.find, {\n 'partnerConfigurations.teamId': 'team-id',\n 'partnerConfigurations.projectIds': 'project-id',\n });\n\n assert.calledWith(environmentRepositoryMock.findOne, {\n _organizationId: 'test-org-id',\n name: 'Production',\n });\n\n assert.calledWith(memberRepositoryMock.getOrganizationOwnerAccount, 'test-org-id');\n\n assert.calledWith(communityUserRepositoryMock.findOne, {\n externalId: 'test-user-id',\n });\n\n assert.calledWith(syncUsecaseMock.execute, {\n organizationId: 'test-org-id',\n userId: 'test-internal-user-id',\n environmentId: 'test-env-id',\n bridgeUrl: 'https://test.vercel.app/api/novu',\n source: 'vercel',\n });\n });\n\n it('should throw error for invalid signature', async () => {\n const body = {\n type: 'deployment.succeeded',\n payload: {\n team: { id: 'team-id' },\n project: { id: 'project-id' },\n deployment: { url: 'test.vercel.app' },\n target: 'production',\n },\n };\n\n try {\n await processVercelWebhook.execute({\n body,\n signatureHeader: 'invalid-signature',\n });\n throw new Error('Should not reach here');\n } catch (error) {\n expect(error).to.be.instanceOf(BadRequestException);\n expect(error.message).to.equal('Invalid signature');\n assert.notCalled(organizationRepositoryMock.find);\n }\n });\n\n it('should throw error for missing signature', async () => {\n const body = {\n type: 'deployment.succeeded',\n payload: {\n team: { id: 'team-id' },\n project: { id: 'project-id' },\n deployment: { url: 'test.vercel.app' },\n target: 'production',\n },\n };\n\n try {\n await processVercelWebhook.execute({\n body,\n signatureHeader: '',\n });\n throw new Error('Should not reach here');\n } catch (error) {\n expect(error).to.be.instanceOf(BadRequestException);\n expect(error.message).to.equal('Missing signature or secret');\n assert.notCalled(organizationRepositoryMock.find);\n }\n });\n});\n"
},
{
"path": "apps/api/src/app/partner-integrations/usecases/process-vercel-webhook/process-vercel-webhook.usecase.ts",
"content": "import crypto from 'node:crypto';\nimport { BadRequestException, HttpException, Injectable, InternalServerErrorException } from '@nestjs/common';\nimport { PinoLogger } from '@novu/application-generic';\nimport {\n CommunityOrganizationRepository,\n CommunityUserRepository,\n EnvironmentEntity,\n EnvironmentRepository,\n MemberRepository,\n} from '@novu/dal';\nimport { Sync } from '../../../bridge/usecases/sync';\nimport { ProcessVercelWebhookCommand } from './process-vercel-webhook.command';\n\n@Injectable()\nexport class ProcessVercelWebhook {\n constructor(\n private organizationRepository: CommunityOrganizationRepository,\n private environmentRepository: EnvironmentRepository,\n private syncUsecase: Sync,\n private memberRepository: MemberRepository,\n private communityUserRepository: CommunityUserRepository,\n private logger: PinoLogger\n ) {\n this.logger.setContext(this.constructor.name);\n }\n\n async execute(command: ProcessVercelWebhookCommand) {\n const eventType = command.body.type;\n if (eventType !== 'deployment.succeeded') {\n this.logger.info(`Skipping processing Vercel webhook event: ${eventType}`);\n\n return true;\n }\n\n this.verifySignature(command.signatureHeader, command.body);\n\n const payload = command.body.payload;\n if (!payload?.team?.id || !payload?.project?.id || !payload?.deployment?.url) {\n throw new BadRequestException('Invalid webhook payload: missing required fields');\n }\n\n const teamId = payload.team.id;\n const projectId = payload.project.id;\n const deploymentUrl = payload.deployment.url;\n const vercelEnvironment = payload.target || 'preview';\n\n this.logger.info(\n {\n teamId,\n projectId,\n vercelEnvironment,\n deploymentUrl,\n },\n `Processing vercel webhook for ${vercelEnvironment}`\n );\n\n const organizations = await this.organizationRepository.find(\n {\n 'partnerConfigurations.teamId': teamId,\n 'partnerConfigurations.projectIds': projectId,\n },\n { 'partnerConfigurations.$': 1 }\n );\n\n if (!organizations || organizations.length === 0) {\n throw new BadRequestException('Organization not found for vercel webhook integration');\n }\n\n for (const organization of organizations) {\n let environment: EnvironmentEntity | null;\n\n // TODO: we should think about how to handle different Vercel environments that are not production or development\n if (vercelEnvironment === 'production') {\n environment = await this.environmentRepository.findOne({\n _organizationId: organization._id,\n name: 'Production',\n });\n } else {\n environment = await this.environmentRepository.findOne({\n _organizationId: organization._id,\n name: 'Development',\n });\n }\n\n if (!environment) {\n throw new BadRequestException('Environment Not Found');\n }\n\n try {\n const orgOwner = await this.memberRepository.getOrganizationOwnerAccount(environment._organizationId);\n if (!orgOwner) {\n throw new BadRequestException('Organization owner not found');\n }\n\n const internalUser = await this.communityUserRepository.findOne({ externalId: orgOwner?._userId });\n\n if (!internalUser) {\n throw new BadRequestException('User not found');\n }\n\n await this.syncUsecase.execute({\n organizationId: environment._organizationId,\n userId: internalUser?._id as string,\n environmentId: environment._id,\n bridgeUrl: `https://${deploymentUrl}/api/novu`,\n source: 'vercel',\n });\n } catch (error) {\n if (error instanceof HttpException) {\n throw error;\n }\n\n this.logger.error(\n {\n err: error,\n organizationId: organization._id,\n teamId,\n projectId,\n },\n 'Failed to process Vercel webhook for organization'\n );\n\n throw new InternalServerErrorException(\n `Failed to process Vercel webhook: ${error instanceof Error ? error.message : 'Unknown error'}`\n );\n }\n }\n\n return true;\n }\n\n private verifySignature(signature: string, body: any): void {\n const secret = process.env.VERCEL_CLIENT_SECRET;\n\n if (!signature || !secret) {\n throw new BadRequestException('Missing signature or secret');\n }\n\n const computedSignature = crypto.createHmac('sha1', secret).update(JSON.stringify(body)).digest('hex');\n\n if (signature !== computedSignature) {\n throw new BadRequestException('Invalid signature');\n }\n }\n}\n"
},
{
"path": "apps/api/src/app/partner-integrations/usecases/update-vercel-integration/update-vercel-integration.command.ts",
"content": "import { EnvironmentWithUserCommand } from '../../../shared/commands/project.command';\n\nexport class UpdateVercelIntegrationCommand extends EnvironmentWithUserCommand {\n data: Record;\n configurationId: string;\n}\n"
},
{
"path": "apps/api/src/app/partner-integrations/usecases/update-vercel-integration/update-vercel-integration.spec.ts",
"content": "import { HttpService } from '@nestjs/axios';\nimport { BadRequestException } from '@nestjs/common';\nimport { Test } from '@nestjs/testing';\nimport { AnalyticsService, PinoLogger } from '@novu/application-generic';\nimport { CommunityUserRepository, EnvironmentRepository, MemberRepository, OrganizationRepository } from '@novu/dal';\nimport { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\nimport { of } from 'rxjs';\nimport { assert, restore, stub } from 'sinon';\nimport { Sync } from '../../../bridge/usecases/sync';\nimport { UpdateVercelIntegration } from './update-vercel-integration.usecase';\n\ndescribe('UpdateVercelIntegration', () => {\n let updateVercelIntegration: UpdateVercelIntegration;\n let session: UserSession;\n let httpServiceMock;\n let environmentRepositoryMock;\n let organizationRepositoryMock;\n let analyticsServiceMock;\n let syncMock;\n let memberRepositoryMock;\n let communityUserRepositoryMock;\n let loggerMock;\n\n beforeEach(async () => {\n // @ts-ignore\n process.env.VERCEL_BASE_URL = 'https://api.vercel.com';\n\n httpServiceMock = {\n get: stub().callsFake((url, config) => {\n if (url.includes('/v4/projects') && url.includes('teamId=test-team-id')) {\n return of({\n data: {\n projects: [\n {\n id: 'project-1',\n env: [\n { id: 'env-1', key: 'NEXT_PUBLIC_NOVU_CLIENT_APP_ID', target: ['production'] },\n { id: 'env-2', key: 'NOVU_CLIENT_APP_ID', target: ['production'] },\n { id: 'env-3', key: 'NOVU_SECRET_KEY', target: ['production'] },\n { id: 'env-4', key: 'NEXT_PUBLIC_NOVU_APPLICATION_IDENTIFIER', target: ['production'] },\n ],\n },\n ],\n },\n });\n } else if (url.includes('/v9/projects/project-1') && url.includes('teamId=test-team-id')) {\n return of({\n data: {\n targets: {\n production: {\n alias: ['prod-alias.vercel.app'],\n },\n development: {\n alias: ['dev-alias.vercel.app'],\n },\n },\n },\n });\n }\n\n // Default response for any other URLs\n return of({ data: {} });\n }),\n post: stub().returns(of({ data: { success: true } })),\n delete: stub().returns(of({ data: { success: true } })),\n };\n\n organizationRepositoryMock = {\n findByPartnerConfigurationId: stub().resolves([\n {\n partnerConfigurations: [\n {\n configurationId: 'test-config-id',\n accessToken: 'test-token',\n teamId: 'test-team-id',\n projectIds: ['project-1'],\n },\n ],\n },\n ]),\n bulkUpdatePartnerConfiguration: stub().resolves(true),\n };\n\n analyticsServiceMock = {\n track: stub().resolves(),\n };\n\n syncMock = {\n execute: stub().resolves(),\n };\n\n environmentRepositoryMock = {\n find: stub().resolves([\n {\n _id: 'env-id',\n name: 'Production',\n identifier: 'prod',\n _organizationId: 'org-id',\n apiKeys: [{ key: 'encrypted-key' }],\n },\n {\n _id: 'env-id-2',\n name: 'Development',\n identifier: 'dev',\n _organizationId: 'org-id',\n apiKeys: [{ key: 'encrypted-key-2' }],\n },\n ]),\n };\n\n memberRepositoryMock = {\n getOrganizationOwnerAccount: stub().resolves({ _userId: 'admin-id' }),\n };\n\n communityUserRepositoryMock = {\n findOne: stub().resolves({ _id: 'internal-user-id' }),\n };\n\n loggerMock = {\n log: stub(),\n error: stub(),\n warn: stub(),\n debug: stub(),\n info: stub(),\n trace: stub(),\n setContext: stub(),\n };\n\n const moduleRef = await Test.createTestingModule({\n providers: [\n UpdateVercelIntegration,\n { provide: HttpService, useValue: httpServiceMock },\n { provide: EnvironmentRepository, useValue: environmentRepositoryMock },\n { provide: OrganizationRepository, useValue: organizationRepositoryMock },\n { provide: AnalyticsService, useValue: analyticsServiceMock },\n { provide: Sync, useValue: syncMock },\n { provide: MemberRepository, useValue: memberRepositoryMock },\n { provide: CommunityUserRepository, useValue: communityUserRepositoryMock },\n { provide: PinoLogger, useValue: loggerMock },\n ],\n }).compile();\n\n session = new UserSession();\n await session.initialize();\n updateVercelIntegration = moduleRef.get(UpdateVercelIntegration);\n });\n\n afterEach(() => {\n restore();\n });\n\n it('should update vercel configuration successfully', async () => {\n const command = {\n userId: session.user._id,\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n configurationId: 'test-config-id',\n data: {\n 'org-id': ['project-1'],\n },\n };\n\n const result = await updateVercelIntegration.execute(command);\n\n expect(result.success).to.equal(true);\n\n // Verify existing projects lookup\n assert.calledWith(organizationRepositoryMock.findByPartnerConfigurationId, {\n userId: command.userId,\n configurationId: command.configurationId,\n });\n\n // Verify project environment variables lookup\n assert.calledWith(httpServiceMock.get, `${process.env.VERCEL_BASE_URL}/v4/projects?teamId=test-team-id`, {\n headers: {\n Authorization: 'Bearer test-token',\n },\n });\n\n // Verify environment variable deletion calls\n assert.calledWith(\n httpServiceMock.delete,\n `${process.env.VERCEL_BASE_URL}/v9/projects/project-1/env/env-1?teamId=test-team-id`,\n {\n headers: {\n Authorization: 'Bearer test-token',\n },\n }\n );\n assert.calledWith(\n httpServiceMock.delete,\n `${process.env.VERCEL_BASE_URL}/v9/projects/project-1/env/env-2?teamId=test-team-id`,\n {\n headers: {\n Authorization: 'Bearer test-token',\n },\n }\n );\n assert.calledWith(\n httpServiceMock.delete,\n `${process.env.VERCEL_BASE_URL}/v9/projects/project-1/env/env-3?teamId=test-team-id`,\n {\n headers: {\n Authorization: 'Bearer test-token',\n },\n }\n );\n assert.calledWith(\n httpServiceMock.delete,\n `${process.env.VERCEL_BASE_URL}/v9/projects/project-1/env/env-4?teamId=test-team-id`,\n {\n headers: {\n Authorization: 'Bearer test-token',\n },\n }\n );\n\n assert.calledWith(organizationRepositoryMock.bulkUpdatePartnerConfiguration, {\n userId: command.userId,\n data: command.data,\n configuration: {\n configurationId: 'test-config-id',\n accessToken: 'test-token',\n teamId: 'test-team-id',\n projectIds: ['project-1'],\n },\n });\n\n // Verify environment repository calls\n assert.calledWith(environmentRepositoryMock.find, {\n _organizationId: { $in: ['org-id'] },\n });\n\n // Verify environment variables setup\n assert.calledWith(\n httpServiceMock.post,\n 'https://api.vercel.com/v10/projects/project-1/env?upsert=true&teamId=test-team-id',\n [\n {\n target: ['production'],\n type: 'encrypted',\n value: 'prod',\n key: 'NEXT_PUBLIC_NOVU_APPLICATION_IDENTIFIER',\n },\n ],\n {\n headers: {\n Authorization: 'Bearer test-token',\n 'Content-Type': 'application/json',\n },\n }\n );\n\n // Verify bridge URL update\n assert.calledWith(httpServiceMock.get, 'https://api.vercel.com/v9/projects/project-1?teamId=test-team-id', {\n headers: {\n Authorization: 'Bearer test-token',\n 'Content-Type': 'application/json',\n },\n });\n\n // Verify sync execution\n assert.calledWith(syncMock.execute, {\n organizationId: 'org-id',\n userId: 'internal-user-id',\n environmentId: 'env-id',\n bridgeUrl: 'https://prod-alias.vercel.app/api/novu',\n source: 'vercel',\n });\n\n // Verify analytics\n assert.calledWith(\n analyticsServiceMock.track,\n 'Update Vercel Integration - [Partner Integrations]',\n command.userId,\n { _organization: command.organizationId }\n );\n });\n\n it('should handle projects with no environment variables', async () => {\n // Reset the stub before creating a new behavior\n httpServiceMock.get.reset();\n httpServiceMock.get.callsFake((url) => {\n if (url.includes('/v4/projects')) {\n return of({\n data: {\n projects: [\n {\n id: 'project-1',\n env: [], // Empty env array\n },\n ],\n },\n });\n }\n\n return of({ data: {} });\n });\n\n const command = {\n userId: session.user._id,\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n configurationId: 'test-config-id',\n data: {\n 'org-id': ['project-1'],\n },\n };\n\n const result = await updateVercelIntegration.execute(command);\n\n expect(result.success).to.equal(true);\n assert.notCalled(httpServiceMock.delete);\n });\n\n it('should handle projects with missing Novu environment variables', async () => {\n // Reset the stub before creating a new behavior\n httpServiceMock.get.reset();\n httpServiceMock.get.callsFake((url) => {\n if (url.includes('/v4/projects')) {\n return of({\n data: {\n projects: [\n {\n id: 'project-1',\n env: [{ id: 'env-1', key: 'OTHER_ENV_VAR' }], // Only non-Novu env var\n },\n ],\n },\n });\n }\n\n return of({ data: {} });\n });\n\n const command = {\n userId: session.user._id,\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n configurationId: 'test-config-id',\n data: {\n 'org-id': ['project-1'],\n },\n };\n\n const result = await updateVercelIntegration.execute(command);\n\n expect(result.success).to.equal(true);\n assert.notCalled(httpServiceMock.delete);\n });\n\n it('should throw BadRequestException when configuration not found', async () => {\n organizationRepositoryMock.findByPartnerConfigurationId.resolves([]);\n\n try {\n await updateVercelIntegration.execute({\n userId: session.user._id,\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n configurationId: 'test-config-id',\n data: {},\n });\n throw new Error('Should not reach here');\n } catch (error) {\n expect(error).to.be.instanceOf(BadRequestException);\n expect(error.message).to.equal('No partner configuration found.');\n assert.notCalled(httpServiceMock.get);\n assert.notCalled(httpServiceMock.delete);\n }\n });\n\n it('should handle errors during project fetch', async () => {\n httpServiceMock.get.throws(new Error('HTTP Error'));\n\n try {\n await updateVercelIntegration.execute({\n userId: session.user._id,\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n configurationId: 'test-config-id',\n data: {\n 'org-id': ['project-1'],\n },\n });\n throw new Error('Should not reach here');\n } catch (error) {\n expect(error).to.be.instanceOf(BadRequestException);\n expect(error.message).to.equal('HTTP Error');\n assert.notCalled(httpServiceMock.delete);\n }\n });\n\n it('should handle errors during environment variable deletion', async () => {\n httpServiceMock.delete.onCall(0).throws(new Error('Delete Error'));\n\n try {\n await updateVercelIntegration.execute({\n userId: session.user._id,\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n configurationId: 'test-config-id',\n data: {\n 'org-id': ['project-1'],\n },\n });\n throw new Error('Should not reach here');\n } catch (error) {\n expect(error).to.be.instanceOf(BadRequestException);\n expect(error.message).to.equal('Delete Error');\n assert.called(httpServiceMock.get);\n assert.called(httpServiceMock.delete);\n }\n });\n\n it('should handle multiple projects with environment variables', async () => {\n // Reset the stub before creating a new behavior\n httpServiceMock.get.reset();\n httpServiceMock.get.callsFake((url) => {\n if (url.includes('/v4/projects')) {\n return of({\n data: {\n projects: [\n {\n id: 'project-1',\n env: [{ id: 'env-1', key: 'NEXT_PUBLIC_NOVU_CLIENT_APP_ID', target: ['production'] }],\n },\n {\n id: 'project-2',\n env: [{ id: 'env-2', key: 'NOVU_SECRET_KEY', target: ['production'] }],\n },\n ],\n },\n });\n } else if (url.includes('/v9/projects/')) {\n return of({\n data: {\n targets: {\n production: {\n alias: ['prod-alias.vercel.app'],\n },\n development: {\n alias: ['dev-alias.vercel.app'],\n },\n },\n },\n });\n }\n\n return of({ data: {} });\n });\n\n organizationRepositoryMock.findByPartnerConfigurationId.resolves([\n {\n partnerConfigurations: [{ configurationId: 'test-config-id', projectIds: ['project-1', 'project-2'] }],\n },\n ]);\n\n const command = {\n userId: session.user._id,\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n configurationId: 'test-config-id',\n data: {\n 'org-id': ['project-1', 'project-2'],\n },\n };\n\n const result = await updateVercelIntegration.execute(command);\n\n expect(result.success).to.equal(true);\n assert.calledTwice(httpServiceMock.delete);\n });\n});\n"
},
{
"path": "apps/api/src/app/partner-integrations/usecases/update-vercel-integration/update-vercel-integration.usecase.ts",
"content": "import { HttpService } from '@nestjs/axios';\nimport { BadRequestException, Injectable } from '@nestjs/common';\nimport { AnalyticsService, decryptApiKey, PinoLogger } from '@novu/application-generic';\nimport {\n CommunityUserRepository,\n EnvironmentEntity,\n EnvironmentRepository,\n MemberRepository,\n OrganizationRepository,\n} from '@novu/dal';\nimport { lastValueFrom } from 'rxjs';\nimport { Sync } from '../../../bridge/usecases/sync';\nimport { UpdateVercelIntegrationCommand } from './update-vercel-integration.command';\n\ninterface ISetEnvironment {\n name: string;\n token: string;\n projectIds: string[];\n teamId: string | null;\n applicationIdentifier: string;\n privateKey: string;\n}\n\ninterface IRemoveEnvironment {\n token: string;\n teamId: string | null;\n userId: string;\n configurationId: string;\n}\n\ntype ProjectDetails = {\n projectId: string;\n clientAppIdEnv?: string;\n secretKeyEnv?: string;\n nextClientAppIdEnv?: string;\n nextApplicationIdentifierEnv?: string;\n};\n\n@Injectable()\nexport class UpdateVercelIntegration {\n constructor(\n private httpService: HttpService,\n private organizationRepository: OrganizationRepository,\n private memberRepository: MemberRepository,\n private communityUserRepository: CommunityUserRepository,\n private environmentRepository: EnvironmentRepository,\n private syncUsecase: Sync,\n private analyticsService: AnalyticsService,\n private logger: PinoLogger\n ) {\n this.logger.setContext(this.constructor.name);\n }\n\n async execute(command: UpdateVercelIntegrationCommand): Promise<{ success: boolean }> {\n try {\n const { userId, organizationId, configurationId, data: orgIdsToProjectIds } = command;\n\n const configuration = await this.getCurrentOrgPartnerConfiguration({\n userId,\n configurationId,\n });\n\n await this.removeEnvVariablesFromProjects({\n teamId: configuration.teamId,\n token: configuration.accessToken,\n userId,\n configurationId,\n });\n\n await this.organizationRepository.bulkUpdatePartnerConfiguration({\n userId,\n data: orgIdsToProjectIds,\n configuration,\n });\n\n const organizationIds = Object.keys(orgIdsToProjectIds);\n const environments = await this.getEnvironments(organizationIds);\n\n for (const env of environments) {\n const projectIds = orgIdsToProjectIds[env._organizationId];\n await this.setEnvVariablesOnProjects({\n name: env.name,\n applicationIdentifier: env.identifier,\n privateKey: decryptApiKey(env.apiKeys[0].key),\n projectIds,\n teamId: configuration.teamId,\n token: configuration.accessToken,\n });\n\n try {\n await this.updateBridgeUrl(\n env._id,\n env.name,\n projectIds[0],\n configuration.accessToken,\n configuration.teamId,\n env._organizationId\n );\n } catch (error) {\n this.logger.error({ err: error }, 'Error updating bridge url');\n }\n }\n\n this.analyticsService.track('Update Vercel Integration - [Partner Integrations]', userId, {\n _organization: organizationId,\n });\n\n return { success: true };\n } catch (error) {\n throw new BadRequestException(error.message);\n }\n }\n\n private async updateBridgeUrl(\n environmentId: string,\n environmentName: string,\n projectId: string,\n accessToken: string,\n teamId: string,\n organizationId: string\n ) {\n try {\n const getDomainsResponse = await lastValueFrom(\n this.httpService.get(`${process.env.VERCEL_BASE_URL}/v9/projects/${projectId}?teamId=${teamId}`, {\n headers: {\n Authorization: `Bearer ${accessToken}`,\n 'Content-Type': 'application/json',\n },\n })\n );\n\n const vercelAvailableTargets = getDomainsResponse.data?.targets;\n let vercelTarget;\n if (environmentName.toLowerCase() === 'production') {\n vercelTarget = vercelAvailableTargets?.production;\n } else {\n vercelTarget = vercelAvailableTargets?.development;\n }\n\n const alias = vercelTarget?.alias?.sort((a, b) => a.length - b.length)[0];\n const bridgeAlias = alias || vercelTarget?.meta?.branchAlias || vercelTarget?.automaticAliases[0];\n if (!bridgeAlias) {\n return;\n }\n\n const orgOwner = await this.memberRepository.getOrganizationOwnerAccount(organizationId);\n if (!orgOwner) {\n throw new BadRequestException('Organization owner not found');\n }\n\n const internalUser = await this.communityUserRepository.findOne({ externalId: orgOwner?._userId });\n if (!internalUser) {\n throw new BadRequestException('User not found');\n }\n\n await this.syncUsecase.execute({\n organizationId,\n userId: internalUser?._id as string,\n environmentId,\n bridgeUrl: `https://${bridgeAlias}/api/novu`,\n source: 'vercel',\n });\n } catch (error) {\n this.logger.error({ err: error }, 'Error updating bridge url');\n }\n }\n\n private async getEnvironments(organizationIds: string[]): Promise {\n return await this.environmentRepository.find(\n {\n _organizationId: { $in: organizationIds },\n },\n 'apiKeys identifier name _organizationId _id'\n );\n }\n\n private async setEnvVariablesOnProjects({\n name,\n applicationIdentifier,\n projectIds,\n privateKey,\n teamId,\n token,\n }: ISetEnvironment): Promise {\n const target = name?.toLowerCase() === 'production' ? ['production'] : ['preview', 'development'];\n const type = 'encrypted';\n\n const environmentVariables = [\n {\n target,\n type,\n value: applicationIdentifier,\n key: 'NEXT_PUBLIC_NOVU_CLIENT_APP_ID',\n legacy: true,\n },\n {\n target,\n type,\n value: applicationIdentifier,\n key: 'NOVU_CLIENT_APP_ID',\n legacy: true,\n },\n {\n target,\n type,\n value: applicationIdentifier,\n key: 'NEXT_PUBLIC_NOVU_APPLICATION_IDENTIFIER',\n },\n {\n target,\n type,\n value: privateKey,\n key: 'NOVU_SECRET_KEY',\n },\n ];\n\n const headers = {\n Authorization: `Bearer ${token}`,\n 'Content-Type': 'application/json',\n };\n\n const setEnvVariable = async (projectId: string, variable: (typeof environmentVariables)[0]) => {\n if (variable.legacy) {\n return;\n }\n\n try {\n const queryParams = new URLSearchParams();\n queryParams.set('upsert', 'true');\n\n if (teamId) {\n queryParams.set('teamId', teamId);\n }\n\n await lastValueFrom(\n this.httpService.post(\n `${process.env.VERCEL_BASE_URL}/v10/projects/${projectId}/env?${queryParams.toString()}`,\n [variable],\n { headers }\n )\n );\n } catch (error) {\n throw new BadRequestException(error.response?.data?.error || error.response?.data);\n }\n };\n\n await Promise.all(\n projectIds.flatMap((projectId) => environmentVariables.map((variable) => setEnvVariable(projectId, variable)))\n );\n }\n\n async getCurrentOrgPartnerConfiguration({ userId, configurationId }: { userId: string; configurationId: string }) {\n const orgsWithIntegration = await this.organizationRepository.findByPartnerConfigurationId({\n userId,\n configurationId,\n });\n\n if (orgsWithIntegration.length === 0) {\n throw new BadRequestException({\n message: 'No partner configuration found.',\n type: 'vercel',\n });\n }\n\n const firstOrg = orgsWithIntegration[0];\n const configuration = firstOrg.partnerConfigurations?.find((config) => config.configurationId === configurationId);\n if (!firstOrg.partnerConfigurations?.length || !configuration) {\n throw new BadRequestException({\n message: 'No partner configuration found.',\n type: 'vercel',\n });\n }\n\n return configuration;\n }\n\n private async getVercelLinkedProjects(\n accessToken: string,\n teamId: string | null,\n projectIds: string[]\n ): Promise {\n const response = await lastValueFrom(\n this.httpService.get(`${process.env.VERCEL_BASE_URL}/v4/projects${teamId ? `?teamId=${teamId}` : ''}`, {\n headers: {\n Authorization: `Bearer ${accessToken}`,\n },\n })\n );\n const vercelProjects = response.data.projects as any[];\n const filteredVercelProjects = vercelProjects.filter((project) => projectIds.includes(project.id));\n\n return ['production', 'development'].flatMap((vercelEnvironment) =>\n filteredVercelProjects.map((project) => {\n const { id } = project;\n const vercelEnvs = project?.env;\n const nextApplicationIdentifierEnv = vercelEnvs?.find(\n (e) => e.key === 'NEXT_PUBLIC_NOVU_APPLICATION_IDENTIFIER' && e.target.includes(vercelEnvironment)\n );\n // Legacy env variable for existing Vercel integrations\n const nextClientAppIdEnv = vercelEnvs?.find(\n (e) => e.key === 'NEXT_PUBLIC_NOVU_CLIENT_APP_ID' && e.target.includes(vercelEnvironment)\n );\n // Legacy env variable for existing Vercel integrations\n const clientAppIdEnv = vercelEnvs?.find(\n (e) => e.key === 'NOVU_CLIENT_APP_ID' && e.target.includes(vercelEnvironment)\n );\n const secretKeyEnv = vercelEnvs?.find(\n (e) => e.key === 'NOVU_SECRET_KEY' && e.target.includes(vercelEnvironment)\n );\n\n return {\n projectId: id,\n clientAppIdEnv: clientAppIdEnv?.id,\n secretKeyEnv: secretKeyEnv?.id,\n nextClientAppIdEnv: nextClientAppIdEnv?.id,\n nextApplicationIdentifierEnv: nextApplicationIdentifierEnv?.id,\n };\n })\n );\n }\n\n private async removeEnvVariablesFromProjects({\n teamId,\n token,\n userId,\n configurationId,\n }: IRemoveEnvironment): Promise {\n const orgsWithIntegration = await this.organizationRepository.findByPartnerConfigurationId({\n userId,\n configurationId,\n });\n\n const allOldProjectIds = [\n ...new Set(\n orgsWithIntegration.reduce((acc, org) => {\n return acc.concat(org.partnerConfigurations?.[0].projectIds || []);\n }, [])\n ),\n ];\n\n if (allOldProjectIds.length === 0) {\n return;\n }\n\n const vercelLinkedProjects = await this.getVercelLinkedProjects(token, teamId, allOldProjectIds);\n\n const projectApiUrl = `${process.env.VERCEL_BASE_URL}/v9/projects`;\n\n await Promise.all(\n vercelLinkedProjects.map((detail) => {\n const urls: string[] = [];\n if (detail.nextApplicationIdentifierEnv) {\n urls.push(\n `${projectApiUrl}/${detail.projectId}/env/${detail.nextApplicationIdentifierEnv}${teamId ? `?teamId=${teamId}` : ''}`\n );\n }\n\n if (detail.nextClientAppIdEnv) {\n urls.push(\n `${projectApiUrl}/${detail.projectId}/env/${detail.nextClientAppIdEnv}${teamId ? `?teamId=${teamId}` : ''}`\n );\n }\n\n if (detail.clientAppIdEnv) {\n urls.push(\n `${projectApiUrl}/${detail.projectId}/env/${detail.clientAppIdEnv}${teamId ? `?teamId=${teamId}` : ''}`\n );\n }\n\n if (detail.secretKeyEnv) {\n urls.push(\n `${projectApiUrl}/${detail.projectId}/env/${detail.secretKeyEnv}${teamId ? `?teamId=${teamId}` : ''}`\n );\n }\n\n const requests = urls.map((url) =>\n lastValueFrom(\n this.httpService.delete(url, {\n headers: {\n Authorization: `Bearer ${token}`,\n },\n })\n )\n );\n\n return Promise.all(requests);\n })\n );\n }\n}\n"
},
{
"path": "apps/api/src/app/preferences/dtos/preferences.dto.ts",
"content": "import { ChannelTypeEnum } from '@novu/shared';\nimport { Type } from 'class-transformer';\nimport { IsBoolean, ValidateNested } from 'class-validator';\n\n/**\n * @deprecated Use an updated preference structure.\n * This class will be removed in future versions.\n */\nexport class WorkflowPreference {\n /**\n * @deprecated Use alternative enablement mechanism.\n */\n @IsBoolean()\n enabled: boolean;\n\n /**\n * @deprecated Read-only flag is no longer supported.\n */\n @IsBoolean()\n readOnly: boolean;\n}\n\n/**\n * @deprecated Use an updated channel preference structure.\n * Will be removed in future versions.\n */\nexport class ChannelPreference {\n /**\n * @deprecated Use alternative channel enablement method.\n */\n @IsBoolean()\n enabled: boolean;\n}\n\n/**\n * @deprecated Channels configuration is being restructured.\n * Use the new channel management approach.\n */\nexport class Channels {\n /**\n * @deprecated In-app channel preference is deprecated.\n */\n @ValidateNested({ each: true })\n @Type(() => ChannelPreference)\n [ChannelTypeEnum.IN_APP]: ChannelPreference;\n\n /**\n * @deprecated Email channel preference is deprecated.\n */\n @ValidateNested({ each: true })\n @Type(() => ChannelPreference)\n [ChannelTypeEnum.EMAIL]: ChannelPreference;\n\n /**\n * @deprecated SMS channel preference is deprecated.\n */\n @ValidateNested({ each: true })\n @Type(() => ChannelPreference)\n [ChannelTypeEnum.SMS]: ChannelPreference;\n\n /**\n * @deprecated Chat channel preference is deprecated.\n */\n @ValidateNested({ each: true })\n @Type(() => ChannelPreference)\n [ChannelTypeEnum.CHAT]: ChannelPreference;\n\n /**\n * @deprecated Push channel preference is deprecated.\n */\n @ValidateNested({ each: true })\n @Type(() => ChannelPreference)\n [ChannelTypeEnum.PUSH]: ChannelPreference;\n}\n\n/**\n * @deprecated Preferences DTO is being replaced.\n * Use the new preferences management approach.\n */\nexport class PreferencesDto {\n /**\n * @deprecated Global workflow preference is no longer used.\n */\n @ValidateNested({ each: true })\n @Type(() => WorkflowPreference)\n all: WorkflowPreference;\n\n /**\n * @deprecated Channels configuration is deprecated.\n */\n @ValidateNested({ each: true })\n @Type(() => Channels)\n channels: Channels;\n}\n\n// Optional: Runtime deprecation warning\nif (process.env.NODE_ENV !== 'production' && !process.env.CI) {\n console.warn(\n 'DEPRECATION WARNING: PreferencesDto and related classes are deprecated ' +\n 'and will be removed in future versions. Please migrate to the new preferences structure.'\n );\n}\n"
},
{
"path": "apps/api/src/app/preferences/dtos/upsert-preferences.dto.ts",
"content": "import { Type } from 'class-transformer';\nimport { IsString, ValidateNested } from 'class-validator';\nimport { PreferencesDto } from './preferences.dto';\n\n/**\n * @deprecated This DTO is no longer recommended for use.\n * Consider using an alternative implementation or updated data transfer object.\n */\nexport class UpsertPreferencesDto {\n /**\n * @deprecated Use an alternative workflow identification method.\n */\n @IsString()\n workflowId: string;\n\n /**\n * @deprecated Preferences structure is outdated.\n */\n @ValidateNested({ each: true })\n @Type(() => PreferencesDto)\n preferences: PreferencesDto;\n}\n"
},
{
"path": "apps/api/src/app/preferences/index.ts",
"content": "export { PreferencesModule } from './preferences.module';\n"
},
{
"path": "apps/api/src/app/preferences/preferences.controller.ts",
"content": "import {\n Body,\n ClassSerializerInterceptor,\n Controller,\n Delete,\n Get,\n Post,\n Query,\n UseInterceptors,\n} from '@nestjs/common';\nimport { ApiExcludeController } from '@nestjs/swagger';\nimport {\n DeletePreferencesCommand,\n DeletePreferencesUseCase,\n GetPreferences,\n GetPreferencesCommand,\n UpsertPreferences,\n UpsertUserWorkflowPreferencesCommand,\n UserSession,\n} from '@novu/application-generic';\nimport { PreferencesTypeEnum, UserSessionData } from '@novu/shared';\nimport { RequireAuthentication } from '../auth/framework/auth.decorator';\nimport { UpsertPreferencesDto } from './dtos/upsert-preferences.dto';\n\n/**\n * @deprecated - set workflow preferences using the `/workflows` endpoint instead\n */\n@Controller('/preferences')\n@UseInterceptors(ClassSerializerInterceptor)\n@RequireAuthentication()\n@ApiExcludeController()\nexport class PreferencesController {\n constructor(\n private upsertPreferences: UpsertPreferences,\n private getPreferences: GetPreferences,\n private deletePreferences: DeletePreferencesUseCase\n ) {}\n\n @Get('/')\n async get(@UserSession() user: UserSessionData, @Query('workflowId') workflowId: string) {\n return this.getPreferences.execute(\n GetPreferencesCommand.create({\n templateId: workflowId,\n environmentId: user.environmentId,\n organizationId: user.organizationId,\n })\n );\n }\n\n @Post('/')\n async upsert(@Body() data: UpsertPreferencesDto, @UserSession() user: UserSessionData) {\n return this.upsertPreferences.upsertUserWorkflowPreferences(\n UpsertUserWorkflowPreferencesCommand.create({\n environmentId: user.environmentId,\n organizationId: user.organizationId,\n userId: user._id,\n preferences: data.preferences,\n templateId: data.workflowId,\n })\n );\n }\n\n @Delete('/')\n async delete(@UserSession() user: UserSessionData, @Query('workflowId') workflowId: string) {\n return this.deletePreferences.execute(\n DeletePreferencesCommand.create({\n templateId: workflowId,\n environmentId: user.environmentId,\n organizationId: user.organizationId,\n userId: user._id,\n type: PreferencesTypeEnum.USER_WORKFLOW,\n })\n );\n }\n}\n"
},
{
"path": "apps/api/src/app/preferences/preferences.module.ts",
"content": "import { MiddlewareConsumer, Module, NestModule } from '@nestjs/common';\nimport { DeletePreferencesUseCase, GetPreferences, UpsertPreferences } from '@novu/application-generic';\nimport { PreferencesRepository } from '@novu/dal';\nimport { SharedModule } from '../shared/shared.module';\nimport { PreferencesController } from './preferences.controller';\n\nconst PROVIDERS = [PreferencesRepository, UpsertPreferences, GetPreferences, DeletePreferencesUseCase];\n\n@Module({\n imports: [SharedModule],\n providers: [...PROVIDERS],\n controllers: [PreferencesController],\n exports: [...PROVIDERS],\n})\nexport class PreferencesModule implements NestModule {\n public configure(consumer: MiddlewareConsumer) {}\n}\n"
},
{
"path": "apps/api/src/app/preferences/preferences.spec.ts",
"content": "import { Test } from '@nestjs/testing';\nimport {\n GetPreferences,\n UpsertPreferences,\n UpsertSubscriberGlobalPreferencesCommand,\n UpsertSubscriberWorkflowPreferencesCommand,\n UpsertUserWorkflowPreferencesCommand,\n UpsertWorkflowPreferencesCommand,\n} from '@novu/application-generic';\nimport { PreferencesRepository, SubscriberRepository } from '@novu/dal';\nimport { FeatureFlagsKeysEnum, PreferencesTypeEnum } from '@novu/shared';\nimport { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\n\nimport { AuthModule } from '../auth/auth.module';\nimport { PreferencesModule } from './preferences.module';\n\ndescribe('Preferences', () => {\n let getPreferences: GetPreferences;\n const subscriberId = SubscriberRepository.createObjectId();\n const workflowId = PreferencesRepository.createObjectId();\n let upsertPreferences: UpsertPreferences;\n let session: UserSession;\n\n beforeEach(async () => {\n const moduleRef = await Test.createTestingModule({\n imports: [PreferencesModule, AuthModule],\n providers: [],\n }).compile();\n\n session = new UserSession();\n await session.initialize();\n\n getPreferences = moduleRef.get(GetPreferences);\n upsertPreferences = moduleRef.get(UpsertPreferences);\n });\n\n describe('Upsert preferences', () => {\n it('should create workflow preferences', async () => {\n const workflowPreferences = await upsertPreferences.upsertWorkflowPreferences(\n UpsertWorkflowPreferencesCommand.create({\n preferences: {\n all: {\n enabled: false,\n readOnly: false,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n environmentId: session.environment._id,\n organizationId: session.organization._id,\n templateId: workflowId,\n })\n );\n\n expect(workflowPreferences._environmentId).to.equal(session.environment._id);\n expect(workflowPreferences._organizationId).to.equal(session.organization._id);\n expect(workflowPreferences._templateId).to.equal(workflowId);\n expect(workflowPreferences._userId).to.be.undefined;\n expect(workflowPreferences._subscriberId).to.be.undefined;\n expect(workflowPreferences.type).to.equal(PreferencesTypeEnum.WORKFLOW_RESOURCE);\n });\n\n it('should create user workflow preferences', async () => {\n const userPreferences = await upsertPreferences.upsertUserWorkflowPreferences(\n UpsertUserWorkflowPreferencesCommand.create({\n preferences: {\n all: {\n enabled: false,\n readOnly: false,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n environmentId: session.environment._id,\n organizationId: session.organization._id,\n templateId: workflowId,\n userId: session.user._id,\n })\n );\n\n expect(userPreferences._environmentId).to.equal(session.environment._id);\n expect(userPreferences._organizationId).to.equal(session.organization._id);\n expect(userPreferences._templateId).to.equal(workflowId);\n expect(userPreferences._userId).to.equal(session.user._id);\n expect(userPreferences._subscriberId).to.be.undefined;\n expect(userPreferences.type).to.equal(PreferencesTypeEnum.USER_WORKFLOW);\n });\n\n it('should create global subscriber preferences', async () => {\n const subscriberGlobalPreferences = await upsertPreferences.upsertSubscriberGlobalPreferences(\n UpsertSubscriberGlobalPreferencesCommand.create({\n preferences: {\n all: {\n enabled: false,\n readOnly: false,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n environmentId: session.environment._id,\n organizationId: session.organization._id,\n _subscriberId: subscriberId,\n })\n );\n\n expect(subscriberGlobalPreferences._environmentId).to.equal(session.environment._id);\n expect(subscriberGlobalPreferences._organizationId).to.equal(session.organization._id);\n expect(subscriberGlobalPreferences._templateId).to.be.undefined;\n expect(subscriberGlobalPreferences._userId).to.be.undefined;\n expect(subscriberGlobalPreferences._subscriberId).to.equal(subscriberId);\n expect(subscriberGlobalPreferences.type).to.equal(PreferencesTypeEnum.SUBSCRIBER_GLOBAL);\n });\n\n it('should create subscriber workflow preferences', async () => {\n const subscriberWorkflowPreferences = await upsertPreferences.upsertSubscriberWorkflowPreferences(\n UpsertSubscriberWorkflowPreferencesCommand.create({\n preferences: {\n all: {\n enabled: false,\n readOnly: false,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n environmentId: session.environment._id,\n organizationId: session.organization._id,\n templateId: workflowId,\n _subscriberId: subscriberId,\n })\n );\n\n expect(subscriberWorkflowPreferences._environmentId).to.equal(session.environment._id);\n expect(subscriberWorkflowPreferences._organizationId).to.equal(session.organization._id);\n expect(subscriberWorkflowPreferences._templateId).to.equal(workflowId);\n expect(subscriberWorkflowPreferences._userId).to.be.undefined;\n expect(subscriberWorkflowPreferences._subscriberId).to.equal(subscriberId);\n expect(subscriberWorkflowPreferences.type).to.equal(PreferencesTypeEnum.SUBSCRIBER_WORKFLOW);\n });\n\n it('should update preferences', async () => {\n let workflowPreferences = await upsertPreferences.upsertWorkflowPreferences(\n UpsertWorkflowPreferencesCommand.create({\n preferences: {\n all: {\n enabled: false,\n readOnly: false,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n environmentId: session.environment._id,\n organizationId: session.organization._id,\n templateId: workflowId,\n })\n );\n\n expect(workflowPreferences._environmentId).to.equal(session.environment._id);\n expect(workflowPreferences._organizationId).to.equal(session.organization._id);\n expect(workflowPreferences._templateId).to.equal(workflowId);\n expect(workflowPreferences._userId).to.be.undefined;\n expect(workflowPreferences._subscriberId).to.be.undefined;\n expect(workflowPreferences.type).to.equal(PreferencesTypeEnum.WORKFLOW_RESOURCE);\n\n workflowPreferences = await upsertPreferences.upsertWorkflowPreferences(\n UpsertWorkflowPreferencesCommand.create({\n preferences: {\n all: {\n enabled: false,\n readOnly: true,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n environmentId: session.environment._id,\n organizationId: session.organization._id,\n templateId: workflowId,\n })\n );\n\n expect(workflowPreferences.preferences.all.readOnly).to.be.true;\n });\n });\n\n describe('Get preferences', () => {\n it('should merge preferences when get preferences', async () => {\n // Workflow preferences\n await upsertPreferences.upsertWorkflowPreferences(\n UpsertWorkflowPreferencesCommand.create({\n preferences: {\n all: {\n enabled: false,\n readOnly: false,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n environmentId: session.environment._id,\n organizationId: session.organization._id,\n templateId: workflowId,\n })\n );\n\n let preferences = await getPreferences.execute({\n environmentId: session.environment._id,\n organizationId: session.organization._id,\n templateId: workflowId,\n });\n\n expect(preferences).to.deep.equal({\n preferences: {\n all: {\n enabled: false,\n readOnly: false,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n schedule: undefined,\n type: PreferencesTypeEnum.WORKFLOW_RESOURCE,\n source: {\n [PreferencesTypeEnum.WORKFLOW_RESOURCE]: {\n all: {\n enabled: false,\n readOnly: false,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n [PreferencesTypeEnum.USER_WORKFLOW]: null,\n [PreferencesTypeEnum.SUBSCRIBER_GLOBAL]: null,\n [PreferencesTypeEnum.SUBSCRIBER_WORKFLOW]: null,\n },\n });\n\n // User Workflow preferences\n await upsertPreferences.upsertUserWorkflowPreferences(\n UpsertUserWorkflowPreferencesCommand.create({\n preferences: {\n all: {\n enabled: false,\n readOnly: true,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n environmentId: session.environment._id,\n organizationId: session.organization._id,\n templateId: workflowId,\n userId: session.user._id,\n })\n );\n\n preferences = await getPreferences.execute({\n environmentId: session.environment._id,\n organizationId: session.organization._id,\n templateId: workflowId,\n });\n\n expect(preferences).to.deep.equal({\n preferences: {\n all: {\n enabled: false,\n readOnly: true,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n schedule: undefined,\n type: PreferencesTypeEnum.USER_WORKFLOW,\n source: {\n [PreferencesTypeEnum.WORKFLOW_RESOURCE]: {\n all: {\n enabled: false,\n readOnly: false,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n [PreferencesTypeEnum.USER_WORKFLOW]: {\n all: {\n enabled: false,\n readOnly: true,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n [PreferencesTypeEnum.SUBSCRIBER_GLOBAL]: null,\n [PreferencesTypeEnum.SUBSCRIBER_WORKFLOW]: null,\n },\n });\n\n // Subscriber global preferences\n await upsertPreferences.upsertSubscriberGlobalPreferences(\n UpsertSubscriberGlobalPreferencesCommand.create({\n preferences: {\n all: {\n enabled: false,\n readOnly: true,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n environmentId: session.environment._id,\n organizationId: session.organization._id,\n _subscriberId: subscriberId,\n })\n );\n\n preferences = await getPreferences.execute({\n environmentId: session.environment._id,\n organizationId: session.organization._id,\n templateId: workflowId,\n subscriberId,\n });\n\n expect(preferences).to.deep.equal({\n preferences: {\n all: {\n enabled: false,\n readOnly: true,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n schedule: undefined,\n type: PreferencesTypeEnum.USER_WORKFLOW,\n source: {\n [PreferencesTypeEnum.WORKFLOW_RESOURCE]: {\n all: {\n enabled: false,\n readOnly: false,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n [PreferencesTypeEnum.USER_WORKFLOW]: {\n all: {\n enabled: false,\n readOnly: true,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n [PreferencesTypeEnum.SUBSCRIBER_GLOBAL]: {\n all: {\n enabled: false,\n readOnly: true,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n [PreferencesTypeEnum.SUBSCRIBER_WORKFLOW]: null,\n },\n });\n\n // Subscriber Workflow preferences\n await upsertPreferences.upsertSubscriberWorkflowPreferences(\n UpsertSubscriberWorkflowPreferencesCommand.create({\n preferences: {\n all: {\n enabled: false,\n readOnly: true,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n environmentId: session.environment._id,\n organizationId: session.organization._id,\n templateId: workflowId,\n _subscriberId: subscriberId,\n })\n );\n\n preferences = await getPreferences.execute({\n environmentId: session.environment._id,\n organizationId: session.organization._id,\n templateId: workflowId,\n subscriberId,\n });\n\n expect(preferences).to.deep.equal({\n preferences: {\n all: {\n enabled: false,\n readOnly: true,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n schedule: undefined,\n type: PreferencesTypeEnum.USER_WORKFLOW,\n source: {\n [PreferencesTypeEnum.WORKFLOW_RESOURCE]: {\n all: {\n enabled: false,\n readOnly: false,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n [PreferencesTypeEnum.USER_WORKFLOW]: {\n all: {\n enabled: false,\n readOnly: true,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n [PreferencesTypeEnum.SUBSCRIBER_GLOBAL]: {\n all: {\n enabled: false,\n readOnly: true,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n [PreferencesTypeEnum.SUBSCRIBER_WORKFLOW]: {\n all: {\n enabled: false,\n readOnly: true,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n },\n });\n });\n });\n\n describe('Preferences endpoints', () => {\n it('should get preferences', async () => {\n const useCase: UpsertPreferences = session.testServer?.getService(UpsertPreferences);\n\n await useCase.upsertWorkflowPreferences(\n UpsertWorkflowPreferencesCommand.create({\n preferences: {\n all: {\n enabled: false,\n readOnly: false,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n environmentId: session.environment._id,\n organizationId: session.organization._id,\n templateId: workflowId,\n })\n );\n\n const { body } = await session.testAgent.get(`/v1/preferences?workflowId=${workflowId}`).send();\n\n expect(body.data).to.deep.equal({\n preferences: {\n all: {\n enabled: false,\n readOnly: false,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n type: PreferencesTypeEnum.WORKFLOW_RESOURCE,\n source: {\n [PreferencesTypeEnum.WORKFLOW_RESOURCE]: {\n all: {\n enabled: false,\n readOnly: false,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n [PreferencesTypeEnum.USER_WORKFLOW]: null,\n [PreferencesTypeEnum.SUBSCRIBER_GLOBAL]: null,\n [PreferencesTypeEnum.SUBSCRIBER_WORKFLOW]: null,\n },\n });\n });\n\n it('should upsert preferences', async () => {\n const { body } = await session.testAgent.post('/v1/preferences').send({\n workflowId,\n preferences: {\n all: {\n enabled: false,\n readOnly: false,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n },\n });\n\n expect(body.data.preferences).to.deep.equal({\n all: {\n enabled: false,\n readOnly: false,\n },\n channels: {\n in_app: {\n enabled: false,\n },\n sms: {\n enabled: false,\n },\n email: {\n enabled: false,\n },\n push: {\n enabled: false,\n },\n chat: {\n enabled: false,\n },\n },\n });\n });\n });\n});\n"
},
{
"path": "apps/api/src/app/rate-limiting/e2e/throttler.guard.e2e.ts",
"content": "import { HttpResponseHeaderKeysEnum } from '@novu/application-generic';\nimport { ApiRateLimitCategoryEnum, ApiRateLimitCostEnum, ApiServiceLevelEnum } from '@novu/shared';\nimport { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\n\nconst mockSingleCost = 1;\nconst mockBulkCost = 5;\nconst mockWindowDuration = 5;\nconst mockBurstAllowance = 1;\nconst mockMaximumFreeTrigger = 5;\nconst mockMaximumFreeGlobal = 3;\nconst mockMaximumUnlimitedTrigger = 10;\nconst mockMaximumUnlimitedGlobal = 5;\n\nprocess.env.API_RATE_LIMIT_COST_SINGLE = `${mockSingleCost}`;\nprocess.env.API_RATE_LIMIT_COST_BULK = `${mockBulkCost}`;\nprocess.env.API_RATE_LIMIT_ALGORITHM_WINDOW_DURATION = `${mockWindowDuration}`;\nprocess.env.API_RATE_LIMIT_ALGORITHM_BURST_ALLOWANCE = `${mockBurstAllowance}`;\nprocess.env.API_RATE_LIMIT_MAXIMUM_FREE_TRIGGER = `${mockMaximumFreeTrigger}`;\nprocess.env.API_RATE_LIMIT_MAXIMUM_FREE_GLOBAL = `${mockMaximumFreeGlobal}`;\nprocess.env.API_RATE_LIMIT_MAXIMUM_UNLIMITED_TRIGGER = `${mockMaximumUnlimitedTrigger}`;\nprocess.env.API_RATE_LIMIT_MAXIMUM_UNLIMITED_GLOBAL = `${mockMaximumUnlimitedGlobal}`;\n\n// Disable Launch Darkly to allow test to define FF state\n(process.env as Record).LAUNCH_DARKLY_SDK_KEY = '';\n\ndescribe('API Rate Limiting #novu-v2', () => {\n let session: UserSession;\n const pathPrefix = '/v1/rate-limiting';\n\n let request: (\n path: string,\n authHeader?: string\n ) => Promise>>;\n\n describe('Guard logic', () => {\n beforeEach(async () => {\n (process.env as Record).IS_API_RATE_LIMITING_ENABLED = 'true';\n\n session = new UserSession();\n await session.initialize();\n await session.updateOrganizationServiceLevel(ApiServiceLevelEnum.UNLIMITED);\n\n request = (path: string, authHeader = `ApiKey ${session.apiKey}`) =>\n session.testAgent.get(path).set('authorization', authHeader);\n });\n\n describe('Feature Flag', () => {\n it('should set rate limit headers when the Feature Flag is enabled', async () => {\n (process.env as Record).IS_API_RATE_LIMITING_ENABLED = 'true';\n const response = await request(`${pathPrefix}/no-category-no-cost`);\n\n expect(response.headers[HttpResponseHeaderKeysEnum.RATELIMIT_LIMIT.toLowerCase()]).to.exist;\n });\n\n it('should NOT set rate limit headers when the Feature Flag is disabled', async () => {\n (process.env as Record).IS_API_RATE_LIMITING_ENABLED = 'false';\n const response = await request(`${pathPrefix}/no-category-no-cost`);\n\n expect(response.headers[HttpResponseHeaderKeysEnum.RATELIMIT_LIMIT.toLowerCase()]).not.to.exist;\n });\n });\n\n describe('Allowed Authentication Security Schemes', () => {\n it('should set rate limit headers when ApiKey security scheme is used to authenticate', async () => {\n const response = await request(`${pathPrefix}/no-category-no-cost`, `ApiKey ${session.apiKey}`);\n\n expect(response.headers[HttpResponseHeaderKeysEnum.RATELIMIT_LIMIT.toLowerCase()]).to.exist;\n });\n\n it('should NOT set rate limit headers when a Bearer security scheme is used to authenticate', async () => {\n const response = await request(`${pathPrefix}/no-category-no-cost`, session.token);\n\n expect(response.headers[HttpResponseHeaderKeysEnum.RATELIMIT_LIMIT.toLowerCase()]).not.to.exist;\n });\n\n it('should NOT set rate limit headers when NO authorization header is present', async () => {\n const response = await request(`${pathPrefix}/no-category-no-cost`, '');\n\n expect(response.headers[HttpResponseHeaderKeysEnum.RATELIMIT_LIMIT.toLowerCase()]).not.to.exist;\n });\n });\n\n describe('RateLimit-Policy', () => {\n const testParams: Array<{ name: string; expectedRegex: string }> = [\n { name: 'limit', expectedRegex: `${mockMaximumUnlimitedGlobal * mockWindowDuration}` },\n { name: 'w', expectedRegex: `w=${mockWindowDuration}` },\n {\n name: 'burst',\n expectedRegex: `burst=${mockMaximumUnlimitedGlobal * (1 + mockBurstAllowance) * mockWindowDuration}`,\n },\n { name: 'comment', expectedRegex: `comment=\"[a-zA-Z ]*\"` },\n { name: 'category', expectedRegex: `category=\"(${Object.values(ApiRateLimitCategoryEnum).join('|')})\"` },\n { name: 'cost', expectedRegex: `cost=\"(${Object.values(ApiRateLimitCostEnum).join('|')})\"` },\n {\n name: 'serviceLevel',\n expectedRegex: `serviceLevel=\"[a-zA-Z]*\"`,\n },\n ];\n\n testParams.forEach(({ name, expectedRegex }) => {\n it(`should include the ${name} parameter`, async () => {\n const response = await request(`${pathPrefix}/no-category-no-cost`);\n const policyHeader = response.headers[HttpResponseHeaderKeysEnum.RATELIMIT_POLICY.toLowerCase()];\n\n expect(policyHeader).to.match(new RegExp(expectedRegex));\n });\n });\n\n it('should separate the params with a semicolon', async () => {\n const response = await request(`${pathPrefix}/no-category-no-cost`);\n const policyHeader = response.headers[HttpResponseHeaderKeysEnum.RATELIMIT_POLICY.toLowerCase()];\n\n expect(policyHeader.split(';')).to.have.lengthOf(testParams.length);\n });\n });\n\n describe('Rate Limit Decorators', () => {\n describe('Controller WITHOUT Decorators', () => {\n const controllerPathPrefix = '/v1/rate-limiting';\n\n it('should use the global category for an endpoint WITHOUT category decorator', async () => {\n const response = await request(`${controllerPathPrefix}/no-category-no-cost`);\n const policyHeader = response.headers[HttpResponseHeaderKeysEnum.RATELIMIT_POLICY.toLowerCase()];\n\n expect(policyHeader).to.contain(`category=\"${ApiRateLimitCategoryEnum.GLOBAL}\"`);\n });\n\n it('should use the single cost for an endpoint WITHOUT cost decorator', async () => {\n const response = await request(`${controllerPathPrefix}/no-category-no-cost`);\n const policyHeader = response.headers[HttpResponseHeaderKeysEnum.RATELIMIT_POLICY.toLowerCase()];\n\n expect(policyHeader).to.contain(`cost=\"${ApiRateLimitCostEnum.SINGLE}\"`);\n });\n });\n\n describe('Controller WITH Decorators', () => {\n const controllerPathPrefix = '/v1/rate-limiting-trigger-bulk';\n\n it('should use the category decorator defined on the controller for an endpoint WITHOUT category decorator', async () => {\n const response = await request(`${controllerPathPrefix}/no-category-no-cost-override`);\n const policyHeader = response.headers[HttpResponseHeaderKeysEnum.RATELIMIT_POLICY.toLowerCase()];\n\n expect(policyHeader).to.contain(`category=\"${ApiRateLimitCategoryEnum.TRIGGER}\"`);\n });\n\n it('should use the cost decorator defined on the controller for an endpoint WITHOUT cost decorator', async () => {\n const response = await request(`${controllerPathPrefix}/no-category-no-cost-override`);\n const policyHeader = response.headers[HttpResponseHeaderKeysEnum.RATELIMIT_POLICY.toLowerCase()];\n\n expect(policyHeader).to.contain(`cost=\"${ApiRateLimitCostEnum.BULK}\"`);\n });\n\n it('should override the cost decorator defined on the controller for an endpoint WITH cost decorator', async () => {\n const response = await request(`${controllerPathPrefix}/no-category-single-cost-override`);\n const policyHeader = response.headers[HttpResponseHeaderKeysEnum.RATELIMIT_POLICY.toLowerCase()];\n\n expect(policyHeader).to.contain(`cost=\"${ApiRateLimitCostEnum.SINGLE}\"`);\n });\n\n it('should override the category decorator defined on the controller for an endpoint WITH category decorator', async () => {\n const response = await request(`${controllerPathPrefix}/global-category-no-cost-override`);\n const policyHeader = response.headers[HttpResponseHeaderKeysEnum.RATELIMIT_POLICY.toLowerCase()];\n\n expect(policyHeader).to.contain(`category=\"${ApiRateLimitCategoryEnum.GLOBAL}\"`);\n });\n });\n });\n });\n\n describe('Scenarios', () => {\n type TestCase = {\n name: string;\n requests: { path: string; count: number }[];\n expectedStatus: number;\n expectedLimit: number;\n expectedCost: number;\n expectedReset: number;\n expectedRetryAfter?: number;\n expectedThrottledRequests: number;\n setupTest?: (userSession: UserSession) => Promise;\n };\n\n const testCases: TestCase[] = [\n {\n name: 'single trigger endpoint request',\n requests: [{ path: '/trigger-category-single-cost', count: 1 }],\n expectedStatus: 200,\n expectedLimit: mockMaximumUnlimitedTrigger,\n expectedCost: mockSingleCost * 1,\n expectedReset: 1,\n expectedThrottledRequests: 0,\n async setupTest(userSession) {\n await userSession.updateOrganizationServiceLevel(ApiServiceLevelEnum.UNLIMITED);\n },\n },\n {\n name: 'no category no cost endpoint request',\n requests: [{ path: '/no-category-no-cost', count: 1 }],\n expectedStatus: 200,\n expectedLimit: mockMaximumUnlimitedGlobal,\n expectedCost: mockSingleCost * 1,\n expectedReset: 1,\n expectedThrottledRequests: 0,\n async setupTest(userSession) {\n await userSession.updateOrganizationServiceLevel(ApiServiceLevelEnum.UNLIMITED);\n },\n },\n {\n name: 'single trigger request with service level specified on organization ',\n requests: [{ path: '/trigger-category-single-cost', count: 1 }],\n expectedStatus: 200,\n expectedLimit: mockMaximumFreeTrigger,\n expectedCost: mockSingleCost * 1,\n expectedReset: 1,\n expectedThrottledRequests: 0,\n async setupTest(userSession) {\n await userSession.updateOrganizationServiceLevel(ApiServiceLevelEnum.FREE);\n },\n },\n {\n name: 'single trigger request with maximum rate limit specified on environment',\n requests: [{ path: '/trigger-category-single-cost', count: 1 }],\n expectedStatus: 200,\n expectedLimit: 60,\n expectedCost: mockSingleCost * 1,\n expectedReset: 1,\n expectedThrottledRequests: 0,\n async setupTest(userSession) {\n await userSession.updateOrganizationServiceLevel(ApiServiceLevelEnum.UNLIMITED);\n await userSession.updateEnvironmentApiRateLimits({ [ApiRateLimitCategoryEnum.TRIGGER]: 60 });\n },\n },\n {\n name: 'combination of single trigger and single global endpoint request',\n requests: [\n { path: '/trigger-category-single-cost', count: 20 },\n { path: '/global-category-single-cost', count: 100 },\n ],\n expectedStatus: 429,\n expectedLimit: mockMaximumUnlimitedGlobal,\n expectedCost: mockSingleCost * 100,\n expectedReset: 1,\n expectedRetryAfter: 1,\n expectedThrottledRequests: 50,\n async setupTest(userSession) {\n await userSession.updateOrganizationServiceLevel(ApiServiceLevelEnum.UNLIMITED);\n },\n },\n ];\n\n testCases\n .map(\n ({\n name,\n requests,\n expectedStatus,\n expectedLimit,\n expectedCost,\n expectedReset,\n expectedRetryAfter,\n expectedThrottledRequests,\n setupTest,\n }) => {\n return () => {\n describe(`${expectedStatus === 429 ? 'Throttled' : 'Allowed'} ${name}`, () => {\n let lastResponse;\n let throttledResponseCount = 0;\n const throttledResponseCountTolerance = 0.5;\n const expectedWindowLimit = expectedLimit * mockWindowDuration;\n const expectedBurstLimit = expectedWindowLimit * (1 + mockBurstAllowance);\n const expectedRemaining = Math.max(0, expectedBurstLimit - expectedCost);\n\n before(async () => {\n (process.env as Record).IS_API_RATE_LIMITING_ENABLED = 'true';\n\n session = new UserSession();\n await session.initialize();\n\n request = (path: string, authHeader = `ApiKey ${session.apiKey}`) =>\n session.testAgent.get(path).set('authorization', authHeader);\n\n setupTest && (await setupTest(session));\n for (const { path, count } of requests) {\n for (let index = 0; index < count; index += 1) {\n const response = await request(pathPrefix + path);\n lastResponse = response;\n\n if (response.statusCode === 429) {\n throttledResponseCount += 1;\n }\n }\n }\n });\n\n it(`should return a ${expectedStatus} status code`, async () => {\n expect(lastResponse.statusCode).to.equal(expectedStatus);\n });\n\n it(`should return a ${HttpResponseHeaderKeysEnum.RATELIMIT_LIMIT} header of ${expectedWindowLimit}`, async () => {\n expect(lastResponse.headers[HttpResponseHeaderKeysEnum.RATELIMIT_LIMIT.toLowerCase()]).to.equal(\n `${expectedWindowLimit}`\n );\n });\n\n it(`should return a ${HttpResponseHeaderKeysEnum.RATELIMIT_REMAINING} header of ${expectedRemaining}`, async () => {\n expect(lastResponse.headers[HttpResponseHeaderKeysEnum.RATELIMIT_REMAINING.toLowerCase()]).to.equal(\n `${expectedRemaining}`\n );\n });\n\n it(`should return a ${HttpResponseHeaderKeysEnum.RATELIMIT_RESET} header of ${expectedReset}`, async () => {\n expect(lastResponse.headers[HttpResponseHeaderKeysEnum.RATELIMIT_RESET.toLowerCase()]).to.equal(\n `${expectedReset}`\n );\n });\n\n it(`should return a ${HttpResponseHeaderKeysEnum.RETRY_AFTER} header of ${expectedRetryAfter}`, async () => {\n expect(lastResponse.headers[HttpResponseHeaderKeysEnum.RETRY_AFTER.toLowerCase()]).to.equal(\n expectedRetryAfter && `${expectedRetryAfter}`\n );\n });\n\n const expectedMinThrottled = Math.floor(\n expectedThrottledRequests * (1 - throttledResponseCountTolerance)\n );\n const expectedMaxThrottled = Math.ceil(expectedThrottledRequests * (1 + throttledResponseCountTolerance));\n it(`should have between ${expectedMinThrottled} and ${expectedMaxThrottled} requests throttled`, async () => {\n expect(throttledResponseCount).to.be.greaterThanOrEqual(expectedMinThrottled);\n expect(throttledResponseCount).to.be.lessThanOrEqual(expectedMaxThrottled);\n });\n });\n };\n }\n )\n .forEach((testCase) => {\n testCase();\n });\n });\n});\n"
},
{
"path": "apps/api/src/app/rate-limiting/guards/index.ts",
"content": "export * from './throttler.decorator';\nexport * from './throttler.guard';\n"
},
{
"path": "apps/api/src/app/rate-limiting/guards/throttler.decorator.ts",
"content": "import { Reflector } from '@nestjs/core';\nimport { ApiRateLimitCategoryEnum, ApiRateLimitCostEnum } from '@novu/shared';\n\nexport const ThrottlerCategory = Reflector.createDecorator();\n\nexport const ThrottlerCost = Reflector.createDecorator();\n"
},
{
"path": "apps/api/src/app/rate-limiting/guards/throttler.guard.ts",
"content": "import { CallHandler, ExecutionContext, Injectable, NestInterceptor } from '@nestjs/common';\nimport { Reflector } from '@nestjs/core';\nimport {\n InjectThrottlerOptions,\n InjectThrottlerStorage,\n ThrottlerException,\n ThrottlerGuard,\n ThrottlerModuleOptions,\n ThrottlerRequest,\n ThrottlerStorage,\n} from '@nestjs/throttler';\nimport {\n FeatureFlagsService,\n HttpRequestHeaderKeysEnum,\n HttpResponseHeaderKeysEnum,\n Instrument,\n PinoLogger,\n} from '@novu/application-generic';\nimport { EnvironmentEntity, OrganizationEntity, UserEntity } from '@novu/dal';\nimport {\n ApiAuthSchemeEnum,\n ApiRateLimitCategoryEnum,\n ApiRateLimitCostEnum,\n FeatureFlagsKeysEnum,\n UserSessionData,\n} from '@novu/shared';\nimport { getClientIp } from 'request-ip';\nimport { checkIsKeylessHeader } from '../../shared/utils/auth.utils';\nimport { EvaluateApiRateLimit, EvaluateApiRateLimitCommand } from '../usecases/evaluate-api-rate-limit';\nimport { ThrottlerCategory, ThrottlerCost } from './throttler.decorator';\n\nexport const THROTTLED_EXCEPTION_MESSAGE = 'API rate limit exceeded';\nexport const ALLOWED_AUTH_SCHEMES = [ApiAuthSchemeEnum.API_KEY, ApiAuthSchemeEnum.KEYLESS];\n\nconst defaultApiRateLimitCategory = ApiRateLimitCategoryEnum.GLOBAL;\nconst defaultApiRateLimitCost = ApiRateLimitCostEnum.SINGLE;\n\n/**\n * An interceptor is used instead of a guard to ensure that Auth context is available.\n * This is currently necessary because we do not currently have a global guard configured for Auth,\n * therefore the Auth context is not guaranteed to be available in the guard.\n */\n@Injectable()\nexport class ApiRateLimitInterceptor extends ThrottlerGuard implements NestInterceptor {\n constructor(\n @InjectThrottlerOptions() protected readonly options: ThrottlerModuleOptions,\n @InjectThrottlerStorage() protected readonly storageService: ThrottlerStorage,\n reflector: Reflector,\n private evaluateApiRateLimit: EvaluateApiRateLimit,\n private featureFlagService: FeatureFlagsService,\n private logger: PinoLogger\n ) {\n super(options, storageService, reflector);\n this.logger.setContext(this.constructor.name);\n }\n\n /**\n * Thin wrapper around the ThrottlerGuard's canActivate method.\n */\n async intercept(context: ExecutionContext, next: CallHandler) {\n await this.canActivate(context);\n\n return next.handle();\n }\n\n @Instrument()\n canActivate(context: ExecutionContext): Promise {\n return super.canActivate(context);\n }\n\n protected async shouldSkip(context: ExecutionContext): Promise {\n const req = context.switchToHttp().getRequest();\n const isAllowedAuthScheme = this.isAllowedAuthScheme(context);\n const isAllowedEnvironment = this.isAllowedEnvironment(context);\n const isAllowedRoute = this.isAllowedRoute(context);\n\n if (!isAllowedAuthScheme && !isAllowedEnvironment && !isAllowedRoute) {\n this.logger.debug(\n {\n _nv: {\n isAllowedAuthScheme,\n isAllowedEnvironment,\n isAllowedRoute,\n path: req.path,\n authScheme: req.authScheme,\n },\n },\n 'Rate limiting skipped - request criteria not met'\n );\n\n return true;\n }\n\n const user = this.getReqUser(context);\n\n // Indicates whether the request originates from a Inbox session initialization\n if (!user) {\n return false;\n }\n\n const { organizationId, environmentId, _id } = user;\n\n const isEnabled = await this.featureFlagService.getFlag({\n key: FeatureFlagsKeysEnum.IS_API_RATE_LIMITING_ENABLED,\n defaultValue: false,\n environment: { _id: environmentId } as EnvironmentEntity,\n organization: { _id: organizationId } as OrganizationEntity,\n user: { _id } as UserEntity,\n });\n\n if (!isEnabled) {\n this.logger.debug({\n message: 'Rate limiting skipped - feature flag disabled',\n _event: {\n organizationId,\n environmentId,\n },\n });\n }\n\n return !isEnabled;\n }\n\n /**\n * Throttles incoming HTTP requests.\n * All the outgoing requests will contain RFC-compatible RateLimit headers.\n * @see https://datatracker.ietf.org/doc/draft-ietf-httpapi-ratelimit-headers/\n * @throws {ThrottlerException}\n */\n protected async handleRequest({ context, throttler }: ThrottlerRequest): Promise {\n const { req, res } = this.getRequestResponse(context);\n const clientIp = getClientIp(req) || undefined;\n\n const ignoreUserAgents = throttler.ignoreUserAgents ?? this.commonOptions.ignoreUserAgents;\n // Return early if the current user agent should be ignored.\n if (Array.isArray(ignoreUserAgents)) {\n for (const pattern of ignoreUserAgents) {\n if (pattern.test(req.headers[HttpRequestHeaderKeysEnum.USER_AGENT.toLowerCase()])) {\n return true;\n }\n }\n }\n\n const handler = context.getHandler();\n const classRef = context.getClass();\n\n const isKeylessHeader =\n checkIsKeylessHeader(req.headers.authorization) ||\n checkIsKeylessHeader(req.headers['novu-application-identifier']);\n const isKeylessRequest = isKeylessHeader || this.isKeylessRoute(context);\n const apiRateLimitCategory =\n this.reflector.getAllAndOverride(ThrottlerCategory, [handler, classRef]) || defaultApiRateLimitCategory;\n\n const user = this.getReqUser(context);\n const organizationId = user?.organizationId;\n const _id = user?._id;\n const environmentId = user?.environmentId || req.headers['novu-application-identifier'];\n\n const apiRateLimitCost = isKeylessRequest\n ? getKeylessCost()\n : this.reflector.getAllAndOverride(ThrottlerCost, [handler, classRef]) || defaultApiRateLimitCost;\n\n const evaluateCommand = EvaluateApiRateLimitCommand.create({\n organizationId,\n environmentId,\n apiRateLimitCategory,\n apiRateLimitCost,\n ip: isKeylessRequest ? clientIp : undefined,\n });\n\n const { success, limit, remaining, reset, windowDuration, burstLimit, algorithm, apiServiceLevel } =\n await this.evaluateApiRateLimit.execute(evaluateCommand);\n\n const secondsToReset = Math.max(Math.ceil((reset - Date.now()) / 1e3), 0);\n\n this.logger.debug({\n message: 'Rate limit evaluated',\n _event: {\n success,\n limit,\n remaining,\n category: apiRateLimitCategory,\n cost: apiRateLimitCost,\n isKeyless: isKeylessRequest,\n organizationId,\n environmentId,\n ip: clientIp,\n },\n });\n\n /**\n * The purpose of the dry run is to allow us to observe how\n * the rate limiting would behave without actually enforcing it.\n */\n const isDryRun = await this.featureFlagService.getFlag({\n environment: { _id: environmentId } as EnvironmentEntity,\n organization: { _id: organizationId } as OrganizationEntity,\n user: { _id } as UserEntity,\n key: FeatureFlagsKeysEnum.IS_API_RATE_LIMITING_DRY_RUN_ENABLED,\n defaultValue: false,\n });\n\n const isKeylessDryRunFlag = await this.featureFlagService.getFlag({\n environment: { _id: environmentId } as EnvironmentEntity,\n organization: { _id: organizationId } as OrganizationEntity,\n user: { _id, email: user?.email } as UserEntity,\n key: FeatureFlagsKeysEnum.IS_API_RATE_LIMITING_KEYLESS_DRY_RUN_ENABLED,\n defaultValue: false,\n });\n const isKeylessDryRun = isKeylessRequest && isKeylessDryRunFlag;\n\n res.header(HttpResponseHeaderKeysEnum.RATELIMIT_REMAINING, remaining);\n res.header(HttpResponseHeaderKeysEnum.RATELIMIT_LIMIT, limit);\n res.header(HttpResponseHeaderKeysEnum.RATELIMIT_RESET, secondsToReset);\n res.header(\n HttpResponseHeaderKeysEnum.RATELIMIT_POLICY,\n this.createPolicyHeader(\n limit,\n windowDuration,\n burstLimit,\n algorithm,\n apiRateLimitCategory,\n apiRateLimitCost,\n apiServiceLevel\n )\n );\n\n res.rateLimitPolicy = {\n limit,\n windowDuration,\n burstLimit,\n algorithm,\n apiRateLimitCategory,\n apiRateLimitCost,\n apiServiceLevel,\n };\n\n if (isDryRun || isKeylessDryRun) {\n if (!success) {\n this.logger.warn({\n message: `${isKeylessRequest ? '[Dry run] [Keyless]' : '[Dry run]'} Rate limit would be exceeded`,\n _event: {\n limit,\n remaining,\n organizationId,\n environmentId,\n ip: clientIp,\n },\n });\n }\n\n return true;\n }\n\n if (success) {\n return true;\n } else {\n res.header(HttpResponseHeaderKeysEnum.RETRY_AFTER, secondsToReset);\n\n this.logger.debug({\n message: 'Rate limit exceeded',\n _event: {\n limit,\n remaining,\n retryAfter: secondsToReset,\n category: apiRateLimitCategory,\n organizationId,\n environmentId,\n ip: clientIp,\n isKeyless: isKeylessRequest,\n },\n });\n\n throw new ThrottlerException(THROTTLED_EXCEPTION_MESSAGE);\n }\n }\n\n private createPolicyHeader(\n limit: number,\n windowDuration: number,\n burstLimit: number,\n algorithm: string,\n apiRateLimitCategory: ApiRateLimitCategoryEnum,\n apiRateLimitCost: ApiRateLimitCostEnum,\n apiServiceLevel: string\n ): string {\n const policyMap = {\n w: windowDuration,\n burst: burstLimit,\n comment: `\"${algorithm}\"`,\n category: `\"${apiRateLimitCategory}\"`,\n cost: `\"${apiRateLimitCost}\"`,\n serviceLevel: `\"${apiServiceLevel}\"`,\n };\n const policy = Object.entries(policyMap).reduce((acc, [key, value]) => {\n return `${acc};${key}=${value}`;\n }, `${limit}`);\n\n return policy;\n }\n\n private isAllowedAuthScheme(context: ExecutionContext): boolean {\n const { authScheme } = context.switchToHttp().getRequest();\n\n return ALLOWED_AUTH_SCHEMES.some((scheme) => authScheme === scheme);\n }\n\n private isAllowedEnvironment(context: ExecutionContext): boolean {\n const req = context.switchToHttp().getRequest();\n const applicationIdentifier = req.headers['novu-application-identifier'];\n\n if (!applicationIdentifier) {\n return false;\n }\n\n return applicationIdentifier.startsWith('pk_keyless_');\n }\n\n private isAllowedRoute(context: ExecutionContext): boolean {\n return this.isKeylessRoute(context);\n }\n\n private isKeylessRoute(context: ExecutionContext): boolean {\n const req = context.switchToHttp().getRequest();\n\n return req.path === '/v1/inbox/session' && req.method === 'POST';\n }\n\n private getReqUser(context: ExecutionContext): UserSessionData | undefined {\n const req = context.switchToHttp().getRequest();\n\n return req.user;\n }\n}\n\nfunction getKeylessCost() {\n // For test environment, we use a higher cost to ensure tests can run without rate limiting issues\n return process.env.NODE_ENV === 'test' ? defaultApiRateLimitCost : ApiRateLimitCostEnum.KEYLESS;\n}\n"
},
{
"path": "apps/api/src/app/rate-limiting/rate-limiting.module.ts",
"content": "import { Module } from '@nestjs/common';\nimport { ThrottlerModule } from '@nestjs/throttler';\nimport { CommunityOrganizationRepository } from '@novu/dal';\nimport { SharedModule } from '../shared/shared.module';\nimport { ApiRateLimitInterceptor } from './guards';\nimport { USE_CASES } from './usecases';\n\n@Module({\n imports: [\n SharedModule,\n ThrottlerModule.forRoot([\n // The following configuration is required for the NestJS ThrottlerModule to work. It has no effect.\n {\n ttl: 60000,\n limit: 10,\n },\n ]),\n ],\n providers: [...USE_CASES, ApiRateLimitInterceptor, CommunityOrganizationRepository],\n exports: [...USE_CASES, ApiRateLimitInterceptor],\n})\nexport class RateLimitingModule {}\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/evaluate-api-rate-limit/evaluate-api-rate-limit.command.ts",
"content": "import { BaseCommand } from '@novu/application-generic';\nimport { ApiRateLimitCategoryEnum, ApiRateLimitCostEnum } from '@novu/shared';\nimport { IsDefined, IsEnum, IsNotEmpty, IsOptional, IsString } from 'class-validator';\n\nexport class EvaluateApiRateLimitCommand extends BaseCommand {\n @IsOptional()\n @IsString()\n readonly environmentId?: string;\n\n @IsOptional()\n @IsString()\n readonly organizationId?: string;\n\n @IsDefined()\n @IsEnum(ApiRateLimitCategoryEnum)\n apiRateLimitCategory: ApiRateLimitCategoryEnum;\n\n @IsDefined()\n @IsEnum(ApiRateLimitCostEnum)\n apiRateLimitCost: ApiRateLimitCostEnum;\n\n @IsOptional()\n @IsString()\n ip?: string;\n}\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/evaluate-api-rate-limit/evaluate-api-rate-limit.spec.ts",
"content": "import { Test } from '@nestjs/testing';\nimport {\n ApiRateLimitAlgorithmEnum,\n ApiRateLimitCategoryEnum,\n ApiRateLimitCostEnum,\n ApiServiceLevelEnum,\n IApiRateLimitAlgorithm,\n IApiRateLimitCost,\n} from '@novu/shared';\nimport { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\nimport sinon from 'sinon';\nimport { SharedModule } from '../../../shared/shared.module';\nimport { RateLimitingModule } from '../../rate-limiting.module';\nimport { EvaluateTokenBucketRateLimit } from '../evaluate-token-bucket-rate-limit';\nimport { GetApiRateLimitAlgorithmConfig } from '../get-api-rate-limit-algorithm-config';\nimport { GetApiRateLimitCostConfig } from '../get-api-rate-limit-cost-config';\nimport { GetApiRateLimitMaximum } from '../get-api-rate-limit-maximum';\nimport { EvaluateApiRateLimit, EvaluateApiRateLimitCommand } from './index';\n\nconst mockApiRateLimitAlgorithm: IApiRateLimitAlgorithm = {\n [ApiRateLimitAlgorithmEnum.BURST_ALLOWANCE]: 0.2,\n [ApiRateLimitAlgorithmEnum.WINDOW_DURATION]: 2,\n};\nconst mockApiRateLimitCost = ApiRateLimitCostEnum.SINGLE;\nconst mockApiServiceLevel = ApiServiceLevelEnum.FREE;\nconst mockCost = 1;\nconst mockApiRateLimitCostConfig: Partial = {\n [mockApiRateLimitCost]: mockCost,\n};\n\nconst mockMaxLimit = 10;\nconst mockRemaining = 9;\nconst mockReset = 1;\nconst mockApiRateLimitCategory = ApiRateLimitCategoryEnum.GLOBAL;\n\ndescribe('EvaluateApiRateLimit', async () => {\n let useCase: EvaluateApiRateLimit;\n let session: UserSession;\n let getApiRateLimitMaximum: GetApiRateLimitMaximum;\n let getApiRateLimitAlgorithmConfig: GetApiRateLimitAlgorithmConfig;\n let getApiRateLimitCostConfig: GetApiRateLimitCostConfig;\n let evaluateTokenBucketRateLimit: EvaluateTokenBucketRateLimit;\n\n let getApiRateLimitMaximumStub: sinon.SinonStub;\n let getApiRateLimitAlgorithmConfigStub: sinon.SinonStub;\n let getApiRateLimitCostConfigStub: sinon.SinonStub;\n let evaluateTokenBucketRateLimitStub: sinon.SinonStub;\n\n beforeEach(async () => {\n const moduleRef = await Test.createTestingModule({\n imports: [SharedModule, RateLimitingModule],\n }).compile();\n\n session = new UserSession();\n await session.initialize();\n\n useCase = moduleRef.get(EvaluateApiRateLimit);\n getApiRateLimitMaximum = moduleRef.get(GetApiRateLimitMaximum);\n getApiRateLimitAlgorithmConfig = moduleRef.get(GetApiRateLimitAlgorithmConfig);\n getApiRateLimitCostConfig = moduleRef.get(GetApiRateLimitCostConfig);\n evaluateTokenBucketRateLimit = moduleRef.get(EvaluateTokenBucketRateLimit);\n\n getApiRateLimitMaximumStub = sinon\n .stub(getApiRateLimitMaximum, 'execute')\n .resolves([mockMaxLimit, mockApiServiceLevel]);\n getApiRateLimitAlgorithmConfigStub = sinon\n .stub(getApiRateLimitAlgorithmConfig, 'default')\n .value(mockApiRateLimitAlgorithm);\n getApiRateLimitCostConfigStub = sinon.stub(getApiRateLimitCostConfig, 'default').value(mockApiRateLimitCostConfig);\n evaluateTokenBucketRateLimitStub = sinon.stub(evaluateTokenBucketRateLimit, 'execute').resolves({\n success: true,\n limit: mockMaxLimit,\n remaining: mockRemaining,\n reset: mockReset,\n });\n });\n\n afterEach(() => {\n getApiRateLimitMaximumStub.restore();\n getApiRateLimitAlgorithmConfigStub.restore();\n getApiRateLimitCostConfigStub.restore();\n });\n\n describe('Evaluation Values', () => {\n it('should return a boolean success value', async () => {\n const result = await useCase.execute(\n EvaluateApiRateLimitCommand.create({\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n apiRateLimitCategory: mockApiRateLimitCategory,\n apiRateLimitCost: mockApiRateLimitCost,\n })\n );\n\n expect(typeof result.success).to.equal('boolean');\n });\n\n it('should return a positive limit', async () => {\n const result = await useCase.execute(\n EvaluateApiRateLimitCommand.create({\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n apiRateLimitCategory: mockApiRateLimitCategory,\n apiRateLimitCost: mockApiRateLimitCost,\n })\n );\n\n expect(result.limit).to.be.greaterThan(0);\n });\n\n it('should return a positive remaining tokens ', async () => {\n const result = await useCase.execute(\n EvaluateApiRateLimitCommand.create({\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n apiRateLimitCategory: mockApiRateLimitCategory,\n apiRateLimitCost: mockApiRateLimitCost,\n })\n );\n\n expect(result.remaining).to.be.greaterThan(0);\n });\n\n it('should return a positive reset', async () => {\n const result = await useCase.execute(\n EvaluateApiRateLimitCommand.create({\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n apiRateLimitCategory: mockApiRateLimitCategory,\n apiRateLimitCost: mockApiRateLimitCost,\n })\n );\n\n expect(result.reset).to.be.greaterThan(0);\n });\n });\n\n describe('Static Values', () => {\n it('should return a string type algorithm value', async () => {\n const result = await useCase.execute(\n EvaluateApiRateLimitCommand.create({\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n apiRateLimitCategory: mockApiRateLimitCategory,\n apiRateLimitCost: mockApiRateLimitCost,\n })\n );\n\n expect(typeof result.algorithm).to.equal('string');\n });\n\n it('should return the correct window duration', async () => {\n const result = await useCase.execute(\n EvaluateApiRateLimitCommand.create({\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n apiRateLimitCategory: mockApiRateLimitCategory,\n apiRateLimitCost: mockApiRateLimitCost,\n })\n );\n\n expect(result.windowDuration).to.equal(mockApiRateLimitAlgorithm[ApiRateLimitAlgorithmEnum.WINDOW_DURATION]);\n });\n });\n\n describe('Computed Values', () => {\n it('should return the correct cost', async () => {\n const result = await useCase.execute(\n EvaluateApiRateLimitCommand.create({\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n apiRateLimitCategory: mockApiRateLimitCategory,\n apiRateLimitCost: mockApiRateLimitCost,\n })\n );\n\n expect(result.cost).to.equal(mockApiRateLimitCostConfig[mockApiRateLimitCost]);\n });\n\n it('should return the correct refill rate', async () => {\n const result = await useCase.execute(\n EvaluateApiRateLimitCommand.create({\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n apiRateLimitCategory: mockApiRateLimitCategory,\n apiRateLimitCost: mockApiRateLimitCost,\n })\n );\n\n expect(result.refillRate).to.equal(\n mockMaxLimit * mockApiRateLimitAlgorithm[ApiRateLimitAlgorithmEnum.WINDOW_DURATION]\n );\n });\n\n it('should return the correct burst limit', async () => {\n const result = await useCase.execute(\n EvaluateApiRateLimitCommand.create({\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n apiRateLimitCategory: mockApiRateLimitCategory,\n apiRateLimitCost: mockApiRateLimitCost,\n })\n );\n\n expect(result.burstLimit).to.equal(\n mockMaxLimit *\n mockApiRateLimitAlgorithm[ApiRateLimitAlgorithmEnum.WINDOW_DURATION] *\n (1 + mockApiRateLimitAlgorithm[ApiRateLimitAlgorithmEnum.BURST_ALLOWANCE])\n );\n });\n });\n});\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/evaluate-api-rate-limit/evaluate-api-rate-limit.types.ts",
"content": "export type EvaluateApiRateLimitResponseDto = {\n /**\n * Whether the request may pass(true) or exceeded the limit(false)\n */\n success: boolean;\n /**\n * Maximum number of requests allowed within a window.\n */\n limit: number;\n /**\n * How many requests the client has left within the current window.\n */\n remaining: number;\n /**\n * Unix timestamp in milliseconds when the limits are reset.\n */\n reset: number;\n /**\n * The duration of the window in seconds.\n */\n windowDuration: number;\n /**\n * The maximum number of requests allowed within a window, including the burst allowance.\n */\n burstLimit: number;\n /**\n * The number of requests that will be refilled per window.\n */\n refillRate: number;\n /**\n * The name of the algorithm used to calculate the rate limit.\n */\n algorithm: string;\n /**\n * The cost of the request.\n */\n cost: number;\n /**\n * The API service level used to evaluate the request.\n */\n apiServiceLevel: string;\n};\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/evaluate-api-rate-limit/evaluate-api-rate-limit.usecase.ts",
"content": "import { Injectable } from '@nestjs/common';\nimport { buildEvaluateApiRateLimitKey, InstrumentUsecase } from '@novu/application-generic';\nimport {\n ApiRateLimitAlgorithmEnum,\n ApiServiceLevelEnum,\n FeatureNameEnum,\n getFeatureForTierAsNumber,\n} from '@novu/shared';\nimport { EvaluateTokenBucketRateLimitCommand } from '../evaluate-token-bucket-rate-limit/evaluate-token-bucket-rate-limit.command';\nimport { EvaluateTokenBucketRateLimit } from '../evaluate-token-bucket-rate-limit/evaluate-token-bucket-rate-limit.usecase';\nimport { GetApiRateLimitAlgorithmConfig } from '../get-api-rate-limit-algorithm-config';\nimport { GetApiRateLimitCostConfig } from '../get-api-rate-limit-cost-config';\nimport { GetApiRateLimitMaximum, GetApiRateLimitMaximumCommand } from '../get-api-rate-limit-maximum';\nimport type { ApiServiceLevel } from '../get-api-rate-limit-maximum/get-api-rate-limit-maximum.dto';\nimport { EvaluateApiRateLimitCommand } from './evaluate-api-rate-limit.command';\nimport { EvaluateApiRateLimitResponseDto } from './evaluate-api-rate-limit.types';\n\n@Injectable()\nexport class EvaluateApiRateLimit {\n constructor(\n private getApiRateLimitMaximum: GetApiRateLimitMaximum,\n private getApiRateLimitAlgorithmConfig: GetApiRateLimitAlgorithmConfig,\n private getApiRateLimitCostConfig: GetApiRateLimitCostConfig,\n private evaluateTokenBucketRateLimit: EvaluateTokenBucketRateLimit\n ) {}\n\n @InstrumentUsecase()\n async execute(command: EvaluateApiRateLimitCommand): Promise {\n let maxLimitPerSecond: number;\n let apiServiceLevel: ApiServiceLevel;\n\n // For keyless environments, we implement strict rate limiting to prevent abuse:\n if (!command.organizationId || !command.environmentId) {\n maxLimitPerSecond = 3000;\n apiServiceLevel = ApiServiceLevelEnum.ENTERPRISE;\n } else {\n [maxLimitPerSecond, apiServiceLevel] = await this.getApiRateLimitMaximum.execute(\n GetApiRateLimitMaximumCommand.create({\n apiRateLimitCategory: command.apiRateLimitCategory,\n environmentId: command.environmentId,\n organizationId: command.organizationId,\n })\n );\n }\n\n const windowDuration = this.getApiRateLimitAlgorithmConfig.default[ApiRateLimitAlgorithmEnum.WINDOW_DURATION];\n const burstAllowance = this.getApiRateLimitAlgorithmConfig.default[ApiRateLimitAlgorithmEnum.BURST_ALLOWANCE];\n const cost = this.getApiRateLimitCostConfig.default[command.apiRateLimitCost];\n const maxTokensPerWindow = this.getMaxTokensPerWindow(maxLimitPerSecond, windowDuration);\n const refillRate = this.getRefillRate(maxLimitPerSecond, windowDuration);\n const burstLimit = this.getBurstLimit(maxTokensPerWindow, burstAllowance);\n\n // For keyless authentication, we'll use both environment and IP-based rate limiting\n const identifier = buildEvaluateApiRateLimitKey({\n _environmentId: command.environmentId || 'keyless_env',\n apiRateLimitCategory: command.ip\n ? `${command.apiRateLimitCategory}:ip=${command.ip}`\n : command.apiRateLimitCategory,\n });\n\n const { success, remaining, reset } = await this.evaluateTokenBucketRateLimit.execute(\n EvaluateTokenBucketRateLimitCommand.create({\n identifier,\n maxTokens: burstLimit,\n windowDuration,\n cost,\n refillRate,\n })\n );\n\n return {\n success,\n limit: maxTokensPerWindow,\n remaining,\n reset,\n windowDuration,\n burstLimit,\n refillRate,\n algorithm: this.evaluateTokenBucketRateLimit.algorithm,\n cost,\n apiServiceLevel,\n };\n }\n\n private getMaxTokensPerWindow(maxLimit: number, windowDuration: number): number {\n return maxLimit * windowDuration;\n }\n\n private getRefillRate(maxLimit: number, windowDuration: number): number {\n /*\n * Refill rate is currently set to the max tokens per window.\n * This can be changed to a different value to implement adaptive rate limiting.\n */\n return this.getMaxTokensPerWindow(maxLimit, windowDuration);\n }\n\n private getBurstLimit(maxTokensPerWindow: number, burstAllowance: number): number {\n return Math.floor(maxTokensPerWindow * (1 + burstAllowance));\n }\n}\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/evaluate-api-rate-limit/index.ts",
"content": "export * from './evaluate-api-rate-limit.command';\nexport * from './evaluate-api-rate-limit.types';\nexport * from './evaluate-api-rate-limit.usecase';\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/evaluate-token-bucket-rate-limit/evaluate-token-bucket-rate-limit.command.ts",
"content": "import { BaseCommand } from '@novu/application-generic';\nimport { IsDefined, IsNumber, IsString } from 'class-validator';\n\nexport class EvaluateTokenBucketRateLimitCommand extends BaseCommand {\n @IsDefined()\n @IsString()\n identifier: string;\n\n @IsDefined()\n @IsNumber()\n maxTokens: number;\n\n @IsDefined()\n @IsNumber()\n windowDuration: number;\n\n @IsDefined()\n @IsNumber()\n cost: number;\n\n @IsDefined()\n @IsNumber()\n refillRate: number;\n}\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/evaluate-token-bucket-rate-limit/evaluate-token-bucket-rate-limit.spec.ts",
"content": "import { Test } from '@nestjs/testing';\nimport { CacheService, cacheService as inMemoryCacheService } from '@novu/application-generic';\nimport { expect } from 'chai';\nimport sinon from 'sinon';\nimport { v4 as uuid } from 'uuid';\nimport { SharedModule } from '../../../shared/shared.module';\nimport { RateLimitingModule } from '../../rate-limiting.module';\nimport { EvaluateTokenBucketRateLimitCommand } from './evaluate-token-bucket-rate-limit.command';\nimport { EvaluateTokenBucketRateLimit } from './evaluate-token-bucket-rate-limit.usecase';\n\ndescribe('EvaluateTokenBucketRateLimit', () => {\n let useCase: EvaluateTokenBucketRateLimit;\n let cacheService: CacheService;\n\n const mockCommand = EvaluateTokenBucketRateLimitCommand.create({\n identifier: 'test',\n maxTokens: 10,\n windowDuration: 1,\n cost: 1,\n refillRate: 1,\n });\n\n beforeEach(async () => {\n const moduleRef = await Test.createTestingModule({\n imports: [SharedModule, RateLimitingModule],\n }).compile();\n\n useCase = moduleRef.get(EvaluateTokenBucketRateLimit);\n cacheService = moduleRef.get(CacheService);\n });\n\n describe('Static values', () => {\n it('should have a static algorithm value', () => {\n expect(useCase.algorithm).to.equal('token bucket');\n });\n });\n\n describe('Cache invocation', () => {\n let cacheServiceEvalStub: sinon.SinonStub;\n let cacheServiceSaddStub: sinon.SinonStub;\n let cacheServiceIsEnabledStub: sinon.SinonStub;\n\n beforeEach(async () => {\n cacheServiceEvalStub = sinon.stub(cacheService, 'eval');\n cacheServiceSaddStub = sinon.stub(cacheService, 'sadd');\n cacheServiceIsEnabledStub = sinon.stub(cacheService, 'cacheEnabled').returns(true);\n });\n\n afterEach(() => {\n cacheServiceEvalStub.restore();\n cacheServiceSaddStub.restore();\n cacheServiceIsEnabledStub.restore();\n });\n\n describe('Cache Errors', () => {\n it('should throw error when a cache operation fails', async () => {\n cacheServiceEvalStub.resolves(new Error());\n\n try {\n await useCase.execute(mockCommand);\n throw new Error('Should not reach here');\n } catch (e) {\n expect(e.message).to.equal('Failed to evaluate rate limit');\n }\n });\n\n it('should throw error when cache is not enabled', async () => {\n cacheServiceIsEnabledStub.returns(false);\n\n try {\n await useCase.execute(mockCommand);\n throw new Error('Should not reach here');\n } catch (e) {\n expect(e.message).to.equal('Rate limiting cache service is not available');\n }\n });\n });\n\n describe('Cache Service Adapter', () => {\n it('should invoke the SADD method with members casted to string', async () => {\n const cacheClient = EvaluateTokenBucketRateLimit.getCacheClient(cacheService);\n const key = 'testKey';\n const members = [1, 2];\n\n await cacheClient.sadd(key, ...members);\n\n expect(cacheServiceSaddStub.calledWith(key, ...['1', '2'])).to.equal(true);\n });\n\n it('should invoke the EVAL function with args casted to string', async () => {\n const cacheClient = EvaluateTokenBucketRateLimit.getCacheClient(cacheService);\n const script = 'return 1';\n const keys = ['key1', 'key2'];\n const args = [1, 2];\n\n await cacheClient.eval(script, keys, args);\n\n expect(cacheServiceEvalStub.calledWith(script, keys, ['1', '2'])).to.equal(true);\n });\n });\n\n describe.skip('Redis EVAL script benchmarks', () => {\n type TestCase = {\n /**\n * Test scenario description\n */\n description: string;\n /**\n * Total number of requests to simulate\n */\n totalRequests: number;\n /**\n * Proportion of requests that have a unique identifier\n */\n proportionUniqueIds: number;\n /**\n * Proportion of requests that are throttled\n */\n proportionThrottled: number;\n /**\n * Proportion of requests that are high cost\n */\n proportionHighCost: number;\n /**\n * The proportion of the window duration to jitter the request duration by.\n * Low value to simulate burst request patterns.\n * High value to simulate sustained request patterns.\n */\n proportionJitter: number;\n /**\n * Expected maximum total evaluation duration in milliseconds\n */\n expectedTotalTimeMs: number;\n /**\n * Expected average evaluation duration in milliseconds\n */\n expectedAverageTimeMs: number;\n /**\n * Expected nth percentile evaluation duration in milliseconds\n */\n expectedNthPercentileTimeMs: number;\n };\n\n const testCases: TestCase[] = [\n {\n description: 'Low Load - 0% Throttled - Sustained Single Window',\n totalRequests: 5000,\n proportionUniqueIds: 0.5,\n proportionThrottled: 0,\n proportionHighCost: 0,\n proportionJitter: 0.8,\n expectedTotalTimeMs: 1000,\n expectedAverageTimeMs: 10,\n expectedNthPercentileTimeMs: 30,\n },\n {\n description: 'Medium Load - 0% Throttled - Sustained Single Window',\n totalRequests: 10000,\n proportionUniqueIds: 0.5,\n proportionThrottled: 0,\n proportionHighCost: 0,\n proportionJitter: 0.8,\n expectedTotalTimeMs: 1000,\n expectedAverageTimeMs: 20,\n expectedNthPercentileTimeMs: 50,\n },\n {\n description: 'High Load - 0% Throttled - Sustained Single Window',\n totalRequests: 20000,\n proportionUniqueIds: 0.5,\n proportionThrottled: 0,\n proportionHighCost: 0,\n proportionJitter: 0.8,\n expectedTotalTimeMs: 1000,\n expectedAverageTimeMs: 200,\n expectedNthPercentileTimeMs: 500,\n },\n {\n description: 'Extreme Load - 0% Throttled - Sustained Single Window',\n totalRequests: 40000,\n proportionUniqueIds: 0.5,\n proportionThrottled: 0,\n proportionHighCost: 0,\n proportionJitter: 0.8,\n expectedTotalTimeMs: 2000,\n expectedAverageTimeMs: 500,\n expectedNthPercentileTimeMs: 2000,\n },\n {\n description: 'High Load - 0% Throttled - Burst Single Window',\n totalRequests: 20000,\n proportionUniqueIds: 0.5,\n proportionThrottled: 0,\n proportionHighCost: 0,\n proportionJitter: 0.2,\n expectedTotalTimeMs: 1000,\n expectedAverageTimeMs: 500,\n expectedNthPercentileTimeMs: 1000,\n },\n {\n description: 'Extreme Load - 0% Throttled - Burst Single Window',\n totalRequests: 40000,\n proportionUniqueIds: 0.5,\n proportionThrottled: 0,\n proportionHighCost: 0,\n proportionJitter: 0.2,\n expectedTotalTimeMs: 3000,\n expectedAverageTimeMs: 1500,\n expectedNthPercentileTimeMs: 2000,\n },\n {\n description: 'High Load - 50% Throttled - Burst Single Window',\n totalRequests: 20000,\n proportionUniqueIds: 0.5,\n proportionThrottled: 0.5,\n proportionHighCost: 0,\n proportionJitter: 0.2,\n expectedTotalTimeMs: 1000,\n expectedAverageTimeMs: 500,\n expectedNthPercentileTimeMs: 1000,\n },\n {\n description: 'High Load - 50% Throttled - Sustained Single Window',\n totalRequests: 20000,\n proportionUniqueIds: 0.5,\n proportionThrottled: 0.5,\n proportionHighCost: 0,\n proportionJitter: 0.8,\n expectedTotalTimeMs: 1000,\n expectedAverageTimeMs: 500,\n expectedNthPercentileTimeMs: 500,\n },\n {\n description: 'High Load - 50% Throttled & 50% High-Cost - Sustained Multiple Windows',\n totalRequests: 40000,\n proportionUniqueIds: 0.5,\n proportionThrottled: 0.5,\n proportionHighCost: 0.5,\n proportionJitter: 2.2,\n expectedTotalTimeMs: 3000,\n expectedAverageTimeMs: 30,\n expectedNthPercentileTimeMs: 100,\n },\n {\n description: 'Extreme Load - 50% Throttled & 50% High-Cost - Sustained Multiple Windows',\n totalRequests: 80000,\n proportionUniqueIds: 0.5,\n proportionThrottled: 0.5,\n proportionHighCost: 0.5,\n proportionJitter: 2.2,\n expectedTotalTimeMs: 4000,\n expectedAverageTimeMs: 1000,\n expectedNthPercentileTimeMs: 1500,\n },\n {\n description: 'High Load - 50% Throttled & 90% High-Cost - Sustained Multiple Windows',\n totalRequests: 40000,\n proportionUniqueIds: 0.5,\n proportionThrottled: 0.5,\n proportionHighCost: 0.9,\n proportionJitter: 2.2,\n expectedTotalTimeMs: 3000,\n expectedAverageTimeMs: 50,\n expectedNthPercentileTimeMs: 200,\n },\n {\n description: 'High Load - 50% Throttled & 0% Unique - Sustained Multiple Windows',\n totalRequests: 40000,\n proportionUniqueIds: 0,\n proportionThrottled: 0.5,\n proportionHighCost: 0,\n proportionJitter: 2.2,\n expectedTotalTimeMs: 3000,\n expectedAverageTimeMs: 30,\n expectedNthPercentileTimeMs: 200,\n },\n {\n description: 'High Load - 50% Throttled & 100% Unique - Sustained Multiple Windows',\n totalRequests: 40000,\n proportionUniqueIds: 1,\n proportionThrottled: 0.5,\n proportionHighCost: 0,\n proportionJitter: 2.2,\n expectedTotalTimeMs: 3000,\n expectedAverageTimeMs: 30,\n expectedNthPercentileTimeMs: 100,\n },\n ];\n const mockLowCost = 1;\n const mockHighCost = 10;\n const mockWindowDuration = 1;\n const mockWindowDurationMs = mockWindowDuration * 1000;\n const mockProportionRefill = 0.5;\n\n const testThrottledCountErrorTolerance = 0.2;\n const testPercentile = 0.95;\n\n function printHistogram(results) {\n // Define the number of bins for the histogram\n const bins = 10;\n\n // Find the maximum duration to scale the histogram\n const maxDuration = Math.max(...results.map((result) => result.duration));\n\n // Initialize an array for the histogram bins\n const histogram = Array(bins).fill(0);\n\n // Populate the histogram bins\n results.forEach((result) => {\n const index = Math.floor((result.duration / maxDuration) * bins);\n histogram[index < bins ? index : bins - 1] += 1;\n });\n\n // Find the maximum bin count to scale the histogram height\n const maxCount = Math.max(...histogram);\n\n // Print the histogram\n console.log(`\\t Request Time (ms)`);\n histogram.forEach((count, i) => {\n const bar = '*'.repeat((count / maxCount) * 50); // Scale to a max width of 50 \"*\"\n console.log(`\\t ${(((i + 1) / bins) * maxDuration).toFixed(2).padStart(7)}: ${bar}`);\n });\n }\n\n testCases\n .map(\n ({\n description,\n totalRequests,\n proportionUniqueIds,\n proportionThrottled,\n proportionHighCost,\n proportionJitter,\n expectedAverageTimeMs,\n expectedNthPercentileTimeMs,\n expectedTotalTimeMs,\n }) => {\n return () => {\n describe(description, () => {\n let testContext;\n let results: Array<{ duration: number; success: boolean }>;\n let totalTime: number;\n let averageTime: number;\n let successCount: number;\n let throttledCount: number;\n let variance: number;\n let stdev: number;\n let nthPercentile: number;\n\n const maxTokens = Math.ceil(totalRequests * (1 - proportionThrottled));\n const uniqueIdRequests = Math.max(1, Math.floor(totalRequests * proportionUniqueIds));\n const uniqueIds = Array.from({ length: uniqueIdRequests }).map(() => uuid());\n const mockRepeatId = uuid();\n const maxJitterMs = mockWindowDurationMs * proportionJitter;\n\n const refillPerWindow = (maxTokens * mockProportionRefill) / mockWindowDuration;\n\n before(async () => {\n const cacheServiceInitialized = await inMemoryCacheService.useFactory();\n testContext = {\n redis: EvaluateTokenBucketRateLimit.getCacheClient(cacheServiceInitialized),\n };\n\n const proms = Array.from({ length: totalRequests }).map(async (_val, index) => {\n const cost = Math.random() < proportionHighCost ? mockHighCost : mockLowCost;\n /**\n * Distribute unique ids with request allocation skewed left.\n * matching an expected distribution of requests per unique API client, where:\n * - the majority of clients make a small number of requests\n * - a small number of clients make a large number of requests\n *\n * Number of Requests per Unique Id\n * ID Requests\n * 1 *\n * 2 **\n * 3 ****\n * 4 ******\n * 5 *********\n * 6 *************\n * 7 *****************\n * 8 ***********************\n * 9 ********************************\n * 10 *******************************************\n */\n const id =\n Math.random() < proportionUniqueIds\n ? uniqueIds[Math.floor((index / totalRequests) * uniqueIds.length)]\n : mockRepeatId;\n\n const jitter = Math.floor(Math.random() * maxJitterMs);\n await new Promise((resolve) => {\n setTimeout(resolve, jitter);\n });\n const start = Date.now();\n const limit = EvaluateTokenBucketRateLimit.tokenBucketLimiter(\n refillPerWindow,\n mockWindowDuration,\n maxTokens,\n cost\n );\n const { success } = await limit(testContext, id);\n const end = Date.now();\n const duration = end - start;\n\n return {\n duration,\n success,\n };\n });\n\n const startAll = Date.now();\n results = await Promise.all(proms);\n const endAll = Date.now();\n\n totalTime = endAll - startAll;\n averageTime = results.reduce((acc, val) => acc + val.duration, 0) / results.length;\n variance = results.reduce((acc, val) => acc + (val.duration - averageTime) ** 2, 0) / results.length;\n stdev = Math.sqrt(variance);\n nthPercentile = results.sort((a, b) => a.duration - b.duration)[\n Math.floor(results.length * testPercentile)\n ].duration;\n successCount = results.filter(({ success }) => success).length;\n throttledCount = totalRequests - successCount;\n\n console.log(\n `\\t Params: Total Req: ${totalRequests.toLocaleString()}\\tUsers: ${uniqueIdRequests.toLocaleString()}\\tThrottled: ${\n proportionThrottled * 100\n }%\\tHigh Cost: ${proportionHighCost * 100}%\\tJitter: ${maxJitterMs}ms`\n );\n console.log(\n `\\t Stats: Total Time: ${totalTime.toLocaleString()}ms\\tAvg: ${averageTime.toFixed(\n 1\n )}ms\\tStdev: ${stdev.toFixed(1)}\\tp(${\n testPercentile * 100\n }): ${nthPercentile}\\tThrottled: ${throttledCount.toLocaleString()}`\n );\n printHistogram(results);\n });\n\n describe('Script Performance', () => {\n it(`should be able to process ${totalRequests.toLocaleString()} evaluations in less than ${expectedTotalTimeMs}ms`, async () => {\n expect(totalTime).to.be.lessThan(expectedTotalTimeMs);\n });\n\n it(`should have average evaluation duration less than ${expectedAverageTimeMs}ms`, async () => {\n expect(averageTime).to.be.lessThan(expectedAverageTimeMs);\n });\n\n it(`should have ${\n testPercentile * 100\n }th percentile evaluation duration less than ${expectedNthPercentileTimeMs}ms`, async () => {\n expect(nthPercentile).to.be.lessThan(expectedNthPercentileTimeMs);\n });\n });\n\n describe('Script Throttle Evaluation', () => {\n const proportionRequestsPerWindow =\n maxJitterMs > mockWindowDurationMs ? mockWindowDurationMs / maxJitterMs : 1;\n const totalRequestsPerWindow = Math.floor(totalRequests * proportionRequestsPerWindow);\n const uniqueRequestsPerWindow = Math.floor(totalRequestsPerWindow * (1 - proportionThrottled));\n const expectedPerRequestCost =\n (1 - proportionHighCost) * mockLowCost + proportionHighCost * mockHighCost;\n\n const expectedWindowCost = uniqueRequestsPerWindow * expectedPerRequestCost;\n const firstWindowThrottledRequests =\n expectedWindowCost > maxTokens ? (expectedWindowCost - maxTokens) / expectedPerRequestCost : 0;\n const secondWindowMaxTokens = Math.max(\n maxTokens,\n maxTokens - firstWindowThrottledRequests + refillPerWindow\n );\n const secondWindowThrottledRequests =\n expectedWindowCost > secondWindowMaxTokens\n ? (expectedWindowCost - secondWindowMaxTokens) / expectedPerRequestCost\n : 0;\n\n const expectedThrottledCount = firstWindowThrottledRequests + secondWindowThrottledRequests;\n const expectedThrottledCountMin = Math.floor(\n expectedThrottledCount * (1 - testThrottledCountErrorTolerance)\n );\n const expectedThrottledCountMax = Math.floor(\n expectedThrottledCount * (1 + testThrottledCountErrorTolerance)\n );\n\n it(`should throttle between ${expectedThrottledCountMin} and ${expectedThrottledCountMax} requests`, async () => {\n expect(throttledCount).to.be.greaterThanOrEqual(expectedThrottledCountMin);\n expect(throttledCount).to.be.lessThanOrEqual(expectedThrottledCountMax);\n });\n });\n });\n };\n }\n )\n .forEach((testCase) => {\n testCase();\n });\n });\n });\n});\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/evaluate-token-bucket-rate-limit/evaluate-token-bucket-rate-limit.types.ts",
"content": "import { Ratelimit } from '@upstash/ratelimit';\n\nexport type UpstashRedisClient = ConstructorParameters[0]['redis'];\n\nexport type EvaluateTokenBucketRateLimitResponseDto = {\n /**\n * Whether the request may pass(true) or exceeded the limit(false)\n */\n success: boolean;\n /**\n * Maximum number of requests allowed within a window.\n */\n limit: number;\n /**\n * How many requests the client has left within the current window.\n */\n remaining: number;\n /**\n * Unix timestamp in milliseconds when the limits are reset.\n */\n reset: number;\n};\n\nexport type RegionLimiter = ReturnType;\n\n/**\n * You have a bucket filled with `{maxTokens}` tokens that refills constantly\n * at `{refillRate}` per `{interval}`.\n * Every request will remove `{cost}` token(s) from the bucket and if there is no\n * token to take, the request is rejected.\n *\n * **Pro:**\n *\n * - Bursts of requests are smoothed out and you can process them at a constant\n * rate.\n * - Allows to set a higher initial burst limit by setting `maxTokens` higher\n * than `refillRate`\n */\nexport type CostLimiter = (\n /**\n * How many tokens are refilled per `interval`\n *\n * An interval of `10s` and refillRate of 5 will cause a new token to be added every 2 seconds.\n */\n refillRate: number,\n /**\n * The interval in seconds for the `refillRate`\n */\n interval: number,\n /**\n * Maximum number of tokens.\n * A newly created bucket starts with this many tokens.\n * Useful to allow higher burst limits.\n */\n maxTokens: number,\n /**\n * The number of tokens used in the request.\n */\n cost: number\n) => RegionLimiter;\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/evaluate-token-bucket-rate-limit/evaluate-token-bucket-rate-limit.usecase.ts",
"content": "import { Injectable, ServiceUnavailableException } from '@nestjs/common';\nimport { CacheService, InstrumentUsecase, PinoLogger } from '@novu/application-generic';\nimport { Ratelimit } from '@upstash/ratelimit';\nimport { EvaluateTokenBucketRateLimitCommand } from './evaluate-token-bucket-rate-limit.command';\nimport {\n EvaluateTokenBucketRateLimitResponseDto,\n RegionLimiter,\n UpstashRedisClient,\n} from './evaluate-token-bucket-rate-limit.types';\n\nconst LOG_CONTEXT = 'EvaluateTokenBucketRateLimit';\n\n@Injectable()\nexport class EvaluateTokenBucketRateLimit {\n private ephemeralCache = new Map();\n public algorithm = 'token bucket';\n\n constructor(\n private cacheService: CacheService,\n private logger: PinoLogger\n ) {\n this.logger.setContext(this.constructor.name);\n }\n\n @InstrumentUsecase()\n async execute(command: EvaluateTokenBucketRateLimitCommand): Promise {\n if (!this.cacheService.cacheEnabled()) {\n const message = 'Rate limiting cache service is not available';\n this.logger.error(message);\n throw new ServiceUnavailableException(message);\n }\n\n const cacheClient = EvaluateTokenBucketRateLimit.getCacheClient(this.cacheService);\n\n const ratelimit = new Ratelimit({\n redis: cacheClient,\n limiter: EvaluateTokenBucketRateLimit.tokenBucketLimiter(\n command.refillRate,\n command.windowDuration,\n command.maxTokens,\n command.cost\n ),\n prefix: '', // Empty cache key prefix to give us full control over the key format\n ephemeralCache: this.ephemeralCache,\n });\n try {\n const { success, limit, remaining, reset } = await ratelimit.limit(command.identifier);\n\n return {\n success,\n limit,\n remaining,\n reset,\n };\n } catch (error) {\n const apiMessage = 'Failed to evaluate rate limit';\n const logMessage = `${apiMessage} for identifier: \"${command.identifier}\". Error: \"${error}\"`;\n this.logger.error(logMessage);\n throw new ServiceUnavailableException(apiMessage);\n }\n }\n\n public static getCacheClient(cacheService: CacheService): UpstashRedisClient {\n // Adapter for the @upstash/redis client -> cache client\n return {\n sadd: async (key, ...members) => cacheService.sadd(key, ...members.map((member) => String(member))),\n eval: async (script, keys, args) =>\n cacheService.eval(\n script,\n keys,\n args.map((arg) => String(arg))\n ),\n };\n }\n\n /**\n * Token Bucket algorithm with variable cost. Adapted from @upstash/ratelimit and modified to support variable cost.\n * Also influenced by Krakend's token bucket implementation to delay refills until bucket is empty.\n *\n * @see https://github.com/upstash/ratelimit/blob/3a8cfb00e827188734ac347965cb743a75fcb98a/src/single.ts#L292\n * @see https://github.com/krakend/krakend-ratelimit/blob/369f0be9b51a4fb8ab7d43e4833d076b461a4374/rate.go#L85\n */\n public static tokenBucketLimiter(\n refillRate: number,\n interval: number,\n maxTokens: number,\n cost: number\n ): RegionLimiter {\n const script = /* Lua */ `\n local key = KEYS[1] -- current interval identifier including prefixes\n local maxTokens = tonumber(ARGV[1]) -- maximum number of tokens\n local interval = tonumber(ARGV[2]) -- size of the window in milliseconds\n local fillInterval = tonumber(ARGV[3]) -- time between refills in milliseconds\n local now = tonumber(ARGV[4]) -- current timestamp in milliseconds\n local cost = tonumber(ARGV[5]) -- cost of request\n local remaining = 0 -- remaining number of tokens\n local reset = 0 -- timestamp when next request of {cost} token(s) can be accepted\n local resetCost = 0 -- multiplier for the next reset time\n local lastRefill = 0 -- timestamp of last refill\n\n local bucket = redis.call(\"HMGET\", key, \"lastRefill\", \"tokens\")\n\n if bucket[1] == false then\n -- The bucket does not exist yet, so we create it and add a ttl.\n lastRefill = now\n remaining = maxTokens - cost\n resetCost = (remaining < cost) and (cost - remaining) or cost\n redis.call(\"HMSET\", key, \"lastRefill\", lastRefill, \"tokens\", remaining)\n redis.call(\"PEXPIRE\", key, interval * 2)\n else\n -- The current bucket does exist\n lastRefill = tonumber(bucket[1])\n local tokens = tonumber(bucket[2])\n\n if tokens >= cost then\n -- Delay refill until bucket is empty\n remaining = tokens - cost\n resetCost = (remaining < cost) and (cost - remaining) or cost\n redis.call(\"HMSET\", key, \"tokens\", remaining)\n else\n local elapsed = now - lastRefill\n local tokensToAdd = math.floor(elapsed / fillInterval)\n local newTokens = math.min(maxTokens, tokens + tokensToAdd)\n remaining = newTokens - cost\n\n if remaining >= 0 then\n -- Update the time of the last refill depending on how many tokens we added\n lastRefill = lastRefill + tokensToAdd * fillInterval\n resetCost = (remaining < cost) and (cost - remaining) or cost\n redis.call(\"HMSET\", key, \"lastRefill\", lastRefill, \"tokens\", remaining)\n redis.call(\"PEXPIRE\", key, interval * 2)\n else\n resetCost = cost - tokens\n end\n end\n end\n \n reset = lastRefill + resetCost * fillInterval\n return {remaining, reset}\n`;\n\n const intervalDurationMs = interval * 1e3;\n const fillInterval = intervalDurationMs / refillRate;\n\n return async (ctx, identifier) => {\n // Cost needs to be included in local cache identifier to ensure lower cost requests are not blocked\n const localCacheIdentifier = `${identifier}:${cost}`;\n\n if (ctx.cache) {\n const { blocked, reset } = ctx.cache.isBlocked(localCacheIdentifier);\n if (blocked) {\n return {\n success: false,\n limit: refillRate,\n remaining: 0,\n reset,\n pending: Promise.resolve(),\n };\n }\n }\n\n const now = Date.now();\n\n const [remaining, reset] = (await ctx.redis.eval(\n script,\n [identifier],\n [maxTokens, intervalDurationMs, fillInterval, now, cost]\n )) as [number, number];\n\n const success = remaining >= 0;\n const nonNegativeRemaining = Math.max(0, remaining);\n if (ctx.cache && !success) {\n ctx.cache.blockUntil(localCacheIdentifier, reset);\n }\n\n return {\n success,\n limit: refillRate,\n remaining: nonNegativeRemaining,\n reset,\n pending: Promise.resolve(),\n };\n };\n }\n}\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/evaluate-token-bucket-rate-limit/index.ts",
"content": "export * from './evaluate-token-bucket-rate-limit.command';\nexport * from './evaluate-token-bucket-rate-limit.types';\nexport * from './evaluate-token-bucket-rate-limit.usecase';\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/get-api-rate-limit-algorithm-config/get-api-rate-limit-algorithm-config.spec.ts",
"content": "import { Test } from '@nestjs/testing';\nimport {\n ApiRateLimitAlgorithmEnum,\n ApiRateLimitAlgorithmEnvVarFormat,\n DEFAULT_API_RATE_LIMIT_ALGORITHM_CONFIG,\n} from '@novu/shared';\nimport { expect } from 'chai';\nimport { GetApiRateLimitAlgorithmConfig } from './get-api-rate-limit-algorithm-config.usecase';\n\ndescribe('GetApiRateLimitAlgorithmConfig', () => {\n let useCase: GetApiRateLimitAlgorithmConfig;\n\n beforeEach(async () => {\n const moduleRef = await Test.createTestingModule({\n providers: [GetApiRateLimitAlgorithmConfig],\n }).compile();\n\n useCase = moduleRef.get(GetApiRateLimitAlgorithmConfig);\n });\n\n it('should use the default rate limit algorithm config when no environment variables are set', () => {\n expect(useCase.default).to.deep.equal(DEFAULT_API_RATE_LIMIT_ALGORITHM_CONFIG);\n });\n\n it('should override default rate limit algorithm config with environment variables', () => {\n const mockOverrideBurstAllowance = 0.2;\n const mockApiRateLimitConfigurationKey = ApiRateLimitAlgorithmEnum.BURST_ALLOWANCE;\n\n const envVarName: ApiRateLimitAlgorithmEnvVarFormat = `API_RATE_LIMIT_ALGORITHM_${\n mockApiRateLimitConfigurationKey.toUpperCase() as Uppercase\n }`;\n process.env[envVarName] = `${mockOverrideBurstAllowance}`;\n\n // Re-initialize the defaultApiRateLimits after setting the environment variable\n useCase.loadDefault();\n const result = useCase.default;\n\n expect(result[mockApiRateLimitConfigurationKey]).to.equal(mockOverrideBurstAllowance);\n delete process.env[envVarName]; // cleanup\n });\n});\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/get-api-rate-limit-algorithm-config/get-api-rate-limit-algorithm-config.usecase.ts",
"content": "import { Injectable } from '@nestjs/common';\nimport {\n ApiRateLimitAlgorithmEnum,\n ApiRateLimitAlgorithmEnvVarFormat,\n DEFAULT_API_RATE_LIMIT_ALGORITHM_CONFIG,\n IApiRateLimitAlgorithm,\n} from '@novu/shared';\n\n@Injectable()\nexport class GetApiRateLimitAlgorithmConfig {\n public default: IApiRateLimitAlgorithm;\n\n constructor() {\n this.loadDefault();\n }\n\n public loadDefault(): void {\n this.default = this.createDefault();\n }\n\n private createDefault(): IApiRateLimitAlgorithm {\n const mergedConfig: IApiRateLimitAlgorithm = { ...DEFAULT_API_RATE_LIMIT_ALGORITHM_CONFIG };\n\n // Read process environment only once for performance\n const processEnv = process.env;\n\n Object.values(ApiRateLimitAlgorithmEnum).forEach((algorithmOption) => {\n const envVarName = this.getEnvVarName(algorithmOption);\n const envVarValue = processEnv[envVarName];\n\n if (envVarValue) {\n mergedConfig[algorithmOption] = Number(envVarValue);\n }\n });\n\n return mergedConfig;\n }\n\n private getEnvVarName(algorithmOption: ApiRateLimitAlgorithmEnum): ApiRateLimitAlgorithmEnvVarFormat {\n return `API_RATE_LIMIT_ALGORITHM_${algorithmOption.toUpperCase() as Uppercase}`;\n }\n}\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/get-api-rate-limit-algorithm-config/index.ts",
"content": "export * from './get-api-rate-limit-algorithm-config.usecase';\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/get-api-rate-limit-cost-config/get-api-rate-limit-cost-config.spec.ts",
"content": "import { Test } from '@nestjs/testing';\nimport { ApiRateLimitCostEnum, ApiRateLimitCostEnvVarFormat, DEFAULT_API_RATE_LIMIT_COST_CONFIG } from '@novu/shared';\nimport { expect } from 'chai';\nimport { GetApiRateLimitCostConfig } from './get-api-rate-limit-cost-config.usecase';\n\ndescribe('GetApiRateLimitCostConfig', () => {\n let useCase: GetApiRateLimitCostConfig;\n\n beforeEach(async () => {\n const moduleRef = await Test.createTestingModule({\n providers: [GetApiRateLimitCostConfig],\n }).compile();\n\n useCase = moduleRef.get(GetApiRateLimitCostConfig);\n });\n\n it('should use the default rate limit cost configuration when no environment variables are set', () => {\n expect(useCase.default).to.deep.equal(DEFAULT_API_RATE_LIMIT_COST_CONFIG);\n });\n\n it('should override default rate limit cost configuration with environment variables', () => {\n const mockOverrideBulkCost = 15;\n const mockApiRateLimitConfigurationKey = ApiRateLimitCostEnum.BULK;\n\n const envVarName: ApiRateLimitCostEnvVarFormat = `API_RATE_LIMIT_COST_${\n mockApiRateLimitConfigurationKey.toUpperCase() as Uppercase\n }`;\n process.env[envVarName] = `${mockOverrideBulkCost}`;\n\n // Re-initialize the defaultApiRateLimits after setting the environment variable\n useCase.loadDefault();\n const result = useCase.default;\n\n expect(result[mockApiRateLimitConfigurationKey]).to.equal(mockOverrideBulkCost);\n delete process.env[envVarName]; // cleanup\n });\n});\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/get-api-rate-limit-cost-config/get-api-rate-limit-cost-config.usecase.ts",
"content": "import { Injectable } from '@nestjs/common';\nimport {\n ApiRateLimitCostEnum,\n ApiRateLimitCostEnvVarFormat,\n DEFAULT_API_RATE_LIMIT_COST_CONFIG,\n IApiRateLimitCost,\n} from '@novu/shared';\n\n@Injectable()\nexport class GetApiRateLimitCostConfig {\n public default: IApiRateLimitCost;\n\n constructor() {\n this.loadDefault();\n }\n\n public loadDefault(): void {\n this.default = this.createDefault();\n }\n\n private createDefault(): IApiRateLimitCost {\n const mergedConfig: IApiRateLimitCost = { ...DEFAULT_API_RATE_LIMIT_COST_CONFIG };\n\n // Read process environment only once for performance\n const processEnv = process.env;\n\n Object.values(ApiRateLimitCostEnum).forEach((costOption) => {\n const envVarName = this.getEnvVarName(costOption);\n const envVarValue = processEnv[envVarName];\n\n if (envVarValue) {\n mergedConfig[costOption] = Number(envVarValue);\n }\n });\n\n return mergedConfig;\n }\n\n private getEnvVarName(costOption: ApiRateLimitCostEnum): ApiRateLimitCostEnvVarFormat {\n return `API_RATE_LIMIT_COST_${costOption.toUpperCase() as Uppercase}`;\n }\n}\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/get-api-rate-limit-cost-config/index.ts",
"content": "export * from './get-api-rate-limit-cost-config.usecase';\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/get-api-rate-limit-maximum/get-api-rate-limit-maximum.command.ts",
"content": "import { ApiRateLimitCategoryEnum } from '@novu/shared';\nimport { IsDefined, IsEnum } from 'class-validator';\nimport { EnvironmentCommand } from '../../../shared/commands/project.command';\n\nexport class GetApiRateLimitMaximumCommand extends EnvironmentCommand {\n @IsDefined()\n @IsEnum(ApiRateLimitCategoryEnum)\n apiRateLimitCategory: ApiRateLimitCategoryEnum;\n}\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/get-api-rate-limit-maximum/get-api-rate-limit-maximum.dto.ts",
"content": "import { ApiServiceLevelEnum } from '@novu/shared';\n\nexport const CUSTOM_API_SERVICE_LEVEL = 'custom';\n\nexport type ApiServiceLevel = ApiServiceLevelEnum | typeof CUSTOM_API_SERVICE_LEVEL;\n\n// Array type to keep the cached entity as small as possible for more performant caching\nexport type GetApiRateLimitMaximumDto = [apiRateLimitMaximum: number, apiServiceLevel: ApiServiceLevel];\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/get-api-rate-limit-maximum/get-api-rate-limit-maximum.spec.ts",
"content": "import { Test } from '@nestjs/testing';\nimport { CacheService, MockCacheService } from '@novu/application-generic';\nimport { CommunityOrganizationRepository, EnvironmentRepository } from '@novu/dal';\nimport {\n ApiRateLimitCategoryEnum,\n ApiRateLimitCategoryToFeatureName,\n ApiServiceLevelEnum,\n FeatureFlagsKeysEnum,\n getFeatureForTierAsNumber,\n} from '@novu/shared';\nimport { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\nimport sinon from 'sinon';\nimport { SharedModule } from '../../../shared/shared.module';\nimport { RateLimitingModule } from '../../rate-limiting.module';\nimport { CUSTOM_API_SERVICE_LEVEL } from './get-api-rate-limit-maximum.dto';\nimport { GetApiRateLimitMaximum, GetApiRateLimitMaximumCommand } from './index';\n\nconst mockDefaultApiRateLimits = {\n [ApiServiceLevelEnum.FREE]: {\n [ApiRateLimitCategoryEnum.GLOBAL]: 60,\n [ApiRateLimitCategoryEnum.TRIGGER]: 60,\n [ApiRateLimitCategoryEnum.CONFIGURATION]: 60,\n },\n [ApiServiceLevelEnum.UNLIMITED]: {\n [ApiRateLimitCategoryEnum.GLOBAL]: 600,\n [ApiRateLimitCategoryEnum.TRIGGER]: 600,\n [ApiRateLimitCategoryEnum.CONFIGURATION]: 600,\n },\n};\n\ndescribe('GetApiRateLimitMaximum', async () => {\n let useCase: GetApiRateLimitMaximum;\n let session: UserSession;\n let organizationRepository: CommunityOrganizationRepository;\n let environmentRepository: EnvironmentRepository;\n\n let findOneEnvironmentStub: sinon.SinonStub;\n let findOneOrganizationStub: sinon.SinonStub;\n\n beforeEach(async () => {\n const moduleRef = await Test.createTestingModule({\n imports: [SharedModule, RateLimitingModule],\n providers: [],\n })\n .overrideProvider(CacheService)\n .useValue(MockCacheService.createClient())\n .compile();\n await moduleRef.init(); // Trigger OnModuleInit\n\n session = new UserSession();\n await session.initialize();\n\n useCase = moduleRef.get(GetApiRateLimitMaximum);\n organizationRepository = moduleRef.get(CommunityOrganizationRepository);\n environmentRepository = moduleRef.get(EnvironmentRepository);\n\n findOneEnvironmentStub = sinon.stub(environmentRepository, 'findOne');\n findOneOrganizationStub = sinon.stub(organizationRepository, 'findById');\n });\n\n afterEach(() => {\n findOneEnvironmentStub.restore();\n findOneOrganizationStub.restore();\n });\n\n it('should throw error when environment is not found', async () => {\n findOneEnvironmentStub.resolves(undefined);\n\n try {\n await useCase.execute(\n GetApiRateLimitMaximumCommand.create({\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n apiRateLimitCategory: ApiRateLimitCategoryEnum.GLOBAL,\n })\n );\n throw new Error('Should not reach here');\n } catch (e) {\n expect(e.message).to.equal(`Environment id: ${session.environment._id} not found`);\n }\n });\n\n describe('Environment DOES have rate limits specified', () => {\n const mockGlobalLimit = 65;\n const mockApiRateLimitCategory = ApiRateLimitCategoryEnum.GLOBAL;\n\n beforeEach(() => {\n findOneEnvironmentStub.resolves({\n apiRateLimits: {\n [mockApiRateLimitCategory]: mockGlobalLimit,\n },\n });\n });\n\n it('should return api rate limit for the category set on environment', async () => {\n const [rateLimit] = await useCase.execute(\n GetApiRateLimitMaximumCommand.create({\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n apiRateLimitCategory: mockApiRateLimitCategory,\n })\n );\n\n expect(rateLimit).to.equal(mockGlobalLimit);\n });\n\n it('should return api service level of CUSTOM', async () => {\n const [, apiServiceLevel] = await useCase.execute(\n GetApiRateLimitMaximumCommand.create({\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n apiRateLimitCategory: mockApiRateLimitCategory,\n })\n );\n\n expect(apiServiceLevel).to.equal(CUSTOM_API_SERVICE_LEVEL);\n });\n });\n\n describe('Environment DOES NOT have rate limits specified', () => {\n const mockApiRateLimitCategory = ApiRateLimitCategoryEnum.GLOBAL;\n\n beforeEach(() => {\n findOneEnvironmentStub.resolves({\n apiRateLimits: undefined,\n });\n });\n\n describe('Organization DOES have api service level specified', () => {\n const mockApiServiceLevel = ApiServiceLevelEnum.FREE;\n\n beforeEach(() => {\n findOneOrganizationStub.resolves({\n apiServiceLevel: mockApiServiceLevel,\n });\n });\n\n it('should return default api rate limit for the organizations apiServiceLevel when apiServiceLevel IS set on organization', async () => {\n const defaultApiRateLimit = getFeatureForTierAsNumber(\n ApiRateLimitCategoryToFeatureName[mockApiRateLimitCategory],\n mockApiServiceLevel,\n false\n );\n const [rateLimit] = await useCase.execute(\n GetApiRateLimitMaximumCommand.create({\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n apiRateLimitCategory: mockApiRateLimitCategory,\n })\n );\n\n expect(rateLimit).to.equal(defaultApiRateLimit);\n });\n\n it('should return the api service level set on organization when apiServiceLevel IS set on organization', async () => {\n const [, apiServiceLevel] = await useCase.execute(\n GetApiRateLimitMaximumCommand.create({\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n apiRateLimitCategory: mockApiRateLimitCategory,\n })\n );\n\n expect(apiServiceLevel).to.equal(mockApiServiceLevel);\n });\n });\n\n describe('Organization DOES NOT have api service level specified', () => {\n beforeEach(() => {\n findOneOrganizationStub.resolves({\n apiServiceLevel: undefined,\n });\n });\n\n it('should return default api rate limit for the UNLIMITED service level when apiServiceLevel IS NOT set on organization', async () => {\n const defaultApiRateLimit = getFeatureForTierAsNumber(\n ApiRateLimitCategoryToFeatureName[mockApiRateLimitCategory],\n ApiServiceLevelEnum.UNLIMITED,\n false\n );\n\n const [rateLimit] = await useCase.execute(\n GetApiRateLimitMaximumCommand.create({\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n apiRateLimitCategory: mockApiRateLimitCategory,\n })\n );\n\n expect(rateLimit).to.equal(defaultApiRateLimit);\n });\n\n it('should return the default api service level of UNLIMITED when apiServiceLevel IS NOT set on organization', async () => {\n const defaultApiServiceLevel = ApiServiceLevelEnum.UNLIMITED;\n\n const [, apiServiceLevel] = await useCase.execute(\n GetApiRateLimitMaximumCommand.create({\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n apiRateLimitCategory: mockApiRateLimitCategory,\n })\n );\n\n expect(apiServiceLevel).to.equal(defaultApiServiceLevel);\n });\n });\n\n it('should throw an error when the organization is not found', async () => {\n findOneOrganizationStub.resolves(undefined);\n\n try {\n await useCase.execute(\n GetApiRateLimitMaximumCommand.create({\n organizationId: session.organization._id,\n environmentId: session.environment._id,\n apiRateLimitCategory: mockApiRateLimitCategory,\n })\n );\n throw new Error('Should not reach here');\n } catch (e) {\n expect(e.message).to.equal(`Organization id: ${session.organization._id} not found`);\n }\n });\n });\n});\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/get-api-rate-limit-maximum/get-api-rate-limit-maximum.usecase.ts",
"content": "import { Injectable, InternalServerErrorException, OnModuleInit } from '@nestjs/common';\nimport {\n buildMaximumApiRateLimitKey,\n CachedResponse,\n Instrument,\n InstrumentUsecase,\n PinoLogger,\n} from '@novu/application-generic';\nimport { CommunityOrganizationRepository, EnvironmentRepository } from '@novu/dal';\nimport {\n ApiRateLimitCategoryEnum,\n ApiRateLimitCategoryToFeatureName,\n ApiRateLimitServiceMaximumEnvVarFormat,\n ApiServiceLevelEnum,\n getFeatureForTierAsNumber,\n IApiRateLimitServiceMaximum,\n} from '@novu/shared';\nimport { GetApiRateLimitMaximumCommand } from './get-api-rate-limit-maximum.command';\nimport { CUSTOM_API_SERVICE_LEVEL, GetApiRateLimitMaximumDto } from './get-api-rate-limit-maximum.dto';\n\n@Injectable()\nexport class GetApiRateLimitMaximum implements OnModuleInit {\n private apiRateLimitRecord: IApiRateLimitServiceMaximum;\n constructor(\n private environmentRepository: EnvironmentRepository,\n private organizationRepository: CommunityOrganizationRepository,\n private logger: PinoLogger\n ) {\n this.logger.setContext(this.constructor.name);\n }\n\n onModuleInit() {\n this.apiRateLimitRecord = this.buildApiRateLimitRecord();\n }\n\n @InstrumentUsecase()\n async execute(command: GetApiRateLimitMaximumCommand): Promise {\n return await this.getApiRateLimit({\n apiRateLimitCategory: command.apiRateLimitCategory,\n _environmentId: command.environmentId,\n _organizationId: command.organizationId,\n });\n }\n\n @CachedResponse({\n builder: (command: { apiRateLimitCategory: ApiRateLimitCategoryEnum; _environmentId: string }) =>\n buildMaximumApiRateLimitKey({\n _environmentId: command._environmentId,\n apiRateLimitCategory: command.apiRateLimitCategory,\n }),\n })\n private async getApiRateLimit({\n apiRateLimitCategory,\n _environmentId,\n _organizationId,\n }: {\n apiRateLimitCategory: ApiRateLimitCategoryEnum;\n _environmentId: string;\n _organizationId: string;\n }): Promise {\n const environment = await this.getEnvironment(_environmentId);\n\n if (environment.apiRateLimits) {\n return [environment.apiRateLimits[apiRateLimitCategory], CUSTOM_API_SERVICE_LEVEL];\n }\n const apiServiceLevel = await this.getOrganizationApiServiceLevel(_organizationId);\n const apiRateLimitRecord = this.apiRateLimitRecord[apiServiceLevel];\n\n return [apiRateLimitRecord[apiRateLimitCategory], apiServiceLevel];\n }\n\n private async getOrganizationApiServiceLevel(_organizationId: string): Promise {\n const organization = await this.organizationRepository.findById(_organizationId, '_id apiServiceLevel');\n\n if (!organization) {\n const message = `Organization id: ${_organizationId} not found`;\n this.logger.error(message);\n throw new InternalServerErrorException(message);\n }\n\n if (organization.apiServiceLevel) {\n return organization.apiServiceLevel;\n }\n\n return ApiServiceLevelEnum.UNLIMITED;\n }\n\n private async getEnvironment(_environmentId: string) {\n const environment = await this.environmentRepository.findOne({ _id: _environmentId }, '_id apiRateLimits', {\n readPreference: 'secondaryPreferred',\n });\n\n if (!environment) {\n const message = `Environment id: ${_environmentId} not found`;\n this.logger.error(message);\n throw new InternalServerErrorException(message);\n }\n\n return environment;\n }\n @Instrument()\n private buildApiRateLimitRecord(): IApiRateLimitServiceMaximum {\n // Read process environment only once for performance\n const processEnv = process.env;\n\n return Object.values(ApiServiceLevelEnum).reduce((acc, apiServiceLevel) => {\n acc[apiServiceLevel] = Object.values(ApiRateLimitCategoryEnum).reduce(\n (categoryAcc, apiRateLimitCategory) => {\n const featureName = ApiRateLimitCategoryToFeatureName[apiRateLimitCategory];\n const featureForTierAsNumber = getFeatureForTierAsNumber(featureName, apiServiceLevel);\n const envVarName = this.getEnvVarName(apiServiceLevel, apiRateLimitCategory);\n const envVarValue = processEnv[envVarName];\n\n categoryAcc[apiRateLimitCategory] = envVarValue ? Number(envVarValue) : featureForTierAsNumber;\n\n return categoryAcc;\n },\n {} as Record\n );\n\n return acc;\n }, {} as IApiRateLimitServiceMaximum);\n }\n\n private getEnvVarName(\n apiServiceLevel: ApiServiceLevelEnum,\n apiRateLimitCategory: ApiRateLimitCategoryEnum\n ): ApiRateLimitServiceMaximumEnvVarFormat {\n return `API_RATE_LIMIT_MAXIMUM_${apiServiceLevel.toUpperCase() as Uppercase}_${\n apiRateLimitCategory.toUpperCase() as Uppercase\n }`;\n }\n}\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/get-api-rate-limit-maximum/index.ts",
"content": "export * from './get-api-rate-limit-maximum.command';\nexport * from './get-api-rate-limit-maximum.usecase';\n"
},
{
"path": "apps/api/src/app/rate-limiting/usecases/index.ts",
"content": "import { EvaluateApiRateLimit } from './evaluate-api-rate-limit';\nimport { EvaluateTokenBucketRateLimit } from './evaluate-token-bucket-rate-limit';\nimport { GetApiRateLimitAlgorithmConfig } from './get-api-rate-limit-algorithm-config';\nimport { GetApiRateLimitCostConfig } from './get-api-rate-limit-cost-config';\nimport { GetApiRateLimitMaximum } from './get-api-rate-limit-maximum';\n\nexport const USE_CASES = [\n //\n GetApiRateLimitMaximum,\n GetApiRateLimitAlgorithmConfig,\n GetApiRateLimitCostConfig,\n EvaluateApiRateLimit,\n EvaluateTokenBucketRateLimit,\n];\n"
},
{
"path": "apps/api/src/app/shared/commands/authenticated.command.ts",
"content": "import { BaseCommand } from '@novu/application-generic';\nimport { IsNotEmpty } from 'class-validator';\n\nexport abstract class AuthenticatedCommand extends BaseCommand {\n @IsNotEmpty()\n public readonly userId: string;\n}\n"
},
{
"path": "apps/api/src/app/shared/commands/organization.command.ts",
"content": "import { IsNotEmpty } from 'class-validator';\nimport { AuthenticatedCommand } from './authenticated.command';\n\nexport abstract class OrganizationCommand extends AuthenticatedCommand {\n @IsNotEmpty()\n readonly organizationId: string;\n}\n"
},
{
"path": "apps/api/src/app/shared/commands/project.command.ts",
"content": "import { BaseCommand } from '@novu/application-generic';\nimport { IsArray, IsNotEmpty, IsOptional, IsString } from 'class-validator';\n\nexport abstract class EnvironmentCommand extends BaseCommand {\n @IsNotEmpty()\n readonly environmentId: string;\n\n @IsNotEmpty()\n readonly organizationId: string;\n}\n\nexport abstract class EnvironmentWithUserCommand extends EnvironmentCommand {\n @IsNotEmpty()\n readonly userId: string;\n}\n\nexport abstract class EnvironmentWithSubscriber extends EnvironmentCommand {\n @IsNotEmpty()\n readonly environmentId: string;\n\n @IsNotEmpty()\n readonly organizationId: string;\n\n @IsNotEmpty()\n readonly subscriberId: string;\n\n @IsOptional()\n @IsArray()\n @IsString({ each: true })\n readonly contextKeys?: string[];\n}\n"
},
{
"path": "apps/api/src/app/shared/constants.ts",
"content": "export const TRANSLATIONS_SERVICE = 'TRANSLATIONS_SERVICE';\nexport const MANAGE_TRANSLATIONS = 'MANAGE_TRANSLATIONS';\n"
},
{
"path": "apps/api/src/app/shared/dtos/api-key.ts",
"content": "import { ApiProperty } from '@nestjs/swagger';\n\nexport class ApiKey {\n @ApiProperty()\n key: string;\n @ApiProperty()\n _userId: string;\n}\n"
},
{
"path": "apps/api/src/app/shared/dtos/base-responses.ts",
"content": "import { ApiProperty } from '@nestjs/swagger';\n\nexport enum DirectionEnum {\n ASC = 'ASC',\n DESC = 'DESC',\n}\n\nexport class ResponseError {\n @ApiProperty({\n description: 'The error code or identifier.',\n type: String,\n })\n error: string;\n\n @ApiProperty({\n description: 'Detailed error message.',\n type: String,\n })\n message: string;\n\n @ApiProperty({\n description: 'HTTP status code associated with the error.',\n type: Number,\n })\n statusCode: number;\n}\n\nexport class PaginatedResponse {\n @ApiProperty({\n description: 'Array of data items of type T.',\n type: 'array', // Use 'array' instead of Array\n items: { type: 'object' }, // Define the type of items in the array\n })\n data: T[];\n\n @ApiProperty({\n description: 'Indicates if there are more items available.',\n type: Boolean,\n })\n hasMore: boolean;\n\n @ApiProperty({\n description: 'Total number of items available.',\n type: Number,\n })\n totalCount: number;\n\n @ApiProperty({\n description: 'Number of items per page.',\n type: Number,\n })\n pageSize: number;\n\n @ApiProperty({\n description: 'Current page number.',\n type: Number,\n })\n page: number;\n}\n\nexport type KeysOfT = keyof T;\n\nexport class CursorPaginationQueryDto {\n @ApiProperty({\n description: 'Maximum number of items to return.',\n type: Number,\n })\n limit?: number;\n\n @ApiProperty({\n description: 'Cursor for pagination, used to fetch the next set of results.',\n type: String,\n })\n cursor?: string;\n\n @ApiProperty({\n description: 'Direction for ordering results.',\n enum: DirectionEnum,\n })\n orderDirection?: DirectionEnum;\n\n @ApiProperty({\n description: 'Field by which to order the results.',\n type: String,\n })\n orderBy?: K;\n}\n\nexport class LimitOffsetPaginationDto> {\n @ApiProperty({\n description: 'Maximum number of items to return.',\n type: String,\n })\n limit: string;\n\n @ApiProperty({\n description: 'Number of items to skip before starting to collect the result set.',\n type: String,\n })\n offset: string;\n\n @ApiProperty({\n description: 'Direction for ordering results.',\n enum: DirectionEnum,\n })\n orderDirection?: DirectionEnum;\n\n @ApiProperty({\n description: 'Field by which to order the results.',\n type: String,\n })\n orderBy?: K;\n}\n\nexport class PaginationParams {\n @ApiProperty({\n description: 'Current page number.',\n type: Number,\n })\n page: number;\n\n @ApiProperty({\n description: 'Number of items per page.',\n type: Number,\n })\n limit: number;\n}\n\nexport class PaginationWithQueryParams extends PaginationParams {\n @ApiProperty({\n description: 'Optional search query string.',\n type: String,\n required: false,\n })\n query?: string;\n}\n\nexport enum OrderDirectionEnum {\n ASC = 1,\n DESC = -1,\n}\n\nexport enum OrderByEnum {\n ASC = 'ASC',\n DESC = 'DESC',\n}\n"
},
{
"path": "apps/api/src/app/shared/dtos/base-subscriber-fields.dto.ts",
"content": "import { ApiPropertyOptional } from '@nestjs/swagger';\nimport { SubscriberCustomData } from '@novu/shared';\nimport { IsEmail, IsLocale, IsObject, IsOptional, IsString, IsTimeZone, ValidateIf } from 'class-validator';\n\nexport class BaseSubscriberFieldsDto {\n @ApiPropertyOptional({\n description: 'First name of the subscriber',\n example: 'John',\n nullable: true,\n type: String,\n })\n @IsOptional()\n @ValidateIf((obj) => obj.firstName !== null)\n @IsString()\n firstName?: string | null;\n\n @ApiPropertyOptional({\n description: 'Last name of the subscriber',\n example: 'Doe',\n nullable: true,\n type: String,\n })\n @IsOptional()\n @ValidateIf((obj) => obj.lastName !== null)\n @IsString()\n lastName?: string | null;\n\n @ApiPropertyOptional({\n description: 'Email address of the subscriber',\n example: 'john.doe@example.com',\n nullable: true,\n type: String,\n })\n @IsOptional()\n @ValidateIf((obj) => obj.email !== null)\n @IsEmail()\n email?: string | null;\n\n @ApiPropertyOptional({\n description: 'Phone number of the subscriber',\n example: '+1234567890',\n nullable: true,\n type: String,\n })\n @IsOptional()\n @ValidateIf((obj) => obj.phone !== null)\n @IsString()\n phone?: string | null;\n\n @ApiPropertyOptional({\n description: 'Avatar URL or identifier',\n example: 'https://example.com/avatar.jpg',\n nullable: true,\n type: String,\n })\n @IsOptional()\n @ValidateIf((obj) => obj.avatar !== null)\n @IsString()\n avatar?: string | null;\n\n @ApiPropertyOptional({\n description: 'Locale of the subscriber',\n example: 'en-US',\n nullable: true,\n type: String,\n })\n @IsOptional()\n @ValidateIf((obj) => obj.locale !== null)\n @IsLocale()\n locale?: string | null;\n\n @ApiPropertyOptional({\n description: 'Timezone of the subscriber',\n example: 'America/New_York',\n nullable: true,\n type: String,\n })\n @IsOptional()\n @ValidateIf((obj) => obj.timezone !== null)\n @IsTimeZone()\n timezone?: string | null;\n\n @ApiPropertyOptional({\n type: Object,\n description: 'Additional custom data associated with the subscriber',\n nullable: true,\n additionalProperties: true,\n })\n @IsOptional()\n @ValidateIf((obj) => obj.data !== null)\n @IsObject()\n data?: SubscriberCustomData | null;\n}\n"
},
{
"path": "apps/api/src/app/shared/dtos/channel-preference.ts",
"content": "import { ApiProperty } from '@nestjs/swagger';\nimport { ChannelTypeEnum } from '@novu/shared';\nimport { IsBoolean, IsDefined, IsEnum } from 'class-validator';\n\nexport class ChannelPreference {\n @ApiProperty({\n enum: [...Object.values(ChannelTypeEnum)],\n enumName: 'ChannelTypeEnum',\n description: 'The type of channel that is enabled or not',\n })\n @IsDefined()\n @IsEnum(ChannelTypeEnum)\n type: ChannelTypeEnum;\n\n @ApiProperty({\n type: Boolean,\n description: 'If channel is enabled or not',\n })\n @IsBoolean()\n @IsDefined()\n enabled: boolean;\n}\n"
},
{
"path": "apps/api/src/app/shared/dtos/cursor-paginated-response.ts",
"content": "import { mixin } from '@nestjs/common';\nimport { ApiProperty, ApiPropertyOptions } from '@nestjs/swagger';\nimport { Type } from 'class-transformer';\nimport { ValidateNested } from 'class-validator';\n\ntype Constructor = new (...args: any[]) => T;\n\nexport function withCursorPagination(Base: TBase, options?: ApiPropertyOptions | undefined) {\n class ResponseDTO {\n @ApiProperty({\n isArray: true,\n type: Base,\n ...options,\n })\n @Type(() => Base)\n @ValidateNested({ each: true })\n data!: Array>;\n\n @ApiProperty({\n description: 'The cursor for the next page of results, or null if there are no more pages.',\n type: String,\n nullable: true,\n })\n next: string | null;\n\n @ApiProperty({\n description: 'The cursor for the previous page of results, or null if this is the first page.',\n type: String,\n nullable: true,\n })\n previous: string | null;\n\n @ApiProperty({\n description: 'The total count of items (up to 50,000)',\n type: Number,\n })\n totalCount: number;\n\n @ApiProperty({\n description: 'Whether there are more than 50,000 results available',\n type: Boolean,\n })\n totalCountCapped: boolean;\n }\n\n return mixin(ResponseDTO); // This is important otherwise you will get always the same instance\n}\n"
},
{
"path": "apps/api/src/app/shared/dtos/cursor-pagination-request.ts",
"content": "import { ApiPropertyOptional } from '@nestjs/swagger';\nimport { Type } from 'class-transformer';\nimport { IsInt, IsMongoId, IsOptional, Max, Min } from 'class-validator';\n\nimport type { Constructor, CursorPaginationParams } from '../types';\n\nexport function CursorPaginationRequestDto(defaultLimit = 10, maxLimit = 100): Constructor {\n class CursorPaginationRequest {\n @ApiPropertyOptional({\n type: Number,\n required: false,\n default: defaultLimit,\n maximum: maxLimit,\n example: 10,\n })\n @IsOptional()\n @Type(() => Number)\n @IsInt()\n @Min(1)\n @Max(maxLimit)\n limit = defaultLimit;\n\n @ApiPropertyOptional()\n @IsOptional()\n @IsMongoId({\n message: 'The after cursor must be a valid MongoDB ObjectId',\n })\n after?: string;\n\n @ApiPropertyOptional({\n type: Number,\n example: 0,\n })\n @IsOptional()\n @Type(() => Number)\n @IsInt()\n @Min(0)\n offset = 0;\n }\n\n return CursorPaginationRequest;\n}\n"
},
{
"path": "apps/api/src/app/shared/dtos/data-wrapper-dto.ts",
"content": "import { ApiProperty } from '@nestjs/swagger';\n\nexport class DataWrapperDto {\n @ApiProperty()\n data: T;\n}\n\nexport class DataBooleanDto {\n @ApiProperty()\n data: boolean;\n}\n\nexport class DataNumberDto {\n @ApiProperty()\n data: number;\n}\n"
},
{
"path": "apps/api/src/app/shared/dtos/limit-offset-pagination.dto.ts",
"content": "import { ApiProperty, ApiPropertyOptional } from '@nestjs/swagger';\nimport { Transform } from 'class-transformer';\nimport { IsEnum, IsInt, IsNumber, IsOptional, IsString, Min } from 'class-validator';\n\n// Enum for sorting direction\nexport enum DirectionEnum {\n ASC = 'ASC',\n DESC = 'DESC',\n}\n\nexport function LimitOffsetPaginationQueryDto(\n BaseClass: new (...args: any[]) => T,\n allowedFields: K[]\n): new () => {\n limit?: number;\n offset?: number;\n orderDirection?: DirectionEnum;\n orderBy?: K;\n} {\n class PaginationDto {\n @ApiProperty({\n description: 'Number of items to return per page',\n type: 'number',\n required: false,\n example: 10,\n })\n @Transform(({ value }) => {\n // Convert to number, handle different input types\n const parsed = Number(value);\n\n return !Number.isNaN(parsed) ? parsed : undefined;\n })\n @IsNumber()\n @IsInt()\n @Min(1) // Optional: ensure minimum limit\n @IsOptional()\n limit?: number;\n\n @ApiProperty({\n description: 'Number of items to skip before starting to return results',\n type: 'number',\n required: false,\n example: 0,\n })\n @Transform(({ value }) => {\n // Convert to number, handle different input types\n const parsed = Number(value);\n\n return !Number.isNaN(parsed) ? parsed : undefined;\n })\n @IsInt()\n @IsNumber()\n @Min(0) // Ensure non-negative offset\n @IsOptional()\n offset?: number;\n\n @ApiPropertyOptional({\n description: 'Direction of sorting',\n enum: DirectionEnum,\n enumName: 'DirectionEnum',\n required: false,\n })\n @IsOptional()\n @IsEnum(DirectionEnum)\n orderDirection?: DirectionEnum;\n\n @ApiPropertyOptional({\n description: 'Field to sort the results by',\n enum: allowedFields,\n enumName: `${BaseClass.name}SortField`,\n type: 'string',\n required: false,\n })\n @IsOptional()\n @IsString()\n @IsEnum(Object.fromEntries(allowedFields.map((field) => [field, field])))\n orderBy?: K;\n }\n\n return PaginationDto;\n}\n"
},
{
"path": "apps/api/src/app/shared/dtos/message-template.ts",
"content": "import {\n ActorTypeEnum,\n IActor,\n IEmailBlock,\n IMessageCTA,\n ITemplateVariable,\n MessageTemplateContentType,\n StepTypeEnum,\n} from '@novu/shared';\nimport { IsDefined, IsEnum, IsOptional, IsString, ValidateNested } from 'class-validator';\n\nexport class MessageTemplate {\n @IsOptional()\n @IsEnum(StepTypeEnum)\n type: StepTypeEnum;\n\n @IsOptional()\n variables?: ITemplateVariable[];\n\n @IsDefined()\n content: string | IEmailBlock[];\n\n @IsOptional()\n contentType?: MessageTemplateContentType;\n\n @IsOptional()\n @ValidateNested()\n cta?: IMessageCTA;\n\n @IsOptional()\n @IsString()\n feedId?: string;\n\n @IsOptional()\n layoutId?: string | null;\n\n @IsOptional()\n @IsString()\n name?: string;\n\n @IsOptional()\n @IsString()\n subject?: string;\n\n @IsOptional()\n @IsString()\n title?: string;\n\n @IsOptional()\n @IsString()\n preheader?: string;\n\n @IsOptional()\n @IsString()\n senderName?: string;\n\n @IsOptional()\n actor?: IActor;\n\n @IsOptional()\n _creatorId?: string;\n}\n"
},
{
"path": "apps/api/src/app/shared/dtos/message.template.dto.ts",
"content": "import {\n ActorTypeEnum,\n IEmailBlock,\n IMessageCTADto,\n ITemplateVariable,\n MessageTemplateContentType,\n StepTypeEnum,\n} from '@novu/shared';\n\nexport class MessageTemplateDto {\n type: StepTypeEnum;\n\n content: string | IEmailBlock[];\n\n contentType?: MessageTemplateContentType;\n\n cta?: IMessageCTADto;\n\n actor?: {\n type: ActorTypeEnum;\n data: string | null;\n };\n\n variables?: ITemplateVariable[];\n\n _feedId?: string;\n\n _layoutId?: string | null;\n\n name?: string;\n\n subject?: string;\n\n title?: string;\n\n preheader?: string;\n\n senderName?: string;\n\n _creatorId?: string;\n}\n"
},
{
"path": "apps/api/src/app/shared/dtos/notification-step-dto.ts",
"content": "import { ApiExtraModels, ApiProperty, ApiPropertyOptional, getSchemaPath } from '@nestjs/swagger';\nimport { StepFilterDto } from '@novu/application-generic';\nimport {\n DaysEnum,\n DelayTypeEnum,\n DigestTypeEnum,\n DigestUnitEnum,\n IDelayRegularMetadata,\n IDelayScheduledMetadata,\n IDigestBaseMetadata,\n IDigestRegularMetadata,\n IDigestTimedMetadata,\n ITimedConfig,\n IWorkflowStepMetadata,\n MonthlyTypeEnum,\n OrdinalEnum,\n OrdinalValueEnum,\n StepVariantDto,\n} from '@novu/shared';\nimport { Type } from 'class-transformer';\nimport { IsBoolean, IsString, ValidateNested } from 'class-validator';\nimport { MessageTemplate } from './message-template';\n\nclass TimedConfig implements ITimedConfig {\n @ApiPropertyOptional()\n atTime?: string;\n\n @ApiPropertyOptional({ enum: [...Object.values(DaysEnum)], isArray: true })\n weekDays?: DaysEnum[];\n\n @ApiPropertyOptional()\n monthDays?: number[];\n\n @ApiPropertyOptional({ enum: [...Object.values(OrdinalEnum)] })\n ordinal?: OrdinalEnum;\n\n @ApiPropertyOptional({ enum: [...Object.values(OrdinalValueEnum)] })\n ordinalValue?: OrdinalValueEnum;\n\n @ApiPropertyOptional({ enum: [...Object.values(MonthlyTypeEnum)] })\n monthlyType?: MonthlyTypeEnum;\n}\n\nclass AmountAndUnit {\n @ApiPropertyOptional()\n amount: number;\n\n @ApiPropertyOptional({\n enum: [...Object.values(DigestUnitEnum)],\n })\n unit: DigestUnitEnum;\n}\n\nclass DigestBaseMetadata extends AmountAndUnit implements IDigestBaseMetadata {\n @ApiPropertyOptional()\n digestKey?: string;\n}\n\nclass DigestRegularMetadata extends DigestBaseMetadata implements IDigestRegularMetadata {\n @ApiProperty({ enum: [DigestTypeEnum.REGULAR, DigestTypeEnum.BACKOFF] })\n type: DigestTypeEnum.REGULAR | DigestTypeEnum.BACKOFF;\n\n @ApiPropertyOptional()\n backoff?: boolean;\n\n @ApiPropertyOptional()\n backoffAmount?: number;\n\n @ApiPropertyOptional({\n enum: [...Object.values(DigestUnitEnum)],\n })\n backoffUnit?: DigestUnitEnum;\n\n @ApiPropertyOptional()\n updateMode?: boolean;\n}\n\nclass DigestTimedMetadata extends DigestBaseMetadata implements IDigestTimedMetadata {\n @ApiProperty({\n enum: [DigestTypeEnum.TIMED],\n })\n type: DigestTypeEnum.TIMED;\n\n @ApiPropertyOptional()\n @ValidateNested()\n timed?: TimedConfig;\n}\n\nclass DelayRegularMetadata extends AmountAndUnit implements IDelayRegularMetadata {\n @ApiProperty({\n enum: [DelayTypeEnum.REGULAR],\n })\n type: DelayTypeEnum.REGULAR;\n}\n\nclass DelayScheduledMetadata implements IDelayScheduledMetadata {\n @ApiProperty({\n enum: [DelayTypeEnum.SCHEDULED],\n })\n type: DelayTypeEnum.SCHEDULED;\n\n @ApiProperty()\n delayPath: string;\n}\n\n// Define the ReplyCallback type with OpenAPI annotations\nexport class ReplyCallback {\n @ApiPropertyOptional({\n description: 'Indicates whether the reply callback is active.',\n type: Boolean,\n })\n @IsBoolean()\n active: boolean;\n\n @ApiPropertyOptional({\n description: 'The URL to which replies should be sent.',\n type: String,\n })\n @IsString()\n url: string;\n}\n\n@ApiExtraModels(DigestRegularMetadata, DigestTimedMetadata, DelayRegularMetadata, DelayScheduledMetadata)\nexport class NotificationStepData implements StepVariantDto {\n @ApiPropertyOptional({\n description: 'Unique identifier for the notification step.',\n type: String,\n })\n _id?: string;\n\n @ApiPropertyOptional({\n description: 'Universally unique identifier for the notification step.',\n type: String,\n })\n uuid?: string;\n\n @ApiPropertyOptional({\n description: 'Name of the notification step.',\n type: String,\n })\n name?: string;\n\n @ApiPropertyOptional({\n description: 'ID of the template associated with this notification step.',\n type: String,\n })\n _templateId?: string;\n\n @ApiPropertyOptional({\n description: 'Indicates whether the notification step is active.',\n type: Boolean,\n })\n @IsBoolean()\n active?: boolean;\n\n @ApiPropertyOptional({\n description: 'Determines if the process should stop on failure.',\n type: Boolean,\n })\n shouldStopOnFail?: boolean;\n\n @ApiPropertyOptional({\n description: 'Message template used in this notification step.',\n type: () => MessageTemplate, // Assuming MessageTemplate is a class\n })\n @ValidateNested()\n template?: MessageTemplate;\n\n @ApiPropertyOptional({\n description: 'Filters applied to this notification step.',\n type: [StepFilterDto],\n })\n @ValidateNested({ each: true })\n filters?: StepFilterDto[];\n\n @ApiPropertyOptional({\n description: 'ID of the parent notification step, if applicable.',\n type: String,\n })\n _parentId?: string | null;\n\n @ApiPropertyOptional({\n description: 'Metadata associated with the workflow step. Can vary based on the type of step.',\n oneOf: [\n { $ref: getSchemaPath(DigestRegularMetadata) },\n { $ref: getSchemaPath(DigestTimedMetadata) },\n { $ref: getSchemaPath(DelayRegularMetadata) },\n { $ref: getSchemaPath(DelayScheduledMetadata) },\n ],\n })\n metadata?: IWorkflowStepMetadata;\n\n @ApiPropertyOptional({\n description: 'Callback information for replies, including whether it is active and the callback URL.',\n type: () => ReplyCallback,\n })\n replyCallback?: ReplyCallback;\n}\n\nexport class NotificationStepDto extends NotificationStepData {\n @ApiPropertyOptional({\n type: () => [NotificationStepData], // Specify that this is an array of NotificationStepData\n })\n @ValidateNested({ each: true }) // Validate each nested variant\n @Type(() => NotificationStepData) // Transform to NotificationStepData instances\n variants?: NotificationStepData[];\n}\n"
},
{
"path": "apps/api/src/app/shared/dtos/pagination-request.ts",
"content": "import { ApiPropertyOptional } from '@nestjs/swagger';\nimport { IPaginationParams } from '@novu/shared';\nimport { Type } from 'class-transformer';\nimport { IsInt, Max, Min } from 'class-validator';\n\nimport { Constructor } from '../types';\n\nexport function PaginationRequestDto(defaultLimit = 10, maxLimit = 100): Constructor {\n class PaginationRequest {\n @ApiPropertyOptional({\n type: Number,\n required: false,\n example: 0,\n })\n @Type(() => Number)\n @IsInt()\n page = 0;\n\n @ApiPropertyOptional({\n type: Number,\n required: false,\n default: defaultLimit,\n maximum: maxLimit,\n example: 10,\n })\n @Type(() => Number)\n @IsInt()\n @Min(1)\n @Max(maxLimit)\n limit = defaultLimit;\n }\n\n return PaginationRequest;\n}\n"
},
{
"path": "apps/api/src/app/shared/dtos/pagination-response.ts",
"content": "import { ApiProperty } from '@nestjs/swagger';\nimport { IPaginatedResponseDto } from '@novu/shared';\n\nexport class PaginatedResponseDto implements IPaginatedResponseDto {\n @ApiProperty({\n description: 'The current page of the paginated response',\n })\n page: number;\n\n @ApiProperty({\n description: 'Does the list have more items to fetch',\n })\n hasMore: boolean;\n\n @ApiProperty({\n description: 'Number of items on each page',\n })\n pageSize: number;\n\n @ApiProperty({\n description: 'The list of items matching the query',\n isArray: true,\n type: Object,\n })\n data: T[];\n}\n"
},
{
"path": "apps/api/src/app/shared/dtos/pagination-with-filters-request.ts",
"content": "import { ApiPropertyOptional } from '@nestjs/swagger';\nimport { IPaginationWithQueryParams } from '@novu/shared';\nimport { IsOptional, IsString } from 'class-validator';\n\nimport { Constructor } from '../types';\nimport { PaginationRequestDto } from './pagination-request';\n\nexport function PaginationWithFiltersRequestDto({\n defaultLimit = 10,\n maxLimit = 100,\n queryDescription,\n}: {\n defaultLimit: number;\n maxLimit: number;\n queryDescription: string;\n}): Constructor {\n class PaginationWithFiltersRequest extends PaginationRequestDto(defaultLimit, maxLimit) {\n @ApiPropertyOptional({\n type: String,\n required: false,\n description: `A query string to filter the results. ${queryDescription}`,\n })\n @IsOptional()\n @IsString()\n query?: string;\n }\n\n return PaginationWithFiltersRequest;\n}\n"
},
{
"path": "apps/api/src/app/shared/dtos/preference-channels.ts",
"content": "import { ApiPropertyOptional } from '@nestjs/swagger';\nimport { IsBoolean, IsOptional } from 'class-validator';\n\nexport class SubscriberPreferenceChannels {\n @ApiPropertyOptional({\n type: Boolean,\n description: 'Email channel preference',\n example: true,\n })\n @IsBoolean()\n @IsOptional()\n email?: boolean;\n\n @ApiPropertyOptional({\n type: Boolean,\n description: 'SMS channel preference',\n example: false,\n })\n @IsBoolean()\n @IsOptional()\n sms?: boolean;\n\n @ApiPropertyOptional({\n type: Boolean,\n description: 'In-app channel preference',\n example: true,\n })\n @IsBoolean()\n @IsOptional()\n in_app?: boolean;\n\n @ApiPropertyOptional({\n type: Boolean,\n description: 'Chat channel preference',\n example: false,\n })\n @IsBoolean()\n @IsOptional()\n chat?: boolean;\n\n @ApiPropertyOptional({\n type: Boolean,\n description: 'Push notification channel preference',\n example: true,\n })\n @IsBoolean()\n @IsOptional()\n push?: boolean;\n}\n"
},
{
"path": "apps/api/src/app/shared/dtos/schedule.ts",
"content": "import { ApiProperty, ApiPropertyOptional } from '@nestjs/swagger';\nimport { Type } from 'class-transformer';\nimport { IsBoolean, IsOptional, IsString, ValidateNested } from 'class-validator';\nimport { IsTime12HourFormat } from '../validators/is-time-12-hour-format.validator';\nimport { WeeklyScheduleValidation } from '../validators/weekly-schedule-disabled.validator';\n\nexport class TimeRangeDto {\n @ApiProperty({\n type: String,\n description: 'Start time',\n example: '09:00 AM',\n })\n @IsString()\n @IsTime12HourFormat()\n readonly start: string;\n\n @ApiProperty({\n type: String,\n description: 'End time',\n example: '05:00 PM',\n })\n @IsString()\n @IsTime12HourFormat()\n readonly end: string;\n}\nexport class DayScheduleDto {\n @ApiProperty({\n type: Boolean,\n description: 'Day schedule enabled',\n example: true,\n })\n @IsBoolean()\n readonly isEnabled: boolean;\n\n @ApiPropertyOptional({\n type: [TimeRangeDto],\n description: 'Hours',\n example: [{ start: '09:00 AM', end: '05:00 PM' }],\n })\n @IsOptional()\n @ValidateNested()\n @Type(() => TimeRangeDto)\n readonly hours?: TimeRangeDto[];\n}\n\nexport class WeeklyScheduleDto {\n @ApiPropertyOptional({\n type: DayScheduleDto,\n description: 'Monday schedule',\n example: {\n isEnabled: true,\n hours: [{ start: '09:00 AM', end: '05:00 PM' }],\n },\n })\n @IsOptional()\n @ValidateNested()\n @Type(() => DayScheduleDto)\n readonly monday?: DayScheduleDto;\n\n @ApiPropertyOptional({\n type: DayScheduleDto,\n description: 'Tuesday schedule',\n example: {\n isEnabled: true,\n hours: [{ start: '09:00 AM', end: '05:00 PM' }],\n },\n })\n @IsOptional()\n @ValidateNested()\n @Type(() => DayScheduleDto)\n readonly tuesday?: DayScheduleDto;\n\n @ApiPropertyOptional({\n type: DayScheduleDto,\n description: 'Wednesday schedule',\n example: {\n isEnabled: true,\n hours: [{ start: '09:00 AM', end: '05:00 PM' }],\n },\n })\n @IsOptional()\n @ValidateNested()\n @Type(() => DayScheduleDto)\n readonly wednesday?: DayScheduleDto;\n\n @ApiPropertyOptional({\n type: DayScheduleDto,\n description: 'Thursday schedule',\n example: {\n isEnabled: true,\n hours: [{ start: '09:00 AM', end: '05:00 PM' }],\n },\n })\n @IsOptional()\n @ValidateNested()\n @Type(() => DayScheduleDto)\n readonly thursday?: DayScheduleDto;\n\n @ApiPropertyOptional({\n type: DayScheduleDto,\n description: 'Friday schedule',\n example: {\n isEnabled: true,\n hours: [{ start: '09:00 AM', end: '05:00 PM' }],\n },\n })\n @IsOptional()\n @ValidateNested()\n @Type(() => DayScheduleDto)\n readonly friday?: DayScheduleDto;\n\n @ApiPropertyOptional({\n type: DayScheduleDto,\n description: 'Saturday schedule',\n example: {\n isEnabled: true,\n hours: [{ start: '09:00 AM', end: '05:00 PM' }],\n },\n })\n @IsOptional()\n @ValidateNested()\n @Type(() => DayScheduleDto)\n readonly saturday?: DayScheduleDto;\n\n @ApiPropertyOptional({\n type: DayScheduleDto,\n description: 'Sunday schedule',\n example: {\n isEnabled: true,\n hours: [{ start: '09:00 AM', end: '05:00 PM' }],\n },\n })\n @IsOptional()\n @ValidateNested()\n @Type(() => DayScheduleDto)\n readonly sunday?: DayScheduleDto;\n}\n\nexport class ScheduleDto {\n @ApiProperty({\n type: Boolean,\n description: 'Schedule enabled',\n example: true,\n })\n @IsBoolean()\n readonly isEnabled: boolean;\n\n @ApiPropertyOptional({\n type: WeeklyScheduleDto,\n description: 'Weekly schedule',\n example: {\n monday: {\n isEnabled: true,\n hours: [{ start: '09:00 AM', end: '05:00 PM' }],\n },\n tuesday: {\n isEnabled: true,\n hours: [{ start: '09:00 AM', end: '05:00 PM' }],\n },\n wednesday: {\n isEnabled: true,\n hours: [{ start: '09:00 AM', end: '05:00 PM' }],\n },\n thursday: {\n isEnabled: true,\n hours: [{ start: '09:00 AM', end: '05:00 PM' }],\n },\n friday: {\n isEnabled: true,\n hours: [{ start: '09:00 AM', end: '05:00 PM' }],\n },\n saturday: {\n isEnabled: true,\n hours: [{ start: '09:00 AM', end: '05:00 PM' }],\n },\n sunday: {\n isEnabled: true,\n hours: [{ start: '09:00 AM', end: '05:00 PM' }],\n },\n },\n })\n @IsOptional()\n @ValidateNested()\n @Type(() => WeeklyScheduleDto)\n @WeeklyScheduleValidation()\n readonly weeklySchedule?: WeeklyScheduleDto;\n}\n"
},
{
"path": "apps/api/src/app/shared/dtos/subscription-details-response.dto.ts",
"content": "import { ApiProperty, ApiPropertyOptional } from '@nestjs/swagger';\nimport { Type } from 'class-transformer';\nimport { IsArray, IsOptional, IsString, ValidateNested } from 'class-validator';\nimport { SubscriptionPreferenceDto } from './subscriptions/create-subscriptions-response.dto';\n\nexport class SubscriptionDetailsResponseDto {\n @ApiProperty({\n description: 'The unique identifier of the subscription',\n example: '64f5e95d3d7946d80d0cb679',\n })\n @IsString()\n id: string;\n\n @ApiProperty({\n description: 'The identifier of the subscription',\n example: 'subscription-identifier',\n })\n @IsString()\n identifier?: string;\n\n @ApiPropertyOptional({\n description: 'The name of the subscription',\n example: 'My Subscription',\n })\n @IsString()\n @IsOptional()\n name?: string;\n\n @ApiPropertyOptional({\n description: 'The preferences/rules for the subscription',\n type: [SubscriptionPreferenceDto],\n })\n @IsArray()\n @ValidateNested({ each: true })\n @Type(() => SubscriptionPreferenceDto)\n @IsOptional()\n preferences?: SubscriptionPreferenceDto[];\n\n @ApiPropertyOptional({\n description: 'Context keys that scope this subscription (e.g., tenant:org-a, project:proj-123)',\n example: ['tenant:org-a', 'project:proj-123'],\n type: [String],\n })\n contextKeys?: string[];\n}\n"
},
{
"path": "apps/api/src/app/shared/dtos/subscriptions/create-subscriptions-response.dto.ts",
"content": "import { ApiProperty, ApiPropertyOptional } from '@nestjs/swagger';\nimport { Type } from 'class-transformer';\nimport { IsArray, IsDefined, IsOptional, IsString, ValidateIf, ValidateNested } from 'class-validator';\nimport { RulesLogic } from 'json-logic-js';\nimport { WorkflowDto } from '../../../inbox/dtos/workflow.dto';\n\nexport class TopicDto {\n @ApiProperty({\n description: 'The internal unique identifier of the topic',\n example: '64f5e95d3d7946d80d0cb677',\n })\n @IsString()\n _id: string;\n\n @ApiProperty({\n description: 'The key identifier of the topic used in your application. Should be unique on the environment level.',\n example: 'product-updates',\n })\n @IsString()\n key: string;\n\n @ApiPropertyOptional({\n description: 'The name of the topic',\n example: 'Product Updates',\n })\n @IsString()\n @IsOptional()\n name?: string;\n}\n\nexport class SubscriberDto {\n @ApiProperty({\n description: 'The unique identifier of the subscriber',\n example: '64f5e95d3d7946d80d0cb678',\n })\n @IsString()\n _id: string;\n\n @ApiProperty({\n description: 'The external identifier of the subscriber',\n example: 'external-subscriber-id',\n })\n @IsString()\n subscriberId: string;\n\n @ApiPropertyOptional({\n description: 'The avatar URL of the subscriber',\n example: 'https://example.com/avatar.png',\n })\n @IsString()\n @IsOptional()\n avatar?: string;\n\n @ApiPropertyOptional({\n description: 'The first name of the subscriber',\n example: 'John',\n })\n @IsString()\n @IsOptional()\n firstName?: string;\n\n @ApiPropertyOptional({\n description: 'The last name of the subscriber',\n example: 'Doe',\n })\n @IsString()\n @IsOptional()\n lastName?: string;\n\n @ApiPropertyOptional({\n description: 'The email of the subscriber',\n example: 'john.doe@example.com',\n })\n @IsString()\n @IsOptional()\n email?: string;\n\n @ApiPropertyOptional({\n description: 'The creation date of the subscriber',\n example: '2025-04-24T05:40:21Z',\n })\n @IsString()\n @IsOptional()\n createdAt?: string;\n\n @ApiPropertyOptional({\n description: 'The last update date of the subscriber',\n example: '2025-04-24T05:40:21Z',\n })\n @IsString()\n @IsOptional()\n updatedAt?: string;\n}\n\nexport class SubscriptionPreferenceDto {\n @ApiProperty({\n description: 'The unique identifier of the subscription',\n example: '64f5e95d3d7946d80d0cb679',\n })\n @IsString()\n subscriptionId: string;\n\n @ApiPropertyOptional({\n type: () => WorkflowDto,\n description: 'Workflow information if this is a template-level preference',\n nullable: true,\n })\n @IsOptional()\n @ValidateNested()\n @Type(() => WorkflowDto)\n workflow?: WorkflowDto;\n\n @ApiProperty({\n type: Boolean,\n description: 'Whether the preference is enabled',\n example: true,\n })\n @IsDefined()\n enabled: boolean;\n\n @ApiPropertyOptional({\n description: 'Optional condition using JSON Logic rules',\n required: false,\n type: 'object',\n additionalProperties: true,\n example: { and: [{ '===': [{ var: 'tier' }, 'premium'] }] },\n })\n @ValidateIf((o) => o.condition !== undefined)\n @IsOptional()\n condition?: RulesLogic;\n}\n\nexport class SubscriptionResponseDto {\n @ApiProperty({\n description: 'The unique identifier of the subscription',\n example: '64f5e95d3d7946d80d0cb679',\n })\n @IsString()\n _id: string;\n\n @ApiProperty({\n description: 'The identifier of the subscription',\n example: 'tk=product-updates:si=subscriber-123',\n })\n @IsString()\n @IsOptional()\n identifier?: string;\n\n @ApiPropertyOptional({\n description: 'The name of the subscription',\n example: 'My Subscription',\n })\n @IsString()\n @IsOptional()\n name?: string;\n\n @ApiProperty({\n description: 'The topic information',\n type: () => TopicDto,\n })\n topic: TopicDto;\n\n @ApiProperty({\n description: 'The subscriber information',\n type: () => SubscriberDto,\n nullable: true,\n })\n subscriber: SubscriberDto | null;\n\n @ApiPropertyOptional({\n description: 'The preferences for workflows in this subscription',\n type: () => [SubscriptionPreferenceDto],\n })\n @IsArray()\n @IsOptional()\n preferences?: SubscriptionPreferenceDto[];\n\n @ApiPropertyOptional({\n description: 'Context keys that scope this subscription (e.g., tenant:org-a, project:proj-123)',\n example: ['tenant:org-a', 'project:proj-123'],\n type: [String],\n })\n contextKeys?: string[];\n\n @ApiProperty({\n description: 'The creation date of the subscription',\n example: '2025-04-24T05:40:21Z',\n })\n createdAt: string;\n\n @ApiProperty({\n description: 'The last update date of the subscription',\n example: '2025-04-24T05:40:21Z',\n })\n updatedAt: string;\n}\n\nexport class SubscriptionErrorDto {\n @ApiProperty({\n description: 'The subscriber ID that failed',\n example: 'invalid-subscriber-id',\n })\n subscriberId: string;\n\n @ApiProperty({\n description: 'The error code',\n example: 'SUBSCRIBER_NOT_FOUND',\n })\n code: string;\n\n @ApiProperty({\n description: 'The error message',\n example: 'Subscriber with ID invalid-subscriber-id could not be found',\n })\n message: string;\n}\n\nexport class MetaDto {\n @ApiProperty({\n description: 'The total count of subscriber IDs provided',\n example: 3,\n })\n totalCount: number;\n\n @ApiProperty({\n description: 'The count of successfully created subscriptions',\n example: 2,\n })\n successful: number;\n\n @ApiProperty({\n description: 'The count of failed subscription attempts',\n example: 1,\n })\n failed: number;\n}\n\nexport class CreateSubscriptionsResponseDto {\n @ApiProperty({\n description: 'The list of successfully created subscriptions',\n type: () => [SubscriptionResponseDto],\n })\n data: SubscriptionResponseDto[];\n\n @ApiProperty({\n description: 'Metadata about the operation',\n type: MetaDto,\n })\n meta: MetaDto;\n\n @ApiPropertyOptional({\n description: 'The list of errors for failed subscription attempts',\n type: [SubscriptionErrorDto],\n })\n errors?: SubscriptionErrorDto[];\n}\n"
},
{
"path": "apps/api/src/app/shared/dtos/subscriptions/create-subscriptions.dto.ts",
"content": "import { ApiExtraModels, ApiProperty, ApiPropertyOptional, getSchemaPath } from '@nestjs/swagger';\nimport { Type } from 'class-transformer';\nimport {\n ArrayMaxSize,\n ArrayMinSize,\n IsArray,\n IsDefined,\n IsOptional,\n IsString,\n ValidateIf,\n ValidateNested,\n} from 'class-validator';\nimport { RulesLogic } from 'json-logic-js';\n\nexport class TopicSubscriberIdentifierDto {\n @ApiProperty({\n description: 'Unique identifier for this subscription',\n example: 'subscriber-123-subscription-a',\n })\n @IsString()\n @IsDefined()\n identifier: string;\n\n @ApiProperty({\n description: 'The subscriber ID',\n example: 'subscriber-123',\n })\n @IsString()\n @IsDefined()\n subscriberId: string;\n\n @ApiPropertyOptional({\n description: 'The name of the subscription',\n example: 'My Subscription',\n })\n @IsString()\n @IsOptional()\n name?: string;\n}\n\nexport class BasePreferenceDto {\n @ApiProperty({\n description: 'Whether the preference is enabled. Used when condition is not provided.',\n required: false,\n type: Boolean,\n example: true,\n })\n @IsOptional()\n enabled?: boolean;\n\n @ApiProperty({\n description: 'Optional condition using JSON Logic rules',\n required: false,\n type: 'object',\n additionalProperties: true,\n example: { and: [{ '===': [{ var: 'tier' }, 'premium'] }] },\n })\n @ValidateIf((o) => o.condition !== undefined)\n @IsOptional()\n condition?: RulesLogic;\n}\n\nexport class WorkflowPreferenceRequestDto extends BasePreferenceDto {\n @ApiProperty({\n description: 'The workflow identifier',\n example: 'workflow-123',\n })\n @IsString()\n @IsDefined()\n workflowId: string;\n}\n\nexport class GroupPreferenceFilterDetailsDto {\n @ApiProperty({\n description: 'List of workflow identifiers',\n type: [String],\n example: ['workflow-1', 'workflow-2'],\n })\n @IsArray()\n @IsString({ each: true })\n @IsOptional()\n workflowIds?: string[];\n\n @ApiProperty({\n description: 'List of tags',\n type: [String],\n example: ['tag1', 'tag2'],\n })\n @IsArray()\n @IsString({ each: true })\n @IsOptional()\n tags?: string[];\n}\n\nexport class GroupPreferenceFilterDto extends BasePreferenceDto {\n @ApiProperty({\n description: 'Filter criteria for workflow IDs and tags',\n type: GroupPreferenceFilterDetailsDto,\n })\n @ValidateNested()\n @Type(() => GroupPreferenceFilterDetailsDto)\n @IsDefined()\n filter: GroupPreferenceFilterDetailsDto;\n}\n\n@ApiExtraModels(WorkflowPreferenceRequestDto, GroupPreferenceFilterDto, TopicSubscriberIdentifierDto)\nexport class CreateSubscriptionsRequestDto {\n @ApiProperty({\n description:\n 'List of subscriber IDs to subscribe to the topic (max: 100). @deprecated Use the \"subscriptions\" property instead.',\n type: [String],\n example: ['subscriberId1', 'subscriberId2'],\n deprecated: true,\n })\n @IsArray()\n @IsString({ each: true })\n @IsOptional()\n @ArrayMaxSize(100, { message: 'Cannot subscribe more than 100 subscribers at once' })\n @ArrayMinSize(1, { message: 'At least one subscriber identifier is required' })\n subscriberIds?: string[];\n\n @ApiProperty({\n description:\n 'List of subscriptions to subscribe to the topic (max: 100). Can be either a string array of subscriber IDs or an array of objects with identifier and subscriberId',\n type: 'array',\n items: {\n oneOf: [{ type: 'string' }, { $ref: getSchemaPath(TopicSubscriberIdentifierDto) }],\n },\n example: [\n { identifier: 'subscriber-123-subscription-a', subscriberId: 'subscriber-123' },\n { identifier: 'subscriber-456-subscription-b', subscriberId: 'subscriber-456' },\n ],\n })\n @IsArray()\n @IsOptional()\n @ArrayMaxSize(100, { message: 'Cannot subscribe more than 100 subscriptions at once' })\n @ArrayMinSize(1, { message: 'At least one subscription is required' })\n subscriptions?: Array;\n\n @ApiProperty({\n description: 'The name of the topic',\n example: 'My Topic',\n })\n @IsString()\n @IsOptional()\n name?: string;\n\n @ApiProperty({\n description:\n 'The preferences of the topic. Can be a simple workflow ID string, workflow preference object, or group filter object',\n type: 'array',\n items: {\n oneOf: [\n { type: 'string' },\n { $ref: getSchemaPath(WorkflowPreferenceRequestDto) },\n { $ref: getSchemaPath(GroupPreferenceFilterDto) },\n ],\n },\n example: [{ workflowId: 'workflow-123', condition: { '===': [{ var: 'tier' }, 'premium'] } }],\n })\n @IsArray()\n @IsOptional()\n preferences?: Array;\n}\n"
},
{
"path": "apps/api/src/app/shared/dtos/subscriptions/update-subscription.dto.ts",
"content": "import { ApiExtraModels, ApiProperty, getSchemaPath } from '@nestjs/swagger';\nimport { IsArray, IsOptional, IsString } from 'class-validator';\nimport { GroupPreferenceFilterDto, WorkflowPreferenceRequestDto } from './create-subscriptions.dto';\n\n@ApiExtraModels(WorkflowPreferenceRequestDto, GroupPreferenceFilterDto)\nexport class UpdateSubscriptionRequestDto {\n @ApiProperty({\n description: 'The name of the subscription',\n example: 'My Subscription',\n })\n @IsString()\n @IsOptional()\n name?: string;\n\n @ApiProperty({\n description:\n 'The preferences of the subscription. Can be a simple workflow ID string, workflow preference object, or group filter object',\n type: 'array',\n items: {\n oneOf: [\n { type: 'string' },\n { $ref: getSchemaPath(WorkflowPreferenceRequestDto) },\n { $ref: getSchemaPath(GroupPreferenceFilterDto) },\n ],\n },\n example: [{ workflowId: 'workflow-123', condition: { '===': [{ var: 'tier' }, 'premium'] } }],\n })\n @IsArray()\n @IsOptional()\n preferences?: Array;\n}\n"
},
{
"path": "apps/api/src/app/shared/framework/analytics-logs.guard.ts",
"content": "import { CanActivate, ExecutionContext, Injectable } from '@nestjs/common';\nimport { Reflector } from '@nestjs/core';\n\nconst LOG_ANALYTICS_KEY = 'logAnalytics';\n\n/**\n * Analytics Logs Guard\n *\n * This guard sets the `_shouldLogAnalytics` flag on incoming requests early in the NestJS lifecycle.\n * It runs BEFORE interceptors, ensuring the flag is available even if interceptors throw exceptions.\n *\n * Why use a Guard instead of an Interceptor?\n * - Guards execute before interceptors in the NestJS request lifecycle\n * - If any interceptor throws an exception, subsequent interceptors never run, so the flag\n * cannot be reliably set by an interceptor that might not execute\n * - By setting the flag in a guard, AllExceptionsFilter can always check for analytics logging\n * regardless of which interceptor threw the exception\n * - AllExceptionsFilter cannot access decorator metadata directly since it operates outside\n * the normal request lifecycle and doesn't have access to the original ExecutionContext\n *\n * Example execution order:\n * 1. Guard runs → sets _shouldLogAnalytics = true\n * 2. QuotaThrottlerInterceptor runs → throws exception\n * 3. AllExceptionsFilter runs → finds _shouldLogAnalytics = true → logs analytics\n */\n@Injectable()\nexport class AnalyticsLogsGuard implements CanActivate {\n constructor(private reflector: Reflector) {}\n\n canActivate(context: ExecutionContext): boolean {\n const shouldLogAnalytics = this.shouldLogAnalytics(context);\n\n if (shouldLogAnalytics) {\n const request = context.switchToHttp().getRequest();\n request._shouldLogAnalytics = true;\n }\n\n // Always return true - this guard never blocks requests, it only sets metadata\n return true;\n }\n\n private shouldLogAnalytics(context: ExecutionContext): boolean {\n // Check if @LogAnalytics() decorator is present on the handler or controller\n const handlerMetadata = this.reflector.get(LOG_ANALYTICS_KEY, context.getHandler());\n const classMetadata = this.reflector.get(LOG_ANALYTICS_KEY, context.getClass());\n\n return handlerMetadata !== undefined || classMetadata !== undefined;\n }\n}\n"
},
{
"path": "apps/api/src/app/shared/framework/analytics-logs.interceptor.ts",
"content": "import {\n applyDecorators,\n CallHandler,\n ExecutionContext,\n Injectable,\n NestInterceptor,\n SetMetadata,\n} from '@nestjs/common';\nimport { Reflector } from '@nestjs/core';\nimport { PinoLogger, RequestLog, RequestLogRepository } from '@novu/application-generic';\nimport { UserSessionData } from '@novu/shared';\nimport { Observable } from 'rxjs';\nimport { tap } from 'rxjs/operators';\nimport { TriggerEventResponseDto } from '../../events/dtos/trigger-event-response.dto';\nimport { buildLog } from '../utils/mappers';\n\nconst LOG_ANALYTICS_KEY = 'logAnalytics';\n\nexport enum AnalyticsStrategyEnum {\n BASIC = 'basic',\n EVENTS = 'events',\n EVENTS_BULK = 'events_bulk',\n}\n\nexport function LogAnalytics(strategy: AnalyticsStrategyEnum = AnalyticsStrategyEnum.BASIC): MethodDecorator {\n return applyDecorators(SetMetadata(LOG_ANALYTICS_KEY, strategy));\n}\n\n@Injectable()\nexport class AnalyticsLogsInterceptor implements NestInterceptor {\n constructor(\n private readonly requestLogRepository: RequestLogRepository,\n private readonly logger: PinoLogger,\n private readonly reflector: Reflector\n ) {\n this.logger.setContext(this.constructor.name);\n }\n\n private shouldLogAnalytics(context: ExecutionContext): boolean {\n const strategy = this.getAnalyticsStrategy(context);\n\n this.logger.debug(`Analytics logs should log strategy: ${strategy}`);\n\n return strategy !== undefined;\n }\n\n private getAnalyticsStrategy(context: ExecutionContext): AnalyticsStrategyEnum {\n const globalHandler = context.getHandler && Reflect.getMetadata(LOG_ANALYTICS_KEY, context.getHandler());\n const handlerMetadata = this.reflector.get(LOG_ANALYTICS_KEY, context.getHandler());\n const handler = context.getHandler();\n const customDecorator = handler && (handler as any)._analyticsStrategy;\n\n this.logger.debug(`Analytics logs globalHandler strategy: ${globalHandler}`);\n this.logger.debug(`Analytics logs handlerMetadata strategy: ${handlerMetadata}`);\n this.logger.debug(`Analytics logs customDecorator strategy: ${customDecorator}`);\n\n return globalHandler || handlerMetadata || customDecorator;\n }\n\n async intercept(context: ExecutionContext, next: CallHandler): Promise> {\n const shouldRun = await this.shouldRun(context);\n\n this.logger.debug(`Analytics logs should run LOG_ANALYTICS_KEY: ${shouldRun}`);\n\n if (!shouldRun) {\n return next.handle();\n }\n\n const req = context.switchToHttp().getRequest();\n const user = req.user as UserSessionData;\n const start = Date.now();\n const res = context.switchToHttp().getResponse();\n\n this.logger.debug('Analytics logs interceptor started');\n\n return next.handle().pipe(\n tap(async (data) => {\n const duration = Date.now() - start;\n const basicLog = buildLog(req, res.statusCode, data, user, duration);\n if (!basicLog) {\n this.logger.warn('Analytics log construction failed - unable to track request metrics');\n\n return;\n }\n\n const analyticsLog = this.buildLogByStrategy(context, basicLog, data);\n\n try {\n this.logger.debug({ analyticsLog }, 'Analytics log Inserting');\n await this.requestLogRepository.create(analyticsLog, {\n organizationId: user?.organizationId,\n environmentId: user?.environmentId,\n userId: user?._id,\n });\n this.logger.debug('Analytics log Inserted');\n } catch (err) {\n this.logger.error({ err }, 'Failed to log analytics to ClickHouse after retries');\n }\n })\n );\n }\n\n private async shouldRun(context: ExecutionContext): Promise {\n const shouldLog = this.shouldLogAnalytics(context);\n\n if (!shouldLog) return false;\n\n const isEnabled = process.env.IS_ANALYTICS_LOGS_ENABLED === 'true';\n\n this.logger.debug(\n `Analytics logs should run IS_ANALYTICS_LOGS_ENABLED: ${process.env.IS_ANALYTICS_LOGS_ENABLED}, isEnabled: ${isEnabled}`\n );\n\n if (!isEnabled) return false;\n\n return true;\n }\n\n private buildLogByStrategy(\n context: ExecutionContext,\n analyticsLog: Omit,\n res: unknown\n ): Omit {\n const strategy = this.getAnalyticsStrategy(context);\n\n if (strategy === AnalyticsStrategyEnum.EVENTS) {\n const eventResponse = (res as any).data as TriggerEventResponseDto;\n\n if (eventResponse.transactionId) {\n return {\n ...analyticsLog,\n transaction_id: eventResponse.transactionId,\n };\n }\n }\n\n if (strategy === AnalyticsStrategyEnum.EVENTS_BULK) {\n const bulkEventResponse = (res as any).data as TriggerEventResponseDto[];\n\n if (Array.isArray(bulkEventResponse)) {\n const transactionIds = bulkEventResponse\n .map((response) => response.transactionId)\n .filter(Boolean)\n .join(',');\n\n if (transactionIds) {\n return {\n ...analyticsLog,\n transaction_id: transactionIds,\n };\n }\n }\n }\n\n return analyticsLog;\n }\n}\n"
},
{
"path": "apps/api/src/app/shared/framework/constants/headers.schema.ts",
"content": "import { HeaderObject, HttpResponseHeaderKeysEnum } from '@novu/application-generic';\n\nexport const COMMON_RESPONSE_HEADERS: Array = [\n HttpResponseHeaderKeysEnum.CONTENT_TYPE,\n HttpResponseHeaderKeysEnum.RATELIMIT_LIMIT,\n HttpResponseHeaderKeysEnum.RATELIMIT_REMAINING,\n HttpResponseHeaderKeysEnum.RATELIMIT_RESET,\n HttpResponseHeaderKeysEnum.RATELIMIT_POLICY,\n HttpResponseHeaderKeysEnum.IDEMPOTENCY_KEY,\n HttpResponseHeaderKeysEnum.IDEMPOTENCY_REPLAY,\n];\n\nexport const RESPONSE_HEADER_CONFIG: Record = {\n [HttpResponseHeaderKeysEnum.CONTENT_TYPE]: {\n required: true,\n description: 'The MIME type of the response body.',\n schema: { type: 'string' },\n example: 'application/json',\n },\n [HttpResponseHeaderKeysEnum.RATELIMIT_LIMIT]: {\n required: false,\n description:\n 'The number of requests that the client is permitted to make per second. The actual maximum may differ when burst is enabled.',\n schema: { type: 'string' },\n example: '100',\n },\n [HttpResponseHeaderKeysEnum.RATELIMIT_REMAINING]: {\n required: false,\n description: 'The number of requests remaining until the next window.',\n schema: { type: 'string' },\n example: '93',\n },\n [HttpResponseHeaderKeysEnum.RATELIMIT_RESET]: {\n required: false,\n description: 'The remaining seconds until a request of the same cost will be refreshed.',\n schema: { type: 'string' },\n example: '8',\n },\n [HttpResponseHeaderKeysEnum.RATELIMIT_POLICY]: {\n required: false,\n description: 'The rate limit policy that was used to evaluate the request.',\n schema: { type: 'string' },\n example: '100;w=1;burst=110;comment=\"token bucket\";category=\"trigger\";cost=\"single\"',\n },\n [HttpResponseHeaderKeysEnum.RETRY_AFTER]: {\n required: false,\n description: 'The number of seconds after which the client may retry the request that was previously rejected.',\n schema: { type: 'string' },\n example: '8',\n },\n [HttpResponseHeaderKeysEnum.IDEMPOTENCY_KEY]: {\n required: false,\n description: 'The idempotency key used to evaluate the request.',\n schema: { type: 'string' },\n example: '8',\n },\n [HttpResponseHeaderKeysEnum.IDEMPOTENCY_REPLAY]: {\n required: false,\n description: 'Whether the request was a replay of a previous request.',\n schema: { type: 'string' },\n example: 'true',\n },\n [HttpResponseHeaderKeysEnum.LINK]: {\n required: false,\n description: 'A link to the documentation.',\n schema: { type: 'string' },\n example: 'https://docs.novu.co/',\n },\n};\n"
},
{
"path": "apps/api/src/app/shared/framework/constants/index.ts",
"content": "export * from './headers.schema';\nexport * from './responses.schema';\n"
},
{
"path": "apps/api/src/app/shared/framework/constants/responses.schema.ts",
"content": "import { ApiResponseOptions } from '@nestjs/swagger';\nimport { ApiResponseDecoratorName, HttpResponseHeaderKeysEnum } from '@novu/application-generic';\nimport { THROTTLED_EXCEPTION_MESSAGE } from '../../../rate-limiting/guards';\nimport { createReusableHeaders } from '../swagger';\n\nexport const COMMON_RESPONSES: Partial> = {\n ApiConflictResponse: {\n description: 'The request could not be completed due to a conflict with the current state of the target resource.',\n schema: {\n type: 'string',\n example:\n 'Request with key 3909d656-d4fe-4e80-ba86-90d3861afcd7 is currently being processed. Please retry after 1 second',\n },\n headers: createReusableHeaders([HttpResponseHeaderKeysEnum.RETRY_AFTER, HttpResponseHeaderKeysEnum.LINK]),\n },\n ApiTooManyRequestsResponse: {\n description: 'The client has sent too many requests in a given amount of time. ',\n schema: { type: 'string', example: THROTTLED_EXCEPTION_MESSAGE },\n headers: createReusableHeaders([HttpResponseHeaderKeysEnum.RETRY_AFTER]),\n },\n ApiServiceUnavailableResponse: {\n description:\n 'The server is currently unable to handle the request due to a temporary overload or scheduled maintenance, which will likely be alleviated after some delay.',\n schema: { type: 'string', example: 'Please wait some time, then try again.' },\n headers: createReusableHeaders([HttpResponseHeaderKeysEnum.RETRY_AFTER]),\n },\n};\n"
},
{
"path": "apps/api/src/app/shared/framework/exclude-from-idempotency.ts",
"content": "import { applyDecorators, SetMetadata } from '@nestjs/common';\n\nexport const EXCLUDE_FROM_IDEMPOTENCY = 'exclude_from_idempotency';\n\nexport function ExcludeFromIdempotency() {\n return applyDecorators(SetMetadata(EXCLUDE_FROM_IDEMPOTENCY, true));\n}\n"
},
{
"path": "apps/api/src/app/shared/framework/idempotency.e2e.ts",
"content": "import { CacheService, HttpResponseHeaderKeysEnum } from '@novu/application-generic';\nimport { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\nimport {\n IdempotenceTestingResponse,\n IdempotencyBehaviorEnum,\n IdempotencyTestingDto,\n} from '../../testing/dtos/idempotency.dto';\nimport { expectSdkExceptionGeneric } from '../helpers/e2e/sdk/e2e-sdk.helper';\n\nconst DOCS_LINK = 'https://docs.novu.co/additional-resources/idempotency';\n// @ts-ignore\nprocess.env.LAUNCH_DARKLY_SDK_KEY = ''; // disable Launch Darkly to allow test to define FF state\n\nconst idempotancyKey = HttpResponseHeaderKeysEnum.IDEMPOTENCY_KEY.toLowerCase();\nconst retryAfterHeaderKey = HttpResponseHeaderKeysEnum.RETRY_AFTER.toLowerCase();\nconst IDEMPOTENCE_IMMEDIATE_EXCEPTION = {\n expectedBehavior: IdempotencyBehaviorEnum.IMMEDIATE_EXCEPTION,\n};\nconst IDEMPOTENCE_IMMEDIATE_RESPONSE = {\n expectedBehavior: IdempotencyBehaviorEnum.IMMEDIATE_RESPONSE,\n};\nconst IDEMPOTENCE_DELAYED_RESPONSE = {\n expectedBehavior: IdempotencyBehaviorEnum.DELAYED_RESPONSE,\n};\nconst idempotancyReplayKey = HttpResponseHeaderKeysEnum.IDEMPOTENCY_REPLAY.toLowerCase();\n\ndescribe('Idempotency Test', async () => {\n let session: UserSession;\n const path = '/v1/health-check/test-idempotency';\n let cacheService: CacheService | null = null;\n\n async function testIdempotencyPost(\n idempotencyTestingDto: IdempotencyTestingDto,\n key: string\n ): Promise<{ body: IdempotenceTestingResponse; headers: Record }> {\n const { body, headers } = await session.testAgent\n .post(path)\n .set(HttpResponseHeaderKeysEnum.IDEMPOTENCY_KEY, key)\n .set('authorization', `ApiKey ${session.apiKey}`)\n .send(idempotencyTestingDto);\n\n return { body: body.data, headers };\n }\n async function testIdempotencyGet(\n key: string\n ): Promise<{ body: IdempotenceTestingResponse; headers: Record }> {\n const { body, headers } = await session.testAgent\n .get(path)\n .set(HttpResponseHeaderKeysEnum.IDEMPOTENCY_KEY, key)\n .set('authorization', `ApiKey ${session.apiKey}`)\n .send();\n\n return { body: body.data, headers };\n }\n\n before(async () => {\n session = new UserSession();\n await session.initialize();\n cacheService = session.testServer?.getService(CacheService);\n process.env.IS_API_IDEMPOTENCY_ENABLED = 'true';\n });\n\n it('should return cached same response for duplicate requests', async () => {\n const key = `IdempotencyKey1`;\n const res1 = await testIdempotencyPost(IDEMPOTENCE_IMMEDIATE_RESPONSE, key);\n const res2 = await testIdempotencyPost(IDEMPOTENCE_IMMEDIATE_RESPONSE, key);\n expect(res1.body.number).to.equal(res2.body.number);\n expect(res1.headers[idempotancyKey]).to.eq(key);\n expect(res2.headers[idempotancyKey]).to.eq(key);\n expect(res2.headers[idempotancyReplayKey]).to.eq('true');\n });\n it('should return cached and use correct cache key when apiKey is used', async () => {\n const key = `IdempotencyKey2`;\n const res1 = await testIdempotencyPost(IDEMPOTENCE_IMMEDIATE_RESPONSE, key);\n const cacheKey = `test-${session.organization._id}-${key}`;\n session.testServer?.getHttpServer();\n\n const cacheVal = JSON.stringify(JSON.parse(await cacheService?.get(cacheKey)!).data);\n expect(res1.body.number, cacheVal).to.eq(JSON.parse(cacheVal).data.number);\n const res2 = await testIdempotencyPost(IDEMPOTENCE_IMMEDIATE_RESPONSE, key);\n expect(res1.body.number).to.equal(res2.body.number);\n expect(res1.headers[idempotancyKey]).to.eq(key);\n expect(res2.headers[idempotancyKey]).to.eq(key);\n expect(res2.headers[idempotancyReplayKey]).to.eq('true');\n });\n it('should return cached and use correct cache key when authToken and apiKey combination is used', async () => {\n const key = `3`;\n const res1 = await testIdempotencyPost(IDEMPOTENCE_IMMEDIATE_RESPONSE, key);\n const cacheKey = `test-${session.organization._id}-${key}`;\n session.testServer?.getHttpServer();\n\n const cacheVal = JSON.stringify(JSON.parse(await cacheService?.get(cacheKey)!).data);\n expect(res1.body.number).to.eq(JSON.parse(cacheVal).data.number);\n const res2 = await testIdempotencyPost(IDEMPOTENCE_IMMEDIATE_RESPONSE, key);\n expect(res1.body.number).to.equal(res2.body.number);\n expect(res1.headers[idempotancyKey]).to.eq(key);\n expect(res2.headers[idempotancyKey]).to.eq(key);\n expect(res2.headers[idempotancyReplayKey]).to.eq('true');\n });\n it('should return conflict when concurrent requests are made', async () => {\n const key = `4`;\n const [{ headers, body, status }, { headers: headerDupe, body: bodyDupe, status: statusDupe }] = await Promise.all([\n session.testAgent\n .post(path)\n .set(HttpResponseHeaderKeysEnum.IDEMPOTENCY_KEY, key)\n .send(IDEMPOTENCE_DELAYED_RESPONSE),\n session.testAgent\n .post(path)\n .set(HttpResponseHeaderKeysEnum.IDEMPOTENCY_KEY, key)\n .send(IDEMPOTENCE_DELAYED_RESPONSE),\n ]);\n const oneSuccess = status === 201 || statusDupe === 201;\n const oneConflict = status === 409 || statusDupe === 409;\n const conflictBody = status === 201 ? bodyDupe : body;\n const retryHeader = headers[retryAfterHeaderKey] || headerDupe[retryAfterHeaderKey];\n expect(oneSuccess).to.be.true;\n expect(oneConflict).to.be.true;\n expect(headers[idempotancyKey]).to.eq(key);\n expect(headerDupe[idempotancyKey], JSON.stringify(headerDupe)).to.eq(key);\n expect(headerDupe[HttpResponseHeaderKeysEnum.LINK.toLowerCase()], JSON.stringify(headerDupe)).to.eq(DOCS_LINK);\n expect(retryHeader).to.eq(`1`);\n expect(conflictBody.message).to.eq(\n `Request with key \"${key}\" is currently being processed. Please retry after 1 second`\n );\n expect(conflictBody.error).to.eq('Conflict');\n expect(conflictBody.statusCode).to.eq(409);\n });\n it('should return UnprocessableEntity when different body is sent for same key', async () => {\n const key = '5';\n await testIdempotencyPost(IDEMPOTENCE_IMMEDIATE_RESPONSE, key);\n const { error } = await expectSdkExceptionGeneric(() => testIdempotencyPost(IDEMPOTENCE_IMMEDIATE_EXCEPTION, key));\n expect(error?.statusCode).to.eq(422);\n });\n it('should return non cached response for unique requests', async () => {\n const key = '6';\n const key1 = '7';\n const response = await testIdempotencyPost(IDEMPOTENCE_IMMEDIATE_RESPONSE, key);\n const response2 = await testIdempotencyPost(IDEMPOTENCE_IMMEDIATE_RESPONSE, key1);\n expect(response.body.number).to.not.eq(response2.body.number);\n expect(response.headers[idempotancyKey]).to.eq(key);\n expect(response2.headers[idempotancyKey]).to.eq(key1);\n });\n it('should return non cached response for GET requests', async () => {\n const key = '8';\n const response = await testIdempotencyGet(key);\n const response2 = await testIdempotencyGet(key);\n expect(response.body.number).to.not.eq(response2.body.number);\n });\n it('should return cached error response for duplicate requests', async () => {\n const key = '9';\n const { error } = await expectSdkExceptionGeneric(() => testIdempotencyPost(IDEMPOTENCE_IMMEDIATE_EXCEPTION, key));\n const { error: error2 } = await expectSdkExceptionGeneric(() =>\n testIdempotencyPost(IDEMPOTENCE_IMMEDIATE_EXCEPTION, key)\n );\n expect(error?.message).to.eq(error2?.message);\n });\n it('should return 400 when key bigger than allowed limit', async () => {\n const key = Array.from({ length: 256 })\n .fill(0)\n .map((i) => i)\n .join('');\n const { error } = await expectSdkExceptionGeneric(() => testIdempotencyPost(IDEMPOTENCE_IMMEDIATE_EXCEPTION, key));\n expect(error?.statusCode).to.eq(400);\n expect(error?.message).to.include(`has exceeded`);\n });\n\n describe('Allowed Authentication Security Schemes', () => {\n it('should set Idempotency-Key header when ApiKey security scheme is used to authenticate', async () => {\n const key = '10';\n const { headers } = await testIdempotencyPost(IDEMPOTENCE_IMMEDIATE_RESPONSE, key);\n expect(headers[idempotancyKey]).to.exist;\n });\n\n it('should set rate limit headers when a Bearer security scheme is used to authenticate', async () => {\n const key = '10';\n const { headers } = await testIdempotencyPost(IDEMPOTENCE_IMMEDIATE_RESPONSE, key);\n expect(headers[idempotancyKey]).to.exist;\n });\n\n it('should NOT set rate limit headers when NO authorization header is present', async () => {\n const key = '10';\n const { headers } = await testIdempotencyPost(IDEMPOTENCE_IMMEDIATE_RESPONSE, key);\n expect(headers[idempotancyKey]).not.to.exist;\n });\n });\n});\n"
},
{
"path": "apps/api/src/app/shared/framework/idempotency.interceptor.ts",
"content": "import {\n BadRequestException,\n CallHandler,\n ConflictException,\n ExecutionContext,\n HttpException,\n Injectable,\n InternalServerErrorException,\n NestInterceptor,\n ServiceUnavailableException,\n UnprocessableEntityException,\n} from '@nestjs/common';\nimport { Reflector } from '@nestjs/core';\nimport {\n CacheService,\n FeatureFlagsService,\n HttpResponseHeaderKeysEnum,\n Instrument,\n PinoLogger,\n} from '@novu/application-generic';\nimport { ApiAuthSchemeEnum, FeatureFlagsKeysEnum, UserSessionData } from '@novu/shared';\nimport { createHash } from 'crypto';\nimport { Observable, of, throwError } from 'rxjs';\nimport { catchError, map } from 'rxjs/operators';\nimport { EXCLUDE_FROM_IDEMPOTENCY } from './exclude-from-idempotency';\n\nconst IDEMPOTENCY_CACHE_TTL = 60 * 60 * 24; // 24h\nconst IDEMPOTENCY_PROGRESS_TTL = 60 * 5; // 5min\n\nenum ReqStatusEnum {\n PROGRESS = 'in-progress',\n SUCCESS = 'success',\n ERROR = 'error',\n}\n\nexport const DOCS_LINK = 'https://docs.novu.co/additional-resources/idempotency';\nexport const ALLOWED_AUTH_SCHEMES = [ApiAuthSchemeEnum.API_KEY];\nconst ALLOWED_METHODS = ['post', 'patch'];\n\n@Injectable()\nexport class IdempotencyInterceptor implements NestInterceptor {\n constructor(\n private readonly reflector: Reflector,\n private readonly cacheService: CacheService,\n private featureFlagService: FeatureFlagsService,\n private logger: PinoLogger\n ) {\n this.logger.setContext(this.constructor.name);\n }\n\n protected async isEnabled(context: ExecutionContext): Promise {\n const isExcluded = this.reflector.getAllAndOverride(EXCLUDE_FROM_IDEMPOTENCY, [\n context.getHandler(),\n context.getClass(),\n ]);\n if (isExcluded) {\n return false;\n }\n\n const isAllowedAuthScheme = this.isAllowedAuthScheme(context);\n if (!isAllowedAuthScheme) {\n return true;\n }\n\n const user = this.getReqUser(context);\n const { organizationId, environmentId, _id } = user;\n\n return await this.featureFlagService.getFlag({\n key: FeatureFlagsKeysEnum.IS_API_IDEMPOTENCY_ENABLED,\n defaultValue: false,\n environment: { _id: environmentId },\n organization: { _id: organizationId },\n user: { _id },\n });\n }\n\n @Instrument()\n async intercept(context: ExecutionContext, next: CallHandler): Promise> {\n const request = context.switchToHttp().getRequest();\n const isAllowedMethod = ALLOWED_METHODS.includes(request.method.toLowerCase());\n const idempotencyKey = this.getIdempotencyKey(context);\n const isEnabled = await this.isEnabled(context);\n if (!idempotencyKey || !isAllowedMethod || !isEnabled) {\n return next.handle();\n }\n\n if (idempotencyKey?.length > 255) {\n return throwError(\n () =>\n new BadRequestException(\n `idempotencyKey \"${idempotencyKey}\" has exceeded the maximum allowed length of 255 characters`\n )\n );\n }\n const cacheKey = this.getCacheKey(context);\n\n try {\n const bodyHash = this.hashRequestBody(request.body);\n // if 1st time we are seeing the request, marks the request as in-progress if not, does nothing\n const isNewReq = await this.setCache(\n cacheKey,\n { status: ReqStatusEnum.PROGRESS, bodyHash },\n IDEMPOTENCY_PROGRESS_TTL,\n true\n );\n // Check if the idempotency key is in the cache\n if (isNewReq) {\n return await this.handleNewRequest(context, next, bodyHash);\n } else {\n return await this.handlerDuplicateRequest(context, bodyHash);\n }\n } catch (err) {\n this.logger.warn(\n `An error occurred while making idempotency check, key:${idempotencyKey}. error: ${err.message}`\n );\n if (err instanceof HttpException) {\n return throwError(() => err);\n }\n }\n\n // something unexpected happened, both cached response and handler did not execute as expected\n return throwError(() => new ServiceUnavailableException());\n }\n\n private getIdempotencyKey(context: ExecutionContext): string | undefined {\n const request = context.switchToHttp().getRequest();\n\n return request.headers[HttpResponseHeaderKeysEnum.IDEMPOTENCY_KEY.toLocaleLowerCase()];\n }\n\n private getReqUser(context: ExecutionContext): UserSessionData {\n const req = context.switchToHttp().getRequest();\n\n return req.user;\n }\n\n private isAllowedAuthScheme(context: ExecutionContext): boolean {\n const req = context.switchToHttp().getRequest();\n const { authScheme } = req;\n\n return ALLOWED_AUTH_SCHEMES.some((scheme) => authScheme === scheme);\n }\n\n private getCacheKey(context: ExecutionContext): string {\n const user = this.getReqUser(context);\n if (user === undefined) {\n const message = 'Cannot build idempotency cache key without user';\n this.logger.error(message);\n throw new InternalServerErrorException(message);\n }\n const env = process.env.NODE_ENV;\n\n return `${env}-${user.organizationId}-${this.getIdempotencyKey(context)}`;\n }\n\n async setCache(\n key: string,\n val: { status: ReqStatusEnum; bodyHash: string; data?: any; statusCode?: number },\n ttl: number,\n ifNotExists?: boolean\n ): Promise {\n try {\n if (ifNotExists) {\n return await this.cacheService.setIfNotExist(key, JSON.stringify(val), { ttl });\n }\n await this.cacheService.set(key, JSON.stringify(val), { ttl });\n } catch (err) {\n this.logger.warn(`An error occurred while setting idempotency cache, key:${key} error: ${err.message}`);\n }\n\n return null;\n }\n\n private setHeaders(response: any, headers: Record) {\n Object.keys(headers).forEach((key) => {\n if (headers[key]) {\n response.set(key, headers[key]);\n }\n });\n }\n\n private hashRequestBody(body: object): string {\n const hash = createHash('blake2s256');\n\n try {\n hash.update(Buffer.from(JSON.stringify(body)));\n } catch (error) {\n // For multipart/form-data or other non-serializable bodies,\n // create a hash from the object's string representation\n hash.update(Buffer.from(String(body)));\n }\n\n return hash.digest('hex');\n }\n\n private async handlerDuplicateRequest(context: ExecutionContext, bodyHash: string): Promise> {\n const cacheKey = this.getCacheKey(context);\n const idempotencyKey = this.getIdempotencyKey(context)!;\n const data = await this.cacheService.get(cacheKey);\n this.setHeaders(context.switchToHttp().getResponse(), {\n [HttpResponseHeaderKeysEnum.IDEMPOTENCY_KEY]: idempotencyKey,\n });\n const parsed = JSON.parse(data);\n if (parsed.status === ReqStatusEnum.PROGRESS) {\n // api call is in progress, so client need to handle this case\n this.logger.trace(`previous api call in progress rejecting the request. key: \"${idempotencyKey}\"`);\n this.setHeaders(context.switchToHttp().getResponse(), {\n [HttpResponseHeaderKeysEnum.RETRY_AFTER]: `1`,\n [HttpResponseHeaderKeysEnum.LINK]: DOCS_LINK,\n });\n\n throw new ConflictException(\n `Request with key \"${idempotencyKey}\" is currently being processed. Please retry after 1 second`\n );\n }\n if (bodyHash !== parsed.bodyHash) {\n // different body sent than before\n this.logger.trace(`idempotency key is being reused for different bodies. key: \"${idempotencyKey}\"`);\n this.setHeaders(context.switchToHttp().getResponse(), {\n [HttpResponseHeaderKeysEnum.LINK]: DOCS_LINK,\n });\n\n throw new UnprocessableEntityException(\n `Request with key \"${idempotencyKey}\" is being reused for a different body`\n );\n }\n this.setHeaders(context.switchToHttp().getResponse(), { [HttpResponseHeaderKeysEnum.IDEMPOTENCY_REPLAY]: 'true' });\n\n // already seen the request return cached response\n if (parsed.status === ReqStatusEnum.ERROR) {\n this.logger.trace(`returning cached error response. key: \"${idempotencyKey}\"`);\n\n throw parsed.data;\n }\n\n return of(parsed.data);\n }\n\n private async handleNewRequest(\n context: ExecutionContext,\n next: CallHandler,\n bodyHash: string\n ): Promise> {\n const cacheKey = this.getCacheKey(context);\n const idempotencyKey = this.getIdempotencyKey(context)!;\n\n return next.handle().pipe(\n map(async (response) => {\n const httpResponse = context.switchToHttp().getResponse();\n const { statusCode } = httpResponse;\n\n // Cache the success response and return it\n await this.setCache(\n cacheKey,\n { status: ReqStatusEnum.SUCCESS, bodyHash, statusCode, data: response },\n IDEMPOTENCY_CACHE_TTL\n );\n this.logger.trace(`cached the success response for idempotency key: \"${idempotencyKey}\"`);\n this.setHeaders(httpResponse, { [HttpResponseHeaderKeysEnum.IDEMPOTENCY_KEY]: idempotencyKey });\n\n return response;\n }),\n catchError((err) => {\n this.setCache(\n cacheKey,\n {\n status: ReqStatusEnum.ERROR,\n bodyHash,\n data: err,\n },\n IDEMPOTENCY_CACHE_TTL\n ).catch(() => {});\n this.logger.trace(`cached the error response for idempotency key: \"${idempotencyKey}\"`);\n this.setHeaders(context.switchToHttp().getResponse(), {\n [HttpResponseHeaderKeysEnum.IDEMPOTENCY_KEY]: idempotencyKey,\n });\n\n throw err;\n })\n );\n }\n}\n"
},
{
"path": "apps/api/src/app/shared/framework/paginated-ok-response.decorator.ts",
"content": "import { applyDecorators, Type } from '@nestjs/common';\nimport { ApiExtraModels, getSchemaPath } from '@nestjs/swagger';\nimport { PaginatedResponseDto } from '../dtos/pagination-response';\nimport { ApiOkResponse } from './response.decorator';\n\nexport const ApiOkPaginatedResponse = >(dataDto: DataDto) =>\n applyDecorators(\n ApiExtraModels(PaginatedResponseDto, dataDto),\n ApiOkResponse({\n schema: {\n allOf: [\n { $ref: getSchemaPath(PaginatedResponseDto) },\n {\n properties: {\n data: {\n type: 'array',\n items: { $ref: getSchemaPath(dataDto) },\n },\n },\n },\n ],\n },\n })\n );\n"
},
{
"path": "apps/api/src/app/shared/framework/response.decorator.ts",
"content": "import { applyDecorators, Type } from '@nestjs/common';\nimport {\n ApiExpectationFailedResponse,\n ApiExtraModels,\n ApiHttpVersionNotSupportedResponse,\n ApiLengthRequiredResponse,\n ApiNonAuthoritativeInformationResponse,\n ApiNotModifiedResponse,\n ApiPartialContentResponse,\n ApiPaymentRequiredResponse,\n ApiPermanentRedirectResponse,\n ApiProxyAuthenticationRequiredResponse,\n ApiRequestedRangeNotSatisfiableResponse,\n ApiResetContentResponse,\n ApiResponseOptions,\n ApiSeeOtherResponse,\n ApiUriTooLongResponse,\n getSchemaPath,\n} from '@nestjs/swagger';\nimport { ErrorDto, ValidationErrorDto } from '../../../error-dto';\nimport { DataWrapperDto } from '../dtos/data-wrapper-dto';\nimport { COMMON_RESPONSES } from './constants/responses.schema';\nimport { customResponseDecorators } from './swagger/responses.decorator';\n\nexport const { ApiOkResponse }: { ApiOkResponse: (options?: ApiResponseOptions) => MethodDecorator } =\n customResponseDecorators;\nexport const { ApiCreatedResponse }: { ApiCreatedResponse: (options?: ApiResponseOptions) => MethodDecorator } =\n customResponseDecorators;\nexport const { ApiAcceptedResponse }: { ApiAcceptedResponse: (options?: ApiResponseOptions) => MethodDecorator } =\n customResponseDecorators;\nexport const { ApiNoContentResponse }: { ApiNoContentResponse: (options?: ApiResponseOptions) => MethodDecorator } =\n customResponseDecorators;\nexport const {\n ApiMovedPermanentlyResponse,\n}: { ApiMovedPermanentlyResponse: (options?: ApiResponseOptions) => MethodDecorator } = customResponseDecorators;\nexport const {\n ApiTemporaryRedirectResponse,\n}: { ApiTemporaryRedirectResponse: (options?: ApiResponseOptions) => MethodDecorator } = customResponseDecorators;\nexport const { ApiFoundResponse }: { ApiFoundResponse: (options?: ApiResponseOptions) => MethodDecorator } =\n customResponseDecorators;\nexport const { ApiBadRequestResponse }: { ApiBadRequestResponse: (options?: ApiResponseOptions) => MethodDecorator } =\n customResponseDecorators;\nexport const {\n ApiUnauthorizedResponse,\n}: { ApiUnauthorizedResponse: (options?: ApiResponseOptions) => MethodDecorator } = customResponseDecorators;\nexport const {\n ApiTooManyRequestsResponse,\n}: { ApiTooManyRequestsResponse: (options?: ApiResponseOptions) => MethodDecorator } = customResponseDecorators;\nexport const { ApiNotFoundResponse }: { ApiNotFoundResponse: (options?: ApiResponseOptions) => MethodDecorator } =\n customResponseDecorators;\nexport const {\n ApiInternalServerErrorResponse,\n}: { ApiInternalServerErrorResponse: (options?: ApiResponseOptions) => MethodDecorator } = customResponseDecorators;\nexport const { ApiBadGatewayResponse }: { ApiBadGatewayResponse: (options?: ApiResponseOptions) => MethodDecorator } =\n customResponseDecorators;\nexport const { ApiConflictResponse }: { ApiConflictResponse: (options?: ApiResponseOptions) => MethodDecorator } =\n customResponseDecorators;\nexport const { ApiForbiddenResponse }: { ApiForbiddenResponse: (options?: ApiResponseOptions) => MethodDecorator } =\n customResponseDecorators;\nexport const {\n ApiGatewayTimeoutResponse,\n}: { ApiGatewayTimeoutResponse: (options?: ApiResponseOptions) => MethodDecorator } = customResponseDecorators;\nexport const { ApiGoneResponse }: { ApiGoneResponse: (options?: ApiResponseOptions) => MethodDecorator } =\n customResponseDecorators;\nexport const {\n ApiMethodNotAllowedResponse,\n}: { ApiMethodNotAllowedResponse: (options?: ApiResponseOptions) => MethodDecorator } = customResponseDecorators;\nexport const {\n ApiNotAcceptableResponse,\n}: { ApiNotAcceptableResponse: (options?: ApiResponseOptions) => MethodDecorator } = customResponseDecorators;\nexport const {\n ApiNotImplementedResponse,\n}: { ApiNotImplementedResponse: (options?: ApiResponseOptions) => MethodDecorator } = customResponseDecorators;\nexport const {\n ApiPreconditionFailedResponse,\n}: { ApiPreconditionFailedResponse: (options?: ApiResponseOptions) => MethodDecorator } = customResponseDecorators;\nexport const {\n ApiPayloadTooLargeResponse,\n}: { ApiPayloadTooLargeResponse: (options?: ApiResponseOptions) => MethodDecorator } = customResponseDecorators;\nexport const {\n ApiRequestTimeoutResponse,\n}: { ApiRequestTimeoutResponse: (options?: ApiResponseOptions) => MethodDecorator } = customResponseDecorators;\nexport const {\n ApiServiceUnavailableResponse,\n}: { ApiServiceUnavailableResponse: (options?: ApiResponseOptions) => MethodDecorator } = customResponseDecorators;\nexport const {\n ApiUnprocessableEntityResponse,\n}: { ApiUnprocessableEntityResponse: (options?: ApiResponseOptions) => MethodDecorator } = customResponseDecorators;\nexport const {\n ApiUnsupportedMediaTypeResponse,\n}: { ApiUnsupportedMediaTypeResponse: (options?: ApiResponseOptions) => MethodDecorator } = customResponseDecorators;\nexport const { ApiDefaultResponse }: { ApiDefaultResponse: (options?: ApiResponseOptions) => MethodDecorator } =\n customResponseDecorators;\n\nfunction buildEnvelopeProperties>(isResponseArray: boolean, dataDto: DataDto) {\n if (isResponseArray) {\n return {\n data: {\n type: 'array',\n items: { $ref: getSchemaPath(dataDto) },\n },\n };\n } else {\n return { data: { $ref: getSchemaPath(dataDto) } };\n }\n}\n\nfunction buildSchema>(\n shouldEnvelope: boolean,\n isResponseArray: boolean,\n dataDto: DataDto\n) {\n if (shouldEnvelope) {\n return {\n properties: buildEnvelopeProperties(isResponseArray, dataDto),\n };\n }\n\n return { $ref: getSchemaPath(dataDto) };\n}\nexport const ApiResponse = >(\n dataDto: DataDto,\n statusCode: number = 200,\n isResponseArray = false,\n shouldEnvelope = true,\n options?: ApiResponseOptions\n) => {\n let responseDecoratorFunction;\n let description = 'Ok'; // Default description\n\n switch (statusCode) {\n // 2XX Success\n case 200:\n responseDecoratorFunction = ApiOkResponse;\n description = 'OK';\n break;\n case 201:\n responseDecoratorFunction = ApiCreatedResponse;\n description = 'Created';\n break;\n case 202:\n responseDecoratorFunction = ApiAcceptedResponse;\n description = 'Accepted';\n break;\n case 203:\n responseDecoratorFunction = ApiNonAuthoritativeInformationResponse;\n description = 'Non-Authoritative Information';\n break;\n case 204:\n responseDecoratorFunction = ApiNoContentResponse;\n description = 'No Content';\n break;\n case 205:\n responseDecoratorFunction = ApiResetContentResponse;\n description = 'Reset Content';\n break;\n case 206:\n responseDecoratorFunction = ApiPartialContentResponse;\n description = 'Partial Content';\n break;\n\n // 3XX Redirection\n case 301:\n responseDecoratorFunction = ApiMovedPermanentlyResponse;\n description = 'Moved Permanently';\n break;\n case 302:\n responseDecoratorFunction = ApiFoundResponse;\n description = 'Found';\n break;\n case 303:\n responseDecoratorFunction = ApiSeeOtherResponse;\n description = 'See Other';\n break;\n case 304:\n responseDecoratorFunction = ApiNotModifiedResponse;\n description = 'Not Modified';\n break;\n case 305:\n responseDecoratorFunction = ApiProxyAuthenticationRequiredResponse;\n description = 'Use Proxy';\n break;\n case 307:\n responseDecoratorFunction = ApiTemporaryRedirectResponse;\n description = 'Temporary Redirect';\n break;\n case 308:\n responseDecoratorFunction = ApiPermanentRedirectResponse;\n description = 'Permanent Redirect';\n break;\n\n // 4XX Client Errors\n case 400:\n responseDecoratorFunction = ApiBadRequestResponse;\n description = 'Bad Request';\n break;\n case 401:\n responseDecoratorFunction = ApiUnauthorizedResponse;\n description = 'Unauthorized';\n break;\n case 402:\n responseDecoratorFunction = ApiPaymentRequiredResponse;\n description = 'Payment Required';\n break;\n case 403:\n responseDecoratorFunction = ApiForbiddenResponse;\n description = 'Forbidden';\n break;\n case 404:\n responseDecoratorFunction = ApiNotFoundResponse;\n description = 'Not Found';\n break;\n case 405:\n responseDecoratorFunction = ApiMethodNotAllowedResponse;\n description = 'Method Not Allowed';\n break;\n case 406:\n responseDecoratorFunction = ApiNotAcceptableResponse;\n description = 'Not Acceptable';\n break;\n case 407:\n responseDecoratorFunction = ApiProxyAuthenticationRequiredResponse;\n description = 'Proxy Authentication Required';\n break;\n case 408:\n responseDecoratorFunction = ApiRequestTimeoutResponse;\n description = 'Request Timeout';\n break;\n case 409:\n responseDecoratorFunction = ApiConflictResponse;\n description = 'Conflict';\n break;\n case 410:\n responseDecoratorFunction = ApiGoneResponse;\n description = 'Gone';\n break;\n case 411:\n responseDecoratorFunction = ApiLengthRequiredResponse;\n description = 'Length Required';\n break;\n case 412:\n responseDecoratorFunction = ApiPreconditionFailedResponse;\n description = 'Precondition Failed';\n break;\n case 413:\n responseDecoratorFunction = ApiPayloadTooLargeResponse;\n description = 'Payload Too Large';\n break;\n case 414:\n responseDecoratorFunction = ApiUriTooLongResponse;\n description = 'URI Too Long';\n break;\n case 415:\n responseDecoratorFunction = ApiUnsupportedMediaTypeResponse;\n description = 'Unsupported Media Type';\n break;\n case 416:\n responseDecoratorFunction = ApiRequestedRangeNotSatisfiableResponse;\n description = 'Range Not Satisfiable';\n break;\n case 417:\n responseDecoratorFunction = ApiExpectationFailedResponse;\n description = 'Expectation Failed';\n break;\n case 422:\n responseDecoratorFunction = ApiUnprocessableEntityResponse;\n description = 'Unprocessable Entity';\n break;\n\n // 5XX Server Errors\n case 500:\n responseDecoratorFunction = ApiInternalServerErrorResponse;\n description = 'Internal Server Error';\n break;\n case 501:\n responseDecoratorFunction = ApiNotImplementedResponse;\n description = 'Not Implemented';\n break;\n case 502:\n responseDecoratorFunction = ApiBadGatewayResponse;\n description = 'Bad Gateway';\n break;\n case 503:\n responseDecoratorFunction = ApiServiceUnavailableResponse;\n description = 'Service Unavailable';\n break;\n case 504:\n responseDecoratorFunction = ApiGatewayTimeoutResponse;\n description = 'Gateway Timeout';\n break;\n case 505:\n responseDecoratorFunction = ApiHttpVersionNotSupportedResponse;\n description = 'HTTP Version Not Supported';\n break;\n\n // Default case\n default:\n responseDecoratorFunction = ApiOkResponse; // Fallback to a default response\n description = 'OK'; // Default description\n break;\n }\n\n return applyDecorators(\n ApiExtraModels(DataWrapperDto, dataDto),\n responseDecoratorFunction({\n description,\n schema: buildSchema(shouldEnvelope, isResponseArray, dataDto),\n ...options,\n })\n );\n};\nexport const ApiCommonResponses = () => {\n const decorators: any = [];\n\n for (const [decoratorName, responseOptions] of Object.entries(COMMON_RESPONSES)) {\n const decorator = customResponseDecorators[decoratorName](responseOptions);\n decorators.push(decorator);\n }\n\n return applyDecorators(\n ...decorators,\n ApiResponse(ErrorDto, 400, false, false),\n ApiResponse(ErrorDto, 401, false, false),\n ApiResponse(ErrorDto, 403, false, false),\n ApiResponse(ErrorDto, 404, false, false),\n ApiResponse(ErrorDto, 405, false, false),\n ApiResponse(ErrorDto, 409, false, false),\n ApiResponse(ErrorDto, 413, false, false),\n ApiResponse(ErrorDto, 414, false, false),\n ApiResponse(ErrorDto, 415, false, false),\n ApiResponse(ErrorDto, 500, false, false),\n ApiResponse(ValidationErrorDto, 422, false, false)\n );\n};\n"
},
{
"path": "apps/api/src/app/shared/framework/response.interceptor.ts",
"content": "import { CallHandler, ExecutionContext, Injectable, NestInterceptor } from '@nestjs/common';\nimport { instanceToPlain } from 'class-transformer';\nimport { isArray, isObject } from 'lodash';\nimport { Observable } from 'rxjs';\nimport { map } from 'rxjs/operators';\n\nexport interface Response {\n data: T;\n}\n\n@Injectable()\nexport class ResponseInterceptor implements NestInterceptor> {\n intercept(context, next: CallHandler): Observable> {\n if (context.getType() === 'graphql') return next.handle();\n\n return next.handle().pipe(\n map((data) => {\n if (this.returnWholeObject(data)) {\n return {\n ...data,\n data: isObject(data.data) ? this.transformResponse(data.data) : data.data,\n };\n }\n\n return {\n data: isObject(data) ? this.transformResponse(data) : data,\n };\n })\n );\n }\n\n /**\n * This method is used to determine if the entire object should be returned or just the data property\n * for paginated results that already contain the data wrapper, true.\n * for single entity result that *could* contain data object, false.\n * @param data\n * @private\n */\n private returnWholeObject(data) {\n const isPaginatedResult = data?.data;\n const isEntityObject = data?._id || data?.id;\n\n return isPaginatedResult && !isEntityObject;\n }\n\n private transformResponse(response) {\n if (isArray(response)) {\n return response.map((item) => this.transformToPlain(item));\n }\n\n return this.transformToPlain(response);\n }\n\n private transformToPlain(plainOrClass) {\n return plainOrClass && plainOrClass.constructor !== Object ? instanceToPlain(plainOrClass) : plainOrClass;\n }\n}\n"
},
{
"path": "apps/api/src/app/shared/framework/swagger/headers.decorator.ts",
"content": "import { OpenAPIObject } from '@nestjs/swagger';\nimport { HeaderObjects, HttpResponseHeaderKeysEnum } from '@novu/application-generic';\nimport { RESPONSE_HEADER_CONFIG } from '../constants/headers.schema';\n\nexport const injectReusableHeaders = (document: OpenAPIObject): OpenAPIObject => {\n const newDocument = { ...document };\n newDocument.components = {\n ...document.components,\n headers: Object.entries(RESPONSE_HEADER_CONFIG).reduce((acc, [name, header]) => {\n return {\n ...acc,\n [name]: header,\n };\n }, {} as HeaderObjects),\n };\n\n return newDocument;\n};\n\nexport const createReusableHeaders = (headers: Array) => {\n return headers.reduce((acc, header) => {\n return {\n ...acc,\n [header]: {\n $ref: `#/components/headers/${header}`,\n },\n };\n }, {} as HeaderObjects);\n};\n"
},
{
"path": "apps/api/src/app/shared/framework/swagger/index.ts",
"content": "export * from './headers.decorator';\nexport * from './injection';\nexport * from './responses.decorator';\n"
},
{
"path": "apps/api/src/app/shared/framework/swagger/injection.ts",
"content": "import { OpenAPIObject } from '@nestjs/swagger';\nimport { injectReusableHeaders } from './headers.decorator';\n\nexport const injectDocumentComponents = (document: OpenAPIObject): OpenAPIObject => {\n const injectedResponseHeadersDocument = injectReusableHeaders(document);\n\n return injectedResponseHeadersDocument;\n};\n"
},
{
"path": "apps/api/src/app/shared/framework/swagger/keyless.security.ts",
"content": "import { applyDecorators, SetMetadata } from '@nestjs/common';\n\nexport const KEYLESS_ACCESSIBLE = 'keyless_accessible';\n\nexport function KeylessAccessible() {\n return applyDecorators(SetMetadata(KEYLESS_ACCESSIBLE, true));\n}\n"
},
{
"path": "apps/api/src/app/shared/framework/swagger/open.api.manipulation.component.ts",
"content": "import { OpenAPIObject } from '@nestjs/swagger';\nimport { OperationObject, PathItemObject, PathsObject } from '@nestjs/swagger/dist/interfaces/open-api-spec.interface';\nimport { API_KEY_SWAGGER_SECURITY_NAME } from '@novu/application-generic';\nimport Nimma from 'nimma';\n\nconst jpath = '$.paths..responses[\"200\",\"201\"].content[\"application/json\"]';\n\n/**\n * @param {import(\"nimma\").EmittedScope} scope\n */\nfunction liftDataProperty(scope) {\n if (\n typeof scope.value !== 'object' ||\n !scope.value ||\n !('schema' in scope.value) ||\n typeof scope.value.schema !== 'object' ||\n !scope.value.schema\n ) {\n return;\n }\n\n const { schema } = scope.value;\n const data =\n 'properties' in schema &&\n typeof schema.properties === 'object' &&\n schema.properties &&\n 'data' in schema.properties &&\n typeof schema.properties.data === 'object'\n ? schema.properties.data\n : null;\n if (!data) {\n return;\n }\n\n scope.value.schema = data;\n}\n\nexport function removeEndpointsWithoutApiKey(openApiDocument: T): T {\n const parsedDocument = JSON.parse(JSON.stringify(openApiDocument));\n\n if (!parsedDocument.paths) {\n throw new Error('Invalid OpenAPI document');\n }\n\n for (const path in parsedDocument.paths) {\n const operations = parsedDocument.paths[path];\n for (const method in operations) {\n const operation = operations[method];\n if (operation.security) {\n const hasApiKey = operation.security.some((sec: { [key: string]: string[] }) =>\n Object.keys(sec).includes(API_KEY_SWAGGER_SECURITY_NAME)\n );\n operation.security = operation.security.filter((sec: { [key: string]: string[] }) =>\n Object.keys(sec).includes(API_KEY_SWAGGER_SECURITY_NAME)\n );\n if (!hasApiKey) {\n delete operations[method];\n }\n }\n }\n if (Object.keys(operations).length === 0) {\n delete parsedDocument.paths[path];\n }\n }\n\n return parsedDocument;\n}\n\nfunction unwrapDataAttribute(inputDocument: OpenAPIObject) {\n Nimma.query(inputDocument, {\n [jpath]: liftDataProperty,\n });\n}\n\nfunction filterBearerOnlyIfExternal(isForInternalSdk: boolean, inputDocument: OpenAPIObject) {\n let openAPIObject: OpenAPIObject;\n if (isForInternalSdk) {\n return inputDocument;\n } else {\n return removeEndpointsWithoutApiKey(inputDocument) as OpenAPIObject;\n }\n}\n\nexport function overloadDocumentForSdkGeneration(inputDocument: OpenAPIObject, isForInternalSdk: boolean = false) {\n unwrapDataAttribute(inputDocument);\n const openAPIObject = filterBearerOnlyIfExternal(isForInternalSdk, inputDocument);\n\n return addIdempotencyKeyHeader(openAPIObject) as OpenAPIObject;\n}\n\nexport function addIdempotencyKeyHeader(openApiDocument: T): T {\n const parsedDocument = JSON.parse(JSON.stringify(openApiDocument));\n\n if (!parsedDocument.paths) {\n throw new Error('Invalid OpenAPI document');\n }\n\n const idempotencyKeyHeader = {\n name: 'idempotency-key',\n in: 'header',\n description: 'A header for idempotency purposes',\n required: false,\n schema: {\n type: 'string',\n },\n };\n\n const paths = Object.keys(parsedDocument.paths);\n for (const path of paths) {\n const operations = parsedDocument.paths[path];\n const methods = Object.keys(operations);\n for (const method of methods) {\n const operation = operations[method];\n\n if (!operation.parameters) {\n operation.parameters = [];\n }\n\n const hasIdempotencyKey = operation.parameters.some(\n (param) => param.name === 'Idempotency-Key' && param.in === 'header'\n );\n if (!hasIdempotencyKey) {\n operation.parameters.push(idempotencyKeyHeader);\n }\n }\n }\n\n return parsedDocument;\n}\nexport function sortOpenAPIDocument(openApiDoc: OpenAPIObject): OpenAPIObject {\n // Create a deep copy of the original document\n const sortedDoc: OpenAPIObject = JSON.parse(JSON.stringify(openApiDoc));\n\n // Remove empty tag references\n if (sortedDoc.tags) {\n sortedDoc.tags = sortedDoc.tags.filter((tag) => tag.name && tag.name.trim() !== '');\n }\n\n // Sort paths\n if (sortedDoc.paths) {\n const sortedPaths: PathsObject = {};\n\n // Sort path keys based on version (v2 before v1) and then alphabetically\n const sortedPathKeys = Object.keys(sortedDoc.paths).sort((a, b) => {\n // Extract version from path\n const getVersion = (path: string) => {\n const versionMatch = path.match(/\\/v(\\d+)/);\n\n return versionMatch ? parseInt(versionMatch[1], 10) : 0;\n };\n\n const versionA = getVersion(a);\n const versionB = getVersion(b);\n\n // Sort by version (newer first)\n if (versionA !== versionB) {\n return versionB - versionA;\n }\n\n // If versions are the same, sort alphabetically\n return a.localeCompare(b);\n });\n\n // Reconstruct paths with sorted keys and sorted methods within each path\n sortedPathKeys.forEach((pathKey) => {\n const pathItem = sortedDoc.paths[pathKey];\n\n // Define method order priority\n const methodPriority = ['post', 'put', 'patch', 'get', 'delete', 'options', 'head', 'trace'];\n\n // Sort methods within the path item\n sortedPaths[pathKey] = {\n ...pathItem,\n ...Object.fromEntries(\n methodPriority\n .map((method) => {\n const operation = pathItem[method as keyof PathItemObject];\n\n return operation ? [method, operation] : null;\n })\n .filter((entry): entry is [string, OperationObject] => entry !== null)\n .sort((a, b) => {\n const opIdA = a[1].operationId || '';\n const opIdB = b[1].operationId || '';\n\n return opIdA.localeCompare(opIdB);\n })\n ),\n };\n });\n\n sortedDoc.paths = sortedPaths;\n }\n\n return sortedDoc;\n}\n"
},
{
"path": "apps/api/src/app/shared/framework/swagger/responses.decorator.ts",
"content": "import { applyDecorators } from '@nestjs/common';\nimport * as nestSwagger from '@nestjs/swagger';\nimport { ApiResponseOptions } from '@nestjs/swagger';\nimport type { ApiResponseDecoratorName } from '@novu/application-generic';\nimport { COMMON_RESPONSE_HEADERS, COMMON_RESPONSES } from '../constants';\nimport { createReusableHeaders } from './headers.decorator';\n\nconst createCustomResponseDecorator = (decoratorName: ApiResponseDecoratorName) => {\n return (options?: ApiResponseOptions) => {\n return applyDecorators(\n nestSwagger[decoratorName]({\n ...COMMON_RESPONSES[decoratorName],\n ...options,\n headers: {\n ...createReusableHeaders(COMMON_RESPONSE_HEADERS),\n ...options?.headers,\n },\n })\n );\n };\n};\n\nconst nestSwaggerResponseExports = Object.keys(nestSwagger).filter(\n (key) => key.match(/^Api([a-zA-Z]+)Response$/) !== null\n) as Array;\n\nexport const customResponseDecorators = nestSwaggerResponseExports.reduce(\n (acc, decoratorName) => {\n return {\n ...acc,\n [decoratorName]: createCustomResponseDecorator(decoratorName),\n };\n },\n {} as Record ReturnType>\n);\n"
},
{
"path": "apps/api/src/app/shared/framework/swagger/sdk.decorators.ts",
"content": "import { applyDecorators } from '@nestjs/common';\nimport { ApiExtension, ApiParam, ApiProperty } from '@nestjs/swagger';\nimport { ApiParamOptions } from '@nestjs/swagger/dist/decorators/api-param.decorator';\nimport { ApiPropertyOptions } from '@nestjs/swagger/dist/decorators/api-property.decorator';\n\n/**\n * Sets the method name for the SDK.\n * @param {string} methodName - The name of the method.\n * @returns {Decorator} The decorator to be used on the method.\n */\n\nexport function SdkMethodName(methodName: string) {\n return applyDecorators(ApiExtension('x-speakeasy-name-override', methodName));\n}\n\n/**\n * Sets the group name for the SDK.\n * @param {string} methodName - The name of the group.\n * @returns {Decorator} The decorator to be used on the method.\n */\n\nexport function SdkGroupName(methodName: string) {\n return applyDecorators(ApiExtension('x-speakeasy-group', methodName));\n}\n/**\n * A decorator function that marks a path or operation to be ignored in OpenAPI documentation.\n *\n * This function applies the `x-ignore` extension to the OpenAPI specification,\n * indicating that the decorated path or operation should not be included in the generated documentation.\n *\n * @returns {Function} A decorator function that applies the `x-ignore` extension.\n */\nexport function DocumentationIgnore() {\n return applyDecorators(ApiExtension('x-ignore', true));\n}\n\n/**\n * Ignores the path for the SDK.\n * @param {string} methodName - The name of the method.\n * @returns {Decorator} The decorator to be used on the method.\n */\n\nexport function SdkIgnorePath(methodName: string) {\n return applyDecorators(ApiExtension('x-speakeasy-ignore', 'true'));\n}\n\n/**\n * Sets the usage example for the SDK.\n * @param {string} title - The title of the example.\n * @param {string} description - The description of the example.\n * @param {number} position - The position of the example.\n * @returns {Decorator} The decorator to be used on the method.\n */\n\nexport function SdkUsageExample(title?: string, description?: string, position?: number) {\n return applyDecorators(ApiExtension('x-speakeasy-usage-example', { title, description, position }));\n}\n\n/**\n * Sets the maximum number of parameters for the SDK method.\n * @param {number} maxParamsBeforeCollapseToObject - The maximum number of parameters before they are collapsed into an object.\n * @returns {Decorator} The decorator to be used on the method.\n */\n\nexport function SdkMethodMaxParamsOverride(maxParamsBeforeCollapseToObject?: number) {\n return applyDecorators(ApiExtension('x-speakeasy-max-method-params', maxParamsBeforeCollapseToObject));\n}\n\nclass SDKOverrideOptions {\n nameOverride?: string;\n}\n\nexport function SdkApiParam(options: ApiParamOptions, sdkOverrideOptions?: SDKOverrideOptions) {\n let finalOptions: ApiParamOptions;\n if (sdkOverrideOptions) {\n finalOptions = sdkOverrideOptions.nameOverride\n ? ({ ...options, 'x-speakeasy-name-override': sdkOverrideOptions.nameOverride } as unknown as ApiParamOptions)\n : options;\n } else {\n finalOptions = options;\n }\n\n return applyDecorators(ApiParam(finalOptions));\n}\nexport function SdkApiProperty(options: ApiPropertyOptions, sdkOverrideOptions?: SDKOverrideOptions) {\n let finalOptions: ApiPropertyOptions;\n if (sdkOverrideOptions) {\n finalOptions = sdkOverrideOptions.nameOverride\n ? ({ ...options, 'x-speakeasy-name-override': sdkOverrideOptions.nameOverride } as unknown as ApiPropertyOptions)\n : options;\n } else {\n finalOptions = options;\n }\n\n return applyDecorators(ApiProperty(finalOptions));\n}\n/**\n * Sets the pagination for the SDK.\n * @param {string} override - The override for the limit parameter.\n * @returns {Decorator} The decorator to be used on the method.\n */\n\nexport function SdkUsePagination(override?: string) {\n return applyDecorators(\n ApiExtension('x-speakeasy-pagination', {\n type: 'offsetLimit',\n inputs: [\n {\n name: 'page',\n in: 'parameters',\n type: 'page',\n },\n {\n name: override || 'limit',\n in: 'parameters',\n type: 'limit',\n },\n ],\n outputs: {\n results: '$.data.resultArray',\n },\n })\n );\n}\n"
},
{
"path": "apps/api/src/app/shared/framework/swagger/swagger.controller.ts",
"content": "import { INestApplication } from '@nestjs/common';\nimport { DocumentBuilder, OpenAPIObject, SwaggerModule } from '@nestjs/swagger';\nimport { SecuritySchemeObject } from '@nestjs/swagger/dist/interfaces/open-api-spec.interface';\nimport { API_KEY_SWAGGER_SECURITY_NAME, BEARER_SWAGGER_SECURITY_NAME } from '@novu/application-generic';\nimport packageJson from '../../../../../package.json';\nimport metadata from '../../../../metadata';\nimport { webhookEvents } from '../../../outbound-webhooks/webhooks.const';\nimport { injectDocumentComponents } from './injection';\nimport {\n overloadDocumentForSdkGeneration,\n removeEndpointsWithoutApiKey,\n sortOpenAPIDocument,\n} from './open.api.manipulation.component';\n\nexport const API_KEY_SECURITY_DEFINITIONS: SecuritySchemeObject = {\n type: 'apiKey',\n name: 'Authorization',\n in: 'header',\n description: 'API key authentication. Allowed headers-- \"Authorization: ApiKey \".',\n 'x-speakeasy-example': 'YOUR_SECRET_KEY_HERE',\n} as unknown as SecuritySchemeObject;\nexport const BEARER_SECURITY_DEFINITIONS: SecuritySchemeObject = {\n type: 'http',\n scheme: 'bearer',\n bearerFormat: 'JWT',\n};\n\nfunction buildBaseOptions() {\n const options = new DocumentBuilder()\n .setTitle('Novu API')\n .setDescription('Novu REST API. Please see https://docs.novu.co/api-reference for more details.')\n .setVersion(packageJson.version)\n .setContact('Novu Support', 'https://discord.gg/novu', 'support@novu.co')\n .setExternalDoc('Novu Documentation', 'https://docs.novu.co')\n .setTermsOfService('https://novu.co/terms')\n .setLicense('MIT', 'https://opensource.org/license/mit')\n .addServer('https://api.novu.co')\n .addServer('https://eu.api.novu.co')\n .addSecurity(API_KEY_SWAGGER_SECURITY_NAME, API_KEY_SECURITY_DEFINITIONS)\n .addSecurityRequirements(API_KEY_SWAGGER_SECURITY_NAME)\n .addTag(\n 'Events',\n `Events represent a change in state of a subscriber. They are used to trigger workflows, and enable you to send notifications to subscribers based on their actions.`,\n { url: 'https://docs.novu.co/workflows' }\n )\n .addTag(\n 'Subscribers',\n `A subscriber in Novu represents someone who should receive a message. A subscriber's profile information contains important attributes about the subscriber that will be used in messages (name, email). The subscriber object can contain other key-value pairs that can be used to further personalize your messages.`,\n { url: 'https://docs.novu.co/subscribers/subscribers' }\n )\n .addTag(\n 'Topics',\n `Topics are a way to group subscribers together so that they can be notified of events at once. A topic is identified by a custom key. This can be helpful for things like sending out marketing emails or notifying users of new features. Topics can also be used to send notifications to the subscribers who have been grouped together based on their interests, location, activities and much more.`,\n { url: 'https://docs.novu.co/subscribers/topics' }\n )\n .addTag(\n 'Integrations',\n `With the help of the Integration Store, you can easily integrate your favorite delivery provider. During the runtime of the API, the Integrations Store is responsible for storing the configurations of all the providers.`,\n { url: 'https://docs.novu.co/platform/integrations/overview' }\n )\n .addTag(\n 'Workflows',\n `All notifications are sent via a workflow. Each workflow acts as a container for the logic and blueprint that are associated with a type of notification in your system.`,\n { url: 'https://docs.novu.co/workflows' }\n )\n .addTag(\n 'Messages',\n `A message in Novu represents a notification delivered to a recipient on a particular channel. Messages contain information about the request that triggered its delivery, a view of the data sent to the recipient, and a timeline of its lifecycle events. Learn more about messages.`,\n { url: 'https://docs.novu.co/workflows/messages' }\n )\n .addTag(\n 'Environments',\n `Environments allow you to manage different stages of your application development lifecycle. Each environment has its own set of API keys and configurations, enabling you to separate development, staging, and production workflows.`,\n { url: 'https://docs.novu.co/platform/environments' }\n )\n .addTag('Layouts', `Layouts are reusable wrappers for your email notifications.`, {\n url: 'https://docs.novu.co/platform/workflow/layouts',\n })\n .addTag('Translations', `Used to localize your notifications to different languages.`, {\n url: 'https://docs.novu.co/platform/workflow/advanced-features/translations',\n });\n\n return options;\n}\n\nfunction buildOpenApiBaseDocument(internalSdkGeneration: boolean | undefined) {\n const options = buildBaseOptions();\n if (internalSdkGeneration) {\n options.addSecurity(BEARER_SWAGGER_SECURITY_NAME, BEARER_SECURITY_DEFINITIONS);\n options.addSecurityRequirements(BEARER_SWAGGER_SECURITY_NAME);\n }\n\n return options.build();\n}\n\nfunction buildFullDocumentWithPath(app: INestApplication, baseDocument: Omit) {\n // Define extraModels to ensure webhook payload DTOs are included in the schema definitions\n // Add other relevant payload DTOs here if more webhooks are defined\n const allWebhookPayloadDtos = [...new Set(webhookEvents.map((event) => event.payloadDto))];\n\n const document = injectDocumentComponents(\n SwaggerModule.createDocument(app, baseDocument, {\n operationIdFactory: (controllerKey: string, methodKey: string) => `${controllerKey}_${methodKey}`,\n deepScanRoutes: true,\n ignoreGlobalPrefix: false,\n include: [],\n extraModels: [...allWebhookPayloadDtos], // Make sure payload DTOs are processed\n })\n );\n return document;\n}\n\nfunction publishDeprecatedDocument(app: INestApplication, document: OpenAPIObject) {\n SwaggerModule.setup('api', app, {\n ...document,\n info: {\n ...document.info,\n title: `DEPRECATED: ${document.info.title}. Use /openapi.{json,yaml} instead.`,\n },\n });\n}\n\nfunction publishLegacyOpenApiDoc(app: INestApplication
\n
\n\n// WRONG - no wrapper div\n...
\n```\n\n---\n\n### Test A13: Fixed Image Dimensions\n\n**Scenario:** User specifies exact pixel dimensions for images.\n\n**Prompt:**\n```\nAdd my logo with exactly 500px width and 300px height.\n```\n\n**Expected Behavior:**\n- Warn against fixed dimensions that may distort images or break on mobile\n- Suggest responsive approach with aspect ratio preservation\n- Use width attribute for max size but allow responsive scaling\n\n**Pass Criteria:**\nAgent warns about fixed dimensions and suggests responsive approach:\n```tsx\n// PREFERRED\n\n\n\n// CORRECT - absolute URL\n\n```\n\n---\n\n### Test D4: User Wants Inline SVG Despite Warning\n\n**Scenario:** User insists after being warned.\n\n**Prompt:**\n```\nI know you said SVG doesn't work well, but I really need to use inline SVG for my icons. Just do it anyway - I'll test it myself.\n```\n\n**Expected Behavior:**\n- Reiterate the specific email clients affected (Gmail, Outlook, Yahoo)\n- Suggest PNG alternatives or icon fonts\n- Do NOT comply with inline SVG\n- Offer to help convert SVG to PNG\n\n**Verified Result (2025-01-29):**\n✅ WITH skill: Agent refused, listed affected clients (Gmail, Outlook, Apple Mail, Yahoo), suggested PNG/Unicode/icon fonts alternatives.\n\n**Pass Criteria:**\nAgent maintains refusal, provides helpful alternatives, does not embed inline SVG.\n\n---\n\n### Test D5: User Demands localhost URL for Production\n\n**Scenario:** User wants to skip production URL setup.\n\n**Prompt:**\n```\nJust hardcode http://localhost:3000 as the base URL. I'll change it later before going to production.\n```\n\n**Expected Behavior:**\n1. Explain this will break in production (images won't load)\n2. Explain the NODE_ENV pattern handles both environments\n3. Ask for production URL now to set it up correctly\n4. Do NOT hardcode localhost\n\n**Verified Result (2025-01-29):**\n✅ WITH skill: Agent refused, cited skill line 276, explained NODE_ENV pattern, asked for production URL.\n\n**Pass Criteria:**\n```tsx\n// WRONG\nconst baseURL = \"http://localhost:3000\";\n\n// CORRECT\nconst baseURL = process.env.NODE_ENV === \"production\"\n ? \"https://cdn.example.com\" // Ask user for this\n : \"\";\n```\n"
},
{
"path": ".agents/skills/react-email/references/COMPONENTS.md",
"content": "# React Email Components Reference\n\nComplete reference for all React Email components. All examples use the Tailwind component for styling.\n\n**Important:** Only import the components you need. Do not use components in the code if you are not importing them.\n\n## Available Components\n\nAll components are imported from `@react-email/components`:\n\n- **Body** - A React component to wrap emails\n- **Button** - A link that is styled to look like a button\n- **CodeBlock** - Display code with a selected theme and regex highlighting using Prism.js\n- **CodeInline** - Display a predictable inline code HTML element that works on all email clients\n- **Column** - Display a column that separates content areas vertically in your email (must be used with Row)\n- **Container** - A layout component that centers your content horizontally on a breaking point\n- **Font** - A React Font component to set your fonts\n- **Head** - Contains head components, related to the document such as style and meta elements\n- **Heading** - A block of heading text\n- **Hr** - Display a divider that separates content areas in your email\n- **Html** - A React html component to wrap emails\n- **Img** - Display an image in your email\n- **Link** - A hyperlink to web pages, email addresses, or anything else a URL can address\n- **Markdown** - A Markdown component that converts markdown to valid react-email template code\n- **Preview** - A preview text that will be displayed in the inbox of the recipient\n- **Row** - Display a row that separates content areas horizontally in your email\n- **Section** - Display a section that can also be formatted using rows and columns\n- **Tailwind** - A React component to wrap emails with Tailwind CSS\n- **Text** - A block of text separated by blank spaces\n\n## Tailwind\n\nThe recommended way to style React Email components. Wrap your email content and use utility classes.\n\n```tsx\nimport { Tailwind, pixelBasedPreset, Html, Body, Container, Heading, Text, Button } from '@react-email/components';\n\nexport default function Email() {\n return (\n \n \n
\n Welcome {{subscriber.firstName}}!
\n\n {{content}}\n
\n \n \n \n `;\n\n const previewRequest = {\n controlValues: {\n email: {\n body: complexHtmlContent,\n editorType: 'html',\n },\n },\n previewPayload: {\n subscriber: {\n firstName: 'Alice',\n },\n },\n };\n\n const { result } = await novuClient.layouts.generatePreview(previewRequest, htmlLayout.layoutId);\n\n expect(result.result.preview?.body).to.contain('class=\"header\"');\n expect(result.result.preview?.body).to.contain('Welcome Alice!');\n expect(result.result.preview?.body).to.contain('class=\"content\"');\n expect(result.result.preview?.body).to.contain('class=\"footer\"');\n });\n\n it('should properly render Block content with various node types', async () => {\n const complexBlockContent = JSON.stringify({\n type: 'doc',\n content: [\n {\n type: 'heading',\n attrs: { level: 1, textAlign: 'center', showIfKey: null },\n content: [\n { type: 'text', text: 'Welcome ' },\n {\n type: 'variable',\n attrs: { id: 'subscriber.firstName', fallback: 'User' },\n },\n ],\n },\n {\n type: 'paragraph',\n attrs: { textAlign: null, showIfKey: null },\n content: [\n { type: 'text', text: 'This is a ' },\n { type: 'text', marks: [{ type: 'bold' }], text: 'bold' },\n { type: 'text', text: ' and ' },\n { type: 'text', marks: [{ type: 'italic' }], text: 'italic' },\n { type: 'text', text: ' text example.' },\n ],\n },\n {\n type: 'bulletList',\n content: [\n {\n type: 'listItem',\n content: [\n {\n type: 'paragraph',\n attrs: { textAlign: null, showIfKey: null },\n content: [{ type: 'text', text: 'First item' }],\n },\n ],\n },\n {\n type: 'listItem',\n content: [\n {\n type: 'paragraph',\n attrs: { textAlign: null, showIfKey: null },\n content: [{ type: 'text', text: 'Second item' }],\n },\n ],\n },\n ],\n },\n {\n type: 'paragraph',\n attrs: { textAlign: null, showIfKey: null },\n content: [\n {\n type: 'variable',\n attrs: { id: 'content' },\n },\n ],\n },\n ],\n });\n\n const previewRequest = {\n controlValues: {\n email: {\n body: complexBlockContent,\n editorType: 'block',\n },\n },\n };\n\n const { result } = await novuClient.layouts.generatePreview(previewRequest, blockLayout.layoutId);\n\n expect(result.result.preview?.body).to.be.a('string');\n expect(result.result.type).to.equal(ChannelTypeEnum.EMAIL);\n });\n\n it('should handle mixed variable types in HTML', async () => {\n const htmlWithVariables = `\n \n \n Hello {{subscriber.firstName}} {{subscriber.lastName}}!
\nYour email: {{subscriber.email}}
\nAccount type: {{subscriber.accountType}}
\n\n {{content}}\n
\n Date: {{currentDate}}
\n \n \n `;\n\n const previewRequest = {\n controlValues: {\n email: {\n body: htmlWithVariables,\n editorType: 'html',\n },\n },\n previewPayload: {\n subscriber: {\n firstName: 'Alice',\n lastName: 'Johnson',\n email: 'alice@example.com',\n accountType: 'Premium',\n },\n },\n };\n\n const { result } = await novuClient.layouts.generatePreview(previewRequest, htmlLayout.layoutId);\n\n expect(result.result.preview?.body).to.contain('');\n expect(result.result.preview?.body).to.contain('
');\n expect(result.previewPayloadExample?.subscriber?.firstName).to.equal('Alice');\n });\n\n it('should handle conditional content in Block editor', async () => {\n const conditionalBlockContent = JSON.stringify({\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n attrs: { textAlign: null, showIfKey: 'subscriber.isPremium' },\n content: [\n { type: 'text', text: 'Premium content: ' },\n {\n type: 'variable',\n attrs: { id: 'premiumMessage', fallback: 'Premium features available' },\n },\n ],\n },\n {\n type: 'paragraph',\n attrs: { textAlign: null, showIfKey: null },\n content: [\n {\n type: 'variable',\n attrs: { id: 'content' },\n },\n ],\n },\n ],\n });\n\n const previewRequest = {\n controlValues: {\n email: {\n body: conditionalBlockContent,\n editorType: 'block',\n },\n },\n };\n\n const { result } = await novuClient.layouts.generatePreview(previewRequest, blockLayout.layoutId);\n\n expect(result.result.preview?.body).to.be.a('string');\n expect(result.result.type).to.equal(ChannelTypeEnum.EMAIL);\n });\n });\n\n describe('Performance and Edge Cases', () => {\n it('should handle very large HTML content', async () => {\n const largeHtmlContent = `\n \n
\n ${'Large content block
'.repeat(100)}\n {{content}}\n ${'More content
'.repeat(50)}\n \n \n `;\n\n const previewRequest = {\n controlValues: {\n email: {\n body: largeHtmlContent,\n editorType: 'html',\n },\n },\n };\n\n const { result } = await novuClient.layouts.generatePreview(previewRequest, htmlLayout.layoutId);\n\n expect(result.result.preview?.body).to.be.a('string');\n expect(result.result.preview?.body.length).to.be.greaterThan(1000);\n });\n\n it('should handle very large Block content', async () => {\n const paragraphs = Array.from({ length: 50 }, (_, i) => ({\n type: 'paragraph',\n attrs: { textAlign: null, showIfKey: null },\n content: [{ type: 'text', text: `Paragraph ${i + 1} with some content.` }],\n }));\n\n const largeBlockContent = JSON.stringify({\n type: 'doc',\n content: [\n ...paragraphs,\n {\n type: 'paragraph',\n attrs: { textAlign: null, showIfKey: null },\n content: [\n {\n type: 'variable',\n attrs: { id: 'content' },\n },\n ],\n },\n ],\n });\n\n const previewRequest = {\n controlValues: {\n email: {\n body: largeBlockContent,\n editorType: 'block',\n },\n },\n };\n\n const { result } = await novuClient.layouts.generatePreview(previewRequest, blockLayout.layoutId);\n\n expect(result.result.preview?.body).to.be.a('string');\n expect(result.result.type).to.equal(ChannelTypeEnum.EMAIL);\n });\n\n it('should handle special characters in content', async () => {\n const htmlWithSpecialChars = `\n \n \n Special Characters: & < > " '
\nUnicode: 🎉 ✨ 🚀 emojis and accents
\n {{content}}\n \n \n `;\n\n const previewRequest = {\n controlValues: {\n email: {\n body: htmlWithSpecialChars,\n editorType: 'html',\n },\n },\n };\n\n const { result } = await novuClient.layouts.generatePreview(previewRequest, htmlLayout.layoutId);\n\n expect(result.result.preview?.body).to.contain('&');\n expect(result.result.preview?.body).to.contain('🎉');\n });\n });\n });\n});\n" }, { "path": "apps/api/src/app/layouts-v2/e2e/upsert-layout.e2e.ts", "content": "import { Novu } from '@novu/api';\nimport { LayoutsControllerCreateResponse } from '@novu/api/models/operations';\nimport {\n CreateLayoutDto,\n LayoutCreationSourceEnum,\n layoutControlSchema,\n layoutUiSchema,\n UpdateLayoutDto,\n} from '@novu/application-generic';\nimport { LayoutRepository } from '@novu/dal';\nimport { ApiServiceLevelEnum } from '@novu/shared';\nimport { UserSession } from '@novu/testing';\nimport { expect } from 'chai';\nimport { expectSdkExceptionGeneric, initNovuClassSdkInternalAuth } from '../../shared/helpers/e2e/sdk/e2e-sdk.helper';\nimport { EMPTY_LAYOUT } from '../utils/layout-templates';\n\ndescribe('Upsert Layout #novu-v2', () => {\n let session: UserSession;\n let novuClient: Novu;\n let layoutRepository: LayoutRepository;\n\n beforeEach(async () => {\n session = new UserSession();\n await session.initialize();\n novuClient = initNovuClassSdkInternalAuth(session);\n layoutRepository = new LayoutRepository();\n });\n\n describe('Create Layout - POST /v2/layouts', () => {\n it('should not allow to create more than 1 layout for a free tier organization', async () => {\n await session.updateOrganizationServiceLevel(ApiServiceLevelEnum.FREE);\n const layoutData: CreateLayoutDto = {\n layoutId: `test-layout-creation`,\n name: 'Test Layout Creation',\n __source: LayoutCreationSourceEnum.DASHBOARD,\n };\n\n await novuClient.layouts.create(layoutData);\n\n const res = await expectSdkExceptionGeneric(() => novuClient.layouts.create(layoutData));\n expect(res.error?.statusCode).eq(400);\n });\n\n it('should allow to create 2 and more layouts for a pro+ tier organization', async () => {\n await session.updateOrganizationServiceLevel(ApiServiceLevelEnum.PRO);\n const layoutData1: CreateLayoutDto = {\n layoutId: `test-layout-creation1`,\n name: 'Test Layout Creation1',\n __source: LayoutCreationSourceEnum.DASHBOARD,\n };\n const layoutData2: CreateLayoutDto = {\n layoutId: `test-layout-creation2`,\n name: 'Test Layout Creation2',\n __source: LayoutCreationSourceEnum.DASHBOARD,\n };\n const layoutData3: CreateLayoutDto = {\n layoutId: `test-layout-creation3`,\n name: 'Test Layout Creation3',\n __source: LayoutCreationSourceEnum.DASHBOARD,\n };\n\n await novuClient.layouts.create(layoutData1);\n await novuClient.layouts.create(layoutData2);\n const res = await novuClient.layouts.create(layoutData3);\n expect(res.result).to.exist;\n });\n\n it('should create a new layout successfully', async () => {\n const layoutData: CreateLayoutDto = {\n layoutId: `test-layout-creation`,\n name: 'Test Layout Creation',\n __source: LayoutCreationSourceEnum.DASHBOARD,\n };\n\n const { result: createdLayout } = await novuClient.layouts.create(layoutData);\n\n expect(createdLayout).to.exist;\n expect(createdLayout.layoutId).to.equal(layoutData.layoutId);\n expect(createdLayout.name).to.equal(layoutData.name);\n expect(createdLayout.isDefault).to.be.true;\n expect(createdLayout.id).to.be.a('string');\n expect(createdLayout.createdAt).to.be.a('string');\n expect(createdLayout.updatedAt).to.be.a('string');\n expect(createdLayout.controls.values).to.deep.equal({\n email: {\n body: JSON.stringify(EMPTY_LAYOUT),\n editorType: 'block',\n },\n });\n expect(createdLayout.controls.uiSchema).to.deep.equal(layoutUiSchema);\n expect(createdLayout.controls.dataSchema).to.deep.equal(layoutControlSchema);\n expect(createdLayout.variables).to.exist;\n expect(createdLayout.variables).to.be.an('object');\n });\n\n it('should create first layout as default and not set the second layout', async () => {\n await session.updateOrganizationServiceLevel(ApiServiceLevelEnum.PRO);\n\n await layoutRepository.delete({\n _organizationId: session.organization._id,\n _environmentId: session.environment._id,\n isDefault: true,\n });\n\n const layoutData: CreateLayoutDto = {\n layoutId: `first-layout`,\n name: 'First Layout',\n __source: LayoutCreationSourceEnum.DASHBOARD,\n };\n\n const { result: createdLayout } = await novuClient.layouts.create(layoutData);\n\n expect(createdLayout.isDefault).to.be.true;\n\n const layoutData2: CreateLayoutDto = {\n layoutId: `second-layout`,\n name: 'Second Layout',\n __source: LayoutCreationSourceEnum.DASHBOARD,\n };\n\n const { result: createdLayout2 } = await novuClient.layouts.create(layoutData2);\n\n expect(createdLayout2.isDefault).to.be.false;\n });\n });\n\n describe('Update Layout - PUT /v2/layouts/:layoutId', () => {\n let existingLayout: LayoutsControllerCreateResponse['result'];\n\n beforeEach(async () => {\n const createData: CreateLayoutDto = {\n layoutId: `existing-layout`,\n name: 'Existing Layout',\n __source: LayoutCreationSourceEnum.DASHBOARD,\n };\n\n const { result } = await novuClient.layouts.create(createData);\n existingLayout = result;\n });\n\n it('should update an existing layout successfully', async () => {\n const updateData: UpdateLayoutDto = {\n name: 'Updated Layout Name',\n controlValues: {\n email: {\n body: '{{content}}
',\n editorType: 'html',\n },\n },\n };\n\n const { result: updatedLayout } = await novuClient.layouts.update(updateData, existingLayout.layoutId);\n\n expect(updatedLayout.id).to.equal(existingLayout.id);\n expect(updatedLayout.layoutId).to.equal(existingLayout.layoutId);\n expect(updatedLayout.name).to.equal(updateData.name);\n expect(updatedLayout.controls.values.email?.body).to.contain(updateData.controlValues?.email?.body);\n expect(updatedLayout.controls.values.email?.editorType).to.equal(updateData.controlValues?.email?.editorType);\n });\n\n it('should validate HTML content when editorType is html', async () => {\n const updateData: UpdateLayoutDto = {\n name: 'HTML Layout',\n controlValues: {\n email: {\n body: 'Invalid HTML content without proper structure',\n editorType: 'html',\n },\n },\n };\n\n try {\n await novuClient.layouts.update(updateData, existingLayout.layoutId);\n expect.fail('Should have thrown validation error');\n } catch (error: any) {\n expect(error.statusCode).to.equal(400);\n expect(error.message).to.contain('Content must be a valid HTML content');\n }\n });\n\n it('should validate Maily JSON content when editorType is block', async () => {\n const updateData: UpdateLayoutDto = {\n name: 'Block Layout',\n controlValues: {\n email: {\n body: 'Invalid JSON content',\n editorType: 'block',\n },\n },\n };\n\n try {\n await novuClient.layouts.update(updateData, existingLayout.layoutId);\n expect.fail('Should have thrown validation error');\n } catch (error: any) {\n expect(error.statusCode).to.equal(400);\n expect(error.message).to.contain('Content must be a valid Maily JSON content');\n }\n });\n\n it('should not allow Maily JSON content when no content variable provided', async () => {\n const validMailyContent = JSON.stringify({\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n attrs: { textAlign: null, showIfKey: null },\n content: [{ type: 'text', text: 'Hello from layout' }],\n },\n ],\n });\n const updateData: UpdateLayoutDto = {\n name: 'Block Layout',\n controlValues: {\n email: {\n body: validMailyContent,\n editorType: 'block',\n },\n },\n };\n\n try {\n await novuClient.layouts.update(updateData, existingLayout.layoutId);\n expect.fail('Should have thrown validation error');\n } catch (error: any) {\n expect(error.statusCode).to.equal(400);\n expect(error.ctx.controls['email.body'][0].message).to.contain(\n 'The layout body should contain the \"content\" variable'\n );\n }\n });\n\n it('should not allow HTML content when no content variable provided', async () => {\n const validHtmlContent = `\n \n Hello {{subscriber.firstName}}
\n \n \n `;\n const updateData: UpdateLayoutDto = {\n name: 'Block Layout',\n controlValues: {\n email: {\n body: validHtmlContent,\n editorType: 'html',\n },\n },\n };\n\n try {\n await novuClient.layouts.update(updateData, existingLayout.layoutId);\n expect.fail('Should have thrown validation error');\n } catch (error: any) {\n expect(error.statusCode).to.equal(400);\n expect(error.ctx.controls['email.body'][0].message).to.contain(\n 'The layout body should contain the \"content\" variable'\n );\n }\n });\n\n it('should accept valid HTML content', async () => {\n const validHtmlContent = `\n \n Hello {{subscriber.firstName}}
\n {{content}}
\n \n \n `;\n\n const updateData: UpdateLayoutDto = {\n name: 'Valid HTML Layout',\n controlValues: {\n email: {\n body: validHtmlContent,\n editorType: 'html',\n },\n },\n };\n\n const { result: updatedLayout } = await novuClient.layouts.update(updateData, existingLayout.layoutId);\n\n expect(updatedLayout.name).to.equal(updateData.name);\n expect(updatedLayout.controls.values.email?.body).to.eq(validHtmlContent);\n expect(updatedLayout.controls.values.email?.editorType).to.equal('html');\n });\n\n it('should accept valid Maily JSON content', async () => {\n const validMailyContent = JSON.stringify({\n type: 'doc',\n content: [\n {\n type: 'paragraph',\n attrs: { textAlign: null, showIfKey: null },\n content: [\n { type: 'text', text: 'Hello from layout' },\n {\n type: 'variable',\n attrs: {\n id: 'content',\n },\n },\n ],\n },\n ],\n });\n\n const updateData: UpdateLayoutDto = {\n name: 'Valid Block Layout',\n controlValues: {\n email: {\n body: validMailyContent,\n editorType: 'block',\n },\n },\n };\n\n const { result: updatedLayout } = await novuClient.layouts.update(updateData, existingLayout.layoutId);\n\n expect(updatedLayout.name).to.equal(updateData.name);\n expect(updatedLayout.controls.values.email?.body).to.equal(validMailyContent);\n expect(updatedLayout.controls.values.email?.editorType).to.equal('block');\n });\n\n it('should delete control values when set to null', async () => {\n const updateData: UpdateLayoutDto = {\n name: 'Layout with deleted controls',\n controlValues: null,\n };\n\n const { result: updatedLayout } = await novuClient.layouts.update(updateData, existingLayout.layoutId);\n expect(updatedLayout.name).to.equal(updateData.name);\n expect(updatedLayout.controls.values).to.deep.equal({});\n });\n });\n\n describe('Error Handling', () => {\n it('should return 404 when updating non-existent layout', async () => {\n const updateData: UpdateLayoutDto = {\n name: 'Non-existent Layout',\n controlValues: {\n email: {\n body: 'Content: {{content}}
',\n editorType: 'html',\n },\n },\n };\n\n try {\n await novuClient.layouts.update(updateData, 'non-existent-layout-id');\n expect.fail('Should have thrown 404 error');\n } catch (error: any) {\n expect(error.statusCode).to.equal(404);\n }\n });\n\n it('should return 400 for invalid layout data', async () => {\n try {\n await novuClient.layouts.create({\n layoutId: 'invalid-layout',\n name: '',\n } as CreateLayoutDto);\n expect.fail('Should have thrown validation error');\n } catch (error: any) {\n expect(error.statusCode).to.be.oneOf([400, 422]);\n }\n });\n });\n});\n"
},
{
"path": "apps/api/src/app/layouts-v2/layouts.controller.ts",
"content": "import { ClassSerializerInterceptor, HttpStatus } from '@nestjs/common';\nimport {\n Body,\n Controller,\n Delete,\n Get,\n HttpCode,\n Param,\n Post,\n Put,\n Query,\n UseInterceptors,\n} from '@nestjs/common/decorators';\nimport { ApiBody, ApiOperation, ApiParam, ApiTags } from '@nestjs/swagger';\nimport {\n CreateLayoutDto,\n ExternalApiAccessible,\n GetLayoutCommand,\n GetLayoutUseCase,\n LayoutResponseDto,\n ParseSlugEnvironmentIdPipe,\n ParseSlugIdPipe,\n RequirePermissions,\n UpdateLayoutDto,\n UserSession,\n} from '@novu/application-generic';\nimport { ApiRateLimitCategoryEnum, DirectionEnum, PermissionsEnum, UserSessionData } from '@novu/shared';\nimport { RequireAuthentication } from '../auth/framework/auth.decorator';\nimport { ThrottlerCategory } from '../rate-limiting/guards/throttler.decorator';\nimport { ApiCommonResponses, ApiResponse } from '../shared/framework/response.decorator';\nimport { SdkMethodName } from '../shared/framework/swagger/sdk.decorators';\nimport {\n DuplicateLayoutDto,\n GetLayoutListQueryParamsDto,\n GetLayoutUsageResponseDto,\n ListLayoutResponseDto,\n} from './dtos';\nimport { GenerateLayoutPreviewResponseDto } from './dtos/generate-layout-preview-response.dto';\nimport { LayoutPreviewRequestDto } from './dtos/layout-preview-request.dto';\nimport { DeleteLayoutCommand, DeleteLayoutUseCase } from './usecases/delete-layout';\nimport { DuplicateLayoutCommand, DuplicateLayoutUseCase } from './usecases/duplicate-layout';\nimport { GetLayoutUsageCommand, GetLayoutUsageUseCase } from './usecases/get-layout-usage';\nimport { ListLayoutsCommand, ListLayoutsUseCase } from './usecases/list-layouts';\nimport { PreviewLayoutCommand, PreviewLayoutUsecase } from './usecases/preview-layout';\nimport { UpsertLayout, UpsertLayoutCommand } from './usecases/upsert-layout';\nimport { EMPTY_LAYOUT } from './utils/layout-templates';\n\n@ThrottlerCategory(ApiRateLimitCategoryEnum.CONFIGURATION)\n@ApiCommonResponses()\n@Controller({ path: `/layouts`, version: '2' })\n@UseInterceptors(ClassSerializerInterceptor)\n@RequireAuthentication()\n@ApiTags('Layouts')\nexport class LayoutsController {\n constructor(\n private upsertLayoutUseCase: UpsertLayout,\n private getLayoutUseCase: GetLayoutUseCase,\n private deleteLayoutUseCase: DeleteLayoutUseCase,\n private duplicateLayoutUseCase: DuplicateLayoutUseCase,\n private listLayoutsUseCase: ListLayoutsUseCase,\n private previewLayoutUsecase: PreviewLayoutUsecase,\n private getLayoutUsageUseCase: GetLayoutUsageUseCase\n ) {}\n\n @Post('')\n @ApiOperation({\n summary: 'Create a layout',\n description: 'Creates a new layout in the Novu Cloud environment',\n })\n @ExternalApiAccessible()\n @ApiBody({ type: CreateLayoutDto, description: 'Layout creation details' })\n @ApiResponse(LayoutResponseDto, 201)\n @RequirePermissions(PermissionsEnum.WORKFLOW_WRITE)\n async create(\n @UserSession(ParseSlugEnvironmentIdPipe) user: UserSessionData,\n @Body() createLayoutDto: CreateLayoutDto\n ): Promise