Widgets: show error message instead of crashing

app/src/main/java/org/isoron/uhabits/widgets/BaseWidgetProvider.java

@@ -107,15 +107,19 @@ public abstract class BaseWidgetProvider extends AppWidgetProvider
         Habit habit = Habit.get(habitId);
         if(habit == null)
         {
-            RemoteViews errorView = new RemoteViews(context.getPackageName(),
+            drawErrorWidget(context, manager, widgetId);
-                    R.layout.widget_error);
-            manager.updateAppWidget(widgetId, errorView);
             return;
         }
 
         new RenderWidgetTask(widgetId, context, habit, manager).execute();
     }
 
+    private void drawErrorWidget(Context context, AppWidgetManager manager, int widgetId)
+    {
+        RemoteViews errorView = new RemoteViews(context.getPackageName(), R.layout.widget_error);
+        manager.updateAppWidget(widgetId, errorView);
+    }
+
     protected abstract void refreshCustomViewData(View widgetView);
 
     private void savePreview(Context context, int widgetId, Bitmap widgetCache)
@@ -231,19 +235,28 @@ public abstract class BaseWidgetProvider extends AppWidgetProvider
         @Override
         protected void onPostExecute(Void aVoid)
         {
-            widgetView.invalidate();
+            try
-            widgetView.setDrawingCacheEnabled(true);
+            {
-            widgetView.buildDrawingCache(true);
+                widgetView.invalidate();
-            Bitmap drawingCache = widgetView.getDrawingCache();
+                widgetView.setDrawingCacheEnabled(true);
-            remoteViews.setTextViewText(R.id.label, habit.name);
+                widgetView.buildDrawingCache(true);
-            remoteViews.setImageViewBitmap(R.id.imageView, drawingCache);
+                Bitmap drawingCache = widgetView.getDrawingCache();
-
+                remoteViews.setTextViewText(R.id.label, habit.name);
-            //savePreview(context, widgetId, drawingCache);
+                remoteViews.setImageViewBitmap(R.id.imageView, drawingCache);
-
+
-            PendingIntent onClickIntent = getOnClickPendingIntent(context, habit);
+                //savePreview(context, widgetId, drawingCache);
-            if(onClickIntent != null) remoteViews.setOnClickPendingIntent(R.id.imageView, onClickIntent);
+
-
+                PendingIntent onClickIntent = getOnClickPendingIntent(context, habit);
-            manager.updateAppWidget(widgetId, remoteViews);
+                if (onClickIntent != null) remoteViews.setOnClickPendingIntent(R.id.imageView,
+                        onClickIntent);
+
+                manager.updateAppWidget(widgetId, remoteViews);
+            }
+            catch (Exception e)
+            {
+                drawErrorWidget(context, manager, widgetId);
+                e.printStackTrace();
+            }
 
             super.onPostExecute(aVoid);
         }

app/src/main/res/layout/widget_error.xml

@@ -35,7 +35,7 @@
         android:layout_width="match_parent"
         android:layout_height="wrap_content"
         android:gravity="center"
-        android:text="Habit not found"
+        android:text="Error drawing widget"
         android:textColor="#ffffff"/>
 
 </LinearLayout>