1 files changed, 22 insertions, 5 deletions
diff --git a/app/src/main/java/fr/gouv/etalab/mastodon/asynctasks/RetrieveScheduledTootsAsyncTask.java b/app/src/main/java/fr/gouv/etalab/mastodon/asynctasks/RetrieveScheduledTootsAsyncTask.java
index f1757ec75..43c3cb2e3 100644
--- a/app/src/main/java/fr/gouv/etalab/mastodon/asynctasks/RetrieveScheduledTootsAsyncTask.java
+++ b/app/src/main/java/fr/gouv/etalab/mastodon/asynctasks/RetrieveScheduledTootsAsyncTask.java
@@ -18,15 +18,21 @@ import android.content.Context;
 import android.database.sqlite.SQLiteDatabase;
 import android.os.AsyncTask;
 import android.os.SystemClock;
+
 import com.evernote.android.job.JobManager;
 import com.evernote.android.job.JobRequest;
+
 import java.lang.ref.WeakReference;
 import java.util.List;
 import java.util.Set;
+
 import fr.gouv.etalab.mastodon.client.Entities.StoredStatus;
+import fr.gouv.etalab.mastodon.fragments.DisplayScheduledTootsFragment;
 import fr.gouv.etalab.mastodon.helper.Helper;
 import fr.gouv.etalab.mastodon.interfaces.OnRetrieveScheduledTootsInterface;
+import fr.gouv.etalab.mastodon.jobs.ScheduledBoostsSyncJob;
 import fr.gouv.etalab.mastodon.jobs.ScheduledTootsSyncJob;
+import fr.gouv.etalab.mastodon.sqlite.BoostScheduleDAO;
 import fr.gouv.etalab.mastodon.sqlite.Sqlite;
 import fr.gouv.etalab.mastodon.sqlite.StatusStoredDAO;
 
@@ -42,20 +48,28 @@ public class RetrieveScheduledTootsAsyncTask extends AsyncTask<Void, Void, Void>
     private OnRetrieveScheduledTootsInterface listener;
     private List<StoredStatus> storedStatuses;
     private WeakReference<Context> contextReference;
+    private DisplayScheduledTootsFragment.typeOfSchedule type;
 
-    public RetrieveScheduledTootsAsyncTask(Context context, OnRetrieveScheduledTootsInterface onRetrieveScheduledTootsInterface){
+    public RetrieveScheduledTootsAsyncTask(Context context, DisplayScheduledTootsFragment.typeOfSchedule type, OnRetrieveScheduledTootsInterface onRetrieveScheduledTootsInterface){
         this.contextReference = new WeakReference<>(context);
         this.listener = onRetrieveScheduledTootsInterface;
-
+        this.type = type;
     }
 
     @Override
     protected Void doInBackground(Void... params) {
         SQLiteDatabase db = Sqlite.getInstance(this.contextReference.get(), Sqlite.DB_NAME, null, Sqlite.DB_VERSION).open();
         //Retrieves job asked by the user
-        storedStatuses = new StatusStoredDAO(this.contextReference.get(), db).getAllScheduled();
+        if( type == DisplayScheduledTootsFragment.typeOfSchedule.TOOT)
+            storedStatuses = new StatusStoredDAO(this.contextReference.get(), db).getAllScheduled();
+        else if(type == DisplayScheduledTootsFragment.typeOfSchedule.BOOST )
+            storedStatuses = new BoostScheduleDAO(this.contextReference.get(), db).getAllScheduled();
         //Retrieves real jobs still waiting
-        Set<JobRequest> jobRequests = JobManager.instance().getAllJobRequestsForTag(ScheduledTootsSyncJob.SCHEDULED_TOOT);
+        Set<JobRequest> jobRequests = null;
+        if( type == DisplayScheduledTootsFragment.typeOfSchedule.TOOT)
+            jobRequests = JobManager.instance().getAllJobRequestsForTag(ScheduledTootsSyncJob.SCHEDULED_TOOT);
+        else if(type == DisplayScheduledTootsFragment.typeOfSchedule.BOOST )
+            jobRequests = JobManager.instance().getAllJobRequestsForTag(ScheduledBoostsSyncJob.SCHEDULED_BOOST);
         int[] jobIds;
         if( jobRequests != null && jobRequests.size() > 0 ){
             int i = 0;
@@ -72,7 +86,10 @@ public class RetrieveScheduledTootsAsyncTask extends AsyncTask<Void, Void, Void>
             for(StoredStatus ss: storedStatuses){
                 if (!Helper.isJobPresent(jobIds, ss.getJobId())){
                     //JobId is fixed to -1 which means an error occured (it was never sent)
-                    new StatusStoredDAO(this.contextReference.get(), db).updateJobId(ss.getId(),-1);
+                    if( type == DisplayScheduledTootsFragment.typeOfSchedule.TOOT)
+                        new StatusStoredDAO(this.contextReference.get(), db).updateJobId(ss.getId(),-1);
+                    else if(type == DisplayScheduledTootsFragment.typeOfSchedule.BOOST )
+                        new BoostScheduleDAO(this.contextReference.get(), db).updateJobId(ss.getId(),-1);
                 }
             }
             //Lets time to update db before dispaying