#!/usr/bin/env python3
#
#   Copyright 20022 - 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.
import time

from acts_contrib.test_utils.power.cellular.ims_api_connector_utils import ImsApiConnector
from acts_contrib.test_utils.power.cellular.ims_api_connector_utils import ImsAppName
from acts_contrib.test_utils.tel.tel_test_utils import set_phone_silent_mode
import acts_contrib.test_utils.power.cellular.cellular_power_preset_base_test as PB


class PowerTelImsPresetTest(PB.PowerCellularPresetLabBaseTest):
    ADB_CMD_ENABLE_IMS = ('am broadcast '
        '-a com.google.android.carrier.action.LOCAL_OVERRIDE '
        '-n com.google.android.carrier/.ConfigOverridingReceiver '
        '--ez carrier_volte_available_bool true '
        '--ez carrier_wfc_ims_available_bool true '
        '--ez carrier_vt_available_bool true '
        '--ez carrier_supports_ss_over_ut_bool true '
        '--ez vonr_setting_visibility_bool true '
        '--ez vonr_enabled_bool true')

    ADB_CMD_DISABLE_IMS = ('am broadcast '
        '-a com.google.android.carrier.action.LOCAL_OVERRIDE '
        '-n com.google.android.carrier/.ConfigOverridingReceiver '
        '--ez carrier_volte_available_bool false '
        '--ez carrier_wfc_ims_available_bool false '
        '--ez carrier_vt_available_bool false '
        '--ez carrier_supports_ss_over_ut_bool false '
        '--ez vonr_setting_visibility_bool false '
        '--ez vonr_enabled_bool false')

    # set NV command
    # !NRCAPA.Gen.VoiceOverNr, 0, 01
    ADB_SET_GOOG_NV = 'echo at+googsetnv="{nv_name}",{index},"{value}" > /dev/umts_router'

    # Key IMS simulator default value
    IMS_CLIENT_DEFAULT_IP = '127.0.0.1'
    IMS_CLIENT_DEFAULT_PORT = 8250
    IMS_CLIENT_DEFAULT_API_TOKEN = 'myclient'
    IMS_API_CONNECTOR_DEFAULT_PORT = 5050

    # IMS available app
    IMS_CLIENT = 'client'
    IMS_SERVER = 'server'

    UE_DEFAULT_NUMBER = '001010123456789'

    def setup_class(self):
        """ Executed only once when initializing the class. """
        super().setup_class()

        # disable mobile data
        self.log.info('Disable mobile data.')
        self.dut.adb.shell('svc data disable')

        # Enable IMS on UE
        self.log.info('Enable VoLTE using adb command.')
        self.dut.adb.shell(self.ADB_CMD_ENABLE_IMS)

        # reboot device for settings to update
        self.log.info('Reboot for VoLTE settings to update.')
        self.dut.reboot()

        # Set voice call volume to minimum
        set_phone_silent_mode(self.log, self.dut)

        # initialize ims simulator connector wrapper
        self.unpack_userparams(api_connector_port=self.IMS_API_CONNECTOR_DEFAULT_PORT,
                               api_token=self.IMS_CLIENT_DEFAULT_API_TOKEN,
                               ims_client_ip=self.IMS_CLIENT_DEFAULT_IP,
                               ims_client_port=self.IMS_CLIENT_DEFAULT_PORT,
                               ue_number=self.UE_DEFAULT_NUMBER)
        self.ims_client = ImsApiConnector(
            self.uxm_ip,
            self.api_connector_port,
            ImsAppName.CLIENT
        )

        self.ims_server = ImsApiConnector(
            self.uxm_ip,
            self.api_connector_port,
            ImsAppName.SERVER
        )

    def setup_test(self):
        # Enable NR if it is VoNR test case
        self.log.info(f'test name: {self.test_name}')
        self.ims_server.restart_server()
        if 'NR' in self.test_name:
            self.log.info('Enable VoNR for UE.')
            self.at_util.enable_ims_nr()
        super().setup_test()

    def power_ims_call_test(self):
        """ Measures power during a VoLTE call.

        Measurement step in this test. Starts the voice call and
        initiates power measurement. Pass or fail is decided with a
        threshold value. """
        # create dedicated bearer
        self.log.info('create dedicated bearer.')
        if 'LTE' in self.test_name:
            self.cellular_simulator.create_dedicated_bearer()

        time.sleep(5)

        # Initiate the voice call
        self.log.info('Callbox initiates call to UE.')
        self.ims_client.initiate_call(self.ue_number)

        time.sleep(5)

        # pick up call
        self.log.info('UE pick up call.')
        self.dut.adb.shell('input keyevent KEYCODE_CALL')

        # Set mac padding
        if 'NR' in self.test_name:
            self.cellular_simulator.modify_dl_ul_mac_padding()

        # Mute the call
        self.dut.droid.telecomCallMute()

        # Turn of screen
        self.dut.droid.goToSleepNow()

        # Measure power and check against threshold
        self.collect_power_data_and_validate()

    def teardown_test(self):
        super().teardown_test()
        # End the call
        self.log.info('Hangup.')
        self.ims_client.hangup_call()

    def teardown_class(self):
        super().teardown_class()
        self.ims_client.tear_down()
        self.ims_server.tear_down()
        self.log.info('Disable IMS.')
        self.dut.adb.shell(self.ADB_CMD_DISABLE_IMS)


class PowerTelIms_Preset_Test(PowerTelImsPresetTest):
    def test_preset_LTE_voice(self):
        self.power_ims_call_test()

    def test_preset_NR_voice(self):
        self.power_ims_call_test()