mirror of
https://github.com/1Panel-dev/1Panel.git
synced 2024-09-21 16:16:00 +08:00
235 lines
7.6 KiB
Go
235 lines
7.6 KiB
Go
package service
|
|
|
|
import (
|
|
"encoding/json"
|
|
"errors"
|
|
"fmt"
|
|
"io/ioutil"
|
|
"net/http"
|
|
"os"
|
|
"runtime"
|
|
"strings"
|
|
"time"
|
|
|
|
"github.com/1Panel-dev/1Panel/backend/app/dto"
|
|
"github.com/1Panel-dev/1Panel/backend/global"
|
|
"github.com/1Panel-dev/1Panel/backend/utils/cmd"
|
|
"github.com/1Panel-dev/1Panel/backend/utils/common"
|
|
"github.com/1Panel-dev/1Panel/backend/utils/files"
|
|
)
|
|
|
|
type UpgradeService struct{}
|
|
|
|
type IUpgradeService interface {
|
|
Upgrade(req dto.Upgrade) error
|
|
LoadNotes(req dto.Upgrade) (string, error)
|
|
SearchUpgrade() (*dto.UpgradeInfo, error)
|
|
}
|
|
|
|
func NewIUpgradeService() IUpgradeService {
|
|
return &UpgradeService{}
|
|
}
|
|
|
|
func (u *UpgradeService) SearchUpgrade() (*dto.UpgradeInfo, error) {
|
|
var upgrade dto.UpgradeInfo
|
|
currentVersion, err := settingRepo.Get(settingRepo.WithByKey("SystemVersion"))
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
latestVersion, err := u.loadVersion(true, currentVersion.Value)
|
|
if err != nil {
|
|
global.LOG.Infof("load latest version failed, err: %v", err)
|
|
return nil, err
|
|
}
|
|
if !common.CompareVersion(string(latestVersion), currentVersion.Value) {
|
|
return nil, err
|
|
}
|
|
upgrade.LatestVersion = latestVersion
|
|
if latestVersion[0:4] == currentVersion.Value[0:4] {
|
|
upgrade.NewVersion = ""
|
|
} else {
|
|
newerVersion, err := u.loadVersion(false, currentVersion.Value)
|
|
if err != nil {
|
|
global.LOG.Infof("load newer version failed, err: %v", err)
|
|
return nil, err
|
|
}
|
|
if newerVersion == currentVersion.Value {
|
|
upgrade.NewVersion = ""
|
|
} else {
|
|
upgrade.NewVersion = newerVersion
|
|
}
|
|
}
|
|
notes, err := u.loadReleaseNotes(fmt.Sprintf("%s/%s/%s/release/1panel-%s-release-notes", global.CONF.System.RepoUrl, global.CONF.System.Mode, upgrade.LatestVersion, upgrade.LatestVersion))
|
|
if err != nil {
|
|
return nil, fmt.Errorf("load relase-notes of version %s failed, err: %v", latestVersion, err)
|
|
}
|
|
upgrade.ReleaseNote = notes
|
|
return &upgrade, nil
|
|
}
|
|
|
|
func (u *UpgradeService) LoadNotes(req dto.Upgrade) (string, error) {
|
|
notes, err := u.loadReleaseNotes(fmt.Sprintf("%s/%s/%s/release/1panel-%s-release-notes", global.CONF.System.RepoUrl, global.CONF.System.Mode, req.Version, req.Version))
|
|
if err != nil {
|
|
return "", fmt.Errorf("load relase-notes of version %s failed, err: %v", req.Version, err)
|
|
}
|
|
return notes, nil
|
|
}
|
|
|
|
func (u *UpgradeService) Upgrade(req dto.Upgrade) error {
|
|
global.LOG.Info("start to upgrade now...")
|
|
fileOp := files.NewFileOp()
|
|
timeStr := time.Now().Format("20060102150405")
|
|
rootDir := fmt.Sprintf("%s/upgrade_%s/downloads", global.CONF.System.TmpDir, timeStr)
|
|
originalDir := fmt.Sprintf("%s/upgrade_%s/original", global.CONF.System.TmpDir, timeStr)
|
|
if err := os.MkdirAll(rootDir, os.ModePerm); err != nil {
|
|
return err
|
|
}
|
|
if err := os.MkdirAll(originalDir, os.ModePerm); err != nil {
|
|
return err
|
|
}
|
|
|
|
downloadPath := fmt.Sprintf("%s/%s/%s/release", global.CONF.System.RepoUrl, global.CONF.System.Mode, req.Version)
|
|
fileName := fmt.Sprintf("1panel-%s-%s-%s.tar.gz", req.Version, "linux", runtime.GOARCH)
|
|
_ = settingRepo.Update("SystemStatus", "Upgrading")
|
|
go func() {
|
|
if err := fileOp.DownloadFile(downloadPath+"/"+fileName, rootDir+"/"+fileName); err != nil {
|
|
global.LOG.Errorf("download service file failed, err: %v", err)
|
|
_ = settingRepo.Update("SystemStatus", "Free")
|
|
return
|
|
}
|
|
global.LOG.Info("download all file successful!")
|
|
defer func() {
|
|
_ = os.Remove(rootDir)
|
|
}()
|
|
if err := handleUnTar(rootDir+"/"+fileName, rootDir); err != nil {
|
|
global.LOG.Errorf("decompress file failed, err: %v", err)
|
|
_ = settingRepo.Update("SystemStatus", "Free")
|
|
return
|
|
}
|
|
tmpDir := rootDir + "/" + strings.ReplaceAll(fileName, ".tar.gz", "")
|
|
|
|
if err := u.handleBackup(fileOp, originalDir); err != nil {
|
|
global.LOG.Errorf("handle backup original file failed, err: %v", err)
|
|
_ = settingRepo.Update("SystemStatus", "Free")
|
|
return
|
|
}
|
|
global.LOG.Info("backup original data successful, now start to upgrade!")
|
|
|
|
if err := cpBinary(tmpDir+"/1panel", "/usr/local/bin/1panel"); err != nil {
|
|
u.handleRollback(fileOp, originalDir, 1)
|
|
global.LOG.Errorf("upgrade 1panel failed, err: %v", err)
|
|
return
|
|
}
|
|
|
|
if err := cpBinary(tmpDir+"/1pctl", "/usr/local/bin/1pctl"); err != nil {
|
|
u.handleRollback(fileOp, originalDir, 2)
|
|
global.LOG.Errorf("upgrade 1pctl failed, err: %v", err)
|
|
return
|
|
}
|
|
if _, err := cmd.Execf("sed -i -e 's#BASE_DIR=.*#BASE_DIR=%s#g' /usr/local/bin/1pctl", global.CONF.System.BaseDir); err != nil {
|
|
u.handleRollback(fileOp, originalDir, 2)
|
|
global.LOG.Errorf("upgrade basedir in 1pctl failed, err: %v", err)
|
|
return
|
|
}
|
|
|
|
if err := cpBinary(tmpDir+"/1panel.service", "/etc/systemd/system/1panel.service"); err != nil {
|
|
u.handleRollback(fileOp, originalDir, 3)
|
|
global.LOG.Errorf("upgrade 1panel.service failed, err: %v", err)
|
|
return
|
|
}
|
|
|
|
global.LOG.Info("upgrade successful!")
|
|
go writeLogs(req.Version)
|
|
_ = settingRepo.Update("SystemVersion", req.Version)
|
|
_ = settingRepo.Update("SystemStatus", "Free")
|
|
_, _ = cmd.Exec("systemctl daemon-reload && systemctl restart 1panel.service")
|
|
}()
|
|
return nil
|
|
}
|
|
|
|
func (u *UpgradeService) handleBackup(fileOp files.FileOp, originalDir string) error {
|
|
if err := fileOp.Copy("/usr/local/bin/1panel", originalDir); err != nil {
|
|
return err
|
|
}
|
|
if err := fileOp.Copy("/usr/local/bin/1pctl", originalDir); err != nil {
|
|
return err
|
|
}
|
|
if err := fileOp.Copy("/etc/systemd/system/1panel.service", originalDir); err != nil {
|
|
return err
|
|
}
|
|
dbPath := global.CONF.System.DbPath + "/" + global.CONF.System.DbFile
|
|
if err := fileOp.Copy(dbPath, originalDir); err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (u *UpgradeService) handleRollback(fileOp files.FileOp, originalDir string, errStep int) {
|
|
dbPath := global.CONF.System.DbPath + "/1Panel.db"
|
|
_ = settingRepo.Update("SystemStatus", "Free")
|
|
if err := cpBinary(originalDir+"/1Panel.db", dbPath); err != nil {
|
|
global.LOG.Errorf("rollback 1panel failed, err: %v", err)
|
|
}
|
|
if err := cpBinary(originalDir+"/1panel", "/usr/local/bin/1panel"); err != nil {
|
|
global.LOG.Errorf("rollback 1pctl failed, err: %v", err)
|
|
}
|
|
if errStep == 1 {
|
|
return
|
|
}
|
|
if err := cpBinary(originalDir+"/1pctl", "/usr/local/bin/1pctl"); err != nil {
|
|
global.LOG.Errorf("rollback 1panel failed, err: %v", err)
|
|
}
|
|
if errStep == 2 {
|
|
return
|
|
}
|
|
if err := cpBinary(originalDir+"/1panel.service", "/etc/systemd/system/1panel.service"); err != nil {
|
|
global.LOG.Errorf("rollback 1panel failed, err: %v", err)
|
|
}
|
|
}
|
|
|
|
func (u *UpgradeService) loadVersion(isLatest bool, currentVersion string) (string, error) {
|
|
path := fmt.Sprintf("%s/%s/latest", global.CONF.System.RepoUrl, global.CONF.System.Mode)
|
|
if !isLatest {
|
|
path = fmt.Sprintf("%s/%s/latest.current", global.CONF.System.RepoUrl, global.CONF.System.Mode)
|
|
}
|
|
latestVersionRes, err := http.Get(path)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer latestVersionRes.Body.Close()
|
|
version, err := ioutil.ReadAll(latestVersionRes.Body)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
if isLatest {
|
|
return string(version), nil
|
|
}
|
|
|
|
versionMap := make(map[string]string)
|
|
if err := json.Unmarshal(version, &versionMap); err != nil {
|
|
return "", fmt.Errorf("load version map failed, err: %v", err)
|
|
}
|
|
|
|
if len(currentVersion) < 4 {
|
|
return "", fmt.Errorf("current version is error format: %s", currentVersion)
|
|
}
|
|
if version, ok := versionMap[currentVersion[0:4]]; ok {
|
|
return version, nil
|
|
}
|
|
return "", errors.New("load version failed in latest.current")
|
|
}
|
|
|
|
func (u *UpgradeService) loadReleaseNotes(path string) (string, error) {
|
|
releaseNotes, err := http.Get(path)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
defer releaseNotes.Body.Close()
|
|
release, err := ioutil.ReadAll(releaseNotes.Body)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
return string(release), nil
|
|
}
|