blob: 8df421bd86fce07370489b5eb40857d0a7a5b520 (
plain)
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
|
package session
import (
"context"
"sync"
"github.com/opencode-ai/opencode/internal/logging"
"github.com/opencode-ai/opencode/internal/pubsub"
)
// Manager handles session management, tracking the currently active session.
type Manager struct {
currentSessionID string
service Service
mu sync.RWMutex
}
// Global instance of the session manager
var globalManager *Manager
// InitManager initializes the global session manager with the provided service.
func InitManager(service Service) {
globalManager = &Manager{
currentSessionID: "",
service: service,
}
// Subscribe to session events to handle session deletions
go func() {
ctx := context.Background()
eventCh := service.Subscribe(ctx)
for event := range eventCh {
if event.Type == pubsub.DeletedEvent && event.Payload.ID == CurrentSessionID() {
// If the current session is deleted, clear the current session
SetCurrentSession("")
}
}
}()
}
// SetCurrentSession changes the active session to the one with the specified ID.
func SetCurrentSession(sessionID string) {
if globalManager == nil {
logging.Warn("Session manager not initialized")
return
}
globalManager.mu.Lock()
defer globalManager.mu.Unlock()
globalManager.currentSessionID = sessionID
logging.Debug("Current session changed", "sessionID", sessionID)
}
// CurrentSessionID returns the ID of the currently active session.
func CurrentSessionID() string {
if globalManager == nil {
logging.Warn("Session manager not initialized")
return ""
}
globalManager.mu.RLock()
defer globalManager.mu.RUnlock()
return globalManager.currentSessionID
}
// CurrentSession returns the currently active session.
// If no session is set or the session cannot be found, it returns nil.
func CurrentSession() *Session {
if globalManager == nil {
logging.Warn("Session manager not initialized")
return nil
}
sessionID := CurrentSessionID()
if sessionID == "" {
return nil
}
session, err := globalManager.service.Get(context.Background(), sessionID)
if err != nil {
logging.Warn("Failed to get current session", "err", err)
return nil
}
return &session
}
|
