/*
 * Copyright (C) 2008 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.systemui.statusbar.connectivity;

import static com.android.settingslib.flags.Flags.newStatusBarIcons;

import com.android.settingslib.AccessibilityContentDescriptions;
import com.android.settingslib.R;
import com.android.settingslib.SignalIcon.IconGroup;

/** */
public class WifiIcons {

    public static final int[] WIFI_FULL_ICONS = getIconsBasedOnFlag();
    public static final int WIFI_NO_SIGNAL = WIFI_FULL_ICONS[0];

    /**
     * Check the aconfig flag to decide on which icons to use. Can be removed once the flag is gone
     */
    private static int[] getIconsBasedOnFlag() {
        if (newStatusBarIcons()) {
            // TODO(b/396664075):
            // The new wifi icons only define a range of [0, 3]. Since this array is indexed on
            // level, we can simulate the range squash by mapping both level 3 to drawn-level 2, and
            // level 4 to drawn-level 3
            return new int[] {
                R.drawable.ic_wifi_0,
                R.drawable.ic_wifi_1,
                R.drawable.ic_wifi_2,
                R.drawable.ic_wifi_2,
                R.drawable.ic_wifi_3
            };
        } else {
            return new int[] {
                com.android.internal.R.drawable.ic_wifi_signal_0,
                com.android.internal.R.drawable.ic_wifi_signal_1,
                com.android.internal.R.drawable.ic_wifi_signal_2,
                com.android.internal.R.drawable.ic_wifi_signal_3,
                com.android.internal.R.drawable.ic_wifi_signal_4
            };
        }
    }

    public static final int[] WIFI_NO_INTERNET_ICONS = getErrorIconsBasedOnFlag();

    private static int [] getErrorIconsBasedOnFlag() {
        if (newStatusBarIcons()) {
            // See above note, new wifi icons only have 3 bars, so levels 2 and 3 are the same
            return new int[] {
                R.drawable.ic_wifi_0_error,
                R.drawable.ic_wifi_1_error,
                R.drawable.ic_wifi_2_error,
                R.drawable.ic_wifi_2_error,
                R.drawable.ic_wifi_3_error,
            };
        } else {
            return new int[] {
                R.drawable.ic_no_internet_wifi_signal_0,
                R.drawable.ic_no_internet_wifi_signal_1,
                R.drawable.ic_no_internet_wifi_signal_2,
                R.drawable.ic_no_internet_wifi_signal_3,
                R.drawable.ic_no_internet_wifi_signal_4
            };
        }
    }

    public static final int[][] QS_WIFI_SIGNAL_STRENGTH = {
            WIFI_NO_INTERNET_ICONS,
            WIFI_FULL_ICONS
    };

    static final int[][] WIFI_SIGNAL_STRENGTH = QS_WIFI_SIGNAL_STRENGTH;

    public static final int QS_WIFI_NO_NETWORK = com.android.internal.R.drawable.ic_wifi_signal_0;

    public static final int WIFI_NO_NETWORK = getNoNetworkBasedOnFlag();

    private static int getNoNetworkBasedOnFlag() {
        if (newStatusBarIcons()) {
            return R.drawable.ic_wifi_0;
        } else {
            return QS_WIFI_NO_NETWORK;
        }
    }

    static final int WIFI_LEVEL_COUNT = WIFI_SIGNAL_STRENGTH[0].length;

    public static final IconGroup UNMERGED_WIFI = new IconGroup(
            "Wi-Fi Icons",
            WifiIcons.WIFI_SIGNAL_STRENGTH,
            WifiIcons.QS_WIFI_SIGNAL_STRENGTH,
            AccessibilityContentDescriptions.WIFI_CONNECTION_STRENGTH,
            WifiIcons.WIFI_NO_NETWORK,
            WifiIcons.QS_WIFI_NO_NETWORK,
            WifiIcons.WIFI_NO_NETWORK,
            WifiIcons.QS_WIFI_NO_NETWORK,
            AccessibilityContentDescriptions.WIFI_NO_CONNECTION
    );
}