feat(项目代码): 首次提交
All checks were successful
continuous-integration/drone/push Build is passing

This commit is contained in:
xiaoqidun 2020-12-24 10:10:59 +08:00
parent 25aca4de3f
commit 776c94c94c
6 changed files with 165 additions and 0 deletions

20
.drone.yml Normal file
View File

@ -0,0 +1,20 @@
kind: pipeline
type: docker
name: default
steps:
- name: build
pull: if-not-exists
image: xiaoqidun/goenv
commands:
- CGO_ENABLED=0 go build -o gocos -trimpath -ldflags '-s -w' gocos.go
- name: docker
pull: if-not-exists
image: plugins/docker
settings:
repo: xiaoqidun/gocos
username:
from_secret: docker_username
password:
from_secret: docker_password
dockerfile: Dockerfile

3
.gitignore vendored Normal file
View File

@ -0,0 +1,3 @@
.idea/
.vscode/
.devcontainer/

11
Dockerfile Normal file
View File

@ -0,0 +1,11 @@
# 基础镜像
FROM alpine
# 作者信息
LABEL MAINTAINER="xiaoqidun@gmail.com"
# 复制程序
COPY gocos /bin/gocos
# 启动命令
ENTRYPOINT /bin/gocos

5
go.mod Normal file
View File

@ -0,0 +1,5 @@
module github.com/xiaoqidun/gocos
go 1.16
require github.com/tencentyun/cos-go-sdk-v5 v0.7.15

13
go.sum Normal file
View File

@ -0,0 +1,13 @@
github.com/QcloudApi/qcloud_sign_golang v0.0.0-20141224014652-e4130a326409/go.mod h1:1pk82RBxDY/JZnPQrtqHlUFfCctgdorsd9M06fMynOM=
github.com/davecgh/go-spew v1.1.0/go.mod h1:J7Y8YcW2NihsgmVo/mv3lAwl/skON4iLHjSsI+c5H38=
github.com/google/go-querystring v1.0.0 h1:Xkwi/a1rcvNg1PPYe5vI8GbeBY/jrVuDX5ASuANWTrk=
github.com/google/go-querystring v1.0.0/go.mod h1:odCYkC5MyYFN7vkCjXpyrEuKhc/BUO6wN/zVPAxq5ck=
github.com/google/uuid v1.1.1 h1:Gkbcsh/GbpXz7lPftLA3P6TYMwjCLYm83jiFQZF/3gY=
github.com/google/uuid v1.1.1/go.mod h1:TIyPZe4MgqvfeYDBFedMoGGpEw/LqOeaOT+nhxU+yHo=
github.com/mozillazg/go-httpheader v0.2.1 h1:geV7TrjbL8KXSyvghnFm+NyTux/hxwueTSrwhe88TQQ=
github.com/mozillazg/go-httpheader v0.2.1/go.mod h1:jJ8xECTlalr6ValeXYdOF8fFUISeBAdw6E61aqQma60=
github.com/pmezard/go-difflib v1.0.0/go.mod h1:iKH77koFhYxTK1pcRnkKkqfTogsbg7gZNVY4sRDYZ/4=
github.com/stretchr/objx v0.1.0/go.mod h1:HFkY916IF+rwdDfMAkV7OtwuqBVzrE8GR6GFx+wExME=
github.com/stretchr/testify v1.3.0/go.mod h1:M5WIy9Dh21IEIfnGCwXGc5bZfKNJtfHm1UVUgZn+9EI=
github.com/tencentyun/cos-go-sdk-v5 v0.7.15 h1:+uKDylojhKkGhBzb8B7J/nIoEl6afGYiJubkVWEctxA=
github.com/tencentyun/cos-go-sdk-v5 v0.7.15/go.mod h1:wQBO5HdAkLjj2q6XQiIfDSP8DXDNrppDRw2Kp/1BODA=

113
gocos.go Normal file
View File

@ -0,0 +1,113 @@
package main
import (
"context"
"github.com/tencentyun/cos-go-sdk-v5"
"io/fs"
"log"
"net/http"
"net/url"
"os"
"path/filepath"
"strings"
)
const (
CodeErr = 1
SecretID = "secret_id"
SecretKey = "secret_key"
BucketURL = "bucket_url"
Source = "source"
Target = "target"
StripPrefix = "strip_prefix"
PathSeparator = "/"
)
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 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) {
os.Exit(CodeErr)
return
}
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 {
os.Exit(CodeErr)
return
}
sourceLen := len(sourceFiles)
if sourceLen == 0 {
return
}
if !strings.HasSuffix(target, PathSeparator) {
target += PathSeparator
}
u, err := url.Parse(bucketURL)
if err != nil {
os.Exit(CodeErr)
return
}
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 {
os.Exit(CodeErr)
return
}
log.Printf("source:%s target:%s\r\n", local, remote)
}
}