#!/usr/bin/env bash

# This script turns many *.sql.go files into a single queries.sql.go file. This
# is due to sqlc's behavior when using multiple sql files to output them to
# multiple Go files. We decided it would be cleaner to move these to a single
# file for readability. We should probably contribute the option to do this
# upstream instead, because this is quite janky.

set -euo pipefail

SCRIPT_DIR=$(dirname "${BASH_SOURCE[0]}")

(
	cd "$SCRIPT_DIR"

	echo generate 1>&2

	# Dump the updated schema (use make to utilize caching).
	if [[ "${SKIP_DUMP_SQL:-0}" != 1 ]]; then
		make -C ../.. --no-print-directory coderd/database/dump.sql
	fi
	# The logic below depends on the exact version being correct :(
	sqlc generate

	# Work directory for formatting before atomic replacement of
	# generated files, ensuring the source tree is never left in a
	# partially written state.
	mkdir -p ../../_gen
	workdir=$(mktemp -d ../../_gen/.dbgen.XXXXXX)
	trap 'rm -rf "$workdir"' EXIT

	first=true
	files=$(find ./queries/ -type f -name "*.sql.go" | LC_ALL=C sort)
	for fi in $files; do
		# Find the last line from the imports section and add 1. We have to
		# disable pipefail temporarily to avoid ERRPIPE errors when piping into
		# `head -n1`.
		set +o pipefail
		cut=$(grep -n ')' "$fi" | head -n 1 | cut -d: -f1)
		set -o pipefail
		cut=$((cut + 1))

		# Copy the header from the first file only, ignoring the source comment.
		if $first; then
			head -n 6 <"$fi" | grep -v "source" >"$workdir/queries.sql.go"
			first=false
		fi

		# Append the file past the imports section into queries.sql.go.
		tail -n "+$cut" <"$fi" >>"$workdir/queries.sql.go"
	done

	# Move sqlc outputs into workdir for formatting.
	mv queries/querier.go "$workdir/querier.go"
	mv queries/models.go "$workdir/models.go"

	# Remove temporary go files.
	rm -f queries/*.go

	# Fix struct/interface names in the workdir (not the source tree).
	gofmt -w -r 'Querier -> sqlcQuerier' -- "$workdir"/*.go
	gofmt -w -r 'Queries -> sqlQuerier' -- "$workdir"/*.go

	# Ensure correct imports exist. Modules must all be downloaded so we
	# get correct suggestions.
	go mod download
	go tool golang.org/x/tools/cmd/goimports -w "$workdir/queries.sql.go"

	# Atomically replace all three target files.
	mv "$workdir/queries.sql.go" queries.sql.go
	mv "$workdir/querier.go" querier.go
	mv "$workdir/models.go" models.go

	go run ../../scripts/dbgen
	# This will error if a view is broken. This is in it's own package to avoid
	# stuff like dbmock breaking builds if it's out of date from the interface.
	go test ./gentest
)