chore: add api serve stop subcommand - update -h to -a

Author: louis-jan

cortex-js/src/command.module.ts

@@ -31,6 +31,7 @@ import { EmbeddingCommand } from './infrastructure/commanders/embeddings.command
 import { BenchmarkCommand } from './infrastructure/commanders/benchmark.command';
 import { EventEmitterModule } from '@nestjs/event-emitter';
 import { DownloadManagerModule } from './download-manager/download-manager.module';
+import { ServeStopCommand } from './infrastructure/commanders/sub-commands/serve-stop.command';
 
 @Module({
   imports: [
@@ -78,6 +79,9 @@ import { DownloadManagerModule } from './download-manager/download-manager.modul
 
     // Shortcuts
     RunCommand,
+
+    // Serve
+    ServeStopCommand,
   ],
 })
 export class CommandModule {}

cortex-js/src/infrastructure/commanders/serve.command.ts

@@ -1,38 +1,29 @@
 import { spawn } from 'child_process';
 import {
-  CORTEX_JS_STOP_API_SERVER_URL,
   defaultCortexJsHost,
   defaultCortexJsPort,
 } from '@/infrastructure/constants/cortex';
 import { CommandRunner, SubCommand, Option } from 'nest-commander';
 import { join } from 'path';
+import { ServeStopCommand } from './sub-commands/serve-stop.command';
 
 type ServeOptions = {
-  host?: string;
+  address?: string;
   port?: number;
   attach: boolean;
 };
 
 @SubCommand({
   name: 'serve',
   description: 'Providing API endpoint for Cortex backend',
+  subCommands: [ServeStopCommand],
 })
 export class ServeCommand extends CommandRunner {
   async run(_input: string[], options?: ServeOptions): Promise<void> {
-    const host = options?.host || defaultCortexJsHost;
+    const host = options?.address || defaultCortexJsHost;
     const port = options?.port || defaultCortexJsPort;
 
-    if (_input[0] === 'stop') {
-      return this.stopServer().then(() => console.log('API server stopped'));
-    } else {
-      return this.startServer(host, port, options);
-    }
-  }
-
-  private async stopServer() {
-    return fetch(CORTEX_JS_STOP_API_SERVER_URL(), {
-      method: 'DELETE',
-    }).catch(() => {});
+    return this.startServer(host, port, options);
   }
 
   private async startServer(
@@ -63,8 +54,8 @@ export class ServeCommand extends CommandRunner {
   }
 
   @Option({
-    flags: '-h, --host <host>',
-    description: 'Host to serve the application',
+    flags: '-a, --address <address>',
+    description: 'Address to use',
   })
   parseHost(value: string) {
     return value;

cortex-js/src/infrastructure/commanders/sub-commands/serve-stop.command.ts

@@ -0,0 +1,18 @@
+import { CORTEX_JS_STOP_API_SERVER_URL } from '@/infrastructure/constants/cortex';
+import { CommandRunner, SubCommand } from 'nest-commander';
+
+@SubCommand({
+  name: 'stop',
+  description: 'Stop the API server',
+})
+export class ServeStopCommand extends CommandRunner {
+  async run(): Promise<void> {
+    return this.stopServer().then(() => console.log('API server stopped'));
+  }
+
+  private async stopServer() {
+    return fetch(CORTEX_JS_STOP_API_SERVER_URL(), {
+      method: 'DELETE',
+    }).catch(() => {});
+  }
+}