drone-cli/drone/encrypt/encrypt.go

package encrypt

import (
	"fmt"
	"io/ioutil"
	"strings"

	"github.com/drone/drone-cli/drone/internal"
	"github.com/drone/drone-go/drone"

	"github.com/urfave/cli"
)

// Command is an encryption cli.Command
var Command = cli.Command{
	Name:      "encrypt",
	Usage:     "encrypt a secret",
	ArgsUsage: "<repo/name> <string>",
	Action:    encryptSecret,
	Flags: []cli.Flag{
		cli.BoolFlag{
			Name:  "allow-pull-request",
			Usage: "permit read access to pull requests",
		},
		cli.BoolFlag{
			Name:  "allow-push-on-pull-request",
			Usage: "permit write access to pull requests (e.g. allow docker push)",
		},
	},
}

func encryptSecret(c *cli.Context) error {
	repo := c.Args().First()
	owner, name, err := internal.ParseRepo(repo)
	if err != nil {
		return err
	}

	client, err := internal.NewClient(c)
	if err != nil {
		return err
	}

	plaintext := c.Args().Get(1)
	if strings.HasPrefix(plaintext, "@") {
		path := strings.TrimPrefix(plaintext, "@")
		data, err := ioutil.ReadFile(path)
		if err != nil {
			return err
		}
		plaintext = string(data)
	}

	secret := &drone.Secret{
		Data:            plaintext,
		PullRequest:     c.Bool("allow-pull-request"),
		PullRequestPush: c.Bool("allow-push-on-pull-request"),
	}
	encrypted, err := client.Encrypt(owner, name, secret)
	if err != nil {
		return err
	}
	fmt.Println(encrypted)
	return nil
}