Initial Commit

src/services/CloudMonitorService.ts

@@ -0,0 +1,65 @@
+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();
+  }
+}