autobuild

#!/usr/bin/env bash

# autobuild client
# Copyright (C) 2024 rail5
# Free software (GNU Affero GPL v3)

temporary_storage_directory="$(mktemp --tmpdir -d autobuild.file.XXXXXXXXXXXX)"
temporary_log_file=$(mktemp --tmpdir autobuild.log.XXXXXXXXXXXX)
autobuild_storage_directory="/var/autobuild"

# shellcheck disable=SC2124
passed_options="${@@Q}"

trap 'graceful_exit' INT
trap 'graceful_exit' TERM
trap 'graceful_exit' HUP

function get_daemon_pid() {
	head -n 1 "$temporary_log_file" | awk '{print $2}'
}

function get_job_id() {
	sed -n '2{p;q;}' "$temporary_log_file" | awk '{print $2}'
}

function graceful_exit() {
	if [[ -f "$temporary_log_file" ]]; then
		pid_to_kill=$(get_daemon_pid)
		socat -t 86400 - UNIX-CONNECT:/var/run/autobuild.socket >/dev/null 2>&1 <<<"-k $pid_to_kill" # Send kill command to daemon
		tail --pid="$pid_to_kill" -f /dev/null 2>/dev/null
		rm -f "/tmp/autobuild.kill.$pid_to_kill" 2>/dev/null
		rm -f "$temporary_log_file" 2>/dev/null
	fi

	if [[ -d "$temporary_storage_directory" ]]; then
		rm -rf "${temporary_storage_directory:?}"
	fi

	exit 0
}

# getopt
TEMP=$(getopt -o bo:p:s --long bell,output:,package:,setup \
              -q -- "$@")

eval set -- "$TEMP"

ring_bell=false
output_directory=$(pwd)

while true; do
	case "$1" in
		-b | --bell )
			ring_bell=true; shift ;;
		-o | --output )
			output_directory=$(realpath "$2")
			shift 2 ;;
		-p | --package )
			# Make sure the autobuild daemon can *read* any files we might give it for the package
			if [[ -f "$2" && "$2" == *".tar.gz" ]]; then
				file_basename="$(basename "$(realpath "$2")")"
				temporary_file_path="$temporary_storage_directory/$file_basename"
				
				cp "$(realpath "$2")" "$temporary_file_path"
				chmod -R 755 "$temporary_storage_directory/"

				passed_options="${passed_options/$2/"$temporary_file_path"}" 
			fi
			shift 2 ;;
		-s | --setup )
			if [[ "$(whoami)" != root ]]; then
				echo "-s must be run as root"
				echo "Try again with 'sudo'"
				graceful_exit
			fi
			sudo -u _autobuild autobuild-setup
			graceful_exit ;;
		-- ) shift; break ;;
		* ) break ;;
	esac
done

# Run the autobuild daemon to build the packages, write the responses to the terminal window and to the temporary log file
socat -t 86400 - UNIX-CONNECT:/var/run/autobuild.socket <<<"$passed_options" | tee "$temporary_log_file"

# Pull the built packages from /var/autobuild/builds and save them in the output directory
cp -r "$autobuild_storage_directory/builds/$(get_job_id)/"* "$output_directory/" 2>/dev/null

# Ask the autobuild daemon to delete the builds from /var/autobuild/builds so we don't waste space
socat - UNIX-CONNECT:/var/run/autobuild.socket <<<"-r $(get_job_id)" >/dev/null

# Ring the bell if the user asked for it
if [[ $ring_bell == true ]]; then
	paplay /usr/share/autobuild/bell.ogg
fi

graceful_exit