dnsmgr/app/controller/Install.php

<?php

namespace app\controller;

use Exception;
use app\BaseController;
use PDO;
use think\facade\Cache;
use think\facade\Db;
use think\facade\Request;

class Install extends BaseController
{
    public function index()
    {
        if (file_exists(app()->getRootPath() . '.env')) {
            return '当前已经安装成功，如果需要重新安装，请手动删除根目录.env文件';
        }
        if ($this->request->isPost()) {
            $mysql_host = $this->request->post('mysql_host');
            $mysql_port = intval($this->request->post('mysql_port', '3306'));
            $mysql_user = $this->request->post('mysql_user', null, 'trim');
            $mysql_pwd = $this->request->post('mysql_pwd', null, 'trim');
            $mysql_name = $this->request->post('mysql_name', null, 'trim');
            $mysql_prefix = $this->request->post('mysql_prefix', 'cloud_', 'trim');
            $admin_username = $this->request->post('admin_username', null, 'trim');
            $admin_password = $this->request->post('admin_password', null, 'trim');

            if (!$mysql_host || !$mysql_user || !$mysql_pwd || !$mysql_name || !$admin_username || !$admin_password) {
                return json(['code' => 0, 'msg' => '必填项不能为空']);
            }

            $configData = file_get_contents(app()->getRootPath() . '.example.env');
            $configData = str_replace(['{dbhost}', '{dbname}', '{dbuser}', '{dbpwd}', '{dbport}', '{dbprefix}'], [$mysql_host, $mysql_name, $mysql_user, $mysql_pwd, $mysql_port, $mysql_prefix], $configData);

            try {
                $DB = new PDO("mysql:host=" . $mysql_host . ";dbname=" . $mysql_name . ";port=" . $mysql_port, $mysql_user, $mysql_pwd);
            } catch (Exception $e) {
                if ($e->getCode() == 2002) {
                    $errorMsg = '连接数据库失败：数据库地址填写错误！';
                } elseif ($e->getCode() == 1045) {
                    $errorMsg = '连接数据库失败：数据库用户名或密码填写错误！';
                } elseif ($e->getCode() == 1049) {
                    $errorMsg = '连接数据库失败：数据库名不存在！';
                } else {
                    $errorMsg = '连接数据库失败：' . $e->getMessage();
                }
                return json(['code' => 0, 'msg' => $errorMsg]);
            }
            $DB->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_SILENT);
            $DB->exec("set sql_mode = ''");
            $DB->exec("set names utf8");

            $sqls = file_get_contents(app()->getAppPath() . 'sql/install.sql');
            $sqls = explode(';', $sqls);

            $password = password_hash($admin_password, PASSWORD_DEFAULT);
            $sqls[] = "REPLACE INTO `" . $mysql_prefix . "config` VALUES ('sys_key', '" . random(16) . "')";
            $sqls[] = "INSERT INTO `" . $mysql_prefix . "user` (`username`,`password`,`level`,`regtime`,`lasttime`,`status`) VALUES ('" . addslashes($admin_username) . "', '$password', 2, NOW(), NOW(), 1)";

            $success = 0;
            $error = 0;
            $errorMsg = null;
            foreach ($sqls as $value) {
                $value = trim($value);
                if (empty($value)) {
                    continue;
                }
                $value = str_replace('dnsmgr_', $mysql_prefix, $value);
                if ($DB->exec($value) === false) {
                    $error++;
                    $dberror = $DB->errorInfo();
                    $errorMsg .= $dberror[2] . "\n";
                } else {
                    $success++;
                }
            }
            if (empty($errorMsg)) {
                if (!file_put_contents(app()->getRootPath() . '.env', $configData)) {
                    return json(['code' => 0, 'msg' => '保存失败，请确保网站根目录有写入权限']);
                }
                Cache::clear();
                return json(['code' => 1, 'msg' => '安装完成！成功执行SQL语句' . $success . '条']);
            } else {
                return json(['code' => 0, 'msg' => $errorMsg]);
            }
        }
        return view();
    }

}