# Copyright 2023 The Pigweed Authors
#
# 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
#
#     https://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.
"""Full test of bazel module."""

from __future__ import annotations

import dataclasses
import re
from collections.abc import Iterator

from recipe_engine import (
    config_types,
    post_process,
    recipe_api,
    recipe_test_api,
)

from PB.recipe_modules.pigweed.bazel.tests.full import InputProperties

DEPS = [
    'pigweed/bazel',
    'recipe_engine/buildbucket',
    'recipe_engine/path',
    'recipe_engine/platform',
    'recipe_engine/properties',
]

INLINE_PROPERTIES_PROTO = """
import "recipe_modules/pigweed/bazel/options.proto";

message InputProperties {
  // Bazel module options.
  recipe_modules.pigweed.bazel.Options bazel_options = 1;

  // List of repository names that will be overridden when calling bazel.
  repeated string overrides = 2;

  // Add --keep_going argument to Bazel.
  bool continue_after_build_error = 3;

  // Download full artifacts for the Bazel build instead of just those necessary
  // for future steps.
  bool download_all_artifacts = 4;
}
"""

PROPERTIES = InputProperties


@dataclasses.dataclass
class _FakeCheckoutContext:
    root: config_types.Path
    bazel_overrides: dict[str, config_types.Path] = dataclasses.field(
        default_factory=dict
    )


def RunSteps(api: recipe_api.RecipeScriptApi, props: InputProperties):
    overrides = {}
    for override in props.overrides:
        overrides[override] = api.path.start_dir / 'bazel_repos' / override

    checkout = _FakeCheckoutContext(
        root=api.path.start_dir,
        bazel_overrides=overrides,
    )

    # pylint: disable=missing-function-docstring
    runner = api.bazel.new_runner(
        checkout=checkout,
        options=props.bazel_options,
        continue_after_build_error=props.continue_after_build_error,
        download_all_artifacts=props.download_all_artifacts,
    )
    runner.run()

    # Coverage only.
    runner.ensure_bazelisk()
    api.bazel.ensure_bazelisk()
    api.bazel.override_arguments(checkout)


def GenTests(api) -> Iterator[recipe_test_api.TestData]:
    def contains_override(step: str, name: str, bzlmod=False, invert=False):
        flag = '--module_override' if bzlmod else '--override_repository'
        pattern = re.compile(rf'{flag}={name}=.*')
        check = post_process.StepCommandContains
        if invert:
            check = post_process.StepCommandDoesNotContain
        return api.post_check(check, f'{step}.bazel', pattern)

    def lacks_override(step: str, name: str, bzlmod=False):
        return contains_override(
            step=step, name=name, bzlmod=bzlmod, invert=True
        )

    yield api.test(
        'fixed-bazelisk',
        api.bazel.properties(bazelisk_version='1.19.0'),
        api.post_check(post_process.MustRun, 'ensure bazelisk'),
        api.post_check(
            post_process.MustRun,
            'default.bazelisk test //....upload foo.so.upload foo.so to bucket',
        ),
        api.post_process(post_process.DropExpectation),
    )

    yield api.test(
        'latest-without-overrides',
        api.bazel.properties(program=['default']),
        api.post_check(post_process.MustRun, 'default.ensure bazelisk'),
        lacks_override('default.bazelisk test //...', 'pigweed'),
        lacks_override('default.bazelisk test //...', 'thirdparty'),
        api.post_process(post_process.DropExpectation),
    )

    yield api.test(
        'latest-with-overrides',
        api.bazel.properties(program=['default']),
        api.properties(overrides=['pigweed', 'thirdparty']),
        api.post_check(post_process.MustRun, 'default.ensure bazelisk'),
        contains_override('default.bazelisk test //...', 'pigweed'),
        contains_override('default.bazelisk test //...', 'thirdparty'),
        lacks_override('default.bazelisk test //...', 'additional'),
        api.post_process(post_process.DropExpectation),
    )

    yield api.test(
        'continue-download',
        api.buildbucket.try_build(experiments=['pigweed.disable_rbe']),
        api.bazel.properties(),
        api.properties(
            continue_after_build_error=True,
            download_all_artifacts=True,
        ),
        api.post_check(
            post_process.StepCommandContains,
            'default.bazelisk build //....bazel',
            ['--keep_going'],
        ),
        api.post_check(
            post_process.StepCommandContains,
            'default.bazelisk test //....bazel',
            ['--keep_going'],
        ),
        api.post_process(post_process.DropExpectation),
    )

    yield api.test(
        'remote',
        api.platform.name('linux'),
        api.buildbucket.ci_build(),
        api.bazel.properties(),
        api.bazel.config(remote=True),
        api.post_check(post_process.MustRun, 'default.ensure bazelisk'),
        api.post_check(
            post_process.StepCommandContains,
            'default.bazelisk build //....bazel',
            ['--config=remote'],
        ),
        api.post_process(post_process.DropExpectation),
    )

    yield api.test(
        'remote-ignored-on-mac',
        api.platform.name('mac'),
        api.buildbucket.ci_build(),
        api.bazel.properties(),
        api.bazel.config(remote=True),
        api.post_check(post_process.MustRun, 'default.ensure bazelisk'),
        api.post_check(
            post_process.MustRun,
            'ignoring remote because not running on Linux',
        ),
        api.post_process(post_process.DropExpectation),
    )

    yield api.test(
        'instance-name-open-ci',
        api.buildbucket.ci_build(project='pigweed'),
        api.bazel.properties(),
        api.bazel.config(remote_cache=True),
        api.post_check(
            post_process.StepCommandContains,
            'default.bazelisk build //....bazel',
            [
                '--bes_instance_name=pigweed-rbe-open',
                '--remote_instance_name=projects/pigweed-rbe-open/instances/default-instance',
            ],
        ),
        api.post_process(post_process.DropExpectation),
    )

    yield api.test(
        'instance-name-private-cq',
        api.buildbucket.try_build(),
        api.bazel.properties(),
        api.bazel.config(remote_cache=True),
        api.post_check(
            post_process.StepCommandContains,
            'default.bazelisk build //....bazel',
            [
                '--bes_instance_name=pigweed-rbe-private-pre',
                '--remote_instance_name=projects/pigweed-rbe-private-pre/instances/default-instance',
            ],
        ),
        api.post_process(post_process.DropExpectation),
    )

    yield api.test(
        'upload-but-no-cache',
        api.buildbucket.ci_build(),
        api.bazel.properties(),
        api.bazel.config(remote_cache=False, upload_local_results=True),
        api.post_check(post_process.MustRun, 'default.ensure bazelisk'),
        api.post_check(
            post_process.MustRun,
            'ignoring upload_local_results since remote_cache is False',
        ),
        api.post_process(post_process.DropExpectation),
    )

    yield api.test(
        'upload',
        api.buildbucket.ci_build(),
        api.bazel.properties(),
        api.bazel.config(remote_cache=True, upload_local_results=True),
        api.post_check(post_process.MustRun, 'default.ensure bazelisk'),
        api.post_check(
            post_process.DoesNotRun,
            'ignoring upload_local_results since remote_cache is False',
        ),
        api.post_process(post_process.DropExpectation),
    )