validate.go

package main

import (
	"fmt"

	"github.com/manifoldco/go-manifold"
	"github.com/urfave/cli"

	"github.com/manifoldco/manifold-cli/errs"
)

func validateTitle(cliCtx *cli.Context, option string, typeName ...string) (string, error) {
	val := cliCtx.String(option)
	if val == "" {
		return val, nil
	}

	if len(typeName) > 1 {
		panic("Only one typeName can be provided")
	} else if len(typeName) == 0 {
		typeName = make([]string, 1)
		typeName[0] = option
	}

	title := manifold.Name(val)
	if err := title.Validate(nil); err != nil {
		return "", errs.NewUsageExitError(cliCtx, cli.NewExitError(
			fmt.Sprintf("You've provided an invalid %s title!", typeName[0]), -1,
		))
	}

	return val, nil
}

func validateName(cliCtx *cli.Context, option string) (string, error) {
	val := cliCtx.String(option)
	if val == "" {
		return val, nil
	}

	name := manifold.Label(val)
	if err := name.Validate(nil); err != nil {
		return "", errs.NewUsageExitError(cliCtx, cli.NewExitError(
			fmt.Sprintf("You've provided an invalid %s!", option), -1,
		))
	}

	return val, nil
}

func requiredName(cliCtx *cli.Context, option string) (string, error) {
	val := cliCtx.String(option)
	if val == "" {
		return "", errs.NewUsageExitError(cliCtx, cli.NewExitError(
			fmt.Sprintf("--%s is required", option), -1,
		))
	}

	return validateName(cliCtx, option)
}

func requiredArgName(cliCtx *cli.Context, idx int, option string) (string, error) {
	args := cliCtx.Args()

	if len(args) < idx+1 {
		return "", errs.ErrTooFewArgs
	}

	val := args[idx]
	name := manifold.Label(val)
	email := manifold.Email(val)

	nErr := name.Validate(nil)
	eErr := email.Validate(nil)
	if nErr != nil && eErr != nil {
		return "", errs.NewUsageExitError(cliCtx, cli.NewExitError(
			fmt.Sprintf("You've provided an invalid %s!", option), -1,
		))
	}

	return val, nil
}

func optionalArgID(cliCtx *cli.Context, idx int, name string) (*manifold.ID, error) {
	args := cliCtx.Args()

	if len(args) < idx+1 {
		return nil, nil
	}

	val := args[idx]
	id, err := manifold.DecodeIDFromString(val)
	if err != nil {
		return nil, cli.NewExitError(fmt.Sprintf("Invalid %s ID: %s", name, err), -1)
	}

	return &id, nil
}

func optionalArgTitle(cliCtx *cli.Context, idx int, title string) (string, error) {
	args := cliCtx.Args()

	if len(args) < idx+1 {
		return "", nil
	}

	val := args[idx]
	l := manifold.Name(val)
	if err := l.Validate(nil); err != nil {
		return "", errs.NewUsageExitError(cliCtx, cli.NewExitError(
			fmt.Sprintf("You've provided an invalid %s!", title), -1,
		))
	}

	return val, nil
}

func optionalArgName(cliCtx *cli.Context, idx int, name string) (string, error) {
	args := cliCtx.Args()

	if len(args) < idx+1 {
		return "", nil
	}

	val := args[idx]
	l := manifold.Label(val)
	if err := l.Validate(nil); err != nil {
		return "", errs.NewUsageExitError(cliCtx, cli.NewExitError(
			fmt.Sprintf("You've provided an invalid %s!", name), -1,
		))
	}

	return val, nil
}

func optionalArgEmail(cliCtx *cli.Context, idx int, name string) (string, error) {
	args := cliCtx.Args()

	if len(args) < idx+1 {
		return "", nil
	}

	val := args[idx]
	n := manifold.Email(val)
	if err := n.Validate(nil); err != nil {
		return "", errs.NewUsageExitError(cliCtx, cli.NewExitError(
			fmt.Sprintf("You've provided an invalid %s email", name), -1,
		))
	}

	return val, nil
}

func optionalArgCode(cliCtx *cli.Context, idx int, name string) (string, error) {
	args := cliCtx.Args()

	if len(args) < idx+1 {
		return "", nil
	}

	val := args[idx]
	c := manifold.Code(val)
	if err := c.Validate(nil); err != nil {
		return "", errs.NewUsageExitError(cliCtx, cli.NewExitError(
			fmt.Sprintf("You've provided an invalid %s code!", name), -1,
		))
	}

	return val, nil
}

func maxOptionalArgsLength(cliCtx *cli.Context, size int) error {
	args := cliCtx.Args()

	if len(args) > size {
		return errs.ErrTooManyArgs
	}
	return nil
}

func exactArgsLength(cliCtx *cli.Context, size int) error {
	args := cliCtx.Args()

	if len(args) > size {
		return errs.ErrTooManyArgs
	} else if len(args) < size {
		return errs.ErrTooFewArgs
	}

	return nil
}

func validateTeamID(cliCtx *cli.Context) (*manifold.ID, error) {
	var teamID *manifold.ID
	cliTeamID := cliCtx.String("team-id")
	if cliTeamID != "" {
		id, err := manifold.DecodeIDFromString(cliTeamID)
		if err != nil {
			return nil, cli.NewExitError(fmt.Sprintf("Invalid team ID: %s", err), -1)
		}

		teamID = &id
	}

	return teamID, nil
}