96 lines
2.8 KiB
Go
96 lines
2.8 KiB
Go
package cmd
|
|
|
|
import (
|
|
"bytes"
|
|
"log"
|
|
"os"
|
|
|
|
"github.com/BBVA/kapow/internal/client"
|
|
|
|
"github.com/spf13/cobra"
|
|
)
|
|
|
|
// RouteCmd is the command line interface for kapow route handling
|
|
var RouteCmd = &cobra.Command{
|
|
Use: "route [action]",
|
|
}
|
|
|
|
func init() {
|
|
var routeListCmd = &cobra.Command{
|
|
Use: "list [flags]",
|
|
Short: "List the current Kapow! routes",
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
controlURL, _ := cmd.Flags().GetString("control-url")
|
|
|
|
if err := client.ListRoutes(controlURL, os.Stdout); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
},
|
|
}
|
|
routeListCmd.Flags().String("control-url", getEnv("KAPOW_CONTROL_URL", "http://localhost:8081"), "Kapow! control interface URL")
|
|
|
|
// TODO: Manage args for url_pattern and command_file (2 exact args)
|
|
var routeAddCmd = &cobra.Command{
|
|
Use: "add [flags] url_pattern [command_file]",
|
|
Short: "Add a route",
|
|
Args: cobra.RangeArgs(1, 2),
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
controlURL, _ := cmd.Flags().GetString("control-url")
|
|
method, _ := cmd.Flags().GetString("method")
|
|
command, _ := cmd.Flags().GetString("command")
|
|
entrypoint, _ := cmd.Flags().GetString("entrypoint")
|
|
urlPattern := args[0]
|
|
|
|
if len(args) > 1 && command == "" {
|
|
commandFile := args[1]
|
|
buf := new(bytes.Buffer)
|
|
if commandFile == "-" {
|
|
_, err := buf.ReadFrom(os.Stdin)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
} else {
|
|
file, err := os.Open(commandFile)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
defer file.Close()
|
|
_, err = buf.ReadFrom(file)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
command = buf.String()
|
|
}
|
|
|
|
// TODO: Read command from parameter, file or stdin
|
|
if err := client.AddRoute(controlURL, urlPattern, method, entrypoint, command, os.Stdout); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
},
|
|
}
|
|
// TODO: Add default values for flags and remove path flag
|
|
routeAddCmd.Flags().String("control-url", getEnv("KAPOW_CONTROL_URL", "http://localhost:8081"), "Kapow! control interface URL")
|
|
routeAddCmd.Flags().StringP("method", "X", "GET", "HTTP method to accept")
|
|
routeAddCmd.Flags().StringP("entrypoint", "e", "/bin/sh -c", "Command to execute")
|
|
routeAddCmd.Flags().StringP("command", "c", "", "Command to pass to the shell")
|
|
|
|
var routeRemoveCmd = &cobra.Command{
|
|
Use: "remove [flags] route_id",
|
|
Short: "Remove the given route",
|
|
Args: cobra.ExactArgs(1),
|
|
Run: func(cmd *cobra.Command, args []string) {
|
|
controlURL, _ := cmd.Flags().GetString("control-url")
|
|
|
|
if err := client.RemoveRoute(controlURL, args[0]); err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
},
|
|
}
|
|
routeRemoveCmd.Flags().String("control-url", getEnv("KAPOW_CONTROL_URL", "http://localhost:8081"), "Kapow! control interface URL")
|
|
|
|
RouteCmd.AddCommand(routeListCmd)
|
|
RouteCmd.AddCommand(routeAddCmd)
|
|
RouteCmd.AddCommand(routeRemoveCmd)
|
|
}
|