/*
 * Copyright (C) 2021 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.server.wifi.util;

import android.net.wifi.WifiManager;
import android.net.wifi.WifiManager.ApiType;
import android.util.SparseArray;

import java.io.PrintWriter;

/**
 * Manage multiple last caller info
 */
public class LastCallerInfoManager {
    private final SparseArray<LastCallerInfo> mLastCallerInfoMap = new SparseArray<>();

    /**
     * Store the last caller information for the API
     */
    public void put(@ApiType int apiName, int tid, int uid, int pid, String packageName,
            boolean toggleState) {
        synchronized (mLastCallerInfoMap) {
            LastCallerInfo callerInfo = new LastCallerInfo(tid, uid, pid, packageName, toggleState);
            mLastCallerInfoMap.put(apiName, callerInfo);
        }
    }

    /**
     * Get the LastCallerInfo for a particular api.
     * @param apiName Name of the API
     * @return The LastCallerInfo, or null if not available.
     */
    public LastCallerInfo get(@ApiType int apiName) {
        synchronized (mLastCallerInfoMap) {
            return mLastCallerInfoMap.get(apiName);
        }
    }

    /**
     * Convert int constant into API String name
     */
    private String convertApiName(@ApiType int key) {
        switch (key) {
            case WifiManager.API_SCANNING_ENABLED:
                return "ScanningEnabled";
            case WifiManager.API_WIFI_ENABLED:
                return "WifiEnabled";
            case WifiManager.API_SOFT_AP:
                return "SoftAp";
            case WifiManager.API_TETHERED_HOTSPOT:
                return "TetheredHotspot";
            case WifiManager.API_AUTOJOIN_GLOBAL:
                return "AutojoinGlobal";
            case WifiManager.API_SET_SCAN_SCHEDULE:
                return "SetScanScanSchedule";
            case WifiManager.API_SET_ONE_SHOT_SCREEN_ON_CONNECTIVITY_SCAN_DELAY:
                return "API_SET_ONE_SHOT_SCREEN_ON_CONNECTIVITY_SCAN_DELAY";
            case WifiManager.API_SET_NETWORK_SELECTION_CONFIG:
                return "API_SET_NETWORK_SELECTION_CONFIG";
            case WifiManager.API_SET_THIRD_PARTY_APPS_ENABLING_WIFI_CONFIRMATION_DIALOG:
                return "API_SET_THIRD_PARTY_APPS_ENABLING_WIFI_CONFIRMATION_DIALOG";
            case WifiManager.API_ADD_NETWORK:
                return "API_ADD_NETWORK";
            case WifiManager.API_UPDATE_NETWORK:
                return "API_UPDATE_NETWORK";
            case WifiManager.API_ALLOW_AUTOJOIN:
                return "API_ALLOW_AUTOJOIN";
            case WifiManager.API_CONNECT_CONFIG:
                return "API_CONNECT_CONFIG";
            case WifiManager.API_CONNECT_NETWORK_ID:
                return "API_CONNECT_NETWORK_ID";
            case WifiManager.API_DISABLE_NETWORK:
                return "API_DISABLE_NETWORK";
            case WifiManager.API_ENABLE_NETWORK:
                return "API_ENABLE_NETWORK";
            case WifiManager.API_FORGET:
                return "API_FORGET";
            case WifiManager.API_SAVE:
                return "API_SAVE";
            case WifiManager.API_START_SCAN:
                return "API_START_SCAN";
            case WifiManager.API_START_LOCAL_ONLY_HOTSPOT:
                return "API_START_LOCAL_ONLY_HOTSPOT";
            case WifiManager.API_P2P_DISCOVER_PEERS:
                return "API_P2P_DISCOVER_PEERS";
            case WifiManager.API_P2P_DISCOVER_PEERS_ON_SOCIAL_CHANNELS:
                return "API_P2P_DISCOVER_PEERS_ON_SOCIAL_CHANNELS";
            case WifiManager.API_P2P_DISCOVER_PEERS_ON_SPECIFIC_FREQUENCY:
                return "API_P2P_DISCOVER_PEERS_ON_SPECIFIC_FREQUENCY";
            case WifiManager.API_P2P_STOP_PEER_DISCOVERY:
                return "API_P2P_STOP_PEER_DISCOVERY";
            case WifiManager.API_P2P_CONNECT:
                return "API_P2P_CONNECT";
            case WifiManager.API_P2P_CANCEL_CONNECT:
                return "API_P2P_CANCEL_CONNECT";
            case WifiManager.API_P2P_CREATE_GROUP:
                return "API_P2P_CREATE_GROUP";
            case WifiManager.API_P2P_CREATE_GROUP_P2P_CONFIG:
                return "API_P2P_CREATE_GROUP_P2P_CONFIG";
            case WifiManager.API_P2P_REMOVE_GROUP:
                return "API_P2P_REMOVE_GROUP";
            case WifiManager.API_P2P_START_LISTENING:
                return "API_P2P_START_LISTENING";
            case WifiManager.API_P2P_STOP_LISTENING:
                return "API_P2P_STOP_LISTENING";
            case WifiManager.API_P2P_SET_CHANNELS:
                return "API_P2P_SET_CHANNELS";
            case WifiManager.API_WIFI_SCANNER_START_SCAN:
                return "API_WIFI_SCANNER_START_SCAN";
            case WifiManager.API_SET_TDLS_ENABLED:
                return "API_SET_TDLS_ENABLED";
            case WifiManager.API_SET_TDLS_ENABLED_WITH_MAC_ADDRESS:
                return "API_SET_TDLS_ENABLED_WITH_MAC_ADDRESS";
            case WifiManager.API_SET_PNO_SCAN_ENABLED:
                return "API_SET_PNO_SCAN_ENABLED";
            default:
                return "Unknown";
        }
    }

    /**
     * Print the last caller info for the APIs tracked
     */
    public void dump(PrintWriter pw) {
        pw.println("Dump of LastCallerInfoManager");
        for (int i = 0; i < mLastCallerInfoMap.size(); i++) {
            String apiName = convertApiName(mLastCallerInfoMap.keyAt(i));
            String callerInfo = mLastCallerInfoMap.valueAt(i).toString();
            pw.println(new StringBuilder()
                    .append("API key=")
                    .append(mLastCallerInfoMap.keyAt(i)).append(" API name=")
                    .append(apiName)
                    .append(": ")
                    .append(callerInfo));
        }
    }

    /**
     * Last caller info
     */
    public static class LastCallerInfo {
        private int mTid;
        private int mUid;
        private int mPid;
        private String mPackageName;
        private boolean mToggleState;

        public LastCallerInfo(int tid, int uid, int pid, String packageName, boolean toggleState) {
            mTid = tid;
            mUid = uid;
            mPid = pid;
            mPackageName = packageName;
            mToggleState = toggleState;
        }

        /**
         * Gets the packageName.
         */
        public String getPackageName() {
            return mPackageName;
        }

        /**
         * Gets the toggleState.
         */
        public boolean getToggleState() {
            return mToggleState;
        }

        /**
         * Convert the last caller info into String format
         */
        public String toString() {
            StringBuilder sb = new StringBuilder();
            sb.append("tid=").append(mTid).append(" uid=").append(mUid)
                    .append(" pid=").append(mPid).append(" packageName=").append(mPackageName)
                    .append(" toggleState=").append(mToggleState);
            return sb.toString();
        }
    }
}