47 lines
985 B
Go
47 lines
985 B
Go
package geminicli
|
|
|
|
import "strings"
|
|
|
|
const maxLogBodyLen = 2048
|
|
|
|
func SanitizeBodyForLogs(body string) string {
|
|
body = truncateBase64InMessage(body)
|
|
if len(body) > maxLogBodyLen {
|
|
body = body[:maxLogBodyLen] + "...[truncated]"
|
|
}
|
|
return body
|
|
}
|
|
|
|
func truncateBase64InMessage(message string) string {
|
|
const maxBase64Length = 50
|
|
|
|
result := message
|
|
offset := 0
|
|
for {
|
|
idx := strings.Index(result[offset:], ";base64,")
|
|
if idx == -1 {
|
|
break
|
|
}
|
|
actualIdx := offset + idx
|
|
start := actualIdx + len(";base64,")
|
|
|
|
end := start
|
|
for end < len(result) && isBase64Char(result[end]) {
|
|
end++
|
|
}
|
|
|
|
if end-start > maxBase64Length {
|
|
result = result[:start+maxBase64Length] + "...[truncated]" + result[end:]
|
|
offset = start + maxBase64Length + len("...[truncated]")
|
|
continue
|
|
}
|
|
offset = end
|
|
}
|
|
|
|
return result
|
|
}
|
|
|
|
func isBase64Char(c byte) bool {
|
|
return (c >= 'A' && c <= 'Z') || (c >= 'a' && c <= 'z') || (c >= '0' && c <= '9') || c == '+' || c == '/' || c == '='
|
|
}
|