1 files changed, 12 insertions, 0 deletions
diff --git a/src/widgets/Decorations.hpp b/src/widgets/Decorations.hpp
index e35e83c..d58ee7f 100644
--- a/src/widgets/Decorations.hpp
+++ b/src/widgets/Decorations.hpp
@@ -3,17 +3,29 @@
 #include "widgets/EdgeDecoration.hpp"
 #include <QWidget>
 #include <QtCore>
+#include <optional>
 #include <qwebengineprofile.h>
 
+enum DecorationType : uint8_t {
+  DecorationTop = 1,
+  DecorationBottom = 2,
+  DecorationLeft = 3,
+  DecorationRight = 4,
+};
+
 class Decorations : public QWidget {
   Q_OBJECT
 
 public:
   Decorations(QWidget *content_widget, QWebEngineProfile *profile, QWidget *parent = nullptr);
 
+  void set_enabled(DecorationType type, bool enabled);
+
 private:
   EdgeDecoration *decoration_top;
   EdgeDecoration *decoration_bottom;
   EdgeDecoration *decoration_left;
   EdgeDecoration *decoration_right;
+
+  std::optional<EdgeDecoration *> get_decoration_widget(DecorationType type);
 };