package service import ( "encoding/json" "fmt" "io/fs" "os" "path" "strings" "time" "github.com/1Panel-dev/1Panel/backend/app/dto" "github.com/1Panel-dev/1Panel/backend/app/model" "github.com/1Panel-dev/1Panel/backend/constant" "github.com/1Panel-dev/1Panel/backend/global" "github.com/1Panel-dev/1Panel/backend/utils/cmd" "github.com/1Panel-dev/1Panel/backend/utils/compose" "github.com/1Panel-dev/1Panel/backend/utils/files" "github.com/pkg/errors" ) func (u *BackupService) WebsiteBackup(req dto.CommonBackup) error { localDir, err := loadLocalDir() if err != nil { return err } website, err := websiteRepo.GetFirst(websiteRepo.WithDomain(req.Name)) if err != nil { return err } timeNow := time.Now().Format("20060102150405") backupDir := fmt.Sprintf("%s/website/%s", localDir, req.Name) fileName := fmt.Sprintf("%s_%s.tar.gz", website.PrimaryDomain, timeNow) if err := handleWebsiteBackup(&website, backupDir, fileName); err != nil { return err } record := &model.BackupRecord{ Type: "website", Name: website.PrimaryDomain, DetailName: "", Source: "LOCAL", BackupType: "LOCAL", FileDir: backupDir, FileName: fileName, } if err := backupRepo.CreateRecord(record); err != nil { global.LOG.Errorf("save backup record failed, err: %v", err) return err } return nil } func (u *BackupService) WebsiteRecover(req dto.CommonRecover) error { website, err := websiteRepo.GetFirst(websiteRepo.WithDomain(req.Name)) if err != nil { return err } fileOp := files.NewFileOp() if !fileOp.Stat(req.File) { return errors.New(fmt.Sprintf("%s file is not exist", req.File)) } global.LOG.Infof("recover website %s from backup file %s", req.Name, req.File) if err := handleWebsiteRecover(&website, req.File, false); err != nil { return err } return nil } func handleWebsiteRecover(website *model.Website, recoverFile string, isRollback bool) error { fileOp := files.NewFileOp() tmpPath := strings.ReplaceAll(recoverFile, ".tar.gz", "") if err := handleUnTar(recoverFile, path.Dir(recoverFile)); err != nil { return err } defer func() { _ = os.RemoveAll(tmpPath) }() temPathWithName := tmpPath + "/" + website.Alias if !fileOp.Stat(tmpPath+"/website.json") || !fileOp.Stat(temPathWithName+".conf") || !fileOp.Stat(temPathWithName+".web.tar.gz") { return errors.New("the wrong recovery package does not have .conf or .web.tar.gz files") } if website.Type == constant.Deployment { if !fileOp.Stat(temPathWithName + ".app.tar.gz") { return errors.New("the wrong recovery package does not have .app.tar.gz files") } } var oldWebsite model.Website websiteJson, err := os.ReadFile(tmpPath + "/website.json") if err != nil { return err } if err := json.Unmarshal(websiteJson, &oldWebsite); err != nil { return fmt.Errorf("unmarshal app.json failed, err: %v", err) } if oldWebsite.Alias != website.Alias || oldWebsite.Type != website.Type || oldWebsite.ID != website.ID { return errors.New("the current backup file does not match the application") } isOk := false if !isRollback { rollbackFile := fmt.Sprintf("%s/original/website/%s_%s.tar.gz", global.CONF.System.BaseDir, website.Alias, time.Now().Format("20060102150405")) if err := handleWebsiteBackup(website, path.Dir(rollbackFile), path.Base(rollbackFile)); err != nil { global.LOG.Errorf("backup website %s for rollback before recover failed, err: %v", website.Alias, err) } defer func() { if !isOk { if err := handleWebsiteRecover(website, rollbackFile, true); err != nil { global.LOG.Errorf("rollback website %s from %s failed, err: %v", website.Alias, rollbackFile, err) return } global.LOG.Infof("rollback website %s from %s successful", website.Alias, rollbackFile) _ = os.RemoveAll(rollbackFile) } else { _ = os.RemoveAll(rollbackFile) } }() } nginxInfo, err := appInstallRepo.LoadBaseInfo(constant.AppOpenresty, "") if err != nil { return err } nginxConfPath := fmt.Sprintf("%s/openresty/%s/conf/conf.d", constant.AppInstallDir, nginxInfo.Name) if err := fileOp.CopyFile(fmt.Sprintf("%s/%s.conf", tmpPath, website.Alias), nginxConfPath); err != nil { return err } if website.Type == constant.Deployment { app, err := appInstallRepo.GetFirst(commonRepo.WithByID(website.AppInstallID)) if err != nil { return err } if err := handleAppRecover(&app, fmt.Sprintf("%s/%s.app.tar.gz", tmpPath, website.Alias), true); err != nil { return err } if _, err := compose.Restart(fmt.Sprintf("%s/%s/%s/docker-compose.yml", constant.AppInstallDir, app.App.Key, app.Name)); err != nil { return err } } siteDir := fmt.Sprintf("%s/openresty/%s/www/sites", constant.AppInstallDir, nginxInfo.Name) if err := handleUnTar(fmt.Sprintf("%s/%s.web.tar.gz", tmpPath, website.Alias), siteDir); err != nil { return err } stdout, err := cmd.Execf("docker exec -i %s nginx -s reload", nginxInfo.ContainerName) if err != nil { return errors.New(string(stdout)) } if err := websiteRepo.SaveWithoutCtx(&oldWebsite); err != nil { return err } return nil } func handleWebsiteBackup(website *model.Website, backupDir, fileName string) error { fileOp := files.NewFileOp() tmpDir := fmt.Sprintf("%s/%s", backupDir, strings.ReplaceAll(fileName, ".tar.gz", "")) if !fileOp.Stat(tmpDir) { if err := os.MkdirAll(tmpDir, os.ModePerm); err != nil { return fmt.Errorf("mkdir %s failed, err: %v", backupDir, err) } } defer func() { _ = os.RemoveAll(tmpDir) }() remarkInfo, _ := json.Marshal(website) if err := fileOp.SaveFile(tmpDir+"/website.json", string(remarkInfo), fs.ModePerm); err != nil { return err } global.LOG.Info("put website.json into tmp dir successful") nginxInfo, err := appInstallRepo.LoadBaseInfo(constant.AppOpenresty, "") if err != nil { return err } nginxConfFile := fmt.Sprintf("%s/openresty/%s/conf/conf.d/%s.conf", constant.AppInstallDir, nginxInfo.Name, website.Alias) if err := fileOp.CopyFile(nginxConfFile, tmpDir); err != nil { return err } global.LOG.Info("put openresty conf into tmp dir successful") if website.Type == constant.Deployment { app, err := appInstallRepo.GetFirst(commonRepo.WithByID(website.AppInstallID)) if err != nil { return err } if err := handleAppBackup(&app, tmpDir, fmt.Sprintf("%s.app.tar.gz", website.Alias)); err != nil { return err } global.LOG.Info("put app.tar.gz into tmp dir successful") } websiteDir := fmt.Sprintf("%s/openresty/%s/www/sites/%s", constant.AppInstallDir, nginxInfo.Name, website.Alias) if err := fileOp.Compress([]string{websiteDir}, tmpDir, fmt.Sprintf("%s.web.tar.gz", website.Alias), files.TarGz); err != nil { return err } global.LOG.Info("put web.tar.gz into tmp dir successful, now start to tar tmp dir") if err := fileOp.Compress([]string{tmpDir}, backupDir, fileName, files.TarGz); err != nil { return err } return nil }