#!/usr/bin/env python3
# coding=utf-8

#
# Copyright (c) 2023 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.
#


import os
import json
import sys
import stat

FLAGS = os.O_WRONLY | os.O_CREAT | os.O_EXCL
MODES = stat.S_IWUSR | stat.S_IRUSR


def _init_sys_config():
    sys.localcoverage_path = os.path.join(current_path, "..")
    sys.path.insert(0, sys.localcoverage_path)


def gen_parts_info_json(folder_list, output_json_path, target_cpu):
    """
    根据部件信息,生成字典至json文件中
    """
    if len(folder_list) != 0:
        data_dict = {}
        for folder_str in folder_list:
            data_dict[folder_str] = f"innerkits/ohos-{target_cpu}/{folder_str}"
        output_json_path = os.path.join(output_json_path, "kits_modules_info.json")
        json_str = json.dumps(data_dict, indent=2)
        if os.path.exists(output_json_path):
            os.remove(output_json_path)
        with os.fdopen(os.open(output_json_path, FLAGS, MODES), 'w') as json_file:
            json_file.write(json_str)
    else:
        print("part_name list information is null")


def get_parts_list(path):
    """
    #获取out/ohos-arm-release/innerkits/ohos-arm内部接口文件夹名称列表
    """
    if os.path.exists(path):
        folder_list = os.listdir(path)
    else:
        print("The folder: %s does not exist" % path)
        folder_list = []
    return folder_list


if __name__ == "__main__":
    current_path = os.getcwd()
    _init_sys_config()
    from localCoverage.utils import get_product_name, coverage_command, get_target_cpu

    root_path = current_path.split("/test/testfwk/developer_test")[0]
    product_name = get_product_name(root_path)
    cpu_type = get_target_cpu(root_path)
    part_info_path = os.path.join(
        root_path, "out", product_name, "innerkits/ohos-%s" % cpu_type
    )
    json_path = os.path.join(
        root_path, "out", product_name, "packages/phone/innerkits/ohos-%s" % cpu_type
    )
    coverage_command("mkdir -p %s" % json_path)
    part_list = get_parts_list(part_info_path)
    gen_parts_info_json(part_list, json_path, cpu_type)