83 lines
3.1 KiB
TypeScript
83 lines
3.1 KiB
TypeScript
import { asNumber, asRecord } from '@sammo-ts/common';
|
|
import {
|
|
createIncomeActionContext,
|
|
getWarGoldIncome,
|
|
type CityIncomeSource,
|
|
type NationIncomeContext,
|
|
type NationTraitModule,
|
|
type TriggerNationalIncomeType,
|
|
} from '@sammo-ts/logic';
|
|
|
|
import type { InMemoryTurnWorld } from './inMemoryWorld.js';
|
|
import type { MonthlyEventActionHandler } from './monthlyEventHandler.js';
|
|
|
|
const toIncomeCity = (city: ReturnType<InMemoryTurnWorld['listCities']>[number]): CityIncomeSource => ({
|
|
id: city.id,
|
|
population: city.population,
|
|
populationMax: city.populationMax,
|
|
agriculture: city.agriculture,
|
|
agricultureMax: city.agricultureMax,
|
|
commerce: city.commerce,
|
|
commerceMax: city.commerceMax,
|
|
security: city.security,
|
|
securityMax: city.securityMax,
|
|
trust: asNumber(city.meta.trust, 50),
|
|
supplyState: city.supplyState,
|
|
defence: city.defence,
|
|
defenceMax: city.defenceMax,
|
|
wall: city.wall,
|
|
wallMax: city.wallMax,
|
|
meta: asRecord(city.meta),
|
|
});
|
|
|
|
export const createProcessWarIncomeHandler = (options: {
|
|
getWorld: () => InMemoryTurnWorld | null;
|
|
nationTraits?: ReadonlyMap<string, NationTraitModule>;
|
|
}): MonthlyEventActionHandler => {
|
|
return () => {
|
|
const world = options.getWorld();
|
|
if (!world) {
|
|
return;
|
|
}
|
|
|
|
// 레거시는 부상병을 인구로 돌려보내기 전에 각 국가의 공급 도시에서
|
|
// dead/10을 합산한다. 국가 타입의 금 수입 modifier도 도시별로 적용된다.
|
|
const cities = world.listCities();
|
|
for (const nation of world.listNations()) {
|
|
if (nation.level <= 0) {
|
|
continue;
|
|
}
|
|
const trait = options.nationTraits?.get(nation.typeCode);
|
|
const actionContext = createIncomeActionContext(nation);
|
|
const modifyIncome = trait?.onCalcNationalIncome
|
|
? (type: TriggerNationalIncomeType, amount: number) =>
|
|
trait.onCalcNationalIncome!(actionContext, type, amount)
|
|
: undefined;
|
|
const incomeContext: NationIncomeContext = {
|
|
rate: asNumber(nation.meta.rate, 20),
|
|
modifyIncome,
|
|
};
|
|
const income = getWarGoldIncome(
|
|
incomeContext,
|
|
cities.filter((city) => city.nationId === nation.id).map(toIncomeCity)
|
|
);
|
|
world.updateNation(nation.id, { gold: nation.gold + income });
|
|
}
|
|
|
|
// 수입 계산이 끝난 뒤 모든 도시가 부상병 20%를 인구로 회복하고
|
|
// dead를 초기화한다. 레거시는 pop_max로 제한하지 않는다.
|
|
for (const city of cities) {
|
|
const dead = asNumber(city.meta.dead, 0);
|
|
world.updateCity(city.id, {
|
|
// MariaDB rounds the SQL expression when assigning it to
|
|
// Ref's integer pop column (including an exact .5 result).
|
|
population: Math.round(city.population + dead * 0.2),
|
|
meta: {
|
|
...city.meta,
|
|
dead: 0,
|
|
},
|
|
});
|
|
}
|
|
};
|
|
};
|