tasks/app/src/main/java/com/todoroo/astrid/service/TaskMover.java

package com.todoroo.astrid.service;

import static com.google.common.collect.Lists.newArrayList;
import static com.google.common.collect.Lists.transform;
import static com.todoroo.andlib.utility.DateUtilities.now;
import static java.util.Collections.emptyList;
import static java.util.Collections.singletonList;

import com.todoroo.astrid.api.CaldavFilter;
import com.todoroo.astrid.api.Filter;
import com.todoroo.astrid.api.GtasksFilter;
import com.todoroo.astrid.dao.TaskDao;
import com.todoroo.astrid.data.SyncFlags;
import com.todoroo.astrid.data.Task;
import java.util.List;
import javax.inject.Inject;
import org.tasks.data.CaldavDao;
import org.tasks.data.CaldavTask;
import org.tasks.data.GoogleTask;
import org.tasks.data.GoogleTaskDao;
import org.tasks.data.GoogleTaskListDao;
import org.tasks.preferences.Preferences;
import org.tasks.sync.SyncAdapters;

public class TaskMover {
  private final TaskDao taskDao;
  private final CaldavDao caldavDao;
  private final GoogleTaskDao googleTaskDao;
  private final SyncAdapters syncAdapters;
  private final GoogleTaskListDao googleTaskListDao;
  private final Preferences preferences;

  @Inject
  public TaskMover(
      TaskDao taskDao,
      CaldavDao caldavDao,
      GoogleTaskDao googleTaskDao,
      SyncAdapters syncAdapters,
      GoogleTaskListDao googleTaskListDao,
      Preferences preferences) {
    this.taskDao = taskDao;
    this.caldavDao = caldavDao;
    this.googleTaskDao = googleTaskDao;
    this.syncAdapters = syncAdapters;
    this.googleTaskListDao = googleTaskListDao;
    this.preferences = preferences;
  }

  public void move(Long task, Filter selectedList) {
    move(singletonList(task), selectedList);
  }

  public void move(List<Long> tasks, Filter selectedList) {
    tasks = newArrayList(tasks);
    tasks.removeAll(googleTaskDao.findChildrenInList(tasks));
    tasks.removeAll(caldavDao.findChildrenInList(tasks));
    for (Task task : taskDao.fetch(tasks)) {
      performMove(task, selectedList);
      task.putTransitory(SyncFlags.GTASKS_SUPPRESS_SYNC, true);
      task.setModificationDate(now());
      taskDao.save(task);
    }
    if (selectedList instanceof CaldavFilter) {
      caldavDao.updateParents((((CaldavFilter) selectedList).getUuid()));
    }
    syncAdapters.sync();
  }

  public Filter getSingleFilter(List<Long> tasks) {
    List<String> caldavCalendars = caldavDao.getCalendars(tasks);
    List<String> googleTaskLists = googleTaskDao.getLists(tasks);
    if (caldavCalendars.isEmpty()) {
      if (googleTaskLists.size() == 1) {
        return new GtasksFilter(googleTaskListDao.getByRemoteId(googleTaskLists.get(0)));
      }
    } else if (googleTaskLists.isEmpty()) {
      if (caldavCalendars.size() == 1) {
        return new CaldavFilter(caldavDao.getCalendar(caldavCalendars.get(0)));
      }
    }
    return null;
  }

  private void performMove(Task task, Filter selectedList) {
    long id = task.getId();
    GoogleTask googleTask = googleTaskDao.getByTaskId(id);
    List<GoogleTask> googleTaskChildren = emptyList();
    List<CaldavTask> caldavChildren = emptyList();
    if (googleTask != null
        && selectedList instanceof GtasksFilter
        && googleTask.getListId().equals(((GtasksFilter) selectedList).getRemoteId())) {
      return;
    }
    CaldavTask caldavTask = caldavDao.getTask(id);
    if (caldavTask != null
        && selectedList instanceof CaldavFilter
        && caldavTask.getCalendar().equals(((CaldavFilter) selectedList).getUuid())) {
      return;
    }
    task.putTransitory(SyncFlags.FORCE_SYNC, true);
    if (googleTask != null) {
      googleTaskChildren = googleTaskDao.getChildren(id);
      googleTaskDao.markDeleted(now(), id);
    }

    if (caldavTask != null) {
      List<Long> toDelete = newArrayList(caldavTask.getTask());
      List<Long> childIds = caldavDao.getChildren(caldavTask.getTask());
      if (!childIds.isEmpty()) {
        caldavChildren = caldavDao.getTasks(childIds);
        toDelete.addAll(childIds);
      }
      caldavDao.markDeleted(now(), toDelete);
    }

    if (selectedList instanceof GtasksFilter) {
      String listId = ((GtasksFilter) selectedList).getRemoteId();
      googleTaskDao.insertAndShift(new GoogleTask(id, listId), preferences.addGoogleTasksToTop());
      if (!googleTaskChildren.isEmpty()) {
        googleTaskDao.insert(
            transform(
                googleTaskChildren,
                child -> {
                  GoogleTask newChild = new GoogleTask(child.getTask(), listId);
                  newChild.setOrder(child.getOrder());
                  newChild.setParent(id);
                  return newChild;
                }));
      }
      if (!caldavChildren.isEmpty()) {
        List<GoogleTask> children = newArrayList();
        for (int i = 0 ; i < caldavChildren.size() ; i++) {
          CaldavTask child = caldavChildren.get(i);
          GoogleTask newChild = new GoogleTask(child.getTask(), listId);
          newChild.setOrder(i);
          newChild.setParent(id);
          children.add(newChild);
        }
        googleTaskDao.insert(children);
      }
    } else if (selectedList instanceof CaldavFilter) {
      String listId = ((CaldavFilter) selectedList).getUuid();
      CaldavTask newParent = caldavTask == null
          ? new CaldavTask(id, listId)
          : new CaldavTask(id, listId, caldavTask.getRemoteId(), caldavTask.getObject());
      caldavDao.insert(newParent);
      caldavDao.insert(transform(googleTaskChildren, child -> {
        CaldavTask newChild = new CaldavTask(child.getTask(), listId);
        newChild.setRemoteParent(newParent.getRemoteId());
        return newChild;
      }));
      caldavDao.insert(transform(caldavChildren, child -> {
        CaldavTask newChild = new CaldavTask(child.getTask(), listId, child.getRemoteId(), child.getObject());
        newChild.setVtodo(child.getVtodo());
        newChild.setRemoteParent(child.getRemoteParent());
        return newChild;
      }));
    }
  }
}