refactor: 프리뷰가 제대로 뜨도록 수정

boostcampwm-2024 · Dec 10, 2024 · 12ab2f0 · 12ab2f0
1 parent 14cbd29
commit 12ab2f0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/MemorialHouse/MHPresentation/MHPresentation/Source/EditBook/View/EditPageCell.swift b/MemorialHouse/MHPresentation/MHPresentation/Source/EditBook/View/EditPageCell.swift
@@ -158,6 +158,7 @@ final class EditPageCell: UITableViewCell {
             let attachmentString = NSAttributedString(attachment: mediaAttachment)
             // Placeholder(공백) 교체
             mutableAttributedString.replaceCharacters(in: range, with: attachmentString)
+            mediaAttachment.dataSource = self
         }
 
         mutableAttributedString.addAttributes(
@@ -264,6 +265,7 @@ final class EditPageCell: UITableViewCell {
             shouldChangeTextIn: NSRange(location: textStorage.length, length: 0),
             replacementText: text
         ) else { return }
+        attachment.dataSource = self
         textStorage.beginEditing()
         textStorage.append(text)
         textStorage.endEditing()