/*
 * Copyright (C) 2020 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.ims.rcs.uce.presence.publish;

import android.content.Context;
import android.net.Uri;
import android.telecom.PhoneAccount;
import android.telephony.PhoneNumberUtils;
import android.telephony.TelephonyManager;
import android.telephony.ims.feature.RcsFeature.RcsImsCapabilities;
import android.telephony.ims.feature.RcsFeature.RcsImsCapabilities.RcsImsCapabilityFlag;
import android.text.TextUtils;
import android.util.Log;

import com.android.i18n.phonenumbers.NumberParseException;
import com.android.i18n.phonenumbers.PhoneNumberUtil;
import com.android.i18n.phonenumbers.Phonenumber;
import com.android.ims.rcs.uce.util.UceUtils;

import java.util.Arrays;

/**
 * The util class of publishing device's capabilities.
 */
public class PublishUtils {
    private static final String LOG_TAG = UceUtils.getLogPrefix() + "PublishUtils";

    private static final String SCHEME_SIP = "sip";
    private static final String SCHEME_TEL = "tel";
    private static final String DOMAIN_SEPARATOR = "@";

    /**
     * @return the contact URI of this device for either a PRESENCE or OPTIONS capabilities request.
     * We will first try to use the IMS service associated URIs from the p-associated-uri header
     * in the IMS registration response. If this is not available, we will fall back to using the
     * SIM card information to generate the URI.
     */
    public static Uri getDeviceContactUri(Context context, int subId,
            DeviceCapabilityInfo deviceCap, boolean isForPresence) {
        boolean preferTelUri = false;
        if (isForPresence) {
            preferTelUri = UceUtils.isTelUriForPidfXmlEnabled(context, subId);
        }
        // Get the uri from the IMS p-associated-uri header which is provided by the IMS service.
        Uri contactUri = deviceCap.getImsAssociatedUri(preferTelUri);
        if (contactUri != null) {
            Uri convertedUri = preferTelUri ? getConvertedTelUri(context, contactUri) : contactUri;
            Log.d(LOG_TAG, "getDeviceContactUri: returning "
                    + (contactUri.equals(convertedUri) ? "found" : "converted")
                    + " ims associated uri");
            return contactUri;
        }

        // No IMS service provided URIs, so generate the contact uri from ISIM.
        TelephonyManager telephonyManager = getTelephonyManager(context, subId);
        if (telephonyManager == null) {
            Log.w(LOG_TAG, "getDeviceContactUri: TelephonyManager is null");
            return null;
        }
        contactUri = getContactUriFromIsim(telephonyManager);
        if (contactUri != null) {
            Log.d(LOG_TAG, "getDeviceContactUri: impu");
            if (preferTelUri) {
                return getConvertedTelUri(context, contactUri);
            } else {
                return contactUri;
            }
        } else {
            Log.d(LOG_TAG, "getDeviceContactUri: line number");
            if (preferTelUri) {
                return getConvertedTelUri(context, getContactUriFromLine1Number(telephonyManager));
            } else {
                return getContactUriFromLine1Number(telephonyManager);
            }
        }
    }

    /**
     * Find all instances of sip/sips/tel URIs containing PII and replace them.
     * <p>
     * This is used for removing PII in logging.
     * @param source The source string to remove the phone numbers from.
     * @return A version of the given string with SIP URIs removed.
     */
    public static String removeNumbersFromUris(String source) {
        // Replace only the number portion in the sip/sips/tel URI
        return source.replaceAll("(?:sips?|tel):(\\+?[\\d\\-]+)", "[removed]");
    }

    private static Uri getContactUriFromIsim(TelephonyManager telephonyManager) {
        // Get the home network domain and the array of the public user identities
        String domain = telephonyManager.getIsimDomain();
        String[] impus = telephonyManager.getIsimImpu();

        if (TextUtils.isEmpty(domain) || impus == null) {
            Log.d(LOG_TAG, "getContactUriFromIsim: domain is null=" + TextUtils.isEmpty(domain));
            Log.d(LOG_TAG, "getContactUriFromIsim: impu is null=" +
                    ((impus == null || impus.length == 0) ? "true" : "false"));
            return null;
        }

        for (String impu : impus) {
            if (TextUtils.isEmpty(impu)) continue;
            Uri impuUri = Uri.parse(impu);
            String scheme = impuUri.getScheme();
            String schemeSpecificPart = impuUri.getSchemeSpecificPart();
            if (SCHEME_SIP.equals(scheme) && !TextUtils.isEmpty(schemeSpecificPart) &&
                    schemeSpecificPart.endsWith(domain)) {
                return impuUri;
            }
        }
        Log.d(LOG_TAG, "getContactUriFromIsim: there is no impu matching the domain");
        return null;
    }

    private static Uri getContactUriFromLine1Number(TelephonyManager telephonyManager) {
        String phoneNumber = formatPhoneNumber(telephonyManager.getLine1Number());
        if (TextUtils.isEmpty(phoneNumber)) {
            Log.w(LOG_TAG, "Cannot get the phone number");
            return null;
        }

        String domain = telephonyManager.getIsimDomain();
        if (!TextUtils.isEmpty(domain)) {
            return Uri.fromParts(SCHEME_SIP, phoneNumber + DOMAIN_SEPARATOR + domain, null);
        } else {
            return Uri.fromParts(SCHEME_TEL, phoneNumber, null);
        }
    }

    private static String formatPhoneNumber(final String phoneNumber) {
        if (TextUtils.isEmpty(phoneNumber)) {
            Log.w(LOG_TAG, "formatPhoneNumber: phone number is empty");
            return null;
        }
        String number = PhoneNumberUtils.stripSeparators(phoneNumber);
        return PhoneNumberUtils.normalizeNumber(number);
    }

    private static TelephonyManager getTelephonyManager(Context context, int subId) {
        TelephonyManager telephonyManager = context.getSystemService(TelephonyManager.class);
        if (telephonyManager == null) {
            return null;
        } else {
            return telephonyManager.createForSubscriptionId(subId);
        }
    }

    /**
     * @return a TEL URI version of the contact URI if given a SIP URI. If given a TEL URI, this
     * method will return the same value given.
     */
    private static Uri getConvertedTelUri(Context context, Uri contactUri) {
        if (contactUri == null) {
            return null;
        }
        if (contactUri.getScheme().equalsIgnoreCase(SCHEME_SIP)) {
            TelephonyManager manager = context.getSystemService(TelephonyManager.class);
            if (manager.getIsimDomain() == null) {
                return contactUri;
            }

            String numbers = contactUri.getSchemeSpecificPart();
            String[] numberParts = numbers.split("[@;:]");
            String number = numberParts[0];

            String simCountryIso = manager.getSimCountryIso();
            if (!TextUtils.isEmpty(simCountryIso)) {
                simCountryIso = simCountryIso.toUpperCase();
                PhoneNumberUtil util = PhoneNumberUtil.getInstance();
                try {
                    Phonenumber.PhoneNumber phoneNumber = util.parse(number, simCountryIso);
                    number = util.format(phoneNumber, PhoneNumberUtil.PhoneNumberFormat.E164);
                    String telUri = SCHEME_TEL + ":" + number;
                    contactUri = Uri.parse(telUri);
                } catch (NumberParseException e) {
                    Log.w(LOG_TAG, "formatNumber: could not format " + number + ", error: " + e);
                }
            }
        }
        return contactUri;
    }

    static @RcsImsCapabilityFlag int getCapabilityType(Context context, int subId) {
        boolean isPresenceSupported = UceUtils.isPresenceSupported(context, subId);
        boolean isSipOptionsSupported = UceUtils.isSipOptionsSupported(context, subId);
        if (isPresenceSupported) {
            return RcsImsCapabilities.CAPABILITY_TYPE_PRESENCE_UCE;
        } else if (isSipOptionsSupported) {
            return RcsImsCapabilities.CAPABILITY_TYPE_OPTIONS_UCE;
        } else {
            // Return NONE when neither OPTIONS nor PRESENCE is supported.
            return RcsImsCapabilities.CAPABILITY_TYPE_NONE;
        }
    }
}