0
0
mirror of https://github.com/thegeeklab/wp-s3-action.git synced 2024-11-21 14:50:39 +00:00

Refactor main for testing and organization

This commit is contained in:
Eric Anderson 2016-03-10 00:54:22 -05:00
parent 94c7d6b998
commit 8b8f1248e6

134
main.go
View File

@ -1,6 +1,7 @@
package main
import (
"errors"
"fmt"
"os"
"path/filepath"
@ -23,25 +24,72 @@ type result struct {
err error
}
type app struct {
vargs *PluginArgs
workspace *drone.Workspace
client AWS
jobs []job
}
var (
buildCommit string
)
var MissingAwsValuesMessage = "Must set access_key, secret_key, and bucket"
func main() {
fmt.Printf("Drone S3 Sync Plugin built from %s\n", buildCommit)
vargs := PluginArgs{}
workspace := drone.Workspace{}
a := newApp()
plugin.Param("vargs", &vargs)
plugin.Param("workspace", &workspace)
if err := plugin.Parse(); err != nil {
err := a.loadVargs()
if err != nil {
fmt.Println(err)
os.Exit(1)
}
if len(vargs.Key) == 0 || len(vargs.Secret) == 0 || len(vargs.Bucket) == 0 {
return
err = a.sanitizeInputs()
if err != nil {
fmt.Println(err)
os.Exit(1)
}
a.createClient()
a.createSyncJobs()
a.createInvalidateJob()
a.runJobs()
fmt.Println("done!")
}
func newApp() *app {
return &app{
vargs: &PluginArgs{},
workspace: &drone.Workspace{},
jobs: make([]job, 1, 1),
}
}
func (a *app) loadVargs() error {
plugin.Param("vargs", a.vargs)
plugin.Param("workspace", a.workspace)
err := plugin.Parse()
return err
}
func (a *app) createClient() {
a.client = NewAWS(*a.vargs)
}
func (a *app) sanitizeInputs() error {
vargs := a.vargs
workspace := a.workspace
if len(a.vargs.Key) == 0 || len(a.vargs.Secret) == 0 || len(a.vargs.Bucket) == 0 {
return errors.New(MissingAwsValuesMessage)
}
if len(vargs.Region) == 0 {
@ -57,15 +105,19 @@ func main() {
vargs.Target = vargs.Target[1:]
}
client := NewAWS(vargs)
remote, err := client.List(vargs.Target)
return nil
}
func (a *app) createSyncJobs() {
vargs := a.vargs
remote, err := a.client.List(vargs.Target)
if err != nil {
fmt.Println(err)
os.Exit(1)
}
local := make([]string, 1, 1)
jobs := make([]job, 1, 1)
err = filepath.Walk(vargs.Source, func(path string, info os.FileInfo, err error) error {
if err != nil || info.IsDir() {
return err
@ -79,7 +131,7 @@ func main() {
}
}
local = append(local, localPath)
jobs = append(jobs, job{
a.jobs = append(a.jobs, job{
local: filepath.Join(vargs.Source, localPath),
remote: filepath.Join(vargs.Target, localPath),
action: "upload",
@ -95,51 +147,59 @@ func main() {
for path, location := range vargs.Redirects {
path = strings.TrimPrefix(path, "/")
local = append(local, path)
jobs = append(jobs, job{
a.jobs = append(a.jobs, job{
local: path,
remote: location,
action: "redirect",
})
}
if (a.vargs.Delete) {
for _, r := range remote {
found := false
for _, l := range local {
if l == r {
found = true
break
}
}
for _, r := range remote {
found := false
for _, l := range local {
if l == r {
found = true
break
}
}
if !found {
a.jobs = append(a.jobs, job{
local: "",
remote: r,
action: "delete",
})
}
}
}
}
if !found {
jobs = append(jobs, job{
local: "",
remote: r,
action: "delete",
})
}
}
if len(vargs.CloudFrontDistribution) > 0 {
jobs = append(jobs, job{
func (a *app) createInvalidateJob() {
if len(a.vargs.CloudFrontDistribution) > 0 {
a.jobs = append(a.jobs, job{
local: "",
remote: filepath.Join("/", vargs.Target, "*"),
remote: filepath.Join("/", a.vargs.Target, "*"),
action: "invalidateCloudFront",
})
}
}
func (a *app) runJobs() {
vargs := a.vargs
client := a.client
jobChan := make(chan struct{}, maxConcurrent)
results := make(chan *result, len(jobs))
results := make(chan *result, len(a.jobs))
fmt.Printf("Synchronizing with bucket \"%s\"\n", vargs.Bucket)
for _, j := range jobs {
for _, j := range a.jobs {
jobChan <- struct{}{}
go func(j job) {
var err error
if j.action == "upload" {
err = client.Upload(j.local, j.remote)
} else if j.action == "redirect" {
err = client.Redirect(j.local, j.remote)
} else if j.action == "delete" && vargs.Delete {
} else if j.action == "delete" {
err = client.Delete(j.remote)
} else if j.action == "invalidateCloudFront" {
client.Invalidate(j.remote)
@ -151,15 +211,13 @@ func main() {
}(j)
}
for _ = range jobs {
for _ = range a.jobs {
r := <-results
if r.err != nil {
fmt.Printf("ERROR: failed to %s %s to %s: %+v\n", r.j.action, r.j.local, r.j.remote, r.err)
os.Exit(1)
}
}
fmt.Println("done!")
}
func debug(format string, args ...interface{}) {