corteza/system/commands/settings.go

package commands

import (
	"context"
	"encoding/json"
	"os"
	"strings"

	"github.com/spf13/cobra"

	"github.com/cortezaproject/corteza-server/pkg/cli"
	"github.com/cortezaproject/corteza-server/pkg/settings"
	"github.com/cortezaproject/corteza-server/system/service"
)

func Settings(ctx context.Context, c *cli.Config) *cobra.Command {
	var (
		cmd = &cobra.Command{
			Use:   "settings",
			Short: "Settings management",
		}
	)

	list := &cobra.Command{
		Use:   "list",
		Short: "List all",
		Run: func(cmd *cobra.Command, args []string) {
			c.InitServices(ctx, c)

			prefix := cmd.Flags().Lookup("prefix").Value.String()
			if kv, err := service.DefaultSettings.FindByPrefix(ctx, prefix); err != nil {
				cli.HandleError(err)
			} else {
				var maxlen int
				for _, v := range kv {
					if l := len(v.Name); l > maxlen {
						maxlen = l
					}
				}

				for _, v := range kv {
					cmd.Printf("%s%s\t%v\n", v.Name, strings.Repeat(" ", maxlen-len(v.Name)), v.Value)
				}
			}
		},
	}

	list.Flags().String("prefix", "", "Filter settings by prefix")

	get := &cobra.Command{
		Use: "get [key to get, ...]",

		Short: "Get value (raw JSON) for a specific key",
		Args:  cobra.ExactArgs(1),
		Run: func(cmd *cobra.Command, args []string) {
			c.InitServices(ctx, c)

			if v, err := service.DefaultSettings.Get(ctx, args[0], 0); err != nil {
				cli.HandleError(err)
			} else if v != nil {
				cmd.Printf("%v\n", v.Value)
			}
		},
	}

	set := &cobra.Command{
		Use:   "set [key to set] [value]",
		Short: "Set value (raw JSON) for a specific key",
		Args:  cobra.ExactArgs(2),
		Run: func(cmd *cobra.Command, args []string) {
			c.InitServices(ctx, c)

			value := args[1]
			v := &settings.Value{
				Name: args[0],
			}

			if err := v.SetRawValue(value); err != nil {
				cli.HandleError(err)
			}

			cli.HandleError(service.DefaultSettings.Set(ctx, v))
		},
	}

	imp := &cobra.Command{
		Use:   "import [file]",
		Short: "Import settings as JSON from stdin or file",
		Args:  cobra.MaximumNArgs(1),
		Run: func(cmd *cobra.Command, args []string) {
			c.InitServices(ctx, c)

			var (
				fh  *os.File
				err error
			)

			if len(args) > 0 {
				fh, err = os.Open(args[0])
				cli.HandleError(err)
			} else {
				fh = os.Stdin
			}

			var (
				decoder = json.NewDecoder(fh)
				input   = map[string]interface{}{}
				vv      settings.ValueSet
			)

			cli.HandleError(decoder.Decode(&input))

			for k, v := range input {
				val := &settings.Value{Name: k}

				cli.HandleError(val.SetValue(v))
				vv = append(vv, val)
			}

			if len(vv) > 0 {
				cli.HandleError(service.DefaultSettings.BulkSet(ctx, vv))
			}
		},
	}

	exp := &cobra.Command{
		Use:   "export [file]",
		Short: "Import settings as JSON to stdout or file",
		Args:  cobra.MaximumNArgs(1),
		Run: func(cmd *cobra.Command, args []string) {
			c.InitServices(ctx, c)

			var (
				fh  *os.File
				err error
			)

			if len(args) > 0 {
				fh, err = os.Create(args[0])
				cli.HandleError(err)
			} else {
				fh = os.Stdin
			}

			var (
				encoder = json.NewEncoder(fh)
			)

			encoder.SetIndent("", "  ")

			if vv, err := service.DefaultSettings.FindByPrefix(ctx); err != nil {
				cli.HandleError(err)
			} else {
				cli.HandleError(encoder.Encode(vv.KV()))
			}
		},
	}

	del := &cobra.Command{
		Use:   "delete [keys, ...]",
		Short: "Set value (raw JSON) for a specific key (or by prefix)",
		Args:  cobra.MinimumNArgs(0),
		Run: func(cmd *cobra.Command, args []string) {
			c.InitServices(ctx, c)

			var names = []string{}

			if prefix := cmd.Flags().Lookup("prefix").Value.String(); len(prefix) > 0 {
				if vv, err := service.DefaultSettings.FindByPrefix(ctx); err != nil {
					cli.HandleError(err)
				} else {
					_ = vv.Walk(func(v *settings.Value) error {
						names = append(names, v.Name)
						return nil
					})
				}
			} else if len(args) > 0 {
				names = args
			}

			for a := 0; a < len(names); a++ {
				cli.HandleError(service.DefaultSettings.Delete(ctx, names[a], 0))
			}
		},
	}

	del.Flags().String("prefix", "", "Filter settings by prefix")

	cmd.AddCommand(
		list,
		get,
		set,
		del,
		imp,
		exp,
	)

	return cmd
}