MTC-9767 add company segement to company import

Assets/js/company_import.js

@@ -0,0 +1,21 @@
+(function($) {
+    'use strict';
+
+    function reorderCompanySegmentsField() {
+        if (!$('form[name="lead_field_import"]').length) return;
+
+        var $segments = $('#lead_field_import_company_segments').closest('.col-xs-4, .col-sm-4');
+        var $owner = $('#lead_field_import_owner').closest('.col-xs-4, .col-sm-4');
+
+        if ($segments.length && $owner.length) {
+            $owner.after($segments);
+        }
+    }
+
+    $(function() {
+        reorderCompanySegmentsField();
+    });
+    $(document).ajaxComplete(function(event, xhr, settings) {
+        reorderCompanySegmentsField();
+    });
+})(mQuery);

EventListener/ImportCompanySegmentSubscriber.php

@@ -0,0 +1,148 @@
+<?php
+
+declare(strict_types=1);
+
+namespace MauticPlugin\LeuchtfeuerCompanySegmentsBundle\EventListener;
+
+use Mautic\LeadBundle\Event\CompanyEvent;
+use Mautic\LeadBundle\Event\ImportEvent;
+use Mautic\LeadBundle\Event\ImportProcessEvent;
+use Mautic\LeadBundle\Event\ImportValidateEvent;
+use Mautic\LeadBundle\LeadEvents;
+use Mautic\LeadBundle\Model\ImportModel;
+use MauticPlugin\LeuchtfeuerCompanySegmentsBundle\Entity\CompanySegment;
+use MauticPlugin\LeuchtfeuerCompanySegmentsBundle\Integration\Config;
+use MauticPlugin\LeuchtfeuerCompanySegmentsBundle\Model\CompanySegmentModel;
+use Symfony\Component\EventDispatcher\EventSubscriberInterface;
+use Symfony\Component\HttpFoundation\RequestStack;
+
+final class ImportCompanySegmentSubscriber implements EventSubscriberInterface
+{
+    /**
+     * @var int|null Tracks the active import ID to determine if we're in import context
+     */
+    private ?int $activeImportId = null;
+
+    public function __construct(
+        private CompanySegmentModel $companySegmentModel,
+        private ImportModel $importModel,
+        private RequestStack $requestStack,
+        private Config $config,
+    ) {
+    }
+
+    public static function getSubscribedEvents(): array
+    {
+        return [
+            LeadEvents::IMPORT_ON_VALIDATE      => ['onValidateImport', 10],
+            LeadEvents::IMPORT_POST_SAVE        => ['onImportPostSave', 0],
+            LeadEvents::IMPORT_ON_PROCESS       => ['onImportProcess', 10],
+            LeadEvents::COMPANY_POST_SAVE       => ['onCompanyPostSave', -10],
+        ];
+    }
+
+    public function onValidateImport(ImportValidateEvent $event): void
+    {
+        if (!$this->config->isPublished()) {
+            return;
+        }
+
+        if (!$event->importIsForRouteObject('companies')) {
+            return;
+        }
+
+        $form = $event->getForm();
+        if (!$form->has('company_segments')) {
+            return;
+        }
+
+        $companySegments = $form->get('company_segments')->getData();
+        if (!is_array($companySegments) || [] === $companySegments) {
+            return;
+        }
+
+        $segmentIds = array_map(
+            fn ($segment): ?int => $segment instanceof CompanySegment ? $segment->getId() : (int) $segment,
+            $companySegments
+        );
+
+        $request = $this->requestStack->getCurrentRequest();
+        if (null !== $request && true === $request->hasSession()) {
+            $request->getSession()->set('mautic.company.import.segments.temp', $segmentIds);
+        }
+    }
+
+    public function onImportPostSave(ImportEvent $event): void
+    {
+        if (!$this->config->isPublished()) {
+            return;
+        }
+
+        $import = $event->getEntity();
+        if ('company' !== $import->getObject()) {
+            return;
+        }
+
+        $request = $this->requestStack->getCurrentRequest();
+        if (null === $request || false === $request->hasSession()) {
+            return;
+        }
+
+        $session  = $request->getSession();
+        $segments = $session->get('mautic.company.import.segments.temp');
+
+        if (!is_array($segments) || [] === $segments) {
+            return;
+        }
+
+        $session->remove('mautic.company.import.segments.temp');
+        $import->mergeToProperties(['company_segments' => $segments]);
+        $this->importModel->saveEntity($import, false);
+    }
+
+    public function onImportProcess(ImportProcessEvent $event): void
+    {
+        if (!$this->config->isPublished()) {
+            return;
+        }
+
+        if ($event->importIsForObject('company')) {
+            $this->activeImportId = $event->import->getId();
+        }
+    }
+
+    public function onCompanyPostSave(CompanyEvent $event): void
+    {
+        if (!$this->config->isPublished()) {
+            return;
+        }
+
+        if (null === $this->activeImportId) {
+            return;
+        }
+
+        $company = $event->getCompany();
+
+        $import = $this->importModel->getEntity($this->activeImportId);
+        if (null === $import || 'company' !== $import->getObject()) {
+            return;
+        }
+
+        $properties = $import->getProperties();
+        if (!isset($properties['company_segments']) || !is_array($properties['company_segments'])) {
+            return;
+        }
+
+        /** @var array<int|string> $segmentIds */
+        $segmentIds = array_values(array_filter(
+            $properties['company_segments'],
+            fn ($value): bool => is_int($value) || is_string($value)
+        ));
+
+        if ([] === $segmentIds) {
+            return;
+        }
+
+        $this->companySegmentModel->addCompany($company, $segmentIds, true);
+    }
+}

Form/Extension/LeadImportFieldTypeExtension.php

@@ -0,0 +1,67 @@
+<?php
+
+declare(strict_types=1);
+
+namespace MauticPlugin\LeuchtfeuerCompanySegmentsBundle\Form\Extension;
+
+use Mautic\LeadBundle\Form\Type\LeadImportFieldType;
+use MauticPlugin\LeuchtfeuerCompanySegmentsBundle\Form\Type\CompanySegmentListType;
+use MauticPlugin\LeuchtfeuerCompanySegmentsBundle\Integration\Config;
+use Symfony\Component\Form\AbstractTypeExtension;
+use Symfony\Component\Form\FormBuilderInterface;
+use Symfony\Component\OptionsResolver\OptionsResolver;
+use Symfony\Contracts\Translation\TranslatorInterface;
+
+/**
+ * Extends LeadImportFieldType to add company segments field for company imports.
+ */
+class LeadImportFieldTypeExtension extends AbstractTypeExtension
+{
+    public function __construct(
+        private TranslatorInterface $translator,
+        private Config $config,
+    ) {
+    }
+
+    /** @phpstan-ignore-next-line */
+    public function buildForm(FormBuilderInterface $builder, array $options): void
+    {
+        if (!$this->config->isPublished()) {
+            return;
+        }
+
+        // Only add field for company imports
+        if (!isset($options['object']) || 'company' !== $options['object']) {
+            return;
+        }
+
+        $builder->add(
+            'company_segments',
+            CompanySegmentListType::class,
+            [
+                'label'      => 'mautic.company_segments.form.import.segments',
+                'label_attr' => ['class' => 'control-label'],
+                'attr'       => [
+                    'class'                        => 'form-control',
+                    'data-placeholder'             => $this->translator->trans('mautic.company_segments.form.import.segments.placeholder'),
+                    'data-company-segments-import' => 'true',
+                ],
+                'required' => false,
+                'multiple' => true,
+                'mapped'   => false,
+            ]
+        );
+    }
+
+    public function configureOptions(OptionsResolver $resolver): void
+    {
+        // Ensure the 'object' option is available
+        $resolver->setDefined(['object']);
+    }
+
+    public static function getExtendedTypes(): iterable
+    {
+        // Specify which form type this extension applies to
+        return [LeadImportFieldType::class];
+    }
+}