Added import handler

Author: kevinwang5658

src/connect/http-routes/handlers/import-handler.ts

@@ -0,0 +1,82 @@
+import { spawn } from '@homebridge/node-pty-prebuilt-multiarch';
+import { ConfigFileSchema } from 'codify-schemas';
+import fs from 'node:fs/promises';
+import os from 'node:os';
+import path from 'node:path';
+import { WebSocket } from 'ws';
+
+import { ConnectOrchestrator } from '../../../orchestrators/connect.js';
+import { ajv } from '../../../utils/ajv.js';
+import { Session, SocketServer } from '../../socket-server.js';
+import { ConnectCommand, createCommandHandler } from './create-command.js';
+
+enum ImportType {
+  REFRESH = 'refresh',
+  NEW_RESOURCES = 'new_resources',
+  NEW_ALL = 'new_all',
+}
+
+const validator = ajv.compile(ConfigFileSchema);
+
+export function importHandler() {
+  const spawnCommand = async (body: Record<string, unknown>, ws: WebSocket, session: Session) => {
+    const { config: codifyConfig, type, resourceTypes } = body;
+    if (!codifyConfig) {
+      throw new Error('Unable to parse codify config');
+    }
+
+    if (!type || !Object.values(ImportType).includes(type as ImportType)) {
+      throw new Error('Unable to parse import type');
+    }
+
+    if (type === ImportType.NEW_RESOURCES && (!resourceTypes || Array.isArray(resourceTypes))) {
+      throw new Error('For new resources import type, a list of resource types must be provided');
+    }
+
+    if (!validator(codifyConfig)) {
+      throw new Error('Invalid codify config');
+    }
+    
+    const tmpDir = await fs.mkdtemp(os.tmpdir());
+    const filePath = path.join(tmpDir, 'codify.jsonc');
+    await fs.writeFile(filePath, JSON.stringify(codifyConfig, null, 2));
+    session.additionalData.filePath = filePath;
+
+    let args = '';
+    switch (type as ImportType) {
+      case ImportType.REFRESH: { 
+        break;
+      }
+
+      case ImportType.NEW_RESOURCES: {
+        args = (resourceTypes as string[]).join(' ');
+        break;
+      }
+ 
+      case ImportType.NEW_ALL: {
+        args = '*'
+        break;
+      } 
+    }
+
+    return spawn('zsh', ['-c', `${ConnectOrchestrator.rootCommand} import ${args} -p ${filePath}`], {
+      name: 'xterm-color',
+      cols: 80,
+      rows: 30,
+      cwd: process.env.HOME,
+      env: process.env
+    });
+  }
+
+  const onExit = async (exitCode: number, ws: WebSocket, session: Session) => {
+    if (session.additionalData.filePath) {
+      await fs.rm(session.additionalData.filePath as string, { recursive: true, force: true });
+    }
+  }
+
+  return createCommandHandler({
+    name: ConnectCommand.APPLY,
+    spawnCommand,
+    onExit
+  });
+}

src/connect/http-routes/router.ts

@@ -1,17 +1,17 @@
 import { Router } from 'express';
 
-import { ConnectCommand, createCommandHandler } from './handlers/create-command.js';
-import defaultHandler from './handlers/index.js';
-import { terminalHandler } from './handlers/terminal-handler.js';
 import { applyHandler } from './handlers/apply-handler.js';
+import { importHandler } from './handlers/import-handler.js';
+import defaultHandler from './handlers/index.js';
 import { planHandler } from './handlers/plan-handler.js';
+import { terminalHandler } from './handlers/terminal-handler.js';
 
 const router = Router();
 
 router.use('/', defaultHandler);
 router.use('/apply', applyHandler());
 router.use('/plan', planHandler())
-// router.use('/import', createCommandHandler(ConnectCommand.IMPORT));
+router.use('/import', importHandler());
 router.use('/terminal', terminalHandler());
 
 export default router;