rl-rock 1.2.2 → 1.2.4

package/dist/index.mjs

@@ -110,7 +110,11 @@ var SandboxStatusResponseSchema = z.object({
   namespace: z.string().optional(),
   cpus: z.number().optional(),
   memory: z.string().optional(),
-  state: z.unknown().optional()
+  state: z.unknown().optional(),
+  // Response headers info
+  cluster: z.string().optional(),
+  requestId: z.string().optional(),
+  eagleeyeTraceid: z.string().optional()
 });
 var CommandResponseSchema = z.object({
   stdout: z.string().default(""),
@@ -282,7 +286,18 @@ var HttpUtils = class {
     const snakeData = objectToSnake(data);
     try {
       const response = await client.post(url, snakeData);
-      return objectToCamel(response.data);
+      const camelData = objectToCamel(response.data);
+      const httpResponse = {
+        status: camelData.status ?? "Success",
+        headers: this.extractHeaders(response)
+      };
+      if (camelData.result !== void 0) {
+        httpResponse.result = camelData.result;
+      }
+      if (camelData.error !== void 0) {
+        httpResponse.error = camelData.error;
+      }
+      return httpResponse;
     } catch (error) {
       if (error instanceof AxiosError) {
         throw new Error(`Failed to POST ${url}: ${error.message}`);
@@ -298,7 +313,18 @@ var HttpUtils = class {
     const client = this.createClient({ headers });
     try {
       const response = await client.get(url);
-      return objectToCamel(response.data);
+      const camelData = objectToCamel(response.data);
+      const httpResponse = {
+        status: camelData.status ?? "Success",
+        headers: this.extractHeaders(response)
+      };
+      if (camelData.result !== void 0) {
+        httpResponse.result = camelData.result;
+      }
+      if (camelData.error !== void 0) {
+        httpResponse.error = camelData.error;
+      }
+      return httpResponse;
     } catch (error) {
       if (error instanceof AxiosError) {
         throw new Error(`Failed to GET ${url}: ${error.message}`);
@@ -306,6 +332,20 @@ var HttpUtils = class {
       throw error;
     }
   }
+  /**
+   * Extract headers from axios response
+   */
+  static extractHeaders(response) {
+    const headers = {};
+    for (const [key, value] of Object.entries(response.headers)) {
+      if (typeof value === "string") {
+        headers[key.toLowerCase()] = value;
+      } else if (Array.isArray(value)) {
+        headers[key.toLowerCase()] = value.join(", ");
+      }
+    }
+    return headers;
+  }
   /**
    * Convert camelCase key to snake_case
    */
@@ -352,7 +392,18 @@ var HttpUtils = class {
     });
     try {
       const response = await client.post(url, formData);
-      return objectToCamel(response.data);
+      const camelData = objectToCamel(response.data);
+      const httpResponse = {
+        status: camelData.status ?? "Success",
+        headers: this.extractHeaders(response)
+      };
+      if (camelData.result !== void 0) {
+        httpResponse.result = camelData.result;
+      }
+      if (camelData.error !== void 0) {
+        httpResponse.error = camelData.error;
+      }
+      return httpResponse;
     } catch (error) {
       if (error instanceof AxiosError) {
         throw new Error(`Failed to POST multipart ${url}: ${error.message}`);
@@ -563,7 +614,7 @@ var EnvHubClient = class {
         this.headers,
         payload
       );
-      return createRockEnvInfo(response);
+      return createRockEnvInfo(response.result);
     } catch (e) {
       throw new EnvHubError(`Failed to register environment: ${e}`);
     }
@@ -580,7 +631,7 @@ var EnvHubClient = class {
         this.headers,
         payload
       );
-      return createRockEnvInfo(response);
+      return createRockEnvInfo(response.result);
     } catch (e) {
       throw new EnvHubError(`Failed to get environment ${envName}: ${e}`);
     }
@@ -600,7 +651,7 @@ var EnvHubClient = class {
         this.headers,
         payload
       );
-      const envsData = response.envs ?? [];
+      const envsData = response.result?.envs ?? [];
       return envsData.map((envData) => createRockEnvInfo(envData));
     } catch (e) {
       throw new EnvHubError(`Failed to list environments: ${e}`);
@@ -633,7 +684,7 @@ var EnvHubClient = class {
         url,
         this.headers
       );
-      return response;
+      return response.result ?? {};
     } catch (e) {
       throw new EnvHubError(`Failed to health check: ${e}`);
     }
@@ -1082,7 +1133,7 @@ var Process = class {
     const scriptPath = `/tmp/${name}`;
     try {
       logger5.info(`[${sandboxId}] Uploading script to ${scriptPath}`);
-      const writeResult = await this.sandbox.write_file({
+      const writeResult = await this.sandbox.writeFile({
         content: scriptContent,
         path: scriptPath
       });
@@ -1422,9 +1473,14 @@ var Sandbox = class extends AbstractSandbox {
     const headers = this.buildHeaders();
     const response = await HttpUtils.get(url, headers);
     if (response.status !== "Success") {
-      throw new Error(`Failed to get status: ${JSON.stringify(response)}`);
+      const errorDetail = response.error ? `, error=${response.error}` : "";
+      throw new Error(`Failed to get status: status=${response.status}${errorDetail}, result=${JSON.stringify(response.result)}`);
     }
-    return response.result;
+    const result = response.result;
+    result.cluster = response.headers["x-rock-gateway-target-cluster"] || this.config.cluster || "N/A";
+    result.requestId = response.headers["x-request-id"] || response.headers["request-id"] || "N/A";
+    result.eagleeyeTraceid = response.headers["eagleeye-traceid"] || "N/A";
+    return result;
   }
   // Command execution
   async execute(command) {
@@ -1444,7 +1500,8 @@ var Sandbox = class extends AbstractSandbox {
         data
       );
       if (response.status !== "Success") {
-        throw new Error(`Failed to execute command: ${JSON.stringify(response)}`);
+        const errorDetail = response.error ? `, error=${response.error}` : "";
+        throw new Error(`Failed to execute command: status=${response.status}${errorDetail}, result=${JSON.stringify(response.result)}`);
       }
       return response.result;
     } catch (e) {
@@ -1466,7 +1523,8 @@ var Sandbox = class extends AbstractSandbox {
         data
       );
       if (response.status !== "Success") {
-        throw new Error(`Failed to create session: ${JSON.stringify(response)}`);
+        const errorDetail = response.error ? `, error=${response.error}` : "";
+        throw new Error(`Failed to create session: status=${response.status}${errorDetail}, result=${JSON.stringify(response.result)}`);
       }
       return response.result;
     } catch (e) {
@@ -1487,7 +1545,8 @@ var Sandbox = class extends AbstractSandbox {
         data
       );
       if (response.status !== "Success") {
-        throw new Error(`Failed to close session: ${JSON.stringify(response)}`);
+        const errorDetail = response.error ? `, error=${response.error}` : "";
+        throw new Error(`Failed to close session: status=${response.status}${errorDetail}, result=${JSON.stringify(response.result)}`);
       }
       return response.result ?? { sessionType: "bash" };
     } catch (e) {
@@ -1533,7 +1592,8 @@ var Sandbox = class extends AbstractSandbox {
         timeoutMs
       );
       if (response.status !== "Success") {
-        throw new Error(`Failed to execute command: ${JSON.stringify(response)}`);
+        const errorDetail = response.error ? `, error=${response.error}` : "";
+        throw new Error(`Failed to run in session: status=${response.status}${errorDetail}, result=${JSON.stringify(response.result)}`);
       }
       return response.result;
     } catch (e) {
@@ -1621,7 +1681,7 @@ var Sandbox = class extends AbstractSandbox {
     return false;
   }
   // File operations
-  async write_file(request) {
+  async writeFile(request) {
     const url = `${this.url}/write_file`;
     const headers = this.buildHeaders();
     const data = {
@@ -1635,7 +1695,7 @@ var Sandbox = class extends AbstractSandbox {
     }
     return { success: true, message: `Successfully write content to file ${request.path}` };
   }
-  async read_file(request) {
+  async readFile(request) {
     const url = `${this.url}/read_file`;
     const headers = this.buildHeaders();
     const data = {