chore: provide flexibility for an optional rudderMessage schema

Signed-off-by: Sai Sankeerth <[email protected]>
rudderlabs · Nov 7, 2023 · 4bcc8db · 4bcc8db
1 parent 0f44408
commit 4bcc8db
Showing 1 changed file with 30 additions and 1 deletion.
diff --git a/src/types/index.ts b/src/types/index.ts
@@ -83,9 +83,37 @@ type Destination = {
 type UserTransformationLibrary = {
   VersionID: string;
 };
+
+type RudderTimestamp = Date | string;
+type RudderMessageType =
+  | 'identify'
+  | 'track'
+  | 'page'
+  | 'screen'
+  | 'group'
+  | 'alias'
+  | 'audiencelist';
+
+type RudderMessageSchema = Partial<{
+  userId: string;
+  anonymousId: string;
+  type: RudderMessageType;
+  channel: string;
+  context: object;
+  originalTimestamp: RudderTimestamp;
+  sentAt: RudderTimestamp;
+  timestamp: RudderTimestamp;
+  event: string;
+  integrations: object;
+  messageId: string;
+  properties: object;
+  traits: object;
+  [key: string]: FixMe;
+}>;
+
 type RouterTransformationRequestData = {
   request?: object;
-  message: RudderMessage;
+  message: RudderMessageSchema;
   metadata: Metadata;
   destination: Destination;
 };
@@ -241,4 +269,5 @@ export {
   UserDeletionResponse,
   Destination,
   ComparatorInput,
+  RudderMessageSchema,
 };