@kapeta/local-cluster-service 0.6.1 → 0.7.1

package/{src → dist/esm/src}/utils/BlockInstanceRunner.js

@@ -1,23 +1,18 @@
-const {spawn} = require('node:child_process');
-const FS = require('node:fs');
-const Path = require('node:path');
-const {Docker} = require('node-docker-api');
-const ClusterConfig = require("@kapeta/local-cluster-config").default;
-const {readYML} = require("./utils");
-const {parseKapetaUri} = require("@kapeta/nodejs-utils");
-const serviceManager = require("../serviceManager");
-const containerManager = require("../containerManager");
-const LogData = require("./LogData");
-const EventEmitter = require("events");
-const md5 = require('md5');
-const {execSync} = require("child_process");
-const clusterService = require("../clusterService");
-
+import FS from 'node:fs';
+import ClusterConfig from '@kapeta/local-cluster-config';
+import { readYML } from './utils';
+import { parseKapetaUri } from '@kapeta/nodejs-utils';
+import { serviceManager } from '../serviceManager';
+import { containerManager } from '../containerManager';
+import { LogData } from './LogData';
+import EventEmitter from 'events';
+import md5 from 'md5';
+import { clusterService } from '../clusterService';
 const KIND_BLOCK_TYPE_OPERATOR = 'core/block-type-operator';
-const KAPETA_SYSTEM_ID = "KAPETA_SYSTEM_ID";
-const KAPETA_BLOCK_REF = "KAPETA_BLOCK_REF";
-const KAPETA_INSTANCE_ID = "KAPETA_INSTANCE_ID";
-const KAPETA_LOCAL_CLUSTER_PORT = "KAPETA_LOCAL_CLUSTER_PORT";
+const KAPETA_SYSTEM_ID = 'KAPETA_SYSTEM_ID';
+const KAPETA_BLOCK_REF = 'KAPETA_BLOCK_REF';
+const KAPETA_INSTANCE_ID = 'KAPETA_INSTANCE_ID';
+const KAPETA_LOCAL_CLUSTER_PORT = 'KAPETA_LOCAL_CLUSTER_PORT';
 /**
  * Needed when running local docker containers as part of plan
  * @type {string[]}
@@ -26,27 +21,22 @@ const DOCKER_ENV_VARS = [
     `KAPETA_LOCAL_SERVER=0.0.0.0`,
     `KAPETA_LOCAL_CLUSTER_HOST=host.docker.internal`,
     `KAPETA_ENVIRONMENT_TYPE=docker`,
-]
-
-
+];
 function getProvider(uri) {
-    return ClusterConfig.getProviderDefinitions().find(provider => {
-        const ref = `${provider.definition.metadata.name}:${provider.version}`
+    return ClusterConfig.getProviderDefinitions().find((provider) => {
+        const ref = `${provider.definition.metadata.name}:${provider.version}`;
         return parseKapetaUri(ref).id === uri.id;
     });
 }
-
 function getProviderPorts(assetVersion) {
-    return assetVersion.definition?.spec?.providers?.map(provider => {
-        return provider.spec?.port?.type
-    }).filter(t => !!t) ?? [];
+    return (assetVersion.definition?.spec?.providers
+        ?.map((provider) => {
+        return provider.spec?.port?.type;
+    })
+        .filter((t) => !!t) ?? []);
 }
-
-class BlockInstanceRunner {
-    /**
-     * @param {string} [planReference]
-     * @param {BlockInstanceInfo[]} [instances]
-     */
+export class BlockInstanceRunner {
+    _systemId;
     constructor(planReference) {
         /**
          *
@@ -55,206 +45,145 @@ class BlockInstanceRunner {
          */
         this._systemId = planReference ?? '';
     }
-
-
-
     /**
      * Start a block
      *
-     * @param {string} blockRef
-     * @param {string} instanceId
-     * @param {any} configuration
-     * @returns {Promise<ProcessInfo>}
      */
     async start(blockRef, instanceId, configuration) {
         return this._execute({
             ref: blockRef,
             id: instanceId,
-            configuration
+            configuration,
         });
     }
-
-    /**
-     *
-     * @param {BlockInstanceInfo} blockInstance
-     * @return {Promise<ProcessInfo>}
-     * @private
-     */
     async _execute(blockInstance) {
         const env = {};
-
         if (this._systemId) {
             env[KAPETA_SYSTEM_ID] = this._systemId;
         }
-
         if (blockInstance.ref) {
             env[KAPETA_BLOCK_REF] = blockInstance.ref;
         }
-
         if (blockInstance.id) {
             env[KAPETA_INSTANCE_ID] = blockInstance.id;
         }
-
         const blockUri = parseKapetaUri(blockInstance.ref);
-
         if (!blockUri.version) {
             blockUri.version = 'local';
         }
-
-        const assetVersion = ClusterConfig.getDefinitions().find(definitions => {
-            const ref = `${definitions.definition.metadata.name}:${definitions.version}`
+        const assetVersion = ClusterConfig.getDefinitions().find((definitions) => {
+            const ref = `${definitions.definition.metadata.name}:${definitions.version}`;
             return parseKapetaUri(ref).id === blockUri.id;
         });
-
         if (!assetVersion) {
             throw new Error(`Block definition not found: ${blockUri.id}`);
         }
-
         const kindUri = parseKapetaUri(assetVersion.definition.kind);
-
         const providerVersion = getProvider(kindUri);
-
         if (!providerVersion) {
             throw new Error(`Kind not found: ${kindUri.id}`);
         }
-
-        /**
-         * @type {ProcessDetails}
-         */
         let processDetails;
-
         if (providerVersion.definition.kind === KIND_BLOCK_TYPE_OPERATOR) {
             processDetails = await this._startOperatorProcess(blockInstance, blockUri, providerVersion, env);
-        } else {
+        }
+        else {
             //We need a port type to know how to connect to the block consistently
             const portTypes = getProviderPorts(assetVersion);
-
             if (blockUri.version === 'local') {
                 processDetails = await this._startLocalProcess(blockInstance, blockUri, env, assetVersion);
-            } else {
+            }
+            else {
                 processDetails = await this._startDockerProcess(blockInstance, blockUri, env);
             }
-
             if (portTypes.length > 0) {
                 processDetails.portType = portTypes[0];
             }
         }
-
         return {
+            name: blockUri.id,
             ...blockInstance,
-            ...processDetails
+            ...processDetails,
         };
     }
-
-
     /**
      * Starts local process
-     * @param {BlockInstanceInfo} blockInstance
-     * @param {BlockInfo} blockInfo
-     * @param {EnvironmentVariables} env
-     * @param assetVersion
-     * @return {ProcessDetails}
-     * @private
      */
     async _startLocalProcess(blockInstance, blockInfo, env, assetVersion) {
-        const baseDir = ClusterConfig.getRepositoryAssetPath(
-            blockInfo.handle,
-            blockInfo.name,
-            blockInfo.version
-        );
-
+        const baseDir = ClusterConfig.getRepositoryAssetPath(blockInfo.handle, blockInfo.name, blockInfo.version);
         if (!FS.existsSync(baseDir)) {
-            throw new Error(
-                `Local block not registered correctly - expected symlink here: ${baseDir}.\n` +
-                `Make sure you've run "blockctl registry link" in your local directory to connect it to Kapeta`
-            );
+            throw new Error(`Local block not registered correctly - expected symlink here: ${baseDir}.\n` +
+                `Make sure you've run "blockctl registry link" in your local directory to connect it to Kapeta`);
         }
-
         if (!assetVersion.definition.spec?.target?.kind) {
             throw new Error('Missing target kind in block definition');
         }
-
         const kindUri = parseKapetaUri(assetVersion.definition.spec?.target?.kind);
-
         const targetVersion = getProvider(kindUri);
-
         if (!targetVersion) {
             throw new Error(`Target not found: ${kindUri.id}`);
         }
-
         const localContainer = targetVersion.definition.spec.local;
-
         if (!localContainer) {
             throw new Error(`Missing local container information from target: ${kindUri.id}`);
         }
-
         const dockerImage = localContainer.image;
         if (!dockerImage) {
             throw new Error(`Missing docker image information: ${JSON.stringify(localContainer)}`);
         }
-
         const containerName = `kapeta-block-instance-${blockInstance.id}`;
         const logs = new LogData();
         logs.addLog(`Starting block ${blockInstance.ref}`);
-        let container = await containerManager.getContainerByName(containerName);
+        let container = (await containerManager.getContainerByName(containerName)) ?? null;
         console.log('Starting dev container', containerName);
-
         if (container) {
             console.log(`Container already exists. Deleting...`);
             try {
                 await container.delete({
-                    force: true
-                })
-            } catch (e) {
+                    force: true,
+                });
+            }
+            catch (e) {
                 throw new Error('Failed to delete existing container: ' + e.message);
             }
             container = null;
         }
-
         logs.addLog(`Creating new container for block: ${containerName}`);
         console.log('Creating new dev container', containerName, dockerImage);
         await containerManager.pull(dockerImage);
-
         const startCmd = localContainer.handlers?.onCreate ? localContainer.handlers.onCreate : '';
         const dockerOpts = localContainer.options ?? {};
         const homeDir = localContainer.userHome ? localContainer.userHome : '/root';
         const workingDir = localContainer.workingDir ? localContainer.workingDir : '/workspace';
-
         const ExposedPorts = {};
         const addonEnv = {};
         const PortBindings = {};
-
         const portTypes = getProviderPorts(assetVersion);
         let port = 80;
-        const promises = portTypes
-            .map(async (portType) => {
-                const publicPort = await serviceManager.ensureServicePort(this._systemId, blockInstance.id, portType);
-                const thisPort = port++; //TODO: Not sure how we should handle multiple ports or non-HTTP ports
-                const dockerPort = `${thisPort}/tcp`;
-                ExposedPorts[dockerPort] = {};
-                addonEnv[`KAPETA_LOCAL_SERVER_PORT_${portType.toUpperCase()}`] = thisPort;
-
-                PortBindings[dockerPort] = [
-                    {
-                        HostIp: "127.0.0.1", //No public
-                        HostPort: `${publicPort}`
-                    }
-                ];
-            });
-
+        const promises = portTypes.map(async (portType) => {
+            const publicPort = await serviceManager.ensureServicePort(this._systemId, blockInstance.id, portType);
+            const thisPort = port++; //TODO: Not sure how we should handle multiple ports or non-HTTP ports
+            const dockerPort = `${thisPort}/tcp`;
+            ExposedPorts[dockerPort] = {};
+            addonEnv[`KAPETA_LOCAL_SERVER_PORT_${portType.toUpperCase()}`] = '' + thisPort;
+            PortBindings[dockerPort] = [
+                {
+                    HostIp: '127.0.0.1',
+                    HostPort: `${publicPort}`,
+                },
+            ];
+        });
         await Promise.all(promises);
-
         let HealthCheck = undefined;
         if (localContainer.healthcheck) {
-            HealthCheck = containerManager.toDockerHealth({cmd: localContainer.healthcheck});
+            HealthCheck = containerManager.toDockerHealth({ cmd: localContainer.healthcheck });
         }
-
         container = await containerManager.startContainer({
             Image: dockerImage,
             name: containerName,
             WorkingDir: workingDir,
             Labels: {
-                'instance': blockInstance.id
+                instance: blockInstance.id,
             },
             HealthCheck,
             ExposedPorts,
@@ -264,118 +193,88 @@ class BlockInstanceRunner {
                 `KAPETA_LOCAL_CLUSTER_PORT=${clusterService.getClusterServicePort()}`,
                 ...Object.entries({
                     ...env,
-                    ...addonEnv
-                }).map(([key, value]) => `${key}=${value}`)
+                    ...addonEnv,
+                }).map(([key, value]) => `${key}=${value}`),
             ],
             HostConfig: {
                 Binds: [
                     `${ClusterConfig.getKapetaBasedir()}:${homeDir}/.kapeta`,
-                    `${baseDir}:${workingDir}` //We mount
+                    `${baseDir}:${workingDir}`, //We mount
                 ],
-                PortBindings
+                PortBindings,
             },
-            ...dockerOpts
+            ...dockerOpts,
         });
-
         try {
             if (HealthCheck) {
                 await containerManager.waitForHealthy(container);
-            } else {
+            }
+            else {
                 await containerManager.waitForReady(container);
             }
-        } catch (e) {
+        }
+        catch (e) {
             logs.addLog(e.message, 'ERROR');
         }
-
         return this._handleContainer(container, logs);
     }
-
-    /**
-     *
-     * @param container
-     * @param logs
-     * @param deleteOnExit
-     * @return {Promise<ProcessDetails>}
-     * @private
-     */
-    async _handleContainer(container, logs , deleteOnExit = false) {
-        const logStream = await container.logs({
+    async _handleContainer(container, logs, deleteOnExit = false) {
+        let localContainer = container;
+        const logStream = (await container.logs({
             follow: true,
             stdout: true,
             stderr: true,
-            tail: LogData.MAX_LINES
-        })
-
+            tail: LogData.MAX_LINES,
+        }));
         const outputEvents = new EventEmitter();
         logStream.on('data', (data) => {
             logs.addLog(data.toString());
             outputEvents.emit('data', data);
         });
-
         logStream.on('error', (data) => {
             logs.addLog(data.toString());
             outputEvents.emit('data', data);
         });
-
         logStream.on('close', async () => {
             const status = await container.status();
+            const data = status.data;
             if (deleteOnExit) {
                 try {
-                    await container.delete()
-                } catch (e) {}
+                    await container.delete();
+                }
+                catch (e) { }
             }
-            outputEvents.emit('exit', status.data?.State?.ExitCode ?? 0);
+            outputEvents.emit('exit', data?.State?.ExitCode ?? 0);
         });
-        /**
-         *
-         * @type {ProcessDetails}
-         */
         return {
             type: 'docker',
             pid: container.id,
             output: outputEvents,
             stop: async () => {
-                if (!container) {
+                if (!localContainer) {
                     return;
                 }
-
                 try {
-                    await container.stop();
+                    await localContainer.stop();
                     if (deleteOnExit) {
-                        await container.delete();
+                        await localContainer.delete();
                     }
-                } catch (e) {}
-                container = null;
+                }
+                catch (e) { }
+                localContainer = null;
             },
             logs: () => {
                 return logs.getLogs();
-            }
+            },
         };
     }
-
-
-    /**
-     * Starts local process using docker
-     * @param {BlockInstanceInfo} blockInstance
-     * @param {BlockInfo} blockInfo
-     * @param {EnvironmentVariables} env
-     * @return {Promise<ProcessDetails>}
-     * @private
-     */
     async _startDockerProcess(blockInstance, blockInfo, env) {
-        const {versionFile} = ClusterConfig.getRepositoryAssetInfoPath(
-            blockInfo.handle,
-            blockInfo.name,
-            blockInfo.version
-        );
-
+        const { versionFile } = ClusterConfig.getRepositoryAssetInfoPath(blockInfo.handle, blockInfo.name, blockInfo.version);
         const versionYml = versionFile;
         if (!FS.existsSync(versionYml)) {
             throw new Error(`Did not find version info at the expected path: ${versionYml}`);
         }
-
         const versionInfo = readYML(versionYml);
-
         if (versionInfo?.artifact?.type !== 'docker') {
             throw new Error(`Unsupported artifact type: ${versionInfo?.artifact?.type}`);
         }
@@ -383,49 +282,45 @@ class BlockInstanceRunner {
         if (!dockerImage) {
             throw new Error(`Missing docker image information: ${JSON.stringify(versionInfo?.artifact?.details)}`);
         }
-
         const containerName = `kapeta-block-instance-${blockInstance.id}`;
         const logs = new LogData();
         let container = await containerManager.getContainerByName(containerName);
-
         if (container) {
-            if (container.data.State === 'running') {
+            const containerData = container.data;
+            if (containerData.State === 'running') {
                 logs.addLog(`Found existing running container for block: ${containerName}`);
-            } else {
+            }
+            else {
                 logs.addLog(`Found existing container for block: ${containerName}. Starting now`);
                 await container.start();
             }
-        } else {
+        }
+        else {
             logs.addLog(`Creating new container for block: ${containerName}`);
             container = await containerManager.startContainer({
                 Image: dockerImage,
                 name: containerName,
                 Labels: {
-                    'instance': blockInstance.id
+                    instance: blockInstance.id,
                 },
                 Env: [
                     ...DOCKER_ENV_VARS,
                     `KAPETA_LOCAL_CLUSTER_PORT=${clusterService.getClusterServicePort()}`,
-                    ...Object.entries(env).map(([key, value]) => `${key}=${value}`)
+                    ...Object.entries(env).map(([key, value]) => `${key}=${value}`),
                 ],
                 HostConfig: {
-                    Binds: [
-                        `${ClusterConfig.getKapetaBasedir()}:${ClusterConfig.getKapetaBasedir()}`
-                    ],
-
-                }
+                    Binds: [`${ClusterConfig.getKapetaBasedir()}:${ClusterConfig.getKapetaBasedir()}`],
+                },
             });
-
             try {
                 await containerManager.waitForReady(container);
-            } catch (e) {
+            }
+            catch (e) {
                 logs.addLog(e.message, 'ERROR');
             }
         }
-
         return this._handleContainer(container, logs);
     }
-
     /**
      *
      * @param blockInstance
@@ -436,98 +331,87 @@ class BlockInstanceRunner {
      * @private
      */
     async _startOperatorProcess(blockInstance, blockUri, providerDefinition, env) {
-        const {assetFile} = ClusterConfig.getRepositoryAssetInfoPath(
-            blockUri.handle,
-            blockUri.name,
-            blockUri.version
-        );
-
+        const { assetFile } = ClusterConfig.getRepositoryAssetInfoPath(blockUri.handle, blockUri.name, blockUri.version);
         const kapetaYmlPath = assetFile;
         if (!FS.existsSync(kapetaYmlPath)) {
             throw new Error(`Did not find kapeta.yml at the expected path: ${kapetaYmlPath}`);
         }
-
         const spec = providerDefinition.definition.spec;
         const providerRef = `${providerDefinition.definition.metadata.name}:${providerDefinition.version}`;
-
         if (!spec?.local?.image) {
             throw new Error(`Provider did not have local image: ${providerRef}`);
         }
-
         const dockerImage = spec?.local?.image;
-
         try {
             await containerManager.pull(dockerImage);
-        } catch (e) {
+        }
+        catch (e) {
             console.warn('Failed to pull image. Continuing...', e);
         }
-
         const containerName = `kapeta-block-instance-${md5(blockInstance.id)}`;
         const logs = new LogData();
-        let container = await containerManager.getContainerByName(containerName);
-
+        let container = (await containerManager.getContainerByName(containerName)) ?? null;
         if (container) {
-            if (container.data.State === 'running') {
+            const containerData = container.data;
+            if (containerData.State === 'running') {
                 logs.addLog(`Found existing running container for block: ${containerName}`);
-            } else {
-                if (container.data.State?.ExitCode > 0) {
-                    logs.addLog(`Container exited with code: ${container.data.State.ExitCode}. Deleting...`);
+            }
+            else {
+                if (containerData.State?.ExitCode > 0) {
+                    logs.addLog(`Container exited with code: ${containerData.State.ExitCode}. Deleting...`);
                     try {
-                        await container.delete()
-                    } catch (e) {}
+                        await container.delete();
+                    }
+                    catch (e) { }
                     container = null;
-                } else {
+                }
+                else {
                     logs.addLog(`Found existing container for block: ${containerName}. Starting now`);
                     try {
                         await container.start();
-                    } catch (e) {
+                    }
+                    catch (e) {
                         console.warn('Failed to start container. Deleting...', e);
                         try {
-                            await container.delete()
-                        } catch (e) {}
+                            await container.delete();
+                        }
+                        catch (e) { }
                         container = null;
                     }
                 }
             }
         }
-
         if (!container) {
             const ExposedPorts = {};
             const addonEnv = {};
             const PortBindings = {};
             let HealthCheck = undefined;
             let Mounts = [];
-            const promises = Object.entries(spec.local.ports)
-                .map(async ([portType, value]) => {
-                    const dockerPort = `${value.port}/${value.type}`;
-                    ExposedPorts[dockerPort] = {};
-                    addonEnv[`KAPETA_LOCAL_SERVER_PORT_${portType.toUpperCase()}`] = value.port;
-                    const publicPort = await serviceManager.ensureServicePort(this._systemId, blockInstance.id, portType);
-                    PortBindings[dockerPort] = [
-                        {
-                            HostIp: "127.0.0.1", //No public
-                            HostPort: `${publicPort}`
-                        }
-                    ];
-                });
-
+            const promises = Object.entries(spec.local.ports).map(async ([portType, value]) => {
+                const dockerPort = `${value.port}/${value.type}`;
+                ExposedPorts[dockerPort] = {};
+                addonEnv[`KAPETA_LOCAL_SERVER_PORT_${portType.toUpperCase()}`] = value.port;
+                const publicPort = await serviceManager.ensureServicePort(this._systemId, blockInstance.id, portType);
+                PortBindings[dockerPort] = [
+                    {
+                        HostIp: '127.0.0.1',
+                        HostPort: `${publicPort}`,
+                    },
+                ];
+            });
             await Promise.all(promises);
-
             if (spec.local?.env) {
                 Object.entries(spec.local.env).forEach(([key, value]) => {
                     addonEnv[key] = value;
                 });
             }
-
             if (spec.local?.mounts) {
                 const mounts = containerManager.createMounts(blockUri.id, spec.local.mounts);
                 Mounts = containerManager.toDockerMounts(mounts);
             }
-
             if (spec.local?.health) {
                 HealthCheck = containerManager.toDockerHealth(spec.local?.health);
             }
-
             logs.addLog(`Creating new container for block: ${containerName}`);
             container = await containerManager.startContainer({
                 Image: dockerImage,
@@ -537,13 +421,13 @@ class BlockInstanceRunner {
                 HostConfig: {
                     Binds: [
                         `${kapetaYmlPath}:/kapeta.yml:ro`,
-                        `${ClusterConfig.getKapetaBasedir()}:${ClusterConfig.getKapetaBasedir()}`
+                        `${ClusterConfig.getKapetaBasedir()}:${ClusterConfig.getKapetaBasedir()}`,
                     ],
                     PortBindings,
-                    Mounts
+                    Mounts,
                 },
                 Labels: {
-                    'instance': blockInstance.id
+                    instance: blockInstance.id,
                 },
                 Env: [
                     `KAPETA_INSTANCE_NAME=${blockInstance.ref}`,
@@ -551,30 +435,27 @@ class BlockInstanceRunner {
                     ...DOCKER_ENV_VARS,
                     ...Object.entries({
                         ...env,
-                        ...addonEnv
-                    }).map(([key, value]) => `${key}=${value}`)
-                ]
+                        ...addonEnv,
+                    }).map(([key, value]) => `${key}=${value}`),
+                ],
             });
-
             try {
                 if (HealthCheck) {
                     await containerManager.waitForHealthy(container);
-                } else {
+                }
+                else {
                     await containerManager.waitForReady(container);
                 }
-            } catch (e) {
+            }
+            catch (e) {
                 logs.addLog(e.message, 'ERROR');
             }
         }
-
         const out = await this._handleContainer(container, logs, true);
         const portTypes = spec.local.ports ? Object.keys(spec.local.ports) : [];
         if (portTypes.length > 0) {
             out.portType = portTypes[0];
         }
-
         return out;
     }
 }
-
-module.exports = BlockInstanceRunner;