1 files changed, 20 insertions, 5 deletions

diff --git a/AFBClient.h b/AFBClient.h
index 5b8bb6d..f8f824a 100644
--- a/AFBClient.h
+++ b/AFBClient.h
@@ -3,6 +3,8 @@
 
 #include <systemd/sd-event.h>
 
+#include <functional>
+
 extern "C"
 {
 #include <json-c/json.h>
@@ -18,19 +20,32 @@ class AFBClient
     virtual~AFBClient();
 
 public:
+    enum EventType {
+       Event_Active = 1,
+       Event_Inactive,
+       Event_Visible,
+       Event_Invisible,
+       Event_SyncDraw,
+       Event_FlushDraw,
+    };
+
     static AFBClient &instance()
     {
         static AFBClient obj;
         return obj;
     }
 
-    bool init();
-    int requestSurface(const char *label);
-    void activateSurface(const char *label);
-    void deactivateSurface(const char *label);
-    void endDraw(const char *label);
+    int init(int port, char const *token);
     int dispatch(uint64_t timeout);
 
+    // WM API
+    int requestSurface(const char *label);
+    int activateSurface(const char *label);
+    int deactivateSurface(const char *label);
+    int endDraw(const char *label);
+
+    void set_event_handler(enum EventType et, std::function<void(char const *label)> f);
+
     static void onCall(void *closure, const char *api, const char *verb, struct afb_wsj1_msg *msg);
     static void onEvent(void *closure, const char *event, struct afb_wsj1_msg *msg);
     static void onHangup(void *closure, struct afb_wsj1 *wsj1);