main.go

package main

import (
	"errors"
	"fmt"
	"log"
	"os"
	"os/exec"
	"os/user"
	"path/filepath"
	"strings"

	"github.com/spf13/viper"
)

var Version string

const remoteName = "drone"

func execute(cmd *exec.Cmd) error {
	fmt.Println("+", strings.Join(cmd.Args, " "))
	cmd.Env = os.Environ()
	cmd.Stderr = os.Stderr
	cmd.Stdin = os.Stdin
	return cmd.Run()
}

func pluginExec() error {
	if viper.GetBool("skip-verify") {
		cmd := exec.Command("git", "config", "--global", "http.sslVerify", "false")
		if err := execute(cmd); err != nil {
			return err
		}
	}
	sshKey := viper.GetString("ssh-key")
	if sshKey == "" {
		return errors.New("missing ssh key")
	}
	home := "/root"
	if currentUser, err := user.Current(); err == nil {
		home = currentUser.HomeDir
	}
	sshpath := filepath.Join(home, ".ssh")
	if err := os.MkdirAll(sshpath, 0o700); err != nil {
		return err
	}
	_ = os.WriteFile(filepath.Join(sshpath, "config"), []byte("StrictHostKeyChecking no\n"), 0o700)
	err := os.WriteFile(filepath.Join(sshpath, "id_rsa"), []byte(sshKey), 0o600)
	if err != nil {
		return err
	}
	remote := viper.GetString("remote")
	if remote != "" {
		cmd := exec.Command("git", "remote", "add", remoteName, remote)
		if err := execute(cmd); err != nil {
			return err
		}
	}
	ref := ""
	switch os.Getenv("DRONE_BUILD_EVENT") {
	case "tag":
		ref = os.Getenv("DRONE_TAG")
		cmd := exec.Command("git", "tag", "-a", ref, "-m", "Ver: "+ref)
		if err := execute(cmd); err != nil {
			return err
		}
	case "push":
		ref = os.Getenv("DRONE_BRANCH")
	}
	if ref == "" {
		return errors.New("missing ref")
	}
	cmd := exec.Command("git", "push", remoteName, ref)
	if viper.GetBool("force") {
		cmd.Args = append(cmd.Args, "--force")
	}
	return execute(cmd)
}

func main() {
	viper.SetEnvPrefix("PLUGIN")
	viper.SetEnvKeyReplacer(strings.NewReplacer("-", "_"))
	viper.AutomaticEnv()
	if err := pluginExec(); err != nil {
		log.Fatal(err)
	}
}