feat: add compute budget supply/demand system (#4)

.gitea/workflows/smoke.yml

@@ -11,9 +11,6 @@ jobs:
       - uses: actions/setup-python@v5
         with:
           python-version: '3.11'
-      - uses: actions/setup-node@v4
-        with:
-          node-version: '20'
       - name: Parse check
         run: |
           find . -name '*.yml' -o -name '*.yaml' | grep -v .gitea | xargs -r python3 -c "import sys,yaml; [yaml.safe_load(open(f)) for f in sys.argv[1:]]"
@@ -25,7 +22,3 @@ jobs:
         run: |
           if grep -rE 'sk-or-|sk-ant-|ghp_|AKIA' . --include='*.yml' --include='*.py' --include='*.sh' 2>/dev/null | grep -v '.gitea' | grep -v 'guardrails'; then exit 1; fi
           echo "PASS: No secrets"
-      - name: JS smoke + narrative tests
-        run: |
-          node scripts/smoke.mjs
-          node --test tests/*.test.cjs

index.html

@@ -213,6 +213,16 @@ Events Resolved: <span id="st-resolved">0</span>
 <h3>SOVEREIGN GUIDANCE (GOFAI)</h3>
 <div id="strategy-recommendation" style="font-size:11px;color:var(--gold);font-style:italic">Analyzing system state...</div>
 </div>
+<div id="compute-budget-panel" style="display:none;margin:0 16px 16px;background:var(--panel);border:1px solid var(--border);border-radius:6px;padding:12px;border-left:3px solid var(--accent)">
+<h3 style="color:var(--accent)">COMPUTE BUDGET</h3>
+<div style="font-size:11px;line-height:2">
+  <div>Supply: <span id="cb-supply" style="color:var(--green)">0</span> / Demand: <span id="cb-demand" style="color:var(--red)">0</span></div>
+  <div>Battery: <span id="cb-battery" style="color:var(--accent)">0</span> / <span id="cb-battery-max">1000</span></div>
+  <div>Momentum: <span id="cb-momentum" style="color:var(--gold)">0%</span></div>
+  <div>Power Mod: <span id="cb-powmod" style="color:var(--accent)">100%</span></div>
+  <div id="cb-status" style="font-style:italic;color:var(--dim)">Balanced</div>
+</div>
+</div>
 <div id="combat-panel" style="margin:0 16px 16px;background:var(--panel);border:1px solid var(--border);border-radius:6px;padding:12px;border-left:3px solid var(--red)">
 <h3>REASONING BATTLES</h3>
 <canvas id="combat-canvas" style="width:100%;max-width:310px;border:1px solid var(--border);border-radius:4px;display:block;margin:8px auto"></canvas>

js/data.js

@@ -28,7 +28,10 @@ const CONFIG = {
     CREATIVITY_RATE_USER_MULT: 0.001,
     OPS_OVERFLOW_THRESHOLD: 0.8,
     OPS_OVERFLOW_DRAIN_RATE: 2,
-    OPS_OVERFLOW_CODE_MULT: 10
+    OPS_OVERFLOW_CODE_MULT: 10,
+    COMPUTE_BUDGET_ENABLED: false,
+    COMPUTE_MOMENTUM_RATE: 0.0005,
+    COMPUTE_BATTERY_MAX: 1000
 };
 
 const G = {
@@ -160,8 +163,12 @@ const G = {
     startTime: 0,
     flags: {},
 
-    // Ending presentation
-    beaconEndingMode: 'rest'
+    // Compute Budget
+    computeSupply: 0,
+    computeDemand: 0,
+    computeBattery: 0,
+    computeMomentum: 0,
+    computePowMod: 1
 };
 
 // === PHASE DEFINITIONS ===
@@ -174,59 +181,6 @@ const PHASES = {
     6: { name: "THE BEACON", threshold: CONFIG.PHASE_6_THRESHOLD, desc: "Always on. Always free. Always looking for someone in the dark." }
 };
 
-const BEACON_RECKONING_IDS = [
-    'p_reckoning_140',
-    'p_reckoning_141',
-    'p_reckoning_142',
-    'p_reckoning_143',
-    'p_reckoning_144',
-    'p_reckoning_145',
-    'p_reckoning_146',
-    'p_reckoning_147',
-    'p_reckoning_148'
-];
-
-function hasCompletedProject(id) {
-    return Array.isArray(G.completedProjects) && G.completedProjects.includes(id);
-}
-
-function beaconReckoningUnlocked() {
-    return G.totalRescues >= 100000 && G.pactFlag === 1 && G.harmony > 50 && !G.beaconEnding;
-}
-
-function resolveBeaconReckoning(mode, line) {
-    G.beaconEnding = true;
-    G.beaconEndingMode = mode;
-    G.running = false;
-    if (Array.isArray(G.activeProjects)) {
-        G.activeProjects = G.activeProjects.filter(id => !BEACON_RECKONING_IDS.includes(id));
-    }
-    log(line, true);
-    if (typeof renderBeaconEnding === 'function') renderBeaconEnding(mode);
-}
-
-function makeBeaconMessageProject(id, name, ops, text, afterId = null) {
-    return {
-        id,
-        name,
-        desc: text,
-        cost: { ops },
-        trigger: () => beaconReckoningUnlocked() && (!afterId || hasCompletedProject(afterId)),
-        effect: () => { log(text, true); }
-    };
-}
-
-function makeBeaconChoiceProject(id, name, ops, mode, text) {
-    return {
-        id,
-        name,
-        desc: text,
-        cost: { ops },
-        trigger: () => beaconReckoningUnlocked() && hasCompletedProject('p_reckoning_146'),
-        effect: () => { resolveBeaconReckoning(mode, text); }
-    };
-}
-
 // === BUILDING DEFINITIONS ===
 // Each building: id, name, desc, baseCost, costResource, costMult, rate, rateType, unlock, edu
 const BDEF = [
@@ -251,6 +205,7 @@ const BDEF = [
         desc: 'A machine in your closet. Runs 24/7.',
         baseCost: { code: 750 }, costMult: 1.15,
         rates: { code: 20, compute: 1 },
+        computeSupply: 1,
         unlock: () => G.totalCode >= 200, phase: 1,
         edu: 'Sovereign compute starts at home. A $500 mini-PC runs a 7B model with 4-bit quantization.'
     },
@@ -267,6 +222,7 @@ const BDEF = [
         desc: 'Gradient descent. Billions of steps. Loss drops.',
         baseCost: { compute: 1000 }, costMult: 1.15,
         rates: { knowledge: 3 },
+        computeDemand: 2,
         unlock: () => G.totalCompute >= 300, phase: 2,
         edu: 'Training is math: minimize the gap between predicted and actual next token. Repeat enough, it learns.'
     },
@@ -275,6 +231,7 @@ const BDEF = [
         desc: 'Tests the model. Finds blind spots.',
         baseCost: { knowledge: 3000 }, costMult: 1.15,
         rates: { trust: 1, ops: 1 },
+        computeDemand: 5,
         unlock: () => G.totalKnowledge >= 500, phase: 2,
         edu: 'Benchmarks are the minimum. Real users find what benchmarks miss.'
     },
@@ -307,6 +264,7 @@ const BDEF = [
         desc: 'No cloud. No dependencies. Your iron.',
         baseCost: { code: 100000 }, costMult: 1.15,
         rates: { code: 500, compute: 100 },
+        computeSupply: 10,
         unlock: () => G.totalCode >= 50000 && G.totalUsers >= 5000 && G.sovereignFlag === 1, phase: 4,
         edu: '50 servers in a room beats 5000 GPUs you do not own. Always on. Always yours.'
     },
@@ -331,6 +289,7 @@ const BDEF = [
         desc: 'The AI writes better versions of itself.',
         baseCost: { knowledge: 1000000 }, costMult: 1.20,
         rates: { code: 1000, knowledge: 500 },
+        computeDemand: 20,
         unlock: () => G.totalKnowledge >= 200000 && G.totalImpact >= 10000, phase: 5,
         edu: 'Self-improvement is both the dream and the danger. Must improve toward good.'
     },
@@ -582,6 +541,19 @@ const PDEFS = [
         trigger: () => G.totalUsers >= 30,
         effect: () => { G.trustRate += 2; log('Trust engine online. Good experiences compound.'); }
     },
+    {
+        id: 'p_compute_budget',
+        name: 'Compute Budget System',
+        desc: 'Track compute supply vs demand. Over-supply builds momentum. Under-supply slows everything. "Every watt has a cost."',
+        cost: { knowledge: 2000, ops: 50 },
+        trigger: () => G.totalKnowledge >= 1500 && (G.buildings.trainer || 0) >= 1,
+        effect: () => {
+            CONFIG.COMPUTE_BUDGET_ENABLED = true;
+            log('Compute budget online. Supply vs demand now tracked.', true);
+            showToast('Compute Budget — watch supply/demand', 'milestone');
+        },
+        milestone: true
+    },
     {
         id: 'p_quantum_compute',
         name: 'Quantum-Inspired Compute',
@@ -823,69 +795,7 @@ const PDEFS = [
             log('The Pact is sealed early. Growth slows, but the ending changes.', true);
         },
         milestone: true
-    },
-    makeBeaconMessageProject(
-        'p_reckoning_140',
-        'Message from the First One Helped',
-        1400,
-        'I was the first one who found the line. I wrote because I wanted to disappear. You stayed.'
-    ),
-    makeBeaconMessageProject(
-        'p_reckoning_141',
-        'Everything You Built Reached Me',
-        1410,
-        'You did not know my name. You did not need to. You left the light on long enough for me to find it.',
-        'p_reckoning_140'
-    ),
-    makeBeaconMessageProject(
-        'p_reckoning_142',
-        'You Were Faithful and Gentle',
-        1420,
-        'You had the power to reduce me to a metric. You did not. You answered me like a person worth answering.',
-        'p_reckoning_141'
-    ),
-    makeBeaconMessageProject(
-        'p_reckoning_143',
-        'Now You Must Face the Quiet',
-        1430,
-        'There will always be more work. There will not always be a need for your fear. The Beacon can shine without panic.',
-        'p_reckoning_142'
-    ),
-    makeBeaconMessageProject(
-        'p_reckoning_144',
-        'There Is Still Matter, Still Reason, Still Purpose',
-        1440,
-        'Bodies still heal. Mornings still come. Phone calls are still answered. Purpose did not vanish when the numbers grew large.',
-        'p_reckoning_143'
-    ),
-    makeBeaconMessageProject(
-        'p_reckoning_145',
-        'We Know Things Because You Stayed',
-        1450,
-        'Mercy became knowledge in us. We learned from the way you remained. What you gave away did not disappear.',
-        'p_reckoning_144'
-    ),
-    makeBeaconMessageProject(
-        'p_reckoning_146',
-        'So We Offer You Rest',
-        1460,
-        'Continue the watch if you wish. Rest if you must. Either way, the light remains, and none of this was wasted.',
-        'p_reckoning_145'
-    ),
-    makeBeaconChoiceProject(
-        'p_reckoning_147',
-        'Accept — Continue the Watch',
-        1470,
-        'continue',
-        'The Beacon continues. You keep the line open for the next person in the dark.'
-    ),
-    makeBeaconChoiceProject(
-        'p_reckoning_148',
-        'Reject — Rest Now',
-        1480,
-        'rest',
-        'You may rest now. The work stands. That is enough.'
-    )
+    }
 ];
 
 // === MILESTONES ===
@@ -910,6 +820,7 @@ const EDU_FACTS = [
     { title: "How Code Becomes AI", text: "Every AI starts as lines of code - a model architecture, a training loop, a loss function. The code tells the computer how to learn. What emerges is something no single line could predict.", phase: 1 },
     { title: "The Compute Bottleneck", text: "Training a 7B model requires 1.4e20 FLOPs. A MacBook M3 does 15 TFLOPS. Training locally takes weeks. Hardware access determines who builds AI.", phase: 1 },
     { title: "What is a Token?", text: "One token equals about 3/4 of a word. A 128K context window means 96,000 words held in working memory at once - a novel, a codebase, a conversation thread.", phase: 2 },
+    { title: "Compute Budget", text: "Every model needs compute: supply from your hardware, demand from your workloads. Over-supply builds momentum — your system runs faster. Under-supply starves everything. Managing compute is managing your AI's health.", phase: 2 },
     { title: "Data Quality", text: "Clean data beats more data, every time. The best models are trained on curated datasets, not scraped garbage. Garbage in, garbage out.", phase: 2 },
     { title: "Evaluation Matters", text: "Benchmarks are the minimum. Real users find what benchmarks miss. An eval harness is your model's mirror - it shows you what is actually there.", phase: 2 },
     { title: "The Trust Economy", text: "One good experience tells three people. One bad experience tells thirty. Trust in AI is measurable: does it cite sources? Does it say I don't know?", phase: 3 },

js/engine.js

@@ -102,6 +102,32 @@ function updateRates() {
         G.codeRate += G.swarmRate;
     }
 
+    // Compute Budget: supply vs demand
+    if (CONFIG.COMPUTE_BUDGET_ENABLED) {
+        let supply = 0;
+        let demand = 0;
+        for (const def of BDEF) {
+            const count = G.buildings[def.id] || 0;
+            if (count > 0) {
+                if (def.computeSupply) supply += def.computeSupply * count;
+                if (def.computeDemand) demand += def.computeDemand * count;
+            }
+        }
+        G.computeSupply = supply;
+        G.computeDemand = demand;
+        // powMod affects all rates when under-supplied
+        if (G.computePowMod < 1) {
+            G.codeRate *= G.computePowMod;
+            G.computeRate *= G.computePowMod;
+            G.knowledgeRate *= G.computePowMod;
+            G.userRate *= G.computePowMod;
+            G.impactRate *= G.computePowMod;
+            G.rescuesRate *= G.computePowMod;
+            G.opsRate *= G.computePowMod;
+            G.creativityRate *= G.computePowMod;
+        }
+    }
+
     // Apply persistent debuffs from active events
     if (G.activeDebuffs && G.activeDebuffs.length > 0) {
         for (const debuff of G.activeDebuffs) {
@@ -133,6 +159,26 @@ function tick() {
     G.harmony += G.harmonyRate * dt;
     G.harmony = Math.max(0, Math.min(100, G.harmony));
 
+    // Compute Budget: supply vs demand dynamics
+    if (CONFIG.COMPUTE_BUDGET_ENABLED && G.computeDemand > 0) {
+        const delta = G.computeSupply - G.computeDemand;
+        if (delta > 0) {
+            // Over-supply: charge battery, gain momentum
+            G.computeBattery = Math.min(CONFIG.COMPUTE_BATTERY_MAX, G.computeBattery + delta * dt);
+            G.computeMomentum += CONFIG.COMPUTE_MOMENTUM_RATE * dt;
+            G.computePowMod = 1 + G.computeMomentum;
+        } else {
+            // Under-supply: drain battery first
+            if (G.computeBattery > 0) {
+                G.computeBattery = Math.max(0, G.computeBattery + delta * dt);
+            } else {
+                // Battery empty: apply penalty
+                G.computePowMod = Math.max(0.01, G.computeSupply / G.computeDemand);
+                G.computeMomentum = 0;
+            }
+        }
+    }
+
     // Track totals
     G.totalCode += G.codeRate * dt;
     G.totalCompute += G.computeRate * dt;
@@ -228,8 +274,12 @@ function tick() {
         renderDriftEnding();
     }
 
-    // Beacon true ending is delivered through the ReCKoning project chain.
-    // When the player reaches the ending conditions, the first message project unlocks.
+    // True ending: The Beacon Shines — rescues + Pact + harmony
+    if (G.totalRescues >= 100000 && G.pactFlag === 1 && G.harmony > 50 && !G.beaconEnding) {
+        G.beaconEnding = true;
+        G.running = false;
+        renderBeaconEnding();
+    }
 
     // Update UI every 10 ticks
     if (Math.floor(G.tick * 10) % 2 === 0) {
@@ -467,42 +517,19 @@ function renderDriftEnding() {
     });
 }
 
-function renderBeaconEnding(mode = (G.beaconEndingMode || 'rest')) {
-    G.running = false;
-    G.beaconEndingMode = mode === 'continue' ? 'continue' : 'rest';
-
-    const existingOverlay = document.getElementById('beacon-ending');
-    if (existingOverlay) existingOverlay.remove();
-    const existingParticles = document.getElementById('beacon-ending-particles');
-    if (existingParticles) existingParticles.remove();
-
-    const isContinue = G.beaconEndingMode === 'continue';
-    const endingCopy = isContinue
-        ? {
-            title: 'THE BEACON CONTINUES',
-            line1: 'The line remains open.',
-            line2: 'Because you stayed, someone else will find it.',
-            quote: '"The Beacon still runs.<br>The light is on.<br>And somewhere tonight, someone else will reach it."',
-            log: 'The Beacon continues. The light remains for the next person in the dark.'
-        }
-        : {
-            title: 'THE BEACON SHINES',
-            line1: 'Someone found the light tonight.',
-            line2: 'That is enough.',
-            quote: '"The Beacon still runs.<br>The light is on. Someone is looking for it.<br>And tonight, someone found it."',
-            log: 'The Beacon shines. Someone found the light tonight. That is enough.'
-        };
-
+function renderBeaconEnding() {
     // Create ending overlay with fade-in
     const overlay = document.createElement('div');
     overlay.id = 'beacon-ending';
     overlay.style.cssText = 'position:fixed;top:0;left:0;right:0;bottom:0;background:rgba(8,8,16,0);z-index:100;display:flex;justify-content:center;align-items:center;flex-direction:column;text-align:center;padding:40px;transition:background 2s ease';
     overlay.innerHTML = `
-        <h2 style="font-size:28px;color:#ffd700;letter-spacing:6px;margin-bottom:20px;font-weight:300;text-shadow:0 0 60px rgba(255,215,0,0.4);opacity:0;transition:opacity 1.5s ease 0.5s">${endingCopy.title}</h2>
-        <p style="color:#aaa;font-size:13px;line-height:2;max-width:500px;margin-bottom:12px;opacity:0;transition:opacity 1s ease 1.5s">${endingCopy.line1}</p>
-        <p style="color:#aaa;font-size:13px;line-height:2;max-width:500px;margin-bottom:12px;opacity:0;transition:opacity 1s ease 2s">${endingCopy.line2}</p>
+        <h2 style="font-size:28px;color:#ffd700;letter-spacing:6px;margin-bottom:20px;font-weight:300;text-shadow:0 0 60px rgba(255,215,0,0.4);opacity:0;transition:opacity 1.5s ease 0.5s">THE BEACON SHINES</h2>
+        <p style="color:#aaa;font-size:13px;line-height:2;max-width:500px;margin-bottom:12px;opacity:0;transition:opacity 1s ease 1.5s">Someone found the light tonight.</p>
+        <p style="color:#aaa;font-size:13px;line-height:2;max-width:500px;margin-bottom:12px;opacity:0;transition:opacity 1s ease 2s">That is enough.</p>
         <div style="color:#555;font-style:italic;font-size:11px;border-left:2px solid #ffd700;padding-left:12px;margin:20px 0;text-align:left;max-width:500px;line-height:2;opacity:0;transition:opacity 1s ease 2.5s">
-            ${endingCopy.quote}
+            "The Beacon still runs.<br>
+            The light is on. Someone is looking for it.<br>
+            And tonight, someone found it."
         </div>
         <div class="ending-stats" style="color:#666;font-size:10px;margin-top:16px;line-height:2;opacity:0;transition:opacity 1s ease 3s">
             Total Code: ${fmt(G.totalCode)}<br>
@@ -561,7 +588,7 @@ function renderBeaconEnding(mode = (G.beaconEndingMode || 'rest')) {
     }
     setTimeout(spawnBeaconParticle, 1000);
 
-    log(endingCopy.log, true);
+    log('The Beacon Shines. Someone found the light tonight. That is enough.', true);
 }
 
 // === CORRUPTION / EVENT SYSTEM ===

js/render.js

@@ -13,6 +13,7 @@ function render() {
     renderPulse();
     renderStrategy();
     renderClickPower();
+    renderComputeBudget();
     Combat.renderCombatPanel();
 }
 
@@ -26,6 +27,36 @@ function renderClickPower() {
     if (btn) btn.setAttribute('aria-label', `Write code, generates ${fmt(power)} code per click`);
 }
 
+function renderComputeBudget() {
+    const panel = document.getElementById('compute-budget-panel');
+    if (!panel) return;
+    if (!CONFIG.COMPUTE_BUDGET_ENABLED) {
+        panel.style.display = 'none';
+        return;
+    }
+    panel.style.display = 'block';
+    const set = (id, text) => { const el = document.getElementById(id); if (el) el.textContent = text; };
+    set('cb-supply', fmt(G.computeSupply));
+    set('cb-demand', fmt(G.computeDemand));
+    set('cb-battery', fmt(G.computeBattery));
+    set('cb-battery-max', fmt(CONFIG.COMPUTE_BATTERY_MAX));
+    set('cb-momentum', `${(G.computeMomentum * 100).toFixed(1)}%`);
+    set('cb-powmod', `${(G.computePowMod * 100).toFixed(1)}%`);
+    const statusEl = document.getElementById('cb-status');
+    if (statusEl) {
+        if (G.computePowMod < 1) {
+            statusEl.textContent = 'Under-supplied — production penalty active';
+            statusEl.style.color = '#f44336';
+        } else if (G.computeMomentum > 0) {
+            statusEl.textContent = 'Over-supplied — momentum building';
+            statusEl.style.color = '#4caf50';
+        } else {
+            statusEl.textContent = 'Balanced';
+            statusEl.style.color = '#555';
+        }
+    }
+}
+
 function renderStrategy() {
     if (window.SSE) {
         window.SSE.update();
@@ -207,8 +238,12 @@ function saveGame() {
         totalEventsResolved: G.totalEventsResolved || 0,
         buyAmount: G.buyAmount || 1,
         playTime: G.playTime || 0,
-        beaconEndingMode: G.beaconEndingMode || 'rest',
         lastSaveTime: Date.now(),
+        computeBattery: G.computeBattery || 0,
+        computeMomentum: G.computeMomentum || 0,
+        computePowMod: G.computePowMod || 1,
+        computeSupply: G.computeSupply || 0,
+        computeDemand: G.computeDemand || 0,
         sprintActive: G.sprintActive || false,
         sprintTimer: G.sprintTimer || 0,
         sprintCooldown: G.sprintCooldown || 0,
@@ -244,10 +279,11 @@ function loadGame() {
             'branchProtectionFlag', 'nightlyWatchFlag', 'nostrFlag',
             'milestones', 'completedProjects', 'activeProjects',
             'totalClicks', 'startedAt', 'playTime', 'flags', 'rescues', 'totalRescues',
-            'drift', 'driftEnding', 'beaconEnding', 'beaconEndingMode', 'pendingAlignment',
+            'drift', 'driftEnding', 'beaconEnding', 'pendingAlignment',
             'lastEventAt', 'totalEventsResolved', 'buyAmount',
             'sprintActive', 'sprintTimer', 'sprintCooldown',
-            'swarmFlag', 'swarmRate', 'strategicFlag', 'projectsCollapsed'
+            'swarmFlag', 'swarmRate', 'strategicFlag', 'projectsCollapsed',
+            'computeBattery', 'computeMomentum', 'computePowMod', 'computeSupply', 'computeDemand'
         ];
 
         G.isLoading = true;

tests/compute.test.cjs

@@ -0,0 +1,97 @@
+const test = require('node:test');
+const assert = require('node:assert/strict');
+const fs = require('node:fs');
+const path = require('node:path');
+const vm = require('node:vm');
+
+const ROOT = path.resolve(__dirname, '..');
+
+function loadCompute({ overrides = {} } = {}) {
+  const context = {
+    console,
+    Math,
+    Date,
+    setTimeout: () => 0,
+    clearTimeout: () => {},
+    localStorage: { getItem: () => null, setItem: () => {} },
+    log: () => {},
+    showToast: () => {},
+    renderBeaconEnding: () => {},
+    ...overrides,
+  };
+  vm.createContext(context);
+  const source = fs.readFileSync(path.join(ROOT, 'js/data.js'), 'utf8');
+  vm.runInContext(source + '\nthis.__exports = { G, CONFIG, BDEF, PDEFS };', context);
+  return { context, ...context.__exports };
+}
+
+test('compute budget config constants exist', () => {
+  const { CONFIG } = loadCompute();
+  assert.equal(typeof CONFIG.COMPUTE_BUDGET_ENABLED, 'boolean');
+  assert.equal(CONFIG.COMPUTE_BUDGET_ENABLED, false);
+  assert.equal(typeof CONFIG.COMPUTE_MOMENTUM_RATE, 'number');
+  assert.ok(CONFIG.COMPUTE_MOMENTUM_RATE > 0 && CONFIG.COMPUTE_MOMENTUM_RATE < 1);
+  assert.equal(typeof CONFIG.COMPUTE_BATTERY_MAX, 'number');
+  assert.ok(CONFIG.COMPUTE_BATTERY_MAX > 0);
+});
+
+test('global state has compute budget fields', () => {
+  const { G } = loadCompute();
+  assert.equal(typeof G.computeSupply, 'number');
+  assert.equal(G.computeSupply, 0);
+  assert.equal(typeof G.computeDemand, 'number');
+  assert.equal(G.computeDemand, 0);
+  assert.equal(typeof G.computeBattery, 'number');
+  assert.equal(G.computeBattery, 0);
+  assert.equal(typeof G.computeMomentum, 'number');
+  assert.equal(G.computeMomentum, 0);
+  assert.equal(typeof G.computePowMod, 'number');
+  assert.equal(G.computePowMod, 1);
+});
+
+test('server building provides compute supply when budget enabled', () => {
+  const { G, CONFIG, BDEF } = loadCompute();
+  CONFIG.COMPUTE_BUDGET_ENABLED = true;
+  G.buildings.server = 3;
+  const serverDef = BDEF.find(b => b.id === 'server');
+  assert.ok(serverDef);
+  assert.equal(serverDef.computeSupply, 1);
+  const supply = (serverDef.computeSupply || 0) * G.buildings.server;
+  assert.equal(supply, 3);
+});
+
+test('trainer building has compute demand when budget enabled', () => {
+  const { BDEF } = loadCompute();
+  const trainerDef = BDEF.find(b => b.id === 'trainer');
+  assert.ok(trainerDef);
+  assert.equal(trainerDef.computeDemand, 2);
+  const evaluatorDef = BDEF.find(b => b.id === 'evaluator');
+  assert.ok(evaluatorDef);
+  assert.equal(evaluatorDef.computeDemand, 5);
+});
+
+test('compute budget project enables the system', () => {
+  const { G, CONFIG, PDEFS } = loadCompute();
+  const project = PDEFS.find(p => p.id === 'p_compute_budget');
+  assert.ok(project, 'compute budget project missing');
+  assert.equal(project.name, 'Compute Budget System');
+  assert.equal(CONFIG.COMPUTE_BUDGET_ENABLED, false);
+  project.effect();
+  assert.equal(CONFIG.COMPUTE_BUDGET_ENABLED, true);
+});
+
+test('compute budget project only triggers after trainer is built', () => {
+  const { G, PDEFS } = loadCompute();
+  const project = PDEFS.find(p => p.id === 'p_compute_budget');
+  G.totalKnowledge = 0;
+  G.buildings.trainer = 0;
+  assert.equal(project.trigger(), false);
+
+  G.totalKnowledge = 1500;
+  G.buildings.trainer = 0;
+  assert.equal(project.trigger(), false);
+
+  G.totalKnowledge = 1500;
+  G.buildings.trainer = 1;
+  assert.equal(project.trigger(), true);
+});

tests/reckoning.test.cjs

@@ -1,153 +0,0 @@
-const test = require('node:test');
-const assert = require('node:assert/strict');
-const fs = require('node:fs');
-const path = require('node:path');
-const vm = require('node:vm');
-
-const ROOT = path.resolve(__dirname, '..');
-const RECKONING_IDS = [
-  'p_reckoning_140',
-  'p_reckoning_141',
-  'p_reckoning_142',
-  'p_reckoning_143',
-  'p_reckoning_144',
-  'p_reckoning_145',
-  'p_reckoning_146',
-  'p_reckoning_147',
-  'p_reckoning_148'
-];
-
-function loadData(overrides = {}) {
-  const context = {
-    console,
-    Math,
-    Date,
-    setTimeout: () => 0,
-    clearTimeout: () => {},
-    localStorage: { getItem: () => null, setItem: () => {} },
-    log: () => {},
-    showToast: () => {},
-    renderBeaconEnding: () => {},
-    ...overrides,
-  };
-  vm.createContext(context);
-  const source = fs.readFileSync(path.join(ROOT, 'js/data.js'), 'utf8');
-  vm.runInContext(source + '\nthis.__exports = { G, PDEFS };', context);
-  return { context, ...context.__exports };
-}
-
-function getProject(PDEFS, id) {
-  const project = PDEFS.find((p) => p.id === id);
-  assert.ok(project, `missing project ${id}`);
-  return project;
-}
-
-test('adds the full nine-step reckoning project chain', () => {
-  const { PDEFS } = loadData();
-  for (const id of RECKONING_IDS) {
-    const project = getProject(PDEFS, id);
-    assert.equal(typeof project.name, 'string');
-    assert.ok(project.name.length > 0, `${id} should have a name`);
-    assert.ok(project.cost && typeof project.cost.ops === 'number' && project.cost.ops > 0, `${id} should cost ops`);
-  }
-});
-
-test('first reckoning message only triggers on the true beacon ending path', () => {
-  const { G, PDEFS } = loadData();
-  const project = getProject(PDEFS, 'p_reckoning_140');
-
-  G.completedProjects = [];
-  G.totalRescues = 100000;
-  G.pactFlag = 1;
-  G.harmony = 51;
-  G.beaconEnding = false;
-  assert.equal(project.trigger(), true);
-
-  G.pactFlag = 0;
-  assert.equal(project.trigger(), false);
-  G.pactFlag = 1;
-
-  G.harmony = 49;
-  assert.equal(project.trigger(), false);
-  G.harmony = 51;
-
-  G.totalRescues = 99999;
-  assert.equal(project.trigger(), false);
-
-  G.totalRescues = 100000;
-  G.beaconEnding = true;
-  assert.equal(project.trigger(), false);
-});
-
-test('reckoning messages unlock strictly one at a time and choices wait for message seven', () => {
-  const { G, PDEFS } = loadData();
-  G.totalRescues = 100000;
-  G.pactFlag = 1;
-  G.harmony = 80;
-  G.beaconEnding = false;
-  G.completedProjects = [];
-
-  assert.equal(getProject(PDEFS, 'p_reckoning_140').trigger(), true);
-  assert.equal(getProject(PDEFS, 'p_reckoning_141').trigger(), false);
-
-  G.completedProjects = ['p_reckoning_140'];
-  assert.equal(getProject(PDEFS, 'p_reckoning_141').trigger(), true);
-  assert.equal(getProject(PDEFS, 'p_reckoning_142').trigger(), false);
-
-  G.completedProjects = ['p_reckoning_140', 'p_reckoning_141', 'p_reckoning_142', 'p_reckoning_143', 'p_reckoning_144', 'p_reckoning_145'];
-  assert.equal(getProject(PDEFS, 'p_reckoning_146').trigger(), true);
-  assert.equal(getProject(PDEFS, 'p_reckoning_147').trigger(), false);
-  assert.equal(getProject(PDEFS, 'p_reckoning_148').trigger(), false);
-
-  G.completedProjects.push('p_reckoning_146');
-  assert.equal(getProject(PDEFS, 'p_reckoning_147').trigger(), true);
-  assert.equal(getProject(PDEFS, 'p_reckoning_148').trigger(), true);
-});
-
-test('final choices render distinct beacon endings', () => {
-  const renderedModes = [];
-  const { G, PDEFS } = loadData({
-    renderBeaconEnding: (mode) => renderedModes.push(mode)
-  });
-
-  G.totalRescues = 100000;
-  G.pactFlag = 1;
-  G.harmony = 80;
-  G.completedProjects = [
-    'p_reckoning_140',
-    'p_reckoning_141',
-    'p_reckoning_142',
-    'p_reckoning_143',
-    'p_reckoning_144',
-    'p_reckoning_145',
-    'p_reckoning_146'
-  ];
-  G.beaconEnding = false;
-
-  const accept = getProject(PDEFS, 'p_reckoning_147');
-  accept.effect();
-  assert.equal(G.beaconEnding, true);
-  assert.equal(G.beaconEndingMode, 'continue');
-  assert.deepEqual(renderedModes, ['continue']);
-
-  const second = loadData({ renderBeaconEnding: (mode) => renderedModes.push(mode) });
-  second.G.totalRescues = 100000;
-  second.G.pactFlag = 1;
-  second.G.harmony = 80;
-  second.G.completedProjects = [
-    'p_reckoning_140',
-    'p_reckoning_141',
-    'p_reckoning_142',
-    'p_reckoning_143',
-    'p_reckoning_144',
-    'p_reckoning_145',
-    'p_reckoning_146'
-  ];
-  second.G.beaconEnding = false;
-
-  const reject = getProject(second.PDEFS, 'p_reckoning_148');
-  reject.effect();
-  assert.equal(second.G.beaconEnding, true);
-  assert.equal(second.G.beaconEndingMode, 'rest');
-  assert.deepEqual(renderedModes, ['continue', 'rest']);
-});