File size: 3,500 Bytes
4bf0fdd |
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 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 |
import fs from 'fs/promises';
import path from 'path';
import { getDataDir } from '../utils/paths.js';
import { FILE_CACHE_TTL } from '../constants/index.js';
import { log } from '../utils/logger.js';
/**
* 负责 token 文件的读写与简单缓存
* 不关心业务字段，只处理 JSON 数组的加载和保存
*/
class TokenStore {
constructor(filePath = path.join(getDataDir(), 'accounts.json')) {
this.filePath = filePath;
this._cache = null;
this._cacheTime = 0;
this._cacheTTL = FILE_CACHE_TTL;
}
async _ensureFileExists() {
const dir = path.dirname(this.filePath);
try {
await fs.mkdir(dir, { recursive: true });
} catch (e) {
// 目录已存在等情况忽略
}
try {
await fs.access(this.filePath);
} catch (e) {
// 文件不存在时创建空数组
await fs.writeFile(this.filePath, '[]', 'utf8');
log.info('✓ 已创建账号配置文件');
}
}
_isCacheValid() {
if (!this._cache) return false;
const now = Date.now();
return (now - this._cacheTime) < this._cacheTTL;
}
/**
* 读取全部 token（包含禁用的），带简单内存缓存
* @returns {Promise<Array<object>>}
*/
async readAll() {
if (this._isCacheValid()) {
return this._cache;
}
await this._ensureFileExists();
try {
const data = await fs.readFile(this.filePath, 'utf8');
const parsed = JSON.parse(data || '[]');
if (!Array.isArray(parsed)) {
log.warn('账号配置文件格式异常，已重置为空数组');
this._cache = [];
} else {
this._cache = parsed;
}
} catch (error) {
log.error('读取账号配置文件失败:', error.message);
this._cache = [];
}
this._cacheTime = Date.now();
return this._cache;
}
/**
* 覆盖写入全部 token，更新缓存
* @param {Array<object>} tokens
*/
async writeAll(tokens) {
await this._ensureFileExists();
const normalized = Array.isArray(tokens) ? tokens : [];
try {
await fs.writeFile(this.filePath, JSON.stringify(normalized, null, 2), 'utf8');
this._cache = normalized;
this._cacheTime = Date.now();
} catch (error) {
log.error('保存账号配置文件失败:', error.message);
throw error;
}
}
/**
* 根据内存中的启用 token 列表，将对应记录合并回文件
* - 仅按 refresh_token 匹配并更新已有记录
* - 未出现在 activeTokens 中的记录（例如已禁用账号）保持不变
* @param {Array<object>} activeTokens - 内存中的启用 token 列表（可能包含 sessionId）
* @param {object|null} tokenToUpdate - 如果只需要单个更新，可传入该 token 以减少遍历
*/
async mergeActiveTokens(activeTokens, tokenToUpdate = null) {
const allTokens = [...await this.readAll()];
const applyUpdate = (targetToken) => {
if (!targetToken) return;
const index = allTokens.findIndex(t => t.refresh_token === targetToken.refresh_token);
if (index !== -1) {
const { sessionId, ...plain } = targetToken;
allTokens[index] = { ...allTokens[index], ...plain };
}
};
if (tokenToUpdate) {
applyUpdate(tokenToUpdate);
} else if (Array.isArray(activeTokens) && activeTokens.length > 0) {
for (const memToken of activeTokens) {
applyUpdate(memToken);
}
}
await this.writeAll(allTokens);
}
}
export default TokenStore;
|