<?php

namespace Autogedal\AutogedalHelper\Ui\Component\Listing\Column;

use Magento\Ui\Component\Listing\Columns\Column;

class BundleSku extends Column
{
    public function prepareDataSource(array $dataSource)
    {
        if (isset($dataSource['data']['items'])) {
            foreach ($dataSource['data']['items'] as &$item) {
                $item[$this->getData('name')] = isset($item['bundle_sku']) ? $this->extractParentBundleSku($item['bundle_sku']) : '';
            }
        }
        return $dataSource;
    }

    public function extractParentBundleSku(string $sku): string
    {
        $parts = explode('-', $sku);
        return $parts[0] ?? $sku;
    }
}