task-manager/server/db.ts

import { drizzle } from 'drizzle-orm/neon-http';
import { neon } from '@neondatabase/serverless';
import * as schema from '@shared/schema';

let db: ReturnType<typeof drizzle> | null = null;

export function getDatabase() {
  if (!db) {
    const databaseUrl = process.env.DATABASE_URL;
    
    if (!databaseUrl) {
      throw new Error('DATABASE_URL environment variable is not set');
    }

    const sql = neon(databaseUrl);
    db = drizzle(sql, { schema });
  }

  return db;
}

export async function initializeDatabase() {
  try {
    const database = getDatabase();
    
    // Create default labels if they don't exist
    const existingLabels = await database.select().from(schema.labels);
    
    if (existingLabels.length === 0) {
      const defaultLabels = [
        { name: "Work", color: "#3B82F6" },
        { name: "Personal", color: "#10B981" },
        { name: "Urgent", color: "#EF4444" },
        { name: "Study", color: "#8B5CF6" },
      ];
      
      await database.insert(schema.labels).values(defaultLabels);
      console.log('✓ Created default labels');
    }
    
    console.log('✓ Database initialized successfully');
  } catch (error) {
    console.error('Failed to initialize database:', error);
    throw error;
  }
}