convert repo plugin to golang

plugins/common/common.go

@@ -0,0 +1,149 @@
+package common
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+	"regexp"
+	"strings"
+
+	sh "github.com/codeskyblue/go-sh"
+)
+
+// DokkuCmd represents a shell command to be run for dokku
+type DokkuCmd struct {
+	Env           map[string]string
+	Command       *exec.Cmd
+	CommandString string
+	Args          []string
+	ShowOutput    bool
+}
+
+// NewDokkuCmd creates a new DokkuCmd
+func NewDokkuCmd(command string) *DokkuCmd {
+	items := strings.Split(command, " ")
+	cmd := items[0]
+	args := items[1:]
+	return &DokkuCmd{
+		Command:       exec.Command(cmd, args...),
+		CommandString: command,
+		Args:          args,
+		ShowOutput:    true,
+	}
+}
+
+// Execute is a lightweight wrapper around exec.Command
+func (dc *DokkuCmd) Execute() bool {
+	env := os.Environ()
+	for k, v := range dc.Env {
+		env = append(env, fmt.Sprintf("%s=%s", k, v))
+	}
+	dc.Command.Env = env
+	if dc.ShowOutput {
+		dc.Command.Stdout = os.Stdout
+		dc.Command.Stderr = os.Stderr
+	}
+	err := dc.Command.Run()
+	if err != nil {
+		return false
+	}
+	return true
+}
+
+// VerifyAppName verifies app name format and app existence"
+func VerifyAppName(appName string) (err error) {
+	dokkuRoot := MustGetEnv("DOKKU_ROOT")
+	appRoot := strings.Join([]string{dokkuRoot, appName}, "/")
+	_, err = os.Stat(appRoot)
+	if os.IsNotExist(err) {
+		return fmt.Errorf("App %s does not exist: %v\n", appName, err)
+	}
+	r, _ := regexp.Compile("^[a-z].*")
+	if !r.MatchString(appName) {
+		return fmt.Errorf("App name (%s) must begin with lowercase alphanumeric character\n", appName)
+	}
+	return err
+}
+
+// MustGetEnv returns env variable or fails if it's not set
+func MustGetEnv(key string) string {
+	dokkuRoot := os.Getenv(key)
+	if dokkuRoot == "" {
+		LogFail(fmt.Sprintf("%s not set!", key))
+	}
+	return dokkuRoot
+}
+
+// LogFail is the failure log formatter
+// prints text to stderr and exits with status 1
+func LogFail(text string) {
+	fmt.Fprintln(os.Stderr, fmt.Sprintf("FAILED: %s", text))
+	os.Exit(1)
+}
+
+// GetDeployingAppImageName returns deploying image identifier for a given app, tag tuple. validate if tag is presented
+func GetDeployingAppImageName(args ...string) (imageName string) {
+	argArray := make([]string, 3)
+	for idx, arg := range args {
+		argArray[idx] = arg
+	}
+	appName := argArray[0]
+	imageTag := argArray[1]
+	imageRepo := argArray[2]
+
+	if appName == "" {
+		LogFail("(GetDeployingAppImageName) APP must not be empty")
+	}
+
+	b, err := sh.Command("plugn", "trigger", "deployed-app-repository", appName).Output()
+	if err != nil {
+		LogFail(err.Error())
+	}
+	imageRemoteRepository := string(b[:])
+
+	b, err = sh.Command("plugn", "trigger", "deployed-app-image-tag", appName).Output()
+	if err != nil {
+		LogFail(err.Error())
+	}
+	newImageTag := string(b[:])
+
+	b, err = sh.Command("plugn", "trigger", "deployed-app-image-repo", appName).Output()
+	if err != nil {
+		LogFail(err.Error())
+	}
+	newImageRepo := string(b[:])
+
+	if newImageRepo != "" {
+		imageRepo = newImageRepo
+	}
+	if newImageTag != "" {
+		imageTag = newImageTag
+	}
+	if imageRepo == "" {
+		imageRepo = GetAppImageRepo(appName)
+	}
+	if imageTag == "" {
+		imageTag = "latest"
+	}
+
+	imageName = fmt.Sprintf("%s%s:%s", imageRemoteRepository, imageRepo, imageTag)
+	if !VerifyImage(imageName) {
+		LogFail(fmt.Sprintf("app image (%s) not found", imageName))
+	}
+	return
+}
+
+// GetAppImageRepo is the central definition of a dokku image repo pattern
+func GetAppImageRepo(appName string) string {
+	return strings.Join([]string{"dokku", appName}, "/")
+}
+
+// VerifyImage returns true if docker image exists in local repo
+func VerifyImage(image string) bool {
+	imageCmd := NewDokkuCmd(fmt.Sprintf("docker inspect %s", image))
+	imageCmd.ShowOutput = false
+	if imageCmd.Execute() {
+		return true
+	}
+	return false
+}