/*
 * Copyright 2016 Peter Kenji Yamanaka
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.pyamsoft.powermanager.dagger.wrapper;

import android.app.AlarmManager;
import android.app.PendingIntent;
import android.content.Context;
import android.content.Intent;
import android.os.Build;
import android.support.annotation.NonNull;
import java.util.Date;
import javax.inject.Inject;
import timber.log.Timber;

class JobQueuerWrapperImpl implements JobQueuerWrapper {

  @NonNull private final Context appContext;
  @NonNull private final AlarmManager alarmManager;

  @Inject JobQueuerWrapperImpl(@NonNull Context appContext) {
    this.appContext = appContext;
    this.alarmManager = (AlarmManager) appContext.getSystemService(Context.ALARM_SERVICE);
  }

  @Override public void cancel(@NonNull Intent intent) {
    Timber.w("Cancel Alarm: %s", intent);
    alarmManager.cancel(
        PendingIntent.getService(appContext, 0, intent, PendingIntent.FLAG_UPDATE_CURRENT));
  }

  @Override public void set(@NonNull Intent intent, long time) {
    final Date date = new Date(time);
    if (Build.VERSION.SDK_INT >= Build.VERSION_CODES.M) {
      Timber.i("Set and allow while idle: %s at %s", intent, date);
      alarmManager.setExactAndAllowWhileIdle(AlarmManager.RTC_WAKEUP, time,
          PendingIntent.getService(appContext, 0, intent, PendingIntent.FLAG_UPDATE_CURRENT));
    } else {
      Timber.i("Set: %s at %s", intent, date);
      alarmManager.setExact(AlarmManager.RTC_WAKEUP, time,
          PendingIntent.getService(appContext, 0, intent, PendingIntent.FLAG_UPDATE_CURRENT));
    }
  }
}

@@ -1380,28 +1380,43 @@
     
-alarmManager.cancel(
+final PendingIntent pendingIntent =
 %0A   
@@ -1498,24 +1498,91 @@
 ATE_CURRENT)
+;%0A    alarmManager.cancel(pendingIntent);%0A    pendingIntent.cancel(
 );%0A  %7D%0A%0A  @O

Make sure to cancel the pending intent too in case it wasn't registered