[node] Big Contracts lib refactoring

7 files changed, 88 insertions, 139 deletions

diff --git a/nodejs/automation/contracting.ts b/nodejs/automation/contracting.ts
index 9ac5d2f..4648b0d 100644
--- a/nodejs/automation/contracting.ts
+++ b/nodejs/automation/contracting.ts
@@ -1,10 +1,8 @@
 import { debugLog } from '../lib/api.ts';
 import { Ship } from '../lib/ships.ts';
-import { Contract } from '../lib/types.ts';
 import * as mining from './mining.js';
 import * as selling from './selling.js';
-import * as dbContracts from '../database/contracts.ts';
-import * as libContracts from '../lib/contracts.ts';
+import { Contract, getContracts } from '../lib/contracts.ts';
 import * as libSystems from '../lib/systems.ts';
 import * as systems from '../lib/systems.ts';
 import {
@@ -12,7 +10,7 @@ import {
 } from '../lib/utils.ts';
 
 export async function run(ship: Ship): Promise<void> {
-	const contracts = await libContracts.getContracts();
+	const contracts = await getContracts();
 	const active = contracts.filter(function(c) {
 		if (c.fulfilled) return false;
 		const deadline = new Date(c.terms.deadline).getTime();
@@ -29,7 +27,7 @@ export async function run(ship: Ship): Promise<void> {
 
 async function runOne(contract: Contract, ship: Ship): Promise<void> {
 	debugLog(contract);
-	await libContracts.accept(contract);
+	await contract.accept();
 	switch(contract.type) {
 		case 'PROCUREMENT':
 			//if (contract.terms.deliver[0].tradeSymbol.match(/_ORE$/)) {
@@ -58,7 +56,7 @@ async function runOreProcurement(contract: Contract, ship: Ship): Promise<void>
 				break;
 			case deliveryPoint.symbol:
 				if (goodCargo !== undefined) { // we could be here if a client restart happens right after selling before we navigate away
-					contract = await libContracts.deliver(contract, ship);
+					await contract.deliver(ship);
 					if (contract.fulfilled) return;
 				}
 				await ship.navigate(asteroid);
@@ -117,7 +115,7 @@ async function runTradeProcurement(contract: Contract, ship: Ship): Promise<void
 		await ship.purchase(wantedCargo, units);
 		// then make a delivery
 		await ship.navigate(deliveryPoint);
-		contract = await libContracts.deliver(contract, ship);
+		await contract.deliver(ship);
 		if (contract.fulfilled) return;
 	}
 	console.log("runTradeProcurement not implemented");
diff --git a/nodejs/automation/init.ts b/nodejs/automation/init.ts
index 1e8ea72..28163b7 100644
--- a/nodejs/automation/init.ts
+++ b/nodejs/automation/init.ts
@@ -1,14 +1,13 @@
 import * as dbAgents from '../database/agents.ts';
 import * as db from '../database/db.ts';
-import * as dbContracts from '../database/contracts.ts';
 import * as dbTokens from '../database/tokens.ts';
 import {
 	Response,
 } from '../lib/api.ts';
 import {
 	Agent,
-	Contract,
 } from '../lib/types.ts';
+import { Contract } from '../lib/contracts.ts';
 import { Ship } from '../lib/ships.ts';
 import * as libContracts from '../lib/contracts.ts';
 
@@ -31,7 +30,6 @@ export async function init(): Promise<void> {
 		switch(json.error?.code) {
 			case 4111:  // 4111 means the agent symbol has already been claimed so no server reset happened
 				// TODO await agents.agents();
-				await libContracts.getContracts();
 				return;
 			default:
 				throw json;
@@ -40,5 +38,4 @@ export async function init(): Promise<void> {
 	db.reset();
 	dbTokens.addToken(json.data.token);
 	dbAgents.addAgent(json.data.agent);
-	dbContracts.setContract(json.data.contract);
 }
diff --git a/nodejs/automation/mining.ts b/nodejs/automation/mining.ts
index cdfcb78..e7ba62f 100644
--- a/nodejs/automation/mining.ts
+++ b/nodejs/automation/mining.ts
@@ -1,8 +1,7 @@
 import * as selling from './selling.js';
-import * as dbContracts from '../database/contracts.js';
+import { Contract } from '../lib/contracts.js';
 import { Ship } from '../lib/ships.js';
 import {
-	Contract,
 	Waypoint,
 } from '../lib/types.ts';
 import { categorizeCargo } from '../lib/utils.ts';
@@ -11,7 +10,6 @@ export async function mineUntilFullFor(contract: Contract, ship: Ship, asteroid:
 	// TODO find a good asteroid
 	while(true) {
 		await mineUntilFull(ship);
-		contract = dbContracts.getContract(contract.id);
 		const deliver = contract.terms.deliver[0];
 		const cargo = categorizeCargo(ship.cargo, deliver.tradeSymbol);
 		const wantedUnits = Object.values(cargo.wanted).reduce((acc, e) => acc += e, 0);
diff --git a/nodejs/database/contracts.ts b/nodejs/database/contracts.ts
deleted file mode 100644
index 9adb4c8..0000000
--- a/nodejs/database/contracts.ts
+++ /dev/null
@@ -1,26 +0,0 @@
-import { Contract } from '../lib/types.ts';
-import { DbData, db } from './db.ts';
-
-const addContractStatement = db.prepare(`INSERT INTO contracts(data) VALUES (json(?));`);
-const getContractStatement = db.prepare(`SELECT data FROM contracts WHERE data->>'id' = ?;`);
-const getContractsStatement = db.prepare(`SELECT data FROM contracts WHERE data->>'fulfilled' = false;`);
-const updateContractStatement = db.prepare(`UPDATE contracts SET data = json(:data) WHERE data->>'id' = :id;`);
-
-export function getContract(id: string): Contract {
-	const data = getContractStatement.get(id) as DbData|undefined;
-	if (!data) throw `invalid id ${id} in getContract database call`;
-	return JSON.parse(data.data);
-}
-
-export function getContracts(): Array<Contract> {
-	const data = getContractsStatement.all() as Array<DbData>;
-	return data.map(contractData => JSON.parse(contractData.data));
-}
-
-export function setContract(data: Contract): void {
-	const changes = updateContractStatement.run({
-		data: JSON.stringify(data),
-		id: data.id,
-	}).changes;
-	if (changes === 0) addContractStatement.run(JSON.stringify(data));
-}
diff --git a/nodejs/lib/contracts.ts b/nodejs/lib/contracts.ts
index 0582cc7..1b54178 100644
--- a/nodejs/lib/contracts.ts
+++ b/nodejs/lib/contracts.ts
@@ -1,7 +1,6 @@
 import {
 	Agent,
 	Cargo,
-	Contract,
 } from './types.ts';
 import {
 	APIError,
@@ -11,84 +10,92 @@ import {
 } from './api.ts';
 import { Ship } from './ships.ts';
 import * as dbAgents from '../database/agents.ts';
-import * as dbContracts from '../database/contracts.ts';
-
-export async function accept(contract: Contract): Promise<Contract> {
-	contract = dbContracts.getContract(contract.id);
-	if (contract.accepted) return contract;
-	const response = await send<{agent: Agent, contract: Contract, type: ''}>({endpoint: `/my/contracts/${contract.id}/accept`, method: 'POST'});
-	if (response.error) {
-		debugLog(response);
-		throw response;
-	}
-	dbAgents.setAgent(response.data.agent);
-	dbContracts.setContract(response.data.contract);
-	return response.data.contract;
-}
 
 export async function getContracts(): Promise<Array<Contract>> {
 	const response = await sendPaginated<Contract>({endpoint: '/my/contracts'});
-	response.forEach(contract => dbContracts.setContract(contract));
-	return response;
-}
-
-export async function getContract(contract: Contract): Promise<Contract> {
-	try {
-		return dbContracts.getContract(contract.id);
-	} catch {}
-	const response = await send<Contract>({endpoint: `/my/contracts/${contract.id}`});
-	if (response.error) {
-		debugLog(response);
-		throw response;
-	}
-	dbContracts.setContract(response.data);
-	return response.data;
+	return response.map(contract => new Contract(contract));
 }
 
-export async function deliver(contract: Contract, ship: Ship): Promise<Contract> {
-	contract = dbContracts.getContract(contract.id);
-	if (contract.terms.deliver[0].unitsRequired <= contract.terms.deliver[0].unitsFulfilled) {
-		return await fulfill(contract);
+export class Contract {
+	accepted: boolean;
+	deadlineToAccept: Date;
+	expiration: Date;
+	factionSymbol: string;
+	fulfilled: boolean;
+	id: string;
+	terms: {
+		deadline: Date;
+		payment: {
+			onAccepted: number;
+			onFulfilled: number;
+		},
+		deliver: Array<{
+			tradeSymbol: string;
+			destinationSymbol: string;
+			unitsRequired: number;
+			unitsFulfilled: number;
+		}>;
+	};
+	type: string;
+	constructor(contract: Contract) {
+		this.accepted = contract.accepted;
+		this.deadlineToAccept = contract.deadlineToAccept;
+		this.expiration = contract.expiration;
+		this.factionSymbol = contract.factionSymbol;
+		this.fulfilled = contract.fulfilled;
+		this.id = contract.id;
+		this.terms = contract.terms;
+		this.type = contract.type;
 	}
-	const tradeSymbol = contract.terms.deliver[0].tradeSymbol;
-	let units = 0;
-	ship.cargo.inventory.forEach(i => {if (i.symbol === tradeSymbol) units = i.units; });
-	await ship.dock(); // we need to be docked to deliver
-	const response = await send<{contract: Contract, cargo: Cargo}>({ endpoint: `/my/contracts/${contract.id}/deliver`, method: 'POST', payload: {
-		shipSymbol: ship.symbol,
-		tradeSymbol: tradeSymbol,
-		units: units,
-	}});
-	if (response.error) {
-		switch(response.error.code) {
-			case 4503: // contract has expired
-				// TODO sell cargo? the next trading loop should take care of it by itself
-				contract.fulfilled = true;
-				return contract;
-			case 4509: // contract delivery terms have been met
-				return await fulfill(contract);
-			default: // yet unhandled error
-				debugLog(response);
-				throw response;
+	async accept(): Promise<void> {
+		if (this.accepted) return;
+		const response = await send<{agent: Agent, contract: Contract, type: ''}>({endpoint: `/my/contracts/${this.id}/accept`, method: 'POST'});
+		if (response.error) {
+			debugLog(response);
+			throw response;
 		}
+		dbAgents.setAgent(response.data.agent);
 	}
-	dbContracts.setContract(response.data.contract);
-	ship.cargo = response.data.cargo;
-	if(response.data.contract.terms.deliver[0].unitsRequired <= response.data.contract.terms.deliver[0].unitsFulfilled) {
-		return await fulfill(response.data.contract);
+	async deliver(ship: Ship): Promise<void> {
+		const unitsRemaining = this.terms.deliver[0].unitsRequired - this.terms.deliver[0].unitsFulfilled;
+		if (unitsRemaining <= 0) return await this.fulfill();
+		const tradeSymbol = this.terms.deliver[0].tradeSymbol;
+		let units = 0;
+		ship.cargo.inventory.forEach(i => {if (i.symbol === tradeSymbol) units = i.units; });
+		if (units === 0) return;
+		if (units > unitsRemaining) units = unitsRemaining;
+		await ship.dock(); // we need to be docked to deliver
+		const response = await send<{contract: Contract, cargo: Cargo}>({ endpoint: `/my/contracts/${this.id}/deliver`, method: 'POST', payload: {
+			shipSymbol: ship.symbol,
+			tradeSymbol: tradeSymbol,
+			units: units,
+		}});
+		if (response.error) {
+			switch(response.error.code) {
+				case 4503: // contract has expired
+					// TODO sell cargo? the next trading loop should take care of it by itself
+					this.fulfilled = true;
+					return;
+				case 4509: // contract delivery terms have been met
+					return await this.fulfill();
+				default: // yet unhandled error
+					debugLog(response);
+					throw response;
+			}
+		}
+		ship.cargo = response.data.cargo;
+		if(response.data.contract.terms.deliver[0].unitsRequired <= response.data.contract.terms.deliver[0].unitsFulfilled) {
+			return await this.fulfill();
+		}
 	}
-	return response.data.contract;
-}
-
-export async function fulfill(contract: Contract): Promise<Contract> {
-	contract = dbContracts.getContract(contract.id);
-	if (contract.fulfilled) return contract;
-	const response = await send<{agent: Agent, contract: Contract}>({ endpoint: `/my/contracts/${contract.id}/fulfill`, method: 'POST'});
-	if (response.error) {
-		debugLog(response);
-		throw response;
+	async fulfill(): Promise<void> {
+		if (this.terms.deliver[0].unitsRequired > this.terms.deliver[0].unitsFulfilled) return;
+		if (this.fulfilled) return;
+		const response = await send<{agent: Agent, contract: Contract}>({ endpoint: `/my/contracts/${this.id}/fulfill`, method: 'POST'});
+		if (response.error) {
+			debugLog(response);
+			throw response;
+		}
+		dbAgents.setAgent(response.data.agent);
 	}
-	dbAgents.setAgent(response.data.agent);
-	dbContracts.setContract(response.data.contract);
-	return response.data.contract;
-}
+};
diff --git a/nodejs/lib/ships.ts b/nodejs/lib/ships.ts
index 74419fc..8df3aea 100644
--- a/nodejs/lib/ships.ts
+++ b/nodejs/lib/ships.ts
@@ -10,11 +10,11 @@ import {
 	ShipIsStillOnCooldownError,
 	ShipRequiresMoreFuelForNavigationError,
 } from './errors.ts';
+import { Contract } from './contracts.ts';
 import * as libSystems from './systems.ts';
 import {
 	Agent,
 	Cargo,
-	Contract,
 	Cooldown,
 	Fuel,
 	Nav,
@@ -25,7 +25,6 @@ import {
 	shortestPath,
 } from './utils.ts';
 import * as dbAgents from '../database/agents.ts';
-import * as dbContracts from '../database/contracts.ts';
 
 export async function getShips(): Promise<Array<Ship>> {
 	const response = await send<Array<Ship>>({endpoint: `/my/ships`, page: 1});
@@ -154,8 +153,7 @@ export class Ship {
 					throw response;
 			}
 		}
-		dbContracts.setContract(response.data.contract);
-		return response.data.contract;
+		return new Contract(response.data.contract);
 	}
 	async orbit(): Promise<void> {
 		if (this.nav.status === 'IN_ORBIT') return;
diff --git a/nodejs/lib/types.ts b/nodejs/lib/types.ts
index a8e748c..e4d750f 100644
--- a/nodejs/lib/types.ts
+++ b/nodejs/lib/types.ts
@@ -30,29 +30,6 @@ export type Chart = {
 	submittedOn: Date;
 };
 
-export type Contract = {
-	id: string;
-	factionSymbol: string;
-	type: string;
-	terms: {
-		deadline: Date;
-		payment: {
-			onAccepted: number;
-			onFulfilled: number;
-		},
-		deliver: Array<{
-			tradeSymbol: string;
-			destinationSymbol: string;
-			unitsRequired: number;
-			unitsFulfilled: number;
-		}>;
-	};
-	accepted: boolean;
-	fulfilled: boolean;
-	expiration: Date;
-	deadlineToAccept: Date;
-};
-
 export type Cooldown = {
 	shipSymbol: string;
 	totalSeconds: number;