From c1e10e09a542e35a9949f76636ec2d4693a577a1 Mon Sep 17 00:00:00 2001
From: Alinson Xavier <axavier@uwaterloo.ca>
Date: Sun, 17 Jul 2016 08:30:07 -0400
Subject: [PATCH] Refresh list after importing DB

---
 .../uhabits/ui/habits/list/ListHabitsActivity.java       | 2 +-
 .../uhabits/ui/habits/list/ListHabitsController.java     | 9 +++++++--
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/app/src/main/java/org/isoron/uhabits/ui/habits/list/ListHabitsActivity.java b/app/src/main/java/org/isoron/uhabits/ui/habits/list/ListHabitsActivity.java
index 56d60a443..7b0ddef6a 100644
--- a/app/src/main/java/org/isoron/uhabits/ui/habits/list/ListHabitsActivity.java
+++ b/app/src/main/java/org/isoron/uhabits/ui/habits/list/ListHabitsActivity.java
@@ -65,7 +65,7 @@ protected void onCreate(Bundle savedInstanceState)
         screen = new ListHabitsScreen(this, rootView);
         menu = new ListHabitsMenu(this, screen, adapter);
         selectionMenu = new ListHabitsSelectionMenu(habits, screen, adapter);
-        controller = new ListHabitsController(habits, screen, system);
+        controller = new ListHabitsController(habits, screen, system, adapter);
 
         adapter.setProgressBar(
             new AndroidProgressBar(rootView.getProgressBar()));
diff --git a/app/src/main/java/org/isoron/uhabits/ui/habits/list/ListHabitsController.java b/app/src/main/java/org/isoron/uhabits/ui/habits/list/ListHabitsController.java
index 1382012ec..1bb34973d 100644
--- a/app/src/main/java/org/isoron/uhabits/ui/habits/list/ListHabitsController.java
+++ b/app/src/main/java/org/isoron/uhabits/ui/habits/list/ListHabitsController.java
@@ -28,6 +28,7 @@
 import org.isoron.uhabits.tasks.*;
 import org.isoron.uhabits.ui.*;
 import org.isoron.uhabits.ui.habits.list.controllers.*;
+import org.isoron.uhabits.ui.habits.list.model.*;
 import org.isoron.uhabits.utils.*;
 
 import java.io.*;
@@ -47,6 +48,8 @@ public class ListHabitsController
     @NonNull
     private final HabitList habitList;
 
+    private HabitCardListAdapter adapter;
+
     @Inject
     Preferences prefs;
 
@@ -55,11 +58,13 @@ public class ListHabitsController
 
     public ListHabitsController(@NonNull HabitList habitList,
                                 @NonNull ListHabitsScreen screen,
-                                @NonNull BaseSystem system)
+                                @NonNull BaseSystem system,
+                                @NonNull HabitCardListAdapter adapter)
     {
         this.screen = screen;
         this.system = system;
         this.habitList = habitList;
+        this.adapter = adapter;
         HabitsApplication.getComponent().inject(this);
     }
 
@@ -115,7 +120,7 @@ public void onImportDataFinished(int result)
         switch (result)
         {
             case ImportDataTask.SUCCESS:
-                screen.invalidate();
+                adapter.refresh();
                 screen.showMessage(R.string.habits_imported);
                 break;