drone-cli/drone/template/template_set.go

package template

import (
	"errors"
	"io/ioutil"
	"strings"

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

	"github.com/urfave/cli"
)

var templateUpdateCmd = cli.Command{
	Name:      "update",
	Usage:     "update a template",
	ArgsUsage: "[namespace] [name] [data]",
	Action:    templateUpdate,
	Flags: []cli.Flag{
		cli.StringFlag{
			Name:  "name",
			Usage: "template name",
		},
		cli.StringFlag{
			Name:  "namespace",
			Usage: "organization name",
		},
		cli.StringFlag{
			Name:  "data",
			Usage: "template file data",
		},
	},
}

func templateUpdate(c *cli.Context) error {
	client, err := internal.NewClient(c)
	if err != nil {
		return err
	}

	namespace := c.String("namespace")
	if namespace == "" {
		return errors.New("missing namespace")
	}

	template := &drone.Template{
		Name: c.String("name"),
	}
	if strings.HasPrefix(c.String("data"), "@") {
		path := strings.TrimPrefix(c.String("data"), "@")
		out, ferr := ioutil.ReadFile(path)
		if ferr != nil {
			return ferr
		}
		template.Data = string(out)
	}
	_, err = client.TemplateUpdate(namespace, template.Name, template)
	return err
}