package cmd
import (
"fmt"
"io"
"strings"
"github.com/go-openapi/swag"
"github.com/spf13/cobra"
"github.com/vmware/dispatch/pkg/client"
"golang.org/x/net/context"
"github.com/vmware/dispatch/pkg/api/v1"
"github.com/vmware/dispatch/pkg/dispatchcli/i18n"
)
var (
createEventDriverLong = i18n.T(
`Create dispatch event driver`)
createEventDriverExample = i18n.T(``)
createEventDriverConfig []string
createEventDriverSecrets []string
createEventDriverName string
)
func NewCmdCreateEventDriver(out io.Writer, errOut io.Writer) *cobra.Command {
cmd := &cobra.Command{
Use: "eventdriver DRIVER_TYPE [--name DRIVER_NAME] [--set KEY=VALUE] [--secret SECRET_NAME]",
Short: i18n.T("Create event driver"),
Long: createEventDriverLong,
Example: createEventDriverExample,
Args: cobra.ExactArgs(1),
Aliases: []string{"eventdrivers", "event-driver", "event-drivers"},
Run: func(cmd *cobra.Command, args []string) {
c := eventManagerClient()
err := createEventDriver(out, errOut, cmd, args, c)
CheckErr(err)
},
}
cmd.Flags().StringVar(&createEventDriverName, "name", "", "name for the event driver. will be automatically generated if not specified.")
cmd.Flags().StringArrayVarP(&createEventDriverConfig, "set", "s", []string{}, "set event driver configurations, default: empty")
cmd.Flags().StringArrayVar(&createEventDriverSecrets, "secret", []string{}, "Configuration passed via secrets, can be specified multiple times or a comma-delimited string")
return cmd
}
func CallCreateEventDriver(c client.EventsClient) ModelAction {
return func(driver interface{}) error {
ev := driver.(*v1.EventDriver)
created, err := c.CreateEventDriver(context.TODO(), "", ev)
if err != nil {
return err
}
*ev = *created
return nil
}
}
func createEventDriver(out, errOut io.Writer, cmd *cobra.Command, args []string, c client.EventsClient) error {
driverType := args[0]
var driverConfig []*v1.Config
for _, conf := range createEventDriverConfig {
result := strings.Split(conf, "=")
switch len(result) {
case 1:
driverConfig = append(driverConfig, &v1.Config{
Key: result[0],
})
case 2:
driverConfig = append(driverConfig, &v1.Config{
Key: result[0],
Value: result[1],
})
default:
fmt.Fprint(errOut, "Invalid Configuration Format, should be --set key=value or --set key")
}
}
eventDriver := &v1.EventDriver{
Name: swag.String(resourceName(createEventDriverName)),
Type: swag.String(driverType),
Config: driverConfig,
Secrets: createEventDriverSecrets,
Tags: []*v1.Tag{},
}
err := CallCreateEventDriver(c)(eventDriver)
if err != nil {
if *eventDriver.Name == "vcenter" {
fmt.Fprintf(errOut,
"Note: this version of Dispatch does not have built-in vcenter event driver anymore."+
"Please go to https://github.com/dispatchframework/dispatch-events-vcenter find out more "+
"how to create a vcenter driver type.")
}
return err
}
if w, err := formatOutput(out, false, eventDriver); w {
return err
}
fmt.Fprintf(out, "Created event driver: %s\n", *eventDriver.Name)
return nil
}