core: verify surface roles on creation of objects

1 files changed, 6 insertions, 4 deletions

diff --git a/src/protocols/LayerShell.hpp b/src/protocols/LayerShell.hpp
index f348e86b..7fa14447 100644
--- a/src/protocols/LayerShell.hpp
+++ b/src/protocols/LayerShell.hpp
@@ -8,18 +8,20 @@
 #include "wlr-layer-shell-unstable-v1.hpp"
 #include "../helpers/Vector2D.hpp"
 #include "../helpers/signal/Signal.hpp"
+#include "types/SurfaceRole.hpp"
 
 class CMonitor;
 class CWLSurfaceResource;
 
-class CLayerShellResource {
+class CLayerShellResource : public ISurfaceRole {
   public:
     CLayerShellResource(SP<CZwlrLayerSurfaceV1> resource_, SP<CWLSurfaceResource> surf_, std::string namespace_, CMonitor* pMonitor, zwlrLayerShellV1Layer layer);
     ~CLayerShellResource();
 
-    bool good();
-    void configure(const Vector2D& size);
-    void sendClosed();
+    bool                 good();
+    void                 configure(const Vector2D& size);
+    void                 sendClosed();
+    virtual eSurfaceRole role();
 
     enum eCommittedState {
         STATE_SIZE          = (1 << 0),