@aifabrix/builder 2.44.6 → 2.45.5

package/lib/datasource/audit-evidence-verify.js

@@ -0,0 +1,108 @@
+/**
+ * @fileoverview Run Plan 407.1 nine-row audit evidence matrix against dataplane APIs
+ * @author AI Fabrix Team
+ * @version 1.0.0
+ */
+
+'use strict';
+
+const { queryExecutions } = require('../api/audit.api');
+const { pollExecutionsUntilReady } = require('./audit-evidence-poll');
+const {
+  buildListMatrixRows,
+  buildPerExecutionAndTailRows
+} = require('./audit-evidence-verify-matrix');
+const { checkRow8CorrelationGroup, checkRow9Traces } = require('./audit-evidence-verify-helpers');
+
+/**
+ * @param {Object} ctx
+ * @returns {number}
+ */
+/**
+ * E2E runs tag executions with testRunId as correlationId; prefer that for list poll.
+ * @param {Object} ctx
+ * @returns {Object}
+ */
+function buildExecutionsPollQuery(ctx) {
+  const base = { pageSize: 100, sort: '-startedAt' };
+  if (ctx.correlationId) {
+    return { ...base, correlationId: ctx.correlationId };
+  }
+  return { ...base, datasourceKey: ctx.datasourceKey };
+}
+
+function resolveMinExpected(ctx) {
+  const fromOps = ctx.expectedOperations && ctx.expectedOperations.length;
+  const fromIds = ctx.executionIds && ctx.executionIds.length;
+  if (fromIds && fromOps) {
+    return Math.max(1, Math.min(fromOps, fromIds));
+  }
+  return Math.max(1, fromOps || fromIds || 1);
+}
+
+/**
+ * @param {string} datasourceKey
+ * @param {Object} ctx
+ * @returns {import('../api/types/audit.types').AuditEvidenceVerification}
+ */
+function failedMissingIds(datasourceKey, ctx) {
+  return {
+    status: 'failed',
+    datasourceKey,
+    correlationId: ctx.correlationId || null,
+    executionIds: [],
+    matrix: [
+      {
+        row: 0,
+        status: 'failed',
+        detail: 'missing audit.executionIds in envelope',
+        code: 'auditIdsMissing'
+      }
+    ]
+  };
+}
+
+/**
+ * @param {Object} ctx
+ * @returns {Promise<import('../api/types/audit.types').AuditEvidenceVerification>}
+ */
+async function verifyAuditEvidenceMatrix(ctx) {
+  const datasourceKey = ctx.datasourceKey || '';
+  const executionIds = Array.isArray(ctx.executionIds) ? ctx.executionIds : [];
+  if (!executionIds.length && !ctx.correlationId) {
+    return failedMissingIds(datasourceKey, ctx);
+  }
+
+  const { dataplaneUrl, authConfig } = ctx;
+  const minExpected = resolveMinExpected(ctx);
+  const execQuery = buildExecutionsPollQuery(ctx);
+  const polled = await pollExecutionsUntilReady({
+    fetchExecutions: () => queryExecutions(dataplaneUrl, authConfig, execQuery),
+    minExpected,
+    executionIds,
+    maxWaitMs: ctx.maxWaitMs,
+    intervalMs: ctx.intervalMs
+  });
+
+  const executions = polled.data || [];
+  const matrix = await buildListMatrixRows(ctx, polled, minExpected);
+  const tail = await buildPerExecutionAndTailRows(ctx, executions, executionIds);
+  matrix.push(...tail.rows);
+
+  const passed = matrix.every(r => r.status === 'passed');
+  return {
+    status: passed ? 'passed' : 'failed',
+    datasourceKey,
+    correlationId: ctx.correlationId || null,
+    executionIds: tail.execIds,
+    matrix
+  };
+}
+
+module.exports = {
+  verifyAuditEvidenceMatrix,
+  buildExecutionsPollQuery,
+  resolveMinExpected,
+  checkRow8CorrelationGroup,
+  checkRow9Traces
+};

package/lib/datasource/bulk-loader-run-batches.js

@@ -0,0 +1,88 @@
+/**
+ * @fileoverview Batch upload loop for datasource load (plan 144)
+ * @author AI Fabrix Team
+ * @version 1.0.0
+ */
+
+'use strict';
+
+const { bulkSyncRecords } = require('../api/records-bulk.api');
+const { withTransientRetry, errorStatusCode } = require('./bulk-loader-service-retry');
+
+/**
+ * @param {Object} state
+ * @param {number} index
+ * @param {Error} err
+ * @param {Object[]} batchRecords
+ */
+function recordBatchFailure(state, index, err, batchRecords) {
+  state.batches.push({
+    index: index + 1,
+    total: state.chunkCount,
+    error: err.message || String(err),
+    statusCode: errorStatusCode(err),
+    recordCount: batchRecords.length,
+    failed: batchRecords.map(r => ({ key: r.key, message: err.message || String(err) }))
+  });
+  state.totals.failedCount += batchRecords.length;
+}
+
+/**
+ * @param {Object} state
+ * @param {number} index
+ * @param {Object} response
+ * @param {Object[]} batchRecords
+ */
+function recordBatchSuccess(state, index, response, batchRecords) {
+  const inserted = response.insertedCount || 0;
+  const updated = response.updatedCount || 0;
+  const processed = response.totalProcessed || batchRecords.length;
+  const failedInBatch = Math.max(0, batchRecords.length - processed);
+
+  state.batches.push({
+    index: index + 1,
+    total: state.chunkCount,
+    insertedCount: inserted,
+    updatedCount: updated,
+    totalProcessed: processed,
+    failed: failedInBatch > 0 ? [{ message: 'Some records were not processed' }] : []
+  });
+  state.totals.insertedCount += inserted;
+  state.totals.updatedCount += updated;
+  state.totals.failedCount += failedInBatch;
+}
+
+/**
+ * @async
+ * @param {Object} params
+ * @returns {Promise<{ batches: Object[], totals: Object }>}
+ */
+async function uploadRecordBatches(params) {
+  const { chunks, syncType, dataplaneUrl, datasourceKey, authConfig } = params;
+  const state = {
+    batches: [],
+    totals: { insertedCount: 0, updatedCount: 0, failedCount: 0 },
+    chunkCount: chunks.length
+  };
+
+  for (let i = 0; i < chunks.length; i += 1) {
+    const batchRecords = chunks[i];
+    const body = { syncType, sync: false, records: batchRecords };
+    try {
+      const response = await withTransientRetry(() =>
+        bulkSyncRecords(dataplaneUrl, datasourceKey, authConfig, body)
+      );
+      recordBatchSuccess(state, i, response, batchRecords);
+    } catch (err) {
+      recordBatchFailure(state, i, err, batchRecords);
+    }
+  }
+
+  return { batches: state.batches, totals: state.totals };
+}
+
+module.exports = {
+  uploadRecordBatches,
+  recordBatchFailure,
+  recordBatchSuccess
+};

package/lib/datasource/bulk-loader-service-retry.js

@@ -0,0 +1,52 @@
+/**
+ * @fileoverview Retry helpers for bulk loader (plan 144)
+ * @author AI Fabrix Team
+ * @version 1.0.0
+ */
+
+'use strict';
+
+const MAX_RETRIES = 3;
+const RETRY_STATUS = new Set([429, 500, 502, 503, 504]);
+
+/**
+ * @param {Error} err
+ * @returns {number|null}
+ */
+function errorStatusCode(err) {
+  if (!err || typeof err !== 'object') return null;
+  if (typeof err.statusCode === 'number') return err.statusCode;
+  if (err.response && typeof err.response.status === 'number') return err.response.status;
+  return null;
+}
+
+function sleep(ms) {
+  return new Promise(resolve => setTimeout(resolve, ms));
+}
+
+/**
+ * @async
+ * @param {Function} fn
+ * @returns {Promise<*>}
+ */
+async function withTransientRetry(fn) {
+  let lastErr;
+  for (let attempt = 0; attempt < MAX_RETRIES; attempt += 1) {
+    try {
+      return await fn();
+    } catch (err) {
+      lastErr = err;
+      const code = errorStatusCode(err);
+      if (!code || !RETRY_STATUS.has(code) || attempt >= MAX_RETRIES - 1) {
+        throw err;
+      }
+      await sleep(500 * (attempt + 1));
+    }
+  }
+  throw lastErr;
+}
+
+module.exports = {
+  withTransientRetry,
+  errorStatusCode
+};

package/lib/datasource/bulk-loader-service.js

@@ -0,0 +1,127 @@
+/**
+ * @fileoverview Load local fixtures via dataplane bulk record sync (plan 144)
+ * @author AI Fabrix Team
+ * @version 1.0.0
+ */
+
+'use strict';
+
+const { resolveDataFilePath } = require('./local-data-paths');
+const { parseRecordFile, estimatePayloadBytes } = require('./record-file-parser');
+const { normalizeRecordsForBulk } = require('./record-mapper');
+const { resolveLoadExportContext } = require('./datasource-load-export-context');
+const { uploadRecordBatches } = require('./bulk-loader-run-batches');
+const { runWithDatasourceLoadExportSpinner } = require('../utils/datasource-load-export-ui');
+
+const DEFAULT_BATCH_SIZE = 100;
+
+/**
+ * @param {Object[]} records
+ * @param {number} batchSize
+ * @returns {Object[][]}
+ */
+function chunkRecords(records, batchSize) {
+  const size = Math.max(1, batchSize);
+  const chunks = [];
+  for (let i = 0; i < records.length; i += size) {
+    chunks.push(records.slice(i, i + size));
+  }
+  return chunks;
+}
+
+/**
+ * @param {Object} params
+ * @returns {Object}
+ */
+function buildLoadResultBase(params) {
+  const { ctx, filePath, format, records, batchSize, syncType, dryRun } = params;
+  return {
+    datasourceKey: ctx.datasourceKey,
+    systemKey: ctx.systemKey,
+    appKey: ctx.appKey,
+    dryRun,
+    file: filePath,
+    format,
+    recordCount: records.length,
+    batchSize,
+    syncType,
+    estimatedPayloadBytes: estimatePayloadBytes(records),
+    batches: [],
+    totals: { insertedCount: 0, updatedCount: 0, failedCount: 0 }
+  };
+}
+
+/**
+ * @async
+ * @param {Object} params
+ * @returns {Promise<Object>}
+ */
+async function executeDatasourceLoadWork(params) {
+  const { ctx, options, batchSize, syncType, dryRun } = params;
+  const { filePath, format } = resolveDataFilePath({
+    systemKey: ctx.systemKey,
+    entitySuffix: ctx.entitySuffix,
+    file: options.file,
+    format: options.format
+  });
+
+  const raw = parseRecordFile(filePath, format);
+  const records = normalizeRecordsForBulk(raw, ctx.datasource);
+  const result = buildLoadResultBase({
+    ctx,
+    filePath,
+    format,
+    records,
+    batchSize,
+    syncType,
+    dryRun
+  });
+
+  if (dryRun) {
+    return { ...result, context: ctx };
+  }
+
+  const { batches, totals } = await uploadRecordBatches({
+    chunks: chunkRecords(records, batchSize),
+    syncType,
+    dataplaneUrl: ctx.dataplaneUrl,
+    datasourceKey: ctx.datasourceKey,
+    authConfig: ctx.authConfig
+  });
+
+  return { ...result, batches, totals, context: ctx };
+}
+
+/**
+ * @async
+ * @param {string} datasourceKey
+ * @param {Object} options
+ * @returns {Promise<Object>}
+ */
+async function runDatasourceLoad(datasourceKey, options = {}) {
+  const ctx = await resolveLoadExportContext(datasourceKey, options);
+  const batchSize = options.batchSize
+    ? parseInt(String(options.batchSize), 10)
+    : DEFAULT_BATCH_SIZE;
+  const syncType = options.syncType || 'incremental';
+  const dryRun = options.dryRun === true;
+  const startedAt = Date.now();
+
+  const payload = await runWithDatasourceLoadExportSpinner(
+    () => executeDatasourceLoadWork({ ctx, options, batchSize, syncType, dryRun }),
+    {
+      operation: dryRun ? 'load-dry-run' : 'load',
+      datasourceKey: ctx.datasourceKey,
+      json: options.json === true
+    }
+  );
+
+  return { ...payload, durationMs: Date.now() - startedAt };
+}
+
+module.exports = {
+  runDatasourceLoad,
+  chunkRecords,
+  buildLoadResultBase,
+  DEFAULT_BATCH_SIZE
+};

package/lib/datasource/capability/run-capability-dimension.js

@@ -43,7 +43,8 @@ async function maybeLoadDimensionCatalogKeys() {
       page: 1,
       pageSize: 500
     });
-    const items = res?.data?.items || res?.data?.data?.items || res?.items || [];
+    const raw = res?.data?.items ?? res?.data?.data ?? res?.items;
+    const items = Array.isArray(raw) ? raw : raw?.items || [];
     const keys = new Set(
       Array.isArray(items) ? items.map((d) => String(d?.key || '').trim()).filter(Boolean) : []
     );

package/lib/datasource/datasource-exporter-resolve.js

@@ -0,0 +1,82 @@
+/**
+ * @fileoverview Export path and search body helpers (plan 144)
+ * @author AI Fabrix Team
+ * @version 1.0.0
+ */
+
+'use strict';
+
+const path = require('path');
+const { resolveDefaultDataOutputPath, inferFormatFromPath } = require('./local-data-paths');
+
+/**
+ * @param {string|undefined} raw
+ * @returns {Object|undefined}
+ */
+function parseFilterJson(raw) {
+  if (!raw || typeof raw !== 'string' || !raw.trim()) {
+    return undefined;
+  }
+  try {
+    const parsed = JSON.parse(raw);
+    if (parsed && typeof parsed === 'object' && !Array.isArray(parsed)) {
+      return parsed;
+    }
+    throw new Error('filter must be a JSON object');
+  } catch (err) {
+    throw new Error(`Invalid --filter JSON: ${err.message}`);
+  }
+}
+
+/**
+ * @param {string|undefined} raw
+ * @returns {string[]|undefined}
+ */
+function parseFieldsCsv(raw) {
+  if (!raw || typeof raw !== 'string' || !raw.trim()) {
+    return undefined;
+  }
+  return raw
+    .split(',')
+    .map(s => s.trim())
+    .filter(Boolean);
+}
+
+/**
+ * @param {Object} opts
+ * @returns {{ outputPath: string, format: 'json'|'ndjson', limit: number, intent: string, filters: Object|undefined, fields: string[]|undefined }}
+ */
+function resolveExportOptions(opts) {
+  let limit = opts.limit ? parseInt(String(opts.limit), 10) : 1000;
+  if (Number.isNaN(limit) || limit < 1) {
+    limit = 1000;
+  }
+  limit = Math.min(limit, 10000);
+
+  const intent = opts.intent || 'validation';
+  const filters = parseFilterJson(opts.filter);
+  const fields = parseFieldsCsv(opts.fields);
+
+  let outputPath;
+  let format = opts.format;
+  if (opts.file && String(opts.file).trim()) {
+    outputPath = path.resolve(String(opts.file).trim());
+    format = format || inferFormatFromPath(outputPath);
+  } else {
+    const resolved = resolveDefaultDataOutputPath({
+      systemKey: opts.systemKey,
+      entitySuffix: opts.entitySuffix,
+      format: format || 'json'
+    });
+    outputPath = resolved.filePath;
+    format = opts.format || resolved.format;
+  }
+
+  return { outputPath, format, limit, intent, filters, fields };
+}
+
+module.exports = {
+  parseFilterJson,
+  parseFieldsCsv,
+  resolveExportOptions
+};

package/lib/datasource/datasource-exporter-service.js

@@ -0,0 +1,144 @@
+/**
+ * @fileoverview Export governed records via Records Search (plan 144)
+ * @author AI Fabrix Team
+ * @version 1.0.0
+ */
+
+'use strict';
+
+const fs = require('fs');
+const path = require('path');
+const { searchRecords } = require('../api/records-search.api');
+const { resolveLoadExportContext } = require('./datasource-load-export-context');
+const { resolveExportOptions } = require('./datasource-exporter-resolve');
+const { runWithDatasourceLoadExportSpinner } = require('../utils/datasource-load-export-ui');
+
+const DEFAULT_LIMIT = 1000;
+const MAX_LIMIT = 10000;
+
+/**
+ * @param {Object[]} data
+ * @param {string[]|undefined} fields
+ * @returns {Object[]}
+ */
+function projectExportRows(data, fields) {
+  if (!fields || !fields.length) {
+    return data;
+  }
+  return data.map(row => {
+    if (!row || typeof row !== 'object') {
+      return row;
+    }
+    const meta =
+      row.metadata && typeof row.metadata === 'object' && !Array.isArray(row.metadata)
+        ? row.metadata
+        : row;
+    const out = {};
+    fields.forEach(field => {
+      if (Object.prototype.hasOwnProperty.call(meta, field)) {
+        out[field] = meta[field];
+      }
+    });
+    return out;
+  });
+}
+
+/**
+ * @param {string} filePath
+ * @param {'json'|'ndjson'} format
+ * @param {Object[]} rows
+ */
+function writeExportFile(filePath, format, rows) {
+  const dir = path.dirname(filePath);
+  fs.mkdirSync(dir, { recursive: true });
+  if (format === 'ndjson') {
+    const body = rows.map(row => JSON.stringify(row)).join('\n');
+    fs.writeFileSync(filePath, body.length ? `${body}\n` : '', 'utf8');
+    return;
+  }
+  fs.writeFileSync(filePath, `${JSON.stringify(rows, null, 2)}\n`, 'utf8');
+}
+
+/**
+ * @async
+ * @param {Object} ctx
+ * @param {Object} resolved
+ * @returns {Promise<Object>}
+ */
+async function executeDatasourceExportWork(ctx, resolved) {
+  const searchBody = {
+    intent: resolved.intent,
+    datasourceKeys: [ctx.datasourceKey],
+    searchMode: 'full',
+    limit: resolved.limit
+  };
+  if (resolved.filters) {
+    searchBody.filters = resolved.filters;
+  }
+
+  const response = await searchRecords(ctx.dataplaneUrl, ctx.authConfig, searchBody);
+  if (!response || response.success === false) {
+    const msg = response?.formattedError || response?.error || 'Records search request failed';
+    throw new Error(msg);
+  }
+
+  const data = Array.isArray(response.data) ? response.data : [];
+  const projected = projectExportRows(data, resolved.fields);
+  writeExportFile(resolved.outputPath, resolved.format, projected);
+
+  const meta = response.meta && typeof response.meta === 'object' ? response.meta : {};
+  const excluded = meta.excluded && typeof meta.excluded === 'object' ? meta.excluded : {};
+
+  return {
+    datasourceKey: ctx.datasourceKey,
+    systemKey: ctx.systemKey,
+    appKey: ctx.appKey,
+    outputFile: resolved.outputPath,
+    format: resolved.format,
+    recordCount: projected.length,
+    limit: resolved.limit,
+    intent: resolved.intent,
+    filters: resolved.filters || null,
+    fields: resolved.fields || null,
+    meta: { excluded, auditRef: meta.auditRef },
+    excludedAbac: Number(excluded.abac) || 0,
+    excludedFilter: Number(excluded.filter) || 0,
+    hitLimitCap: projected.length >= resolved.limit,
+    context: ctx
+  };
+}
+
+/**
+ * @async
+ * @param {string} datasourceKey
+ * @param {Object} options
+ * @returns {Promise<Object>}
+ */
+async function runDatasourceExport(datasourceKey, options = {}) {
+  const ctx = await resolveLoadExportContext(datasourceKey, options);
+  const resolved = resolveExportOptions({
+    ...options,
+    systemKey: ctx.systemKey,
+    entitySuffix: ctx.entitySuffix
+  });
+  const startedAt = Date.now();
+
+  const payload = await runWithDatasourceLoadExportSpinner(
+    () => executeDatasourceExportWork(ctx, resolved),
+    {
+      operation: 'export',
+      datasourceKey: ctx.datasourceKey,
+      json: options.json === true
+    }
+  );
+
+  return { ...payload, durationMs: Date.now() - startedAt };
+}
+
+module.exports = {
+  runDatasourceExport,
+  projectExportRows,
+  writeExportFile,
+  DEFAULT_LIMIT,
+  MAX_LIMIT
+};