ltsecurityinc安防产品在线商城代码

app/code/Plumrocket/Base/Setup/AbstractUninstall.php

@@ -0,0 +1,280 @@
+<?php
+/**
+ * @package     Plumrocket_Base
+ * @copyright   Copyright (c) 2020 Plumrocket Inc. (https://plumrocket.com)
+ * @license     https://plumrocket.com/license/  End-user License Agreement
+ */
+
+namespace Plumrocket\Base\Setup;
+
+use Magento\Cms\Model\BlockFactory;
+use Magento\Eav\Setup\EavSetupFactory;
+use Magento\Framework\App\DeploymentConfig;
+use Magento\Framework\Config\ConfigOptionsListConstants;
+use Magento\Framework\Config\File\ConfigFilePool;
+use Magento\Framework\Module\ModuleList\Loader;
+use Magento\Framework\Module\Setup\Context as SetupContext;
+use Magento\Framework\Setup\ModuleContextInterface;
+use Magento\Framework\Setup\SchemaSetupInterface;
+use Magento\Framework\Setup\UninstallInterface;
+use Magento\Framework\ShellInterface;
+use Psr\Log\LoggerInterface;
+
+abstract class AbstractUninstall implements UninstallInterface
+{
+    /**
+     * @var string
+     */
+    protected $_configSectionId;
+
+    /**
+     * @var array
+     */
+    protected $_attributes = [];
+
+    /**
+     * @var array
+     */
+    protected $_tables = [];
+
+    /**
+     * List of entity types to uninstall
+     *
+     * @var int[]|string[]
+     */
+    protected $eavEntityTypes = [];
+
+    /**
+     * @var array
+     */
+    protected $_tablesFields = [];
+
+    /**
+     * @var array
+     */
+    protected $_pathes = [];
+
+    /**
+     * @var array
+     */
+    protected $_cmsBlocks = [];
+
+    /**
+     * @var SetupContext
+     */
+    protected $context;
+
+    /**
+     * @var \Magento\Eav\Setup\EavSetupFactory
+     */
+    protected $eavSetupFactory;
+
+    /**
+     * @var \Magento\Cms\Model\BlockFactory
+     */
+    protected $_cmsBlockFactory;
+
+    /**
+     * @var \Magento\Framework\ShellInterface
+     */
+    private $shell;
+
+    /**
+     * @var \Magento\Framework\App\DeploymentConfig\Writer
+     */
+    private $deploymentConfigWriter;
+
+    /**
+     * @var \Magento\Framework\App\DeploymentConfig
+     */
+    private $deploymentConfig;
+
+    /**
+     * @var \Magento\Framework\Module\ModuleList\Loader
+     */
+    private $moduleListLoader;
+
+    /**
+     * @var \Psr\Log\LoggerInterface
+     */
+    private $logger;
+
+    /**
+     * AbstractUninstall constructor.
+     *
+     * @param \Magento\Framework\Module\Setup\Context        $context
+     * @param \Magento\Eav\Setup\EavSetupFactory             $eavSetupFactory
+     * @param \Magento\Cms\Model\BlockFactory                $cmsBlockFactory
+     * @param \Magento\Framework\ShellInterface              $shell
+     * @param \Magento\Framework\App\DeploymentConfig        $deploymentConfig
+     * @param \Magento\Framework\App\DeploymentConfig\Writer $deploymentConfigWriter
+     * @param \Magento\Framework\Module\ModuleList\Loader    $moduleListLoader
+     * @param \Psr\Log\LoggerInterface                       $logger
+     */
+    public function __construct(
+        SetupContext $context,
+        EavSetupFactory $eavSetupFactory,
+        BlockFactory $cmsBlockFactory,
+        ShellInterface $shell,
+        DeploymentConfig $deploymentConfig,
+        DeploymentConfig\Writer $deploymentConfigWriter,
+        Loader $moduleListLoader,
+        LoggerInterface $logger
+    ) {
+        $this->context          = $context;
+        $this->eavSetupFactory  = $eavSetupFactory;
+        $this->_cmsBlockFactory = $cmsBlockFactory;
+        $this->shell = $shell;
+        $this->deploymentConfig = $deploymentConfig;
+        $this->deploymentConfigWriter = $deploymentConfigWriter;
+        $this->moduleListLoader = $moduleListLoader;
+        $this->logger = $logger;
+    }
+
+    /**
+     * Uninstall
+     *
+     * @param SchemaSetupInterface   $setup
+     * @param ModuleContextInterface $context
+     * @return void
+     * @throws \Magento\Framework\Exception\LocalizedException
+     */
+    public function uninstall(SchemaSetupInterface $setup, ModuleContextInterface $context)
+    {
+        $setup->startSetup();
+
+        $moduleName = $this->getModuleName();
+        $dataSetup = new \Magento\Setup\Module\DataSetup($this->context);
+        /** @var \Magento\Eav\Setup\EavSetup $eavSetup */
+        $eavSetup = $this->eavSetupFactory->create(['setup' => $dataSetup]);
+
+        //remove attribute
+        foreach ($this->_attributes as $entityTypeName => $attributeNames) {
+            $entityTypeId = $eavSetup->getEntityTypeId($entityTypeName);
+            foreach ($attributeNames as $attributeName) {
+                $eavSetup->removeAttribute($entityTypeId, $attributeName);
+            }
+        }
+
+        //remove tables
+        foreach ($this->_tables as $_tableName) {
+            $_tableName = $setup->getTable($_tableName);
+            $setup->getConnection()->dropTable($_tableName);
+        }
+
+        //remove Eav Entity Type
+        foreach ($this->eavEntityTypes as $eavEntityType) {
+            $eavSetup->removeEntityType($eavEntityType);
+        }
+
+        //remove tables fields
+        foreach ($this->_tablesFields as $_tableName => $_fields) {
+            $_tableName = $setup->getTable($_tableName);
+            foreach ($_fields as $_field) {
+                try {
+                    $setup->getConnection()->dropColumn($_tableName, $_field);
+                } catch (\Exception $e) {
+                    $this->logger->critical($e->getMessage());
+                }
+            }
+        }
+
+        $this->removeStaticBlocks();
+
+        $this->removeConfigData($setup, $moduleName);
+
+        $this->removeModuleFromConfigFile($moduleName);
+
+        $this->shell->execute('rm -rf ' . BP . '/var/cache');
+
+        foreach ($this->_pathes as $path) {
+            $this->shell->execute('rm -rf ' . BP . $path);
+        }
+
+        $setup->endSetup();
+    }
+
+    /**
+     * Get module name.
+     *
+     * @return string
+     */
+    public function getModuleName()
+    {
+        $className = get_class($this);
+        $namespace = substr(
+            $className,
+            0,
+            strpos($className, '\\' . 'Setup')
+        );
+
+        return str_replace('\\', '_', $namespace);
+    }
+
+    /**
+     * Remove static blocks.
+     *
+     * @return void
+     * @throws \Exception
+     */
+    private function removeStaticBlocks(): void
+    {
+        foreach ($this->_cmsBlocks as $_identifier) {
+            /** @var \Magento\Cms\Model\Block $cmsBlock */
+            $cmsBlock = $this->_cmsBlockFactory->create();
+            $cmsBlock->load($_identifier);
+            if ($cmsBlock->getId()) {
+                $cmsBlock->delete();
+            }
+        }
+    }
+
+    /**
+     * Remove configs.
+     *
+     * @param \Magento\Framework\Setup\SchemaSetupInterface $setup
+     * @param string                                        $moduleName
+     * @return void
+     */
+    private function removeConfigData(SchemaSetupInterface $setup, string $moduleName): void
+    {
+        if ($this->_configSectionId) {
+            $setup->getConnection()->delete(
+                $setup->getTable('core_config_data'),
+                ['path LIKE ?' => $this->_configSectionId . '/%']
+            );
+
+            $setup->getConnection()->delete(
+                $setup->getTable('setup_module'),
+                ['module = ?' => $moduleName]
+            );
+        }
+    }
+
+    /**
+     * Remove from config.php
+     *
+     * @param string $moduleName
+     * @return void
+     * @throws \Magento\Framework\Exception\FileSystemException
+     * @throws \Magento\Framework\Exception\LocalizedException
+     */
+    private function removeModuleFromConfigFile(string $moduleName): void
+    {
+        $configuredModules = $this->deploymentConfig->getConfigData(
+            ConfigOptionsListConstants::KEY_MODULES
+        );
+        $existingModules = $this->moduleListLoader->load([$moduleName]);
+        $newModules = [];
+        foreach (array_keys($existingModules) as $module) {
+            $newModules[$module] = $configuredModules[$module] ?? 0;
+        }
+
+        $this->deploymentConfigWriter->saveConfig(
+            [
+                ConfigFilePool::APP_CONFIG => [ConfigOptionsListConstants::KEY_MODULES => $newModules]
+            ],
+            true
+        );
+    }
+}