/*
 * Copyright (C) 2018 The Android Open Source Project
 *
 * 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.android.tradefed.invoker.shard.token;

import com.android.tradefed.device.DeviceNotAvailableException;
import com.android.tradefed.device.ITestDevice;
import com.android.tradefed.device.StubDevice;
import com.android.tradefed.device.helper.TelephonyHelper;
import com.android.tradefed.device.helper.TelephonyHelper.SimCardInformation;
import com.android.tradefed.log.LogUtil.CLog;

import com.google.common.annotations.VisibleForTesting;
import com.google.common.base.Strings;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;

/** Token provider for telephony related tokens. */
public class TelephonyTokenProvider implements ITokenProvider {

    public static final String ORANGE_SIM_ID = "20801";
    public static final String THALES_GEMALTO_SIM_ID = "00101";
    public static final String GSM_OPERATOR_PROP = "gsm.sim.operator.numeric";

    private static final String[] SECURE_ELEMENT_SIM_IDS = {ORANGE_SIM_ID, THALES_GEMALTO_SIM_ID};

    @Override
    public boolean hasToken(ITestDevice device, TokenProperty token) {
        if (device.getIDevice() instanceof StubDevice) {
            return false;
        }
        try {
            SimCardInformation info = getSimInfo(device);
            if (info == null || !info.mHasTelephonySupport) {
                CLog.e("SimcardInfo: %s", info);
                return false;
            }
            switch (token) {
                case SIM_CARD:
                    // 5 is sim state ready
                    if ("5".equals(info.mSimState)) {
                        return true;
                    }
                    CLog.w(
                            "%s cannot run with token '%s' - Sim info: %s",
                            device.getSerialNumber(), token, info);
                    return false;
                case UICC_SIM_CARD:
                    if (info.mCarrierPrivileges) {
                        return true;
                    }
                    CLog.w(
                            "%s cannot run with token '%s' - Sim info: %s",
                            device.getSerialNumber(), token, info);
                    return false;
                case SECURE_ELEMENT_SIM_CARD:
                    if (info.mHasSecuredElement && info.mHasSeService) {
                        List<String> simProp =
                                getOptionalDualSimProperty(device, GSM_OPERATOR_PROP);
                        // Check whether simProp contains any secure element SIM IDs.
                        // TODO(b/180568443): Create a mechanism of checking whether a SIM card
                        // matches the requirements of secure element test.
                        if (!Collections.disjoint(simProp, Arrays.asList(SECURE_ELEMENT_SIM_IDS))) {
                            return true;
                        } else {
                            CLog.w(
                                    "%s doesn't have a Orange Sim card (%s) for secured elements."
                                            + " %s returned: '%s'",
                                    device.getSerialNumber(),
                                    ORANGE_SIM_ID,
                                    GSM_OPERATOR_PROP,
                                    simProp);
                        }
                    }

                    CLog.w(
                            "%s cannot run with token '%s' - Sim info: %s",
                            device.getSerialNumber(), token, info);
                    return false;
                default:
                    CLog.w("Token '%s' doesn't match any TelephonyTokenProvider tokens.", token);
                    return false;
            }
        } catch (DeviceNotAvailableException e) {
            CLog.e("Ignoring DNAE: %s", e);
        }
        return false;
    }

    @VisibleForTesting
    SimCardInformation getSimInfo(ITestDevice device) throws DeviceNotAvailableException {
        return TelephonyHelper.getSimInfo(device);
    }

    private List<String> getOptionalDualSimProperty(ITestDevice device, String property)
            throws DeviceNotAvailableException {
        String propRes = device.getProperty(property);
        CLog.d("queried sim property: '%s'. result: '%s'", property, propRes);
        if (Strings.isNullOrEmpty(propRes)) {
            return new ArrayList<>();
        }
        String[] splitProp = propRes.split(",");
        List<String> results = new ArrayList<>();
        for (String p : splitProp) {
            results.add(p);
        }
        return results;
    }
}