Fix capturing self and calling additional configuration

appsailor · Feb 5, 2019 · e5719f3 · e5719f3
1 parent bc1a15e
commit e5719f3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/ChattoAdditions/Source/Chat Items/CompoundMessage/CompoundMessagePresenter.swift b/ChattoAdditions/Source/Chat Items/CompoundMessage/CompoundMessagePresenter.swift
@@ -91,7 +91,8 @@ public final class CompoundMessagePresenter<ViewModelBuilderT, InteractionHandle
             return
         }
 
-        super.configureCell(compoundCell, decorationAttributes: decorationAttributes, animated: animated) {
+        super.configureCell(compoundCell, decorationAttributes: decorationAttributes, animated: animated) { [weak self] in
+            guard let self = self else { return }
             guard compoundCell.lastDisplayedModel != self.messageModel else { return }
             compoundCell.lastDisplayedModel = self.messageModel
             let modules = self.contentFactories.map { $0.createMessageModule(forModel: self.messageModel) }
@@ -100,6 +101,7 @@ public final class CompoundMessagePresenter<ViewModelBuilderT, InteractionHandle
             bubbleView.style = self.compoundCellStyle
             bubbleView.decoratedContentViews = modules.map { .init(module: $0) }
             bubbleView.layoutProvider = self.layoutProvider
+            additionalConfiguration?()
         }
     }