@alwaysai/device-agent 0.0.10 → 0.0.11-internal-qa-2

package/src/index.ts

@@ -10,9 +10,10 @@ import { root } from './root';
 import { runDeviceAgentCloudInterface } from './cloud-connection/device-agent-cloud-connection';
 import { AgentConfigFile } from './infrastructure/agent-config';
 import { ALWAYSAI_DEVICE_AGENT_MODE } from './environment';
+import { logger } from './util/logger';
 
 if (module === require.main) {
-  console.log('Starting alwaysAI Device Agent');
+  logger.info('Starting alwaysAI Device Agent');
   if (!AgentConfigFile().exists()) {
     AgentConfigFile().initialize();
   }

package/src/infrastructure/agent-config.ts

@@ -82,14 +82,16 @@ export function AgentConfigFile(dir = ALWAYSAI_CONFIG_DIR) {
 
   async function getApps() {
     const config = configFile.read();
-    return config.applications;
+    return config && config.applications ? config.applications : [];
   }
 
   async function getReadyApps() {
     const config = configFile.read();
-    return config.applications.filter((app) => {
-      return app.ready;
-    });
+    return config && config.applications
+      ? config.applications.filter((app) => {
+          return app.ready;
+        })
+      : [];
   }
 
   async function getApp(props: { projectId: string }) {
@@ -124,6 +126,8 @@ export function AgentConfigFile(dir = ALWAYSAI_CONFIG_DIR) {
   async function removeApp(props: { projectId: string }) {
     const { projectId } = props;
     const config = configFile.read();
+    if (!config) return;
+
     const newList = config.applications.filter((app) => {
       return app.projectId !== projectId;
     });

package/src/infrastructure/certificates-and-tokens.ts

@@ -9,6 +9,7 @@ import {
   refreshDevice,
 } from 'alwaysai/lib/infrastructure';
 import { JsSpawner, Spawner, writeCertificate, writeTokens } from 'alwaysai/lib/util';
+import { logger } from '../util/logger';
 
 // NOTE: This is reimplemented from the CLI to work for local checks
 export async function getTargetHardwareUuid(spawner: Spawner) {
@@ -56,7 +57,7 @@ export async function writeCertificateAndToken(props: { deviceUuid: string }) {
   let iotKeys = device.iot_keys;
   if (!iotKeys) {
     // there are no certificates, we need to generate new ones
-    console.log('Generating new certificate');
+    logger.info('Generating new certificate');
     const response = await getDeviceCertificates(device.uuid, device.mode);
     iotKeys = response.iotKeys;
   }
@@ -67,5 +68,5 @@ export async function writeCertificateAndToken(props: { deviceUuid: string }) {
     privateKey: iotKeys.keyPair.PrivateKey,
     rootCa: iotKeys.rootCA,
   });
-  console.log('Updated tokens and certificate');
+  logger.info('Updated tokens and certificate');
 }

package/src/subcommands/app/app.ts

@@ -24,6 +24,7 @@ import {
   updateModels,
 } from '../../application-control';
 import { AgentConfigFile } from '../../infrastructure/agent-config';
+import { logger } from '../../util/logger';
 
 export const listAppsCliLeaf = CliLeaf({
   name: 'list',
@@ -47,7 +48,7 @@ export const listAppReleasesCliLeaf = CliLeaf({
   async action(_, opts) {
     const { project } = opts;
     const releaseHistory = await listAppReleases({ projectId: project });
-    console.log(releaseHistory);
+    logger.info(releaseHistory);
   },
 });
 
@@ -66,7 +67,7 @@ export const listAppLatestReleaseCliLeaf = CliLeaf({
     if (latestReleaseHash === undefined) {
       throw new CliTerseError('This application has not been published yet');
     }
-    console.log(latestReleaseHash);
+    logger.info(latestReleaseHash);
   },
 });
 
@@ -107,7 +108,7 @@ export const getAppStatusCliLeaf = CliLeaf({
   async action(_, opts) {
     const { project } = opts;
     const appStatus = await getAppStatus({ projectId: project });
-    console.log(appStatus);
+    logger.info(appStatus);
   },
 });
 
@@ -143,7 +144,7 @@ export const getAppLogsCliLeaf = CliLeaf({
     const readable = await getAppLogs({ projectId: project });
     readable.setEncoding('utf8');
     for await (const chunk of readable) {
-      console.log(chunk);
+      logger.info(chunk);
     }
   },
 });
@@ -307,7 +308,7 @@ export const getAllEnvsCLiLeaf = CliLeaf({
   },
   async action(_, opts) {
     const { project } = opts;
-    console.log(await getAllEnvs({ projectId: project }));
+    logger.info(await getAllEnvs({ projectId: project }));
   },
 });
 

package/src/subcommands/device/device.ts

@@ -8,6 +8,7 @@ import {
   getTargetHardwareUuid,
   writeCertificateAndToken,
 } from '../../infrastructure/certificates-and-tokens';
+import { logger } from '../../util/logger';
 
 export const initCliLeaf = CliLeaf({
   name: 'init',
@@ -24,7 +25,7 @@ export const initCliLeaf = CliLeaf({
   },
   async action(_, opts) {
     const { name, description } = opts;
-    console.log('Initializing device');
+    logger.info('Initializing device');
     await checkUserIsLoggedInComponent({ yes: true });
     if (!(await checkPaidPlan())) {
       throw new Error(`This action only supported for Enterprise alwaysAI accounts!`);
@@ -44,7 +45,7 @@ export const initCliLeaf = CliLeaf({
     };
     const response = await addDevice(deviceToAdd, 'production');
     await writeCertificateAndToken({ deviceUuid: response.deviceUUID });
-    console.log(`Initialized device as ${response.deviceUUID}`);
+    logger.info(`Initialized device as ${response.deviceUUID}`);
   },
 });
 

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

@@ -2,6 +2,7 @@ import { CliLeaf, CliNumberInput, CliStringInput } from '@alwaysai/alwayscli';
 import { appInstallModel } from 'alwaysai/lib/core/app';
 import { CliRpcClient } from 'alwaysai/lib/infrastructure';
 import { JsSpawner } from 'alwaysai/lib/util';
+import { logger } from '../util/logger';
 
 export const getModelPackageCliLeaf = CliLeaf({
   name: 'get-model-package',
@@ -27,6 +28,6 @@ export const getModelPackageCliLeaf = CliLeaf({
     const version =
       opts.version || (await CliRpcClient().getModelVersion({ id })).version;
     await appInstallModel(spawner, id, version);
-    console.log(`Completed downloading ${id} to ${path}`);
+    logger.info(`Completed downloading ${id} to ${path}`);
   },
 });

package/src/util/logger.ts

@@ -0,0 +1,28 @@
+import * as winston from 'winston';
+import 'winston-daily-rotate-file';
+import * as path from 'path';
+import { AAI_DIR } from 'alwaysai/lib/constants';
+import { ALWAYSAI_LOG_LEVEL, ALWAYSAI_LOG_TO_CONSOLE } from '../environment';
+
+const LOG_LEVEL = ALWAYSAI_LOG_LEVEL || 'info';
+const transports = ALWAYSAI_LOG_TO_CONSOLE
+  ? [new winston.transports.Console({ level: LOG_LEVEL })]
+  : [
+      new winston.transports.DailyRotateFile({
+        filename: path.join(AAI_DIR, 'agent-logs', 'agent-logs.txt'),
+        maxSize: '5m',
+        maxFiles: '2d',
+      }),
+    ];
+
+export const logger = winston.createLogger({
+  level: LOG_LEVEL,
+  format: winston.format.combine(
+    winston.format.errors({ stack: true }),
+    winston.format.splat(),
+    winston.format.simple(),
+  ),
+  transports,
+});
+
+logger.info(`Initialized logger with log level: ${LOG_LEVEL}`);