# Copyright 2017 the V8 project authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

from testrunner.local import testsuite
from testrunner.objects import testcase

# We use the 'wasm-3.0' branch on the main spec repo, so enable all proposals
# that were merged into that branch.
default_flags = ['--experimental-wasm-exnref']

proposal_flags = [
    {
        'name': 'js-types',
        'flags': ['--experimental-wasm-type-reflection']
    },
    {
        'name': 'tail-call',
        'flags': []
    },
    {
        'name':
            'memory64',
        'flags': [
            # The memory64 repository is rebased on the upstream 'wasm-3.0'
            # branch, which contains exnref.
            '--experimental-wasm-exnref'
        ]
    },
    {
        'name': 'extended-const',
        'flags': []
    },
    {
        'name': 'function-references',
        'flags': []
    },
    {
        'name': 'gc',
        'flags': []
    },
    {
        'name': 'multi-memory',
        'flags': ['--experimental-wasm-multi-memory']
    },
    {
        'name': 'exception-handling',
        # This flag enables the *new* exception handling proposal. The legacy
        # proposal is enabled by default.
        'flags': ['--experimental-wasm-exnref']
    },
    {
        'name':
            'js-promise-integration',
        'flags': [
            '--experimental-wasm-jspi',
            # The jspi repository is rebased on the upstream 'wasm-3.0'
            # branch, which contains exnref.
            '--experimental-wasm-exnref'
        ]
    }
]


class TestLoader(testsuite.JSTestLoader):
  pass

class TestSuite(testsuite.TestSuite):

  def __init__(self, ctx, *args, **kwargs):
    super(TestSuite, self).__init__(ctx, *args, **kwargs)
    self.test_root = self.root / "tests"
    self._test_loader.test_root = self.test_root

  def _test_loader_class(self):
    return TestLoader

  def _test_class(self):
    return TestCase

class TestCase(testcase.D8TestCase):
  def _get_files_params(self):
    return [self.suite.test_root / self.path_js]

  def _get_source_flags(self):
    for proposal in proposal_flags:
      if f"proposals/{proposal['name']}" in self.name:
        return proposal['flags']
    return default_flags