// Copyright © 2023 Ory Corp
// SPDX-License-Identifier: Apache-2.0

package expand

import (
	"fmt"

	"github.com/ory/x/cmdx"
	"github.com/spf13/cobra"

	"github.com/ory/keto/cmd/client"
	"github.com/ory/keto/cmd/helpers"
	rts "github.com/ory/keto/gen/go/ory/keto/relation_tuples/v1alpha2"
	"github.com/ory/keto/ketoapi"
)

const FlagMaxDepth = "max-depth"

func NewExpandCmd() *cobra.Command {
	var maxDepth int32

	cmd := &cobra.Command{
		Use:   "expand <relation> <object_namespace>:<object_id>",
		Short: "Expand all subjects that have the relation on the object",
		Long: "Expand all subjects that have the relation on the object, including how access is granted.\n\n" +
			"Example:\n" +
			"	keto expand view Doc:readme\n\n",
		Args: cobra.RangeArgs(2, 3),
		RunE: func(cmd *cobra.Command, args []string) error {
			namespace, object, err := helpers.ParseNamespaceObject(cmd, args[1:])
			if err != nil {
				return err
			}

			conn, err := client.GetReadConn(cmd)
			if err != nil {
				return err
			}
			defer func() { _ = conn.Close() }()

			cl := rts.NewExpandServiceClient(conn)
			resp, err := cl.Expand(cmd.Context(), &rts.ExpandRequest{
				Subject:  rts.NewSubjectSet(namespace, object, args[0]),
				MaxDepth: maxDepth,
			})
			if err != nil {
				_, _ = fmt.Fprintf(cmd.ErrOrStderr(), "Could not make request: %s\n", err)
				return cmdx.FailSilently(cmd)
			}

			var tree *ketoapi.Tree[*ketoapi.RelationTuple]
			if resp.Tree != nil {
				tree = ketoapi.TreeFromProto[*ketoapi.RelationTuple](resp.Tree)
			}

			cmdx.PrintJSONAble(cmd, tree)
			format, err := cmd.Flags().GetString(cmdx.FlagFormat)
			if err != nil {
				return err
			}
			quiet, err := cmd.Flags().GetBool(cmdx.FlagQuiet)
			if err != nil {
				return err
			}
			switch format {
			case string(cmdx.FormatDefault), "":
				if tree == nil && !quiet {
					_, _ = fmt.Fprint(cmd.OutOrStdout(), "Got an empty tree. This probably means that the requested relation tuple is not present in Keto.")
				}
				_, _ = fmt.Fprintln(cmd.OutOrStdout())
			}
			return nil
		},
	}

	client.RegisterRemoteURLFlags(cmd.Flags())
	cmdx.RegisterJSONFormatFlags(cmd.Flags())
	cmdx.RegisterNoiseFlags(cmd.Flags())
	cmd.Flags().Int32VarP(&maxDepth, FlagMaxDepth, "d", 0, "Maximum depth of the tree to be returned. If the value is less than 1 or greater than the global max-depth then the global max-depth will be used instead.")

	return cmd
}

func RegisterCommandsRecursive(parent *cobra.Command) {
	parent.AddCommand(NewExpandCmd())
}