WeKnora deployment source with the reviewed wiki identity, summary and recovery fixes

cli/cmd/session/view.go

@@ -0,0 +1,199 @@
+package sessioncmd
+
+import (
+	"context"
+	"fmt"
+	"time"
+
+	"github.com/spf13/cobra"
+
+	"github.com/Tencent/WeKnora/cli/internal/cmdutil"
+	"github.com/Tencent/WeKnora/cli/internal/iostreams"
+	sdk "github.com/Tencent/WeKnora/client"
+)
+
+const (
+	defaultFullLimit = 50
+	maxFullLimit     = 1000
+)
+
+// sessionViewFields enumerates the fields surfaced for `--format json` discovery on
+// `session view`. Mirrors sdk.Session json tags; adds the synthesized
+// `messages` projection surfaced by `--full`.
+var sessionViewFields = []string{
+	"id", "tenant_id", "title", "description", "created_at", "updated_at",
+	"messages",
+}
+
+type ViewOptions struct {
+	// Full instructs runView to fetch chat history via LoadMessages and
+	// render it after the session metadata.
+	Full bool
+	// Limit caps the number of messages loaded when Full is true.
+	// Must be 1..maxFullLimit.
+	Limit int
+	// LimitSet records whether the caller explicitly set --limit, so we
+	// can reject `--limit` without `--full` (vs. silently ignoring the
+	// default).
+	LimitSet bool
+}
+
+// ViewService is the narrow SDK surface this command depends on. LoadMessages
+// is only invoked under --full but lives on the same interface so the runView
+// dependency surface stays minimal.
+type ViewService interface {
+	GetSession(ctx context.Context, id string) (*sdk.Session, error)
+	LoadMessages(ctx context.Context, sessionID string, limit int, beforeTime *time.Time, opts ...sdk.ResourceURLOptions) ([]sdk.Message, error)
+}
+
+// NewCmdView builds `weknora session view <id>`. Renders session metadata
+// only by default. With `--full`, also loads the chat history via
+// `LoadMessages` and renders messages (or projects them into the JSON
+// payload under `messages`).
+func NewCmdView(f *cmdutil.Factory) *cobra.Command {
+	opts := &ViewOptions{Limit: defaultFullLimit}
+	cmd := &cobra.Command{
+		Use:   "view <session-id>",
+		Short: "Show a chat session by ID",
+		Long: `Show a chat session.
+
+By default renders the session metadata (id, title, description, timestamps).
+
+Pass --full to also load the chat history (LoadMessages SDK call). Use
+--limit to cap the number of messages loaded (1..1000, default 50).
+--limit without --full is rejected as input.invalid_argument.`,
+		Args: cobra.ExactArgs(1),
+		RunE: func(c *cobra.Command, args []string) error {
+			opts.LimitSet = c.Flags().Changed("limit")
+			fopts, err := cmdutil.CheckFormatFlag(c)
+			if err != nil {
+				return err
+			}
+			fopts.ResolveDefault(iostreams.IO.IsStdoutTTY())
+			// Validate static input before building the client so a bad --limit
+			// returns input.invalid_argument (exit 5), not an auth error (exit 3).
+			if err := validateViewOpts(opts); err != nil {
+				return err
+			}
+			cli, err := f.Client()
+			if err != nil {
+				return err
+			}
+			return runView(c.Context(), opts, fopts, cli, args[0])
+		},
+	}
+	cmd.Flags().BoolVar(&opts.Full, "full", false, "Also load chat history via LoadMessages")
+	cmd.Flags().IntVarP(&opts.Limit, "limit", "L", defaultFullLimit, "Max messages to load when --full is set (1..1000)")
+	cmdutil.AddFormatFlag(cmd, sessionViewFields...)
+	cmdutil.SetAgentHelp(cmd, cmdutil.AgentHelp{
+		UsedFor:       "fetch one chat session by id; --full also loads its message history",
+		RequiredFlags: []string{"<session-id> (positional)"},
+		Examples:      []string{"weknora session view sess_abc", "weknora session view sess_abc --full --limit 50"},
+		Output:        "envelope.data is the session object; with --full it also carries the loaded messages",
+	})
+	return cmd
+}
+
+// validateViewOpts checks the --limit/--full invariants. Called from RunE
+// before the client is built (so a bad value surfaces as exit 5, not an auth
+// error) and at runView's top for direct callers; idempotent.
+func validateViewOpts(opts *ViewOptions) error {
+	if !opts.Full && opts.LimitSet {
+		return &cmdutil.Error{
+			Code:    cmdutil.CodeInputInvalidArgument,
+			Message: "--limit requires --full",
+		}
+	}
+	if opts.Full {
+		if opts.Limit < 1 || opts.Limit > maxFullLimit {
+			return &cmdutil.Error{
+				Code:    cmdutil.CodeInputInvalidArgument,
+				Message: fmt.Sprintf("--limit must be in 1..%d, got %d", maxFullLimit, opts.Limit),
+			}
+		}
+	}
+	return nil
+}
+
+func runView(ctx context.Context, opts *ViewOptions, fopts *cmdutil.FormatOptions, svc ViewService, id string) error {
+	if err := validateViewOpts(opts); err != nil {
+		return err
+	}
+
+	s, err := svc.GetSession(ctx, id)
+	if err != nil {
+		return cmdutil.WrapHTTP(err, "get session %q", id)
+	}
+
+	var msgs []sdk.Message
+	if opts.Full {
+		msgs, err = svc.LoadMessages(ctx, id, opts.Limit, nil)
+		if err != nil {
+			return cmdutil.WrapHTTP(err, "load messages for session %q", id)
+		}
+		if msgs == nil {
+			msgs = []sdk.Message{}
+		}
+	}
+
+	if fopts.WantsJSON() {
+		if !opts.Full {
+			return fopts.Emit(iostreams.IO.Out, s, nil)
+		}
+		// Project session + messages into a single bare object. Use the
+		// SDK json tags via an embedded *Session so existing keys stay
+		// stable.
+		payload := struct {
+			*sdk.Session
+			Messages []sdk.Message `json:"messages"`
+		}{Session: s, Messages: msgs}
+		return fopts.Emit(iostreams.IO.Out, payload, nil)
+	}
+
+	w := iostreams.IO.Out
+	fmt.Fprintf(w, "ID:        %s\n", s.ID)
+	if s.Title != "" {
+		fmt.Fprintf(w, "TITLE:     %s\n", s.Title)
+	}
+	if s.Description != "" {
+		fmt.Fprintf(w, "DESC:      %s\n", s.Description)
+	}
+	if t, ok := parseTS(s.CreatedAt); ok {
+		fmt.Fprintf(w, "CREATED:   %s\n", t.Format("2006-01-02 15:04:05"))
+	} else if s.CreatedAt != "" {
+		fmt.Fprintf(w, "CREATED:   %s\n", s.CreatedAt)
+	}
+	if t, ok := parseTS(s.UpdatedAt); ok {
+		fmt.Fprintf(w, "UPDATED:   %s\n", t.Format("2006-01-02 15:04:05"))
+	} else if s.UpdatedAt != "" {
+		fmt.Fprintf(w, "UPDATED:   %s\n", s.UpdatedAt)
+	}
+
+	if opts.Full {
+		fmt.Fprintln(w)
+		fmt.Fprintf(w, "Messages (%d):\n", len(msgs))
+		for _, m := range msgs {
+			fmt.Fprintln(w)
+			ts := ""
+			if !m.CreatedAt.IsZero() {
+				ts = " " + m.CreatedAt.Format("2006-01-02 15:04:05")
+			}
+			fmt.Fprintf(w, "[%s]%s\n", m.Role, ts)
+			if m.Content != "" {
+				fmt.Fprintln(w, m.Content)
+			}
+		}
+	}
+	return nil
+}
+
+func parseTS(s string) (time.Time, bool) {
+	if s == "" {
+		return time.Time{}, false
+	}
+	t, err := time.Parse(time.RFC3339, s)
+	if err != nil {
+		return time.Time{}, false
+	}
+	return t, true
+}