shopware / platform

// eslint-disable-next-line sw-deprecation-rules/private-feature-declarations

export default function initializeRepositoryFactory(container: InitContainer) {

    const httpClient = container.httpClient;

    const factoryContainer = Shopware.Application.getContainer('factory');

    const serviceContainer = Shopware.Application.getContainer('service');

    return httpClient.get('_info/entity-schema.json', {

        headers: {

            Authorization: `Bearer ${serviceContainer.loginService.getToken()}`,

        },

    }).then(({ data }) => {

        const entityDefinitionFactory = factoryContainer.entityDefinition;

        const customEntityDefinitionService = serviceContainer.customEntityDefinitionService;

        const cmsPageTypeService = serviceContainer.cmsPageTypeService;

        let hasCmsAwareDefinitions = false;

        // eslint-disable-next-line @typescript-eslint/no-unsafe-argument

        Object.entries(data).forEach(([key, value]) => {

            entityDefinitionFactory.add(key, value);

            if (key.startsWith('custom_entity_') || key.startsWith('ce_')) {

                // @ts-expect-error - value is defined

                customEntityDefinitionService.addDefinition(value);

                // @ts-expect-error - value is defined

                // eslint-disable-next-line @typescript-eslint/no-unsafe-member-access

                hasCmsAwareDefinitions = hasCmsAwareDefinitions || !!value?.flags?.['cms-aware'];

            }

        });

        if (hasCmsAwareDefinitions) {

            customEntityTypes.forEach((customEntityType) => {

                cmsPageTypeService.register(customEntityType);

            });

        }

        const hydrator = new EntityHydrator();

        const changesetGenerator = new ChangesetGenerator();

        const entityFactory = new EntityFactory();

        const errorResolver = new ErrorResolverError();

        Shopware.Application.addServiceProvider('repositoryFactory', () => {

            return new RepositoryFactory(

                hydrator,

                changesetGenerator,

                entityFactory,

                httpClient,

                errorResolver,

            );

        });

        Shopware.Application.addServiceProvider('entityHydrator', () => {

            return hydrator;

        });

        Shopware.Application.addServiceProvider('entityFactory', () => {

            return entityFactory;