import Database from 'better-sqlite3';
import { readFileSync } from 'fs';
import { join } from 'path';

/** 读取 ETL 的 schema.sql 并在给定 db 上建表 + 注入测试数据。*/
export function seedDatabase(db: Database.Database): void {
  const schemaPath = join(
    __dirname,
    '..',
    '..',
    '..',
    '..',
    'app',
    'etl',
    'schema.sql',
  );
  db.exec(readFileSync(schemaPath, 'utf-8'));

  const cat = db.prepare(
    'INSERT INTO category(name, subcat) VALUES (?, ?)',
  );
  const elec = cat.run('电力机车', '').lastInsertRowid as number;
  const emu = cat.run('动车组', '复兴号').lastInsertRowid as number;
  const insp = cat.run('检测车', '普速检测').lastInsertRowid as number;

  const model = db.prepare(
    `INSERT INTO model(category_id, series, model_code, full_name, manufacturer,
       country, country_type, first_year, last_year, status, usage,
       max_speed_value, max_speed_unit, weight_value, weight_unit, raw_json)
     VALUES (@category_id, @series, @model_code, @full_name, @manufacturer,
       @country, @country_type, @first_year, @last_year, @status, @usage,
       @max_speed_value, @max_speed_unit, @weight_value, @weight_unit, @raw_json)`,
  );
  model.run({
    category_id: elec, series: '和谐', model_code: 'HXD1型', full_name: '',
    manufacturer: '中车株洲', country: '中国', country_type: '国产',
    first_year: 2006, last_year: null, status: '现役', usage: '干线货运',
    max_speed_value: 120, max_speed_unit: 'km/h', weight_value: 150,
    weight_unit: 't', raw_json: JSON.stringify({ model_code: 'HXD1型' }),
  });
  model.run({
    category_id: emu, series: '', model_code: 'CR400AF', full_name: '复兴号',
    manufacturer: '中车四方', country: '中国', country_type: '国产',
    first_year: 2017, last_year: null, status: '现役', usage: '高速客运',
    max_speed_value: 350, max_speed_unit: 'km/h', weight_value: null,
    weight_unit: '', raw_json: JSON.stringify({ model_code: 'CR400AF' }),
  });

  db.prepare(
    `INSERT INTO unit(category_id, car_number, model_name, function, depot, status)
     VALUES (?, ?, ?, ?, ?, ?)`,
  ).run(insp, 'SYJZ-0001', '移动式线路动态加载试验车', '试验车', '国铁', '半封存');
}