add documentype generic viewmodel

Author: thomas-kl1

ViewModel/DocumentType.php

@@ -0,0 +1,74 @@
+<?php
+/**
+ * Copyright © OpenGento, All rights reserved.
+ * See LICENSE bundled with this library for license details.
+ */
+declare(strict_types=1);
+
+namespace Opengento\DocumentWidget\ViewModel;
+
+use Magento\Framework\Api\SearchCriteriaBuilder;
+use Magento\Framework\Exception\LocalizedException;
+use Magento\Framework\View\Element\Block\ArgumentInterface;
+use Opengento\Document\Api\Data\DocumentTypeInterface;
+use Opengento\Document\Api\DocumentTypeRepositoryInterface;
+use Psr\Log\LoggerInterface;
+
+final class DocumentType implements ArgumentInterface
+{
+    /**
+     * @var DocumentTypeRepositoryInterface
+     */
+    private $docTypeRepository;
+
+    /**
+     * @var SearchCriteriaBuilder
+     */
+    private $criteriaBuilder;
+
+    /**
+     * @var LoggerInterface
+     */
+    private $logger;
+
+    /**
+     * @var DocumentTypeInterface[]
+     */
+    private $documentTypes;
+
+    /**
+     * @var string[]
+     */
+    private $visibilities;
+
+    public function __construct(
+        DocumentTypeRepositoryInterface $docTypeRepository,
+        SearchCriteriaBuilder $criteriaBuilder,
+        LoggerInterface $logger,
+        array $visibilities
+    ) {
+        $this->docTypeRepository = $docTypeRepository;
+        $this->criteriaBuilder = $criteriaBuilder;
+        $this->logger = $logger;
+        $this->visibilities = $visibilities;
+    }
+
+    /**
+     * @return DocumentTypeInterface[]
+     */
+    public function getList(): array
+    {
+        if (!$this->documentTypes) {
+            $this->criteriaBuilder->addFilter('visibility', $this->visibilities, 'in');
+            try {
+                $this->documentTypes = $this->docTypeRepository->getList($this->criteriaBuilder->create());
+            } catch (LocalizedException $e) {
+                $this->logger->error($e->getLogMessage(), $e->getTrace());
+
+                $this->documentTypes = [];
+            }
+        }
+
+        return $this->documentTypes->getItems();
+    }
+}

composer.json

@@ -23,9 +23,9 @@
         "roave/security-advisories": "dev-master"
     },
     "suggest": {
-        "opengento/module-document-product-link": "",
-        "opengento/module-document-product-search": "",
-        "opengento/module-document-search": ""
+        "opengento/module-document-product-link": "This module aims to help merchants to link their documents to products in Magento 2.",
+        "opengento/module-document-product-search": "This module aims to make documents searchable with product keywords in Magento 2.",
+        "opengento/module-document-search": "This module aims to make documents searchable for customers in Magento 2."
     },
     "license": [
         "MIT"

etc/frontend/di.xml

@@ -41,4 +41,11 @@
             </argument>
         </arguments>
     </type>
+    <virtualType name="Opengento\DocumentWidget\ViewModel\VisibleDocumentType" type="Opengento\DocumentWidget\ViewModel\DocumentType">
+        <arguments>
+            <argument name="visibilities" xsi:type="array">
+                <item name="public" xsi:type="const">Opengento\Document\Model\DocumentType\Visibility::VISIBILITY_PUBLIC</item>
+            </argument>
+        </arguments>
+    </virtualType>
 </config>