script par carregar dados dummy no servidor

2026-06-04 18:55:26 +01:00
parent 8f51987342
commit 521987f824
2 changed files with 142 additions and 1 deletions
@@ -0,0 +1,140 @@
+import Database from 'better-sqlite3';
+import fs from 'fs';
+import path from 'path';
+import crypto from 'crypto';
+
+// 1. Load DATABASE_URL from .env
+let dbPath = 'local.db';
+try {
+	const envContent = fs.readFileSync('.env', 'utf8');
+	const match = envContent.match(/^DATABASE_URL=(.+)$/m);
+	if (match && match[1]) {
+		dbPath = match[1].trim().replace(/['"]/g, '');
+	}
+} catch (e) {
+	console.log('Aviso: .env não encontrado ou erro ao ler. Usando local.db por defeito.');
+}
+
+console.log(`A ligar à base de dados em: ${dbPath}`);
+const db = new Database(dbPath);
+
+try {
+	// Começar transação
+	db.transaction(() => {
+		// 1. Limpar a tabela de entregas (deliveries)
+		console.log('A limpar tabela de entregas (deliveries)...');
+		db.prepare('DELETE FROM deliveries').run();
+
+		// 2. Garantir que existe pelo menos um turno (shift)
+		let shift = db.prepare("SELECT * FROM shifts LIMIT 1").get();
+		if (!shift) {
+			console.log('Nenhum turno encontrado. A criar turno padrão T1...');
+			const nowTs = Date.now();
+			const shiftId = crypto.randomUUID();
+			db.prepare(`
+				INSERT INTO shifts (id, code, start_time, end_time, days, created_at, updated_at)
+				VALUES (?, 'T1', '14:30', '16:30', '2,4', ?, ?)
+			`).run(shiftId, nowTs, nowTs);
+			
+			shift = db.prepare("SELECT * FROM shifts LIMIT 1").get();
+		}
+
+		if (!shift) {
+			throw new Error('Não foi possível encontrar ou criar um turno padrão.');
+		}
+		console.log(`Turno utilizado: ${shift.code} (ID: ${shift.id})`);
+
+		// 3. Garantir que os beneficiários de teste #1, #2, #3 existem e estão ativos/parent
+		const testCreationTime = Date.now() - 35 * 24 * 60 * 60 * 1000;
+		for (const num of [1, 2, 3]) {
+			const existing = db.prepare('SELECT * FROM beneficiaries WHERE number = ?').get(num);
+
+			if (!existing) {
+				const bId = crypto.randomUUID();
+				console.log(`A criar Beneficiário Teste #${num}...`);
+				db.prepare(`
+					INSERT INTO beneficiaries (id, number, name, contact, household_size, status, is_parent, created_at, updated_at)
+					VALUES (?, ?, ?, ?, ?, 'ativo', 1, ?, ?)
+				`).run(
+					bId,
+					num,
+					`Beneficiário Teste #${num}`,
+					`91234560${num}`,
+					num,
+					testCreationTime,
+					Date.now()
+				);
+			} else {
+				console.log(`A atualizar Beneficiário Teste #${num}...`);
+				db.prepare(`
+					UPDATE beneficiaries
+					SET status = 'ativo', is_parent = 1, created_at = ?, updated_at = ?
+					WHERE id = ?
+				`).run(testCreationTime, Date.now(), existing.id);
+			}
+		}
+
+		// 4. Buscar todos os beneficiários ativos que são chefes de família (is_parent = 1)
+		const activeBeneficiaries = db.prepare(`
+			SELECT * FROM beneficiaries
+			WHERE status = 'ativo' AND is_parent = 1
+		`).all();
+
+		console.log(`Encontrados ${activeBeneficiaries.length} beneficiários ativos (chefes de família).`);
+
+		// 5. Calcular as terças (2) e quintas (4) nos últimos 30 dias
+		const dates = [];
+		const today = new Date();
+		const todayStr = today.toLocaleDateString('sv-SE', { timeZone: 'Europe/Lisbon' });
+
+		for (let i = 1; i <= 30; i++) {
+			const d = new Date();
+			d.setDate(d.getDate() - i);
+			const dayOfWeek = d.getDay();
+			
+			if (dayOfWeek === 2 || dayOfWeek === 4) {
+				const dateStr = d.toLocaleDateString('sv-SE', { timeZone: 'Europe/Lisbon' });
+				if (dateStr < todayStr) {
+					dates.push(dateStr);
+				}
+			}
+		}
+
+		dates.sort();
+
+		if (dates.length === 0) {
+			throw new Error('Não foram encontradas datas de entregas válidas nos últimos 30 dias.');
+		}
+
+		console.log(`Datas calculadas para entregas (Terças/Quintas dos últimos 30 dias):`, dates);
+
+		// 6. Inserir entregas de teste
+		let insertedCount = 0;
+		for (const b of activeBeneficiaries) {
+			let datesToInsert = [...dates];
+
+			// Logica para criar diferentes níveis de ausências nos dados de teste
+			if (b.number === 1) {
+				datesToInsert = dates.slice(1);
+			} else if (b.number === 2) {
+				datesToInsert = dates.slice(2);
+			} else if (b.number === 3) {
+				datesToInsert = dates.slice(3);
+			}
+
+			for (const dateStr of datesToInsert) {
+				const deliveryId = crypto.randomUUID();
+				db.prepare(`
+					INSERT INTO deliveries (id, beneficiary_id, shift_id, date, created_at)
+					VALUES (?, ?, ?, ?, ?)
+				`).run(deliveryId, b.id, shift.id, dateStr, Date.now());
+				insertedCount++;
+			}
+		}
+
+		console.log(`Sucesso! Foram geradas ${insertedCount} entregas para os beneficiários ativos.`);
+	})();
+} catch (err) {
+	console.error('Erro ao gerar dados de teste:', err);
+	process.exit(1);
+}