strapi/packages/core/data-transfer/lib/providers/local-strapi-destination-provider.ts

import { createLogger } from '@strapi/logger';
import chalk from 'chalk';
import { Duplex } from 'stream';
import type { IDestinationProvider, IMetadata, ProviderType, Stream } from '../../types';

interface ILocalStrapiDestinationProviderOptions {
  getStrapi(): Promise<Strapi.Strapi>;
}

const log = createLogger();

export class LocalStrapiDestinationProvider implements IDestinationProvider {
  name: string = 'provider::destination.local-strapi';
  type: ProviderType = 'destination';

  options: ILocalStrapiDestinationProviderOptions;
  strapi?: Strapi.Strapi;

  constructor(options: ILocalStrapiDestinationProviderOptions) {
    this.options = options;
  }

  async bootstrap(): Promise<void> {
    this.strapi = await this.options.getStrapi();
  }

  async close(): Promise<void> {
    await this.strapi.destroy?.();
  }

  getMetadata(): IMetadata | Promise<IMetadata> {
    return null;
  }

  getEntitiesStream(): Stream {
    const self = this;

    return new Writable({
      objectMode: true,
      async write(entity, _encoding, callback) {
        if (!self.strapi) {
          callback(new Error('Strapi instance not found'));
        }

        const { type: uid, id, data } = entity;

        try {
          await strapi.entityService.create(uid, { data });
        } catch (e) {
          log.warn(
            chalk.bold(`Failed to import ${chalk.yellowBright(uid)} (${chalk.greenBright(id)})`)
          );
          e.details.errors
            .map((err, i) => {
              const info = {
                uid: chalk.yellowBright(`[${uid}]`),
                path: chalk.blueBright(`[${err.path.join('.')}]`),
                id: chalk.greenBright(`[${id}]`),
                message: err.message,
              };

              return `(${i}) ${info.uid}${info.id}${info.path}: ${info.message}`;
            })
            .forEach((message) => log.warn(message));
        } finally {
          callback();
        }
      },
    });
  }
}