From c21ae7452b2e046cae948537c019783c611af93b Mon Sep 17 00:00:00 2001
From: gibbyb <gib@gibbyb.com>
Date: Fri, 9 Aug 2024 12:39:53 -0500
Subject: [PATCH] Add to db schema

---
 src/server/db/schema.ts | 130 ++++++++++++++++++++++++++++++++++------
 1 file changed, 113 insertions(+), 17 deletions(-)

diff --git a/src/server/db/schema.ts b/src/server/db/schema.ts
index a13f408..1130e50 100644
--- a/src/server/db/schema.ts
+++ b/src/server/db/schema.ts
@@ -3,8 +3,10 @@ import {
   timestamp,
   pgTable,
   text,
+  pgEnum,
   primaryKey,
   integer,
+  numeric,
 } from "drizzle-orm/pg-core"
 import postgres from "postgres"
 import { drizzle } from "drizzle-orm/postgres-js"
@@ -12,18 +14,29 @@ import type { AdapterAccountType } from "next-auth/adapters"
  
 const connectionString = process.env.DATABASE_URL ?? "";
 const pool = postgres(connectionString, { max: 1 })
- 
 export const db = drizzle(pool)
  
-export const users = pgTable("user", {
-  id: text("id")
-    .primaryKey()
-    .$defaultFn(() => crypto.randomUUID()),
-  name: text("name"),
-  email: text("email").unique(),
-  emailVerified: timestamp("emailVerified", { mode: "date" }),
-  image: text("image"),
-})
+export const frequencyEnum = pgEnum("frequency", ["Monthly", "Bi-weekly", "Weekly"])
+export const workOrderStatusEnum = pgEnum("workOrderStatus", ["Pending", "Open", "Closed"])
+export const preferredDaysofWeekEnum = pgEnum("preferredDaysofWeek", 
+  ["Monday", "Tuesday", "Wednesday", "Thursday", "Friday", "Saturday", "Sunday"])
+export const propertyTypeEnum = pgEnum("propertyType", ["Apartment", "Condominium",
+  "Mobile Home", "Multi-Unit Home", "Single-Family Residence", "Townhouse"])
+
+export const users = pgTable(
+  "user",
+  {
+    id: text("id")
+      .primaryKey()
+      .$defaultFn(() => crypto.randomUUID()),
+    name: text("name"),
+    email: text("email").unique().notNull(),
+    emailVerified: timestamp("emailVerified", { mode: "date" }),
+    image: text("image"),
+    tenantID: text("tenantID").references(() => tenants.id, { onDelete: "cascade" }),
+    stripeCustomerID: text("stripeCustomerID"),
+  }
+)
  
 export const accounts = pgTable(
   "account",
@@ -49,13 +62,16 @@ export const accounts = pgTable(
   })
 )
  
-export const sessions = pgTable("session", {
-  sessionToken: text("sessionToken").primaryKey(),
-  userId: text("userId")
-    .notNull()
-    .references(() => users.id, { onDelete: "cascade" }),
-  expires: timestamp("expires", { mode: "date" }).notNull(),
-})
+export const sessions = pgTable(
+  "session",
+  {
+    sessionToken: text("sessionToken").primaryKey(),
+    userId: text("userId")
+      .notNull()
+      .references(() => users.id, { onDelete: "cascade" }),
+    expires: timestamp("expires", { mode: "date" }).notNull(),
+  }
+)
  
 export const verificationTokens = pgTable(
   "verificationToken",
@@ -91,3 +107,83 @@ export const authenticators = pgTable(
     }),
   })
 )
+
+export const admins = pgTable(
+  "admin",
+  {
+    id: text("id").primaryKey(),
+    userID: text("userID").notNull().references(() => users.id, { onDelete: "cascade" }),
+  }
+)
+
+export const properties = pgTable(
+  "property",
+  {
+    id: text("id").primaryKey(),
+    address1: text("address1").unique().notNull(),
+    address2: text("address2"),
+    city: text("city").notNull(),
+    state: text("state").notNull(),
+    zip: text("zip").notNull(),
+    monthlyRent: numeric("monthlyRent").notNull(),
+    securityDeposit: numeric("securityDeposit"),
+    propertyType: propertyTypeEnum("propertyType").notNull(),
+  }
+)
+
+export const tenants = pgTable(
+  "tenant",
+  {
+    id: text("id").primaryKey(),
+    propertyID: text("propertyID").notNull().references(() => properties.id, { onDelete: "cascade" }),
+  }
+)
+
+export const payments = pgTable(
+  "payment",
+  {
+    id: text("id").primaryKey(),
+    userID: text("userID").notNull().references(() => users.id, { onDelete: "cascade" }),
+    stripePaymentID: text("stripeID"),
+    amount: numeric("amount").notNull(),
+    paymentDate: timestamp("paymentDate").notNull(),
+    status: text("status").notNull(),
+  }
+)
+
+export const autoPayments = pgTable(
+  "autoPayment",
+  {
+    id: text("id").primaryKey(),
+    userID: text("userID").notNull().references(() => users.id, { onDelete: "cascade" }),
+    amount: numeric("amount").notNull(),
+    frequency: frequencyEnum("frequency").notNull(),
+    preferredDayofWeek: preferredDaysofWeekEnum("preferredDayofWeek").notNull(),
+    startDate: timestamp("startDate").notNull(),
+    nextPaymentDate: timestamp("nextPaymentDate").notNull(),
+    stripePaymentMethodID: text("stripePaymentMethodID").notNull(),
+  }
+)
+
+export const workorders = pgTable(
+  "workorder",
+  {
+    id: text("id").primaryKey(),
+    userID: text("userID").notNull().references(() => users.id, { onDelete: "cascade" }),
+    date: timestamp("date").notNull(),
+    status: workOrderStatusEnum("status").notNull().default("Pending"),
+    title: text("title").notNull(),
+    description: text("description"),
+  }
+)
+
+export const documents = pgTable(
+  "document",
+  {
+    id: text("id").primaryKey(),
+    tenantID: text("tenantID").notNull().references(() => tenants.id, { onDelete: "cascade" }),
+    name: text("name").notNull(),
+    type: text("type").notNull(),
+    file: text("file").notNull(),
+  }
+)