Sunnysideup\UUDI\Extensions\UUIDExtension

Complexity

Total Complexity

18

Size/Duplication

Total Lines	100
Duplicated Lines	0 %

Importance

Changes	14
Bugs	0	Features	0

7 Methods

Rating	Name	Duplication	Size	Complexity
A	updateSettingsFields()	0	5	2
A	updateCMSFieldsForHashId()	0	23	3
A	getHashID()	0	8	2
A	updateCMSFields()	0	5	2
A	calculatePublicUUID()	0	3	1
A	onAfterWrite()	0	6	3
A	onBeforeWrite()	0	12	5

<?php

namespace Sunnysideup\UUDI\Extensions;

use SilverStripe\CMS\Model\SiteTree;

The type SilverStripe\CMS\Model\SiteTree was not found. Maybe you did not declare it correctly or list all dependencies?

use SilverStripe\Forms\CheckboxField;
use SilverStripe\Forms\FieldList;
use SilverStripe\Forms\ReadonlyField;
use SilverStripe\ORM\DataExtension;
use SilverStripe\ORM\DataObject;
use Sunnysideup\UUDI\Api\HashCreator;

/**
 * Class \Sunnysideup\UUDI\Extensions\UUIDExtension.
 *
 * @property string $UUID
 * @property string $PublicUUID
 */
class UUIDExtension extends DataExtension
{
    private static $db = [
        'RequiresUUID' => 'Boolean(0)', //32 + 1 + 32
        'UUID' => 'Varchar(65)', //32 + 1 + 32
        'PublicUUID' => 'Varchar(12)', //32 + 1 + 32
    ];

    private static $indexes = [
        'RequiresUUID' => true,
        'UUID' => false,
        'PublicUUID' => true,
    ];

    private $UUIDNeverAgainRaceCondition = false;

    public function onBeforeWrite()
    {
        $owner = $this->getOwner();
        if ($owner->RequiresUUID) {
            if(!$owner->UUID) {
                $owner->UUID = $this->getHashID();
            }
        } else {
            $owner->UUID = '';
        }
        if (! $owner->PublicUUID || 'ERROR' === $owner->PublicUUID) {
            $owner->PublicUUID = $this->calculatePublicUUID();
        }
    }

    public function onAfterWrite()
    {
        $owner = $this->getOwner();
        if (! $owner->UUID && false === ! $this->UUIDNeverAgainRaceCondition) {
            $this->UUIDNeverAgainRaceCondition = true;
            $owner->write();
        }
    }

    public function calculatePublicUUID(): string
    {
        return HashCreator::generate_hash(12);
    }

    public function updateCMSFields(FieldList $fields)
    {
        $owner = $this->owner;
        if (! ($owner instanceof SiteTree)) {
            $this->updateCMSFieldsForHashId($fields);
        }
    }

    public function updateSettingsFields(FieldList $fields)
    {
        $owner = $this->owner;
        if ($owner instanceof SiteTree) {
            $this->updateCMSFieldsForHashId($fields);
        }
    }

    public function updateCMSFieldsForHashId(FieldList $fields)
    {
        /** @var DataObject $owner */
        $owner = $this->owner;
        $fields->removeByName(
            [
                'RequiresUUID',
                'UUID',
                'PublicUUID',
            ]
        );
        if ($owner->hasMethod('ShowUUIDInCMS')) {
            $tab = 'Root.UUID';
            if ($owner->hasMethod('UUIDTabInCMS')) {
                $tab = $owner->UUIDTabInCMS();
            }

            $fields->addFieldsToTab(
                $tab,
                [
                    // ReadonlyField::create('MyUUID', 'Private UUID', $owner->UUID),
                    CheckboxField::create('RequiresUUID', 'Has UUID', $owner->PublicUUID)->performDisabledTransformation,
                    ReadonlyField::create('MyPublicUUID', 'Public UUID', $owner->PublicUUID),
                ]
            );
        }
    }

    /**
     * Gets a truly unique identifier to the classname and ID.
     */
    protected function getHashID(): ?string
    {
        $owner = $this->getOwner();
        if ($owner->ID) {
            return HashCreator::create_hash_id($owner->ClassName, $owner->ID) . '_' . HashCreator::generate_hash(32);
        }

        return '';
    }
}