15f9996aaSopenharmony_ci#!/usr/bin/env python3
25f9996aaSopenharmony_ci# -*- coding: utf-8 -*-
35f9996aaSopenharmony_ci# Copyright (c) 2023 Huawei Device Co., Ltd.
45f9996aaSopenharmony_ci# Licensed under the Apache License, Version 2.0 (the "License");
55f9996aaSopenharmony_ci# you may not use this file except in compliance with the License.
65f9996aaSopenharmony_ci# You may obtain a copy of the License at
75f9996aaSopenharmony_ci#
85f9996aaSopenharmony_ci#     http://www.apache.org/licenses/LICENSE-2.0
95f9996aaSopenharmony_ci#
105f9996aaSopenharmony_ci# Unless required by applicable law or agreed to in writing, software
115f9996aaSopenharmony_ci# distributed under the License is distributed on an "AS IS" BASIS,
125f9996aaSopenharmony_ci# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
135f9996aaSopenharmony_ci# See the License for the specific language governing permissions and
145f9996aaSopenharmony_ci# limitations under the License.
155f9996aaSopenharmony_ci
165f9996aaSopenharmony_ciimport sys
175f9996aaSopenharmony_ciimport os
185f9996aaSopenharmony_cifrom containers.status import throw_exception
195f9996aaSopenharmony_cifrom exceptions.ohos_exception import OHOSException
205f9996aaSopenharmony_cisys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
215f9996aaSopenharmony_cifrom scripts.util import file_utils  # noqa: E402
225f9996aaSopenharmony_ci
235f9996aaSopenharmony_ci
245f9996aaSopenharmony_ciclass BundlePartObj(object):
255f9996aaSopenharmony_ci    def __init__(self, bundle_config_file, exclusion_modules_config_file, load_test_config):
265f9996aaSopenharmony_ci        self._build_config_file = bundle_config_file
275f9996aaSopenharmony_ci        self._exclusion_modules_config_file = exclusion_modules_config_file
285f9996aaSopenharmony_ci        self._load_test_config = load_test_config
295f9996aaSopenharmony_ci        self._loading_config()
305f9996aaSopenharmony_ci
315f9996aaSopenharmony_ci    @throw_exception
325f9996aaSopenharmony_ci    def _loading_config(self):
335f9996aaSopenharmony_ci        if not os.path.exists(self._build_config_file):
345f9996aaSopenharmony_ci            raise OHOSException("file '{}' doesn't exist.".format(
355f9996aaSopenharmony_ci                self._build_config_file), "2011")
365f9996aaSopenharmony_ci        self.bundle_info = file_utils.read_json_file(self._build_config_file)
375f9996aaSopenharmony_ci        if self.bundle_info is None:
385f9996aaSopenharmony_ci            raise OHOSException("read file '{}' failed.".format(
395f9996aaSopenharmony_ci                self._build_config_file), "2011")
405f9996aaSopenharmony_ci        self._check_format()
415f9996aaSopenharmony_ci        self.exclusion_modules_info = file_utils.read_json_file(
425f9996aaSopenharmony_ci            self._exclusion_modules_config_file)
435f9996aaSopenharmony_ci
445f9996aaSopenharmony_ci    @throw_exception
455f9996aaSopenharmony_ci    def _check_format(self):
465f9996aaSopenharmony_ci        _tip_info = "bundle.json info is incorrect in '{}'".format(
475f9996aaSopenharmony_ci            self._build_config_file)
485f9996aaSopenharmony_ci        if 'component' not in self.bundle_info:
495f9996aaSopenharmony_ci            raise OHOSException(
505f9996aaSopenharmony_ci                "{}, 'component' is required.".format(_tip_info), "2011")
515f9996aaSopenharmony_ci        _component_info = self.bundle_info.get('component')
525f9996aaSopenharmony_ci        if 'name' not in _component_info:
535f9996aaSopenharmony_ci            raise OHOSException(
545f9996aaSopenharmony_ci                "{}, 'component.name' is required.".format(_tip_info), "2011")
555f9996aaSopenharmony_ci        if 'subsystem' not in _component_info:
565f9996aaSopenharmony_ci            raise OHOSException(
575f9996aaSopenharmony_ci                "{}, 'component.subsystem' is required.".format(_tip_info), "2011")
585f9996aaSopenharmony_ci        if 'build' not in _component_info:
595f9996aaSopenharmony_ci            raise OHOSException(
605f9996aaSopenharmony_ci                "{}, 'component.build' is required.".format(_tip_info), "2011")
615f9996aaSopenharmony_ci        _bundle_build = _component_info.get('build')
625f9996aaSopenharmony_ci        if 'sub_component' not in _bundle_build and 'group_type' not in _bundle_build \
635f9996aaSopenharmony_ci                and 'modules' not in _bundle_build:
645f9996aaSopenharmony_ci            raise OHOSException(
655f9996aaSopenharmony_ci                "{}, 'component.build.sub_component','component.build.group_type' or \
665f9996aaSopenharmony_ci                'component.build.modules' is required.".format(_tip_info), "2011")
675f9996aaSopenharmony_ci        if 'group_type' in _bundle_build:
685f9996aaSopenharmony_ci            group_list = ['base_group', 'fwk_group', 'service_group']
695f9996aaSopenharmony_ci            _module_groups = _bundle_build.get('group_type')
705f9996aaSopenharmony_ci            for _group_type, _module_list in _module_groups.items():
715f9996aaSopenharmony_ci                if _group_type not in group_list:
725f9996aaSopenharmony_ci                    raise OHOSException(
735f9996aaSopenharmony_ci                        "{}, incorrect group type".format(_tip_info), "2011")
745f9996aaSopenharmony_ci
755f9996aaSopenharmony_ci    def to_ohos_build(self):
765f9996aaSopenharmony_ci        _component_info = self.bundle_info.get('component')
775f9996aaSopenharmony_ci        _subsystem_name = _component_info.get('subsystem')
785f9996aaSopenharmony_ci        _part_name = _component_info.get('name')
795f9996aaSopenharmony_ci        _bundle_build = _component_info.get('build')
805f9996aaSopenharmony_ci        _exclusion_modules_info = self.exclusion_modules_info
815f9996aaSopenharmony_ci        _ohos_build_info = {}
825f9996aaSopenharmony_ci        _ohos_build_info['subsystem'] = _subsystem_name
835f9996aaSopenharmony_ci        _part_info = {}
845f9996aaSopenharmony_ci        module_list = []
855f9996aaSopenharmony_ci        if _component_info.get('build').__contains__('sub_component'):
865f9996aaSopenharmony_ci            _part_info['module_list'] = _component_info.get('build').get(
875f9996aaSopenharmony_ci                'sub_component')
885f9996aaSopenharmony_ci        elif _component_info.get('build').__contains__('modules'):
895f9996aaSopenharmony_ci            _part_info['module_list'] = _component_info.get(
905f9996aaSopenharmony_ci                'build').get('modules')
915f9996aaSopenharmony_ci        elif _component_info.get('build').__contains__('group_type'):
925f9996aaSopenharmony_ci            _module_groups = _component_info.get('build').get('group_type')
935f9996aaSopenharmony_ci            for _group_type, _module_list in _module_groups.items():
945f9996aaSopenharmony_ci                _key = '{}:{}'.format(_subsystem_name, _part_name)
955f9996aaSopenharmony_ci                if not _exclusion_modules_info.get(_key):
965f9996aaSopenharmony_ci                    module_list.extend(_module_list)
975f9996aaSopenharmony_ci                elif _group_type not in _exclusion_modules_info.get(_key):
985f9996aaSopenharmony_ci                    module_list.extend(_module_list)
995f9996aaSopenharmony_ci            _part_info['module_list'] = module_list
1005f9996aaSopenharmony_ci        if 'inner_kits' in _bundle_build:
1015f9996aaSopenharmony_ci            _part_info['inner_kits'] = _bundle_build.get('inner_kits')
1025f9996aaSopenharmony_ci        elif 'inner_api' in _bundle_build:
1035f9996aaSopenharmony_ci            _part_info['inner_kits'] = _bundle_build.get('inner_api')
1045f9996aaSopenharmony_ci        if 'test' in _bundle_build and self._load_test_config:
1055f9996aaSopenharmony_ci            _part_info['test_list'] = _bundle_build.get('test')
1065f9996aaSopenharmony_ci        if 'features' in _component_info:
1075f9996aaSopenharmony_ci            _part_info['feature_list'] = _component_info.get('features')
1085f9996aaSopenharmony_ci        if 'syscap' in _component_info:
1095f9996aaSopenharmony_ci            _part_info['system_capabilities'] = _component_info.get('syscap')
1105f9996aaSopenharmony_ci        if 'hisysevent_config' in _component_info:
1115f9996aaSopenharmony_ci            _part_info['hisysevent_config'] = _component_info.get(
1125f9996aaSopenharmony_ci                'hisysevent_config')
1135f9996aaSopenharmony_ci        _part_info['part_deps'] = _component_info.get('deps', {})
1145f9996aaSopenharmony_ci        _part_info['part_deps']['build_config_file'] = self._build_config_file
1155f9996aaSopenharmony_ci        _ohos_build_info['parts'] = {_part_name: _part_info}
1165f9996aaSopenharmony_ci        return _ohos_build_info
117