establish db

2026-06-05 20:30:28 -04:00
parent 820a872f07
commit 58b5804d7a
95 changed files with 42873 additions and 233 deletions
--- a/apps/web/src/lib/db/schema.ts
+++ b/apps/web/src/lib/db/schema.ts
@@ -0,0 +1,104 @@
+/**
+ * Drizzle ORM Schema for the Plant Disease Knowledge Base.
+ *
+ * Uses Turso (libSQL) with SQLite dialect.
+ * Arrays (symptoms, causes, treatment, prevention, lookalike_ids)
+ * are stored as JSON text columns and typed via Drizzle's $type().
+ */
+
+import { sql } from "drizzle-orm";
+import { sqliteTable, text, integer, index } from "drizzle-orm/sqlite-core";
+
+// ─── Plants Table ────────────────────────────────────────────────────────────
+
+export const plants = sqliteTable(
+  "plants",
+  {
+    id: text("id").primaryKey(),
+    commonName: text("common_name").notNull(),
+    scientificName: text("scientific_name").notNull(),
+    family: text("family").notNull(),
+    category: text("category").notNull(),
+    careSummary: text("care_summary").notNull().default(""),
+    imageUrl: text("image_url").notNull().default(""),
+    createdAt: text("created_at")
+      .notNull()
+      .default(sql`(datetime('now'))`),
+    updatedAt: text("updated_at")
+      .notNull()
+      .default(sql`(datetime('now'))`),
+  },
+  (table) => ({
+    categoryIdx: index("idx_plants_category").on(table.category),
+    commonNameIdx: index("idx_plants_common_name").on(table.commonName),
+  }),
+);
+
+// ─── Diseases Table ──────────────────────────────────────────────────────────
+
+export const diseases = sqliteTable(
+  "diseases",
+  {
+    id: text("id").primaryKey(),
+    plantId: text("plant_id")
+      .notNull()
+      .references(() => plants.id),
+    name: text("name").notNull(),
+    scientificName: text("scientific_name").notNull().default(""),
+    causalAgentType: text("causal_agent_type", {
+      enum: ["fungal", "bacterial", "viral", "environmental"],
+    }).notNull(),
+    description: text("description").notNull().default(""),
+    symptoms: text("symptoms", { mode: "json" }).notNull().default([]).$type<string[]>(),
+    causes: text("causes", { mode: "json" }).notNull().default([]).$type<string[]>(),
+    treatment: text("treatment", { mode: "json" }).notNull().default([]).$type<string[]>(),
+    prevention: text("prevention", { mode: "json" }).notNull().default([]).$type<string[]>(),
+    lookalikeIds: text("lookalike_ids", { mode: "json" }).notNull().default([]).$type<string[]>(),
+    severity: text("severity", {
+      enum: ["low", "moderate", "high", "critical"],
+    }).notNull(),
+    sourceUrl: text("source_url").notNull().default(""),
+    createdAt: text("created_at")
+      .notNull()
+      .default(sql`(datetime('now'))`),
+    updatedAt: text("updated_at")
+      .notNull()
+      .default(sql`(datetime('now'))`),
+  },
+  (table) => ({
+    plantIdIdx: index("idx_diseases_plant_id").on(table.plantId),
+    causalAgentIdx: index("idx_diseases_causal_agent").on(table.causalAgentType),
+    severityIdx: index("idx_diseases_severity").on(table.severity),
+  }),
+);
+
+// ─── Scrape Sources Table ────────────────────────────────────────────────────
+
+export const scrapeSources = sqliteTable("scrape_sources", {
+  id: text("id").primaryKey(),
+  sourceType: text("source_type", {
+    enum: ["wikipedia", "university_extension", "cabi", "other"],
+  }).notNull(),
+  sourceUrl: text("source_url").notNull(),
+  lastScrapedAt: text("last_scraped_at"),
+  entriesCount: integer("entries_count").default(0),
+  status: text("status", { enum: ["pending", "success", "error"] })
+    .notNull()
+    .default("pending"),
+  errorMessage: text("error_message"),
+  createdAt: text("created_at")
+    .notNull()
+    .default(sql`(datetime('now'))`),
+});
+
+// ─── Relation Inference ──────────────────────────────────────────────────────
+
+export const plantsRelations = {};
+export const diseasesRelations = {};
+
+// ─── Type helpers ────────────────────────────────────────────────────────────
+
+export type PlantRow = typeof plants.$inferSelect;
+export type PlantInsert = typeof plants.$inferInsert;
+export type DiseaseRow = typeof diseases.$inferSelect;
+export type DiseaseInsert = typeof diseases.$inferInsert;