mtab/extend/Upgrade2.php

<?php
ini_set('max_execution_time', 0);
ini_set('memory_limit', '500M');

class Upgrade2
{
    protected string $archiveFile = "";//升级文件地址
    protected string $extractPath = "";//解压目录地址
    protected string $root_path = "";//程序根目录
    public string $update_download_url = "";//升级zip文件下载地址
    public string $update_sql_url = "";//升级sql脚本文件地址
    public string $update_script = "";//升级后执行的脚本地址
    protected bool $isLog = false;

    //构造方法初始化一些数据
    function __construct($update_download_url = null, $update_sql_url = null, $update_script = null)
    {
        $this->archiveFile = runtime_path() . 'mtab.zip';
        $this->extractPath = runtime_path();
        $this->root_path = root_path();
        if ($update_download_url !== null) {
            $this->update_download_url = $update_download_url;
        }
        if ($update_sql_url !== null) {
            $this->update_sql_url = $update_sql_url;
        }
        if ($update_script !== null) {
            $this->update_script = $update_script;
        }
    }

    //运行入口
    function run($cli = false): bool
    {
        if ($cli) {
            $this->isLog = true;
        }
        return $this->startUpgrade();
    }

    public function log($msg)
    {
        if ($this->isLog) {
            print_r($msg . "\n");
        }
    }

    //新的进程启动升级
    private function startUpgrade(): bool
    {
        //如果有程序代码的更新资源则更新程序代码
        if (strlen($this->update_download_url) > 1) {
            //如果有遗留的解压资源则删除
            $this->log("正在检查是否有旧版本的安装包，并删除。");
            $this->deleteDirectory("{$this->extractPath}mtab");
            //如果存在旧的升级包则删除
            $this->delZip();
            //下载远程更新包
            $this->log("正在下载升级包...");
            if (!$this->fileDownload()) {
                $this->log('资源下载失败');
                abort(0, '资源下载失败');
            }
            //解压升级包
            $this->log("正在解压升级包...");
            if (!$this->unzip($this->archiveFile, $this->extractPath)) {
                $this->delZip();
                abort(0, '升级资源包解压失败');
            }
            $this->log("正在更新程序...");
            $this->deleteDirectory(public_path() . 'dist/');//删除旧的网站文件
            //拷贝覆盖
            $this->copy();
            //删除下载的更新包
            $this->log("正在删除升级包...");
            $this->delZip();
            //更新完后的一些操作
        }
        //如果有数据库的更新资源则更新程序代码
        if (strlen($this->update_sql_url) > 1) {
            $this->log("正在更新数据库...");
            $this->updateSql();
        }
        if (file_exists("{$this->root_path}install.sql")) {
            $this->log("正在更新数据库...");
            $this->updateSql("{$this->root_path}install.sql");
        }
        //退出
        return true;
    }

    private function fileDownload(): bool
    {
        $length = 0;
        try {
            $f = fopen($this->update_download_url, 'r');
            $w = fopen($this->archiveFile, 'wb+');
            $fileSize = $this->getFileSize($this->update_download_url);
            do {
                $a = fread($f, 1024 * 64);
                $length += strlen($a);
                fwrite($w, $a);
                // 计算下载进度
                $progress = ($fileSize > 0) ? round($length / $fileSize * 100, 2) : 0;
                // 打印进度条，在一行内更新
                if ($this->isLog) {
                    if ($progress <= 100) {
                        $this->printProgress((int)$progress);
                    }
                }
            } while ($a);
            fclose($w);
            fclose($f);
            $this->log("\n下载完成");
        } catch (ErrorException $e) {
            return false;
        }
        return true;
    }

    private function printProgress(float $progress)
    {
        try {
            $barLength = 50; // 进度条的总长度
            $completed = round($progress / 100 * $barLength); // 完成的部分
            $bar = str_repeat('=', $completed) . str_repeat(' ', max($barLength - $completed, 0)); // 拼接进度条
            echo "\r[" . $bar . "] " . $progress . "%";
        } catch (Exception $e) {

        }
    }

    private function getFileSize(string $url): int
    {
        // 使用 HEAD 请求获取文件大小
        $headers = get_headers($url, 1);
        return isset($headers['Content-Length']) ? (int)$headers['Content-Length'] : 0;
    }

    //删除升级包
    function delZip()
    {
        if (file_exists($this->archiveFile)) {
            unlink($this->archiveFile);
        }
    }

    //解压
    private function unzip($archiveFile, $extractPath): bool
    {
        $zip = new ZipArchive();
        if ($zip->open($archiveFile) === TRUE) {
            $zip->extractTo($extractPath, null);
            $zip->close();
        } else {
            return false;
        }
        return true;
    }

    //升级的数据库
    function updateSql($path = null)
    {
        if ($path) {
            $f = fopen($path, 'r');
        } else {
            $f = fopen($this->update_sql_url, 'r');
        }
        $sql = "";
        do {
            $sqlTmp = fread($f, 1024);
            $sql = $sql . $sqlTmp;
        } while ($sqlTmp);
        fclose($f);
        // 解析SQL文件内容并执行
        $sql_statements = explode(';', trim($sql));
        foreach ($sql_statements as $sql_statement) {
            if (!empty($sql_statement)) {
                try {
                    \think\facade\Db::execute($sql_statement);
                } catch (Exception $e) {
                }
            }
        }
    }

    //递归删除目录
    function deleteDirectory($dir)
    {
        if (!is_dir($dir)) {
            return;
        }
        $files = scandir($dir);
        foreach ($files as $file) {
            if ($file != '.' && $file != '..') {
                if (is_dir("$dir/$file")) {
                    $this->deleteDirectory("$dir/$file");
                } else {
                    unlink("$dir/$file");
                }
            }
        }
        rmdir($dir);
    }

    // 递归复制目录及其内容
    function copyDir($source, $dest)
    {
        if (!is_dir($dest)) {
            mkdir($dest, 0777, true);
        }
        $files = scandir($source);
        foreach ($files as $file) {
            if ($file !== '.' && $file !== '..') {
                $src = $source . '/' . $file;
                $dst = $dest . '/' . $file;
                if (is_dir($src)) {
                    $this->copyDir($src, $dst);
                } else {
                    copy($src, $dst);
                }
            }
        }
    }

    //覆盖原来的程序
    private function copy()
    {
        //移动覆盖
        $this->copyDir("{$this->extractPath}mtab/", "{$this->root_path}");
        //删除解压目录
        $this->deleteDirectory("{$this->extractPath}mtab");
    }

}