// Copyright 2022 The Cobalt 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 dev.cobalt.coat;

import static dev.cobalt.util.Log.TAG;

import android.content.Context;
import com.google.android.gms.ads.identifier.AdvertisingIdClient;
import com.google.android.gms.common.GooglePlayServicesNotAvailableException;
import com.google.android.gms.common.GooglePlayServicesRepairableException;
import dev.cobalt.util.Log;
import java.io.IOException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

/** A class for managing the IfA for CoAT. https://developer.android.com/training/articles/ad-id */
public class AdvertisingId {
  private final Context context;
  private final ExecutorService singleThreadExecutor;
  private long lastRefreshed;
  private final long cacheTtlMs = 1000 * 60 * 10; // 10 minutes.
  private AdvertisingIdClient.Info advertisingIdInfo;

  public AdvertisingId(Context context) {
    this.context = context;
    this.lastRefreshed = 0;
    this.singleThreadExecutor = Executors.newSingleThreadExecutor();
    refresh();
  }

  private void refresh() {
    if (System.currentTimeMillis() - lastRefreshed < cacheTtlMs) {
      // Cache is up to date.
      return;
    }
    singleThreadExecutor.execute(
        () -> {
          try {
            advertisingIdInfo = AdvertisingIdClient.getAdvertisingIdInfo(context);
            lastRefreshed = System.currentTimeMillis();
            Log.i(TAG, "Successfully retrieved Advertising ID (IfA).");
          } catch (IOException
              | GooglePlayServicesNotAvailableException
              | GooglePlayServicesRepairableException e) {
            Log.e(TAG, "Failed to retrieve Advertising ID (IfA).");
          }
        });
  }

  public String getId() {
    String result = "";
    if (lastRefreshed != 0) {
      result = advertisingIdInfo.getId();
      refresh();
    }
    Log.d(TAG, "Returning IfA getId: " + result);
    return result;
  }

  public boolean isLimitAdTrackingEnabled() {
    boolean result = false;
    if (lastRefreshed != 0) {
      result = advertisingIdInfo.isLimitAdTrackingEnabled();
      refresh();
    }
    Log.d(TAG, "Returning IfA LimitedAdTrackingEnabled: " + Boolean.toString(result));
    return result;
  }
}