File size: 2,257 Bytes
7421850 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 | /**
* @license
* Copyright 2025 Google LLC
* SPDX-License-Identifier: Apache-2.0
*/
import { Storage, debugLogger } from '@google/gemini-cli-core';
import * as fs from 'node:fs';
import * as path from 'node:path';
const STATE_FILENAME = 'state.json';
interface PersistentStateData {
defaultBannerShownCount?: Record<string, number>;
terminalSetupPromptShown?: boolean;
tipsShown?: number;
hasSeenScreenReaderNudge?: boolean;
focusUiEnabled?: boolean;
startupWarningCounts?: Record<string, number>;
// Add other persistent state keys here as needed
}
export class PersistentState {
private cache: PersistentStateData | null = null;
private filePath: string | null = null;
private getPath(): string {
if (!this.filePath) {
this.filePath = path.join(Storage.getGlobalGeminiDir(), STATE_FILENAME);
}
return this.filePath;
}
private load(): PersistentStateData {
if (this.cache) {
return this.cache;
}
try {
const filePath = this.getPath();
if (fs.existsSync(filePath)) {
const content = fs.readFileSync(filePath, 'utf-8');
// eslint-disable-next-line @typescript-eslint/no-unsafe-assignment
this.cache = JSON.parse(content);
} else {
this.cache = {};
}
} catch (error) {
debugLogger.warn('Failed to load persistent state:', error);
// If error reading (e.g. corrupt JSON), start fresh
this.cache = {};
}
return this.cache!;
}
private save() {
if (!this.cache) return;
try {
const filePath = this.getPath();
const dir = path.dirname(filePath);
if (!fs.existsSync(dir)) {
fs.mkdirSync(dir, { recursive: true });
}
fs.writeFileSync(filePath, JSON.stringify(this.cache, null, 2));
} catch (error) {
debugLogger.warn('Failed to save persistent state:', error);
}
}
get<K extends keyof PersistentStateData>(
key: K,
): PersistentStateData[K] | undefined {
return this.load()[key];
}
set<K extends keyof PersistentStateData>(
key: K,
value: PersistentStateData[K],
): void {
this.load(); // ensure loaded
this.cache![key] = value;
this.save();
}
}
export const persistentState = new PersistentState();
|