/**
* @param \Magento\Catalog\Model\Product $product
* @return \Magento\Catalog\Model\Product
*/
public function afterLoad(\Magento\Catalog\Model\Product $product)
{
if ($product->getTypeId() != \Magento\Downloadable\Model\Product\Type::TYPE_DOWNLOADABLE) {
return $product;
}
$productExtension = $product->getExtensionAttributes();
if ($productExtension === null) {
$productExtension = $this->productExtensionFactory->create();
}
$links = $this->linkRepository->getLinksByProduct($product);
if ($links !== null) {
$productExtension->setDownloadableProductLinks($links);
}
$samples = $this->linkRepository->getSamplesByProduct($product);
if ($samples !== null) {
$productExtension->setDownloadableProductSamples($samples);
}
$product->setExtensionAttributes($productExtension);
return $product;
}
