@layr-labs/ecloud-cli 0.1.0-rc.1 → 0.1.0

package/dist/commands/compute/app/upgrade.js

@@ -11,38 +11,15 @@ import {
   watchUpgrade
 } from "@layr-labs/ecloud-sdk";
 
-// src/flags.ts
-import { Flags } from "@oclif/core";
-
-// src/utils/prompts.ts
-import { input, select, password, confirm as inquirerConfirm } from "@inquirer/prompts";
-import fs3 from "fs";
-import path3 from "path";
-import os3 from "os";
-import { isAddress as isAddress2 } from "viem";
-import { privateKeyToAccount as privateKeyToAccount2 } from "viem/accounts";
+// src/telemetry.ts
 import {
-  getEnvironmentConfig,
-  getAvailableEnvironments,
-  isEnvironmentAvailable,
-  getAllAppsByDeveloper as getAllAppsByDeveloper2,
-  getCategoryDescriptions,
-  fetchTemplateCatalog,
-  PRIMARY_LANGUAGES,
-  validateAppName,
-  validateImageReference,
-  validateFilePath,
-  validatePrivateKeyFormat,
-  extractAppNameFromImage,
-  UserApiClient as UserApiClient2
-} from "@layr-labs/ecloud-sdk";
-
-// src/utils/appResolver.ts
-import { isAddress } from "viem";
-import { privateKeyToAccount } from "viem/accounts";
-import {
-  UserApiClient,
-  getAllAppsByDeveloper
+  createTelemetryClient,
+  createAppEnvironment,
+  createMetricsContext,
+  addMetric,
+  addMetricWithDimensions,
+  emitMetrics,
+  getBuildType as getBuildType2
 } from "@layr-labs/ecloud-sdk";
 
 // src/utils/globalConfig.ts
@@ -51,6 +28,7 @@ import * as path from "path";
 import * as os from "os";
 import { load as loadYaml, dump as dumpYaml } from "js-yaml";
 import { getBuildType } from "@layr-labs/ecloud-sdk";
+import * as crypto from "crypto";
 var GLOBAL_CONFIG_FILE = "config.yaml";
 var PROFILE_CACHE_TTL_MS = 24 * 60 * 60 * 1e3;
 function getGlobalConfigDir() {
@@ -93,6 +71,14 @@ function saveGlobalConfig(config) {
   const content = dumpYaml(config, { lineWidth: -1 });
   fs.writeFileSync(configPath, content, { mode: 420 });
 }
+function getDefaultEnvironment() {
+  const config = loadGlobalConfig();
+  return config.default_environment;
+}
+function getGlobalTelemetryPreference() {
+  const config = loadGlobalConfig();
+  return config.telemetry_enabled;
+}
 function getProfileCache(environment) {
   const config = loadGlobalConfig();
   const cacheEntry = config.profile_cache?.[environment];
@@ -116,6 +102,107 @@ function setProfileCache(environment, profiles) {
   };
   saveGlobalConfig(config);
 }
+function getOrCreateUserUUID() {
+  const config = loadGlobalConfig();
+  if (config.user_uuid) {
+    return config.user_uuid;
+  }
+  const uuid = generateUUID();
+  config.user_uuid = uuid;
+  config.first_run = false;
+  saveGlobalConfig(config);
+  return uuid;
+}
+function generateUUID() {
+  const bytes = crypto.randomBytes(16);
+  bytes[6] = bytes[6] & 15 | 64;
+  bytes[8] = bytes[8] & 63 | 128;
+  const hex = Array.from(bytes, (b) => b.toString(16).padStart(2, "0"));
+  return hex.slice(0, 4).join("") + hex.slice(4, 6).join("") + "-" + hex.slice(6, 8).join("") + "-" + hex.slice(8, 10).join("") + "-" + hex.slice(10, 12).join("") + "-" + hex.slice(12, 16).join("");
+}
+
+// src/telemetry.ts
+function createCLITelemetryClient() {
+  const userUUID = getOrCreateUserUUID();
+  const environment = createAppEnvironment(userUUID);
+  const telemetryEnabled = getGlobalTelemetryPreference();
+  return createTelemetryClient(environment, "ecloud-cli", {
+    telemetryEnabled: telemetryEnabled === true
+    // Only enabled if explicitly set to true
+  });
+}
+async function withTelemetry(command, action) {
+  const client = createCLITelemetryClient();
+  const metrics = createMetricsContext();
+  metrics.properties["source"] = "ecloud-cli";
+  metrics.properties["command"] = command.id || command.constructor.name;
+  const environment = getDefaultEnvironment() || "sepolia";
+  metrics.properties["environment"] = environment;
+  const buildType = getBuildType2() || "prod";
+  metrics.properties["build_type"] = buildType;
+  const cliVersion = command.config.version;
+  if (cliVersion) {
+    metrics.properties["cli_version"] = cliVersion;
+  }
+  addMetric(metrics, "Count", 1);
+  let actionError;
+  let result;
+  try {
+    result = await action();
+    return result;
+  } catch (err) {
+    actionError = err instanceof Error ? err : new Error(String(err));
+    throw err;
+  } finally {
+    const resultValue = actionError ? "Failure" : "Success";
+    const dimensions = {};
+    if (actionError) {
+      dimensions["error"] = actionError.message;
+    }
+    addMetricWithDimensions(metrics, resultValue, 1, dimensions);
+    const duration = Date.now() - metrics.startTime.getTime();
+    addMetric(metrics, "DurationMilliseconds", duration);
+    try {
+      await emitMetrics(client, metrics);
+      await client.close();
+    } catch {
+    }
+  }
+}
+
+// src/flags.ts
+import { Flags } from "@oclif/core";
+
+// src/utils/prompts.ts
+import { input, select, password, confirm as inquirerConfirm } from "@inquirer/prompts";
+import fs3 from "fs";
+import path3 from "path";
+import os3 from "os";
+import { isAddress as isAddress2 } from "viem";
+import { privateKeyToAccount as privateKeyToAccount2 } from "viem/accounts";
+import {
+  getEnvironmentConfig,
+  getAvailableEnvironments,
+  isEnvironmentAvailable,
+  getAllAppsByDeveloper as getAllAppsByDeveloper2,
+  getCategoryDescriptions,
+  fetchTemplateCatalog,
+  PRIMARY_LANGUAGES,
+  validateAppName,
+  validateImageReference,
+  validateFilePath,
+  validatePrivateKeyFormat,
+  extractAppNameFromImage,
+  UserApiClient as UserApiClient2
+} from "@layr-labs/ecloud-sdk";
+
+// src/utils/appResolver.ts
+import { isAddress } from "viem";
+import { privateKeyToAccount } from "viem/accounts";
+import {
+  UserApiClient,
+  getAllAppsByDeveloper
+} from "@layr-labs/ecloud-sdk";
 
 // src/utils/appNames.ts
 import * as fs2 from "fs";
@@ -163,7 +250,7 @@ function listApps(environment) {
 
 // src/utils/version.ts
 function getCliVersion() {
-  return true ? "0.1.0-rc.1" : "0.0.0";
+  return true ? "0.1.0" : "0.0.0";
 }
 function getClientId() {
   return `ecloud-cli/v${getCliVersion()}`;
@@ -444,7 +531,7 @@ async function getImageReferenceInteractive(imageRef, buildFromDockerfile = fals
   if (buildFromDockerfile) {
     console.log("\n\u{1F4E6} Build & Push Configuration");
     console.log("Your Docker image will be built and pushed to a registry");
-    console.log("so that EigenX can pull and run it in the TEE.");
+    console.log("so that Ecloud CLI can pull and run it in the TEE.");
     console.log();
     if (registries.length > 0) {
       displayDetectedRegistries(registries, appName);
@@ -969,88 +1056,102 @@ var AppUpgrade = class _AppUpgrade extends Command {
     })
   };
   async run() {
-    const { args, flags } = await this.parse(_AppUpgrade);
-    const logger = {
-      info: (msg) => this.log(msg),
-      warn: (msg) => this.warn(msg),
-      error: (msg) => this.error(msg),
-      debug: (msg) => flags.verbose && this.log(msg)
-    };
-    const environment = flags.environment || "sepolia";
-    const environmentConfig = getEnvironmentConfig2(environment);
-    const rpcUrl = flags["rpc-url"] || environmentConfig.defaultRPCURL;
-    const privateKey = await getPrivateKeyInteractive(flags["private-key"]);
-    const appID = await getOrPromptAppID({
-      appID: args["app-id"],
-      environment,
-      privateKey,
-      rpcUrl,
-      action: "upgrade"
-    });
-    const dockerfilePath = await getDockerfileInteractive(flags.dockerfile);
-    const buildFromDockerfile = dockerfilePath !== "";
-    const imageRef = await getImageReferenceInteractive(flags["image-ref"], buildFromDockerfile);
-    const envFilePath = await getEnvFileInteractive(flags["env-file"]);
-    let currentInstanceType = "";
-    try {
-      const userApiClient = new UserApiClient3(environmentConfig, privateKey, rpcUrl, getClientId());
-      const infos = await userApiClient.getInfos([appID], 1);
-      if (infos.length > 0) {
-        currentInstanceType = infos[0].machineType || "";
-      }
-    } catch {
-    }
-    const availableTypes = await fetchAvailableInstanceTypes(environmentConfig, privateKey, rpcUrl);
-    const instanceType = await getInstanceTypeInteractive(
-      flags["instance-type"],
-      currentInstanceType,
-      availableTypes
-    );
-    const logSettings = await getLogSettingsInteractive(
-      flags["log-visibility"]
-    );
-    const resourceUsageMonitoring = await getResourceUsageMonitoringInteractive(
-      flags["resource-usage-monitoring"]
-    );
-    const logVisibility = logSettings.publicLogs ? "public" : logSettings.logRedirect ? "private" : "off";
-    const { prepared, gasEstimate } = await prepareUpgrade(
-      {
-        appId: appID,
+    return withTelemetry(this, async () => {
+      const { args, flags } = await this.parse(_AppUpgrade);
+      const logger = {
+        info: (msg) => this.log(msg),
+        warn: (msg) => this.warn(msg),
+        error: (msg) => this.error(msg),
+        debug: (msg) => flags.verbose && this.log(msg)
+      };
+      const environment = flags.environment || "sepolia";
+      const environmentConfig = getEnvironmentConfig2(environment);
+      const rpcUrl = flags["rpc-url"] || environmentConfig.defaultRPCURL;
+      const privateKey = await getPrivateKeyInteractive(flags["private-key"]);
+      const appID = await getOrPromptAppID({
+        appID: args["app-id"],
+        environment,
         privateKey,
         rpcUrl,
-        environment,
-        dockerfilePath,
-        imageRef,
-        envFilePath,
-        instanceType,
-        logVisibility,
-        resourceUsageMonitoring
-      },
-      logger
-    );
-    this.log(`
+        action: "upgrade"
+      });
+      const dockerfilePath = await getDockerfileInteractive(flags.dockerfile);
+      const buildFromDockerfile = dockerfilePath !== "";
+      const imageRef = await getImageReferenceInteractive(flags["image-ref"], buildFromDockerfile);
+      const envFilePath = await getEnvFileInteractive(flags["env-file"]);
+      let currentInstanceType = "";
+      try {
+        const userApiClient = new UserApiClient3(
+          environmentConfig,
+          privateKey,
+          rpcUrl,
+          getClientId()
+        );
+        const infos = await userApiClient.getInfos([appID], 1);
+        if (infos.length > 0) {
+          currentInstanceType = infos[0].machineType || "";
+        }
+      } catch {
+      }
+      const availableTypes = await fetchAvailableInstanceTypes(
+        environmentConfig,
+        privateKey,
+        rpcUrl
+      );
+      const instanceType = await getInstanceTypeInteractive(
+        flags["instance-type"],
+        currentInstanceType,
+        availableTypes
+      );
+      const logSettings = await getLogSettingsInteractive(
+        flags["log-visibility"]
+      );
+      const resourceUsageMonitoring = await getResourceUsageMonitoringInteractive(
+        flags["resource-usage-monitoring"]
+      );
+      const logVisibility = logSettings.publicLogs ? "public" : logSettings.logRedirect ? "private" : "off";
+      const { prepared, gasEstimate } = await prepareUpgrade(
+        {
+          appId: appID,
+          privateKey,
+          rpcUrl,
+          environment,
+          dockerfilePath,
+          imageRef,
+          envFilePath,
+          instanceType,
+          logVisibility,
+          resourceUsageMonitoring,
+          skipTelemetry: true
+        },
+        logger
+      );
+      this.log(`
 Estimated transaction cost: ${chalk.cyan(gasEstimate.maxCostEth)} ETH`);
-    if (isMainnet(environmentConfig)) {
-      const confirmed = await confirm(`Continue with upgrade?`);
-      if (!confirmed) {
-        this.log(`
+      if (isMainnet(environmentConfig)) {
+        const confirmed = await confirm(`Continue with upgrade?`);
+        if (!confirmed) {
+          this.log(`
 ${chalk.gray(`Upgrade cancelled`)}`);
-        return;
+          return;
+        }
       }
-    }
-    const res = await executeUpgrade(
-      prepared,
-      {
-        maxFeePerGas: gasEstimate.maxFeePerGas,
-        maxPriorityFeePerGas: gasEstimate.maxPriorityFeePerGas
-      },
-      logger
-    );
-    await watchUpgrade(res.appId, privateKey, rpcUrl, environment, logger, getClientId());
-    this.log(
-      `
+      const res = await executeUpgrade(
+        prepared,
+        {
+          maxFeePerGas: gasEstimate.maxFeePerGas,
+          maxPriorityFeePerGas: gasEstimate.maxPriorityFeePerGas
+        },
+        logger,
+        true
+        // skipTelemetry
+      );
+      await watchUpgrade(res.appId, privateKey, rpcUrl, environment, logger, getClientId(), true);
+      this.log(
+        `
 \u2705 ${chalk.green(`App upgraded successfully ${chalk.bold(`(id: ${res.appId}, image: ${res.imageRef})`)}`)}`
-    );
+      );
+    });
   }
 };
 async function fetchAvailableInstanceTypes(environmentConfig, privateKey, rpcUrl) {