1Panel/agent/utils/cloud_storage/client/local.go

package client

import (
	"context"
	"fmt"
	"os"
	"path"
	"time"

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

type localClient struct{}

func NewLocalClient(vars map[string]interface{}) (*localClient, error) {
	return &localClient{}, nil
}

func (c localClient) ListBuckets() ([]interface{}, error) {
	return nil, nil
}

func (c localClient) Exist(file string) (bool, error) {
	_, err := os.Stat(file)
	return err == nil, err
}

func (c localClient) Size(file string) (int64, error) {
	fileInfo, err := os.Stat(file)
	if err != nil {
		return 0, err
	}
	return fileInfo.Size(), nil
}

func (c localClient) Delete(file string) (bool, error) {
	if err := os.RemoveAll(file); err != nil {
		return false, err
	}
	return true, nil
}

func (c localClient) Upload(ctx context.Context, src, target string) (bool, error) {
	if _, ok := ctx.Deadline(); !ok {
		var cancel context.CancelFunc
		ctx, cancel = context.WithTimeout(ctx, 24*time.Hour)
		defer cancel()
	}
	fileOp := files.NewFileOp()
	if _, err := os.Stat(path.Dir(target)); err != nil {
		if os.IsNotExist(err) {
			if err = os.MkdirAll(path.Dir(target), os.ModePerm); err != nil {
				return false, err
			}
		} else {
			return false, err
		}
	}

	if err := fileOp.CopyAndReNameWithContext(ctx, src, target, "", true); err != nil {
		return false, fmt.Errorf("cp file failed, err: %v", err)
	}
	return true, nil
}

func (c localClient) Download(src, target string) (bool, error) {
	fileOp := files.NewFileOp()
	if _, err := os.Stat(path.Dir(target)); err != nil {
		if os.IsNotExist(err) {
			if err = os.MkdirAll(path.Dir(target), os.ModePerm); err != nil {
				return false, err
			}
		} else {
			return false, err
		}
	}

	if err := fileOp.CopyAndReName(src, target, "", true); err != nil {
		return false, fmt.Errorf("cp file failed, err: %v", err)
	}
	return true, nil
}

func (c localClient) ListObjects(prefix string) ([]string, error) {
	var files []string
	if _, err := os.Stat(prefix); err != nil {
		return files, nil
	}
	list, err := os.ReadDir(prefix)
	if err != nil {
		return files, err
	}
	for i := 0; i < len(list); i++ {
		if !list[i].IsDir() {
			files = append(files, list[i].Name())
		}
	}

	return files, nil
}