# Copyright (c) 2024 PaddlePaddle Authors. All Rights Reserved.
#
# 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 argparse
import json
import os
import re
import sys

import requests

SKIP_COVERAGE_CHECKING_LABELS = [
    "cinn",
    "typing",
    "codestyle",
    "phi_ops",
    r"3.1\d",  # SOT Python 3.13 support
]

SKIP_DISTRIBUTE_TEST_CHECKING_LABELS = [
    "cinn",
    "BUAA",
]

SKIP_COVERAGE_CHECKING_REGEX = re.compile(
    rf"[\[【]({'|'.join(SKIP_COVERAGE_CHECKING_LABELS)})[\]】]",
    re.IGNORECASE,
)

SKIP_DISTRIBUTE_TEST_CHECKING_REGEX = re.compile(
    rf"[\[【]({'|'.join(SKIP_DISTRIBUTE_TEST_CHECKING_LABELS)})[\]】]",
    re.IGNORECASE,
)


def check_skip_coverage(title):
    if match_obj := SKIP_COVERAGE_CHECKING_REGEX.search(title):
        print(f'The title starts with {match_obj.group(0)}')
    else:
        print(
            f'The title does not start with {" or ".join(SKIP_COVERAGE_CHECKING_LABELS)}'
        )
        sys.exit(1)


def check_skip_distribute_test(title):
    if match_obj := SKIP_DISTRIBUTE_TEST_CHECKING_REGEX.search(title):
        print(f'The title starts with {match_obj.group(0)}')
    else:
        print(
            f'The title does not start with {" or ".join(SKIP_COVERAGE_CHECKING_LABELS)}'
        )
        sys.exit(1)


def main(args):
    pr_id = os.getenv('GIT_PR_ID')
    if not pr_id:
        print('PREC No PR ID')
        sys.exit(1)
    response = requests.get(
        f'https://api.github.com/repos/PaddlePaddle/Paddle/pulls/{pr_id}',
        headers={'Accept': 'application/vnd.github.v3+json'},
    )

    data = json.loads(response.text)

    title = data['title']
    if args.type == 'skip_coverage_check':
        check_skip_coverage(title)
    elif args.type == 'skip_distribute_test':
        print(check_skip_distribute_test(title))


if __name__ == "__main__":
    parser = argparse.ArgumentParser(description='Process some arguments.')
    parser.add_argument('type', help='The type of check to perform.')

    args = parser.parse_args()
    main(args)