ory-am/hydra

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

package cmd

import (
    "github.com/spf13/cobra"

    "github.com/ory/hydra/v2/jwk"
    "github.com/ory/x/flagx"

    "github.com/ory/hydra/v2/cmd/cliclient"
    "github.com/ory/x/cmdx"
)

func NewGetJWKSCmd() *cobra.Command {
    cmd := &cobra.Command{
        Use:     "jwk set-1 [set-2] ...",
        Aliases: []string{"jwks"},
        Args:    cobra.MinimumNArgs(1),
        Short:   "Get one or more JSON Web Key Set by its ID(s)",
        Long:    `This command gets all the details about an JSON Web Key. You can use this command in combination with jq.`,
        Example: `To get the JSON Web Key Set's use, run:

    {{ .CommandPath }} <set-id> | jq -r '.[].use'

To get the JSON Web Key Set as only public keys:

    {{ .CommandPath }} --public <set-id>'
`,
        RunE: func(cmd *cobra.Command, args []string) error {
            m, _, err := cliclient.NewClient(cmd)
            if err != nil {
                return err
            }

            var sets outputJSONWebKeyCollection
            for _, set := range args {
                key, _, err := m.JwkAPI.GetJsonWebKeySet(cmd.Context(), set).Execute() //nolint:bodyclose
                if err != nil {
                    return cmdx.PrintOpenAPIError(cmd, err)
                }

                sets.Keys = append(sets.Keys, key.Keys...)
            }

            if flagx.MustGetBool(cmd, "public") {
                sets.Keys, err = jwk.OnlyPublicSDKKeys(sets.Keys)
                if err != nil {
                    return err
                }
            }

            if len(sets.Keys) == 1 {
                cmdx.PrintRow(cmd, outputJsonWebKey{Set: args[0], JsonWebKey: sets.Keys[0]})
            } else if len(sets.Keys) > 1 {
                cmdx.PrintTable(cmd, sets)
            }

            return nil
        },
    }
    cmd.Flags().Bool("public", false, "Only return public keys")
    return cmd
}