| 143 | return err |
| 144 | } |
| 145 | return s.SaveStateWithRecovery(sessionID, state, recovery, tasks) |
| 146 | } |
| 147 | |
| 148 | func (s *Store) LoadState(sessionID string) *agent.AgentState { |
| 149 | data := loadJSONMap(s.stateReadPath(sessionID)) |
| 150 | if data == nil { |
| 151 | return nil |
| 152 | } |
| 153 | payload, _ := data["state"].(map[string]any) |
| 154 | if payload == nil { |
| 155 | payload = data |
| 156 | } |
| 157 | permissionRaw, _ := payload["permission_state"].(map[string]any) |
| 158 | cacheBreakpointsRaw := payload["cache_breakpoints"] |
| 159 | if cacheBreakpointsRaw == nil { |
| 160 | cacheBreakpointsRaw = payload["cache_break_points"] |
| 161 | } |
| 162 | delete(payload, "permission_state") |
| 163 | delete(payload, "cache_breakpoints") |
| 164 | delete(payload, "cache_break_points") |
| 165 | state, err := agent.GetAgentStateFromMap(payload) |
| 166 | if err != nil { |
| 167 | return nil |
| 168 | } |
| 169 | if permissionRaw != nil { |
| 170 | if permissionState, err := security.GetPermissionStateFromMap(permissionRaw); err == nil { |
| 171 | state.PermissionState = &permissionState |
| 172 | } |
| 173 | } |
| 174 | if state.PermissionState == nil { |
| 175 | state.PermissionState = security.DefaultPermissionState() |
| 176 | } |
| 177 | state.CacheBreakPoints = mapset.NewSet[int]() |
| 178 | for _, value := range anySlice(cacheBreakpointsRaw) { |
| 179 | state.CacheBreakPoints.Add(intFromAny(value)) |
| 180 | } |
| 181 | return &state |
| 182 | } |