feat: allow numeric timestamp in listMessages

xmtp · Oct 27, 2023 · 98de258 · 98de258
1 parent c262034
commit 98de258
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 10 deletions.
diff --git a/src/index.ts b/src/index.ts
@@ -117,16 +117,16 @@ export async function listMessages(
   clientAddress: string,
   conversationTopic: string,
   limit?: number | undefined,
-  before?: Date | undefined,
-  after?: Date | undefined,
+  before?: number | Date | undefined,
+  after?: number | Date | undefined,
   direction?: "SORT_DIRECTION_ASCENDING" | "SORT_DIRECTION_DESCENDING" | undefined,
 ): Promise<DecodedMessage[]> {
   const messages = await XMTPModule.loadMessages(
     clientAddress,
     conversationTopic,
     limit,
-    before?.getTime(),
-    after?.getTime(),
+    typeof before === 'number' ? before : before?.getTime(),
+    typeof after === 'number' ? after : after?.getTime(),
     direction || "SORT_DIRECTION_DESCENDING",
   );
   return messages.map((json: string) => {
@@ -142,8 +142,8 @@ export async function listBatchMessages(
     return JSON.stringify({
       limit: item.pageSize || 0,
       topic: item.contentTopic,
-      after: item.startTime?.getTime() || 0,
-      before: item.endTime?.getTime() || 0,
+      after: (typeof item.startTime === 'number' ? item.startTime :  item.startTime?.getTime()) || 0,
+      before: (typeof item.endTime === 'number' ? item.endTime :  item.endTime?.getTime()) || 0,
       direction: item.direction || "SORT_DIRECTION_DESCENDING",
     });
   });

diff --git a/src/lib/Conversation.ts b/src/lib/Conversation.ts
@@ -39,8 +39,8 @@ export class Conversation {
   // TODO: Support pagination and conversation ID here
   async messages(
     limit?: number | undefined,
-    before?: Date | undefined,
-    after?: Date | undefined,
+    before?: number | Date | undefined,
+    after?: number | Date | undefined,
     direction?: "SORT_DIRECTION_ASCENDING" | "SORT_DIRECTION_DESCENDING" | undefined,
   ): Promise<DecodedMessage[]> {
     try {

diff --git a/src/lib/Query.ts b/src/lib/Query.ts
@@ -1,6 +1,6 @@
 export type Query = {
-  startTime?: Date | undefined;
-  endTime?: Date | undefined;
+  startTime?: number | Date | undefined;
+  endTime?: number | Date | undefined;
   contentTopic: string;
   pageSize?: number | undefined;
   direction?: "SORT_DIRECTION_ASCENDING" | "SORT_DIRECTION_DESCENDING" | undefined;