Files
worldmonitor/scripts/seed-spr-policies.mjs
Elie Habib 1af73975b9 feat(energy): SPR policy classification layer (#2881)
* feat(energy): add SPR policy classification layer with 66-country registry

Static JSON registry classifying strategic petroleum reserve regimes for
66 countries (all IEA members + major producers/consumers). Integrates
into energy profile handler, shock model limitations, analyst context,
spine seeder, and CDP UI.

- scripts/data/spr-policies.json: 66-entry registry with regime, source, asOf
- scripts/seed-spr-policies.mjs: seeder following chokepoint-baselines pattern
- Proto fields 51-59 on GetCountryEnergyProfileResponse
- Handler reads SPR registry from Redis, populates proto fields
- Shock model adds fuel-mode-gated SPR limitations for non-IEA gov SPR
- Analyst context refactored to accumulator pattern (IEA + SPR parts)
- CDP UI: SPR badge for non-IEA government_spr, muted text for spare_capacity
- Spine integration: SPR fields in shockInputs + hasSprPolicy coverage flag
- Cache keys, health, bootstrap, seed-health registrations
- Tests: registry shape, ISO2, regime enum, required entries, no estimatedFillPct

* fix(energy): remove SPR from bootstrap (server-only); narrow SPR hasAny gate to renderable regimes

* feat(energy): render "no known SPR" risk note for countries with regime=none

* fix(energy): human-readable SPR regime labels; parallelize spine+registry reads in analyst
2026-04-09 22:16:24 +04:00

70 lines
2.2 KiB
JavaScript

#!/usr/bin/env node
import { readFileSync } from 'node:fs';
import { dirname, resolve } from 'node:path';
import { fileURLToPath } from 'node:url';
import { loadEnvFile, runSeed } from './_seed-utils.mjs';
loadEnvFile(import.meta.url);
export const CANONICAL_KEY = 'energy:spr-policies:v1';
export const SPR_POLICIES_TTL_SECONDS = 34_560_000; // ~400 days
const VALID_REGIMES = new Set([
'mandatory_stockholding',
'government_spr',
'spare_capacity',
'commercial_only',
'none',
]);
const REQUIRED_COUNTRIES = ['CN', 'IN', 'JP', 'SA', 'US'];
export function buildPayload() {
const __dirname = dirname(fileURLToPath(import.meta.url));
const raw = readFileSync(resolve(__dirname, 'data', 'spr-policies.json'), 'utf-8');
const registry = JSON.parse(raw);
return {
...registry,
updatedAt: new Date().toISOString(),
};
}
export function validateFn(data) {
if (!data?.policies || typeof data.policies !== 'object') return false;
const entries = Object.entries(data.policies);
if (entries.length < 30) return false;
const iso2Re = /^[A-Z]{2}$/;
for (const [key, entry] of entries) {
if (!iso2Re.test(key)) return false;
if (!VALID_REGIMES.has(entry.regime)) return false;
if (typeof entry.source !== 'string' || entry.source.length === 0) return false;
if (typeof entry.asOf !== 'string' || entry.asOf.length === 0) return false;
if ('capacityMb' in entry) {
if (typeof entry.capacityMb !== 'number' || !Number.isFinite(entry.capacityMb) || entry.capacityMb < 0) return false;
}
if ('estimatedFillPct' in entry) return false;
}
for (const reqCode of REQUIRED_COUNTRIES) {
if (!(reqCode in data.policies)) return false;
}
return true;
}
const isMain = process.argv[1]?.endsWith('seed-spr-policies.mjs');
if (isMain) {
runSeed('energy', 'spr-policies', CANONICAL_KEY, buildPayload, {
validateFn,
ttlSeconds: SPR_POLICIES_TTL_SECONDS,
sourceVersion: 'spr-policies-registry-v1',
recordCount: (data) => Object.keys(data?.policies ?? {}).length,
}).catch((err) => {
const cause = err.cause ? ` (cause: ${err.cause.message || err.cause.code || err.cause})` : '';
console.error('FATAL:', (err.message || err) + cause);
process.exit(1);
});
}