diff --git a/src/decorator/BaseDecorator.java b/src/decorator/BaseDecorator.java index 103ccdf..9fd2618 100644 --- a/src/decorator/BaseDecorator.java +++ b/src/decorator/BaseDecorator.java @@ -1,6 +1,7 @@ package decorator; import entities.Notification; +import notificationsDecorations.NotificationDecoration; public abstract class BaseDecorator { protected Notification notification; @@ -9,5 +10,17 @@ public abstract class BaseDecorator { this.notification = notification; } + public void transitionState() { + notification.transitionState(); + } + + public void setKoState() { + notification.setKoState(); + } + + public void addDecoration(NotificationDecoration decoration) { + notification.addDecoration(decoration); + } + public abstract String getContent(); }