package cli

import (

"fmt"

"golang.org/x/xerrors"

"github.com/coder/coder/v2/cli/cliui"

"github.com/coder/coder/v2/codersdk"

"github.com/coder/serpent"

)

func (r *RootCmd) userOIDCClaims() *serpent.Command {

formatter := cliui.NewOutputFormatter(

cliui.ChangeFormatterData(

cliui.TableFormat([]claimRow{}, []string{"key", "value"}),

func(data any) (any, error) {

resp, ok := data.(codersdk.OIDCClaimsResponse)

if !ok {

return nil, xerrors.Errorf("expected type %T, got %T", resp, data)

}

rows := make([]claimRow, 0, len(resp.Claims))

for k, v := range resp.Claims {

rows = append(rows, claimRow{

Key: k,

Value: fmt.Sprintf("%v", v),

})

}

return rows, nil

},

),

cliui.JSONFormat(),

)

cmd := &serpent.Command{

Use: "oidc-claims",

Short: "Display the OIDC claims for the authenticated user.",

Long: FormatExamples(

Example{

Description: "Display your OIDC claims",

Command: "coder users oidc-claims",

},

Example{

Description: "Display your OIDC claims as JSON",

Command: "coder users oidc-claims -o json",

},

),

Middleware: serpent.Chain(

serpent.RequireNArgs(0),

),

Handler: func(inv *serpent.Invocation) error {

client, err := r.InitClient(inv)

if err != nil {

return err

}

resp, err := client.UserOIDCClaims(inv.Context())

if err != nil {

return xerrors.Errorf("get oidc claims: %w", err)

}

out, err := formatter.Format(inv.Context(), resp)

if err != nil {

return err

}

_, err = fmt.Fprintln(inv.Stdout, out)

return err

},

}

formatter.AttachOptions(&cmd.Options)

return cmd

}

type claimRow struct {

Key string `json:"-" table:"key,default_sort"`

Value string `json:"-" table:"value"`

}