course-tencent-cloud/app/Console/Tasks/CleanDemoDataTask.php

<?php
/**
 * @copyright Copyright (c) 2021 深圳市酷瓜软件有限公司
 * @license https://opensource.org/licenses/GPL-2.0
 * @link https://www.koogua.com
 */

namespace App\Console\Tasks;

use App\Caches\CategoryAllList as CategoryAllListCache;
use App\Caches\CategoryList as CategoryListCache;
use App\Caches\CategoryTreeList as CategoryTreeListCache;
use App\Caches\IndexSlideList as IndexSlideListCache;
use App\Models\Account as AccountModel;
use App\Models\Category as CategoryModel;
use App\Repos\User as UserRepo;
use App\Services\Utils\IndexPageCache as IndexPageCacheUtil;

class CleanDemoDataTask extends Task
{

    public function mainAction()
    {
        if ($this->isDemoEnv()) {

            $this->truncateTables();
            $this->createRootUser();
            $this->cleanSearchIndex();
            $this->cleanCache();

        } else {

            echo '------ access denied ------' . PHP_EOL;
        }
    }

    protected function truncateTables()
    {
        echo '------ start truncate tables ------' . PHP_EOL;

        $excludeTables = [
            'kg_area', 'kg_migration', 'kg_migration_task', 'kg_nav',
            'kg_page', 'kg_role', 'kg_setting', 'kg_vip',
        ];

        $tables = $this->db->listTables();

        foreach ($tables as $table) {
            if (!in_array($table, $excludeTables)) {
                $this->db->execute("TRUNCATE TABLE {$table}");
            }
        }

        echo '------ end truncate tables ------' . PHP_EOL;
    }

    protected function createRootUser()
    {
        echo '------ start create root user ------' . PHP_EOL;

        $account = new AccountModel();

        $account->create([
            'id' => 10000,
            'email' => '10000@163.com',
            'password' => '1a1e4568f1a3740b8853a8a16e29bc87',
            'salt' => 'MbZWxN3L',
            'create_time' => time(),
        ]);

        $userRepo = new UserRepo();

        $user = $userRepo->findById($account->id);

        $user->update([
            'admin_role' => 1,
            'edu_role' => 2,
        ]);

        echo '------ end create root user ------' . PHP_EOL;
    }

    protected function cleanCache()
    {
        $util = new IndexPageCacheUtil();
        $util->rebuild();

        $slideListCache = new IndexSlideListCache();
        $slideListCache->rebuild();

        $categoryListCache = new CategoryListCache();
        $categoryAllListCache = new CategoryAllListCache();
        $categoryTreeListCache = new CategoryTreeListCache();

        foreach (CategoryModel::types() as $key => $value) {
            $categoryListCache->rebuild($key);
            $categoryAllListCache->rebuild($key);
            $categoryTreeListCache->rebuild($key);
        }
    }

    protected function cleanSearchIndex()
    {
        $articleIndexTask = new ArticleIndexTask();
        $articleIndexTask->cleanAction();

        $courseIndexTask = new CourseIndexTask();
        $courseIndexTask->cleanAction();

        $questionIndexTask = new QuestionIndexTask();
        $questionIndexTask->cleanAction();
    }

    protected function isDemoEnv()
    {
        $userRepo = new UserRepo();

        $user = $userRepo->findById(100015);

        return (bool)$user;
    }

}