init

server/utils/postgres.ts

@@ -0,0 +1,87 @@
+import { Client, type QueryResult } from "pg";
+import config from "../config.ts";
+
+export let postgres: Client | undefined = undefined;
+if (config.DATABASE_URL) {
+  postgres = new Client({
+    connectionString: config.DATABASE_URL,
+    ssl: { rejectUnauthorized: false },
+  });
+  postgres.connect();
+}
+
+/**
+ * Use this if we need a new connection instead of sharing.
+ * Guarantees we'll return a client because we throw if we don't have it configured
+ * @returns
+ */
+export function newPostgres() {
+  if (!config.DATABASE_URL) {
+    throw new Error("postgres not configured");
+  }
+  const postgres = new Client({
+    connectionString: config.DATABASE_URL,
+    ssl: { rejectUnauthorized: false },
+  });
+  postgres.connect();
+  return postgres;
+}
+
+export async function updateObject(
+  postgres: Client,
+  table: string,
+  object: AnyDict,
+  condition: AnyDict,
+): Promise<QueryResult<any>> {
+  const columns = Object.keys(object);
+  const values = Object.values(object);
+  // TODO support compound conditions, not just one
+  let query = `UPDATE ${table} SET ${columns
+    .map((c, i) => `"${c}" = $${i + 1}`)
+    .join(",")}
+    WHERE "${Object.keys(condition)[0]}" = $${Object.keys(object).length + 1}
+    RETURNING *`;
+  //console.log(query);
+  const result = await postgres.query(query, [
+    ...values,
+    condition[Object.keys(condition)[0]],
+  ]);
+  return result;
+}
+
+export async function insertObject(
+  postgres: Client,
+  table: string,
+  object: AnyDict,
+): Promise<QueryResult<any>> {
+  const columns = Object.keys(object);
+  const values = Object.values(object);
+  let query = `INSERT INTO ${table} (${columns.map((c) => `"${c}"`).join(",")})
+    VALUES (${values.map((_, i) => "$" + (i + 1)).join(",")})
+    RETURNING *`;
+  // console.log(query);
+  const result = await postgres.query(query, values);
+  return result;
+}
+
+export async function upsertObject(
+  postgres: Client,
+  table: string,
+  object: AnyDict,
+  conflict: BooleanDict,
+): Promise<QueryResult<any>> {
+  const columns = Object.keys(object);
+  const values = Object.values(object);
+  let query = `INSERT INTO ${table} (${columns.map((c) => `"${c}"`).join(",")})
+    VALUES (${values.map((_, i) => "$" + (i + 1)).join(",")})
+    ON CONFLICT (${Object.keys(conflict)
+      .map((k) => `"${k}"`)
+      .join(",")})
+    DO UPDATE SET ${Object.keys(object)
+      .map((c) => `"${c}" = EXCLUDED."${c}"`)
+      .join(",")}
+    RETURNING *`;
+  // console.log(query);
+  const result = await postgres.query(query, values);
+  return result;
+}