@aifabrix/builder 2.45.0 → 2.46.0

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
+};

package/lib/datasource/datasource-identifier.js

@@ -0,0 +1,148 @@
+/**
+ * @fileoverview Match datasource CLI identifiers to integration files (key or filename stem).
+ * @author AI Fabrix Team
+ * @version 1.0.0
+ */
+
+'use strict';
+
+const path = require('path');
+const { getIntegrationPath } = require('../utils/paths');
+const { resolveApplicationConfigPath } = require('../utils/app-config-resolver');
+const { loadConfigFile } = require('../utils/config-format');
+
+/**
+ * @param {string} fileRef
+ * @returns {string}
+ */
+function datasourceFileStem(fileRef) {
+  return path.basename(String(fileRef || ''), path.extname(String(fileRef || '')));
+}
+
+/**
+ * True when identifier matches parsed `key` or the datasource JSON filename (with or without .json).
+ * @param {string} identifier
+ * @param {string} fileRef - Manifest-relative datasource path
+ * @param {Object|null} parsed
+ * @returns {boolean}
+ */
+function datasourceIdentifierMatchesFile(identifier, fileRef, parsed) {
+  const id = String(identifier || '').trim();
+  if (!id || !parsed || typeof parsed !== 'object') {
+    return false;
+  }
+  if (parsed.key === id) {
+    return true;
+  }
+  const stem = datasourceFileStem(fileRef);
+  if (stem === id) {
+    return true;
+  }
+  if (`${stem}.json` === id.toLowerCase()) {
+    return true;
+  }
+  return false;
+}
+
+/**
+ * Pick integration app folder for a datasource identifier (longest matching prefix wins).
+ * @param {string} identifier
+ * @param {string[]} appNames
+ * @returns {string|null}
+ */
+function pickIntegrationAppForDatasourceIdentifier(identifier, appNames) {
+  const id = String(identifier || '').trim();
+  if (!id) {
+    return null;
+  }
+  const matches = appNames.filter(app => id === app || id.startsWith(`${app}-`));
+  if (matches.length === 0) {
+    return null;
+  }
+  matches.sort((a, b) => b.length - a.length);
+  return matches[0];
+}
+
+/**
+ * List manifest datasource files for an integration app.
+ * @param {string} appKey
+ * @returns {{ appPath: string, schemaBasePath: string, datasourceFiles: string[] }|null}
+ */
+function listDatasourceFilesForApp(appKey) {
+  const appPath = getIntegrationPath(appKey);
+  let config;
+  try {
+    const configPath = resolveApplicationConfigPath(appPath);
+    config = loadConfigFile(configPath);
+  } catch {
+    return null;
+  }
+  const schemaBasePath = config.externalIntegration?.schemaBasePath || './';
+  const datasourceFiles = config.externalIntegration?.dataSources || [];
+  return { appPath, schemaBasePath, datasourceFiles };
+}
+
+/**
+ * @param {string} appPath
+ * @param {string} schemaBasePath
+ * @param {string} fileRef
+ * @returns {string}
+ */
+function resolveDatasourceFilePath(appPath, schemaBasePath, fileRef) {
+  if (path.isAbsolute(schemaBasePath)) {
+    return path.join(schemaBasePath, fileRef);
+  }
+  return path.join(appPath, schemaBasePath, fileRef);
+}
+
+/**
+ * Collect datasource file refs in one app that match the identifier.
+ * @param {string} appKey
+ * @param {string} identifier
+ * @returns {string[]}
+ */
+function collectDatasourceFileRefsMatchingIdentifier(appKey, identifier) {
+  const listed = listDatasourceFilesForApp(appKey);
+  if (!listed) {
+    return [];
+  }
+  const fsSync = require('fs');
+  const hits = [];
+  for (const fileRef of listed.datasourceFiles) {
+    if (!fileRef || typeof fileRef !== 'string') {
+      continue;
+    }
+    const fullPath = resolveDatasourceFilePath(listed.appPath, listed.schemaBasePath, fileRef);
+    if (!fsSync.existsSync(fullPath)) {
+      continue;
+    }
+    try {
+      const parsed = loadConfigFile(fullPath);
+      if (datasourceIdentifierMatchesFile(identifier, fileRef, parsed)) {
+        hits.push(fileRef);
+      }
+    } catch {
+      // skip unreadable or invalid files
+    }
+  }
+  return hits;
+}
+
+/**
+ * @param {string} appKey
+ * @param {string} identifier
+ * @returns {boolean}
+ */
+function appHasDatasourceIdentifier(appKey, identifier) {
+  return collectDatasourceFileRefsMatchingIdentifier(appKey, identifier).length > 0;
+}
+
+module.exports = {
+  datasourceFileStem,
+  datasourceIdentifierMatchesFile,
+  pickIntegrationAppForDatasourceIdentifier,
+  listDatasourceFilesForApp,
+  resolveDatasourceFilePath,
+  collectDatasourceFileRefsMatchingIdentifier,
+  appHasDatasourceIdentifier
+};

package/lib/datasource/datasource-load-export-context.js

@@ -0,0 +1,95 @@
+/**
+ * @fileoverview Shared context resolution for datasource load/export (plan 144)
+ * @author AI Fabrix Team
+ * @version 1.0.0
+ */
+
+'use strict';
+
+const { getConfig } = require('../core/config');
+const { setupIntegrationTestAuth } = require('../external-system/test-auth');
+const { applyLabDataplaneApiKeyBearer } = require('../utils/token-manager');
+const { resolveAppKeyForDatasource } = require('./resolve-app');
+const { getSystemKeyFromAppKey } = require('./integration-context');
+const { loadDatasourceForApp } = require('./unified-validation-run-resolve');
+const { findDatasourceFileByKey } = require('./integration-context');
+const { getIntegrationPath } = require('../utils/paths');
+const { resolveApplicationConfigPath } = require('../utils/app-config-resolver');
+const { computeEntitySuffix } = require('./local-data-paths');
+
+/**
+ * @param {string} appKey
+ * @throws {Error}
+ */
+async function assertExternalIntegrationApp(appKey) {
+  const pathsUtil = require('../utils/paths');
+  const appType = await pathsUtil.detectAppType(appKey).catch(() => null);
+  if (!appType || appType.baseDir !== 'integration') {
+    throw new Error(
+      'datasource load/export apply to external integration folders only (integration/<systemKey>/).'
+    );
+  }
+}
+
+/**
+ * @async
+ * @param {string} datasourceKey
+ * @param {Object} options
+ * @returns {Promise<Object>}
+ */
+async function resolveLoadExportContext(datasourceKey, options = {}) {
+  const { appKey } = await resolveAppKeyForDatasource(datasourceKey, options.app);
+  await assertExternalIntegrationApp(appKey);
+  const systemKey = await getSystemKeyFromAppKey(appKey);
+  const {
+    datasource,
+    datasourcePath,
+    datasourceKey: canonicalDatasourceKey
+  } = loadDatasourceForApp(appKey, datasourceKey, findDatasourceFileByKey);
+  const appPath = getIntegrationPath(appKey);
+  const manifestPath = resolveApplicationConfigPath(appPath);
+  const entitySuffix = computeEntitySuffix(systemKey, canonicalDatasourceKey);
+
+  let authConfig = options.authConfig;
+  let dataplaneUrl = options.dataplaneUrl || options.dataplane;
+  if (!authConfig || !dataplaneUrl) {
+    const configObj = await getConfig();
+    const auth = await setupIntegrationTestAuth(
+      appKey,
+      { environment: options.environment || options.env, dataplane: options.dataplane },
+      configObj
+    );
+    authConfig = auth.authConfig;
+    dataplaneUrl = auth.dataplaneUrl;
+  }
+
+  authConfig = applyLabDataplaneApiKeyBearer(authConfig);
+
+  return {
+    appKey,
+    systemKey,
+    datasourceKey: canonicalDatasourceKey,
+    datasource,
+    datasourcePath,
+    manifestPath,
+    entitySuffix,
+    authConfig,
+    dataplaneUrl,
+    environment: options.environment || options.env
+  };
+}
+
+/**
+ * Gray manifest label segment for TTY (integration/<appKey>).
+ * @param {string} appKey
+ * @returns {string}
+ */
+function integrationManifestLabel(appKey) {
+  return `integration/${appKey}`;
+}
+
+module.exports = {
+  assertExternalIntegrationApp,
+  resolveLoadExportContext,
+  integrationManifestLabel
+};

package/lib/datasource/integration-context.js

@@ -9,6 +9,10 @@ const fs = require('fs').promises;
 const { getIntegrationPath } = require('../utils/paths');
 const { resolveApplicationConfigPath } = require('../utils/app-config-resolver');
 const { loadConfigFile } = require('../utils/config-format');
+const {
+  datasourceIdentifierMatchesFile,
+  resolveDatasourceFilePath
+} = require('./datasource-identifier');
 
 /**
  * @param {string} appKey - Integration app key
@@ -43,13 +47,11 @@ function findDatasourceFileByKey(appPath, schemaBasePath, datasourceFiles, datas
   const fsSync = require('fs');
   for (const f of datasourceFiles) {
     if (!f || typeof f !== 'string') continue;
-    const fullPath = path.isAbsolute(schemaBasePath)
-      ? path.join(schemaBasePath, f)
-      : path.join(appPath, schemaBasePath, f);
+    const fullPath = resolveDatasourceFilePath(appPath, schemaBasePath, f);
     if (!fsSync.existsSync(fullPath)) continue;
     try {
       const parsed = loadConfigFile(fullPath);
-      if (parsed && parsed.key === datasourceKey) return f;
+      if (datasourceIdentifierMatchesFile(datasourceKey, f, parsed)) return f;
     } catch {
       // skip unreadable or invalid files
     }

package/lib/datasource/local-data-paths.js

@@ -0,0 +1,145 @@
+/**
+ * @fileoverview Resolve integration/.data/ paths for datasource load/export (plan 144)
+ * @author AI Fabrix Team
+ * @version 1.0.0
+ */
+
+'use strict';
+
+const path = require('path');
+const fs = require('fs');
+const { getCwdIntegrationRoot, getIntegrationRoot } = require('../utils/paths');
+
+const DATA_DIR_NAME = '.data';
+
+/**
+ * Absolute path to integration/.data (cwd integration root preferred).
+ * @returns {string}
+ */
+function getIntegrationDataDir() {
+  const cwdRoot = getCwdIntegrationRoot();
+  const base = cwdRoot || getIntegrationRoot();
+  return path.join(base, DATA_DIR_NAME);
+}
+
+/**
+ * Entity suffix from datasource key and system key.
+ * @param {string} systemKey
+ * @param {string} datasourceKey
+ * @returns {string}
+ */
+function computeEntitySuffix(systemKey, datasourceKey) {
+  const sys = String(systemKey || '').trim();
+  const key = String(datasourceKey || '').trim();
+  const prefix = `${sys}-`;
+  if (sys && key.startsWith(prefix)) {
+    return key.slice(prefix.length) || key;
+  }
+  const lastHyphen = key.lastIndexOf('-');
+  if (lastHyphen > 0) {
+    return key.slice(lastHyphen + 1);
+  }
+  return key;
+}
+
+/**
+ * Default fixture basename (without extension).
+ * @param {string} systemKey
+ * @param {string} entitySuffix
+ * @returns {string}
+ */
+function defaultDataBasename(systemKey, entitySuffix) {
+  return `${systemKey}-data-${entitySuffix}`;
+}
+
+/**
+ * @param {string} dir
+ * @param {string} basename
+ * @param {'json'|'ndjson'} format
+ * @returns {string}
+ */
+function dataFilePath(dir, basename, format) {
+  const ext = format === 'ndjson' ? '.ndjson' : '.json';
+  return path.join(dir, `${basename}${ext}`);
+}
+
+/**
+ * Infer format from file extension.
+ * @param {string} filePath
+ * @returns {'json'|'ndjson'}
+ */
+function inferFormatFromPath(filePath) {
+  const ext = path.extname(filePath).toLowerCase();
+  if (ext === '.ndjson' || ext === '.jsonl') {
+    return 'ndjson';
+  }
+  return 'json';
+}
+
+/**
+ * Default integration/.data output path (file need not exist).
+ * @param {Object} opts
+ * @param {string} opts.systemKey
+ * @param {string} opts.entitySuffix
+ * @param {'json'|'ndjson'} [opts.format]
+ * @returns {{ filePath: string, format: 'json'|'ndjson' }}
+ */
+function resolveDefaultDataOutputPath(opts) {
+  const dir = getIntegrationDataDir();
+  const basename = defaultDataBasename(opts.systemKey, opts.entitySuffix);
+  const format = opts.format === 'ndjson' ? 'ndjson' : 'json';
+  return { filePath: dataFilePath(dir, basename, format), format };
+}
+
+/**
+ * Resolve data file: --file override or default integration/.data paths.
+ * @param {Object} opts
+ * @param {string} opts.systemKey
+ * @param {string} opts.entitySuffix
+ * @param {string} [opts.file] - CLI --file
+ * @param {'json'|'ndjson'} [opts.format] - explicit format
+ * @returns {{ filePath: string, format: 'json'|'ndjson' }}
+ */
+function resolveDataFilePath(opts) {
+  if (opts.file && String(opts.file).trim()) {
+    const filePath = path.resolve(String(opts.file).trim());
+    const format = opts.format || inferFormatFromPath(filePath);
+    return { filePath, format };
+  }
+
+  const dir = getIntegrationDataDir();
+  const basename = defaultDataBasename(opts.systemKey, opts.entitySuffix);
+  const jsonPath = dataFilePath(dir, basename, 'json');
+  const ndjsonPath = dataFilePath(dir, basename, 'ndjson');
+
+  if (opts.format === 'json' && fs.existsSync(jsonPath)) {
+    return { filePath: jsonPath, format: 'json' };
+  }
+  if (opts.format === 'ndjson' && fs.existsSync(ndjsonPath)) {
+    return { filePath: ndjsonPath, format: 'ndjson' };
+  }
+  if (!opts.format || opts.format === 'json') {
+    if (fs.existsSync(jsonPath)) {
+      return { filePath: jsonPath, format: 'json' };
+    }
+  }
+  if (!opts.format || opts.format === 'ndjson') {
+    if (fs.existsSync(ndjsonPath)) {
+      return { filePath: ndjsonPath, format: 'ndjson' };
+    }
+  }
+
+  throw new Error(
+    `No local data file found. Expected one of:\n  ${jsonPath}\n  ${ndjsonPath}\n` +
+      'Use --file <path> to override.'
+  );
+}
+
+module.exports = {
+  getIntegrationDataDir,
+  computeEntitySuffix,
+  defaultDataBasename,
+  inferFormatFromPath,
+  resolveDefaultDataOutputPath,
+  resolveDataFilePath
+};