const Database = require('better-sqlite3');
const path = require('path');
const fs = require('fs');

const DB_DIR = path.join(__dirname, '..', 'database');
const DB_PATH = path.join(DB_DIR, 'gardentrack.db');

// Ensure database directory exists
if (!fs.existsSync(DB_DIR)) {
  fs.mkdirSync(DB_DIR, { recursive: true });
}

let db;
try {
  db = new Database(DB_PATH);
  console.log('Connected to SQLite database');
} catch (err) {
  console.error('Error creating database:', err.message);
  process.exit(1);
}

// Create tables
try {
  // Plants table
  db.exec(`
    CREATE TABLE IF NOT EXISTS plants (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      purchase_location TEXT,
      seedling_age INTEGER,
      type TEXT NOT NULL,
      variety TEXT NOT NULL,
      seedling_height REAL,
      planting_date DATE NOT NULL,
      current_height REAL,
      health_status TEXT DEFAULT 'good' CHECK(health_status IN ('good', 'needs-attention', 'dead')),
      photo_url TEXT,
      current_photo_url TEXT,
      notes TEXT,
      created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
      updated_at DATETIME DEFAULT CURRENT_TIMESTAMP
    )
  `);

  // Plant history table
  db.exec(`
    CREATE TABLE IF NOT EXISTS plant_history (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      plant_id INTEGER NOT NULL,
      year INTEGER NOT NULL,
      blooming_date DATE,
      fruiting_date DATE,
      created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
      updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (plant_id) REFERENCES plants (id) ON DELETE CASCADE
    )
  `);

  // Harvest records table
  db.exec(`
    CREATE TABLE IF NOT EXISTS harvest_records (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      plant_id INTEGER NOT NULL,
      date DATE NOT NULL,
      quantity REAL NOT NULL,
      unit TEXT NOT NULL,
      notes TEXT,
      created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
      updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (plant_id) REFERENCES plants (id) ON DELETE CASCADE
    )
  `);

  // Maintenance records table
  db.exec(`
    CREATE TABLE IF NOT EXISTS maintenance_records (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      plant_id INTEGER NOT NULL,
      date DATE NOT NULL,
      type TEXT NOT NULL CHECK(type IN ('chemical', 'fertilizer', 'watering', 'pruning', 'transplanting', 'other')),
      description TEXT NOT NULL,
      amount TEXT,
      created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
      updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (plant_id) REFERENCES plants (id) ON DELETE CASCADE
    )
  `);

  // Tasks table
  db.exec(`
    CREATE TABLE IF NOT EXISTS tasks (
      id INTEGER PRIMARY KEY AUTOINCREMENT,
      plant_id INTEGER,
      title TEXT NOT NULL,
      description TEXT,
      deadline DATE NOT NULL,
      completed BOOLEAN DEFAULT 0,
      created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
      updated_at DATETIME DEFAULT CURRENT_TIMESTAMP,
      FOREIGN KEY (plant_id) REFERENCES plants (id) ON DELETE SET NULL
    )
  `);

  console.log('Database tables created successfully!');

  // Insert sample data
  console.log('Inserting sample data...');
  
  // Sample plants
  const insertPlant = db.prepare(`
    INSERT INTO plants (purchase_location, seedling_age, type, variety, seedling_height, planting_date, current_height, health_status, photo_url, current_photo_url, notes) 
    VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?, ?, ?)
  `);
  
  insertPlant.run('Local Nursery', 12, 'tree', 'Apple - Honeycrisp', 45.0, '2022-04-15', 180.0, 'good', null, null, 'Growing well, good fruit production');
  insertPlant.run('Garden Center', 8, 'shrub', 'Blueberry - Bluecrop', 25.0, '2023-03-20', 85.0, 'needs-attention', null, null, 'Leaves showing slight discoloration');
  insertPlant.run('Online Store', 3, 'herb', 'Basil - Sweet Genovese', 8.0, '2024-05-10', 35.0, 'good', null, null, 'Producing well, regular harvests');

  // Sample tasks
  const insertTask = db.prepare(`
    INSERT INTO tasks (plant_id, title, description, deadline, completed) 
    VALUES (?, ?, ?, ?, ?)
  `);
  
  insertTask.run(1, 'Apply fertilizer', 'Apply spring fertilizer to apple trees', '2024-03-15', 0);
  insertTask.run(2, 'Prune blueberry bushes', 'Annual pruning before spring growth', '2024-02-28', 1);
  insertTask.run(3, 'Harvest basil', 'Regular harvest to encourage growth', '2024-06-01', 0);

  // Sample maintenance records
  const insertMaintenance = db.prepare(`
    INSERT INTO maintenance_records (plant_id, date, type, description, amount) 
    VALUES (?, ?, ?, ?, ?)
  `);
  
  insertMaintenance.run(1, '2024-01-10', 'pruning', 'Winter pruning - removed dead branches', null);
  insertMaintenance.run(2, '2024-01-05', 'fertilizer', 'Applied organic compost', '2 cups');
  insertMaintenance.run(3, '2024-05-15', 'watering', 'Deep watering during dry spell', '1 gallon');

  // Sample harvest records
  const insertHarvest = db.prepare(`
    INSERT INTO harvest_records (plant_id, date, quantity, unit, notes) 
    VALUES (?, ?, ?, ?, ?)
  `);
  
  insertHarvest.run(1, '2023-09-15', 25, 'lbs', 'Excellent harvest, apples were sweet and crisp');
  insertHarvest.run(2, '2023-07-20', 3, 'cups', 'First harvest of the season, berries were plump and sweet');
  insertHarvest.run(3, '2024-05-25', 0.5, 'cups', 'Fresh basil for cooking, very aromatic');

  // Sample observations
  const insertObservation = db.prepare(`
    INSERT INTO plant_observations (plant_id, date, title, observation, weather_conditions, temperature, tags) 
    VALUES (?, ?, ?, ?, ?, ?, ?)
  `);
  
  insertObservation.run(1, '2024-01-15', 'Winter dormancy check', 'Apple tree showing normal winter dormancy. Buds are tight and healthy looking. No signs of pest damage on bark.', 'Clear, cold', 2.5, 'dormancy,health-check,winter');
  insertObservation.run(2, '2024-01-20', 'Pruning completed', 'Finished annual pruning of blueberry bushes. Removed about 20% of old wood and opened up center for better air circulation.', 'Overcast', 8.0, 'pruning,maintenance');
  insertObservation.run(3, '2024-05-12', 'First true leaves', 'Basil seedlings showing first set of true leaves. Growth is vigorous and color is deep green. Ready for transplanting soon.', 'Sunny', 22.0, 'growth,seedling,transplant-ready');

  console.log('Sample data inserted successfully!');
} catch (err) {
  console.error('Error initializing database:', err);
  process.exit(1);
}

db.close();
console.log('Database initialization completed!');