@remit/drizzle-service 0.0.11 → 0.0.12

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@remit/drizzle-service",
-  "version": "0.0.11",
+  "version": "0.0.12",
   "description": "Drizzle ORM service parameterized by dialect (Postgres / SQLite)",
   "type": "module",
   "main": "src/index.ts",

package/src/repos/thread-message.sqlite.test.ts

@@ -212,6 +212,76 @@ describe("DrizzleThreadMessageRepository (sqlite)", () => {
 		);
 	});
 
+	test("listByThread defaults to oldest first, by the date the mail was sent", async () => {
+		const acct = "acct-conversation-dates";
+		const threadId = "t-conversation-dates";
+		const base = Date.now();
+		// The folder a message was delivered to decides its internalDate, so a
+		// reply synced from Sent can carry an earlier internalDate than the
+		// message it answers. sentDate is what the conversation is ordered by.
+		const turns = [
+			{ messageId: "m-question", sentDate: base, internalDate: base + 5000 },
+			{ messageId: "m-answer", sentDate: base + 1000, internalDate: base },
+		];
+		for (const turn of turns) {
+			await repo.create(
+				makeInput({
+					accountConfigId: acct,
+					threadId,
+					messageId: turn.messageId,
+					sentDate: turn.sentDate,
+					internalDate: turn.internalDate,
+				}),
+			);
+		}
+
+		const result = await repo.listByThread(threadId, acct, {
+			excludeDeleted: true,
+		});
+		assert.deepEqual(
+			result.items.map((item) => item.messageId),
+			["m-question", "m-answer"],
+			"the reply follows the message it answers",
+		);
+	});
+
+	test("listByThread paginates in sentDate order without skipping or repeating", async () => {
+		const acct = "acct-conversation-pages";
+		const threadId = "t-conversation-pages";
+		const base = Date.now();
+		for (let index = 0; index < 5; index++) {
+			await repo.create(
+				makeInput({
+					accountConfigId: acct,
+					threadId,
+					messageId: `m-page-${index}`,
+					sentDate: base + index * 1000,
+					internalDate: base,
+				}),
+			);
+		}
+
+		const seen: string[] = [];
+		let continuationToken: string | undefined;
+		do {
+			const page = await repo.listByThread(threadId, acct, {
+				limit: 2,
+				continuationToken,
+				excludeDeleted: true,
+			});
+			seen.push(...page.items.map((item) => item.messageId));
+			continuationToken = page.continuationToken;
+		} while (continuationToken);
+
+		assert.deepEqual(seen, [
+			"m-page-0",
+			"m-page-1",
+			"m-page-2",
+			"m-page-3",
+			"m-page-4",
+		]);
+	});
+
 	test("listByThread excludes soft-deleted messages but keeps the rest of the conversation", async () => {
 		const acct = "acct-conversation-deleted";
 		const threadId = "t-conversation-deleted";

package/src/repos/thread-message.ts

@@ -56,7 +56,6 @@ export const clampThreadSearchLimit = (limit?: number): number => {
 // ─── Cursor ──────────────────────────────────────────────────────────────────
 
 type DateCursor = { s: number; id: string };
-type ThreadCursor = { d: number; id: string };
 type AccountCursor = { id: string };
 
 function encodeDateCursor(sentDate: number, threadMessageId: string): string {
@@ -73,23 +72,6 @@ function decodeDateCursor(token: string): DateCursor | null {
 	}
 }
 
-function encodeThreadCursor(
-	internalDate: number,
-	threadMessageId: string,
-): string {
-	return Buffer.from(
-		JSON.stringify({ d: internalDate, id: threadMessageId }),
-	).toString("base64");
-}
-
-function decodeThreadCursor(token: string): ThreadCursor | null {
-	try {
-		return JSON.parse(Buffer.from(token, "base64").toString()) as ThreadCursor;
-	} catch {
-		return null;
-	}
-}
-
 function encodeAccountCursor(threadMessageId: string): string {
 	return Buffer.from(JSON.stringify({ id: threadMessageId })).toString(
 		"base64",
@@ -563,22 +545,22 @@ export class DrizzleThreadMessageRepository
 	): Promise<ResultList<ThreadMessageItem>> {
 		const order = options?.order ?? "asc";
 		const cursor = options?.continuationToken
-			? decodeThreadCursor(options.continuationToken)
+			? decodeDateCursor(options.continuationToken)
 			: null;
 
 		const cursorCond = cursor
 			? order === "asc"
 				? or(
-						gt(threadMessageTable.internalDate, cursor.d),
+						gt(threadMessageTable.sentDate, cursor.s),
 						and(
-							eq(threadMessageTable.internalDate, cursor.d),
+							eq(threadMessageTable.sentDate, cursor.s),
 							gt(threadMessageTable.threadMessageId, cursor.id),
 						),
 					)
 				: or(
-						lt(threadMessageTable.internalDate, cursor.d),
+						lt(threadMessageTable.sentDate, cursor.s),
 						and(
-							eq(threadMessageTable.internalDate, cursor.d),
+							eq(threadMessageTable.sentDate, cursor.s),
 							gt(threadMessageTable.threadMessageId, cursor.id),
 						),
 					)
@@ -599,8 +581,8 @@ export class DrizzleThreadMessageRepository
 			)
 			.orderBy(
 				order === "asc"
-					? asc(threadMessageTable.internalDate)
-					: desc(threadMessageTable.internalDate),
+					? asc(threadMessageTable.sentDate)
+					: desc(threadMessageTable.sentDate),
 				asc(threadMessageTable.threadMessageId),
 			)
 			.limit(options?.limit ?? 100);
@@ -612,7 +594,7 @@ export class DrizzleThreadMessageRepository
 			items: rows.map(toItem),
 			continuationToken:
 				hasMore && lastRow
-					? encodeThreadCursor(lastRow.internalDate, lastRow.threadMessageId)
+					? encodeDateCursor(lastRow.sentDate, lastRow.threadMessageId)
 					: undefined,
 		};
 	}