Increase integer column types to bigint, fix mysql errors (fixes #136)

Author: ffried

standalone/src/db.js

@@ -1,6 +1,6 @@
 import fs from "node:fs";
 import { join } from "node:path";
-import { SQL } from "bun";
+import { SQL, sql } from "bun";
 
 fs.mkdirSync(process.env.DATA_PATH || "./.data", {
   recursive: true,
@@ -13,66 +13,82 @@ async function initDb() {
 
   db = new SQL(dbUrl);
 
+  const isSqlite = db.options.adapter === "sqlite";
+  const isPostgres = db.options.adapter === "postgres";
+  const changeIntToBigInt = async (tbl, col) => {
+    if (isSqlite) return; // Irrelevant in SQLite.
+    if (isPostgres) {
+      await db`alter table ${sql(tbl)} alter column ${sql(col)} type bigint`.simple();
+    } else {
+      await db`alter table ${sql(tbl)} modify column ${sql(col)} bigint`.simple();
+    }
+  };
+  // MySQL requires a prefix-length for indexing text. 4096 is an arbitrarily chosen number.
+  const indexableTextColType = isPostgres || isSqlite ? sql`text` : sql`varchar(4096)`;
+
   await db`create table if not exists sessions (
-    token text primary key not null,
-    expires integer not null,
-    created integer not null
+    token ${indexableTextColType} primary key not null,
+    expires bigint not null,
+    created bigint not null
   )`.simple();
+  await changeIntToBigInt("sessions", "expires");
+  await changeIntToBigInt("sessions", "created");
 
   await db`create table if not exists keys (
-    siteKey text primary key not null,
+    siteKey ${indexableTextColType} primary key not null,
     name text not null,
     secretHash text not null,
     config text not null,
-    created integer not null
+    created bigint not null
   )`.simple();
+  await changeIntToBigInt("keys", "created");
 
   await db`create table if not exists solutions (
-    siteKey text not null,
-    bucket integer not null,
+    siteKey ${indexableTextColType} not null,
+    bucket bigint not null,
     count integer default 0,
     primary key (siteKey, bucket)
   )`.simple();
+  await changeIntToBigInt("solutions", "bucket");
 
   await db`create table if not exists challenges (
-    siteKey text not null,
-    token text not null,
+    siteKey ${indexableTextColType} not null,
+    token ${indexableTextColType} not null,
     data text not null,
-    expires integer not null,
+    expires bigint not null,
     primary key (siteKey, token)
   )`.simple();
+  await changeIntToBigInt("challenges", "expires");
 
   await db`create table if not exists tokens (
-    siteKey text not null,
-    token text not null,
-    expires integer not null,
+    siteKey ${indexableTextColType} not null,
+    token ${indexableTextColType} not null,
+    expires bigint not null,
     primary key (siteKey, token)
   )`.simple();
+  await changeIntToBigInt("tokens", "expires");
 
   await db`create table if not exists api_keys (
-    id text not null,
+    id ${indexableTextColType} not null,
     name text not null,
-    tokenHash text not null,
-    created integer not null,
+    tokenHash ${indexableTextColType} not null,
+    created bigint not null,
     primary key (id, tokenHash)
   )`.simple();
+  await changeIntToBigInt("api_keys", "created");
 
-  setInterval(async () => {
-    const now = Date.now();
-
-    await db`delete from sessions where expires < ${now}`;
-    await db`delete from tokens where expires < ${now}`;
-    await db`delete from challenges where expires < ${now}`;
-  }, 60 * 1000);
+  setInterval(periodicCleanup, 60 * 1000);
+  await periodicCleanup();
 
+  return db;
+}
 
+async function periodicCleanup() {
   const now = Date.now();
 
   await db`delete from sessions where expires < ${now}`;
   await db`delete from tokens where expires < ${now}`;
   await db`delete from challenges where expires < ${now}`;
-
-  return db;
 }
 
 db = await initDb();