hoi4-agent-tools 4.5.0 → 4.6.0

package/dist/hoi4_agent_tools/map/operations.js

@@ -9,12 +9,14 @@ import { derivedStateCapital, encodeTextDocument, LEVEL_STATE_BUILDINGS, MapWork
 import { planReferenceRemap, ReferenceRemapError } from './references.js';
 import { generateWorld } from './world.js';
 import { resolveEdge } from './world-edge.js';
+import { DEFAULT_AUTO_STRAIT_GAP_PX } from './gameplay/crossings.js';
 import { planMapPositions, planSupplyRebuild } from './derived.js';
 import { buildWorldContext, countryTagCatalog, WORLD_COUNTRY_TAG_FILE, } from './world-context.js';
 import { planWorldOutputs } from './world-outputs.js';
 import { resolvePartitionMode } from './partition.js';
+import { applyAssignTerritory, applyRepartitionStates, } from './territory-operations.js';
 import { unsupportedGeographyParts } from './relief/index.js';
-function* cancellationCheckpoint(signal, iteration = 0, stride = 1) {
+export function* cancellationCheckpoint(signal, iteration = 0, stride = 1) {
     if (iteration % stride !== 0)
         return;
     signal?.throwIfAborted();
@@ -43,7 +45,7 @@ async function completeCooperatively(steps, signal) {
 }
 /** Paths listed per kind in diagnostics and tool results; the artifact lists them all. */
 export const WORLD_OUTPUT_PATH_SAMPLE = 200;
-function normalizeRelative(value) {
+export function normalizeRelative(value) {
     return value.replaceAll('\\', '/').replace(/^\.\//u, '').toLowerCase();
 }
 function numberText(value) {
@@ -61,10 +63,10 @@ function required(value, code, message) {
 function quote(value) {
     return `"${value.replaceAll('\\', '\\\\').replaceAll('"', '\\"')}"`;
 }
-function uniqueSorted(values) {
+export function uniqueSorted(values) {
     return [...new Set(values)].sort((left, right) => left - right);
 }
-function uniqueStrings(values) {
+export function uniqueStrings(values) {
     return [...new Set(values)].sort((left, right) => compareCodeUnits(left, right));
 }
 function lineIndent(text, offset) {
@@ -110,7 +112,7 @@ function blockInsertion(block, documentText, content, newline) {
         description: 'Insert map source assignment',
     };
 }
-function cloneState(state) {
+export function cloneState(state) {
     return {
         id: state.id,
         name: state.name,
@@ -145,7 +147,7 @@ function compileHistory(data, indent, newline) {
     rows.push(...(data.datedHistory ?? []).map((block) => `${indent}\t${block}`));
     return rows.length === 0 ? `{ }` : `{${newline}${rows.join(newline)}${newline}${indent}}`;
 }
-function compileState(data, newline = '\n') {
+export function compileState(data, newline = '\n') {
     const resources = data.resources.size === 0
         ? ''
         : `\tresources = ${renderScalarMap(data.resources, '\t', newline)}${newline}`;
@@ -438,7 +440,7 @@ function patchDefinitionLine(source, updated) {
         replacements.set(7, String(updated.continent));
     return replaceDelimitedFields(line.text, replacements);
 }
-function applyTextRanges(document, replacements) {
+export function applyTextRanges(document, replacements) {
     const ordered = [...replacements].sort((left, right) => left.start - right.start || left.end - right.end);
     let cursor = 0;
     const pieces = [];
@@ -495,7 +497,7 @@ function patchDefinitions(index, updates, removals, additions) {
     }
     return applyTextRanges(document, replacements);
 }
-function rootEvidence(records) {
+export function rootEvidence(records) {
     const groups = new Map();
     for (const record of records) {
         const key = `${record.file.rootKind}:${record.file.loadOrder}`;
@@ -532,7 +534,7 @@ function contiguous(ids, start) {
     }
     return true;
 }
-function maximumObservedId(values, fallback) {
+export function maximumObservedId(values, fallback) {
     let maximum = fallback;
     for (const value of values) {
         if (!Number.isSafeInteger(value)) {
@@ -745,7 +747,7 @@ export function indexWithProposedChanges(base, changes, replacedRoots = []) {
     }
     return MapWorkspaceIndex.build(files, base.sourceRoots, base.sharedIndex.rebuild(files));
 }
-function addChange(changes, relativePath, content, operationId, mediaType) {
+export function addChange(changes, relativePath, content, operationId, mediaType) {
     const key = normalizeRelative(relativePath);
     const current = changes.get(key);
     if (current === undefined) {
@@ -774,10 +776,10 @@ function proposedChanges(changes) {
 }
 /** Source folders a new world replaces, per plan, so lower layers stop counting. */
 /** Diagnostics an operation raises while applying, drained into the plan after it succeeds. */
-const diagnosticsByPlan = new WeakMap();
+export const diagnosticsByPlan = new WeakMap();
 const replacedRootsByPlan = new WeakMap();
 const worldOutputsByPlan = new WeakMap();
-function currentIndex(base, changes) {
+export function currentIndex(base, changes) {
     return indexWithProposedChanges(base, proposedChanges(changes), replacedRootsByPlan.get(changes) ?? []);
 }
 function samePolicy(actual, expected) {
@@ -857,7 +859,7 @@ function requireSplitPolicy(operation) {
     }
     return policy;
 }
-function requireSafeLocalisationDocument(document, relativePath) {
+export function requireSafeLocalisationDocument(document, relativePath) {
     if (document.encoding !== 'utf8-bom') {
         throw new ServiceError('MAP_STATE_LOCALISATION_ENCODING_INVALID', 'State localisation edits require an existing UTF-8 BOM source file', { file: relativePath, encoding: document.encoding });
     }
@@ -866,7 +868,7 @@ function requireSafeLocalisationDocument(document, relativePath) {
         throw new ServiceError('MAP_STATE_LOCALISATION_SOURCE_INVALID', 'State localisation cannot target a malformed source file', { file: relativePath, diagnostics: errors.map(({ code }) => code) });
     }
 }
-function safeLocalisationRelativePath(index, requested) {
+export function safeLocalisationRelativePath(index, requested) {
     const value = requested.replaceAll('\\', '/');
     const segments = value.split('/');
     if (value === '' ||
@@ -884,7 +886,7 @@ function safeLocalisationRelativePath(index, requested) {
     }
     return value;
 }
-function escapedLocalisationValue(value) {
+export function escapedLocalisationValue(value) {
     return value
         .replaceAll('\\', '\\\\')
         .replaceAll('"', '\\"')
@@ -892,7 +894,7 @@ function escapedLocalisationValue(value) {
         .replaceAll('\r', '\\n')
         .replaceAll('\n', '\\n');
 }
-function localisationEntryLine(document, entry, value) {
+export function localisationEntryLine(document, entry, value) {
     const line = document.text.slice(entry.start, entry.end);
     const opening = line.indexOf('"');
     if (opening < 0)
@@ -918,7 +920,7 @@ function localisationEntryLine(document, entry, value) {
         throw new ServiceError('MAP_STATE_LOCALISATION_SOURCE_INVALID', 'State localisation source entry has an unclosed quoted value', { file: document.path, line: entry.line });
     return `${line.slice(0, opening + 1)}${escapedLocalisationValue(value)}${line.slice(closing)}`;
 }
-function appendLocalisationEntry(document, key, value) {
+export function appendLocalisationEntry(document, key, value) {
     let englishHeaders = 0;
     const matcher = /^\s*l_english:\s*(?:#.*)?$/gmu;
     while (matcher.exec(document.text) !== null && englishHeaders < 2)
@@ -1011,7 +1013,7 @@ function splitScalar(total, ratio, policy) {
  * count of slot buildings. A level applies to the whole state, so a split keeps it in both parts
  * and a merge keeps the highest; slot buildings are divided and summed.
  */
-function isLevelStateBuilding(index, building) {
+export function isLevelStateBuilding(index, building) {
     const definition = index.buildingDefinitions.get(building);
     if (definition !== undefined)
         return !definition.sharesSlots && !definition.provincial;
@@ -1079,7 +1081,7 @@ function mergeBounds(left, right) {
         count: left.count + right.count,
     };
 }
-function regionMembershipId(index, provinceId) {
+export function regionMembershipId(index, provinceId) {
     const regions = index.regionForProvince(provinceId);
     if (regions.length !== 1)
         throw new ServiceError('MAP_REGION_MEMBERSHIP_AMBIGUOUS', `Province ${provinceId} must have exactly one strategic region`);
@@ -1091,7 +1093,7 @@ function stateMembershipId(index, provinceId) {
         throw new ServiceError('MAP_STATE_MEMBERSHIP_AMBIGUOUS', `Province ${provinceId} belongs to multiple states`);
     return states[0]?.id;
 }
-function commitState(changes, state, data, fields, operationId, datedRewrite, datedRewriteFollowsProvinces = datedRewrite !== undefined) {
+export function commitState(changes, state, data, fields, operationId, datedRewrite, datedRewriteFollowsProvinces = datedRewrite !== undefined) {
     // Distributed dated history already follows the new province membership.
     assertDatedStateHistorySafe(state, datedRewriteFollowsProvinces
         ? { ...fields, provinces: false, victoryPoints: false, buildings: false }
@@ -1184,7 +1186,7 @@ function distributeDatedHistory(index, state, moved, ratio) {
     }
     return result;
 }
-function assertDatedStateHistorySafe(state, fields, operationId) {
+export function assertDatedStateHistorySafe(state, fields, operationId) {
     if (state.historyBlock === undefined)
         return;
     for (const entry of assignments(state.historyBlock)) {
@@ -1272,14 +1274,13 @@ function exactNumericMap(values, label) {
     }
     return result;
 }
-function applyUpdateState(base, changes, operation) {
-    const index = currentIndex(base, changes);
-    const state = index.statesById.get(operation.stateId);
+/** Apply one state update against an index that already holds every earlier change. */
+export function updateStateAgainst(index, changes, stateId, values, operationId) {
+    const state = index.statesById.get(stateId);
     if (state === undefined)
-        throw new ServiceError('MAP_STATE_NOT_FOUND', `State ${operation.stateId} does not exist`);
+        throw new ServiceError('MAP_STATE_NOT_FOUND', `State ${stateId} does not exist`);
     const desired = cloneState(state);
     const fields = {};
-    const values = operation.changes;
     if (values.capital !== undefined) {
         if (values.victoryPoints === undefined)
             throw new ServiceError('MAP_STATE_CAPITAL_ASSERTION_REQUIRES_VICTORY_POINTS', 'A derived state-capital assertion requires an exact victoryPoints payload');
@@ -1352,7 +1353,25 @@ function applyUpdateState(base, changes, operation) {
         desired.provinceBuildings = provinceBuildings;
         fields.buildings = true;
     }
-    commitState(changes, state, desired, fields, operation.id);
+    commitState(changes, state, desired, fields, operationId);
+}
+function applyUpdateState(base, changes, operation) {
+    updateStateAgainst(currentIndex(base, changes), changes, operation.stateId, operation.changes, operation.id);
+}
+/**
+ * Update many states in one operation: the index is built once and each state is patched in
+ * place, so a politics pass over thousands of states costs one validation. All or nothing.
+ */
+function applyUpdateStates(base, changes, operation) {
+    const seen = new Set();
+    for (const { stateId } of operation.updates) {
+        if (seen.has(stateId))
+            throw new ServiceError('MAP_UPDATE_STATES_DUPLICATE', `State ${stateId} appears twice in updates; merge its changes into one entry`, { stateId });
+        seen.add(stateId);
+    }
+    const index = currentIndex(base, changes);
+    for (const { stateId, changes: values } of operation.updates)
+        updateStateAgainst(index, changes, stateId, values, operation.id);
 }
 function applyMoveStateProvinces(base, changes, operation) {
     const policy = requireMovePolicy(operation);
@@ -1561,7 +1580,7 @@ function requireMergeStatePolicy(operation) {
         throw new ServiceError('MAP_STATE_DISTRIBUTION_REQUIRED', 'State merge requires a complete explicit value, ownership, port, network, position, and province-data distribution policy');
     }
 }
-function removalContent(index, file, label) {
+export function removalContent(index, file, label) {
     const lowerExists = index.sourceFiles.some((candidate) => normalizeRelative(candidate.relativePath) === normalizeRelative(file.relativePath) &&
         candidate.displayPath !== file.displayPath &&
         candidate.loadOrder < file.loadOrder);
@@ -2787,7 +2806,7 @@ function handleSimpleOperation(base, changes, operation) {
         addChange(changes, file.relativePath, content, operation.id, 'text/plain');
     }
 }
-function buildingPositionLine(value) {
+export function buildingPositionLine(value) {
     return [
         value.stateId,
         value.building,
@@ -3317,6 +3336,17 @@ function applyCreateWorld(base, changes, operation, context, signal) {
         throw new ServiceError('MAP_LOOK_CUSTOM_PALETTE_REQUIRED', 'look "custom" needs colormap.palette with at least one class colour');
     if (colormapProfile === 'legacy' && look !== undefined)
         throw new ServiceError('MAP_LOOK_REQUIRES_CALIBRATED_COLORMAP', 'look selects a calibrated palette; remove colormap.profile "legacy" or look');
+    if (!hasGeography && operation.development?.model === 'geography')
+        throw new ServiceError('MAP_DEVELOPMENT_REQUIRES_GEOGRAPHY', 'development.model "geography" plans the economy from a geography model; give geography or geographyPath, or use "legacy"');
+    if (!hasGeography && operation.autoStraits !== undefined && operation.autoStraits !== false)
+        throw new ServiceError('MAP_AUTO_STRAITS_REQUIRES_GEOGRAPHY', 'autoStraits detects channels between the landmasses of a geography world; give geography or geographyPath, or remove autoStraits');
+    const autoStraits = operation.autoStraits === undefined || operation.autoStraits === false
+        ? undefined
+        : {
+            maxGapPx: operation.autoStraits === true
+                ? DEFAULT_AUTO_STRAIT_GAP_PX
+                : (operation.autoStraits.maxGapPx ?? DEFAULT_AUTO_STRAIT_GAP_PX),
+        };
     const workspaceCountries = operation.countries !== undefined && !Array.isArray(operation.countries)
         ? operation.countries
         : undefined;
@@ -3334,6 +3364,12 @@ function applyCreateWorld(base, changes, operation, context, signal) {
     const partitionMode = resolvePartitionMode(operation.partition, hasGeography);
     if (partitionMode === 'legacy' && operation.partitionTuning !== undefined)
         throw new ServiceError('MAP_PARTITION_TUNING_REQUIRES_FEATURE_AWARE', 'partitionTuning applies to the feature-aware partition; set partition to "feature-aware" or remove it');
+    if (operation.synthesisCheck === true && context.synthesisCheck !== true)
+        throw new ServiceError('MAP_SYNTHESIS_CHECK_REQUIRES_PREVIEW', "synthesisCheck checks the model's rivers without writing; call hoi4.map_rewrite with preview: true, or remove synthesisCheck to write the world");
+    if (operation.synthesisCheck === true && !hasGeography)
+        throw new ServiceError('MAP_SYNTHESIS_CHECK_REQUIRES_GEOGRAPHY', 'synthesisCheck runs the relief and river synthesis of a geography model; give geography or geographyPath');
+    if (partitionMode === 'legacy' && operation.borderQuality === 'enforce')
+        throw new ServiceError('MAP_BORDER_QUALITY_ENFORCE_NEEDS_FEATURE_AWARE', 'borderQuality "enforce" needs the feature-aware partition: the legacy partition cuts at chance level against rivers and slopes and would block every write. Set partition to "feature-aware", or use "report"');
     const world = generateWorld({
         startingScenario,
         ...(operation.worldName === undefined ? {} : { worldName: operation.worldName }),
@@ -3408,6 +3444,10 @@ function applyCreateWorld(base, changes, operation, context, signal) {
             }),
         ...(edge.mode === 'none' ? {} : { edge }),
         ...(operation.ambientFrame === true ? { ambientFrame: true } : {}),
+        ...(autoStraits === undefined ? {} : { autoStraits }),
+        ...(operation.development?.model === undefined
+            ? {}
+            : { developmentModel: operation.development.model }),
         colormapProfile,
         ...(partitionMode === 'legacy'
             ? {}
@@ -3423,7 +3463,25 @@ function applyCreateWorld(base, changes, operation, context, signal) {
         ...(operation.colormap?.palette === undefined
             ? {}
             : { colormapPalette: colormapPaletteOverrides(operation.colormap.palette) }),
-    }, buildWorldContext(base), { signal });
+    }, buildWorldContext(base), { signal, ...(operation.synthesisCheck === true ? { riverCheck: true } : {}) });
+    if (operation.synthesisCheck === true)
+        return [
+            ...world.notes.map((note) => ({
+                code: note.code,
+                severity: note.severity,
+                category: 'map',
+                operationId: operation.id,
+                message: note.message,
+                ...(note.details === undefined ? {} : { details: note.details }),
+            })),
+            {
+                code: 'MAP_SYNTHESIS_CHECK_READY',
+                severity: 'info',
+                category: 'map',
+                operationId: operation.id,
+                message: `Every river of the model reaches its mouth over the synthesised relief of a ${world.summary.width}x${world.summary.height} map; nothing was written`,
+            },
+        ];
     const roots = operation.replaceBaseGame === false
         ? []
         : [...WORLD_REPLACED_ROOTS, ...(startingScenario ? STARTING_SCENARIO_REPLACED_ROOTS : [])];
@@ -3449,7 +3507,10 @@ function applyCreateWorld(base, changes, operation, context, signal) {
     for (const file of planMapPositions(generated, { mode: 'replace' }))
         addChange(changes, file.relativePath, file.content, operation.id, 'text/plain');
     const positioned = currentIndex(base, changes);
-    for (const file of planSupplyRebuild(positioned, { mode: 'replace' }))
+    for (const file of planSupplyRebuild(positioned, {
+        mode: 'replace',
+        ...(world.railCost === undefined ? {} : { railCost: world.railCost }),
+    }))
         addChange(changes, file.relativePath, file.content, operation.id, 'text/plain');
     const outputs = planWorldOutputs(base.sourceFiles, proposedChanges(changes), operation.id, mapRoot, base.sourceRoots.states, base.sourceRoots.localisation, { releaseCountryOutputs: workspaceCountries !== undefined });
     for (const file of outputs.changes)
@@ -3644,7 +3705,7 @@ function applyRebuildSupply(base, changes, operation) {
  * Rewrite the script references of one entity kind in the proposed changes. Returns the number
  * of script files changed; an empty result is an error only when the caller asks for it.
  */
-function remapScriptReferences(base, changes, operationId, request) {
+export function remapScriptReferences(base, changes, operationId, request) {
     const index = currentIndex(base, changes);
     let plan;
     try {
@@ -3727,6 +3788,7 @@ function* planMapOperationSteps(base, operations, signal, context = {}) {
     const allocations = [];
     let expectedChangedBounds;
     const manifestOperations = [];
+    const territory = [];
     const worldOutputs = [];
     const ids = new Set();
     for (const [operationIndex, operation] of operations.entries()) {
@@ -3756,6 +3818,7 @@ function* planMapOperationSteps(base, operations, signal, context = {}) {
         ]));
         const allocationsCheckpoint = allocations.length;
         const boundsCheckpoint = expectedChangedBounds;
+        const territoryCheckpoint = territory.length;
         try {
             if (operation.kind === 'create_world') {
                 diagnostics.push(...applyCreateWorld(base, changes, operation, context, signal));
@@ -3766,8 +3829,20 @@ function* planMapOperationSteps(base, operations, signal, context = {}) {
             }
             else if (operation.kind === 'move_state_provinces')
                 applyMoveStateProvinces(base, changes, operation);
+            else if (operation.kind === 'assign_territory') {
+                const outcome = yield* applyAssignTerritory(base, changes, allocations, operation, signal);
+                diagnostics.push(...outcome.diagnostics);
+                territory.push(outcome.report);
+            }
+            else if (operation.kind === 'repartition_states') {
+                const outcome = yield* applyRepartitionStates(base, changes, allocations, operation, signal);
+                diagnostics.push(...outcome.diagnostics);
+                territory.push(outcome.report);
+            }
             else if (operation.kind === 'update_state')
                 applyUpdateState(base, changes, operation);
+            else if (operation.kind === 'update_states')
+                applyUpdateStates(base, changes, operation);
             else if (operation.kind === 'split_state' || operation.kind === 'create_state')
                 yield* applySplitState(base, changes, allocations, operation, signal);
             else if (operation.kind === 'merge_states')
@@ -3807,6 +3882,7 @@ function* planMapOperationSteps(base, operations, signal, context = {}) {
                 changes.set(key, change);
             allocations.length = allocationsCheckpoint;
             expectedChangedBounds = boundsCheckpoint;
+            territory.length = territoryCheckpoint;
             const serviceError = error instanceof ServiceError
                 ? error
                 : new ServiceError('MAP_OPERATION_FAILED', error instanceof Error ? error.message : String(error));
@@ -3840,6 +3916,7 @@ function* planMapOperationSteps(base, operations, signal, context = {}) {
         ...(expectedChangedBounds === undefined ? {} : { expectedChangedBounds }),
         finalIndex,
         operations: manifestOperations,
+        ...(territory.length === 0 ? {} : { territory }),
         ...(worldOutputs.length === 0 ? {} : { worldOutputs }),
     };
 }