123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178 |
- <?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 = "";//升级后执行的脚本地址
- //构造方法初始化一些数据
- 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()
- {
- return $this->startUpgrade();
- }
- //新的进程启动升级
- private function startUpgrade()
- {
- //如果有程序代码的更新资源则更新程序代码
- if (strlen($this->update_download_url) > 1) {
- //如果有遗留的解压资源则删除
- $this->deleteDirectory("{$this->extractPath}mtab");
- //如果存在旧的升级包则删除
- $this->delZip();
- //下载远程更新包
- if(!$this->fileDownload()){
- return "资源下载失败";
- }
- //解压升级包
- if (!$this->unzip($this->archiveFile, $this->extractPath)) {
- $this->delZip();
- return '升级资源包解压失败';
- }
- $this->deleteDirectory(public_path().'dist/');//删除旧的网站文件
- //拷贝覆盖
- $this->copy();
- //删除下载的更新包
- $this->delZip();
- //更新完后的一些操作
- }
- //如果有数据库的更新资源则更新程序代码
- if (strlen($this->update_sql_url) > 1) {
- $this->updateSql();
- }
- //退出
- return true;
- }
- private function fileDownload(): bool
- {
- try {
- $f = fopen($this->update_download_url, 'r');
- $w = fopen($this->archiveFile, 'wb+');
- do {
- $a = fread($f, 1024);
- fwrite($w, $a);
- } while ($a);
- fclose($w);
- fclose($f);
- } catch (ErrorException $e) {
- return false;
- }
- return true;
- }
- //删除升级包
- 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()
- {
- $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");
- }
- }
|