fix: render data import tree errors in a table (#1190)

* fix: create table for data import tree JSON Error

* chore: ci-rerun

* fix: update the changes

* fix: remove extra function

* fix: add error message

* fix: update the error for json output

* fix: preserve original error on invalid json payload

---------

Co-authored-by: Cristian Dominguez <cdominguez@salesforce.com>

Author: soridalac

src/commands/data/import/tree.ts

@@ -5,12 +5,13 @@
  * For full license text, see LICENSE.txt file in the repo root or https://opensource.org/licenses/BSD-3-Clause
  */
 
-import { Messages } from '@salesforce/core';
+import { Messages, SfError } from '@salesforce/core';
 import { SfCommand, Flags } from '@salesforce/sf-plugins-core';
+import { ensureString, isObject } from '@salesforce/ts-types';
 import { importFromPlan } from '../../../api/data/tree/importPlan.js';
 import { importFromFiles } from '../../../api/data/tree/importFiles.js';
 import { orgFlags } from '../../../flags.js';
-import type { ImportResult } from '../../../api/data/tree/importTypes.js';
+import type { ImportResult, TreeResponse } from '../../../api/data/tree/importTypes.js';
 
 Messages.importMessagesDirectoryFromMetaUrl(import.meta.url);
 const messages = Messages.loadMessages('@salesforce/plugin-data', 'tree.import');
@@ -49,19 +50,71 @@ export default class Import extends SfCommand<ImportResult[]> {
     const { flags } = await this.parse(Import);
 
     const conn = flags['target-org'].getConnection(flags['api-version']);
-    const results = flags.plan
-      ? await importFromPlan(conn, flags.plan)
-      : await importFromFiles(conn, flags.files ?? []);
 
-    this.table({
-      data: results,
-      columns: [
-        { key: 'refId', name: 'Reference ID' },
-        { key: 'type', name: 'Type' },
-        { key: 'id', name: 'ID' },
-      ],
-      title: 'Import Results',
-    });
-    return results;
+    try {
+      const results = flags.plan
+        ? await importFromPlan(conn, flags.plan)
+        : await importFromFiles(conn, flags.files ?? []);
+
+      this.table({
+        data: results,
+        columns: [
+          { key: 'refId', name: 'Reference ID' },
+          { key: 'type', name: 'Type' },
+          { key: 'id', name: 'ID' },
+        ],
+        title: 'Import Results',
+      });
+      return results;
+    } catch (err) {
+      const error = err as Error;
+      if (
+        error.cause &&
+        error.cause instanceof Error &&
+        'data' in error.cause &&
+        isObject(error.cause.data) &&
+        'message' in error.cause.data
+      ) {
+        const getTreeResponse = (payload: string): TreeResponse => {
+          try {
+            return JSON.parse(payload) as TreeResponse;
+          } catch (parseErr) {
+            // throw original error on invalid JSON payload
+            if (parseErr instanceof Error && parseErr.name === 'SyntaxError') {
+              throw error;
+            }
+            throw parseErr;
+          }
+        };
+        const errorData = getTreeResponse(ensureString(error.cause.data.message));
+
+        if (errorData.hasErrors) {
+          const errorResults = errorData.results
+            .map((result) =>
+              result.errors.map((errors) => ({
+                referenceId: result.referenceId,
+                StatusCode: errors.statusCode,
+                Message: errors.message,
+                fields: errors.fields.join(', ') || 'N/A',
+              }))
+            )
+            .flat();
+          this.table({
+            data: errorResults,
+            columns: [
+              { key: 'referenceId', name: 'Reference ID' },
+              { key: 'StatusCode', name: 'Status Code' },
+              { key: 'Message', name: 'Error Message' },
+              { key: 'fields', name: 'Fields' },
+            ],
+            title: 'Tree import errors',
+          });
+          const errors = new SfError('Data Import failed');
+          errors.setData(errorResults);
+          throw errors;
+        }
+      }
+      throw error;
+    }
   }
 }