#ifndef CROSSDESK_GUI_KEYBOARD_CONTROLLER_H_
#define CROSSDESK_GUI_KEYBOARD_CONTROLLER_H_

#include <cstdint>
#include <mutex>
#include <string>
#include <unordered_map>

#include <remote_action.h>

#include "keyboard_state.h"

namespace crossdesk {

class GuiRuntime;

// Synchronizes local and remote keyboard state, including heartbeat recovery
// for keys whose key-up event was lost during a connection interruption.
class KeyboardController {
public:
  explicit KeyboardController(GuiRuntime &owner);

  int SendKeyCommand(int key_code, bool is_down, uint32_t scan_code = 0,
                     bool extended = false);
  void ForceReleasePressedKeys();
  void RecoverCapturedState(const PressedKeys& pressed);
  void SendHeartbeat(bool force);
  void ApplyRemoteEvent(const std::string &remote_id,
                        const RemoteAction &remote_action);
  void ApplyRemoteText(const std::string& remote_id, const std::string& text);
  void ApplyRemoteState(const std::string &remote_id,
                        const RemoteAction &remote_action);
  void ReleaseRemotePressedKeys(const std::string &remote_id,
                                const char *reason);
  void ReleaseAllRemotePressedKeys(const char* reason);
  void CheckRemoteTimeouts();

private:
  struct RemoteState {
    PressedKeys pressed_keys;
    uint32_t last_seq = 0;
    uint32_t last_seen_tick = 0;
    bool keyboard_state_seen = false;
  };

  void TrackPressedKey(int key_code, bool is_down, uint32_t scan_code,
                       bool extended);
  bool InjectRemoteKey(int key_code, bool is_down, uint32_t scan_code,
                       bool extended);
  // Caller holds remote_input_mutex_. Copy the id because this erases state.
  void ReleaseRemotePressedKeysLocked(std::string remote_id, const char* reason);

  GuiRuntime &owner_;
  PressedKeys pressed_keys_;
  std::mutex pressed_keys_mutex_;
  uint32_t state_sequence_ = 0;
  uint32_t last_heartbeat_tick_ = 0;
  std::unordered_map<std::string, RemoteState> remote_states_;
  // Guards all remote key state and serializes injection, cleanup and retries.
  std::mutex remote_input_mutex_;
  // Successful new input supersedes cleanup of an older press of that key.
  PressedKeys pending_key_releases_;
  uint32_t last_release_retry_tick_ = 0;
};

} // namespace crossdesk

#endif // CROSSDESK_GUI_KEYBOARD_CONTROLLER_H_