[FIX] ArticleDetail 북마크 버튼 재사용 이슈 해결 (Team-LionHeart#113)

bricksky · Jul 21, 2023 · bcc3762 · bcc3762
1 parent 67f05c7
commit bcc3762
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 8 deletions.
diff --git a/LionHeart-iOS/LionHeart-iOS/Scenes/Article/ArticleDetail/Cells/ThumnailTableViewCell.swift b/LionHeart-iOS/LionHeart-iOS/Scenes/Article/ArticleDetail/Cells/ThumnailTableViewCell.swift
@@ -123,10 +123,16 @@ private extension ThumnailTableViewCell {
 }
 
 extension ThumnailTableViewCell {
+    func setThumbnailImageView() {
+        bookMarkButton.isHidden = false
+        gradientImageView.isHidden = false
+    }
+
     func setImageTypeCell() {
         imageCaptionLabel.snp.updateConstraints { make in
             make.bottom.equalToSuperview().inset(72)
         }
+        gradientImageView.isHidden = true
         bookMarkButton.isHidden = true
     }
 }
diff --git a/...nHeart-iOS/Scenes/Article/ArticleDetail/ViewControllers/ArticleDetailViewController.swift b/...nHeart-iOS/Scenes/Article/ArticleDetail/ViewControllers/ArticleDetailViewController.swift
@@ -33,17 +33,15 @@ final class ArticleDetailViewController: UIViewController {
 
     // MARK: - Properties
 
+    private var isBookMarked: Bool?
+
     private var articleDatas: [BlockTypeAppData]? {
         didSet {
             self.articleTableView.reloadData()
             hideLoading()
         }
     }
 
-//    private var articleDatas: [BlockTypeAppData] = ArticleDetail.dummy().toAppData()
-
-    private var isArticleMarked: Bool?
-
     private var articleId: Int?
 
     private var contentOffsetY: CGFloat = 0
@@ -91,6 +89,7 @@ extension ArticleDetailViewController {
             do {
                 let bookmarkRequest = BookmarkRequest(articleId: articleId, bookmarkStatus: isSelected)
                 try await BookmarkService.shared.postBookmark(bookmarkRequest)
+                isBookMarked = isSelected
             } catch {
                 guard let error = error as? NetworkError else { return }
                 self.handleError(error)
@@ -191,19 +190,24 @@ extension ArticleDetailViewController: UITableViewDataSource {
 
     func tableView(_ tableView: UITableView, cellForRowAt indexPath: IndexPath) -> UITableViewCell {
         guard let articleDatas else { return UITableViewCell() }
+
         switch articleDatas[indexPath.row] {
         case .thumbnail(let isMarked, let thumbnailModel):
             let cell = ThumnailTableViewCell.dequeueReusableCell(to: articleTableView)
-            self.isArticleMarked = isMarked
-            cell.isMarked = self.isArticleMarked
             cell.inputData = thumbnailModel
-
             cell.selectionStyle = .none
             cell.bookmarkButtonDidTap = { isSelected in
                 guard let articleId = self.articleId else { return }
-                
+
                 self.articleBookMark(articleId: articleId, isSelected: isSelected)
             }
+
+            if let isBookMarked {
+                cell.isMarked = isBookMarked
+            } else {
+                cell.isMarked = isMarked
+            }
+            cell.setThumbnailImageView()
             return cell
         case .articleTitle(let titleModel):
             let cell = TitleTableViewCell.dequeueReusableCell(to: articleTableView)