9.5/_additional_column_service_8php_source.html

<?php

namespace ‪TYPO3\CMS\Workspaces\Service;


/*

 * This file is part of the TYPO3 CMS project.

 *

 * It is free software; you can redistribute it and/or modify it under

 * the terms of the GNU General Public License, either version 2

 * of the License, or any later version.

 *

 * For the full copyright and license information, please read the

 * LICENSE.txt file that was distributed with this source code.

 *

 * The TYPO3 project - inspiring people to share!

 */


use ‪TYPO3\CMS\Core\SingletonInterface;

use ‪TYPO3\CMS\Core\Utility\GeneralUtility;

use ‪TYPO3\CMS\Extbase\Object\ObjectManager;

use ‪TYPO3\CMS\Workspaces\ColumnDataProviderInterface;

use ‪TYPO3\CMS\Workspaces\Domain\Model\CombinedRecord;


class ‪AdditionalColumnService implements ‪SingletonInterface

{

    protected ‪$columns = [];


    public static function ‪getInstance()

    {

        return ‪self::getObjectManager()->‪get(AdditionalColumnService::class);

    }


    public static function ‪getObjectManager()

    {

        return GeneralUtility::makeInstance(ObjectManager::class);

    }


    public function register($columnName, $dataProviderClassOrObject)

    {

        if (is_object($dataProviderClassOrObject)) {

            $dataProvider = $dataProviderClassOrObject;

        } else {

            $dataProvider = GeneralUtility::makeInstance($dataProviderClassOrObject);

        }


        if (!$dataProvider instanceof ‪ColumnDataProviderInterface) {

            throw new \RuntimeException('Data provider needs to implement ColumnDataProviderInterface', 1374309323);

        }


        $this->columns[$columnName] = $dataProvider;

    }


    public function ‪getDefinition()

    {

        $columnSettings = [];

        foreach ($this->columns as $columnName => $dataProvider) {

            $definition = $dataProvider->getDefinition();


            if (!is_array($definition)) {

                $definition = [];

            }


            $definition['name'] = $columnName;

            $columnSettings[] = $definition;

        }

        return $columnSettings;

    }


    public function ‪getHandler()

    {

        $columnSettings = [];

        foreach ($this->columns as $columnName => $_) {

            $columnSettings[] = 'TYPO3.Workspaces.extension.AdditionalColumn.' . $columnName;

        }

        return $columnSettings;

    }


    public function ‪getData(‪CombinedRecord $combinedRecord)

    {

        $recordData = [];

        foreach ($this->columns as $columnName => $dataProvider) {

            $data = $dataProvider->getData($combinedRecord);


            if ($data !== null) {

                $recordData[$columnName] = $data;

            }

        }

        return $recordData;

    }

}