@alwaysai/device-agent 0.0.1-2.1-beta-provision

package/src/util/directories.ts

@@ -0,0 +1,82 @@
+import {
+  DEVICE_TOKEN_FILE_NAME,
+  LOCAL_AAI_CFG_DIR,
+  LOCAL_CERT_AND_KEY_DIR,
+  ALWAYSAI_CONFIG_DIR,
+} from "alwaysai/lib/constants";
+import { join } from "path";
+import { getSystemId } from "alwaysai/lib/infrastructure";
+
+export const shortenSystemId = () => {
+  const id = getSystemId();
+  if (id === "production") {
+    return "prod";
+  } else if (id === "development") {
+    return "dev";
+  }
+};
+
+export const BOOTSTRAP_DIR_NAME = "bootstrap-certificates";
+export const BOOTSTRAP_DEVICE_PRIVATE_KEY_FILE_NAME = `aai-claim-private-key_${shortenSystemId()}.pem.key`;
+export const BOOTSTRAP_DEVICE_CERTIFICATE_FILE_NAME = `aai-claim-cert_${shortenSystemId()}.pem.crt`;
+export const BOOTSTRAP_CLAIM_ID_FILE_NAME = `aai-claim-cert-id_${shortenSystemId()}.txt`;
+export const AWS_ROOT_CERTIFICATE_FILE_NAME = "AmazonRootCA1.pem";
+export const CERTIFICATE_OWNERSHIP_TOKEN_FILE_NAME =
+  "certificate-ownership-token.txt";
+export const AGENT_CONFIG_FILE_NAME = "alwaysai.agent.json";
+export const DEVICE_PRIVATE_KEY_FILE_NAME = `aai-claim-private-key_${shortenSystemId()}.pem.key`;
+export const DEVICE_CERTIFICATE_FILE_NAME = `aai-claim-cert_${shortenSystemId()}.pem.crt`;
+export const DEVICE_CLAIM_ID_FILE_NAME = `aai-claim-cert-id_${shortenSystemId()}.txt`;
+
+export function getPrivateKeyFilePath() {
+  return join(LOCAL_CERT_AND_KEY_DIR, DEVICE_PRIVATE_KEY_FILE_NAME);
+}
+
+export function getCertificateFilePath() {
+  return join(LOCAL_CERT_AND_KEY_DIR, DEVICE_CERTIFICATE_FILE_NAME);
+}
+
+export function getRootCertificateFilePath() {
+  return join(LOCAL_CERT_AND_KEY_DIR, AWS_ROOT_CERTIFICATE_FILE_NAME);
+}
+
+export function getCredentialsFilePath() {
+  return join(LOCAL_AAI_CFG_DIR, DEVICE_TOKEN_FILE_NAME);
+}
+
+export const BOOTSTRAP_CERTIFICATES_DIR_PATH = join(
+  LOCAL_CERT_AND_KEY_DIR,
+  BOOTSTRAP_DIR_NAME
+);
+export const DEVICE_CLAIM_ID_FILE_PATH = join(
+  LOCAL_CERT_AND_KEY_DIR,
+  DEVICE_TOKEN_FILE_NAME
+);
+
+export const BOOTSTRAP_DEVICE_PRIVATE_KEY_FILE_PATH = join(
+  LOCAL_CERT_AND_KEY_DIR,
+  BOOTSTRAP_DIR_NAME,
+  BOOTSTRAP_DEVICE_PRIVATE_KEY_FILE_NAME
+);
+
+export const BOOTSTRAP_DEVICE_CERTIFICATE_FILE_PATH = join(
+  LOCAL_CERT_AND_KEY_DIR,
+  BOOTSTRAP_DIR_NAME,
+  BOOTSTRAP_DEVICE_CERTIFICATE_FILE_NAME
+);
+
+export const BOOTSTRAP_CLAIM_ID_FILE_PATH = join(
+  LOCAL_CERT_AND_KEY_DIR,
+  BOOTSTRAP_DIR_NAME,
+  BOOTSTRAP_CLAIM_ID_FILE_NAME
+);
+
+export const AWS_ROOT_CERTIFICATE_FILE_PATH = join(
+  LOCAL_CERT_AND_KEY_DIR,
+  AWS_ROOT_CERTIFICATE_FILE_NAME
+);
+
+export const AGENT_CONFIG_FILE_PATH = join(
+  ALWAYSAI_CONFIG_DIR,
+  AGENT_CONFIG_FILE_NAME
+);

package/src/util/get-device-id.ts

@@ -0,0 +1,22 @@
+import { getCredentialsFilePath } from '../util/directories';
+import { CliTerseError } from '@alwaysai/alwayscli';
+import { ConfigFile } from '@alwaysai/config-nodejs';
+import * as t from 'io-ts';
+
+export function getDeviceId() {
+  const path = getCredentialsFilePath();
+  const codec = t.type({
+    deviceId: t.union([t.string, t.undefined]),
+  });
+  const credentialsJsonFile = ConfigFile({
+    path,
+    codec,
+  });
+  const config = credentialsJsonFile.readIfExists();
+  if (!config || !config.deviceId) {
+    throw new CliTerseError(
+      'Invalid device credentials! Please ensure you are using the device agent on a valid device.',
+    );
+  }
+  return config.deviceId;
+}

package/src/util/http-client.ts

@@ -0,0 +1,35 @@
+import fetch from "node-fetch";
+import { serviceEndpointBuilder } from "../endpoints";
+import { CliAuthenticationClient } from "alwaysai/lib/infrastructure";
+
+export async function httpClient(
+  url: string,
+  method: string,
+  data?: string,
+  headers?: object
+) {
+  const options = { method, ...(data ? { body: data } : {}) };
+  try {
+    const response = await fetch(url, options);
+    const code = response.status;
+
+    const contentType = response.headers.get("content-type");
+    if (contentType === "application/json") {
+      return await response.json();
+    } else if (contentType!.includes("text/plain")) {
+      return await response.text();
+    }
+  } catch (e) {}
+}
+
+export async function microServiceHttpClient(
+  service: string,
+  path: string,
+  method: string,
+  data?: string
+) {
+  const endpoint = serviceEndpointBuilder(service, path);
+  const { getIdAuthorizationHeader } = CliAuthenticationClient();
+  const authorizationHeader = await getIdAuthorizationHeader();
+  return await httpClient(endpoint, method, data);
+}

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.colorize(),
+    winston.format.errors({ stack: true }),
+    winston.format.splat(),
+    winston.format.simple()
+  ),
+  transports,
+});

package/src/util/run-in-dir.ts

@@ -0,0 +1,15 @@
+export async function runInDir<T extends any[], R extends any>(
+  func: (...args: T) => R,
+  args: T,
+  dir: string,
+) {
+  const origCwd = process.cwd();
+  let out: R;
+  try {
+    process.chdir(dir);
+    out = await func(...args);
+  } finally {
+    process.chdir(origCwd);
+  }
+  return out;
+}

package/src/util/sleep.ts

@@ -0,0 +1,5 @@
+export default function sleep(ms: number) {
+  return new Promise((resolve) => {
+    setTimeout(resolve, ms);
+  });
+}