tasks/astrid/src/main/java/com/todoroo/astrid/widget/WidgetUpdateService.java

package com.todoroo.astrid.widget;

import android.app.PendingIntent;
import android.app.Service;
import android.appwidget.AppWidgetManager;
import android.content.ComponentName;
import android.content.Context;
import android.content.Intent;
import android.content.SharedPreferences;
import android.content.res.Resources;
import android.os.IBinder;
import android.util.Log;
import android.view.View;
import android.widget.RemoteViews;

import com.todoroo.andlib.data.TodorooCursor;
import com.todoroo.andlib.service.Autowired;
import com.todoroo.andlib.service.ContextManager;
import com.todoroo.andlib.service.DependencyInjectionService;
import com.todoroo.andlib.utility.Preferences;
import com.todoroo.astrid.actfm.TagViewFragment;
import com.todoroo.astrid.api.Filter;
import com.todoroo.astrid.api.FilterWithCustomIntent;
import com.todoroo.astrid.core.SortHelper;
import com.todoroo.astrid.dao.Database;
import com.todoroo.astrid.dao.TaskListMetadataDao;
import com.todoroo.astrid.data.Task;
import com.todoroo.astrid.service.AstridDependencyInjector;
import com.todoroo.astrid.service.TagDataService;
import com.todoroo.astrid.service.TaskService;
import com.todoroo.astrid.service.ThemeService;
import com.todoroo.astrid.subtasks.SubtasksHelper;
import com.todoroo.astrid.utility.AstridPreferences;
import com.todoroo.astrid.utility.Constants;

import org.tasks.R;
import org.tasks.widget.WidgetHelper;

public class WidgetUpdateService extends Service {

    static {
        AstridDependencyInjector.initialize();
    }

    private static final int NUM_VISIBLE_TASKS = 25;

    public static final String EXTRA_WIDGET_ID = "widget_id"; //$NON-NLS-1$

    @Autowired Database database;

    @Autowired TaskService taskService;

    @Autowired TaskListMetadataDao taskListMetadataDao;

    @Autowired TagDataService tagDataService;

    private final WidgetHelper widgetHelper = new WidgetHelper();

    @Override
    public void onStart(final Intent intent, int startId) {
        ContextManager.setContext(this);
        new Thread(new Runnable() {
            @Override
            public void run() {
                startServiceInBackgroundThread(intent);
            }
        }).start();
    }

    public void startServiceInBackgroundThread(Intent intent) {
        ComponentName thisWidget = new ComponentName(this,
                TasksWidget.class);
        AppWidgetManager manager = AppWidgetManager.getInstance(this);

        int extrasId = AppWidgetManager.INVALID_APPWIDGET_ID;
        if(intent != null) {
            extrasId = intent.getIntExtra(EXTRA_WIDGET_ID, extrasId);
        }
        if(extrasId == AppWidgetManager.INVALID_APPWIDGET_ID) {
            int[] ids;
            try {
                ids = manager.getAppWidgetIds(thisWidget);
                for(int id : ids) {
                    RemoteViews updateViews = buildUpdate(this, id);
                    manager.updateAppWidget(id, updateViews);
                }
            } catch (RuntimeException e) {
                // "System server dead" was sometimes thrown here by the OS. Abort if that happens
            }
        } else {
            RemoteViews updateViews = buildUpdate(this, extrasId);
            manager.updateAppWidget(extrasId, updateViews);
        }

        stopSelf();
    }

    @Override
    public IBinder onBind(Intent intent) {
        return null;
    }

    public RemoteViews buildUpdate(Context context, int widgetId) {
        DependencyInjectionService.getInstance().inject(this);

        RemoteViews views = getThemedRemoteViews(context);

        int numberOfTasks = NUM_VISIBLE_TASKS;

        TodorooCursor<Task> cursor = null;
        Filter filter = null;
        try {
            filter = widgetHelper.getFilter(context, widgetId);
            if (SubtasksHelper.isTagFilter(filter)) {
                ((FilterWithCustomIntent) filter).customTaskList = new ComponentName(context, TagViewFragment.class); // In case legacy widget was created with subtasks fragment
            }
            views.setTextViewText(R.id.widget_title, filter.title);
            views.removeAllViews(R.id.taskbody);

            SharedPreferences publicPrefs = AstridPreferences.getPublicPrefs(this);
            int flags = publicPrefs.getInt(SortHelper.PREF_SORT_FLAGS, 0);
            int sort = publicPrefs.getInt(SortHelper.PREF_SORT_SORT, 0);
            String query = SortHelper.adjustQueryForFlagsAndSort(
                    filter.getSqlQuery(), flags, sort).replaceAll("LIMIT \\d+", "") + " LIMIT " + numberOfTasks;

            String tagName = Preferences.getStringValue(WidgetConfigActivity.PREF_TITLE + widgetId);
            query = SubtasksHelper.applySubtasksToWidgetFilter(tagDataService, taskListMetadataDao, filter, query, tagName, numberOfTasks);

            database.openForReading();
            cursor = taskService.fetchFiltered(query, null, Task.ID, Task.TITLE, Task.DUE_DATE, Task.COMPLETION_DATE);
            Task task = new Task();
            int i;
            for (i = 0; i < cursor.getCount() && i < numberOfTasks; i++) {
                cursor.moveToPosition(i);
                task.readFromCursor(cursor);

                String textContent;
                Resources r = context.getResources();
                int textColor = r
                        .getColor(ThemeService.isDarkWidgetTheme() ? R.color.widget_text_color_dark : R.color.widget_text_color_light);

                textContent = task.getTitle();

                if(task.isCompleted()) {
                    textColor = r.getColor(R.color.task_list_done);
                } else if(task.hasDueDate() && task.isOverdue()) {
                    textColor = r.getColor(R.color.task_list_overdue);
                }

                RemoteViews row = new RemoteViews(Constants.PACKAGE, R.layout.widget_row);

                row.setTextViewText(R.id.text, textContent);
                row.setTextColor(R.id.text, textColor);

                views.addView(R.id.taskbody, row);

                RemoteViews separator = new RemoteViews(Constants.PACKAGE, R.layout.widget_separator);
                boolean isLastRow = (i == cursor.getCount() - 1) || (i == numberOfTasks - 1);
                if (!isLastRow) {
                    views.addView(R.id.taskbody, separator);
                }
            }
            for (; i < numberOfTasks; i++) {
                RemoteViews row = new RemoteViews(Constants.PACKAGE, R.layout.widget_row);
                row.setViewVisibility(R.id.text, View.INVISIBLE);
                views.addView(R.id.taskbody, row);
            }

        } catch (Exception e) {
            // can happen if database is not ready
            Log.e("WIDGET-UPDATE", "Error updating widget", e);
        } finally {
            if(cursor != null) {
                cursor.close();
            }
        }

        PendingIntent pListIntent = widgetHelper.getListIntent(context, filter, widgetId);
        if (pListIntent != null) {
            views.setOnClickPendingIntent(R.id.taskbody, pListIntent);
        }

        PendingIntent pEditIntent = widgetHelper.getNewTaskIntent(context, filter, widgetId);
        if (pEditIntent != null) {
            views.setOnClickPendingIntent(R.id.widget_button, pEditIntent);
            views.setOnClickPendingIntent(R.id.widget_title, pEditIntent);
        }

        return views;
    }

    /**
     * The reason we use a bunch of different but almost identical layouts is that there is a bug with
     * Android 2.1 (level 7) that doesn't allow setting backgrounds on remote views. I know it's lame,
     * but I didn't see a better solution. Alternatively, we could disallow theming widgets on
     * Android 2.1.
     */
    private RemoteViews getThemedRemoteViews(Context context) {
        String packageName = context.getPackageName();
        Resources r = context.getResources();
        int layout;
        RemoteViews views;

        int titleColor;
        int buttonDrawable;

        if (ThemeService.isDarkWidgetTheme()) {
            layout = R.layout.widget_initialized_dark;
            titleColor = r.getColor(R.color.widget_text_color_dark);
            buttonDrawable = R.drawable.ic_action_new_light;
        } else {
            layout = R.layout.widget_initialized;
            titleColor = r.getColor(R.color.widget_text_color_light);
            buttonDrawable = R.drawable.ic_action_new;
        }

        views = new RemoteViews(packageName, layout);
        views.setTextColor(R.id.widget_title, titleColor);
        views.setInt(R.id.widget_button, "setImageResource", buttonDrawable);
        return views;
    }
}