lsky-pro/application/index/controller/admin/System.php

<?php
/**
 * User: Wisp X
 * Date: 2018/9/27
 * Time: 上午10:31
 * Link: https://github.com/wisp-x
 */

namespace app\index\controller\admin;

use app\common\model\Config;
use think\Db;
use app\common\model\Images;
use think\Exception;

/**
 * 系统设置
 *
 * Class System
 * @package app\index\controller\admin
 */
class System extends Base
{
    public function initialize()
    {
        parent::initialize();
        $this->assign('configs', parent::getConfigs(['basics', 'upload', 'user', 'mail', 'audit', 'other']));
    }

    public function index()
    {
        if ($this->request->isPost()) {
            Db::startTrans();
            try {
                $data = $this->request->post();
                foreach ($data as $key => $value) {
                    Config::where('name', $key)->setField('value', $value);
                }
                Db::commit();
            } catch (Exception $e) {
                Db::rollback();
                $this->error($e->getMessage());
            }
            $this->success(lang('Saved successfully'));
        }
        // 命名规则
        $naming = \think\facade\Config::pull('naming');
        $this->assign('naming', $naming);
        return $this->fetch();
    }

    public function console()
    {
        $storage = Images::sum('size');
        $imagesCount = Images::count('*');
        $suspiciousImagesCount = Images::where('suspicious', 1)->count();
        $users_count = \app\common\model\Users::count('*');
        $today = Images::whereTime('create_time', 'today')->count('*');
        $yesterday = Images::whereTime('create_time', 'yesterday')->count('*');
        $month = Images::whereTime('create_time', 'month')->count('*');
        $tourists = Images::where('user_id', 0)->count();

        $this->assign([
            'storage' => format_size($storage, true), // 占用储存
            'images_num' => $imagesCount, // 图片数量
            'suspicious_images_num' => $suspiciousImagesCount, // 可疑图片
            'users_num' => $users_count, // 用户数量
            'today' => $today, // 今日上传
            'yesterday' => $yesterday, // 昨日上传
            'month' => $month, // 本月上传
            'tourists' => $tourists, // 游客上传
        ]);

        return $this->fetch();
    }

    public function testMail()
    {
        if ($this->request->isPost()) {
            $email = $this->request->post('email');
            $err = $this->sendMail($email, lang('Test'), lang('This is a test email!'));
            if (true !== $err) {
                $this->error($err);
            }
            $this->success(lang('Sent successfully'));
        }
    }

    /**
     * 更新系统
     */
    public function upgrade()
    {
        $upgrade = null;
        try {
            if (!class_exists('ZipArchive')) {
                throw new \Exception(lang('Cannot continue, please make sure ZipArrive is installed correctly'));
            }

            ignore_user_abort(true);
            set_time_limit(0);
            ini_set('max_execution_time', 0);
            ini_set('memory_limit', '256M');

            $upgrade = new \Upgrade(app()->getRootPath(), $this->getConfig('system_version'));
            $release = $upgrade->release(); // 获取最新版
            // 判断是否已经是最新版
            if ($upgrade->check($release->version)) {
                throw new \Exception(lang('The current system is the latest version'));
            }
            $upgradeFile = app()->getRuntimePath() . 'upgrade.zip';// 判断是否存在安装包
            $file = file_exists($upgradeFile) ? $upgradeFile : $upgrade->download($release->url);

            // 校验 MD5
            if (strtolower(md5_file($file)) !== strtolower($release->md5)) {
                throw new \Exception(lang('The installation package is corrupt. Please try again later'));
            }

            $dir = $upgrade->unzip($file, $upgrade->getWorkspace()); // 解压安装包到工作区目录
            $path = rtrim($dir . strtolower($release->path), '/') . '/'; // 新版本程序解压后的根目录
            $updateSql = $path . $release->sql; // 更新数据库结构 sql 文件路径
            if (!$sql = @file_get_contents($updateSql)) {
                throw new \Exception(lang('SQL file acquisition failed'));
            }

            // 创建安装锁文件
            if (!@fopen($path . 'application/install.lock', 'w')) {
                throw new \Exception(lang('Setup lock file creation failed'));
            }

            Db::startTrans();

            // 检测新增表字段
            if (!$tableFields = @include($path . 'config/table.php')) {
                throw new \Exception(lang('Failed to get table field configuration file'));
            }
            foreach ($tableFields as $table => $fields) {
                foreach ($fields as $field => $sql) {
                    $fetchFields = array_column(Db::query("DESCRIBE `{$table}`;"), 'Field');
                    if (!in_array($field, $fetchFields)) {
                        Db::execute($sql);
                    }
                }
            }

            // 执行 sql 导入
            if (is_file($path . $release->sql)) {
                $lines = file($path . $release->sql);
                $temp = '';
                foreach ($lines as &$line) {
                    $line = trim($line);
                    if (substr($line, 0, 2) == '--' || $line == '' || substr($line, 0, 2) == '/*') continue;
                    $temp .= $line;
                    if (substr($line, -1, 1) == ';') {
                        Db::execute($temp);
                        $temp = '';
                    }
                }
            }

            // 检测需要的删除文件或文件夹
            foreach ($release->removes as $key => $items) {
                foreach ($items as $item) {
                    $pathname = trim($upgrade->getRootPath(), '/') . '/' . trim($item, '/');
                    if ($key === 'folders') {
                        $upgrade->rmdir($pathname);
                    } else {
                        @unlink($pathname);
                    }
                }
            }

            // 复制文件夹
            $upgrade->copyDirs($path, $upgrade->getRootPath());

            Db::commit();
        } catch (\Exception $e) {
            Db::rollback();
            @$upgrade->rmdir($upgrade->getWorkspace());
            $this->error($e->getMessage());
        } catch (\PDOException $e) {
            Db::rollback();
            @$upgrade->rmdir($upgrade->getWorkspace());
            $this->error($e->getMessage());
        } catch (\HttpException $e) {
            Db::rollback();
            @$upgrade->rmdir($upgrade->getWorkspace());
            $this->error($e->getMessage());
        } catch (\Throwable $e) {
            Db::rollback();
            @$upgrade->rmdir($upgrade->getWorkspace());
            $this->error($e->getMessage());
        }
        $this->success(lang('Upgrade Complete'));
    }

    /**
     * 备份系统
     */
    public function backup()
    {
        $backup = 'backups/' . date('YmdHis') . '.zip';
        try {
            $upgrade = new \Upgrade(app()->getRootPath(), $this->getConfig('system_version'));
            $upgrade->backup($backup);
        } catch (\Exception $e) {
            $this->error($e->getMessage());
        } catch (\PDOException $e) {
            $this->error($e->getMessage());
        } catch (\HttpException $e) {
            $this->error($e->getMessage());
        } catch (\Throwable $e) {
            $this->error($e->getMessage());
        }
        $this->success(lang('Backup complete, %s', [$backup]));
    }
}