first commit

backend/src/db/index.ts

@@ -0,0 +1,33 @@
+import { drizzle as drizzleSqlite } from 'drizzle-orm/better-sqlite3';
+import { drizzle as drizzlePg } from 'drizzle-orm/node-postgres';
+import Database from 'better-sqlite3';
+import pg from 'pg';
+import * as schema from './schema.js';
+import { existsSync, mkdirSync } from 'fs';
+import { dirname } from 'path';
+
+const dbType = process.env.DB_TYPE || 'sqlite';
+
+let db: ReturnType<typeof drizzleSqlite> | ReturnType<typeof drizzlePg>;
+
+if (dbType === 'postgres') {
+  const pool = new pg.Pool({
+    connectionString: process.env.DATABASE_URL || 'postgresql://localhost:5432/spanglish',
+  });
+  db = drizzlePg(pool, { schema });
+} else {
+  const dbPath = process.env.DATABASE_URL || './data/spanglish.db';
+  
+  // Ensure data directory exists
+  const dir = dirname(dbPath);
+  if (!existsSync(dir)) {
+    mkdirSync(dir, { recursive: true });
+  }
+  
+  const sqlite = new Database(dbPath);
+  sqlite.pragma('journal_mode = WAL');
+  db = drizzleSqlite(sqlite, { schema });
+}
+
+export { db };
+export * from './schema.js';