const { getPool } = require('./partnerModel');

async function findAll({ company_id, date_from, date_to, status, page = 1, limit = 50 } = {}) {
  const db = getPool();
  let sql = `SELECT ps.*, pc.company_name, su.name AS registered_by_name
             FROM partner_schedules ps
             LEFT JOIN partner_companies pc ON ps.company_id = pc.id
             LEFT JOIN sso_users su ON ps.registered_by = su.user_id
             WHERE 1=1`;
  const params = [];
  if (company_id) { sql += ' AND ps.company_id = ?'; params.push(company_id); }
  if (date_from) { sql += ' AND ps.work_date >= ?'; params.push(date_from); }
  if (date_to) { sql += ' AND ps.work_date <= ?'; params.push(date_to); }
  if (status) { sql += ' AND ps.status = ?'; params.push(status); }
  sql += ' ORDER BY ps.work_date DESC, ps.created_at DESC';
  const offset = (page - 1) * limit;
  sql += ' LIMIT ? OFFSET ?';
  params.push(limit, offset);
  const [rows] = await db.query(sql, params);
  return rows;
}

async function findById(id) {
  const db = getPool();
  const [rows] = await db.query(
    `SELECT ps.*, pc.company_name, su.name AS registered_by_name
     FROM partner_schedules ps
     LEFT JOIN partner_companies pc ON ps.company_id = pc.id
     LEFT JOIN sso_users su ON ps.registered_by = su.user_id
     WHERE ps.id = ?`, [id]);
  return rows[0] || null;
}

async function findByCompanyToday(companyId) {
  const db = getPool();
  const [rows] = await db.query(
    `SELECT ps.*, pc.company_name
     FROM partner_schedules ps
     LEFT JOIN partner_companies pc ON ps.company_id = pc.id
     WHERE ps.company_id = ? AND ps.work_date = CURDATE()
     ORDER BY ps.created_at DESC`, [companyId]);
  return rows;
}

async function create(data) {
  const db = getPool();
  const [result] = await db.query(
    `INSERT INTO partner_schedules (company_id, work_date, work_description, workplace_name, expected_workers, registered_by, notes)
     VALUES (?, ?, ?, ?, ?, ?, ?)`,
    [data.company_id, data.work_date, data.work_description || null,
     data.workplace_name || null, data.expected_workers || null,
     data.registered_by, data.notes || null]);
  return findById(result.insertId);
}

async function update(id, data) {
  const db = getPool();
  const fields = [];
  const values = [];
  if (data.company_id !== undefined) { fields.push('company_id = ?'); values.push(data.company_id); }
  if (data.work_date !== undefined) { fields.push('work_date = ?'); values.push(data.work_date); }
  if (data.work_description !== undefined) { fields.push('work_description = ?'); values.push(data.work_description || null); }
  if (data.workplace_name !== undefined) { fields.push('workplace_name = ?'); values.push(data.workplace_name || null); }
  if (data.expected_workers !== undefined) { fields.push('expected_workers = ?'); values.push(data.expected_workers || null); }
  if (data.notes !== undefined) { fields.push('notes = ?'); values.push(data.notes || null); }
  if (data.status !== undefined) { fields.push('status = ?'); values.push(data.status); }
  if (fields.length === 0) return findById(id);
  values.push(id);
  await db.query(`UPDATE partner_schedules SET ${fields.join(', ')} WHERE id = ?`, values);
  return findById(id);
}

async function updateStatus(id, status) {
  const db = getPool();
  await db.query('UPDATE partner_schedules SET status = ? WHERE id = ?', [status, id]);
  return findById(id);
}

async function deleteSchedule(id) {
  const db = getPool();
  await db.query('DELETE FROM partner_schedules WHERE id = ?', [id]);
}

module.exports = { findAll, findById, findByCompanyToday, create, update, updateStatus, deleteSchedule };