Test date plugin

Author: holzmaster

src/storage/db/date-plugin.ts

@@ -0,0 +1,46 @@
+import type {
+    KyselyPlugin,
+    PluginTransformQueryArgs,
+    PluginTransformResultArgs,
+    QueryResult,
+    RootOperationNode,
+    UnknownRow,
+} from "kysely";
+
+import log from "@log";
+
+export default {
+    transformQuery(args: PluginTransformQueryArgs): RootOperationNode {
+        return args.node;
+    },
+    async transformResult(args: PluginTransformResultArgs): Promise<QueryResult<UnknownRow>> {
+        return {
+            ...args.result,
+            rows: args.result.rows.map(transformRow),
+        };
+    },
+} satisfies KyselyPlugin;
+
+const sqlitePattern = /^\d\d\d\d-\d\d-\d\d \d\d:\d\d:\d\d$/;
+const iso8601Pattern = /^\d\d\d\d-\d\d-\d\dT\d\d:\d\d:\d\d.\\\d\d\d/; // not matching till the end, so we also match `+00:00` timestamps
+function transformRow(row: UnknownRow): UnknownRow {
+    // See: https://github.com/NullDev/CSZ-Bot/issues/504
+
+    const res = { ...row }; // spread all, so we can keep the same hidden class
+
+    for (const [k, v] of Object.entries(row)) {
+        if (typeof v !== "string") {
+            continue;
+        }
+
+        if (
+            (k.endsWith("At") || k.endsWith("Until")) &&
+            (sqlitePattern.test(v) || iso8601Pattern.test(v))
+        ) {
+            log.info(`Would do: "${v}" -> ${new Date(v)}`);
+            // res[k] = new Date(v);
+        }
+    }
+
+    return res;
+}

src/storage/db/db.ts

@@ -3,47 +3,27 @@ import { fileURLToPath } from "node:url";
 import * as fs from "node:fs/promises";
 import * as path from "node:path";
 
-import {
-    type KyselyPlugin,
-    FileMigrationProvider,
-    Kysely,
-    Migrator,
-    type PluginTransformResultArgs,
-    type QueryResult,
-    type UnknownRow,
-    type PluginTransformQueryArgs,
-    type RootOperationNode,
-} from "kysely";
+import { FileMigrationProvider, Kysely, Migrator } from "kysely";
 import { captureException, startInactiveSpan } from "@sentry/bun";
 import { BunSqliteDialect } from "kysely-bun-sqlite";
 
 import type { Database } from "./model.js";
+import datePlugin from "./date-plugin.js";
 import assertNever from "@/utils/assertNever.js";
 import log from "@log";
 
 let kysely: Kysely<Database>;
 
 export default () => kysely;
 
-const plugin = {
-    transformQuery(args: PluginTransformQueryArgs): RootOperationNode {
-        return args.node;
-    },
-    async transformResult(args: PluginTransformResultArgs): Promise<QueryResult<UnknownRow>> {
-        const r = args.result.rows;
-        log.info(r[0], "Row");
-        return args.result;
-    },
-} satisfies KyselyPlugin;
-
 export async function connectToDb(databasePath: string) {
     if (kysely) {
         return;
     }
 
     const nativeDb = new SqliteDatabase(databasePath);
     const db = new Kysely<Database>({
-        plugins: [plugin],
+        plugins: [datePlugin],
         dialect: new BunSqliteDialect({
             database: nativeDb,
         }),

src/storage/db/model.ts

@@ -3,6 +3,10 @@ import type { ColumnType, Generated, GeneratedAlways, Insertable, Selectable } f
 
 export type Date = ColumnType<string, string, string>; // TODO: Date is not supported by DB Driver
 
+// !! IMPORTANT !!
+// When handling dates, use `At` suffix to make sure the sqlite converter can convert it to a Date object
+// !! / IMPORTANT !!
+
 export interface Database {
     birthdays: BirthdayTable;
     stempels: StempelTable;