#===============================================================================
# Copyright 2021 Intel Corporation
#
# 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 re
import subprocess
import sys
import os
import pathlib
import pytest
from _models_info import TO_SKIP


def get_branch(s):
    if len(s) == 0:
        return 'NO INFO'
    for i in s:
        if 'failed to run accelerated version, fallback to original Scikit-learn' in i:
            return 'was in OPT, but go in Scikit'
    for i in s:
        if 'running accelerated version' in i:
            return 'OPT'
    return 'Scikit'


def run_parse(mas, result):
    name, dtype = mas[0].split()
    temp = []
    INFO_POS = 16
    for i in range(1, len(mas)):
        mas[i] = mas[i][INFO_POS:]  # remove 'SKLEARNEX INFO: '
        if not mas[i].startswith('sklearn'):
            ind = name + ' ' + dtype + ' ' + mas[i]
            result[ind] = get_branch(temp)
            temp.clear()
        else:
            temp.append(mas[i])


def get_result_log():
    os.environ['SKLEARNEX_VERBOSE'] = 'INFO'
    absolute_path = str(pathlib.Path(__file__).parent.absolute())
    try:
        process = subprocess.check_output(
            [
                sys.executable,
                absolute_path + '/utils/_launch_algorithms.py'
            ]
        )
    except subprocess.CalledProcessError as e:
        print(e)
        exit(1)
    mas = []
    result = {}
    for i in process.decode().split('\n'):
        if i.startswith('SKLEARNEX WARNING'):
            continue
        if not i.startswith('SKLEARNEX INFO') and len(mas) != 0:
            run_parse(mas, result)
            mas.clear()
            mas.append(i.strip())
        else:
            mas.append(i.strip())
    del os.environ['SKLEARNEX_VERBOSE']
    return result


result_log = get_result_log()


@pytest.mark.parametrize('configuration', result_log)
def test_patching(configuration):
    if 'OPT' in result_log[configuration]:
        return
    for skip in TO_SKIP:
        if re.search(skip, configuration) is not None:
            pytest.skip("SKIPPED", allow_module_level=False)
    raise ValueError('Test patching failed: ' + configuration)