1ba991379Sopenharmony_ci#!/usr/bin/env python
2ba991379Sopenharmony_ci#coding=utf-8
3ba991379Sopenharmony_ci
4ba991379Sopenharmony_ci#
5ba991379Sopenharmony_ci# Copyright (c) 2023-2024 Huawei Device Co., Ltd.
6ba991379Sopenharmony_ci# Licensed under the Apache License, Version 2.0 (the "License");
7ba991379Sopenharmony_ci# you may not use this file except in compliance with the License.
8ba991379Sopenharmony_ci# You may obtain a copy of the License at
9ba991379Sopenharmony_ci#
10ba991379Sopenharmony_ci#     http://www.apache.org/licenses/LICENSE-2.0
11ba991379Sopenharmony_ci#
12ba991379Sopenharmony_ci# Unless required by applicable law or agreed to in writing, software
13ba991379Sopenharmony_ci# distributed under the License is distributed on an "AS IS" BASIS,
14ba991379Sopenharmony_ci# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
15ba991379Sopenharmony_ci# See the License for the specific language governing permissions and
16ba991379Sopenharmony_ci# limitations under the License.
17ba991379Sopenharmony_ci#
18ba991379Sopenharmony_ci
19ba991379Sopenharmony_ciimport os
20ba991379Sopenharmony_ciimport json
21ba991379Sopenharmony_ci
22ba991379Sopenharmony_ci
23ba991379Sopenharmony_ciclass BaseRule(object):
24ba991379Sopenharmony_ci    RULE_NAME = ""
25ba991379Sopenharmony_ci
26ba991379Sopenharmony_ci    def __init__(self, mgr, args):
27ba991379Sopenharmony_ci        self._args = args
28ba991379Sopenharmony_ci        self._mgr = mgr
29ba991379Sopenharmony_ci        self._white_lists = self.__load_files__("whitelist.json")
30ba991379Sopenharmony_ci
31ba991379Sopenharmony_ci    # To be override
32ba991379Sopenharmony_ci    def __check__(self):
33ba991379Sopenharmony_ci        # Default pass
34ba991379Sopenharmony_ci        return True
35ba991379Sopenharmony_ci
36ba991379Sopenharmony_ci    def __load_files__(self, name):
37ba991379Sopenharmony_ci        rules_dir = []
38ba991379Sopenharmony_ci        rules_dir.append(os.path.join(os.path.dirname(os.path.realpath(__file__)), "../rules"))
39ba991379Sopenharmony_ci        if self._args and self._args.rules:
40ba991379Sopenharmony_ci            rules_dir = rules_dir + self._args.rules
41ba991379Sopenharmony_ci
42ba991379Sopenharmony_ci        res = []
43ba991379Sopenharmony_ci        for d in rules_dir:
44ba991379Sopenharmony_ci            rules_file = os.path.join(d, self.__class__.RULE_NAME, name)
45ba991379Sopenharmony_ci            try:
46ba991379Sopenharmony_ci                with open(rules_file, "r") as f:
47ba991379Sopenharmony_ci                    jsonstr = "".join([line.strip() for line in f if not line.strip().startswith("//")])
48ba991379Sopenharmony_ci                    res = res + json.loads(jsonstr)
49ba991379Sopenharmony_ci            except:
50ba991379Sopenharmony_ci                pass
51ba991379Sopenharmony_ci
52ba991379Sopenharmony_ci        return res
53ba991379Sopenharmony_ci
54ba991379Sopenharmony_ci    def get_mgr(self):
55ba991379Sopenharmony_ci        return self._mgr
56ba991379Sopenharmony_ci
57ba991379Sopenharmony_ci    def get_white_lists(self):
58ba991379Sopenharmony_ci        return self._white_lists
59ba991379Sopenharmony_ci
60ba991379Sopenharmony_ci    def log(self, info):
61ba991379Sopenharmony_ci        print(info)
62ba991379Sopenharmony_ci
63ba991379Sopenharmony_ci    def warn(self, info):
64ba991379Sopenharmony_ci        print("\033[35m[WARNING]\x1b[0m: %s" % info)
65ba991379Sopenharmony_ci
66ba991379Sopenharmony_ci    def error(self, info):
67ba991379Sopenharmony_ci        print("\033[91m[NOT ALLOWED]\x1b[0m: %s" % info)
68ba991379Sopenharmony_ci
69ba991379Sopenharmony_ci    def get_help_url(self):
70ba991379Sopenharmony_ci        return "https://gitee.com/openharmony/developtools_integration_verification/tree/master/tools/startup_guard/rules/%s/README.md" % self.__class__.RULE_NAME
71