gocos/gocos.go

package main

import (
	"context"
	"errors"
	"github.com/tencentyun/cos-go-sdk-v5"
	"io/fs"
	"log"
	"net/http"
	"net/url"
	"os"
	"path/filepath"
	"strings"
	"time"
	_ "time/tzdata"
)

const (
	SecretID      = "secret_id"
	SecretKey     = "secret_key"
	BucketURL     = "bucket_url"
	Source        = "source"
	Target        = "target"
	StripPrefix   = "strip_prefix"
	PathSeparator = "/"
)

func ErrExit(err error) {
	log.Println(err.Error())
	os.Exit(1)
}

func GetConfig(key string) string {
	key = "PLUGIN_" + strings.ToUpper(key)
	return os.Getenv(key)
}

func VarIsEmpty(a ...interface{}) bool {
	for _, v := range a {
		switch v := v.(type) {
		case string:
			if "" == v {
				return true
			}
		case []byte:
			if 0 == len(v) {
				return true
			}
		case []string:
			if 0 == len(v) {
				return true
			}
		}
	}
	return false
}

func init() {
	tz := os.Getenv("TZ")
	if tz == "" {
		tz = "Asia/Shanghai"
	}
	time.Local, _ = time.LoadLocation(tz)
}

func main() {
	var (
		err         error
		secretID    = GetConfig(SecretID)
		secretKey   = GetConfig(SecretKey)
		bucketURL   = GetConfig(BucketURL)
		source      = GetConfig(Source)
		target      = GetConfig(Target)
		stripPrefix = GetConfig(StripPrefix)
	)
	if VarIsEmpty(secretID, secretKey, bucketURL, source, target) {
		ErrExit(errors.New("input error"))
	}
	sourceFiles := make([]string, 0, 0)
	if err = filepath.WalkDir(source, func(path string, d fs.DirEntry, err error) error {
		if !d.IsDir() {
			sourceFiles = append(sourceFiles, path)
		}
		return nil
	}); err != nil {
		ErrExit(err)
	}
	sourceLen := len(sourceFiles)
	if sourceLen == 0 {
		return
	}
	if !strings.HasSuffix(target, PathSeparator) {
		target += PathSeparator
	}
	u, err := url.Parse(bucketURL)
	if err != nil {
		ErrExit(err)
	}
	cosClient := cos.NewClient(
		&cos.BaseURL{
			BucketURL: u,
		},
		&http.Client{
			Transport: &cos.AuthorizationTransport{
				SecretID:  secretID,
				SecretKey: secretKey,
			},
		})
	for i := 0; i < sourceLen; i++ {
		local := sourceFiles[i]
		remote := strings.TrimPrefix(local, stripPrefix)
		if strings.HasPrefix(remote, PathSeparator) {
			remote = strings.TrimPrefix(remote, PathSeparator)
		}
		remote = target + remote
		if _, err = cosClient.Object.PutFromFile(context.Background(), remote, local, &cos.ObjectPutOptions{
			ACLHeaderOptions:       &cos.ACLHeaderOptions{},
			ObjectPutHeaderOptions: &cos.ObjectPutHeaderOptions{},
		}); err != nil {
			ErrExit(err)
		}
		log.Printf("source:%s target:%s\r\n", local, remote)
	}
}