66 lines
2.3 KiB
TypeScript
66 lines
2.3 KiB
TypeScript
import { CloudAccountRepo } from '../ipc/database/cloudHandler';
|
|
import { GoogleAPIService } from './GoogleAPIService';
|
|
import { AutoSwitchService } from './AutoSwitchService';
|
|
import { logger } from '../utils/logger';
|
|
|
|
export class CloudMonitorService {
|
|
private static intervalId: NodeJS.Timeout | null = null;
|
|
private static POLL_INTERVAL = 1000 * 60 * 5; // 5 minutes
|
|
|
|
static start() {
|
|
if (this.intervalId) return;
|
|
logger.info('Starting CloudMonitorService...');
|
|
|
|
// Initial Poll
|
|
this.poll().catch((e) => logger.error('Initial poll failed', e));
|
|
|
|
this.intervalId = setInterval(() => {
|
|
this.poll().catch((e) => logger.error('Scheduled poll failed', e));
|
|
}, this.POLL_INTERVAL);
|
|
}
|
|
|
|
static stop() {
|
|
if (this.intervalId) {
|
|
clearInterval(this.intervalId);
|
|
this.intervalId = null;
|
|
logger.info('Stopped CloudMonitorService');
|
|
}
|
|
}
|
|
|
|
static async poll() {
|
|
logger.info('CloudMonitor: Polling quotas...');
|
|
const accounts = await CloudAccountRepo.getAccounts();
|
|
const now = Math.floor(Date.now() / 1000);
|
|
|
|
for (const account of accounts) {
|
|
try {
|
|
// 1. Check/Refresh Token if needed (give it a 10 min buffer here for safety)
|
|
let accessToken = account.token.access_token;
|
|
if (account.token.expiry_timestamp < now + 600) {
|
|
logger.info(`Monitor: Refreshing token for ${account.email}`);
|
|
const newToken = await GoogleAPIService.refreshAccessToken(account.token.refresh_token);
|
|
account.token.access_token = newToken.access_token;
|
|
account.token.expires_in = newToken.expires_in;
|
|
account.token.expiry_timestamp = now + newToken.expires_in;
|
|
await CloudAccountRepo.updateToken(account.id, account.token);
|
|
accessToken = newToken.access_token;
|
|
}
|
|
|
|
// 2. Fetch Quota
|
|
// We delay slightly between requests to act human/avoid spike
|
|
await new Promise((r) => setTimeout(r, 1000));
|
|
const quota = await GoogleAPIService.fetchQuota(accessToken);
|
|
|
|
// 3. Update DB
|
|
await CloudAccountRepo.updateQuota(account.id, quota);
|
|
} catch (error) {
|
|
logger.error(`Monitor: Failed to update ${account.email}`, error);
|
|
// Could mark status as 'error' or 'rate_limited' if 429
|
|
}
|
|
}
|
|
|
|
// 4. Check for Auto-Switch
|
|
await AutoSwitchService.checkAndSwitchIfNeeded();
|
|
}
|
|
}
|