refatora NotificationStrategy

TreinaDev · Feb 16, 2024 · 984bdb0 · 984bdb0
1 parent 5dd5b91
commit 984bdb0
Showing 1 changed file with 10 additions and 9 deletions.
diff --git a/app/models/notification_strategy.rb b/app/models/notification_strategy.rb
@@ -1,13 +1,14 @@
 class NotificationStrategy < SimpleDelegator
-  STRATEGY = {
-    Invitation => ->(notifiable) { notifiable },
-    Post => ->(notifiable) { notifiable },
-    Comment => ->(notifiable) { notifiable.post },
-    Connection => ->(notifiable) { notifiable.follower },
-    Like => ->(notifiable) { notifiable.likeable.is_a?(Comment) ? notifiable.likeable.post : notifiable.likeable }
-  }.freeze
-
   def redirect_after_click
-    STRATEGY[notifiable.class].call(notifiable)
+    case notifiable
+    when Comment
+      notifiable.post
+    when Connection
+      notifiable.follower
+    when Like
+      notifiable.likeable.is_a?(Comment) ? notifiable.likeable.post : notifiable.likeable
+    else
+      notifiable
+    end
   end
 end