import db from "../db";

export function getAllContracts() {
	return db
		.prepare(
			`
			SELECT 
				contract_id,
				contract_number,
				contract_name,
				customer,
				investor,
				date_signed,
				finish_date
			FROM contracts
			ORDER BY contract_number
		`
		)
		.all();
}

export function getContractById(id) {
	return db
		.prepare(
			`
			SELECT * FROM contracts
			WHERE contract_id = ?
		`
		)
		.get(id);
}

export function deleteContract(id) {
	db.prepare("DELETE FROM contracts WHERE contract_id = ?").run(id);
}

export function getContractWithProjects(id) {
	const contract = getContractById(id);
	if (!contract) return null;

	const projects = db
		.prepare(
			`
			SELECT 
				project_id,
				project_name,
				project_number,
				finish_date
			FROM projects
			WHERE contract_id = ?
			ORDER BY project_name
		`
		)
		.all(id);

	return { ...contract, projects };
}