rl-rock 1.2.3 → 1.2.4

package/dist/index.js

@@ -94,21 +94,14 @@ var ChmodRequestSchema = zod.z.object({
   mode: zod.z.string().default("755"),
   recursive: zod.z.boolean().default(false)
 });
-var HeaderFieldsSchema = {
-  cluster: zod.z.string().optional(),
-  requestId: zod.z.string().optional(),
-  eagleeyeTraceid: zod.z.string().optional()
-};
 var SandboxResponseSchema = zod.z.object({
   code: zod.z.nativeEnum(Codes).optional(),
   exitCode: zod.z.number().optional(),
-  failureReason: zod.z.string().optional(),
-  ...HeaderFieldsSchema
+  failureReason: zod.z.string().optional()
 });
 var IsAliveResponseSchema = zod.z.object({
   isAlive: zod.z.boolean(),
-  message: zod.z.string().default(""),
-  ...HeaderFieldsSchema
+  message: zod.z.string().default("")
 });
 var SandboxStatusResponseSchema = zod.z.object({
   sandboxId: zod.z.string().optional(),
@@ -126,67 +119,57 @@ var SandboxStatusResponseSchema = zod.z.object({
   cpus: zod.z.number().optional(),
   memory: zod.z.string().optional(),
   state: zod.z.unknown().optional(),
-  ...HeaderFieldsSchema
+  // Response headers info
+  cluster: zod.z.string().optional(),
+  requestId: zod.z.string().optional(),
+  eagleeyeTraceid: zod.z.string().optional()
 });
 var CommandResponseSchema = zod.z.object({
   stdout: zod.z.string().default(""),
   stderr: zod.z.string().default(""),
-  exitCode: zod.z.number().optional(),
-  ...HeaderFieldsSchema
+  exitCode: zod.z.number().optional()
 });
 var WriteFileResponseSchema = zod.z.object({
   success: zod.z.boolean().default(false),
-  message: zod.z.string().default(""),
-  ...HeaderFieldsSchema
+  message: zod.z.string().default("")
 });
 var ReadFileResponseSchema = zod.z.object({
-  content: zod.z.string().default(""),
-  ...HeaderFieldsSchema
+  content: zod.z.string().default("")
 });
 var UploadResponseSchema = zod.z.object({
   success: zod.z.boolean().default(false),
   message: zod.z.string().default(""),
-  fileName: zod.z.string().optional(),
-  ...HeaderFieldsSchema
+  fileName: zod.z.string().optional()
 });
 var ObservationSchema = zod.z.object({
   output: zod.z.string().default(""),
   exitCode: zod.z.number().optional(),
   failureReason: zod.z.string().default(""),
-  expectString: zod.z.string().default(""),
-  ...HeaderFieldsSchema
+  expectString: zod.z.string().default("")
 });
 var CreateSessionResponseSchema = zod.z.object({
   output: zod.z.string().default(""),
-  sessionType: zod.z.literal("bash").default("bash"),
-  ...HeaderFieldsSchema
+  sessionType: zod.z.literal("bash").default("bash")
 });
 var CloseSessionResponseSchema = zod.z.object({
-  sessionType: zod.z.literal("bash").default("bash"),
-  ...HeaderFieldsSchema
-});
-var CloseResponseSchema = zod.z.object({
-  ...HeaderFieldsSchema
+  sessionType: zod.z.literal("bash").default("bash")
 });
+var CloseResponseSchema = zod.z.object({});
 var ChownResponseSchema = zod.z.object({
   success: zod.z.boolean().default(false),
-  message: zod.z.string().default(""),
-  ...HeaderFieldsSchema
+  message: zod.z.string().default("")
 });
 var ChmodResponseSchema = zod.z.object({
   success: zod.z.boolean().default(false),
-  message: zod.z.string().default(""),
-  ...HeaderFieldsSchema
+  message: zod.z.string().default("")
 });
 var ExecuteBashSessionResponseSchema = zod.z.object({
   success: zod.z.boolean().default(false),
-  message: zod.z.string().default(""),
-  ...HeaderFieldsSchema
+  message: zod.z.string().default("")
 });
 var OssSetupResponseSchema = zod.z.object({
   success: zod.z.boolean().default(false),
-  message: zod.z.string().default(""),
-  ...HeaderFieldsSchema
+  message: zod.z.string().default("")
 });
 
 // src/common/constants.ts
@@ -302,7 +285,6 @@ var HttpUtils = class {
    * Send POST request
    * Automatically converts request body from camelCase to snake_case
    * Automatically converts response from snake_case to camelCase
-   * Returns structured response with headers
    */
   static async post(url, headers, data, readTimeout) {
     const client = this.createClient({
@@ -312,18 +294,18 @@ var HttpUtils = class {
     const snakeData = objectToSnake(data);
     try {
       const response = await client.post(url, snakeData);
-      const result = objectToCamel(response.data);
-      const responseHeaders = {};
-      for (const [key, value] of Object.entries(response.headers)) {
-        if (typeof value === "string") {
-          responseHeaders[key.toLowerCase()] = value;
-        }
-      }
-      return {
-        status: result?.status ?? "Success",
-        result,
-        headers: responseHeaders
+      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 axios3.AxiosError) {
         throw new Error(`Failed to POST ${url}: ${error.message}`);
@@ -334,24 +316,23 @@ var HttpUtils = class {
   /**
    * Send GET request
    * Automatically converts response from snake_case to camelCase
-   * Returns structured response with headers
    */
   static async get(url, headers) {
     const client = this.createClient({ headers });
     try {
       const response = await client.get(url);
-      const result = objectToCamel(response.data);
-      const responseHeaders = {};
-      for (const [key, value] of Object.entries(response.headers)) {
-        if (typeof value === "string") {
-          responseHeaders[key.toLowerCase()] = value;
-        }
-      }
-      return {
-        status: result?.status ?? "Success",
-        result,
-        headers: responseHeaders
+      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 axios3.AxiosError) {
         throw new Error(`Failed to GET ${url}: ${error.message}`);
@@ -359,6 +340,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
    */
@@ -369,7 +364,6 @@ var HttpUtils = class {
    * Send multipart/form-data request
    * Automatically converts form data keys to snake_case
    * Automatically converts response from snake_case to camelCase
-   * Returns structured response with headers
    */
   static async postMultipart(url, headers, data, files) {
     const formData = new FormData();
@@ -406,18 +400,18 @@ var HttpUtils = class {
     });
     try {
       const response = await client.post(url, formData);
-      const result = objectToCamel(response.data);
-      const responseHeaders = {};
-      for (const [key, value] of Object.entries(response.headers)) {
-        if (typeof value === "string") {
-          responseHeaders[key.toLowerCase()] = value;
-        }
-      }
-      return {
-        status: result?.status ?? "Success",
-        result,
-        headers: responseHeaders
+      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 axios3.AxiosError) {
         throw new Error(`Failed to POST multipart ${url}: ${error.message}`);
@@ -698,7 +692,7 @@ var EnvHubClient = class {
         url,
         this.headers
       );
-      return response.result;
+      return response.result ?? {};
     } catch (e) {
       throw new EnvHubError(`Failed to health check: ${e}`);
     }
@@ -1406,16 +1400,6 @@ var Sandbox = class extends AbstractSandbox {
       headers["X-Namespace"] = this.config.namespace;
     }
   }
-  /**
-   * Extract header-derived fields from response headers
-   */
-  extractHeaderFields(headers) {
-    return {
-      cluster: headers["x-rock-gateway-target-cluster"] || this.config.cluster || "N/A",
-      requestId: headers["x-request-id"] || headers["request-id"] || "N/A",
-      eagleeyeTraceid: headers["eagleeye-traceid"] || "N/A"
-    };
-  }
   // Lifecycle methods
   async start() {
     logger8.info("Starting sandbox...");
@@ -1497,7 +1481,8 @@ 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)}`);
     }
     const result = response.result;
     result.cluster = response.headers["x-rock-gateway-target-cluster"] || this.config.cluster || "N/A";
@@ -1523,11 +1508,10 @@ 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)}`);
       }
-      const result = response.result;
-      Object.assign(result, this.extractHeaderFields(response.headers));
-      return result;
+      return response.result;
     } catch (e) {
       throw new Error(`Failed to execute command: ${e}`);
     }
@@ -1547,11 +1531,10 @@ 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)}`);
       }
-      const result = response.result;
-      Object.assign(result, this.extractHeaderFields(response.headers));
-      return result;
+      return response.result;
     } catch (e) {
       throw new Error(`Failed to create session: ${e}`);
     }
@@ -1570,11 +1553,10 @@ 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)}`);
       }
-      const result = response.result ?? { sessionType: "bash" };
-      Object.assign(result, this.extractHeaderFields(response.headers));
-      return result;
+      return response.result ?? { sessionType: "bash" };
     } catch (e) {
       throw new Error(`Failed to close session: ${e}`);
     }
@@ -1618,11 +1600,10 @@ 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)}`);
       }
-      const result = response.result;
-      Object.assign(result, this.extractHeaderFields(response.headers));
-      return result;
+      return response.result;
     } catch (e) {
       throw new Error(`Failed to run in session: ${e}`);
     }
@@ -1661,10 +1642,7 @@ var Sandbox = class extends AbstractSandbox {
         output: "Failed to submit command, nohup failed to extract PID",
         exitCode: 1,
         failureReason: "PID extraction failed",
-        expectString: "",
-        cluster: response.cluster,
-        requestId: response.requestId,
-        eagleeyeTraceid: response.eagleeyeTraceid
+        expectString: ""
       };
     }
     const success = await this.waitForProcessCompletion(pid, tmpSession, waitTimeout, waitInterval);
@@ -1673,10 +1651,7 @@ var Sandbox = class extends AbstractSandbox {
         output: `Command executed in nohup mode. Output file: ${tmpFile}`,
         exitCode: success ? 0 : 1,
         failureReason: success ? "" : "Process did not complete successfully",
-        expectString: "",
-        cluster: response.cluster,
-        requestId: response.requestId,
-        eagleeyeTraceid: response.eagleeyeTraceid
+        expectString: ""
       };
     }
     const readCmd = responseLimitedBytesInNohup ? `head -c ${responseLimitedBytesInNohup} ${tmpFile}` : `cat ${tmpFile}`;
@@ -1688,10 +1663,7 @@ var Sandbox = class extends AbstractSandbox {
       output: outputResult.output,
       exitCode: success ? 0 : 1,
       failureReason: success ? "" : "Process did not complete successfully",
-      expectString: "",
-      cluster: outputResult.cluster,
-      requestId: outputResult.requestId,
-      eagleeyeTraceid: outputResult.eagleeyeTraceid
+      expectString: ""
     };
   }
   async waitForProcessCompletion(pid, session, waitTimeout, waitInterval) {
@@ -1726,11 +1698,10 @@ var Sandbox = class extends AbstractSandbox {
       sandboxId: this.sandboxId
     };
     const response = await HttpUtils.post(url, headers, data);
-    const headerFields = this.extractHeaderFields(response.headers);
     if (response.status !== "Success") {
-      return { success: false, message: `Failed to write file ${request.path}`, ...headerFields };
+      return { success: false, message: `Failed to write file ${request.path}` };
     }
-    return { success: true, message: `Successfully write content to file ${request.path}`, ...headerFields };
+    return { success: true, message: `Successfully write content to file ${request.path}` };
   }
   async readFile(request) {
     const url = `${this.url}/read_file`;
@@ -1746,10 +1717,7 @@ var Sandbox = class extends AbstractSandbox {
       headers,
       data
     );
-    return {
-      content: response.result?.content ?? "",
-      ...this.extractHeaderFields(response.headers)
-    };
+    return { content: response.result?.content ?? "" };
   }
   // Upload
   async upload(request) {
@@ -1761,7 +1729,7 @@ var Sandbox = class extends AbstractSandbox {
     try {
       const fs = await import('fs');
       if (!fs.existsSync(sourcePath)) {
-        return { success: false, message: `File not found: ${sourcePath}`, cluster: "N/A", requestId: "N/A", eagleeyeTraceid: "N/A" };
+        return { success: false, message: `File not found: ${sourcePath}` };
       }
       const fileBuffer = fs.readFileSync(sourcePath);
       const fileName = sourcePath.split("/").pop() ?? "file";
@@ -1771,13 +1739,12 @@ var Sandbox = class extends AbstractSandbox {
         { targetPath, sandboxId: this.sandboxId ?? "" },
         { file: [fileName, fileBuffer, "application/octet-stream"] }
       );
-      const headerFields = this.extractHeaderFields(response.headers);
       if (response.status !== "Success") {
-        return { success: false, message: "Upload failed", ...headerFields };
+        return { success: false, message: "Upload failed" };
       }
-      return { success: true, message: `Successfully uploaded file ${fileName} to ${targetPath}`, ...headerFields };
+      return { success: true, message: `Successfully uploaded file ${fileName} to ${targetPath}` };
     } catch (e) {
-      return { success: false, message: `Upload failed: ${e}`, cluster: "N/A", requestId: "N/A", eagleeyeTraceid: "N/A" };
+      return { success: false, message: `Upload failed: ${e}` };
     }
   }
   // Close
@@ -2104,7 +2071,6 @@ exports.EnvHubClient = EnvHubClient;
 exports.EnvHubClientConfigSchema = EnvHubClientConfigSchema;
 exports.EnvHubError = EnvHubError;
 exports.ExecuteBashSessionResponseSchema = ExecuteBashSessionResponseSchema;
-exports.HeaderFieldsSchema = HeaderFieldsSchema;
 exports.HttpUtils = HttpUtils;
 exports.InternalServerRockError = InternalServerRockError;
 exports.InvalidParameterRockException = InvalidParameterRockException;