1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
// This file contains functions for formatting and printing chat logs to the terminal.
// These functions operate exclusively on the in-memory protobuf data structures.
// They do NOT access the filesystem; all content (from content files, snippets, etc.)
// is expected to have been loaded into the 'Content' fields of the protobuf messages
// before these functions are called.
package main
import (
"fmt"
"path/filepath"
"strings"
"go.wit.com/lib/protobuf/chatpb"
)
const termWidth = 100 // The target width for the formatted output boxes.
// prettyFormatChat is the main entry point to print a detailed view of a Chat topic.
func prettyFormatChat(chat *chatpb.Chat) {
fmt.Printf("\n========================================================\n")
fmt.Printf("== Chat Topic: %s (UUID: %s)\n", chat.GetChatName(), chat.GetUuid())
fmt.Printf("========================================================\n\n")
for _, entry := range chat.GetEntries() {
author := entry.GetFrom().String()
var formattedTime string
if ctime := entry.GetCtime(); ctime != nil {
t := ctime.AsTime()
formattedTime = t.Format("2006-01-02 15:04:05")
} else {
formattedTime = "No Timestamp"
}
// Use the in-memory Content field directly.
content := entry.GetContent()
if content != "" {
fmt.Printf("✦ %s (%s):\n%s\n", author, formattedTime, strings.TrimSpace(content))
}
if table := entry.GetTable(); table != nil {
printTable(table)
}
for _, toolCall := range entry.GetToolCalls() {
printToolCallBox(toolCall)
}
if snippets := entry.GetSnippets(); snippets != nil {
for _, snippet := range snippets {
printCodeSnippet(snippet)
}
}
fmt.Println()
}
}
func printTable(table *chatpb.Table) {
if table == nil || len(table.GetRows()) == 0 {
return
}
fmt.Println("┌─[ Table Data ]──────────────────────────────────────────")
for _, row := range table.GetRows() {
fmt.Printf("│ %s\n", strings.Join(row.GetFields(), " │ "))
}
fmt.Printf("└─────────────────────────────────────────────────────────\n\n")
}
func printCodeSnippet(snippet *chatpb.CodeSnippet) {
// Use the in-memory Content field directly.
code := snippet.GetContent()
language := filepath.Base(snippet.GetFilename()) // Still useful for display
fmt.Printf("┌─[ Code Snippet: %s ]──────────────────────────────────\n", language)
for _, line := range strings.Split(strings.TrimSpace(code), "\n") {
fmt.Printf("│ %s\n", line)
}
fmt.Printf("└─────────────────────────────────────────────────────────\n\n")
}
func printToolCallBox(tc *chatpb.ToolCall) {
boxWidth := termWidth - 2
fmt.Printf(" ╭%s╮\n", strings.Repeat("─", boxWidth))
header := fmt.Sprintf(" ✔ %s %s (%s)", tc.GetName(), tc.GetInput(), tc.GetDescription())
printWrappedLine(header, boxWidth)
printEmptyLine(boxWidth)
if stdout := tc.GetOutputStdout(); stdout != "" {
for _, line := range strings.Split(stdout, "\n") {
printWrappedLine(" "+line, boxWidth)
}
}
if stderr := tc.GetOutputStderr(); stderr != "" {
for _, line := range strings.Split(stderr, "\n") {
printWrappedLine(" "+line, boxWidth)
}
}
printEmptyLine(boxWidth)
fmt.Printf(" ╰%s╯\n", strings.Repeat("─", boxWidth))
}
func printWrappedLine(text string, width int) {
if len(text) == 0 {
printEmptyLine(width)
return
}
for len(text) > width {
fmt.Printf(" │ %-*s │\n", width, text[:width])
text = text[width:]
}
fmt.Printf(" │ %-*s │\n", width, text)
}
func printEmptyLine(width int) {
fmt.Printf(" │ %*s │\n", width, "")
}
|
