127 lines
5.1 KiB
TypeScript
127 lines
5.1 KiB
TypeScript
/**
|
|
* 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[]>(),
|
|
prevalence: text("prevalence", {
|
|
enum: ["common", "uncommon", "rare"],
|
|
})
|
|
.notNull()
|
|
.default("uncommon"),
|
|
severity: text("severity", {
|
|
enum: ["low", "moderate", "high", "critical"],
|
|
}).notNull(),
|
|
imageUrl: text("image_url").notNull().default(""),
|
|
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),
|
|
prevalenceIdx: index("idx_diseases_prevalence").on(table.prevalence),
|
|
}),
|
|
);
|
|
|
|
// ─── 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'))`),
|
|
});
|
|
|
|
// ─── Plant Views Table ───────────────────────────────────────────────────────
|
|
|
|
export const plantViews = sqliteTable(
|
|
"plant_views",
|
|
{
|
|
plantId: text("plant_id")
|
|
.primaryKey()
|
|
.references(() => plants.id),
|
|
viewCount: integer("view_count").notNull().default(0),
|
|
},
|
|
(table) => ({
|
|
viewCountIdx: index("idx_plant_views_count").on(table.viewCount),
|
|
}),
|
|
);
|
|
|
|
// ─── 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;
|