// Copyright 2018 The Bazel Authors. All rights reserved.
//
// 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.google.devtools.build.android.desugar.runtime;

import android.app.DirectAction;
import android.os.Bundle;
import android.os.CancellationSignal;
import android.service.voice.VoiceInteractionSession;
import android.telephony.AvailableNetworkInfo;
import android.telephony.TelephonyManager;
import java.util.List;
import java.util.concurrent.Executor;
import java.util.function.Consumer;

/**
 * Conversion from desugared to built-in {@link Consumer} for calling built-in Android APIs (see
 * b/128638076).
 */
// TODO(b/74087778): Unnecessary when j$.u.f.Consumer becomes subtype of built-in j.u.f.Consumer
@SuppressWarnings("AndroidJdkLibsChecker")
public final class ConsumerWrapper<T> implements Consumer<T> {

  private final j$.util.function.Consumer<T> wrapped;

  private ConsumerWrapper(j$.util.function.Consumer<T> wrapped) {
    this.wrapped = wrapped;
  }

  @Override
  public void accept(T t) {
    wrapped.accept(t);
  }

  public static void setPreferredOpportunisticDataSubscription(
      TelephonyManager receiver,
      int subId,
      boolean needValidation,
      Executor executor,
      j$.util.function.Consumer<Integer> callback) {
    receiver.setPreferredOpportunisticDataSubscription(
        subId,
        needValidation,
        executor,
        callback != null ? new ConsumerWrapper<Integer>(callback) : null);
  }

  public static void updateAvailableNetworks(
      TelephonyManager receiver,
      List<AvailableNetworkInfo> availableNetworks,
      Executor executor,
      j$.util.function.Consumer<Integer> callback) {
    receiver.updateAvailableNetworks(
        availableNetworks,
        executor,
        callback != null ? new ConsumerWrapper<Integer>(callback) : null);
  }

  public static void performDirectAction(
      VoiceInteractionSession receiver,
      DirectAction action,
      Bundle extras,
      CancellationSignal cancellationSignal,
      Executor resultExecutor,
      j$.util.function.Consumer<Bundle> resultListener) {
    receiver.performDirectAction(
        action,
        extras,
        cancellationSignal,
        resultExecutor,
        resultListener != null ? new ConsumerWrapper<Bundle>(resultListener) : null);
  }

  public static void requestDirectActions(
      VoiceInteractionSession receiver,
      VoiceInteractionSession.ActivityId activityId,
      CancellationSignal cancellationSignal,
      Executor resultExecutor,
      j$.util.function.Consumer<List<DirectAction>> callback) {
    receiver.requestDirectActions(
        activityId,
        cancellationSignal,
        resultExecutor,
        callback != null ? new ConsumerWrapper<List<DirectAction>>(callback) : null);
  }
}

@@ -688,23 +688,113 @@
 id.a
-pp.DirectAction
+ccessibilityservice.AccessibilityService;%0Aimport android.app.DirectAction;%0Aimport android.graphics.Bitmap
 ;%0Aim
@@ -3377,11 +3377,334 @@
 ll);%0A  %7D
+%0A%0A  public static boolean takeScreenshot(%0A      AccessibilityService receiver,%0A      int displayId,%0A      Executor resultExecutor,%0A      j$.util.function.Consumer%3CBitmap%3E callback) %7B%0A    return receiver.takeScreenshot(%0A        displayId, resultExecutor, callback != null ? new ConsumerWrapper%3CBitmap%3E(callback) : null);%0A  %7D
 %0A%7D%0A

Support AccessibilityService.takeScreenshot with Java 8 library desugaring.