#!/usr/bin/env bash

set -eu

eval $(go env)

echo "Checking for license headers..."
EXPECTED_HEADER='//
// 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.'

HEADER_CHECK_FAILED=0

for file in $(find config internal validate tests butane -name \*.go -type f -not -name schema.go); do
    # Don't check the first line because the year will vary
    HEADER="$(head -n 13 ${file} | tail -n 12)"
    if [ "${HEADER}" != "${EXPECTED_HEADER}" ]; then
        echo "${file} is missing the Apache License header"
        HEADER_CHECK_FAILED=1
    fi
done

if [ "${HEADER_CHECK_FAILED}" -ne 0 ]; then
    exit 1
fi

SRC=$(find . -name '*.go' -not -path "./vendor/*")

PKG=$(go list ./... | \
	grep --invert-match vendor | grep --invert-match tests)

# https://github.com/golang/go/issues/15067
PKG_VET=$(go list ./... | \
	grep --invert-match vendor | \
	grep --invert-match internal/log)

echo "Checking gofix..."
go_version=$(go version | awk '{print $3}')
# Go 1.26+ removed the -go flag from go tool fix
go_major=$(echo "$go_version" | sed 's/go\([0-9]*\)\.\([0-9]*\).*/\1/')
go_minor=$(echo "$go_version" | sed 's/go\([0-9]*\)\.\([0-9]*\).*/\2/')
if [ "$go_major" -lt 1 ] || { [ "$go_major" -eq 1 ] && [ "$go_minor" -lt 26 ]; }; then
    go tool fix -go="$go_version" -diff $SRC
else
    # Go 1.26+ uses unitchecker and exits non-zero when it finds fixable issues
    # Make this non-fatal since finding fixable code shouldn't fail the build
    go tool fix -diff $SRC || true
fi

echo "Checking gofmt..."
res=$(gofmt -d -e -s $SRC)
echo "${res}"
if [ -n "${res}" ]; then
	exit 1
fi

echo "Checking govet..."
go vet $PKG_VET

echo "Running tests..."
if [ "$GOARCH" == amd64 ]; then
	go test -timeout 60s -cover "${@}" ${PKG} --race
else
	go test -timeout 60s -cover "${@}" ${PKG}
fi

echo "Checking docs..."
go run internal/util/tools/docs/docs.go

echo "Checking Butane docs..."
# Butane configs in the docs embed local files; supply the ones they reference
# so translation fails on a broken config rather than on a missing file.
butane_files_dir=$(mktemp -d)
trap 'rm -rf "${butane_files_dir}"' EXIT
mkdir -p "${butane_files_dir}/tree"
touch "${butane_files_dir}"/{config.ign,ca.pem,example.conf,example.service,file,file-epilogue,local-file3}
echo "ssh-rsa AAAA" > "${butane_files_dir}/id_rsa.pub"
echo "ssh-ed25519 AAAA" > "${butane_files_dir}/id_ed25519.pub"
echo '{"ignition": {"version": "3.5.0"}}' > "${butane_files_dir}/ignition.ign"
go run internal/util/tools/docs/docs.go -root butane/docs -files-dir "${butane_files_dir}"
# Ensure every platform is listed in supported-platforms.md
platforms=$(grep -A 1 -h platform.Register internal/providers/*/* | grep Name: | cut -f2 -d\")
if [ -z "${platforms}" ]; then
    echo "Found no platform IDs; update $0" >&2
    exit 1
fi
for id in ${platforms}; do
    case "${id}" in
    file) ;;  # Undocumented platform ID for testing
    *)
        if ! grep -qF "\`${id}\`" docs/supported-platforms.md; then
            echo "Undocumented platform ID: $id" >&2
            exit 1
        fi
        ;;
    esac
done

echo "Success"
