@alwaysai/device-agent 0.1.4 → 0.1.5

package/src/util/directories.ts

@@ -4,7 +4,7 @@ import {
   DOCKER_COMPOSE_FILE,
   LOCAL_AAI_CFG_DIR,
   LOCAL_CERT_AND_KEY_DIR
-} from 'alwaysai/lib/constants';
+} from 'alwaysai/lib/paths';
 import { join } from 'path';
 import { getSystemId } from '../infrastructure/system-id';
 
@@ -94,3 +94,14 @@ export const DEVICE_CERTIFICATE_FILE_PATH = join(
   LOCAL_CERT_AND_KEY_DIR,
   DEVICE_CERTIFICATE_FILE_NAME
 );
+
+/*===================================================================
+                      Secure Tunnel bin directory
+===================================================================*/
+export const SECURE_TUNNEL_BIN_NAME = 'localproxy';
+export const SECURE_TUNNEL_BIN_DIR = 'secure-tunnel';
+export const SECURE_TUNNEL_BIN_PATH = join(
+  AAI_DIR,
+  SECURE_TUNNEL_BIN_DIR,
+  SECURE_TUNNEL_BIN_NAME
+);

package/src/util/download-file.ts

@@ -0,0 +1,25 @@
+import * as fs from 'fs';
+import { fetchWithTimeout } from './fetch-with-timeout';
+import { logger } from './logger';
+
+export async function downloadFile(props: {
+  url: string;
+  path: string;
+  errorMessage: string;
+}): Promise<void> {
+  const { url, path, errorMessage } = props;
+  logger.debug(`Downloading package from ${url}`);
+  let response: any;
+  try {
+    response = await fetchWithTimeout(url);
+  } catch (e) {
+    const errorBody = e.type === 'aborted' ? e : await e.response.text();
+    throw new Error(`${errorMessage}: Error=${e}\n${errorBody}`);
+  }
+
+  const stream = response.body.pipe(fs.createWriteStream(path));
+  await new Promise((resolve, reject) => {
+    stream.on('finish', resolve);
+    stream.on('error', reject);
+  });
+}

package/src/util/fetch-with-timeout.ts

@@ -1,5 +1,22 @@
 import nodeFetch from 'node-fetch';
 
+class HTTPResponseError extends Error {
+  public response;
+  constructor(response) {
+    super(`HTTP Error Response: ${response.status} ${response.statusText}`);
+    this.response = response;
+  }
+}
+
+const checkStatus = (response) => {
+  if (response.ok) {
+    // response.status >= 200 && response.status < 300
+    return response;
+  } else {
+    throw new HTTPResponseError(response);
+  }
+};
+
 export async function fetchWithTimeout(
   url: string,
   options: { timeout?: number } = {}
@@ -14,5 +31,5 @@ export async function fetchWithTimeout(
     signal: controller.signal
   });
   clearTimeout(id);
-  return response;
+  return checkStatus(response);
 }

package/src/util/logger.ts

@@ -1,7 +1,7 @@
 import * as winston from 'winston';
 import 'winston-daily-rotate-file';
 import * as path from 'path';
-import { AAI_DIR } from 'alwaysai/lib/constants';
+import { AAI_DIR } from 'alwaysai/lib/paths';
 import { ALWAYSAI_LOG_LEVEL, ALWAYSAI_LOG_TO_CONSOLE } from '../environment';
 
 const LOG_LEVEL = ALWAYSAI_LOG_LEVEL || 'info';

package/src/util/system-info.ts

@@ -0,0 +1,25 @@
+import { JsSpawner } from 'alwaysai/lib/util';
+
+export async function getDistribution() {
+  const spawner = JsSpawner();
+  const os = await spawner.run({
+    exe: 'lsb_release',
+    args: ['-is']
+  });
+  return os.toLowerCase();
+}
+export async function getOsVersion() {
+  const spawner = JsSpawner();
+  return await spawner.run({
+    exe: 'lsb_release',
+    args: ['-sr']
+  });
+}
+
+export async function getArch() {
+  const spawner = JsSpawner();
+  return await spawner.run({
+    exe: 'dpkg',
+    args: ['--print-architecture']
+  });
+}