craft-migration-assistant/src/extensions/LinkFieldExtension.php at craft-5 · dgrigg/craft-migration-assistant · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
<?php

namespace dgrigg\migrationassistant\extensions;

use Craft;
use yii\base\Event;
use dgrigg\migrationassistant\events\ExportEvent;
use dgrigg\migrationassistant\events\ImportEvent;
use dgrigg\migrationassistant\services\BaseMigration;
use dgrigg\migrationassistant\services\BaseContentMigration;
use dgrigg\migrationassistant\helpers\ElementHelper;


class LinkFieldExtension
{

    function __construct(){

        Event::on(BaseMigration::class, BaseMigration::EVENT_BEFORE_EXPORT_FIELD_VALUE, function (ExportEvent $event) {
            $element = $event->element;

            if ($element->className() == 'lenz\linkfield\fields\LinkField') {
                $value = $event->value;
                $linkType = $value->getLinkType();
                $value = (object)(array) $value;
                $value->type = $linkType->name;

                if (isset($value->linkedId) && !is_null($value->linkedId)) {
                    $linkedElement = Craft::$app->elements->getElementById($value->linkedId, $linkType->elementType, $value->linkedSiteId );
                    $value->element = [ElementHelper::getSourceHandle($linkedElement)];
                }

                if (isset($value->linkedSiteId) && !is_null($value->linkedSiteId)) {
                    $site = Craft::$app->sites->getSiteById($value->linkedSiteId);
                    $value->site = $site->handle;
                }

                unset($value->linkedId);
                unset($value->linkedSiteId);
                $event->value = $value;
            }
        });

        Event::on(BaseContentMigration::class, BaseMigration::EVENT_BEFORE_IMPORT_FIELD_VALUE, function (ImportEvent $event) {
            $element = $event->element;
            $value = $event->value;

            if ($element->className() == 'lenz\linkfield\fields\LinkField') {
                if (key_exists('element', $value)){
                    $ids =  ElementHelper::populateIds($value['element']);
                    if (count($ids) > 0){
                        $value['linkedId'] = $ids[0];
                        unset($value['element']);

                    }
                }

                if (key_exists('site', $value)){
                    $site = Craft::$app->sites->getSiteByHandle($value['site']);
                    if ($site){
                        $value['linkedSiteId'] = $site->id;
                        unset($value['site']);
                    }
                }

                $event->value = $value;
            }
        });
    }

}