bugfix result code on changes save

Author: gt4dev

app/src/main/java/com/example/android/architecture/blueprints/todoapp/addedittask/AddEditTaskFragment.kt

@@ -27,6 +27,7 @@ import com.example.android.architecture.blueprints.todoapp.EventObserver
 import com.example.android.architecture.blueprints.todoapp.R
 import com.example.android.architecture.blueprints.todoapp.databinding.AddtaskFragBinding
 import com.example.android.architecture.blueprints.todoapp.tasks.ADD_EDIT_RESULT_OK
+import com.example.android.architecture.blueprints.todoapp.tasks.EDIT_RESULT_OK
 import com.example.android.architecture.blueprints.todoapp.util.getViewModelFactory
 import com.example.android.architecture.blueprints.todoapp.util.setupRefreshLayout
 import com.example.android.architecture.blueprints.todoapp.util.setupSnackbar
@@ -73,8 +74,9 @@ class AddEditTaskFragment : Fragment() {
         viewModel.taskUpdatedEvent.observe(
             viewLifecycleOwner,
             EventObserver {
+                val resultCode = if (viewModel.isNewTask) ADD_EDIT_RESULT_OK else EDIT_RESULT_OK
                 val action = AddEditTaskFragmentDirections
-                    .actionAddEditTaskFragmentToTasksFragment(ADD_EDIT_RESULT_OK)
+                    .actionAddEditTaskFragmentToTasksFragment(resultCode)
                 findNavController().navigate(action)
             }
         )

app/src/main/java/com/example/android/architecture/blueprints/todoapp/addedittask/AddEditTaskViewModel.kt

@@ -51,7 +51,7 @@ class AddEditTaskViewModel(
 
     private var taskId: String? = null
 
-    private var isNewTask: Boolean = false
+    var isNewTask: Boolean = false
 
     private var isDataLoaded = false