1Panel/agent/app/service/snapshot_rollback.go

package service

import (
	"fmt"
	"os"
	"path"

	"github.com/1Panel-dev/1Panel/agent/app/dto"
	"github.com/1Panel-dev/1Panel/agent/app/task"
	"github.com/1Panel-dev/1Panel/agent/constant"
	"github.com/1Panel-dev/1Panel/agent/global"
	"github.com/1Panel-dev/1Panel/agent/i18n"
	"github.com/1Panel-dev/1Panel/agent/utils/files"
)

func (u *SnapshotService) SnapshotRollback(req dto.SnapshotRecover) error {
	global.LOG.Info("start to rollback now")
	snap, err := snapshotRepo.Get(commonRepo.WithByID(req.ID))
	if err != nil {
		return err
	}
	if len(snap.TaskRollbackID) != 0 {
		req.TaskID = snap.TaskRollbackID
	} else {
		_ = snapshotRepo.Update(snap.ID, map[string]interface{}{"task_rollback_id": req.TaskID})
	}
	taskItem, err := task.NewTaskWithOps(snap.Name, task.TaskRollback, task.TaskScopeSnapshot, req.TaskID, snap.ID)
	if err != nil {
		global.LOG.Errorf("new task for create snapshot failed, err: %v", err)
		return err
	}
	go func() {
		rootDir := fmt.Sprintf("%s/1panel_original/original_%s", global.CONF.System.BaseDir, snap.Name)
		baseDir := path.Join(rootDir, "base")

		FileOp := files.NewFileOp()
		taskItem.AddSubTask(
			i18n.GetWithName("SnapCopy", "/usr/local/bin/1pctl"),
			func(t *task.Task) error {
				return FileOp.CopyFile(path.Join(baseDir, "1pctl"), "/usr/local/bin")
			},
			nil,
		)
		taskItem.AddSubTask(
			i18n.GetWithName("SnapCopy", "/usr/local/bin/1panel"),
			func(t *task.Task) error {
				return FileOp.CopyFile(path.Join(baseDir, "1panel"), "/usr/local/bin")
			},
			nil,
		)
		taskItem.AddSubTask(
			i18n.GetWithName("SnapCopy", "/usr/local/bin/1panel_agent"),
			func(t *task.Task) error {
				return FileOp.CopyFile(path.Join(baseDir, "1panel_agent"), "/usr/local/bin")
			},
			nil,
		)
		taskItem.AddSubTask(
			i18n.GetWithName("SnapCopy", "/etc/systemd/system/1panel.service"),
			func(t *task.Task) error {
				return FileOp.CopyFile(path.Join(baseDir, "1panel.service"), "/etc/systemd/system")
			},
			nil,
		)
		taskItem.AddSubTask(
			i18n.GetWithName("SnapCopy", "/etc/systemd/system/1panel_agent.service"),
			func(t *task.Task) error {
				return FileOp.CopyFile(path.Join(baseDir, "1panel.service"), "/etc/systemd/system")
			},
			nil,
		)
		taskItem.AddSubTask(
			i18n.GetWithName("SnapCopy", "/etc/docker/daemon.json"),
			func(t *task.Task) error {
				return FileOp.CopyFile(path.Join(baseDir, "daemon.json"), "/etc/docker")
			},
			nil,
		)
		taskItem.AddSubTask(
			i18n.GetWithName("SnapCopy", global.CONF.System.Backup),
			func(t *task.Task) error {
				return FileOp.CopyDir(path.Join(rootDir, "backup"), global.CONF.System.Backup)
			},
			nil,
		)
		taskItem.AddSubTask(
			i18n.GetWithName("SnapCopy", global.CONF.System.BaseDir),
			func(t *task.Task) error {
				return FileOp.CopyDir(path.Join(rootDir, "1panel"), global.CONF.System.BaseDir)
			},
			nil,
		)
		if err := taskItem.Execute(); err != nil {
			_ = snapshotRepo.Update(req.ID, map[string]interface{}{"rollback_status": constant.StatusFailed, "rollback_message": err.Error()})
			return
		}
		_ = snapshotRepo.Update(req.ID, map[string]interface{}{
			"recover_status":   "",
			"recover_message":  "",
			"rollback_status":  "",
			"rollback_message": "",
			"interrupt_step":   "",
		})
		_ = os.RemoveAll(rootDir)
	}()
	return nil
}