Using .env for secrets #4

src/app/api/db.ts

@@ -1,11 +1,12 @@
+import { envEnsure, envNumber } from "@/lib/env";
 import { Pool } from "pg";
 
 const pool = new Pool({
-  user: "klestiselimaj",
-  host: "localhost",
+  user: envEnsure("DB_USER"),
+  host: envEnsure("DB_HOST"),
   database: "postgres",
-  password: "",
-  port: 5432,
+  password: envEnsure("DB_PASSWORD"),
+  port: envNumber("DB_PORT") || 5432,
 });
 
 const createUsersTableQuery = `

src/lib/env.ts

@@ -0,0 +1,32 @@
+export function env(name: string): string | undefined {
+  return process.env[name];
+}
+
+export function envEnsure(name: string): string {
+  const v = process.env[name];
+
+  if (v === undefined) {
+    throw new Error(
+      `Environment variable '${name}' is required but doesn't exist`
+    );
+  }
+
+  return v;
+}
+
+export function envNumber(name: string): number | undefined {
+  const v = process.env[name];
+  return v ? parseInt(v) : undefined;
+}
+
+export function envNumberEnsure(name: string): number {
+  const v = process.env[name];
+
+  if (v === undefined) {
+    throw new Error(
+      `Environment variable '${name}' is required but doesn't exist`
+    );
+  }
+
+  return parseInt(v);
+}