Merge pull request vercel#56 from erasta/patch-1

ensureTableExists to create users table on the first run

Author: leerob

app/db.ts

@@ -10,19 +10,41 @@ import { genSaltSync, hashSync } from 'bcrypt-ts';
 let client = postgres(`${process.env.POSTGRES_URL!}?sslmode=require`);
 let db = drizzle(client);
 
-let users = pgTable('User', {
-  id: serial('id').primaryKey(),
-  email: varchar('email', { length: 64 }),
-  password: varchar('password', { length: 64 }),
-});
-
 export async function getUser(email: string) {
+  const users = await ensureTableExists();
   return await db.select().from(users).where(eq(users.email, email));
 }
 
 export async function createUser(email: string, password: string) {
+  const users = await ensureTableExists();
   let salt = genSaltSync(10);
   let hash = hashSync(password, salt);
 
   return await db.insert(users).values({ email, password: hash });
 }
+
+async function ensureTableExists() {
+  const result = await client`
+    SELECT EXISTS (
+      SELECT FROM information_schema.tables 
+      WHERE table_schema = 'public' 
+      AND table_name = 'User'
+    );`;
+
+  if (!result[0].exists) {
+    await client`
+      CREATE TABLE "User" (
+        id SERIAL PRIMARY KEY,
+        email VARCHAR(64),
+        password VARCHAR(64)
+      );`;
+  }
+
+  const table = pgTable('User', {
+    id: serial('id').primaryKey(),
+    email: varchar('email', { length: 64 }),
+    password: varchar('password', { length: 64 }),
+  });
+
+  return table;
+}