NerdsAndCompany\Schematic\Services\AssetSources

Complexity

Total Complexity

15

Size/Duplication

Total Lines	138
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	1

Test Coverage

Coverage

72.72%

Importance

Changes

0

6 Methods

Rating	Name	Duplication	Size	Complexity
A	getAssetSourcesService()	0	4	1
A	export()	0	12	2
A	getAssetSourceDefinition()	0	12	1
A	resetCraftAssetSourcesServiceCache()	0	15	3
B	import()	0	31	6
A	populateAssetSource()	0	21	2

<?php

namespace NerdsAndCompany\Schematic\Services;

use Craft\Craft;
use Craft\AssetSourceModel;

/**
 * Schematic Assets Service.
 *
 * Sync Craft Setups.
 *
 * @author    Nerds & Company
 * @copyright Copyright (c) 2015-2017, Nerds & Company
 * @license   MIT
 *
 * @link      http://www.nerds.company
 */
class AssetSources extends Base
{
    /**
     * @return AssetSourcesService
     */
    private function getAssetSourcesService()
    {
        return Craft::app()->assetSources;
    }

    /**
     * Export all asset sources.
     *
     * @param AssetSourceModel[] $assetSources
     *
     * @return array
     */
    public function export(array $assetSources = [])
    {
        Craft::log(Craft::t('Exporting Asset Sources'));

        $assetSourceDefinitions = [];

        foreach ($assetSources as $assetSource) {
            $assetSourceDefinitions[$assetSource->handle] = $this->getAssetSourceDefinition($assetSource);
        }

        return $assetSourceDefinitions;
    }

    /**
     * @param AssetSourceModel $assetSource
     *
     * @return array
     */
    private function getAssetSourceDefinition(AssetSourceModel $assetSource)
    {
        $fieldLayout = Craft::app()->fields->getLayoutById($assetSource->fieldLayoutId);

        return [
            'type' => $assetSource->type,
            'name' => $assetSource->name,
            'sortOrder' => $assetSource->sortOrder,
            'settings' => $assetSource->settings,
            'fieldLayout' => Craft::app()->schematic_fields->getFieldLayoutDefinition($fieldLayout),
        ];
    }

    /**
     * Import asset source definitions.
     *
     * @param array $assetSourceDefinitions
     * @param bool  $force
     *
     * @return Result
     */
    public function import(array $assetSourceDefinitions, $force = false)
    {
        Craft::log(Craft::t('Importing Asset Sources'));

        $this->resetCraftAssetSourcesServiceCache();
        $assetSources = $this->getAssetSourcesService()->getAllSources('handle');

        foreach ($assetSourceDefinitions as $assetSourceHandle => $assetSourceDefinition) {
            $assetSource = array_key_exists($assetSourceHandle, $assetSources)
                ? $assetSources[$assetSourceHandle]
                : new AssetSourceModel();

            unset($assetSources[$assetSourceHandle]);

            $this->populateAssetSource($assetSource, $assetSourceDefinition, $assetSourceHandle);

            if (!$this->getAssetSourcesService()->saveSource($assetSource)) { // Save assetsource via craft
                $this->addErrors($assetSource->getAllErrors());

                continue;
            }
        }

        if ($force) {
            foreach ($assetSources as $assetSource) {
                $this->getAssetSourcesService()->deleteSourceById($assetSource->id);
            }
        }

        return $this->getResultModel();
    }

    /**
     * Populate asset source.
     *
     * @param AssetSourceModel $assetSource
     * @param array            $assetSourceDefinition
     * @param string           $assetSourceHandle
     *
     * @return AssetSourceModel
     */
    private function populateAssetSource(AssetSourceModel $assetSource, array $assetSourceDefinition, $assetSourceHandle)
    {
        $defaultAssetSourceSettings = array(
            'publicURLs' => true,
        );

        $assetSource->setAttributes([
            'handle' => $assetSourceHandle,
            'type' => $assetSourceDefinition['type'],
            'name' => $assetSourceDefinition['name'],
            'sortOrder' => $assetSourceDefinition['sortOrder'],
            'settings' => array_merge($defaultAssetSourceSettings, $assetSourceDefinition['settings']),
        ]);

        if (array_key_exists('fieldLayout', $assetSourceDefinition)) {
            $fieldLayout = Craft::app()->schematic_fields->getFieldLayout($assetSourceDefinition['fieldLayout']);
            $assetSource->setFieldLayout($fieldLayout);
        }

        return $assetSource;
    }

    /**
     * Reset craft fields service cache using reflection.
     */
    private function resetCraftAssetSourcesServiceCache()
    {
        $obj = $this->getAssetSourcesService();
        $refObject = new \ReflectionObject($obj);
        if ($refObject->hasProperty('_fetchedAllSources')) {
            $refProperty = $refObject->getProperty('_fetchedAllSources');
            $refProperty->setAccessible(true);
            $refProperty->setValue($obj, false);
        }
        if ($refObject->hasProperty('_sourcesById')) {
            $refProperty = $refObject->getProperty('_sourcesById');
            $refProperty->setAccessible(true);
            $refProperty->setValue($obj, array());
        }
    }
}