agent_session.h

Go to the documentation of this file.

#ifndef YAZE_APP_EDITOR_AGENT_AGENT_SESSION_H_
#define YAZE_APP_EDITOR_AGENT_AGENT_SESSION_H_
 
#include <functional>
#include <memory>
#include <string>
#include <vector>
 
#include "app/editor/agent/agent_state.h"
 
namespace yaze {
namespace editor {
 
struct AgentSession {
  std::string agent_id;           // Unique ID (e.g., "agent_1", "agent_2")
  std::string display_name;       // User-visible name (e.g., "Agent 1", "Code Review")
  AgentUIContext context;         // Chat history, config, state (shared between views)
  bool is_active = false;         // Currently selected in sidebar tabs
  bool has_card_open = false;     // Full card visible in docking space
 
  // Callbacks for this session (set by parent controller)
  ChatCallbacks chat_callbacks;
  ProposalCallbacks proposal_callbacks;
  CollaborationCallbacks collaboration_callbacks;
};
 
class AgentSessionManager {
 public:
  using SessionCreatedCallback = std::function<void(const std::string& agent_id)>;
  using SessionClosedCallback = std::function<void(const std::string& agent_id)>;
  using PanelOpenedCallback = std::function<void(const std::string& agent_id)>;
  using PanelClosedCallback = std::function<void(const std::string& agent_id)>;
 
  AgentSessionManager();
  ~AgentSessionManager() = default;
 
  // ============================================================================
  // Session Lifecycle
  // ============================================================================
 
  std::string CreateSession(const std::string& name = "");
 
  void CloseSession(const std::string& agent_id);
 
  void RenameSession(const std::string& agent_id, const std::string& new_name);
 
  // ============================================================================
  // Active Session Management
  // ============================================================================
 
  AgentSession* GetActiveSession();
  const AgentSession* GetActiveSession() const;
 
  void SetActiveSession(const std::string& agent_id);
 
  AgentSession* GetSession(const std::string& agent_id);
  const AgentSession* GetSession(const std::string& agent_id) const;
 
  // ============================================================================
  // Panel Management (Pop-out Dockable Panels)
  // ============================================================================
 
  void OpenPanelForSession(const std::string& agent_id);
 
  void ClosePanelForSession(const std::string& agent_id);
 
  bool IsPanelOpenForSession(const std::string& agent_id) const;
 
  std::vector<std::string> GetOpenPanelSessionIds() const;
 
  // ============================================================================
  // Iteration
  // ============================================================================
 
  std::vector<AgentSession>& GetAllSessions() { return sessions_; }
  const std::vector<AgentSession>& GetAllSessions() const { return sessions_; }
 
  size_t GetSessionCount() const { return sessions_.size(); }
 
  bool HasSessions() const { return !sessions_.empty(); }
 
  // ============================================================================
  // Callbacks
  // ============================================================================
 
  void SetSessionCreatedCallback(SessionCreatedCallback cb) {
    on_session_created_ = std::move(cb);
  }
  void SetSessionClosedCallback(SessionClosedCallback cb) {
    on_session_closed_ = std::move(cb);
  }
  void SetPanelOpenedCallback(PanelOpenedCallback cb) {
    on_card_opened_ = std::move(cb);
  }
  void SetPanelClosedCallback(PanelClosedCallback cb) {
    on_card_closed_ = std::move(cb);
  }
 
 private:
  std::vector<AgentSession> sessions_;
  std::string active_session_id_;
  int next_session_number_ = 1;  // For auto-generating names
 
  // Callbacks
  SessionCreatedCallback on_session_created_;
  SessionClosedCallback on_session_closed_;
  PanelOpenedCallback on_card_opened_;
  PanelClosedCallback on_card_closed_;
 
  std::string GenerateAgentId();
 
  int FindSessionIndex(const std::string& agent_id) const;
};
 
}  // namespace editor
}  // namespace yaze
 
#endif  // YAZE_APP_EDITOR_AGENT_AGENT_SESSION_H_