@jack200714/mafw 4.5.1

package/gateway/dist/usage/usage-poller.js

@@ -0,0 +1,216 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.UsagePoller = void 0;
+const logger_1 = require("../core/utils/logger");
+const WINDOW_MS = {
+    '5h': 5 * 60 * 60 * 1000,
+    '7d': 7 * 24 * 60 * 60 * 1000,
+    'month': 30 * 24 * 60 * 60 * 1000,
+};
+class UsagePoller {
+    store;
+    limitsGetter;
+    budgetsGetter;
+    lastGood = null;
+    pluginLoader;
+    constructor(store, limitsGetter, budgetsGetter = () => ({}), pluginLoader) {
+        this.store = store;
+        this.limitsGetter = limitsGetter;
+        this.budgetsGetter = budgetsGetter;
+        this.pluginLoader = pluginLoader;
+    }
+    get limits() { return this.limitsGetter(); }
+    get budgets() { return this.budgetsGetter(); }
+    async poll() {
+        try {
+            const providers = [];
+            const seen = new Set();
+            const configuredProviders = Object.keys(this.limits);
+            for (const name of configuredProviders) {
+                const p = this.aggregateProvider(name, this.limits[name]);
+                if (p.windows.length > 0) {
+                    providers.push(p);
+                    seen.add(name);
+                }
+            }
+            const dbProviders = this.store.getDistinctProviders();
+            const providerMap = new Map();
+            for (const name of dbProviders) {
+                if (seen.has(name))
+                    continue;
+                const totalCost = this.store.getProviderTotalCost(name);
+                const totalTokens = this.store.getProviderTotalTokens(name);
+                const tokenTotal = totalTokens.input + totalTokens.output + totalTokens.reasoning + totalTokens.cache.read + totalTokens.cache.write;
+                if (totalCost <= 0 && tokenTotal <= 0)
+                    continue;
+                const budget = this.budgets[name];
+                const used = Math.round(totalCost * 100) / 100;
+                if (budget && budget > 0) {
+                    const pct = Math.round((used / budget) * 100);
+                    providerMap.set(name, {
+                        name,
+                        type: 'api',
+                        plan: 'budget',
+                        windows: [{
+                                window: 'balance',
+                                used,
+                                limit: budget,
+                                unit: '$',
+                                pct,
+                                tokens: tokenTotal,
+                            }],
+                        severity: pct >= 90 ? 'critical' : pct >= 75 ? 'high' : pct >= 50 ? 'mid' : 'low',
+                    });
+                }
+                else {
+                    providerMap.set(name, {
+                        name,
+                        type: 'api',
+                        windows: [{
+                                window: 'balance',
+                                used,
+                                limit: 0,
+                                unit: '$',
+                                pct: 0,
+                                tokens: tokenTotal,
+                                projectedCost: used,
+                            }],
+                        severity: 'low',
+                    });
+                }
+            }
+            const pluginAdapters = this.pluginLoader?.getAdapters() ?? [];
+            const allExternal = pluginAdapters;
+            const externalResults = await Promise.allSettled(allExternal.map(a => a.fetch()));
+            for (const result of externalResults) {
+                if (result.status === 'fulfilled' && result.value) {
+                    const name = result.value.name;
+                    const budget = this.budgets[name];
+                    // Budget 折叠只适用于纯 balance 插件；含 day/7d/month 等窗口的插件
+                    // 折叠会静默吞掉窗口，故跳过并 warn。
+                    const allBalance = result.value.windows.every(w => w.window === 'balance');
+                    if (budget && budget > 0 && allBalance) {
+                        const balWindow = result.value.windows.find(w => w.window === 'balance');
+                        const remaining = balWindow?.remaining;
+                        const used = remaining !== undefined
+                            ? Math.max(0, budget - remaining)
+                            : this.store.getProviderTotalCost(name);
+                        const tokens = this.store.getProviderTotalTokens(name);
+                        const tokenTotal = tokens.input + tokens.output + tokens.reasoning + tokens.cache.read + tokens.cache.write;
+                        const pct = Math.round((used / budget) * 100);
+                        providerMap.set(name, {
+                            ...result.value,
+                            plan: 'budget',
+                            windows: [{
+                                    window: 'balance',
+                                    used: Math.round(used * 100) / 100,
+                                    limit: budget,
+                                    unit: '$',
+                                    pct,
+                                    tokens: tokenTotal,
+                                    remaining: remaining !== undefined ? Math.round(remaining * 100) / 100 : undefined,
+                                }],
+                            severity: pct >= 90 ? 'critical' : pct >= 75 ? 'high' : pct >= 50 ? 'mid' : 'low',
+                        });
+                    }
+                    else {
+                        if (budget && budget > 0 && !allBalance) {
+                            logger_1.log.warn(`[UsagePoller] budgets.${name} configured but plugin has non-balance windows; skipping budget collapse`);
+                        }
+                        const tokens = this.store.getProviderTotalTokens(name);
+                        const tokenTotal = tokens.input + tokens.output + tokens.reasoning + tokens.cache.read + tokens.cache.write;
+                        const windows = result.value.windows.map(w => w.limit === 0
+                            ? { ...w, tokens: tokenTotal, projectedCost: w.used }
+                            : w);
+                        providerMap.set(result.value.name, { ...result.value, windows });
+                    }
+                }
+            }
+            providers.push(...providerMap.values());
+            const result = { providers, updatedAt: Date.now() };
+            this.lastGood = result;
+            return result;
+        }
+        catch {
+            if (this.lastGood)
+                return this.lastGood;
+            return { providers: [], updatedAt: Date.now() };
+        }
+    }
+    aggregateProvider(name, limits) {
+        const now = Date.now();
+        const windows = [];
+        for (const [windowKey, limit] of Object.entries(limits)) {
+            if (windowKey === 'balance') {
+                const totalCost = this.store.getProviderTotalCost(name);
+                if (totalCost > 0) {
+                    const pct = Math.round((totalCost / limit) * 100);
+                    windows.push({
+                        window: 'balance',
+                        used: Math.round(totalCost * 100) / 100,
+                        limit,
+                        unit: '$',
+                        pct,
+                    });
+                }
+                continue;
+            }
+            const windowType = windowKey;
+            const windowMs = WINDOW_MS[windowType];
+            if (!windowMs)
+                continue;
+            const cutoff = now - windowMs;
+            const cost = this.store.getProviderCostInWindow(name, cutoff);
+            if (cost > 0) {
+                const pct = Math.round((cost / limit) * 100);
+                const earliest = this.store.getProviderEarliestTurnInWindow(name, cutoff);
+                const resetAt = earliest ? earliest + windowMs : now + windowMs;
+                const pacing = this.calculatePacing(pct, cutoff, now, windowMs);
+                const projected = this.calculateProjected(pct, cutoff, now, windowMs);
+                windows.push({
+                    window: windowType,
+                    used: Math.round(cost * 100) / 100,
+                    limit,
+                    unit: '$',
+                    resetAt,
+                    pct,
+                    pacing,
+                    projected,
+                });
+            }
+        }
+        const severity = this.calculateSeverity(windows);
+        return { name, windows, severity };
+    }
+    calculatePacing(pctUsed, cutoff, now, windowMs) {
+        const elapsed = now - cutoff;
+        const pctElapsed = Math.round((elapsed / windowMs) * 100);
+        const diff = pctUsed - pctElapsed;
+        if (diff > 10)
+            return 'ahead';
+        if (diff < -10)
+            return 'under';
+        return 'on-track';
+    }
+    calculateProjected(pctUsed, cutoff, now, windowMs) {
+        const elapsed = now - cutoff;
+        if (elapsed <= 0)
+            return pctUsed;
+        const rate = pctUsed / elapsed;
+        const projected = Math.round(rate * windowMs);
+        return Math.min(projected, 999);
+    }
+    calculateSeverity(windows) {
+        if (windows.length === 0)
+            return 'low';
+        const maxPct = Math.max(...windows.map(w => w.pct));
+        if (maxPct >= 90)
+            return 'critical';
+        if (maxPct >= 75)
+            return 'high';
+        if (maxPct >= 50)
+            return 'mid';
+        return 'low';
+    }
+}
+exports.UsagePoller = UsagePoller;

package/gateway/package.json

@@ -0,0 +1,37 @@
+{
+  "name": "mafw-gateway",
+  "version": "5.0.0",
+  "description": "MAFW Gateway v5.0 — SSE Event-Driven Scheduler",
+  "main": "dist/index.js",
+  "scripts": {
+    "build": "tsc && node -e \"const fs=require('fs'),p=require('path');const cp=(s,d)=>{if(fs.existsSync(s)){fs.cpSync(s,d,{recursive:true});console.log('[copy] '+d)}else{console.log('[skip] '+s)};};cp(p.join(__dirname,'src','dashboard','public'),p.join(__dirname,'dist','dashboard','public'));cp(p.join(__dirname,'src','tray','tray.ps1'),p.join(__dirname,'dist','tray','tray.ps1'));cp(p.join(__dirname,'src','usage','builtin-plugins'),p.join(__dirname,'dist','usage','builtin-plugins'));\"",
+    "start": "node dist/index.js",
+    "dev": "ts-node src/index.ts",
+    "test": "jest --runInBand"
+  },
+  "dependencies": {
+    "@a2a-js/sdk": "^1.0.1",
+    "@earendil-works/pi-coding-agent": "^0.84.1",
+    "@huggingface/transformers": "^4.2.0",
+    "@modelcontextprotocol/sdk": "^1.30.0",
+    "@opencode-ai/sdk": "^1.18.21",
+    "better-sqlite3": "^12.11.1",
+    "bonjour-service": "^1.4.4",
+    "cron": "^3.1.7",
+    "js-yaml": "^4.1.0",
+    "langchain-mcp-adapters": "^0.6.0",
+    "ws": "^8.21.3",
+    "zeromq": "^6.1.2"
+  },
+  "devDependencies": {
+    "@types/better-sqlite3": "^7.6.13",
+    "@types/cross-spawn": "^6.0.6",
+    "@types/jest": "^30.0.0",
+    "@types/node": "^20.0.0",
+    "@types/ws": "^8.18.1",
+    "jest": "^30.4.2",
+    "ts-jest": "^29.4.12",
+    "ts-node": "^10.9.0",
+    "typescript": "^5.3.0"
+  }
+}

package/package.json

@@ -0,0 +1,78 @@
+{
+  "name": "@jack200714/mafw",
+  "version": "4.5.1",
+  "description": "MAFW Loop Agent Plugin for OpenCode - Phase Relay + TMEM + Dynamic Compression",
+  "main": "dist/plugin.js",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/Jackli-2022/mafw.git"
+  },
+  "opencode": {
+    "plugin": "dist/plugin.js"
+  },
+  "bin": {
+    "mafw": "./bin/mafw.js"
+  },
+  "scripts": {
+    "build": "tsc && cd gateway && npm run build && cd .. && npm run build -w @mafw/sdk && npm run build -w @mafw/tui",
+    "postinstall": "node -e \"var p=require('path'),f=require('fs'),d=p.join(__dirname,'gateway'),ts=p.join(d,'node_modules','typescript','bin','tsc');if(f.existsSync(d)){if(f.existsSync(ts)){console.log('[mafw] gateway dev deps present, skip prod install')}else{require('child_process').execSync('npm install --omit=dev --no-audit --no-fund --loglevel=error',{cwd:d,stdio:'inherit'})}}\"",
+    "prepare": "node -e \"var p=require('path'),f=require('fs'),ts=p.join(__dirname,'gateway','node_modules','typescript','bin','tsc');if(!f.existsSync(ts)){console.log('[mafw] gateway dev deps missing, skip prepare build (fresh CI install)');process.exit(0)}require('child_process').execSync('npm run build',{cwd:__dirname,stdio:'inherit'})\"",
+    "pack:install": "npm run build && npm pack && npm install -g jack200714-mafw-*.tgz && mafw version",
+    "test": "npm test --prefix gateway",
+    "test:unit": "npm test --prefix gateway",
+    "test:tui": "node --test packages/tui/tests/*.test.ts"
+  },
+  "files": [
+    "dist/",
+    "gateway/dist/",
+    "gateway/package.json",
+    ".opencode/",
+    "bin/",
+    "packages/tui/dist/",
+    "README.md"
+  ],
+  "dependencies": {
+    "@a2a-js/sdk": "^1.0.1",
+    "@earendil-works/pi-tui": "^0.84.1",
+    "@huggingface/transformers": "^4.2.0",
+    "@langchain/core": "^1.1.48",
+    "@langchain/langgraph": "^1.4.7",
+    "@modelcontextprotocol/sdk": "^1.29.0",
+    "@opencode-ai/sdk": "^1.17.12",
+    "axios": "^1.6.0",
+    "better-sqlite3": "^12.11.1",
+    "bonjour-service": "^1.4.4",
+    "chalk": "^5.6.2",
+    "cron": "^3.1.7",
+    "js-yaml": "^4.1.0",
+    "langchain-mcp-adapters": "^0.6.0",
+    "opencode-ai": "^1.0.0",
+    "simple-git": "^3.20.0",
+    "ws": "^8.21.3",
+    "zeromq": "^6.1.2",
+    "zod": "^3.25.32"
+  },
+  "devDependencies": {
+    "@types/better-sqlite3": "^7.6.13",
+    "@types/jest": "^29.5.0",
+    "@types/js-yaml": "^4.0.5",
+    "@types/node": "^20.0.0",
+    "@types/ws": "^8.18.1",
+    "eslint": "^8.54.0",
+    "jest": "^29.7.0",
+    "ts-jest": "^29.4.11",
+    "ts-node": "^10.9.0",
+    "typescript": "^5.3.0"
+  },
+  "engines": {
+    "node": ">=18.0.0"
+  },
+  "overrides": {
+    "electron-builder": {
+      "chalk": "4.1.2"
+    }
+  },
+  "workspaces": [
+    "packages/*"
+  ]
+}