on blog update success

Author: mitchtabian

app/src/main/java/com/codingwithmitch/openapi/ui/main/blog/UpdateBlogFragment.kt

@@ -7,6 +7,8 @@ import androidx.lifecycle.Observer
 import androidx.navigation.fragment.findNavController
 import com.codingwithmitch.openapi.R
 import com.codingwithmitch.openapi.ui.main.blog.state.BlogStateEvent
+import com.codingwithmitch.openapi.ui.main.blog.viewmodel.onBlogPostUpdateSuccess
+import com.codingwithmitch.openapi.ui.main.blog.viewmodel.setUpdatedBlogFields
 import kotlinx.android.synthetic.main.fragment_update_blog.*
 import okhttp3.MultipartBody
 
@@ -35,7 +37,9 @@ class UpdateBlogFragment : BaseBlogFragment(){
 
                     // if this is not null, the blogpost was updated
                     viewState.viewBlogFields.blogPost?.let{ blogPost ->
-                        // TODO("onBlogPostUpdateSuccess")
+                        viewModel.onBlogPostUpdateSuccess(blogPost).let {
+                            findNavController().popBackStack()
+                        }
                     }
                 }
             }
@@ -86,6 +90,14 @@ class UpdateBlogFragment : BaseBlogFragment(){
         return super.onOptionsItemSelected(item)
     }
 
+    override fun onPause() {
+        super.onPause()
+        viewModel.setUpdatedBlogFields(
+            uri = null,
+            title = blog_title.text.toString(),
+            body = blog_body.text.toString()
+        )
+    }
 }
 
 

app/src/main/java/com/codingwithmitch/openapi/ui/main/blog/viewmodel/Setters.kt

@@ -81,8 +81,30 @@ fun BlogViewModel.setUpdatedBlogFields(title: String?, body: String?, uri: Uri?)
 }
 
 
+fun BlogViewModel.updateListItem(newBlogPost: BlogPost){
+    val update = getCurrentViewStateOrNew()
+    val list = update.blogFields.blogList.toMutableList()
+    for(i in 0..(list.size - 1)){
+        if(list[i].pk == newBlogPost.pk){
+            list[i] = newBlogPost
+            break
+        }
+    }
+    update.blogFields.blogList = list
+    setViewState(update)
+}
 
 
+fun BlogViewModel.onBlogPostUpdateSuccess(blogPost: BlogPost){
+    setUpdatedBlogFields(
+        uri = null,
+        title = blogPost.title,
+        body = blogPost.body
+    ) // update UpdateBlogFragment (not really necessary since navigating back)
+    setBlogPost(blogPost) // update ViewBlogFragment
+    updateListItem(blogPost) // update BlogFragment
+}
+