# python3
# Copyright 2021 The Chromium Authors
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

import re
import unittest

from lib import compiler


class CompilerTestCase(unittest.TestCase):
    def assertListSortedEqual(self, a, b, msg=None):
        a.sort()
        b.sort()
        if msg:
            self.assertListEqual(a, b, msg=msg)
        else:
            self.assertListEqual(a, b)

    def matching_archs(self, matching: str) -> set[str]:
        return {
            arch
            for arch in compiler._RUSTC_ARCH_TO_BUILD_CONDITION
            if re.search(matching, arch)
        }

    def not_matching_archs(self, matching: str) -> set[str]:
        """The inverse of matching_archs()."""
        return {
            arch
            for arch in compiler._RUSTC_ARCH_TO_BUILD_CONDITION
            if not re.search(matching, arch)
        }


class TestGnConditions(CompilerTestCase):
    def test_all_platforms(self):
        s = compiler.BuildConditionSet(compiler.ArchSet.ALL())
        self.assertListSortedEqual([], s.get_gn_conditions())

    def test_one_platform(self):
        for a in compiler.ArchSet.ALL().as_strings():
            s = compiler.BuildConditionSet(compiler.ArchSet(initial={a}))
            mode: compiler.BuildCondition = \
                compiler._RUSTC_ARCH_TO_BUILD_CONDITION[a]
            self.assertListSortedEqual([mode.gn_condition()],
                                       s.get_gn_conditions())

    def test_os(self):
        # One OS.
        for (matching, mode) in [
            (compiler._RUSTC_ARCH_MATCH_ANDROID,
             compiler.BuildCondition.ALL_ANDROID),
            (compiler._RUSTC_ARCH_MATCH_FUCHSIA,
             compiler.BuildCondition.ALL_FUCHSIA),
            (compiler._RUSTC_ARCH_MATCH_IOS, compiler.BuildCondition.ALL_IOS),
            (compiler._RUSTC_ARCH_MATCH_WINDOWS,
             compiler.BuildCondition.ALL_WINDOWS),
            (compiler._RUSTC_ARCH_MATCH_LINUX,
             compiler.BuildCondition.ALL_LINUX),
            (compiler._RUSTC_ARCH_MATCH_MAC, compiler.BuildCondition.ALL_MAC),
        ]:
            archs = self.matching_archs(matching)
            s = compiler.BuildConditionSet(compiler.ArchSet(initial=archs))
            cond = mode.gn_condition()
            self.assertListSortedEqual([cond],
                                       s.get_gn_conditions(),
                                       msg=repr(archs))

        # Two OSs.
        archs = self.matching_archs(compiler._RUSTC_ARCH_MATCH_WINDOWS + r"|" +
                                    compiler._RUSTC_ARCH_MATCH_MAC)
        s = compiler.BuildConditionSet(compiler.ArchSet(initial=archs))
        cond1 = compiler.BuildCondition.ALL_WINDOWS.gn_condition()
        cond2 = compiler.BuildCondition.ALL_MAC.gn_condition()
        self.assertListSortedEqual([cond1, cond2], s.get_gn_conditions())

        # All but one OS.
        for (matching_os, mode) in [
            (compiler._RUSTC_ARCH_MATCH_ANDROID,
             compiler.BuildCondition.NOT_ANDROID),
            (compiler._RUSTC_ARCH_MATCH_FUCHSIA,
             compiler.BuildCondition.NOT_FUCHSIA),
            (compiler._RUSTC_ARCH_MATCH_IOS, compiler.BuildCondition.NOT_IOS),
            (compiler._RUSTC_ARCH_MATCH_WINDOWS,
             compiler.BuildCondition.NOT_WINDOWS),
            (compiler._RUSTC_ARCH_MATCH_LINUX,
             compiler.BuildCondition.NOT_LINUX),
            (compiler._RUSTC_ARCH_MATCH_MAC, compiler.BuildCondition.NOT_MAC),
        ]:
            s = compiler.BuildConditionSet(
                compiler.ArchSet(initial=self.not_matching_archs(matching_os)))
            cond = mode.gn_condition()
            self.assertListSortedEqual([cond], s.get_gn_conditions())

    def test_one_cpu(self):
        for (matching, mode) in [
            (compiler._RUSTC_ARCH_MATCH_X86, compiler.BuildCondition.ALL_X86),
            (compiler._RUSTC_ARCH_MATCH_X64, compiler.BuildCondition.ALL_X64),
            (compiler._RUSTC_ARCH_MATCH_ARM32,
             compiler.BuildCondition.ALL_ARM32),
            (compiler._RUSTC_ARCH_MATCH_ARM64,
             compiler.BuildCondition.ALL_ARM64)
        ]:
            s = compiler.BuildConditionSet(
                compiler.ArchSet(initial=self.matching_archs(matching)))
            cond = mode.gn_condition()
            self.assertListSortedEqual([cond], s.get_gn_conditions())

    def test_combining_os_and_cpu(self):
        # One Cpu and one OS (with overlap).
        archs = self.matching_archs(compiler._RUSTC_ARCH_MATCH_LINUX + r"|" +
                                    compiler._RUSTC_ARCH_MATCH_X86)
        s = compiler.BuildConditionSet(compiler.ArchSet(initial=archs))
        cond1 = compiler.BuildCondition.ALL_LINUX.gn_condition()
        cond2 = compiler.BuildCondition.ALL_X86.gn_condition()
        self.assertListSortedEqual([cond1, cond2], s.get_gn_conditions())

        # One Cpu and one OS (without overlap).
        archs = self.matching_archs(compiler._RUSTC_ARCH_MATCH_MAC + r"|" +
                                    compiler._RUSTC_ARCH_MATCH_X86)
        s = compiler.BuildConditionSet(compiler.ArchSet(initial=archs))
        cond1 = compiler.BuildCondition.ALL_MAC.gn_condition()
        cond2 = compiler.BuildCondition.ALL_X86.gn_condition()
        self.assertListSortedEqual([cond1, cond2], s.get_gn_conditions())

    def test_invert(self):
        all = compiler.BuildConditionSet.ALL()
        none = compiler.BuildConditionSet.EMPTY()
        self.assertEqual(none, all.inverted())
        self.assertEqual(all, none.inverted())

        one = compiler.BuildConditionSet(
            compiler.ArchSet(
                initial=self.matching_archs(compiler._RUSTC_ARCH_MATCH_MAC)))
        the_rest = compiler.BuildConditionSet(
            compiler.ArchSet(initial=self.not_matching_archs(
                compiler._RUSTC_ARCH_MATCH_MAC)))
        self.assertListSortedEqual(one.inverted().get_gn_conditions(),
                                   the_rest.get_gn_conditions())


class TestCompiler(unittest.TestCase):
    def test_all_and_one(self):
        self.assertEqual(len(compiler.ArchSet.ALL().as_strings()),
                         len(compiler._RUSTC_ARCH_TO_BUILD_CONDITION))
        self.assertEqual(len(compiler.ArchSet.ONE()), 1)


class TestArchSet(CompilerTestCase):
    def test_construct(self):
        a = compiler.ArchSet(
            initial=self.matching_archs(compiler._RUSTC_ARCH_MATCH_ARM32))
        self.assertSetEqual({
            "armv7-linux-androideabi",
            "armv7-apple-ios",
        }, a.as_strings())

        a = compiler.ArchSet.EMPTY()
        self.assertSetEqual(set(), a.as_strings())

        a = compiler.ArchSet(
            initial=self.matching_archs(compiler._RUSTC_ARCH_MATCH_ARM32))
        self.assertSetEqual({
            "armv7-linux-androideabi",
            "armv7-apple-ios",
        }, a.as_strings())

        a = compiler.ArchSet(
            initial=self.matching_archs(compiler._RUSTC_ARCH_MATCH_ARM32))
        self.assertTrue(a.has_arch("armv7-linux-androideabi"))
        self.assertFalse(a.has_arch("i686-pc-windows-msvc"))

    def test_bool(self):
        a = compiler.ArchSet.EMPTY()
        self.assertFalse(bool(a))

        a = compiler.ArchSet.ONE()
        self.assertTrue(bool(a))

        a = compiler.ArchSet.ALL()
        self.assertTrue(bool(a))

    def test_eq(self):
        self.assertEqual(compiler.ArchSet.EMPTY(), compiler.ArchSet.EMPTY())
        self.assertEqual(compiler.ArchSet.ONE(), compiler.ArchSet.ONE())
        self.assertEqual(compiler.ArchSet.ALL(), compiler.ArchSet.ALL())

    def test_len(self):
        self.assertEqual(len(compiler.ArchSet.EMPTY()), 0)
        self.assertEqual(len(compiler.ArchSet.ONE()), 1)
        self.assertEqual(len(compiler.ArchSet.ALL()),
                         len(compiler._RUSTC_ARCH_TO_BUILD_CONDITION))

    def test_add_archset(self):
        a = compiler.ArchSet.EMPTY()
        a.add_archset(compiler.ArchSet.ALL())
        self.assertEqual(a, compiler.ArchSet.ALL())

        a = compiler.ArchSet.ONE()
        a.add_archset(compiler.ArchSet.ALL())
        self.assertEqual(a, compiler.ArchSet.ALL())

        a = compiler.ArchSet.ALL()
        a.add_archset(compiler.ArchSet.ALL())
        self.assertEqual(a, compiler.ArchSet.ALL())

    def test_and(self):
        a = compiler.ArchSet.EMPTY()
        a = a & compiler.ArchSet.ALL()
        self.assertSetEqual(set(), a.as_strings())

        a = compiler.ArchSet.EMPTY()
        a &= compiler.ArchSet.ALL()
        self.assertSetEqual(set(), a.as_strings())

        a = compiler.ArchSet.EMPTY()
        a = a & compiler.ArchSet.EMPTY()
        self.assertSetEqual(set(), a.as_strings())

        a = compiler.ArchSet.EMPTY()
        a &= compiler.ArchSet.EMPTY()
        self.assertSetEqual(set(), a.as_strings())

        a = compiler.ArchSet.ALL()
        a = a & compiler.ArchSet.ALL()
        self.assertSetEqual(compiler.ArchSet.ALL().as_strings(), a.as_strings())

        a = compiler.ArchSet.ALL()
        a &= compiler.ArchSet.ALL()
        self.assertSetEqual(compiler.ArchSet.ALL().as_strings(), a.as_strings())

        a = compiler.ArchSet.ALL()
        a = a & compiler.ArchSet.ONE()
        self.assertSetEqual(compiler.ArchSet.ONE().as_strings(), a.as_strings())

        a = compiler.ArchSet.ALL()
        a &= compiler.ArchSet.ONE()
        self.assertSetEqual(compiler.ArchSet.ONE().as_strings(), a.as_strings())

        a = compiler.ArchSet.ALL()
        a = a & compiler.ArchSet.EMPTY()
        self.assertSetEqual(set(), a.as_strings())

        a = compiler.ArchSet.ALL()
        a &= compiler.ArchSet.EMPTY()
        self.assertSetEqual(set(), a.as_strings())