package list

import (
	"strings"

	"github.com/MakeNowJust/heredoc/v2"
	"github.com/spf13/cobra"

	gitlab "gitlab.com/gitlab-org/api/client-go/v2"

	"gitlab.com/gitlab-org/cli/internal/cmdutils"
	"gitlab.com/gitlab-org/cli/internal/iostreams"
	"gitlab.com/gitlab-org/cli/internal/mcpannotations"
	"gitlab.com/gitlab-org/cli/internal/tableprinter"
	"gitlab.com/gitlab-org/cli/internal/utils"
)

type options struct {
	gitlabClient func() (*gitlab.Client, error)
	io           *iostreams.IOStreams

	showKeyIDs   bool
	outputFormat string
}

func NewCmdList(f cmdutils.Factory) *cobra.Command {
	opts := &options{
		io:           f.IO(),
		gitlabClient: f.GitLabClient,
	}
	cmd := &cobra.Command{
		Use:   "list",
		Short: "Get a list of GPG keys for the currently authenticated user.",
		Long: heredoc.Docf(`Each row shows the key and its creation date. Pass %[1]s--show-id%[1]s to
		also display the key ID, which the %[1]sget%[1]s and %[1]sdelete%[1]s commands accept
		as an argument.
		`, "`"),
		Example: heredoc.Doc(`
			# List your GPG keys
			glab gpg-key list

			# Include the key ID in the output
			glab gpg-key list --show-id`),
		Args: cobra.MaximumNArgs(1),
		Annotations: map[string]string{
			mcpannotations.Safe: "true",
		},
		RunE: func(cmd *cobra.Command, args []string) error {
			return opts.run()
		},
	}

	cmd.Flags().BoolVarP(&opts.showKeyIDs, "show-id", "", false, "Shows IDs of GPG keys.")
	cmdutils.EnableJSONOutput(cmd, opts.io, &opts.outputFormat)

	return cmd
}

func (o *options) run() error {
	client, err := o.gitlabClient()
	if err != nil {
		return err
	}

	keys, _, err := client.Users.ListGPGKeys()
	if err != nil {
		return cmdutils.WrapError(err, "failed to list GPG keys.")
	}

	if o.outputFormat == "json" {
		return o.io.PrintJSON(keys)
	}

	cs := o.io.Color()
	table := tableprinter.NewTablePrinter()
	isTTy := o.io.IsOutputTTY()

	if len(keys) > 0 {
		if o.showKeyIDs {
			table.AddRow("ID", "Key", "Created At")
		} else {
			table.AddRow("Key", "Created At")
		}
	}

	for _, key := range keys {
		createdAt := key.CreatedAt.String()
		if isTTy {
			createdAt = utils.TimeToPrettyTimeAgo(*key.CreatedAt)
		}
		// Replace newlines in key with spaces for better table display
		keyDisplay := strings.ReplaceAll(key.Key, "\n", " ")
		if o.showKeyIDs {
			table.AddRow(key.ID, keyDisplay, cs.Gray(createdAt))
		} else {
			table.AddRow(keyDisplay, cs.Gray(createdAt))
		}
	}

	o.io.LogInfo(table.String())

	return nil
}
