#pragma once

#include <list>
#include "../../defines.hpp"
#include "../../helpers/WLClasses.hpp"

class CInputManager;
struct STextInputV1;

class CInputMethodRelay {
  public:
    CInputMethodRelay();

    void                 onNewIME(wlr_input_method_v2*);
    void                 onNewTextInput(wlr_text_input_v3*);

    wlr_input_method_v2* m_pWLRIME = nullptr;

    void                 commitIMEState(STextInput* pInput);
    void                 removeTextInput(STextInput* pInput);

    void                 onKeyboardFocus(wlr_surface*);

    STextInput*          getFocusedTextInput();
    STextInput*          getFocusableTextInput();

    void                 setPendingSurface(STextInput*, wlr_surface*);

    SIMEKbGrab*          getIMEKeyboardGrab(SKeyboard*);

    void                 setIMEPopupFocus(SIMEPopup*, wlr_surface*);
    void                 updateInputPopup(SIMEPopup*);
    void                 damagePopup(SIMEPopup*);
    void                 removePopup(SIMEPopup*);

  private:
    std::unique_ptr<SIMEKbGrab> m_pKeyboardGrab;

    std::list<STextInput>       m_lTextInputs;
    std::list<SIMEPopup>        m_lIMEPopups;

    DYNLISTENER(textInputNew);
    DYNLISTENER(IMECommit);
    DYNLISTENER(IMEDestroy);
    DYNLISTENER(IMEGrab);
    DYNLISTENER(IMENewPopup);

    void         createNewTextInput(wlr_text_input_v3*, STextInputV1* tiv1 = nullptr);
    wlr_surface* focusedSurface(STextInput* pInput);

    friend class CHyprRenderer;
    friend class CInputManager;
    friend class CTextInputV1ProtocolManager;
};