formidable/internal/command/set.go

package command

import (
	"fmt"
	"os"

	"forge.cadoles.com/wpetit/formidable/internal/jsonpointer"
	"github.com/pkg/errors"
	"github.com/santhosh-tekuri/jsonschema/v5"
	_ "github.com/santhosh-tekuri/jsonschema/v5/httploader"
	"github.com/urfave/cli/v2"
)

func Set() *cli.Command {
	flags := []cli.Flag{
		&cli.BoolFlag{
			Name:    "force",
			Aliases: []string{"f"},
			Usage:   "Force data tree creation",
			Value:   false,
		},
	}

	flags = append(flags, commonFlags()...)

	return &cli.Command{
		Name:  "set",
		Usage: "Set value at specific path",
		Flags: flags,
		Action: func(ctx *cli.Context) error {
			schema, err := loadSchema(ctx)
			if err != nil {
				return errors.Wrap(err, "could not load schema")
			}

			_, values, err := loadData(ctx)
			if err != nil {
				return errors.Wrap(err, "could not load data")
			}

			rawPointer := ctx.Args().Get(0)
			rawValue := ctx.Args().Get(1)

			pointer := jsonpointer.New(rawPointer)

			var updatedValues interface{}

			force := ctx.Bool("force")

			if force {
				updatedValues, err = pointer.Force(values, rawValue)
				if err != nil {
					return errors.Wrapf(err, "could not force value '%v' to pointer '%v'", rawValue, rawPointer)
				}
			} else {
				updatedValues, err = pointer.Set(values, rawValue)
				if err != nil {
					return errors.Wrapf(err, "could not set value '%v' to pointer '%v'", rawValue, rawPointer)
				}
			}

			if err := schema.Validate(updatedValues); err != nil {
				if _, ok := err.(*jsonschema.ValidationError); ok {
					fmt.Printf("%#v\n", err)

					os.Exit(1)
				}

				return errors.Wrap(err, "could not validate resulting json")
			}

			if err := outputValues(ctx, updatedValues); err != nil {
				return errors.Wrap(err, "could not output updated values")
			}

			return nil
		},
	}
}