prime-sdk 3.0.6 → 3.0.7

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "prime-sdk",
-  "version": "3.0.6",
+  "version": "3.0.7",
   "description": "swiss army knife",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",

package/src/external/aws/sns/sns.provider.ts

@@ -12,6 +12,7 @@ export type SNSSetup = {
   endpoint?: string;
   credentials: AwsCredentialIdentityProvider | AwsCredentialIdentity;
   keepAlive?: boolean;
+  messageAttributes?: boolean;
 };
 
 const SNSProvider = (params: SNSSetup) => {
@@ -36,8 +37,14 @@ export class SNSModule {
 
     return {
       module: SNSModule,
-      providers: [snsProvider],
-      exports: [snsProvider],
+      providers: [
+        snsProvider,
+        {
+          provide: 'SNS_OPTIONS',
+          useValue: params,
+        },
+      ],
+      exports: [snsProvider, 'SNS_OPTIONS'],
     };
   }
 }

package/src/external/aws/sns/sns.service.ts

@@ -3,23 +3,47 @@ import {
   PublishCommandOutput,
   SNSClient,
 } from '@aws-sdk/client-sns';
-import { Inject, Injectable } from '@nestjs/common';
+import { Inject, Injectable, Optional } from '@nestjs/common';
 import { IEvent } from 'src/lib/events/interfaces/event.interface';
 import { EventData, EventSmsData } from 'src/lib/events/types/event.type';
+import { SNSSetup } from './sns.provider';
 
 @Injectable()
 export class Sns implements IEvent {
-  constructor(@Inject('CONNECTION_SNS') private readonly client: SNSClient) {}
+  constructor(
+    @Inject('CONNECTION_SNS') private readonly client: SNSClient,
+    @Optional() @Inject('SNS_OPTIONS') private readonly options?: SNSSetup,
+  ) {}
 
   async publishSns(event: EventData): Promise<PublishCommandOutput> {
+    const messageAttributes = this.getMessageAttributes(event);
+
     const command = new PublishCommand({
       TopicArn: event.topic,
       Message: event.message,
+      ...(messageAttributes && { MessageAttributes: messageAttributes }),
     });
 
     return await this.client.send(command);
   }
 
+  private getMessageAttributes(event: EventData) {
+    if (event.messageAttributes) {
+      return event.messageAttributes;
+    }
+
+    if (this.options?.messageAttributes) {
+      return {
+        enqueuedAt: {
+          DataType: 'Number',
+          StringValue: String(Date.now()),
+        },
+      };
+    }
+
+    return undefined;
+  }
+
   async publishSms(event: EventSmsData): Promise<PublishCommandOutput> {
     const command = new PublishCommand({
       PhoneNumber: event.phoneNumber,

package/src/external/redis/interfaces/redis.interface.ts

@@ -12,6 +12,12 @@ export interface IRedisService {
   deleteRedisKey(key: string): Promise<void>;
 
   createPubSubClient(): Promise<PubSubRedis>;
+
+  setRedisKeyIfNotExists(
+    key: string,
+    value: Date | number | Buffer | string,
+    ttl?: number,
+  ): Promise<boolean>;
 }
 
 export const IRedisService = Symbol('IRedisService');

package/src/external/redis/redis.service.ts

@@ -37,4 +37,21 @@ export class RedisService implements IRedisService {
       sub: this.client.duplicate(),
     };
   }
+
+  async setRedisKeyIfNotExists(
+    key: string,
+    value: Date | number | Buffer | string,
+    ttl?: number,
+  ): Promise<boolean> {
+    if (value instanceof Date) {
+      value = value.toISOString();
+    }
+    let result: string;
+    if (ttl) {
+      result = await this.client.set(key, value, 'EX', ttl, 'NX');
+    } else {
+      result = await this.client.set(key, value, 'NX');
+    }
+    return result === 'OK';
+  }
 }

package/src/index.ts

@@ -34,6 +34,8 @@ import {
   PubSubRedis,
 } from './external/redis/types/ioredis.types';
 
+import { SNSSetup } from './external/aws/sns/sns.provider';
+
 export {
   QueueModule,
   Queue,
@@ -44,6 +46,7 @@ export {
   EventModule,
   Events,
   IEvent,
+  SNSSetup,
   EmailModule,
   Email,
   IEmail,

package/src/lib/cache/cache.ts

@@ -24,4 +24,12 @@ export class Cache implements ICache {
   async createPubSubClient(): Promise<PubSubRedis> {
     return await this.client.createPubSubClient();
   }
+
+  async setIfNotExists(
+    key: string,
+    value: Date | number | Buffer | string,
+    ttl?: number,
+  ): Promise<boolean> {
+    return await this.client.setRedisKeyIfNotExists(key, value, ttl);
+  }
 }

package/src/lib/cache/interfaces/cache.interface.ts

@@ -12,6 +12,12 @@ export interface ICache {
   delete(key: string): Promise<void>;
 
   createPubSubClient(): Promise<PubSubRedis>;
+
+  setIfNotExists(
+    key: string,
+    value: Date | number | Buffer | string,
+    ttl?: number,
+  ): Promise<boolean>;
 }
 
 export const ICache = Symbol('ICache');

package/src/lib/events/types/event.type.ts

@@ -1,6 +1,9 @@
+import { MessageAttributeValue } from '@aws-sdk/client-sns';
+
 export type EventData = {
   topic: string;
   message: string;
+  messageAttributes?: Record<string, MessageAttributeValue>;
 };
 
 export type EventSmsData = {