@alwaysai/device-agent 1.3.0-1 → 1.3.1-1

package/src/cloud-connection/cmd-status.ts

@@ -1,71 +0,0 @@
-import { logger } from '../util/logger';
-
-type CmdStatusType = 'idle' | 'in_progress';
-
-class CmdStatus {
-  private projectId: string;
-  private status: CmdStatusType;
-
-  constructor(projectId: string, status: CmdStatusType) {
-    this.projectId = projectId;
-    this.status = status;
-  }
-
-  public getProjectId() {
-    return this.projectId;
-  }
-
-  public update(status: CmdStatusType) {
-    this.status = status;
-  }
-
-  public getStatus(): CmdStatusType {
-    return this.status;
-  }
-}
-
-export class CmdStatusManager {
-  private apps: { [projectId: string]: CmdStatus } = {};
-
-  public async start(projectId: string) {
-    if (!(projectId in this.apps)) {
-      const cmdStatus = new CmdStatus(projectId, 'in_progress');
-      this.apps[projectId] = cmdStatus;
-    } else if (this.apps[projectId].getStatus() !== 'in_progress') {
-      this.apps[projectId].update('in_progress');
-    } else {
-      logger.debug(
-        `Ignoring start for ${projectId} since it already has a command in progress`
-      );
-      return;
-    }
-    logger.debug(`Started command for ${projectId}`);
-  }
-
-  public async stop(projectId: string) {
-    if (
-      !(projectId in this.apps) ||
-      this.apps[projectId].getStatus() === 'idle'
-    ) {
-      throw new Error(`No ongoing command to stop for ${projectId}`);
-    }
-    this.apps[projectId].update('idle');
-    logger.debug(`Stopped command for ${projectId}`);
-  }
-
-  public isCmdInProgress(projectId: string): boolean {
-    if (!(projectId in this.apps)) {
-      return false;
-    }
-    return this.apps[projectId].getStatus() === 'in_progress';
-  }
-
-  public anyCmdInProgress() {
-    for (const projectId in this.apps) {
-      if (this.apps[projectId].getStatus() === 'in_progress') {
-        return true;
-      }
-    }
-    return false;
-  }
-}

package/src/cloud-connection/message-builder.ts

@@ -1,117 +0,0 @@
-import {
-  AppLogsMessage,
-  AppLogsMessageType,
-  AppLogsPayload,
-  AppStateMessage,
-  AppStateMessageType,
-  AppStatePayload,
-  BaseToClientMessage,
-  DeviceStatsMessage,
-  DeviceStatsMessageType,
-  DeviceStatsPayload,
-  SignedUrlsRequestMessage,
-  SignedUrlsRequestMessageType,
-  SignedUrlsRequestPayload,
-  StatusResponseMessage,
-  StatusResponseMessageType,
-  StatusResponsePayload,
-  generateTxId,
-  getToClientTopic,
-  keyMirrors,
-  validateToDeviceAgentMessage
-} from '@alwaysai/device-agent-schemas';
-import { getDeviceUuid } from '../util/get-device-id';
-
-function buildBaseClientMessage(txId?: string): BaseToClientMessage {
-  const clientId = getDeviceUuid();
-  const baseClientMessage = {
-    timestamp: new Date().toUTCString(),
-    deviceId: clientId,
-    topic: getToClientTopic(clientId),
-    txId: txId || generateTxId()
-  };
-  return baseClientMessage;
-}
-
-// Public Message Builders
-
-// app logs
-export async function buildAppLogsMessage(
-  payload: AppLogsPayload,
-  txId?: string
-): Promise<AppLogsMessage> {
-  const baseMessage = buildBaseClientMessage(txId);
-  const messageType: AppLogsMessageType =
-    keyMirrors.toClientMessageType.app_logs;
-  const message = {
-    ...baseMessage,
-    messageType,
-    payload
-  };
-  return message;
-}
-
-// app state
-export async function buildAppStateMessage(
-  payload: AppStatePayload,
-  txId?: string
-): Promise<AppStateMessage> {
-  const baseMessage = buildBaseClientMessage(txId);
-  const messageType: AppStateMessageType =
-    keyMirrors.toClientMessageType.app_state;
-  const message = {
-    ...baseMessage,
-    messageType,
-    payload
-  };
-  return message;
-}
-
-// signed URLs
-export async function buildSignedUrlsRequestMessage(
-  payload: SignedUrlsRequestPayload,
-  txId?: string
-): Promise<SignedUrlsRequestMessage> {
-  const baseMessage = buildBaseClientMessage(txId);
-  const messageType: SignedUrlsRequestMessageType =
-    keyMirrors.toCloudMessageType.signed_urls_request;
-  const message = {
-    ...baseMessage,
-    messageType,
-    payload
-  };
-  return message;
-}
-
-// status response
-export async function buildStatusResponseMessage(
-  payload: StatusResponsePayload,
-  txId?: string
-): Promise<StatusResponseMessage> {
-  const baseMessage = buildBaseClientMessage(txId);
-  const messageType: StatusResponseMessageType =
-    keyMirrors.toClientMessageType.status_response;
-  const message = {
-    ...baseMessage,
-    messageType,
-    payload
-  };
-  const valid = validateToDeviceAgentMessage(message);
-  return message;
-}
-
-// device stats
-export async function buildDeviceStatsMessage(
-  payload: DeviceStatsPayload,
-  txId?: string
-): Promise<DeviceStatsMessage> {
-  const messageType: DeviceStatsMessageType =
-    keyMirrors.toClientMessageType.device_stats;
-  const baseMessage = buildBaseClientMessage(txId);
-  const message = {
-    ...baseMessage,
-    messageType,
-    payload
-  };
-  return message;
-}

package/src/secure-tunneling/index.ts

@@ -1,74 +0,0 @@
-import { downloadFile } from '../util/download-file';
-import {
-  SECURE_TUNNEL_BIN_DIR,
-  SECURE_TUNNEL_BIN_PATH,
-  SECURE_TUNNEL_BIN_NAME,
-  AWS_ROOT_CERTIFICATE_FILE_PATH
-} from '../util/directories';
-import { AAI_DIR } from 'alwaysai/lib/paths';
-import { JsSpawner } from 'alwaysai/lib/util';
-import { join } from 'path';
-import { logger } from '../util/logger';
-import { aaiArtifactsBucketUrl } from '../urls';
-import { getDistribution, getOsVersion, getArch } from '../util/system-info';
-
-export async function secureTunnelNotifyHandler(message: {
-  clientAccessToken: string;
-  region: string;
-}) {
-  const { clientAccessToken, region } = message;
-  await terminateSecureTunnelProcess();
-  await downloadSecureTunnel();
-  logger.info('Starting Secure Tunnel');
-
-  const args = [
-    '--destination-app',
-    '22',
-    '--region',
-    region,
-    '--capath',
-    AWS_ROOT_CERTIFICATE_FILE_PATH,
-    '--local-bind-address',
-    '0.0.0.0',
-    '-t',
-    clientAccessToken
-  ];
-
-  await JsSpawner().run({
-    exe: `${SECURE_TUNNEL_BIN_PATH}`,
-    args
-  });
-}
-
-async function downloadSecureTunnel() {
-  const arch = await getArch();
-  const linuxDistro = await getDistribution();
-  const osVersion = await getOsVersion();
-
-  const url = `${aaiArtifactsBucketUrl}/securetunnel/${linuxDistro}/${osVersion}/${arch}/${SECURE_TUNNEL_BIN_NAME}`;
-
-  await JsSpawner().mkdirp(join(AAI_DIR, SECURE_TUNNEL_BIN_DIR));
-  await downloadFile({
-    url,
-    path: SECURE_TUNNEL_BIN_PATH,
-    errorMessage: `Secure Tunnel bin for ${linuxDistro} ${osVersion} ${arch} not found}`
-  });
-
-  const spawner = JsSpawner();
-  await spawner.run({
-    exe: 'chmod',
-    args: ['+x', SECURE_TUNNEL_BIN_PATH]
-  });
-}
-
-async function terminateSecureTunnelProcess() {
-  const spawner = JsSpawner();
-  try {
-    await spawner.run({
-      exe: 'pkill',
-      args: [SECURE_TUNNEL_BIN_NAME]
-    });
-  } catch (e) {
-    logger.debug('Proccess terminated if status code is 1');
-  }
-}