fix: Move more directories

cloud/packages/core/src/drizzle/index.ts

@@ -0,0 +1,16 @@
+import { Resource } from "sst";
+import postgres from "postgres";
+import { drizzle } from "drizzle-orm/postgres-js";
+
+const client = postgres({
+    idle_timeout: 30000,
+    connect_timeout: 30000,
+    host: Resource.Database.host,
+    database: Resource.Database.database,
+    user: Resource.Database.username,
+    password: Resource.Database.password,
+    port: Resource.Database.port,
+    max: parseInt(process.env.POSTGRES_POOL_MAX || "1"),
+});
+
+export const db = drizzle(client, {});

cloud/packages/core/src/drizzle/transaction.ts

@@ -0,0 +1,63 @@
+import { db } from ".";
+import {
+  PgTransaction,
+  PgTransactionConfig
+} from "drizzle-orm/pg-core";
+import {
+  PostgresJsQueryResultHKT
+} from "drizzle-orm/postgres-js";
+import { ExtractTablesWithRelations } from "drizzle-orm";
+import { createContext } from "../context";
+
+export type Transaction = PgTransaction<
+  PostgresJsQueryResultHKT,
+  Record<string, never>,
+  ExtractTablesWithRelations<Record<string, never>>
+>;
+
+type TxOrDb = Transaction | typeof db;
+
+const TransactionContext = createContext<{
+  tx: Transaction;
+  effects: (() => void | Promise<void>)[];
+}>();
+
+export async function useTransaction<T>(callback: (trx: TxOrDb) => Promise<T>) {
+  try {
+    const { tx } = TransactionContext.use();
+    return callback(tx);
+  } catch {
+    return callback(db);
+  }
+}
+
+export async function afterTx(effect: () => any | Promise<any>) {
+  try {
+    const { effects } = TransactionContext.use();
+    effects.push(effect);
+  } catch {
+    await effect();
+  }
+}
+
+export async function createTransaction<T>(
+  callback: (tx: Transaction) => Promise<T>,
+  isolationLevel?: PgTransactionConfig["isolationLevel"],
+): Promise<T> {
+  try {
+    const { tx } = TransactionContext.use();
+    return callback(tx);
+  } catch {
+    const effects: (() => void | Promise<void>)[] = [];
+    const result = await db.transaction(
+      async (tx) => {
+        return TransactionContext.provide({ tx, effects }, () => callback(tx));
+      },
+      {
+        isolationLevel: isolationLevel || "read committed",
+      },
+    );
+    await Promise.all(effects.map((x) => x()));
+    return result as T;
+  }
+}

cloud/packages/core/src/drizzle/types.ts

@@ -0,0 +1,39 @@
+import { char, timestamp as rawTs } from "drizzle-orm/pg-core";
+
+export const ulid = (name: string) => char(name, { length: 26 + 4 });
+
+export const id = {
+  get id() {
+    return ulid("id").primaryKey().notNull();
+  },
+};
+
+export const teamID = {
+  get id() {
+    return ulid("id").notNull();
+  },
+  get teamID() {
+    return ulid("team_id").notNull();
+  },
+};
+
+export const userID = {
+  get id() {
+    return ulid("id").notNull();
+  },
+  get userID() {
+    return ulid("user_id").notNull();
+  },
+};
+
+export const utc = (name: string) =>
+  rawTs(name, {
+    withTimezone: true,
+    // mode: "date"
+  });
+
+export const timestamps = {
+  timeCreated: utc("time_created").notNull().defaultNow(),
+  timeUpdated: utc("time_updated").notNull().defaultNow(),
+  timeDeleted: utc("time_deleted"),
+};