package cli

import (
	"encoding/json"
	"fmt"
	"net/url"

	"github.com/spf13/cobra"

	"gitea.mynodes.xyz/ilker/llmgw/internal/control/adminapi"
)

func newRequestCmd(client func() *ctlClient, asJSON *bool) *cobra.Command {
	return &cobra.Command{
		Use:   "request ID",
		Short: "What one request went through: replica, queue, first token, tokens, guardrails",
		Long: "Every response carries its id in X-Request-Id, and a client may send its\n" +
			"own. Given one, this says which replica answered, how long admission held\n" +
			"it, how long the first token took and what it cost — the answer to \"why\n" +
			"was this slow\". The access log has the same fields under \"request\".",
		Args: cobra.ExactArgs(1),
		RunE: func(cmd *cobra.Command, args []string) error {
			var v adminapi.RequestView
			if err := client().get(cmd.Context(), "/api/v1/requests/"+url.PathEscape(args[0]), &v); err != nil {
				return err
			}
			if *asJSON {
				return json.NewEncoder(cmd.OutOrStdout()).Encode(v)
			}
			model := v.Model
			if v.ServedModel != "" && v.ServedModel != v.Model {
				model += " (served by " + v.ServedModel + ")"
			}
			out := cmd.OutOrStdout()
			fmt.Fprintf(out, "request   %s at %s\n", v.RequestID, v.At.UTC().Format("2006-01-02 15:04:05Z"))
			fmt.Fprintf(out, "model     %s\n", model)
			fmt.Fprintf(out, "replica   %s\n", or(v.Replica, "- (none answered)"))
			fmt.Fprintf(out, "status    %d\n", v.Status)
			fmt.Fprintf(out, "time      %d ms total: %d queued, %d to the first token\n", v.LatencyMs, v.QueueMs, v.TTFTMs)
			fmt.Fprintf(out, "tokens    %d in, %d out\n", v.PromptTokens, v.CompletionTokens)
			fmt.Fprintf(out, "who       key %s, project %s%s\n", v.KeyID, v.ProjectID, prefixed(", user ", v.UserID))
			if v.GuardAction != "" {
				fmt.Fprintf(out, "guard     %s by policy %s (%s), %d ms\n", v.GuardAction, v.GuardPolicy,
					or(v.GuardCategory, "-"), v.GuardMs)
			} else if v.GuardMs > 0 {
				fmt.Fprintf(out, "guard     allowed, %d ms\n", v.GuardMs)
			}
			return nil
		},
	}
}