1Panel/agent/utils/files/archiver.go

package files

import (
	"context"

	"github.com/1Panel-dev/1Panel/agent/buserr"
	"github.com/1Panel-dev/1Panel/agent/utils/cmd"
)

type ShellArchiver interface {
	Extract(ctx context.Context, filePath, dstDir string, secret string) error
	Compress(ctx context.Context, sourcePaths []string, dstFile string, secret string) error
}

func NewShellArchiver(compressType CompressType) (ShellArchiver, error) {
	switch compressType {
	case Tar, Gz, Bz2, TarBz2, Tgz, Xz, TarXz:
		if err := checkCmdAvailability("tar"); err != nil {
			return nil, err
		}
		return NewTarArchiver(compressType), nil
	case TarGz:
		if err := checkCmdAvailability("tar"); err != nil {
			return nil, err
		}
		return NewTarGzArchiver(), nil
	case Zip:
		if err := checkCmdAvailability("zip"); err != nil {
			return nil, err
		}
		return NewZipArchiver(), nil
	case Rar:
		if err := checkCmdAvailability("rar"); err != nil {
			return nil, err
		}
		return NewRarArchiver(), nil
	case X7z:
		if err := checkCmdAvailability("7z"); err != nil {
			return nil, err
		}
		return NewX7zArchiver(), nil
	default:
		return nil, buserr.New("unsupported compress type")
	}
}

func NewExtractShellArchiver(compressType CompressType) (ShellArchiver, error) {
	switch compressType {
	case Tar, Gz, Bz2, TarBz2, Tgz, Xz, TarXz:
		if err := checkCmdAvailability("tar"); err != nil {
			return nil, err
		}
		return NewTarArchiver(compressType), nil
	case TarGz:
		if err := checkCmdAvailability("tar"); err != nil {
			return nil, err
		}
		return NewTarGzArchiver(), nil
	case Zip:
		if err := checkCmdAvailability("unzip"); err != nil {
			return nil, err
		}
		return NewZipArchiver(), nil
	case Rar:
		if err := checkCmdAvailability("unrar"); err != nil {
			return nil, err
		}
		return NewRarArchiver(), nil
	case X7z:
		if err := checkCmdAvailability("7z"); err != nil {
			return nil, err
		}
		return NewX7zArchiver(), nil
	default:
		return nil, buserr.New("unsupported decompress type")
	}
}

func checkCmdAvailability(cmdStr string) error {
	if cmd.Which(cmdStr) {
		return nil
	}
	return buserr.WithName("ErrCmdNotFound", cmdStr)
}