opencode-antigravity-autopilot 2.2.1 → 2.2.3

package/dist/scripts/src/quota/ApiQuotaPoller.js

@@ -0,0 +1,340 @@
+"use strict";
+var __extends = (this && this.__extends) || (function () {
+    var extendStatics = function (d, b) {
+        extendStatics = Object.setPrototypeOf ||
+            ({ __proto__: [] } instanceof Array && function (d, b) { d.__proto__ = b; }) ||
+            function (d, b) { for (var p in b) if (Object.prototype.hasOwnProperty.call(b, p)) d[p] = b[p]; };
+        return extendStatics(d, b);
+    };
+    return function (d, b) {
+        if (typeof b !== "function" && b !== null)
+            throw new TypeError("Class extends value " + String(b) + " is not a constructor or null");
+        extendStatics(d, b);
+        function __() { this.constructor = d; }
+        d.prototype = b === null ? Object.create(b) : (__.prototype = b.prototype, new __());
+    };
+})();
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __generator = (this && this.__generator) || function (thisArg, body) {
+    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype);
+    return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
+    function verb(n) { return function (v) { return step([n, v]); }; }
+    function step(op) {
+        if (f) throw new TypeError("Generator is already executing.");
+        while (g && (g = 0, op[0] && (_ = 0)), _) try {
+            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
+            if (y = 0, t) op = [op[0] & 2, t.value];
+            switch (op[0]) {
+                case 0: case 1: t = op; break;
+                case 4: _.label++; return { value: op[1], done: false };
+                case 5: _.label++; y = op[1]; op = [0]; continue;
+                case 7: op = _.ops.pop(); _.trys.pop(); continue;
+                default:
+                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
+                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
+                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
+                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
+                    if (t[2]) _.ops.pop();
+                    _.trys.pop(); continue;
+            }
+            op = body.call(thisArg, _);
+        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
+        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
+    }
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ApiQuotaPoller = exports.AuthenticationError = void 0;
+var fs_1 = __importDefault(require("fs"));
+var LOG_FILE = '/tmp/autopilot.log';
+function logToFile(message) {
+    var timestamp = new Date().toISOString();
+    fs_1.default.appendFileSync(LOG_FILE, "[".concat(timestamp, "] ").concat(message, "\n"));
+}
+var ANTIGRAVITY_CLIENT_ID = '1071006060591-tmhssin2h21lcre235vtolojh4g403ep.apps.googleusercontent.com';
+var ANTIGRAVITY_CLIENT_SECRET = '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';
+var GOOGLE_TOKEN_URL = 'https://oauth2.googleapis.com/token';
+var CLOUDCODE_BASE_URL = 'https://cloudcode-pa.googleapis.com';
+var CLOUDCODE_METADATA = {
+    ideType: 'ANTIGRAVITY',
+    platform: 'PLATFORM_UNSPECIFIED',
+    pluginType: 'GEMINI',
+};
+var AuthenticationError = /** @class */ (function (_super) {
+    __extends(AuthenticationError, _super);
+    function AuthenticationError(message) {
+        var _this = _super.call(this, message) || this;
+        _this.name = 'AuthenticationError';
+        return _this;
+    }
+    return AuthenticationError;
+}(Error));
+exports.AuthenticationError = AuthenticationError;
+var ApiQuotaPoller = /** @class */ (function () {
+    function ApiQuotaPoller() {
+        if (!(this instanceof ApiQuotaPoller)) {
+            // @ts-ignore
+            return new ApiQuotaPoller();
+        }
+    }
+    ApiQuotaPoller.prototype.refreshAccessToken = function (refreshToken) {
+        return __awaiter(this, void 0, void 0, function () {
+            var params, controller, timeout, response, data, e_1;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        params = new URLSearchParams({
+                            client_id: ANTIGRAVITY_CLIENT_ID,
+                            client_secret: ANTIGRAVITY_CLIENT_SECRET,
+                            refresh_token: refreshToken,
+                            grant_type: 'refresh_token',
+                        });
+                        controller = new AbortController();
+                        timeout = setTimeout(function () { return controller.abort(); }, 10000);
+                        _a.label = 1;
+                    case 1:
+                        _a.trys.push([1, 4, 5, 6]);
+                        return [4 /*yield*/, fetch(GOOGLE_TOKEN_URL, {
+                                method: 'POST',
+                                headers: { 'Content-Type': 'application/x-www-form-urlencoded' },
+                                body: params.toString(),
+                                signal: controller.signal,
+                            })];
+                    case 2:
+                        response = _a.sent();
+                        if (!response.ok) {
+                            logToFile("Token refresh failed: ".concat(response.status));
+                            if (response.status === 401 || response.status === 403) {
+                                throw new AuthenticationError("Token refresh failed (".concat(response.status, ")"));
+                            }
+                            throw new Error("Token refresh failed (".concat(response.status, ")"));
+                        }
+                        return [4 /*yield*/, response.json()];
+                    case 3:
+                        data = (_a.sent());
+                        return [2 /*return*/, data.access_token];
+                    case 4:
+                        e_1 = _a.sent();
+                        logToFile("Token refresh error: ".concat(e_1.message));
+                        throw e_1;
+                    case 5:
+                        clearTimeout(timeout);
+                        return [7 /*endfinally*/];
+                    case 6: return [2 /*return*/];
+                }
+            });
+        });
+    };
+    ApiQuotaPoller.prototype.loadCodeAssist = function (accessToken) {
+        return __awaiter(this, void 0, void 0, function () {
+            var controller, timeout, response;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        controller = new AbortController();
+                        timeout = setTimeout(function () { return controller.abort(); }, 10000);
+                        _a.label = 1;
+                    case 1:
+                        _a.trys.push([1, , 4, 5]);
+                        return [4 /*yield*/, fetch("".concat(CLOUDCODE_BASE_URL, "/v1internal:loadCodeAssist"), {
+                                method: 'POST',
+                                headers: {
+                                    Authorization: "Bearer ".concat(accessToken),
+                                    'Content-Type': 'application/json',
+                                    'User-Agent': 'antigravity',
+                                },
+                                body: JSON.stringify({ metadata: CLOUDCODE_METADATA }),
+                                signal: controller.signal,
+                            })];
+                    case 2:
+                        response = _a.sent();
+                        if (!response.ok) {
+                            logToFile("loadCodeAssist failed: ".concat(response.status));
+                            throw new Error("loadCodeAssist failed (".concat(response.status, ")"));
+                        }
+                        return [4 /*yield*/, response.json()];
+                    case 3: return [2 /*return*/, (_a.sent())];
+                    case 4:
+                        clearTimeout(timeout);
+                        return [7 /*endfinally*/];
+                    case 5: return [2 /*return*/];
+                }
+            });
+        });
+    };
+    ApiQuotaPoller.prototype.extractProjectId = function (cloudaicompanionProject) {
+        if (!cloudaicompanionProject)
+            return undefined;
+        if (typeof cloudaicompanionProject === 'string') {
+            var match = cloudaicompanionProject.match(/projects\/([^/]+)/);
+            return match ? match[1] : undefined;
+        }
+        return undefined;
+    };
+    ApiQuotaPoller.prototype.fetchAvailableModels = function (accessToken, projectId) {
+        return __awaiter(this, void 0, void 0, function () {
+            var payload, controller, timeout, response;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        payload = projectId ? { project: projectId } : {};
+                        controller = new AbortController();
+                        timeout = setTimeout(function () { return controller.abort(); }, 10000);
+                        _a.label = 1;
+                    case 1:
+                        _a.trys.push([1, , 4, 5]);
+                        return [4 /*yield*/, fetch("".concat(CLOUDCODE_BASE_URL, "/v1internal:fetchAvailableModels"), {
+                                method: 'POST',
+                                headers: {
+                                    Authorization: "Bearer ".concat(accessToken),
+                                    'Content-Type': 'application/json',
+                                    'User-Agent': 'antigravity',
+                                },
+                                body: JSON.stringify(payload),
+                                signal: controller.signal,
+                            })];
+                    case 2:
+                        response = _a.sent();
+                        if (!response.ok) {
+                            logToFile("fetchAvailableModels failed: ".concat(response.status));
+                            throw new Error("fetchModels failed (".concat(response.status, ")"));
+                        }
+                        return [4 /*yield*/, response.json()];
+                    case 3: return [2 /*return*/, (_a.sent())];
+                    case 4:
+                        clearTimeout(timeout);
+                        return [7 /*endfinally*/];
+                    case 5: return [2 /*return*/];
+                }
+            });
+        });
+    };
+    ApiQuotaPoller.prototype.checkQuota = function (account) {
+        return __awaiter(this, void 0, void 0, function () {
+            var accessToken, projectId, codeAssist, quotaResponse, models, _i, _a, _b, modelKey, info, quotaInfo, label, lowerLabel, remainingFraction, error_1;
+            var _c;
+            return __generator(this, function (_d) {
+                switch (_d.label) {
+                    case 0:
+                        _d.trys.push([0, 5, , 6]);
+                        return [4 /*yield*/, this.refreshAccessToken(account.refreshToken)];
+                    case 1:
+                        accessToken = _d.sent();
+                        projectId = account.projectId || account.managedProjectId;
+                        if (!!projectId) return [3 /*break*/, 3];
+                        return [4 /*yield*/, this.loadCodeAssist(accessToken)];
+                    case 2:
+                        codeAssist = _d.sent();
+                        projectId = this.extractProjectId(codeAssist.cloudaicompanionProject);
+                        _d.label = 3;
+                    case 3: return [4 /*yield*/, this.fetchAvailableModels(accessToken, projectId)];
+                    case 4:
+                        quotaResponse = _d.sent();
+                        if (!quotaResponse.models)
+                            return [2 /*return*/, []];
+                        models = [];
+                        for (_i = 0, _a = Object.entries(quotaResponse.models); _i < _a.length; _i++) {
+                            _b = _a[_i], modelKey = _b[0], info = _b[1];
+                            quotaInfo = info.quotaInfo;
+                            if (!quotaInfo)
+                                continue;
+                            label = info.displayName || modelKey;
+                            lowerLabel = label.toLowerCase();
+                            // Filter out unwanted models
+                            if (lowerLabel.startsWith('chat_') ||
+                                lowerLabel.startsWith('rev19') ||
+                                lowerLabel.includes('gemini 2.5') ||
+                                lowerLabel.includes('gemini 3 pro image')) {
+                                continue;
+                            }
+                            remainingFraction = Math.min(1, Math.max(0, (_c = quotaInfo.remainingFraction) !== null && _c !== void 0 ? _c : 0));
+                            models.push({
+                                model: info.model || modelKey,
+                                displayName: label,
+                                remainingFraction: remainingFraction,
+                                isExhausted: remainingFraction <= 0,
+                                resetTime: quotaInfo.resetTime,
+                            });
+                        }
+                        return [2 /*return*/, models];
+                    case 5:
+                        error_1 = _d.sent();
+                        if (error_1 instanceof AuthenticationError) {
+                            throw error_1;
+                        }
+                        logToFile("Failed to check quota via API: ".concat(error_1));
+                        return [2 /*return*/, []];
+                    case 6: return [2 /*return*/];
+                }
+            });
+        });
+    };
+    ApiQuotaPoller.prototype.checkQuotaForModel = function (account, modelName) {
+        return __awaiter(this, void 0, void 0, function () {
+            var models, model, normalizedSearch_1;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0: return [4 /*yield*/, this.checkQuota(account)];
+                    case 1:
+                        models = _a.sent();
+                        model = models.find(function (m) { return m.model === modelName || m.displayName === modelName; });
+                        // Try partial match (e.g., "gemini-3-pro" matches "Gemini 3 Pro")
+                        if (!model) {
+                            normalizedSearch_1 = modelName.toLowerCase().replace(/[-_]/g, ' ');
+                            model = models.find(function (m) {
+                                return m.model.toLowerCase().includes(normalizedSearch_1) ||
+                                    m.displayName.toLowerCase().includes(normalizedSearch_1);
+                            });
+                        }
+                        if (!model)
+                            return [2 /*return*/, null];
+                        return [2 /*return*/, {
+                                remainingFraction: model.remainingFraction,
+                                resetTime: model.resetTime,
+                                model: model.model,
+                            }];
+                }
+            });
+        });
+    };
+    ApiQuotaPoller.prototype.getAllQuotas = function (account) {
+        return __awaiter(this, void 0, void 0, function () {
+            var models, quotaMap, _i, models_1, model;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0: return [4 /*yield*/, this.checkQuota(account)];
+                    case 1:
+                        models = _a.sent();
+                        quotaMap = new Map();
+                        for (_i = 0, models_1 = models; _i < models_1.length; _i++) {
+                            model = models_1[_i];
+                            quotaMap.set(model.model, {
+                                remainingFraction: model.remainingFraction,
+                                resetTime: model.resetTime,
+                                model: model.model,
+                            });
+                            // Also add by display name for easier lookup
+                            quotaMap.set(model.displayName, {
+                                remainingFraction: model.remainingFraction,
+                                resetTime: model.resetTime,
+                                model: model.model,
+                            });
+                        }
+                        return [2 /*return*/, quotaMap];
+                }
+            });
+        });
+    };
+    return ApiQuotaPoller;
+}());
+exports.ApiQuotaPoller = ApiQuotaPoller;

package/dist/scripts/src/quota/LSPFinder.js

@@ -0,0 +1,86 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LSPFinder = void 0;
+var child_process_1 = require("child_process");
+var readline = __importStar(require("readline"));
+var LSPFinder = /** @class */ (function () {
+    function LSPFinder() {
+        if (!(this instanceof LSPFinder)) {
+            // @ts-ignore
+            return new LSPFinder();
+        }
+    }
+    LSPFinder.prototype.findProcess = function () {
+        return new Promise(function (resolve) {
+            var ps = (0, child_process_1.spawn)('ps', ['aux']);
+            var rl = readline.createInterface({
+                input: ps.stdout,
+                crlfDelay: Infinity,
+            });
+            ps.on('error', function () {
+                rl.close();
+                resolve(null);
+            });
+            var found = false;
+            rl.on('line', function (line) {
+                if (found)
+                    return;
+                if (line.includes('language_server_antigravity')) {
+                    var pidMatch = line.trim().split(/\s+/)[1];
+                    var csrfMatch = line.match(/--csrf_token=([a-zA-Z0-9_\-]+)/);
+                    var portMatch = line.match(/--extension_server_port=(\d+)/);
+                    if (pidMatch && csrfMatch && portMatch) {
+                        found = true;
+                        resolve({
+                            pid: parseInt(pidMatch, 10),
+                            csrfToken: csrfMatch[1],
+                            port: parseInt(portMatch[1], 10),
+                        });
+                        ps.kill();
+                        rl.close();
+                    }
+                }
+            });
+            ps.on('close', function () {
+                if (!found) {
+                    resolve(null);
+                }
+            });
+        });
+    };
+    return LSPFinder;
+}());
+exports.LSPFinder = LSPFinder;

package/dist/scripts/src/quota/QuotaCacheUpdater.js

@@ -0,0 +1,212 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __generator = (this && this.__generator) || function (thisArg, body) {
+    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype);
+    return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
+    function verb(n) { return function (v) { return step([n, v]); }; }
+    function step(op) {
+        if (f) throw new TypeError("Generator is already executing.");
+        while (g && (g = 0, op[0] && (_ = 0)), _) try {
+            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
+            if (y = 0, t) op = [op[0] & 2, t.value];
+            switch (op[0]) {
+                case 0: case 1: t = op; break;
+                case 4: _.label++; return { value: op[1], done: false };
+                case 5: _.label++; y = op[1]; op = [0]; continue;
+                case 7: op = _.ops.pop(); _.trys.pop(); continue;
+                default:
+                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
+                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
+                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
+                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
+                    if (t[2]) _.ops.pop();
+                    _.trys.pop(); continue;
+            }
+            op = body.call(thisArg, _);
+        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
+        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
+    }
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.QuotaCacheUpdater = void 0;
+exports.writeQuotaToCache = writeQuotaToCache;
+exports.startQuotaCacheService = startQuotaCacheService;
+var manager_1 = require("../manager");
+var fs_1 = __importDefault(require("fs"));
+var path_1 = __importDefault(require("path"));
+var os_1 = __importDefault(require("os"));
+var LOG_FILE = '/tmp/autopilot.log';
+function logToFile(message) {
+    var timestamp = new Date().toISOString();
+    fs_1.default.appendFileSync(LOG_FILE, "[".concat(timestamp, "] [QuotaCacheUpdater] ").concat(message, "\n"));
+}
+function getQuotaCachePath() {
+    var xdgConfigHome = process.env.XDG_CONFIG_HOME;
+    if (xdgConfigHome) {
+        return path_1.default.join(xdgConfigHome, 'opencode', 'quota-cache.json');
+    }
+    var homeDir = os_1.default.homedir();
+    return path_1.default.join(homeDir, '.config', 'opencode', 'quota-cache.json');
+}
+function writeQuotaToCache(quota) {
+    return __awaiter(this, void 0, void 0, function () {
+        var cache, cachePath, cacheDir;
+        return __generator(this, function (_a) {
+            try {
+                cache = {
+                    percentage: Math.round(quota.remainingFraction * 100),
+                    model: quota.model || 'unknown',
+                    timestamp: Date.now(),
+                };
+                cachePath = getQuotaCachePath();
+                cacheDir = path_1.default.dirname(cachePath);
+                if (!fs_1.default.existsSync(cacheDir)) {
+                    fs_1.default.mkdirSync(cacheDir, { recursive: true });
+                }
+                fs_1.default.writeFileSync(cachePath, JSON.stringify(cache, null, 2), 'utf-8');
+            }
+            catch (error) {
+                logToFile("Failed to write quota cache: ".concat(error));
+            }
+            return [2 /*return*/];
+        });
+    });
+}
+var DEFAULT_IDLE_POLL_INTERVAL_MS = 5 * 60 * 1000; // 5 minutes
+var QuotaCacheUpdater = /** @class */ (function () {
+    function QuotaCacheUpdater(manager, idlePollIntervalMs) {
+        if (idlePollIntervalMs === void 0) { idlePollIntervalMs = DEFAULT_IDLE_POLL_INTERVAL_MS; }
+        if (!(this instanceof QuotaCacheUpdater)) {
+            // @ts-ignore
+            return new QuotaCacheUpdater(manager, idlePollIntervalMs);
+        }
+        this.idleTimeoutId = null;
+        this.manager = manager;
+        this.idlePollIntervalMs = idlePollIntervalMs;
+        this.lastQueryTime = 0;
+    }
+    QuotaCacheUpdater.prototype.updateCache = function () {
+        return __awaiter(this, void 0, void 0, function () {
+            var quota, cache, cachePath, cacheDir, error_1;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        _a.trys.push([0, 2, , 3]);
+                        return [4 /*yield*/, this.manager.getQuotaViaApi()];
+                    case 1:
+                        quota = _a.sent();
+                        if (!quota) {
+                            return [2 /*return*/];
+                        }
+                        cache = {
+                            percentage: Math.round(quota.remainingFraction * 100),
+                            model: quota.model || 'unknown',
+                            timestamp: Date.now(),
+                        };
+                        cachePath = getQuotaCachePath();
+                        cacheDir = path_1.default.dirname(cachePath);
+                        if (!fs_1.default.existsSync(cacheDir)) {
+                            fs_1.default.mkdirSync(cacheDir, { recursive: true });
+                        }
+                        fs_1.default.writeFileSync(cachePath, JSON.stringify(cache, null, 2), 'utf-8');
+                        logToFile("Cache updated: ".concat(cache.percentage, "% remaining for ").concat(cache.model));
+                        return [3 /*break*/, 3];
+                    case 2:
+                        error_1 = _a.sent();
+                        logToFile("Failed to update quota cache: ".concat(error_1));
+                        return [3 /*break*/, 3];
+                    case 3: return [2 /*return*/];
+                }
+            });
+        });
+    };
+    /**
+     * Called after a query is made to refresh quota and reset idle timer.
+     */
+    QuotaCacheUpdater.prototype.onQueryCompleted = function () {
+        return __awaiter(this, void 0, void 0, function () {
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        this.lastQueryTime = Date.now();
+                        return [4 /*yield*/, this.updateCache()];
+                    case 1:
+                        _a.sent();
+                        this.resetIdleTimer();
+                        return [2 /*return*/];
+                }
+            });
+        });
+    };
+    /**
+     * Start the updater - polls immediately and sets up idle polling.
+     */
+    QuotaCacheUpdater.prototype.start = function () {
+        logToFile('Starting QuotaCacheUpdater');
+        this.lastQueryTime = Date.now();
+        this.updateCache();
+        this.resetIdleTimer();
+    };
+    /**
+     * Reset the idle timer. Called after each query.
+     */
+    QuotaCacheUpdater.prototype.resetIdleTimer = function () {
+        var _this = this;
+        if (this.idleTimeoutId) {
+            clearTimeout(this.idleTimeoutId);
+        }
+        this.idleTimeoutId = setTimeout(function () { return __awaiter(_this, void 0, void 0, function () {
+            var timeSinceLastQuery;
+            return __generator(this, function (_a) {
+                switch (_a.label) {
+                    case 0:
+                        timeSinceLastQuery = Date.now() - this.lastQueryTime;
+                        logToFile("Idle poll triggered. Time since last query: ".concat(Math.round(timeSinceLastQuery / 1000), "s"));
+                        return [4 /*yield*/, this.updateCache()];
+                    case 1:
+                        _a.sent();
+                        this.resetIdleTimer(); // Schedule next idle poll
+                        return [2 /*return*/];
+                }
+            });
+        }); }, this.idlePollIntervalMs);
+    };
+    QuotaCacheUpdater.prototype.stop = function () {
+        if (this.idleTimeoutId) {
+            clearTimeout(this.idleTimeoutId);
+            this.idleTimeoutId = null;
+        }
+        logToFile('QuotaCacheUpdater stopped');
+    };
+    return QuotaCacheUpdater;
+}());
+exports.QuotaCacheUpdater = QuotaCacheUpdater;
+function startQuotaCacheService() {
+    return __awaiter(this, arguments, void 0, function (idlePollIntervalMs) {
+        var manager, updater;
+        if (idlePollIntervalMs === void 0) { idlePollIntervalMs = DEFAULT_IDLE_POLL_INTERVAL_MS; }
+        return __generator(this, function (_a) {
+            switch (_a.label) {
+                case 0:
+                    manager = new manager_1.QuotaManager();
+                    return [4 /*yield*/, manager.initialize()];
+                case 1:
+                    _a.sent();
+                    updater = new QuotaCacheUpdater(manager, idlePollIntervalMs);
+                    updater.start();
+                    return [2 /*return*/, updater];
+            }
+        });
+    });
+}