Initial Commit

src/ipc/cloud/router.ts

@@ -0,0 +1,79 @@
+import { z } from 'zod';
+import { os } from '@orpc/server';
+import {
+  addGoogleAccount,
+  listCloudAccounts,
+  deleteCloudAccount,
+  refreshAccountQuota,
+  switchCloudAccount,
+  getAutoSwitchEnabled,
+  setAutoSwitchEnabled,
+  forcePollCloudMonitor,
+  startAuthFlow,
+} from './handler';
+import { CloudAccountSchema } from '../../types/cloudAccount';
+import { CloudAccountRepo } from '../database/cloudHandler';
+
+export const cloudRouter = os.router({
+  addGoogleAccount: os
+    .input(z.object({ authCode: z.string() }))
+    .output(CloudAccountSchema)
+    .handler(async ({ input }) => {
+      return addGoogleAccount(input.authCode);
+    }),
+
+  listCloudAccounts: os.output(z.array(CloudAccountSchema)).handler(async () => {
+    return listCloudAccounts();
+  }),
+
+  deleteCloudAccount: os
+    .input(z.object({ accountId: z.string() }))
+    .output(z.void())
+    .handler(async ({ input }) => {
+      await deleteCloudAccount(input.accountId);
+    }),
+
+  refreshAccountQuota: os
+    .input(z.object({ accountId: z.string() }))
+    .output(CloudAccountSchema)
+    .handler(async ({ input }) => {
+      return refreshAccountQuota(input.accountId);
+    }),
+
+  switchCloudAccount: os
+    .input(z.object({ accountId: z.string() }))
+    .output(z.void())
+    .handler(async ({ input }) => {
+      await switchCloudAccount(input.accountId);
+    }),
+
+  getAutoSwitchEnabled: os.output(z.boolean()).handler(async () => {
+    return getAutoSwitchEnabled();
+  }),
+
+  setAutoSwitchEnabled: os
+    .input(z.object({ enabled: z.boolean() }))
+    .output(z.void())
+    .handler(async ({ input }) => {
+      await setAutoSwitchEnabled(input.enabled);
+    }),
+
+  forcePollCloudMonitor: os.output(z.void()).handler(async () => {
+    await forcePollCloudMonitor();
+  }),
+
+  startAuthFlow: os.output(z.void()).handler(async () => {
+    await startAuthFlow();
+  }),
+
+  syncLocalAccount: os.output(CloudAccountSchema.nullable()).handler(async () => {
+    try {
+      const result = await CloudAccountRepo.syncFromIDE();
+
+      return result;
+    } catch (error: any) {
+      console.error('[ORPC] syncLocalAccount error:', error.message, error.stack);
+      throw error;
+    }
+  }),
+});