#!/usr/bin/env python3
#-*- coding: utf-8 -*-

# Copyright (c) 2024 Huawei Device Co., Ltd.
# 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.

from hypium.action.host import host


def push_source(source_path, driver, sn, update_param=False):
    """
    @func: push resources to devices
    @param source_path: the path of resources required by a case
    @param driver: the device
    @param sn: device sn
    """
    host.shell("hdc -t {} shell kill -9 `pidof listen_test`".format(sn))
    host.shell("hdc -t {} target mount".format(sn))
    host.shell("hdc -t {} shell rm -r /data/log/hilog".format(sn))
    host.shell("hdc -t {} shell hilog -d /system/bin/samgr".format(sn))
    if "sa_listen_cfg_path" in source_path:
        driver.Storage.push_file(local_path=source_path["sa_listen_cfg_path"], device_path="/system/etc/init")
        host.shell("hdc -t {} shell chmod 644 /system/etc/init/listen_test.cfg".format(sn))
        driver.Storage.push_file(local_path=source_path["sa_listen_json_path"], device_path="/system/profile/")
        host.shell("hdc -t {} shell chmod 644 /system/profile/listen_test.json".format(sn))

    if "sa_lib_listen_test_path" in source_path:
        driver.Storage.push_file(local_path=source_path["sa_lib_listen_test_path"], device_path="/system/lib/")
        host.shell("hdc -t {} shell chmod 644 /system/lib/liblisten_test.z.so".format(sn))

    if "sa_proxy_path" in source_path:
        driver.Storage.push_file(local_path=source_path["sa_proxy_path"], device_path="/system/lib/")
        host.shell("hdc -t {} shell chmod 644 /system/lib/libtest_sa_proxy.z.so".format(sn))

    if "sa_lib_fwk_path" in source_path:
        driver.Storage.push_file(local_path=source_path["sa_lib_fwk_path"], device_path="/system/lib/")
        host.shell("hdc -t {} shell chmod 644 /system/lib/libsystem_ability_fwk.z.so".format(sn))

    if "sa_lib_audio_ability" in source_path:
        driver.Storage.push_file(local_path=source_path["sa_lib_audio_ability"], device_path="/system/lib/")
        host.shell("hdc -t {} shell chmod 644 /system/lib/libtest_audio_ability.z.so".format(sn))

    if "sa_ondemand_path" in source_path:
        driver.Storage.push_file(local_path=source_path["sa_ondemand_path"], device_path="/system/bin/")
        host.shell("hdc -t {} shell chmod 755 /system/bin/ondemand".format(sn))

    if "sa_para_path" in source_path:
        driver.Storage.push_file(local_path=source_path["sa_para_path"], device_path="/system/etc/param/")
        host.shell("hdc -t {} shell chmod 755 /system/etc/param/samgr.para".format(sn))
        driver.Storage.push_file(local_path=source_path["sa_para_dac_path"], device_path="/system/etc/param/")
        host.shell("hdc -t {} shell chmod 755 /system/etc/param/samgr.para.dac".format(sn))
    if update_param:
        driver.System.execute_command("ondemand param true")
    driver.System.reboot()


def remove_source(source_path, driver, sn):
    """
    @func: push resources from devices
    @param source_path: the path of resources required by a case
    @param driver: the device
    @param sn: device sn
    """
    host.shell("hdc -t {} shell kill -9 `pidof listen_test`".format(sn))
    host.shell("hdc -t {} target mount".format(sn))
    if "sa_listen_cfg_path" in source_path:
        driver.Storage.remove_file("/system/etc/init/listen_test.cfg")
        driver.Storage.remove_file("/system/etc/init/listen_test.json")

    if "sa_lib_listen_test_path" in source_path:
        driver.Storage.remove_file("/system/lib/liblisten_test.z.so")

    if "sa_proxy_path" in source_path:
        driver.Storage.remove_file("/system/lib/libtest_sa_proxy.z.so")

    if "sa_lib_fwk_path" in source_path:
        driver.Storage.remove_file("/system/lib/libsystem_ability_fwk.z.so")

    if "sa_lib_audio_ability" in source_path:
        driver.Storage.remove_file("/system/lib/libtest_audio_ability.z.so")

    if "sa_ondemand_path" in source_path:
        driver.Storage.remove_file("/system/bin/ondemand")

    if "sa_para_path" in source_path:
        driver.Storage.remove_file("/system/etc/param/samgr.para")
        driver.Storage.remove_file("/system/etc/param/samgr.para.dac")
        driver.Storage.push_file(local_path=source_path["sa_para_origin"], device_path="/system/etc/param/")
        driver.Storage.push_file(local_path=source_path["sa_para_dac_origin"], device_path="/system/etc/param/")