Article bookmark button functional

Author: sameersyd

BoxFeed.xcodeproj/project.xcworkspace/xcuserdata/sameernawaz.xcuserdatad/UserInterfaceState.xcuserstate

@@ -12,6 +12,11 @@ struct ArticleView: View {
     @Environment(\.dismiss) var dismiss
     @StateObject var viewModel: ArticleViewModel
 
+    // CoreData
+    @Environment(\.managedObjectContext) var moc
+    @FetchRequest(fetchRequest: ArticleCD.getAllArticles()) var articles: FetchedResults<ArticleCD>
+    
+    
     var body: some View {
         NavigationView {
             ScrollView(.vertical, showsIndicators: false) {
@@ -82,8 +87,9 @@ extension ArticleView {
 
                 VStack(spacing: 0) {
                     HStack(alignment: .center) {
-                        Button(action: {  }) {
-                            Image.bookmark_filled.resizable()
+                        Button(action: { viewModel.bookmarkArticle(articles, moc) }) {
+                            (viewModel.isBookmarked(articles) ? Image.bookmark_filled : Image.bookmark)
+                                .resizable()
                                 .renderingMode(.template)
                                 .foregroundColor(.white)
                                 .frame(width: 20, height: 20)

BoxFeed/Screens/Article/ArticleView.swift

@@ -5,7 +5,8 @@
 //  Created by Sameer Nawaz on 18/09/21.
 //
 
-import Foundation
+import SwiftUI
+import CoreData
 
 class ArticleViewModel: ObservableObject {
 
@@ -14,4 +15,49 @@ class ArticleViewModel: ObservableObject {
     init(model: NewsModel) {
         self.model = model
     }
+    
+    func isBookmarked(_ articles: FetchedResults<ArticleCD>) -> Bool {
+        for article in articles {
+            if article.url == model.url?.absoluteString {
+                return true
+            }
+        }
+        return false
+    }
+    
+    func bookmarkArticle(
+        _ articles: FetchedResults<ArticleCD>,
+        _ moc: NSManagedObjectContext
+    ) {
+        for article in articles {
+            if article.url == model.url?.absoluteString {
+                removeBookmark(article, moc)
+                return
+            }
+        }
+        addBookmark(moc)
+    }
+    
+    private func addBookmark(_ moc: NSManagedObjectContext) {
+        let articleObj = ArticleCD(context: moc)
+        articleObj.id = UUID()
+        articleObj.createdAt = Date()
+        articleObj.sourceId = model.id?.rawValue
+        articleObj.author = model.author
+        articleObj.content = model.content
+        articleObj.desc = model.description
+        articleObj.publishedAt = model.publishedAt
+        articleObj.title = model.title
+        articleObj.url = model.url?.absoluteString
+        articleObj.urlToImage = model.urlToImage?.absoluteString
+        do { try moc.save() }
+        catch { print(error.localizedDescription) }
+    }
+    
+    private func removeBookmark(_ article: ArticleCD, _ moc: NSManagedObjectContext) {
+        moc.delete(article)
+        do {
+            try moc.save()
+        } catch { print(error.localizedDescription) }
+    }
 }