Files
2026-07-13 13:00:08 +08:00

95 lines
2.8 KiB
Go

package agent
import (
"bytes"
"os"
"time"
"reasonix/internal/provider"
"reasonix/internal/store"
)
// SessionsShareContent reports whether two saved sessions decode to the same
// transcript. It replaces byte-comparing the .jsonl checkpoints, which stopped
// implying transcript equality once the event log became authoritative: two
// identical checkpoints can hide diverged event logs.
func SessionsShareContent(pathA, pathB string) (bool, error) {
msgsA, _, _, err := loadSessionMessages(pathA)
if err != nil {
return false, err
}
msgsB, _, _, err := loadSessionMessages(pathB)
if err != nil {
return false, err
}
digestA, err := digestSessionMessages(msgsA)
if err != nil {
return false, err
}
digestB, err := digestSessionMessages(msgsB)
if err != nil {
return false, err
}
return bytes.Equal(digestA[:], digestB[:]), nil
}
// SessionUserMessage is one user-role message with the best-known wall-clock
// time. Messages restored from a replace event (compaction, rewind) lose their
// per-turn times and report zero; callers apply their own fallback.
type SessionUserMessage struct {
Text string
At time.Time
}
// LoadSessionUserMessages returns the session's user-role messages in
// transcript order, event-log aware. Direct .jsonl decoding misses everything
// after the first save once an event log exists, so surfaces like prompt
// history must use this instead.
func LoadSessionUserMessages(path string) ([]SessionUserMessage, error) {
if probe, err := probeSessionEventLog(path); err == nil && probe.native && probe.size > 0 {
replay, err := replaySessionEventLog(store.SessionEventLog(path))
if err == nil && replay.records > 0 {
out := make([]SessionUserMessage, 0, len(replay.msgs))
for i, m := range replay.msgs {
if m.Role != provider.RoleUser {
continue
}
at := time.Time{}
if i < len(replay.times) {
at = replay.times[i]
}
out = append(out, SessionUserMessage{Text: m.Content, At: at})
}
return out, nil
}
}
msgs, err := loadSessionMessagesFromJSONL(path)
if err != nil {
return nil, err
}
out := make([]SessionUserMessage, 0, len(msgs))
for _, m := range msgs {
if m.Role != provider.RoleUser {
continue
}
out = append(out, SessionUserMessage{Text: m.Content})
}
return out, nil
}
// SessionContentModTime returns when the session transcript last changed on
// disk: the newer of the .jsonl checkpoint and the event log. The checkpoint
// alone goes stale between checkpoints, so recency ordering must use this.
func SessionContentModTime(path string) time.Time {
var mod time.Time
if info, err := os.Stat(path); err == nil && !info.IsDir() {
mod = info.ModTime()
}
if logPath := store.SessionEventLog(path); logPath != "" {
if info, err := os.Stat(logPath); err == nil && !info.IsDir() && info.ModTime().After(mod) {
mod = info.ModTime()
}
}
return mod
}