feat(db): add staging

Author: GalvinPython

.env.example

@@ -29,3 +29,4 @@ CONFIG_DISCORD_COMPONENTS_PAGE_SIZE='10' # The number of channels to display per
 # Postgres URLs
 POSTGRES_URL='postgresql://user:password@server:port/database'
 POSTGRES_DEV_URL='postgresql://user:password@server:port/database'
+POSTGRES_STAGING_URL='postgresql://user:password@server:port/database'

drizzle.config.ts

@@ -3,7 +3,9 @@ import { defineConfig } from "drizzle-kit";
 const databaseUrl =
     process.env.NODE_ENV === "development"
         ? process.env.POSTGRES_DEV_URL
-        : process.env.POSTGRES_URL;
+        : process.env.NODE_ENV === "staging"
+          ? process.env.POSTGRES_STAGING_URL
+          : process.env.POSTGRES_URL;
 
 console.log("Using database URL:", databaseUrl);
 

package.json

@@ -24,7 +24,10 @@
   "scripts": {
     "db:generate": "bun drizzle-kit generate",
     "db:push:dev": "cross-env NODE_ENV=development bun drizzle-kit push",
+    "db:push:staging": "cross-env NODE_ENV=staging bun drizzle-kit push",
     "db:push:prod": "bun drizzle-kit push",
+    "db:migrate:staging": "bun src/utils/migratedb.ts --staging",
+    "db:migrate:prod": "bun src/utils/migratedb.ts",
     "dev": "concurrently --names \"WEB,API,BOT\" --prefix-colors \"blue,green,magenta\" \"cd web && bun dev\" \"cd api && cargo watch -x \\\"run -- --dev\\\"\" \"bun --watch . --dev\"",
     "dev:bot": "bun --watch . --dev",
     "test:jetstream": "bun --watch src/utils/bluesky/jetstream.ts",

src/config.ts

@@ -1,5 +1,15 @@
 // FILL IN THIS INFORMATION IN .ENV
 export const runningInDevMode: boolean = process.argv.includes("--dev");
+
+// Staging mode is for only testing the production database before breaking the actual production bot
+// Run `bun db:migrate:staging` to migrate the staging database and check for any mistakes before running `bun db:migrate:prod`
+// Do NOT use this mode for regular testing, use --dev for that
+export const runningInStagingMode: boolean = process.argv.includes("--staging");
+
+if (runningInDevMode && runningInStagingMode) {
+    throw new Error("Cannot run in both dev and staging mode!");
+}
+
 export interface Config {
     youtubeInnertubeProxyUrl: string | null;
     updateIntervalYouTube: number;
@@ -20,7 +30,9 @@ export const config: Config = {
         : 60_000,
     databaseUrl: runningInDevMode
         ? process.env?.POSTGRES_DEV_URL
-        : process.env?.POSTGRES_URL,
+        : runningInStagingMode
+          ? process.env?.POSTGRES_STAGING_URL
+          : process.env?.POSTGRES_URL,
     discordWaitForGuildCacheTime: process.env
         ?.CONFIG_DISCORD_WAIT_FOR_GUILD_CACHE_TIME
         ? parseInt(process.env?.CONFIG_DISCORD_WAIT_FOR_GUILD_CACHE_TIME) * 1000