pruncalc/ts/buy.ts

import {setupPopover} from './popover';

const warehouseNames = {
	'AI1': 'ANT',
	'CI1': 'BEN',
	'IC1': 'HRT',
	'NC1': 'MOR',
} as const;

const username = document.querySelector('#username') as HTMLInputElement;
const apiKey = document.querySelector('#api-key') as HTMLInputElement;
{
	const storedUsername = localStorage.getItem('fio-username');
	if (storedUsername)
		username.value = storedUsername;
	const storedApiKey = localStorage.getItem('fio-api-key');
	if (storedApiKey)
		apiKey.value = storedApiKey;
}
const xitAct = document.querySelector('textarea#xit-act') as HTMLTextAreaElement;
document.querySelector('#fetch')!.addEventListener('click', async () => {
	const supplyForDays = parseInt((document.querySelector('#days') as HTMLInputElement).value, 10);
	const cx = (document.querySelector('#cx') as HTMLInputElement).value;
	if (!(cx in warehouseNames))
		throw new Error(`invalid CX ${cx}`);
	const loader = document.querySelector('#loader') as HTMLElement;
	loader.innerHTML = '';
	loader.style.display = 'block';
	try {
		const output = await calculate(username.value, apiKey.value, supplyForDays, cx as keyof typeof warehouseNames);
		console.log(output);
		localStorage.setItem('fio-username', username.value);
		localStorage.setItem('fio-api-key', apiKey.value);
	} catch (e) {
		console.error(e);
	}
	loader.style.display = 'none';
});
document.querySelector('#copy-xit-act')!.addEventListener('click', () => {
	navigator.clipboard.writeText(xitAct.value);
});
setupPopover();

async function calculate(username: string, apiKey: string, supplyForDays: number, cx: keyof typeof warehouseNames): Promise<void> {
	const tbody = document.querySelector('tbody')!;
	tbody.innerHTML = '';

	const prices = await getPrices(cx);
	const planets = await getPlanets(username, apiKey);
	const avail = await warehouseInventory(username, apiKey, warehouseNames[cx]);
	const {bids, orders} = await getBids(username, apiKey, cx);
	const buy = new Map<string, number>();
	for (const planet of planets) {
		for (const [mat, amount] of planet.supplyForDays(supplyForDays))
			buy.set(mat, (buy.get(mat) ?? 0) + amount);
		for (const mat of planet.exporting) {
			const planetExport = planet.inventory.get(mat);
			if (planetExport)
				avail.set(mat, (avail.get(mat) ?? 0) + planetExport);
		}
	}

	// what's left to buy
	const materials: Material[] = [];
	for (const [mat, amount] of buy) {
		const remaining = Math.max(amount - (bids.get(mat) ?? 0) - (avail.get(mat) ?? 0), 0);
		const price = prices.get(mat);
		if (!price || price.Bid === null || price.Ask === null) {
			console.log(mat, 'has no bid/ask');
			continue;
		}
		const spread = price.Ask - price.Bid;
		materials.push({
			ticker: mat,
			amount,
			bids: bids.get(mat) ?? 0,
			have: avail.get(mat) ?? 0,
			spread,
			savings: spread * remaining,
		});
	}
	materials.sort((a, b) => b.savings - a.savings);

	const toBuy: Record<string, number> = {};
	const priceLimits: Record<string, number> = {};
	const format = new Intl.NumberFormat(undefined, {maximumFractionDigits: 0}).format;
	for (const m of materials) {
		const tr = document.createElement('tr');
		const buyAmount = Math.max(m.amount - m.bids - m.have, 0);
		tr.innerHTML = `
			<td>${m.ticker}</td>
			<td>${format(m.amount)}</td>
			<td>${format(m.bids)}</td>
			<td>${format(m.have)}</td>
			<td>${format(buyAmount)}</td>
			<td>${format(m.spread)}</td>
			<td>${format(m.savings)}</td>
		`;
		if (buyAmount > 0) {
			if (m.bids === 0)
				tr.children[2].classList.add('red');
			toBuy[m.ticker] = buyAmount;
			const bid = prices.get(m.ticker)!.Bid!;
			const epsilon = 10 ** (Math.floor(Math.log10(bid)) - 2);
			const limit = bid + 2 * epsilon;
			priceLimits[m.ticker] = limit;
		}
		tbody.appendChild(tr);
	}

	xitAct.value = JSON.stringify({
		'actions': [
			{'name': 'BuyItems', 'type': 'CX Buy', 'group': 'A1', 'exchange': cx,
				'priceLimits': priceLimits, 'buyPartial': true, 'allowUnfilled': true, 'useCXInv': false},
		],
		'global': {'name': `buy orders for ${supplyForDays} days`},
		'groups': [{'type': 'Manual', 'name': 'A1', 'materials': toBuy}],
	});

	// deposits of current bids
	orders.sort((a, b) => (b.Limit * b.Amount) - (a.Limit * a.Amount));
	for (const order of orders) {
		const deposit = order.Limit * order.Amount;
		console.log(`${order.MaterialTicker.padEnd(4)} ${deposit}\n`);
	}
}

async function getPrices(cx: string) {
	const rawPrices= await cachedFetchJSON('https://refined-prun.github.io/refined-prices/all.json');
	const prices = new Map<string, RawPrice>();
	for (const p of rawPrices)
		if (p.ExchangeCode === cx)
			prices.set(p.MaterialTicker, p);
	return prices;
}

async function getPlanets(username: string, apiKey: string) {
	const fioBurns: FIOBurn[] = await cachedFetchJSON('https://rest.fnar.net/fioweb/burn/user/' + username,
			{headers: {'Authorization': apiKey}});
	const planets = fioBurns.map(burn => new Planet(burn));
	return planets;
}

async function warehouseInventory(username: string, apiKey: string, whName: string): Promise<Map<string, number>> {
	const warehouses: Warehouse[] = await cachedFetchJSON('https://rest.fnar.net/sites/warehouses/' + username,
			{headers: {'Authorization': apiKey}});
	
	const inventory = new Map<string, number>();
	for (const warehouse of warehouses)
		if (warehouse.LocationNaturalId === whName) {
			const storage: Storage = await cachedFetchJSON(`https://rest.fnar.net/storage/${username}/${warehouse.StoreId}`,
					{headers: {'Authorization': apiKey}});
			for (const item of storage.StorageItems)
				inventory.set(item.MaterialTicker, item.MaterialAmount);
			break;
		}
	return inventory;
}

async function getBids(username: string, apiKey: string, cx: string) {
	const allOrders: ExchangeOrder[] = await cachedFetchJSON('https://rest.fnar.net/cxos/' + username,
			{headers: {'Authorization': apiKey}});
	const orders = allOrders.filter(order =>
			order.OrderType === 'BUYING' && order.Status !== 'FILLED' && order.ExchangeCode === cx);
	const bids = new Map<string, number>();
	for (const order of orders)
		bids.set(order.MaterialTicker, (bids.get(order.MaterialTicker) ?? 0) + order.Amount);
	return {bids, orders};
}

const fetchCache = new Map<string, any>();
async function cachedFetchJSON(url: string, options?: RequestInit): Promise<any> {
	if (fetchCache.has(url))
		return fetchCache.get(url);
	const response = await fetch(url, options).then((r) => r.json());
	fetchCache.set(url, response);
	return response;
}

class Planet {
	name: string;
	inventory: Map<string, number>;
	netConsumption: Amount[];
	exporting: Set<string>; // producing more than consumption

	constructor(fioBurn: FIOBurn) {
		this.name = fioBurn.PlanetName || fioBurn.PlanetNaturalId;
		this.inventory = new Map();
		for (const item of fioBurn.Inventory)
			this.inventory.set(item.MaterialTicker, item.MaterialAmount);

		const producing = new Map<string, Amount>();
		for (const item of fioBurn.OrderProduction)
			producing.set(item.MaterialTicker, item);

		this.netConsumption = [];
		for (const c of [...fioBurn.OrderConsumption, ...fioBurn.WorkforceConsumption]) {
			let net = c.DailyAmount;
			const production = producing.get(c.MaterialTicker);
			if (production) {
				net -= production.DailyAmount;
				if (net < 0)
					continue;
			}
			c.netConsumption = net;
			this.netConsumption.push(c);
		}

		const consuming = new Set(this.netConsumption.map(item => item.MaterialTicker));
		this.exporting = new Set<string>();
		for (const item of fioBurn.OrderProduction)
			if (!consuming.has(item.MaterialTicker))
				this.exporting.add(item.MaterialTicker);
	}

	supplyForDays(targetDays: number): Map<string, number> {
		const buy = new Map<string, number>();
		for (const consumption of this.netConsumption) {
			const ticker = consumption.MaterialTicker;
			const avail = this.inventory.get(ticker) ?? 0;
			const dailyConsumption = consumption.netConsumption!;
			const days = avail / dailyConsumption;
			if (days < targetDays)
				buy.set(ticker, Math.ceil((targetDays - days) * dailyConsumption));
		}
		return buy;
	}
}

interface Material {
	ticker: string;
	amount: number;
	bids: number;
	have: number;
	spread: number;
	savings: number;
}

interface RawPrice {
	MaterialTicker: string;
	ExchangeCode: string;
	Bid: number | null;
	Ask: number | null;
	FullTicker: string;
}

interface ExchangeOrder {
	MaterialTicker: string;
	ExchangeCode: string;
	OrderType: 'SELLING' | 'BUYING';
	Status: 'FILLED' | 'PARTIALLY_FILLED';
	Amount: number;
	Limit: number;
}

interface StorageItem {
	MaterialTicker: string;
	MaterialAmount: number;
}

interface Storage {
	Name: string;
	StorageItems: StorageItem[];
	WeightLoad: number;
	VolumeLoad: number;
	Type: 'STORE' | 'WAREHOUSE_STORE' | 'FTL_FUEL_STORE' | 'STL_FUEL_STORE' | 'SHIP_STORE';
}

interface Warehouse {
	StoreId: string;
	LocationNaturalId: string;
}

interface Amount {
	MaterialTicker: string;
	DailyAmount: number;
	netConsumption?: number;
}

interface FIOBurn {
	PlanetName: string;
	PlanetNaturalId: string;
	Error: any;
	OrderConsumption: Amount[];
	WorkforceConsumption: Amount[];
	Inventory: StorageItem[];
	OrderProduction: Amount[];
}