112 lines
3 KiB
Go
112 lines
3 KiB
Go
package main
|
|
|
|
import (
|
|
"fmt"
|
|
"os"
|
|
"path/filepath"
|
|
"regexp"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
func parseRepo(str string) (host, owner, repo string) {
|
|
var parts = strings.Split(str, "/")
|
|
if len(parts) != 3 {
|
|
return
|
|
}
|
|
host = parts[0]
|
|
owner = parts[1]
|
|
repo = parts[2]
|
|
return
|
|
}
|
|
|
|
// getGoPath checks the source codes absolute path
|
|
// in reference to the host operating system's GOPATH
|
|
// to correctly determine the code's package path. This
|
|
// is Go-specific, since Go code must exist in
|
|
// $GOPATH/src/github.com/{owner}/{name}
|
|
func getGoPath(dir string) (string, bool) {
|
|
path := os.Getenv("GOPATH")
|
|
if len(path) == 0 {
|
|
return "", false
|
|
}
|
|
// append src to the GOPATH, since
|
|
// the code will be stored in the src dir
|
|
path = filepath.Join(path, "src")
|
|
if !filepath.HasPrefix(dir, path) {
|
|
return "", false
|
|
}
|
|
|
|
// remove the prefix from the directory
|
|
// this should leave us with the go package name
|
|
return dir[len(path):], true
|
|
}
|
|
|
|
var gopathExp = regexp.MustCompile("./src/(github.com/[^/]+/[^/]+|bitbucket.org/[^/]+/[^/]+|code.google.com/[^/]+/[^/]+)")
|
|
|
|
// getRepoPath checks the source codes absolute path
|
|
// on the host operating system in an attempt
|
|
// to correctly determine the code's package path. This
|
|
// is Go-specific, since Go code must exist in
|
|
// $GOPATH/src/github.com/{owner}/{name}
|
|
func getRepoPath(dir string) (path string, ok bool) {
|
|
// let's get the package directory based
|
|
// on the path in the host OS
|
|
indexes := gopathExp.FindStringIndex(dir)
|
|
if len(indexes) == 0 {
|
|
return
|
|
}
|
|
|
|
index := indexes[len(indexes)-1]
|
|
|
|
// if the dir is /home/ubuntu/go/src/github.com/foo/bar
|
|
// the index will start at /src/github.com/foo/bar.
|
|
// We'll need to strip "/src/" which is where the
|
|
// magic number 5 comes from.
|
|
index = strings.LastIndex(dir, "/src/")
|
|
return dir[index+5:], true
|
|
}
|
|
|
|
// GetRepoMap returns a map of enivronment variables that
|
|
// should be injected into the .drone.yml
|
|
func getParamMap(prefix string) map[string]string {
|
|
envs := map[string]string{}
|
|
|
|
for _, item := range os.Environ() {
|
|
env := strings.SplitN(item, "=", 2)
|
|
if len(env) != 2 {
|
|
continue
|
|
}
|
|
|
|
key := env[0]
|
|
val := env[1]
|
|
if strings.HasPrefix(key, prefix) {
|
|
envs[strings.TrimPrefix(key, prefix)] = val
|
|
}
|
|
}
|
|
return envs
|
|
}
|
|
|
|
// prints the time as a human readable string
|
|
func humanizeDuration(d time.Duration) string {
|
|
if seconds := int(d.Seconds()); seconds < 1 {
|
|
return "Less than a second"
|
|
} else if seconds < 60 {
|
|
return fmt.Sprintf("%d seconds", seconds)
|
|
} else if minutes := int(d.Minutes()); minutes == 1 {
|
|
return "About a minute"
|
|
} else if minutes < 60 {
|
|
return fmt.Sprintf("%d minutes", minutes)
|
|
} else if hours := int(d.Hours()); hours == 1 {
|
|
return "About an hour"
|
|
} else if hours < 48 {
|
|
return fmt.Sprintf("%d hours", hours)
|
|
} else if hours < 24*7*2 {
|
|
return fmt.Sprintf("%d days", hours/24)
|
|
} else if hours < 24*30*3 {
|
|
return fmt.Sprintf("%d weeks", hours/24/7)
|
|
} else if hours < 24*365*2 {
|
|
return fmt.Sprintf("%d months", hours/24/30)
|
|
}
|
|
return fmt.Sprintf("%f years", d.Hours()/24/365)
|
|
}
|