Add boolean plugin

Author: holzmaster

src/storage/db/boolean-plugin.ts

@@ -0,0 +1,46 @@
+import {
+    type KyselyPlugin,
+    OperationNodeTransformer,
+    type PluginTransformQueryArgs,
+    type PluginTransformResultArgs,
+    PrimitiveValueListNode,
+    type QueryResult,
+    type RootOperationNode,
+    type UnknownRow,
+    type ValueNode,
+} from "kysely";
+
+// Source:
+// https://github.com/kysely-org/kysely/issues/123#issuecomment-1194184342
+
+export class SqliteBooleanPlugin implements KyselyPlugin {
+    readonly #transformer = new SqliteBooleanTransformer();
+
+    transformQuery(args: PluginTransformQueryArgs): RootOperationNode {
+        return this.#transformer.transformNode(args.node);
+    }
+
+    transformResult(args: PluginTransformResultArgs): Promise<QueryResult<UnknownRow>> {
+        return Promise.resolve(args.result);
+    }
+}
+
+class SqliteBooleanTransformer extends OperationNodeTransformer {
+    override transformValue(node: ValueNode): ValueNode {
+        return {
+            ...super.transformValue(node),
+            value: this.serialize(node.value),
+        };
+    }
+
+    transformPrimitiveValueList(node: PrimitiveValueListNode): PrimitiveValueListNode {
+        return {
+            ...super.transformPrimitiveValueList(node),
+            values: node.values.map(value => this.serialize(value)),
+        };
+    }
+
+    private serialize(value: unknown) {
+        return typeof value === "boolean" ? (value ? 1 : 0) : value;
+    }
+}

src/storage/db/db.ts

@@ -8,6 +8,7 @@ import { captureException, startInactiveSpan } from "@sentry/node";
 
 import type { Database } from "./model.js";
 import datePlugin from "./date-plugin.js";
+import { SqliteBooleanPlugin } from "./boolean-plugin.js";
 import assertNever from "@/utils/assertNever.js";
 import log from "@log";
 
@@ -22,7 +23,7 @@ export async function connectToDb(databasePath: string) {
 
     const nativeDb = new SQLite(databasePath);
     const db = new Kysely<Database>({
-        plugins: [datePlugin],
+        plugins: [datePlugin, new SqliteBooleanPlugin()],
         dialect: new SqliteDialect({
             database: nativeDb,
         }),

src/utils/smoke.test.ts

@@ -43,12 +43,13 @@ describe("database smoke", () => {
         // TODO: Check if there is a builtin way of handling DB lifecycle
         await createDatabase();
         await using _ = defer(closeDatabase);
+
         await db()
             .insertInto("emoteUse")
             .values({
-                channelId: "0",
-                emoteId: 0,
-                messageGuildId: "0",
+                channelId: "13",
+                emoteId: 37,
+                messageGuildId: "42",
                 isReaction: false,
             })
             .execute();