tests/utils/dts-export.ts - develop - githubmirror/strapi

tests/utils/dts-export.ts
106 строк3 KB
Ben Irvin
revert: move back test files
18 ноя 2025, 17:15
18 ноя 2025, 17:152f1e80a
import type { Core } from '@strapi/strapi';

import dts from '@strapi/data-transfer';
import { createStrapi } from '@strapi/strapi';
import { ALLOWED_CONTENT_TYPES } from '../e2e/constants';

const {
  file: {
    providers: { createLocalFileDestinationProvider },
  },
  strapi: {
    providers: { createLocalStrapiSourceProvider },
  },
  engine: { createTransferEngine },
} = dts;

export const exportData = async (): Promise<void> => {
  const args = process.argv.slice(2);

  if (args.length !== 1) {
    console.error('Please provide the export file name as a parameter.');
    process.exit(1);
  }

  const strapi = await createStrapiInstance();

  const source = createSourceProvider(strapi);
  const destination = createDestinationProvider(args[0]);

  const engine = createTransferEngine(source, destination, {
    versionStrategy: 'ignore', // for an export to file, versionStrategy will always be skipped
    schemaStrategy: 'ignore', // for an export to file, schemaStrategy will always be skipped
    transforms: {
      links: [
        {
          filter(link) {
            return (
              ALLOWED_CONTENT_TYPES.includes(link.left.type) &&
              ALLOWED_CONTENT_TYPES.includes(link.right.type)
            );
          },
        },
      ],
      entities: [
        {
          filter(entity) {
            return ALLOWED_CONTENT_TYPES.includes(entity.type);
          },
        },
      ],
    },
  });

  engine.diagnostics.onDiagnostic((diagnostic) => {
    if (diagnostic.kind !== 'info') console.log(diagnostic);
  });

  try {
    const results = await engine.transfer();
    const { destination: dest, engine: engineResults } = results;

    // Enhanced output with details (from CLI version)
    if (dest?.file?.path) {
      const path = require('path');
      const relativeArchivePath = path.relative(process.cwd(), dest.file.path);
      console.log(`Dataset exported to: ${relativeArchivePath}`);
      console.log('The export contains:');
      console.log(`  - ${engineResults.schemas?.count ?? 0} schemas`);
      console.log(`  - ${engineResults.entities?.count ?? 0} entities`);
      console.log(`  - ${engineResults.links?.count ?? 0} links`);
      console.log(`  - ${engineResults.assets?.count ?? 0} assets`);
      console.log(`  - ${engineResults.configuration?.count ?? 0} configs`);
    } else {
      // Fallback to JSON output (browser version)
      console.log(JSON.stringify(results.engine, null, 2));
    }
  } catch {
    console.error('Export process failed.');
    process.exit(1);
  }

  process.exit(0);
};

const createSourceProvider = (strapi: Core.Strapi) =>
  createLocalStrapiSourceProvider({
    async getStrapi() {
      return strapi;
    },
  });

const createDestinationProvider = (filePath: string) =>
  createLocalFileDestinationProvider({
    file: { path: filePath },
    encryption: { enabled: false },
    compression: { enabled: false },
  });

const createStrapiInstance = async (logLevel = 'error'): Promise<Core.Strapi> => {
  const app = createStrapi();

  app.log.level = logLevel;
  const loadedApp = await app.load();

  return loadedApp;
};