#!/usr/bin/env bash
#
# Copyright 2016 The Bazel 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.
#
# An end-to-end test for bazel-specific parts of the build-event stream.

set -euo pipefail
# --- begin runfiles.bash initialization ---
if [[ ! -d "${RUNFILES_DIR:-/dev/null}" && ! -f "${RUNFILES_MANIFEST_FILE:-/dev/null}" ]]; then
  if [[ -f "$0.runfiles_manifest" ]]; then
    export RUNFILES_MANIFEST_FILE="$0.runfiles_manifest"
  elif [[ -f "$0.runfiles/MANIFEST" ]]; then
    export RUNFILES_MANIFEST_FILE="$0.runfiles/MANIFEST"
  elif [[ -f "$0.runfiles/bazel_tools/tools/bash/runfiles/runfiles.bash" ]]; then
    export RUNFILES_DIR="$0.runfiles"
  fi
fi
if [[ -f "${RUNFILES_DIR:-/dev/null}/bazel_tools/tools/bash/runfiles/runfiles.bash" ]]; then
  source "${RUNFILES_DIR}/bazel_tools/tools/bash/runfiles/runfiles.bash"
elif [[ -f "${RUNFILES_MANIFEST_FILE:-/dev/null}" ]]; then
  source "$(grep -m1 "^bazel_tools/tools/bash/runfiles/runfiles.bash " \
            "$RUNFILES_MANIFEST_FILE" | cut -d ' ' -f 2-)"
else
  echo >&2 "ERROR: cannot find @bazel_tools//tools/bash/runfiles:runfiles.bash"
  exit 1
fi
# --- end runfiles.bash initialization ---

source "$(rlocation "io_bazel/src/test/shell/integration_test_setup.sh")" \
  || { echo "integration_test_setup.sh not found!" >&2; exit 1; }

#### SETUP #############################################################

function set_up() {
  mkdir -p pkg
  touch remote_file
  if is_windows; then
    # Windows needs "file:///c:/foo/bar".
    FILE_URL="file:///$(cygpath -m "$PWD")/remote_file"
  else
    # Non-Windows needs "file:///foo/bar".
    FILE_URL="file://${PWD}/remote_file"
  fi

  cat > MODULE.bazel <<EOF
http_file = use_repo_rule("@bazel_tools//tools/build_defs/repo:http.bzl", "http_file")
http_file(name="remote", urls=["${FILE_URL}"])
EOF
  cat > pkg/BUILD <<'EOF'
genrule(
  name="main",
  srcs=["@remote//file"],
  outs = ["main.out"],
  cmd = "cp $< $@",
)
EOF
}

#### TESTS #############################################################

function test_fetch_test() {
  # We expect the "fetch" command to generate at least a minimally useful
  # build-event stream.
  bazel clean --expunge
  rm -f "${TEST_log}"
  bazel fetch --build_event_text_file="${TEST_log}" //pkg:main \
      || fail "bazel fetch failed"
  [ -f "${TEST_log}" ] \
      || fail "fetch did not generate requested build-event file"
  expect_log '^started'
  expect_log '^finished'
  expect_log 'name: "SUCCESS"'
  expect_log 'uuid: "'
  expect_log '^fetch'
  # on second attempt, the fetched file should already be cached.
  bazel shutdown
  rm -f "${TEST_log}"
  bazel fetch --build_event_text_file="${TEST_log}" //pkg:main \
      || fail "bazel fetch failed"
  [ -f "${TEST_log}" ] \
      || fail "fetch did not generate requested build-event file"
  expect_log '^started'
  expect_log '^finished'
  expect_log 'name: "SUCCESS"'
  expect_log 'uuid: "'
  expect_not_log '^fetch'
}

function test_fetch_in_build() {
  # We expect a fetch that happens as a consequence of a build to be reported.
  bazel clean --expunge
  bazel build --build_event_text_file="${TEST_log}" //pkg:main \
      || fail "bazel build failed"
  expect_log 'name: "SUCCESS"'
  expect_log '^fetch'
  bazel shutdown
  bazel build --build_event_text_file="${TEST_log}" //pkg:main \
      || fail "bazel build failed"
  expect_log 'name: "SUCCESS"'
  expect_not_log '^fetch'
}

function test_query() {
  # Verify that at least a minimally meaningful event stream is generated
  # for non-build. In particular, we expect bazel not to crash.
  bazel query --build_event_text_file=$TEST_log  //pkg:main \
    || fail "bazel query failed"
  expect_log '^started'
  expect_log 'command: "query"'
  expect_log 'args: "--build_event_text_file='
  expect_log 'build_finished'
  expect_not_log 'aborted'
  expect_log '^finished'
  expect_log 'name: "SUCCESS"'
  expect_log 'last_message: true'
}

function test_fetch_failure() {
  # We expect that if a build is failing due to an error fetching an external
  # repository, we get a reasonable attribution of the root cause.
  cat > MODULE.bazel <<'EOF'
failing = use_repo_rule("//:failing_repo.bzl", "failing")

failing(name="remote")
EOF
  touch BUILD
  cat > failing_repo.bzl <<'EOF'
def _impl(ctx):
  fail("This is the error message")

failing = repository_rule(
  implementation = _impl,
  attrs = {},
)
EOF
  cat > pkg/BUILD <<'EOF'
genrule(
  name="main",
  srcs=["@remote//file"],
  outs = ["main.out"],
  cmd = "cp $< $@",
)
genrule(
  name="main2",
  srcs=["@remote//file2"],
  outs = ["main2.out"],
  cmd = "cp $< $@",
)
EOF
  bazel build -k --build_event_text_file=$TEST_log  //pkg:main //pkg:main2 \
    && fail "expected failure" || :

  # TODO: https://github.com/bazelbuild/bazel/issues/23240
  # Create a new event id type that doesn't use "//external"
  expect_log 'label: "//external:+failing+remote"'
  expect_log 'description:.*This is the error message'
  expect_not_log 'label.*@+failing+remote//file'
}

function test_residue_in_run_bep(){
  add_rules_shell "MODULE.bazel"
  mkdir -p a
  cat > a/BUILD <<'EOF'
load("@rules_shell//shell:sh_binary.bzl", "sh_binary")
sh_binary(
    name = 'arg',
    srcs = ['arg.sh'],
)
EOF

  cat > a/arg.sh <<'EOF'
#!/usr/bin/env bash

COUNTER=1
for i in "$@"; do
  echo "ARG $COUNTER": $i;
  COUNTER=$(($COUNTER+1))
done
EOF

  chmod +x a/arg.sh
  bazel run --experimental_run_bep_event_include_residue=true \
   --build_event_json_file=bep.json //a:arg -- 'arg1' 'arg2' \
    >&"$TEST_log" || fail "run failed"

  expect_log "ARG 1: arg1"
  expect_log "ARG 2: arg2"

  ls >& "$TEST_log"
  cat bep.json >> "$TEST_log"

  expect_log "execRequest"
  expect_log "argv"
  expect_log "arg1"
  expect_log "arg2"
}

function test_no_residue_in_run_bep(){
  add_rules_shell "MODULE.bazel"
  mkdir -p a
  cat > a/BUILD <<'EOF'
load("@rules_shell//shell:sh_binary.bzl", "sh_binary")
sh_binary(
    name = 'arg',
    srcs = ['arg.sh'],
)
EOF

  cat > a/arg.sh <<'EOF'
#!/usr/bin/env bash

COUNTER=1
for i in "$@"; do
  echo "ARG $COUNTER": $i;
  COUNTER=$(($COUNTER+1))
done
EOF

  chmod +x a/arg.sh
  bazel run --build_event_json_file=bep.json \
  --experimental_run_bep_event_include_residue=false //a:arg -- \
    'arg1' 'arg2' \
    >&"$TEST_log" || fail "run failed"

  expect_log "ARG 1: arg1"
  expect_log "ARG 2: arg2"

  ls >& "$TEST_log"
  cat bep.json >> "$TEST_log"

  expect_log "//a:arg"
  expect_log "execRequest"
  expect_log "argv"
  expect_log "REDACTED"
  expect_not_log "arg1"
  expect_not_log "arg2"
}


function test_residue_in_run_test_bep(){
  add_rules_shell "MODULE.bazel"
  mkdir -p a
  cat > a/BUILD <<'EOF'
load("@rules_shell//shell:sh_test.bzl", "sh_test")
sh_test(
    name = 'arg',
    srcs = ['arg_test.sh'],
)
EOF

  cat > a/arg_test.sh <<'EOF'
#!/usr/bin/env bash

COUNTER=1
for i in "$@"; do
  echo "ARG $COUNTER": $i;
  (( COUNTER++ ))
done
EOF

  chmod +x a/arg_test.sh
  bazel run --experimental_run_bep_event_include_residue=true \
   --build_event_json_file=bep.json //a:arg -- 'arg1' 'arg2' \
    >&"$TEST_log" || fail "run failed"

  expect_log "ARG 1: arg1"
  expect_log "ARG 2: arg2"

  ls >& "$TEST_log"
  cat bep.json >> "$TEST_log"

  expect_log "execRequest"
  expect_log "argv"
  expect_log "arg1"
  expect_log "arg2"
}

function test_no_residue_in_run_test_bep(){
  add_rules_shell "MODULE.bazel"
  mkdir -p a
  cat > a/BUILD <<'EOF'
load("@rules_shell//shell:sh_test.bzl", "sh_test")
sh_test(
    name = 'arg',
    srcs = ['arg_test.sh'],
)
EOF

  cat > a/arg_test.sh <<'EOF'
#!/usr/bin/env bash

COUNTER=1
for i in "$@"; do
  echo "ARG $COUNTER": $i;
  (( COUNTER++ ))
done
EOF

  chmod +x a/arg_test.sh
  bazel run --build_event_json_file=bep.json \
  --experimental_run_bep_event_include_residue=false //a:arg -- \
    'arg1' 'arg2' \
    >&"$TEST_log" || fail "run failed"

  expect_log "ARG 1: arg1"
  expect_log "ARG 2: arg2"

  ls >& "$TEST_log"
  cat bep.json >> "$TEST_log"

  expect_log "execRequest"
  expect_log "argv"
  expect_log "REDACTED"
  expect_not_log "arg1"
  expect_not_log "arg2"
}

run_suite "Bazel-specific integration tests for the build-event stream"
