feat: initial commit

internal/command/admin/layer/create.go

@@ -0,0 +1,72 @@
+package layer
+
+import (
+	"os"
+
+	"forge.cadoles.com/cadoles/bouncer/internal/client"
+	"forge.cadoles.com/cadoles/bouncer/internal/command/admin/apierr"
+	clientFlag "forge.cadoles.com/cadoles/bouncer/internal/command/admin/flag"
+	"forge.cadoles.com/cadoles/bouncer/internal/command/admin/layer/flag"
+	layerFlag "forge.cadoles.com/cadoles/bouncer/internal/command/admin/layer/flag"
+	proxyFlag "forge.cadoles.com/cadoles/bouncer/internal/command/admin/proxy/flag"
+	"forge.cadoles.com/cadoles/bouncer/internal/format"
+	"github.com/pkg/errors"
+	"github.com/urfave/cli/v2"
+)
+
+func CreateCommand() *cli.Command {
+	return &cli.Command{
+		Name:  "create",
+		Usage: "Create layer",
+		Flags: layerFlag.WithLayerCreateFlags(),
+		Action: func(ctx *cli.Context) error {
+			baseFlags := clientFlag.GetBaseFlags(ctx)
+
+			token, err := clientFlag.GetToken(baseFlags)
+			if err != nil {
+				return errors.WithStack(apierr.Wrap(err))
+			}
+
+			layerName, err := flag.AssertLayerName(ctx)
+			if err != nil {
+				return errors.WithStack(err)
+			}
+
+			proxyName, err := proxyFlag.AssertProxyName(ctx)
+			if err != nil {
+				return errors.WithStack(err)
+			}
+
+			layerType, err := flag.AssertLayerType(ctx)
+			if err != nil {
+				return errors.WithStack(err)
+			}
+
+			layerOptions, err := flag.AssertLayerOptions(ctx)
+			if err != nil {
+				return errors.WithStack(err)
+			}
+
+			client := client.New(baseFlags.ServerURL, client.WithToken(token))
+
+			layer, err := client.CreateLayer(
+				ctx.Context,
+				proxyName,
+				layerName,
+				layerType,
+				layerOptions,
+			)
+			if err != nil {
+				return errors.WithStack(apierr.Wrap(err))
+			}
+
+			hints := layerHints(baseFlags.OutputMode)
+
+			if err := format.Write(baseFlags.Format, os.Stdout, hints, layer); err != nil {
+				return errors.WithStack(err)
+			}
+
+			return nil
+		},
+	}
+}

internal/command/admin/layer/delete.go

@@ -0,0 +1,62 @@
+package layer
+
+import (
+	"os"
+
+	"forge.cadoles.com/cadoles/bouncer/internal/client"
+	"forge.cadoles.com/cadoles/bouncer/internal/command/admin/apierr"
+	clientFlag "forge.cadoles.com/cadoles/bouncer/internal/command/admin/flag"
+	layerFlag "forge.cadoles.com/cadoles/bouncer/internal/command/admin/layer/flag"
+	proxyFlag "forge.cadoles.com/cadoles/bouncer/internal/command/admin/proxy/flag"
+	"forge.cadoles.com/cadoles/bouncer/internal/format"
+	"forge.cadoles.com/cadoles/bouncer/internal/store"
+	"github.com/pkg/errors"
+	"github.com/urfave/cli/v2"
+)
+
+func DeleteCommand() *cli.Command {
+	return &cli.Command{
+		Name:  "delete",
+		Usage: "Delete layer",
+		Flags: layerFlag.WithLayerFlags(),
+		Action: func(ctx *cli.Context) error {
+			baseFlags := clientFlag.GetBaseFlags(ctx)
+
+			token, err := clientFlag.GetToken(baseFlags)
+			if err != nil {
+				return errors.WithStack(apierr.Wrap(err))
+			}
+
+			proxyName, err := proxyFlag.AssertProxyName(ctx)
+			if err != nil {
+				return errors.WithStack(err)
+			}
+
+			layerName, err := layerFlag.AssertLayerName(ctx)
+			if err != nil {
+				return errors.WithStack(err)
+			}
+
+			client := client.New(baseFlags.ServerURL, client.WithToken(token))
+
+			layerName, err = client.DeleteLayer(ctx.Context, proxyName, layerName)
+			if err != nil {
+				return errors.WithStack(apierr.Wrap(err))
+			}
+
+			hints := format.Hints{
+				OutputMode: baseFlags.OutputMode,
+			}
+
+			if err := format.Write(baseFlags.Format, os.Stdout, hints, struct {
+				Name store.LayerName `json:"id"`
+			}{
+				Name: layerName,
+			}); err != nil {
+				return errors.WithStack(err)
+			}
+
+			return nil
+		},
+	}
+}

internal/command/admin/layer/flag/flag.go

@@ -0,0 +1,76 @@
+package flag
+
+import (
+	"encoding/json"
+
+	proxyFlag "forge.cadoles.com/cadoles/bouncer/internal/command/admin/proxy/flag"
+	"forge.cadoles.com/cadoles/bouncer/internal/store"
+	"github.com/pkg/errors"
+	"github.com/urfave/cli/v2"
+)
+
+const (
+	FlagLayerName    = "layer-name"
+	FlagLayerType    = "layer-type"
+	FlagLayerOptions = "layer-options"
+)
+
+func WithLayerFlags(flags ...cli.Flag) []cli.Flag {
+	baseFlags := proxyFlag.WithProxyFlags(
+		&cli.StringFlag{
+			Name:     FlagLayerName,
+			Usage:    "use `LAYER_NAME` as targeted layer",
+			Value:    "",
+			Required: true,
+		},
+	)
+
+	flags = append(flags, baseFlags...)
+
+	return flags
+}
+
+func WithLayerCreateFlags(flags ...cli.Flag) []cli.Flag {
+	return WithLayerFlags(
+		&cli.StringFlag{
+			Name:     FlagLayerType,
+			Usage:    "Set `LAYER_TYPE` as layer's type",
+			Value:    "",
+			Required: true,
+		},
+		&cli.StringFlag{
+			Name:  FlagLayerOptions,
+			Usage: "Set `LAYER_OPTIONS` as layer's options",
+			Value: "{}",
+		},
+	)
+}
+
+func AssertLayerName(ctx *cli.Context) (store.LayerName, error) {
+	rawLayerName := ctx.String(FlagLayerName)
+
+	name, err := store.ValidateName(rawLayerName)
+	if err != nil {
+		return "", errors.WithStack(err)
+	}
+
+	return store.LayerName(name), nil
+}
+
+func AssertLayerType(ctx *cli.Context) (store.LayerType, error) {
+	rawLayerType := ctx.String(FlagLayerType)
+
+	return store.LayerType(rawLayerType), nil
+}
+
+func AssertLayerOptions(ctx *cli.Context) (store.LayerOptions, error) {
+	rawLayerOptions := ctx.String(FlagLayerOptions)
+
+	layerOptions := store.LayerOptions{}
+
+	if err := json.Unmarshal([]byte(rawLayerOptions), &layerOptions); err != nil {
+		return nil, errors.WithStack(err)
+	}
+
+	return layerOptions, nil
+}

internal/command/admin/layer/get.go

@@ -0,0 +1,55 @@
+package layer
+
+import (
+	"os"
+
+	"forge.cadoles.com/cadoles/bouncer/internal/client"
+	"forge.cadoles.com/cadoles/bouncer/internal/command/admin/apierr"
+	clientFlag "forge.cadoles.com/cadoles/bouncer/internal/command/admin/flag"
+	layerFlag "forge.cadoles.com/cadoles/bouncer/internal/command/admin/layer/flag"
+	proxyFlag "forge.cadoles.com/cadoles/bouncer/internal/command/admin/proxy/flag"
+	"forge.cadoles.com/cadoles/bouncer/internal/format"
+	"github.com/pkg/errors"
+	"github.com/urfave/cli/v2"
+)
+
+func GetCommand() *cli.Command {
+	return &cli.Command{
+		Name:  "get",
+		Usage: "Get layer",
+		Flags: layerFlag.WithLayerFlags(),
+		Action: func(ctx *cli.Context) error {
+			baseFlags := clientFlag.GetBaseFlags(ctx)
+
+			token, err := clientFlag.GetToken(baseFlags)
+			if err != nil {
+				return errors.WithStack(apierr.Wrap(err))
+			}
+
+			proxyName, err := proxyFlag.AssertProxyName(ctx)
+			if err != nil {
+				return errors.WithStack(err)
+			}
+
+			layerName, err := layerFlag.AssertLayerName(ctx)
+			if err != nil {
+				return errors.WithStack(err)
+			}
+
+			client := client.New(baseFlags.ServerURL, client.WithToken(token))
+
+			layer, err := client.GetLayer(ctx.Context, proxyName, layerName)
+			if err != nil {
+				return errors.WithStack(apierr.Wrap(err))
+			}
+
+			hints := layerHints(baseFlags.OutputMode)
+
+			if err := format.Write(baseFlags.Format, os.Stdout, hints, layer); err != nil {
+				return errors.WithStack(err)
+			}
+
+			return nil
+		},
+	}
+}

internal/command/admin/layer/query.go

@@ -0,0 +1,69 @@
+package layer
+
+import (
+	"os"
+
+	"forge.cadoles.com/cadoles/bouncer/internal/client"
+	"forge.cadoles.com/cadoles/bouncer/internal/command/admin/apierr"
+	clientFlag "forge.cadoles.com/cadoles/bouncer/internal/command/admin/flag"
+	proxyFlag "forge.cadoles.com/cadoles/bouncer/internal/command/admin/proxy/flag"
+	"forge.cadoles.com/cadoles/bouncer/internal/format"
+	"forge.cadoles.com/cadoles/bouncer/internal/store"
+	"github.com/pkg/errors"
+	"github.com/urfave/cli/v2"
+)
+
+func QueryCommand() *cli.Command {
+	return &cli.Command{
+		Name:  "query",
+		Usage: "Query layers",
+		Flags: proxyFlag.WithProxyFlags(
+			&cli.StringSliceFlag{
+				Name:  "with-name",
+				Usage: "use `WITH_NAME` as query filter",
+			},
+		),
+		Action: func(ctx *cli.Context) error {
+			baseFlags := clientFlag.GetBaseFlags(ctx)
+
+			token, err := clientFlag.GetToken(baseFlags)
+			if err != nil {
+				return errors.WithStack(apierr.Wrap(err))
+			}
+
+			proxyName, err := proxyFlag.AssertProxyName(ctx)
+			if err != nil {
+				return errors.WithStack(err)
+			}
+
+			options := make([]client.QueryLayerOptionFunc, 0)
+
+			rawNames := ctx.StringSlice("with-name")
+			if rawNames != nil {
+				layerNames := func(names []string) []store.LayerName {
+					layerNames := make([]store.LayerName, len(names))
+					for i, name := range names {
+						layerNames[i] = store.LayerName(name)
+					}
+					return layerNames
+				}(rawNames)
+				options = append(options, client.WithQueryLayerNames(layerNames...))
+			}
+
+			client := client.New(baseFlags.ServerURL, client.WithToken(token))
+
+			proxies, err := client.QueryLayer(ctx.Context, proxyName, options...)
+			if err != nil {
+				return errors.WithStack(apierr.Wrap(err))
+			}
+
+			hints := layerHeaderHints(baseFlags.OutputMode)
+
+			if err := format.Write(baseFlags.Format, os.Stdout, hints, clientFlag.AsAnySlice(proxies)...); err != nil {
+				return errors.WithStack(err)
+			}
+
+			return nil
+		},
+	}
+}

internal/command/admin/layer/root.go

@@ -0,0 +1,19 @@
+package layer
+
+import (
+	"github.com/urfave/cli/v2"
+)
+
+func Root() *cli.Command {
+	return &cli.Command{
+		Name:  "layer",
+		Usage: "Execute actions related to layers",
+		Subcommands: []*cli.Command{
+			CreateCommand(),
+			GetCommand(),
+			QueryCommand(),
+			UpdateCommand(),
+			DeleteCommand(),
+		},
+	}
+}

internal/command/admin/layer/update.go

@@ -0,0 +1,91 @@
+package layer
+
+import (
+	"encoding/json"
+	"os"
+
+	"forge.cadoles.com/cadoles/bouncer/internal/client"
+	"forge.cadoles.com/cadoles/bouncer/internal/command/admin/apierr"
+	clientFlag "forge.cadoles.com/cadoles/bouncer/internal/command/admin/flag"
+	layerFlag "forge.cadoles.com/cadoles/bouncer/internal/command/admin/layer/flag"
+	proxyFlag "forge.cadoles.com/cadoles/bouncer/internal/command/admin/proxy/flag"
+	"forge.cadoles.com/cadoles/bouncer/internal/format"
+	"forge.cadoles.com/cadoles/bouncer/internal/store"
+	"github.com/pkg/errors"
+	"github.com/urfave/cli/v2"
+)
+
+func UpdateCommand() *cli.Command {
+	return &cli.Command{
+		Name:  "update",
+		Usage: "Update layer",
+		Flags: layerFlag.WithLayerFlags(
+			&cli.BoolFlag{
+				Name:  "enabled",
+				Usage: "Enable or disable proxy",
+			},
+			&cli.IntFlag{
+				Name:  "weight",
+				Usage: "Set `WEIGHT` as proxy's weight",
+			},
+			&cli.StringFlag{
+				Name:  "options",
+				Usage: "Set `OPTIONS` as proxy's options",
+			},
+		),
+		Action: func(ctx *cli.Context) error {
+			baseFlags := clientFlag.GetBaseFlags(ctx)
+
+			token, err := clientFlag.GetToken(baseFlags)
+			if err != nil {
+				return errors.WithStack(apierr.Wrap(err))
+			}
+
+			proxyName, err := proxyFlag.AssertProxyName(ctx)
+			if err != nil {
+				return errors.WithStack(err)
+			}
+
+			layerName, err := layerFlag.AssertLayerName(ctx)
+			if err != nil {
+				return errors.WithStack(err)
+			}
+
+			opts := &client.UpdateLayerOptions{}
+
+			if ctx.IsSet("options") {
+				var options store.LayerOptions
+				if err := json.Unmarshal([]byte(ctx.String("options")), &options); err != nil {
+					return errors.Wrap(err, "could not parse options")
+				}
+
+				opts.Options = &options
+			}
+
+			if ctx.IsSet("weight") {
+				weight := ctx.Int("weight")
+				opts.Weight = &weight
+			}
+
+			if ctx.IsSet("enabled") {
+				enabled := ctx.Bool("enabled")
+				opts.Enabled = &enabled
+			}
+
+			client := client.New(baseFlags.ServerURL, client.WithToken(token))
+
+			proxy, err := client.UpdateLayer(ctx.Context, proxyName, layerName, opts)
+			if err != nil {
+				return errors.WithStack(apierr.Wrap(err))
+			}
+
+			hints := layerHints(baseFlags.OutputMode)
+
+			if err := format.Write(baseFlags.Format, os.Stdout, hints, proxy); err != nil {
+				return errors.WithStack(err)
+			}
+
+			return nil
+		},
+	}
+}

internal/command/admin/layer/util.go

@@ -0,0 +1,33 @@
+package layer
+
+import (
+	"forge.cadoles.com/cadoles/bouncer/internal/format"
+	"forge.cadoles.com/cadoles/bouncer/internal/format/table"
+)
+
+func layerHeaderHints(outputMode format.OutputMode) format.Hints {
+	return format.Hints{
+		OutputMode: outputMode,
+		Props: []format.Prop{
+			format.NewProp("Name", "Name"),
+			format.NewProp("Type", "Type"),
+			format.NewProp("Enabled", "Enabled"),
+			format.NewProp("Weight", "Weight"),
+		},
+	}
+}
+
+func layerHints(outputMode format.OutputMode) format.Hints {
+	return format.Hints{
+		OutputMode: outputMode,
+		Props: []format.Prop{
+			format.NewProp("Name", "Name"),
+			format.NewProp("Type", "Type"),
+			format.NewProp("Enabled", "Enabled"),
+			format.NewProp("Weight", "Weight"),
+			format.NewProp("Options", "Options"),
+			format.NewProp("CreatedAt", "CreatedAt", table.WithCompactModeMaxColumnWidth(20)),
+			format.NewProp("UpdatedAt", "UpdatedAt", table.WithCompactModeMaxColumnWidth(20)),
+		},
+	}
+}