1 files changed, 50 insertions, 43 deletions

diff --git a/nodejs/automation/contracting.js b/nodejs/automation/contracting.js
index 525253c..c051b03 100644
--- a/nodejs/automation/contracting.js
+++ b/nodejs/automation/contracting.js
@@ -1,55 +1,62 @@
 import * as mining from './mining.js';
+import * as dbContracts from '../database/contracts.js';
 import * as dbShips from '../database/ships.js';
 import * as api from '../lib/api.js';
 import * as contracts from '../lib/contracts.js';
-import * as ships from '../lib/ships.js';
+import * as libShips from '../lib/ships.js';
 import * as systems from '../lib/systems.js';
+import * as utils from './utils.js';
 
-export async function auto(ctx) {
-	let ship = dbShips.getShip(ctx.symbol);
-	// Fetch our contracts in the system the ship currently is in
-	let cs = await contracts.contracts();
-	cs = cs.data.filter(c => c.terms.deliver[0].destinationSymbol.startsWith(ship.nav.systemSymbol));
-	if (cs === []) throw `No contract at ${ctx.symbol}'s location`;
-	let contract = cs[0];
-	if (!contract.accepted) {
-		console.log(new Date(), `accepting contract ${contract.id}`);
-		await contracts.accept({contract: contract.id});
+export async function init() {
+	const cs = dbContracts.getContracts();
+	cs.forEach(contract => run(contract));
+}
+
+async function run(contract) {
+	await contracts.accept({id: contract.id});
+	const contractSystem = utils.waypointToSystem(contract.terms.deliver[0].destinationSymbol);
+	let ships = dbShips.getShipsAt(contractSystem);
+	ships = ships.filter(ship => ship.registration.role !== 'SATELLITE'); // filter out probes
+
+	switch(contract.type) {
+	case 'PROCUREMENT':
+		await runProcurement(contract, ships);
+		break;
+	default:
+		throw `Handling of contract type ${contract.type} is not implemented yet`;
 	}
-	const good = contract.terms.deliver[0].tradeSymbol;
+}
+
+async function runProcurement(contract, ships) {
+	// TODO check if contract is fulfilled!
+	const wantedCargo = contract.terms.deliver[0].tradeSymbol;
 	const deliveryPoint = contract.terms.deliver[0].destinationSymbol;
-	const asteroidFields = await systems.type({symbol: ship.nav.systemSymbol, type: 'ASTEROID_FIELD'});
+	const asteroidFields = await systems.type({symbol: ships[0].nav.systemSymbol, type: 'ASTEROID_FIELD'});
 	const asteroidField = asteroidFields[0].symbol;
-	while (true) {
-		ship = dbShips.getShip(ctx.symbol);
-		// If we are in transit, we wait until we arrive
-		const delay = new Date(ship.nav.route.arrival) - new Date();
-		if (delay > 0) await api.sleep(delay);
-		// Then it depends on where we are
-		let goodCargo = ship.cargo.inventory.filter(i => i.symbol === good)[0];
-		// the switch makes this 'resumable'
-		switch (ship.nav.waypointSymbol) {
-		case asteroidField:
-			let response = await mining.mineUntilFullOf({good: good, symbol: ctx.symbol});
-			await ships.navigate({symbol: ctx.symbol, waypoint: deliveryPoint});
-			break;
-		case deliveryPoint:
-			await ships.dock({symbol: ctx.symbol});
-			await ships.refuel({symbol: ctx.symbol});
-			if (goodCargo !== undefined) {
-				console.log(`delivering ${goodCargo.units} of ${good}`);
-				await contracts.deliver({contract: contract.id, symbol: ctx.symbol, good: good, units: goodCargo.units });
+	ships.forEach(async function(ship) {
+		while (!contract.fulfilled) {
+			ship = dbShips.getShip(ship.symbol);
+			let goodCargo = ship.cargo.inventory.filter(i => i.symbol === wantedCargo)[0];
+			// If we are in transit, we wait until we arrive
+			const delay = new Date(ship.nav.route.arrival) - new Date();
+			if (delay > 0) await api.sleep(delay);
+			// Then it depends on where we are
+			switch (ship.nav.waypointSymbol) {
+			case asteroidField:
+				let response = await mining.mineUntilFullOf({good: wantedCargo, symbol: ship.symbol});
+				await libShips.navigate({symbol: ship.symbol, waypoint: deliveryPoint});
+				break;
+			case deliveryPoint:
+				if (goodCargo !== undefined) { // we could be here if a client restart happens right after selling before we navigate away
+					console.log(`delivering ${goodCargo.units} of ${wantedCargo}`);
+					await contracts.deliver({contract: contract.id, symbol: ship.symbol, good: wantedCargo, units: goodCargo.units });
+					// TODO check if contract is fulfilled!
+				}
+				await libShips.navigate({symbol: ship.symbol, waypoint: asteroidField});
+				break;
+			default:
+				await libShips.navigate({symbol: ship.symbol, waypoint: asteroidField});
 			}
-			await ships.navigate({symbol: ctx.symbol, waypoint: asteroidField});
-			await ships.dock({symbol: ctx.symbol});
-			await ships.refuel({symbol: ctx.symbol});
-			await ships.orbit({symbol: ctx.symbol});
-			break;
-		default:
-			await ships.navigate({symbol: ctx.symbol, waypoint: asteroidField});
-			await ships.dock({symbol: ctx.symbol});
-			await ships.refuel({symbol: ctx.symbol});
-			await ships.orbit({symbol: ctx.symbol});
 		}
-	}
+	});
 }