@alwaysai/device-agent 0.0.11 → 0.0.13

package/src/subcommands/app/app.ts

@@ -2,7 +2,7 @@ import {
   CliLeaf,
   CliStringArrayInput,
   CliStringInput,
-  CliTerseError,
+  CliTerseError
 } from '@alwaysai/alwayscli';
 import {
   addModel,
@@ -21,7 +21,7 @@ import {
   startApp,
   stopApp,
   uninstallApp,
-  updateModels,
+  updateModels
 } from '../../application-control';
 import { AgentConfigFile } from '../../infrastructure/agent-config';
 import { logger } from '../../util/logger';
@@ -33,7 +33,7 @@ export const listAppsCliLeaf = CliLeaf({
   async action(_, opts) {
     const apps = await AgentConfigFile().getApps();
     console.table(apps);
-  },
+  }
 });
 
 export const listAppReleasesCliLeaf = CliLeaf({
@@ -42,14 +42,14 @@ export const listAppReleasesCliLeaf = CliLeaf({
   namedInputs: {
     project: CliStringInput({
       description: 'Project ID',
-      required: true,
-    }),
+      required: true
+    })
   },
   async action(_, opts) {
     const { project } = opts;
     const releaseHistory = await listAppReleases({ projectId: project });
     logger.info(releaseHistory);
-  },
+  }
 });
 
 export const listAppLatestReleaseCliLeaf = CliLeaf({
@@ -58,17 +58,19 @@ export const listAppLatestReleaseCliLeaf = CliLeaf({
   namedInputs: {
     project: CliStringInput({
       description: 'Project ID',
-      required: true,
-    }),
+      required: true
+    })
   },
   async action(_, opts) {
     const { project } = opts;
-    const latestReleaseHash = await listAppLatestRelease({ projectId: project });
+    const latestReleaseHash = await listAppLatestRelease({
+      projectId: project
+    });
     if (latestReleaseHash === undefined) {
       throw new CliTerseError('This application has not been published yet');
     }
     logger.info(latestReleaseHash);
-  },
+  }
 });
 
 export const installAppCliLeaf = CliLeaf({
@@ -77,11 +79,11 @@ export const installAppCliLeaf = CliLeaf({
   namedInputs: {
     project: CliStringInput({
       description: 'Project ID',
-      required: true,
+      required: true
     }),
     releaseHash: CliStringInput({
-      description: 'Release Hash',
-    }),
+      description: 'Release Hash'
+    })
   },
   async action(_, opts) {
     const project = opts.project;
@@ -93,7 +95,7 @@ export const installAppCliLeaf = CliLeaf({
       throw new CliTerseError('This application has not been published yet');
     }
     await installApp({ projectId: project, appReleaseHash: releaseHash });
-  },
+  }
 });
 
 export const getAppStatusCliLeaf = CliLeaf({
@@ -102,14 +104,14 @@ export const getAppStatusCliLeaf = CliLeaf({
   namedInputs: {
     project: CliStringInput({
       description: 'Project ID',
-      required: true,
-    }),
+      required: true
+    })
   },
   async action(_, opts) {
     const { project } = opts;
     const appStatus = await getAppStatus({ projectId: project });
     logger.info(appStatus);
-  },
+  }
 });
 
 export const startAppCliLeaf = CliLeaf({
@@ -118,16 +120,16 @@ export const startAppCliLeaf = CliLeaf({
   namedInputs: {
     project: CliStringInput({
       description: 'Project ID',
-      required: true,
+      required: true
     }),
     dockerLoginToken: CliStringInput({
-      description: 'Docker login token',
-    }),
+      description: 'Docker login token'
+    })
   },
   async action(_, opts) {
     const { project, dockerLoginToken } = opts;
     await startApp({ projectId: project, dockerLoginToken });
-  },
+  }
 });
 
 export const getAppLogsCliLeaf = CliLeaf({
@@ -136,8 +138,8 @@ export const getAppLogsCliLeaf = CliLeaf({
   namedInputs: {
     project: CliStringInput({
       description: 'Project ID',
-      required: true,
-    }),
+      required: true
+    })
   },
   async action(_, opts) {
     const { project } = opts;
@@ -146,7 +148,7 @@ export const getAppLogsCliLeaf = CliLeaf({
     for await (const chunk of readable) {
       logger.info(chunk);
     }
-  },
+  }
 });
 
 export const stopAppCliLeaf = CliLeaf({
@@ -155,13 +157,13 @@ export const stopAppCliLeaf = CliLeaf({
   namedInputs: {
     project: CliStringInput({
       description: 'Project ID',
-      required: true,
-    }),
+      required: true
+    })
   },
   async action(_, opts) {
     const { project } = opts;
     await stopApp({ projectId: project });
-  },
+  }
 });
 
 export const restartAppCliLeaf = CliLeaf({
@@ -170,13 +172,13 @@ export const restartAppCliLeaf = CliLeaf({
   namedInputs: {
     project: CliStringInput({
       description: 'Project ID',
-      required: true,
-    }),
+      required: true
+    })
   },
   async action(_, opts) {
     const { project } = opts;
     await restartApp({ projectId: project });
-  },
+  }
 });
 
 export const uninstallAppCliLeaf = CliLeaf({
@@ -185,13 +187,13 @@ export const uninstallAppCliLeaf = CliLeaf({
   namedInputs: {
     project: CliStringInput({
       description: 'Project ID',
-      required: true,
-    }),
+      required: true
+    })
   },
   async action(_, opts) {
     const { project } = opts;
     await uninstallApp({ projectId: project });
-  },
+  }
 });
 
 export const rollbackAppCliLeaf = CliLeaf({
@@ -200,13 +202,13 @@ export const rollbackAppCliLeaf = CliLeaf({
   namedInputs: {
     project: CliStringInput({
       description: 'Project ID',
-      required: true,
-    }),
+      required: true
+    })
   },
   async action(_, opts) {
     const { project } = opts;
     await rollbackApp({ projectId: project });
-  },
+  }
 });
 
 export const showAppModelsCliLeaf = CliLeaf({
@@ -215,14 +217,14 @@ export const showAppModelsCliLeaf = CliLeaf({
   namedInputs: {
     project: CliStringInput({
       description: 'Project ID',
-      required: true,
-    }),
+      required: true
+    })
   },
   async action(_, opts) {
     const { project } = opts;
     const appModels = await getAppModels({ projectId: project });
     console.table(appModels);
-  },
+  }
 });
 
 export const addModelCliLeaf = CliLeaf({
@@ -231,17 +233,17 @@ export const addModelCliLeaf = CliLeaf({
   namedInputs: {
     project: CliStringInput({
       description: 'Project ID',
-      required: true,
+      required: true
     }),
     model: CliStringInput({
       description: 'Model ID',
-      required: true,
-    }),
+      required: true
+    })
   },
   async action(_, opts) {
     const { project, model } = opts;
     await addModel({ projectId: project, modelId: model });
-  },
+  }
 });
 
 export const removeModelCliLeaf = CliLeaf({
@@ -250,17 +252,17 @@ export const removeModelCliLeaf = CliLeaf({
   namedInputs: {
     project: CliStringInput({
       description: 'Project ID',
-      required: true,
+      required: true
     }),
     model: CliStringInput({
       description: 'Model ID',
-      required: true,
-    }),
+      required: true
+    })
   },
   async action(_, opts) {
     const { project, model } = opts;
     await removeModel({ projectId: project, modelId: model });
-  },
+  }
 });
 
 export const replaceModelsCliLeaf = CliLeaf({
@@ -269,17 +271,17 @@ export const replaceModelsCliLeaf = CliLeaf({
   namedInputs: {
     project: CliStringInput({
       description: 'Project Id',
-      required: true,
+      required: true
     }),
     models: CliStringArrayInput({
       description: 'One or more model IDs',
-      required: true,
-    }),
+      required: true
+    })
   },
   async action(_, opts) {
     const { project, models } = opts;
     await replaceModels({ projectId: project, modelIds: models });
-  },
+  }
 });
 
 export const updateModelsCliLeaf = CliLeaf({
@@ -288,13 +290,13 @@ export const updateModelsCliLeaf = CliLeaf({
   namedInputs: {
     project: CliStringInput({
       description: 'Project Id',
-      required: true,
-    }),
+      required: true
+    })
   },
   async action(_, opts) {
     const { project } = opts;
     await updateModels({ projectId: project });
-  },
+  }
 });
 
 export const getAllEnvsCLiLeaf = CliLeaf({
@@ -303,13 +305,13 @@ export const getAllEnvsCLiLeaf = CliLeaf({
   namedInputs: {
     project: CliStringInput({
       description: 'Project Id',
-      required: true,
-    }),
+      required: true
+    })
   },
   async action(_, opts) {
     const { project } = opts;
     logger.info(await getAllEnvs({ projectId: project }));
-  },
+  }
 });
 
 export const setEnvCLiLeaf = CliLeaf({
@@ -317,21 +319,21 @@ export const setEnvCLiLeaf = CliLeaf({
   description: 'Set environment variables for a service',
   positionalInput: CliStringArrayInput({
     placeholder: '<NAME=VALUE> [<NAME=VALUE> ...]',
-    required: true,
+    required: true
   }),
   namedInputs: {
     project: CliStringInput({
       description: 'Project Id',
-      required: true,
+      required: true
     }),
     service: CliStringInput({
       description:
         'The name of the docker-compose service to apply environment variable to',
-      required: false,
-    }),
+      required: false
+    })
   },
   async action(args, opts) {
     const { project, service } = opts;
     await setEnv({ projectId: project, vars: args, service });
-  },
+  }
 });

package/src/subcommands/app/index.ts

@@ -17,7 +17,7 @@ import {
   replaceModelsCliLeaf,
   showAppModelsCliLeaf,
   getAllEnvsCLiLeaf,
-  setEnvCLiLeaf,
+  setEnvCLiLeaf
 } from './app';
 
 export const appCliBranch = CliBranch({
@@ -41,6 +41,6 @@ export const appCliBranch = CliBranch({
     replaceModelsCliLeaf,
     updateModelsCliLeaf,
     getAllEnvsCLiLeaf,
-    setEnvCLiLeaf,
-  ],
+    setEnvCLiLeaf
+  ]
 });

package/src/subcommands/device/clean.ts

@@ -0,0 +1,26 @@
+import { CliLeaf } from '@alwaysai/alwayscli';
+import * as rimraf from 'rimraf';
+import { LOCAL_CERT_AND_KEY_DIR } from 'alwaysai/lib/constants';
+import { logger } from '../../util/logger';
+import { AgentConfigFile } from '../../infrastructure/agent-config';
+import { DeviceConfigFile } from 'alwaysai/lib/core/device';
+import { APP_ROOT, CREDENTIALS_FILE_PATH } from '../../util/directories';
+
+export const cleanCliLeaf = CliLeaf({
+  name: 'clean',
+  description: 'Remove all provisioning files',
+  async action(_, opts) {
+    logger.info('Cleaning device configuration');
+    logger.debug(`Removing ${LOCAL_CERT_AND_KEY_DIR}`);
+    rimraf.sync(LOCAL_CERT_AND_KEY_DIR);
+    logger.debug(`Removing ${AgentConfigFile().path}`);
+    AgentConfigFile().remove();
+    logger.debug(`Removing ${DeviceConfigFile().path}`);
+    DeviceConfigFile().remove();
+    logger.debug(`Removing ${CREDENTIALS_FILE_PATH}`);
+    rimraf.sync(CREDENTIALS_FILE_PATH);
+    logger.debug(`Removing ${APP_ROOT}`);
+    rimraf.sync(APP_ROOT);
+    logger.info('Device configuration cleaned');
+  }
+});

package/src/subcommands/device/device.ts

@@ -1,45 +1,49 @@
-import { CliLeaf, CliStringInput } from "@alwaysai/alwayscli";
-import { checkUserIsLoggedInComponent } from "alwaysai/lib/components/user";
-import { checkPaidPlan } from "alwaysai/lib/core/project";
-import { v4 as uuidv4 } from "uuid";
-import { CliAuthenticationClient } from "alwaysai/lib/infrastructure";
-
-import { httpClient, microServiceHttpClient } from "../../util/http-client";
+import { CliLeaf, CliStringInput } from '@alwaysai/alwayscli';
+import { checkUserIsLoggedInComponent } from 'alwaysai/lib/components/user';
+import { getTargetHardwareUuid } from 'alwaysai/lib/core/app';
+import { checkPaidPlan } from 'alwaysai/lib/core/project';
+import { v4 as uuidv4 } from 'uuid';
+import { CliAuthenticationClient } from 'alwaysai/lib/infrastructure';
+import { existsSync } from 'fs';
+import { httpClient, microServiceHttpClient } from '../../util/http-client';
 import {
   AWS_ROOT_CERTIFICATE_FILE_NAME,
   BOOTSTRAP_CERTIFICATES_DIR_PATH,
-} from "../../util/directories";
+  getPrivateKeyFilePath
+} from '../../util/directories';
 
-import { LOCAL_CERT_AND_KEY_DIR } from "alwaysai/lib/constants";
-import fetch from "node-fetch";
-import { JsSpawner } from "alwaysai/lib/util";
+import { LOCAL_CERT_AND_KEY_DIR } from 'alwaysai/lib/constants';
+import { JsSpawner } from 'alwaysai/lib/util';
 import {
   getCpuUtil,
   getDiskUtil,
-  getMemUtil,
-} from "../../device-control/device-control";
-import {
-  getTargetHardwareUuid,
-  writeCertificateAndToken,
-} from "../../infrastructure/certificates-and-tokens";
-import { logger } from "../../util/logger";
+  getMemUtil
+} from '../../device-control/device-control';
+import { writeTokenAndDeviceCfg } from '../../infrastructure/tokens-and-device-cfg';
+import { logger } from '../../util/logger';
+import { DeviceConfigFile } from 'alwaysai/lib/core/device';
 
 export const initCliLeaf = CliLeaf({
-  name: "init",
-  description: "Initialize device",
+  name: 'init',
+  description: 'Initialize device',
   namedInputs: {
     name: CliStringInput({
-      description: "Device name",
-      required: true,
+      description: 'Device name',
+      required: true
     }),
     description: CliStringInput({
-      description: "Device description",
-      required: false,
-    }),
+      description: 'Device description',
+      required: false
+    })
   },
   async action(_, opts) {
     const { name, description } = opts;
-    logger.info("Initializing device");
+    if (DeviceConfigFile().exists() || existsSync(getPrivateKeyFilePath())) {
+      throw new Error(
+        "Device has been previously provisioned. Run 'aai-agent device clean' to re-provision"
+      );
+    }
+    logger.info(`Initializing device ${name}`);
     await checkUserIsLoggedInComponent({ yes: true });
     if (!(await checkPaidPlan())) {
       throw new Error(
@@ -47,54 +51,66 @@ export const initCliLeaf = CliLeaf({
       );
     }
     const { username } = await CliAuthenticationClient().getInfo();
-    const spawner = JsSpawner();
-    const hardwareId = await getTargetHardwareUuid(spawner);
+    const hardwareId = await getTargetHardwareUuid(JsSpawner());
     const device = {
-      deviceMode: "production",
+      deviceMode: 'production',
       hardwareId,
       txId: uuidv4(),
       userName: username,
-      friendlyName: name,
+      friendlyName: name
     };
 
-    const response = await microServiceHttpClient(
-      "fleet-provision",
-      "addDevice",
-      "POST",
-      JSON.stringify(device)
-    );
+    const response: { deviceUuid: string; claimCertificate: string[] } =
+      await microServiceHttpClient(
+        'fleet-provision',
+        'addDevice',
+        'POST',
+        JSON.stringify(device)
+      );
+    logger.debug(`addDevice Response: ${JSON.stringify(response, null, 2)}`);
 
+    if (!Object.hasOwn(response, 'claimCertificate')) {
+      throw new Error(
+        "Device cannot be provisioned. Run 'aai-agent device clean' and retry."
+      );
+    }
     const ROOT_CERT_AWS = await httpClient(
-      "https://www.amazontrust.com/repository/AmazonRootCA1.pem",
-      "GET"
+      'https://www.amazontrust.com/repository/AmazonRootCA1.pem',
+      'GET',
+      {}
     );
 
-    await writeCertificateAndToken({ deviceUuid: response.deviceUuid });
+    await writeTokenAndDeviceCfg({
+      deviceUuid: response.deviceUuid
+    });
 
-    await JsSpawner().mkdirp(BOOTSTRAP_CERTIFICATES_DIR_PATH);
-    JsSpawner({ path: LOCAL_CERT_AND_KEY_DIR }).writeFile(
+    await JsSpawner().mkdirp(BOOTSTRAP_CERTIFICATES_DIR_PATH());
+
+    await JsSpawner({ path: LOCAL_CERT_AND_KEY_DIR }).writeFile(
       AWS_ROOT_CERTIFICATE_FILE_NAME,
       ROOT_CERT_AWS
     );
-    const certSpawner = JsSpawner({ path: BOOTSTRAP_CERTIFICATES_DIR_PATH });
+    const certSpawner = JsSpawner({
+      path: BOOTSTRAP_CERTIFICATES_DIR_PATH()
+    });
     for (const key in response.claimCertificate) {
       await certSpawner.writeFile(key, response.claimCertificate[key]);
     }
 
-    logger.info(`Initialized device as ${response.deviceUuid}`);
-  },
+    logger.info(`Initialized device ${name}: UUID=${response.deviceUuid}`);
+  }
 });
 
 export const getInfoCliLeaf = CliLeaf({
-  name: "get-info",
-  description: "Get device info",
+  name: 'get-info',
+  description: 'Get device info',
   namedInputs: {},
   async action(_, opts) {
     const deviceInfo = [
-      ["CPU Utilization", `${String(await getCpuUtil())} %`],
-      ["Disk Utilization", `${String(await getDiskUtil())} %`],
-      ["Memory Utilization", `${String(await getMemUtil())} %`],
+      ['CPU Utilization', `${String(await getCpuUtil())} %`],
+      ['Disk Utilization', `${String(await getDiskUtil())} %`],
+      ['Memory Utilization', `${String(await getMemUtil())} %`]
     ];
     console.table(deviceInfo);
-  },
+  }
 });

package/src/subcommands/device/index.ts

@@ -1,8 +1,9 @@
 import { CliBranch } from '@alwaysai/alwayscli';
 import { getInfoCliLeaf, initCliLeaf } from './device';
+import { cleanCliLeaf } from './clean';
 
 export const deviceCliBranch = CliBranch({
   name: 'device',
   description: 'Manage current device',
-  subcommands: [initCliLeaf, getInfoCliLeaf],
+  subcommands: [initCliLeaf, getInfoCliLeaf, cleanCliLeaf]
 });

package/src/subcommands/get-model-package.ts

@@ -10,17 +10,17 @@ export const getModelPackageCliLeaf = CliLeaf({
   positionalInput: CliStringInput({
     description: 'For example, "alwaysai/mobilenet_ssd"',
     required: true,
-    placeholder: '<id>',
+    placeholder: '<id>'
   }),
   namedInputs: {
     version: CliNumberInput({
       description: 'The version number of the model',
-      required: false,
+      required: false
     }),
     path: CliStringInput({
       description: 'The output path to write model package to',
-      required: false,
-    }),
+      required: false
+    })
   },
   async action(id, opts) {
     const path = opts.path || process.cwd();
@@ -29,5 +29,5 @@ export const getModelPackageCliLeaf = CliLeaf({
       opts.version || (await CliRpcClient().getModelVersion({ id })).version;
     await appInstallModel(spawner, id, version);
     logger.info(`Completed downloading ${id} to ${path}`);
-  },
+  }
 });

package/src/subcommands/index.ts

@@ -7,5 +7,5 @@ export const subcommands = [
   loginCliLeaf,
   appCliBranch,
   deviceCliBranch,
-  getModelPackageCliLeaf,
+  getModelPackageCliLeaf
 ];

package/src/subcommands/login.ts

@@ -1,6 +1,6 @@
 import { CliLeaf, CliStringInput } from '@alwaysai/alwayscli';
 import { alwaysaiUserLoginYesComponent } from 'alwaysai/lib/components/user';
-import { writeCertificateAndToken } from '../infrastructure/certificates-and-tokens';
+import { writeTokenAndDeviceCfg } from '../infrastructure/tokens-and-device-cfg';
 
 export const loginCliLeaf = CliLeaf({
   name: 'login',
@@ -8,26 +8,26 @@ export const loginCliLeaf = CliLeaf({
   namedInputs: {
     email: CliStringInput({
       description: 'alwaysAI email',
-      required: true,
+      required: true
     }),
     password: CliStringInput({
       description: 'Account password',
-      required: true,
+      required: true
     }),
     device: CliStringInput({
       description: 'The device UUID',
-      required: false,
-    }),
+      required: false
+    })
   },
   async action(_, opts) {
     const { email, password, device } = opts;
 
     await alwaysaiUserLoginYesComponent({
       alwaysaiUserEmail: email,
-      alwaysaiUserPassword: password,
+      alwaysaiUserPassword: password
     });
     if (device) {
-      await writeCertificateAndToken({ deviceUuid: device });
+      await writeTokenAndDeviceCfg({ deviceUuid: device });
     }
-  },
+  }
 });

package/src/util/clean-certs.ts

@@ -0,0 +1,12 @@
+import { JsSpawner } from 'alwaysai/lib/util';
+import { logger } from '../util/logger';
+import { LOCAL_CERT_AND_KEY_DIR } from 'alwaysai/lib/constants';
+import { BOOTSTRAP_CERTIFICATES_DIR_PATH } from '../util/directories';
+
+export const rmBootstrapCertsAndClose = async () => {
+  const spawner = JsSpawner();
+  await spawner.rimraf(BOOTSTRAP_CERTIFICATES_DIR_PATH());
+  await spawner.rimraf(LOCAL_CERT_AND_KEY_DIR);
+  logger.error('Could not provision device. Try again.');
+  process.exit(1);
+};