plugin.go

package main

import (
	"encoding/json"
	"fmt"
	"io/ioutil"

	"github.com/jmccann/drone-artifactory/artifactory"
	"github.com/sirupsen/logrus"
)

type (
	// Action to perform
	Action struct {
		Name          string `json:"action"`
		RawArguments  json.RawMessage
		Arguments     interface{}
		ArgumentsFile string `json:"args_file"`
	}

	// Plugin struct
	Plugin struct {
		Actions []Action
		Config  artifactory.Config
	}
)

// Exec run the plugin
func (p Plugin) Exec() error {
	err := p.Config.Validate()

	if err != nil {
		return err
	}

	logrus.Info("Creating Client")
	arti, err := artifactory.New(p.Config)

	if err != nil {
		return err
	}

	logrus.Info("Pinging Artifactory")
	err = arti.Ping()

	if err != nil {
		return err
	}

	logrus.Info("Validating Actions")
	for idx := range p.Actions {
		err = parseArgs(&p.Actions[idx])

		if err != nil {
			return err
		}
	}

	logrus.Info("Running Actions")
	for _, action := range p.Actions {
		err = do(arti, action)

		if err != nil {
			return err
		}
	}

	return nil
}

func do(arti *artifactory.Artifactory, action Action) error {
	if action.Name == "copy" {
		return arti.Copy(action.Arguments.(artifactory.CopyArgs))
	}

	if action.Name == "delete" {
		return arti.Delete(action.Arguments.(artifactory.DeleteArgs))
	}

	if action.Name == "set-props" {
		return arti.SetProps(action.Arguments.(artifactory.PropsArgs))
	}

	if action.Name == "upload" {
		return arti.Upload(action.Arguments.(artifactory.UploadArgs))
	}

	return fmt.Errorf("action '%s' not supported", action)
}

func parseArgs(action *Action) error {
	var err error
	args := action.RawArguments

	if action.ArgumentsFile != "" {
		args, err = ioutil.ReadFile(action.ArgumentsFile)

		if err != nil {
			return err
		}
	}

	if action.Name == "copy" {
		var copyArgs artifactory.CopyArgs
		err = json.Unmarshal(args, &copyArgs)

		if err != nil {
			return err
		}

		action.Arguments = copyArgs
		return copyArgs.Validate()
	}

	if action.Name == "delete" {
		var deleteArgs artifactory.DeleteArgs
		err = json.Unmarshal(args, &deleteArgs)

		if err != nil {
			return err
		}

		action.Arguments = deleteArgs
		return deleteArgs.Validate()
	}

	if action.Name == "set-props" {
		var propsArgs artifactory.PropsArgs
		err = json.Unmarshal(args, &propsArgs)

		if err != nil {
			return err
		}

		action.Arguments = propsArgs
		return propsArgs.Validate()
	}

	if action.Name == "upload" {
		var uploadArgs artifactory.UploadArgs
		err := json.Unmarshal(args, &uploadArgs)

		if err != nil {
			return err
		}

		action.Arguments = uploadArgs
		return uploadArgs.Validate()
	}

	return fmt.Errorf("action '%s' not supported", action.Name)
}