path: root/kern

#!/bin/sh

set -e

KERNFRAG=${KERNFRAG:-/etc/kernfrag}

err() {
	printf "kern: %s\n" "$@" >&2
}

errx() {
	err "$@"
	exit 1
}

has_boot_mount() {
	awk '{print $2}' /etc/fstab | grep -q ^/boot$
}

is_boot_mounted() {
	mountpoint -q /boot
}

mount_boot_on_demand() {
	if has_boot_mount && ! is_boot_mounted; then
		sudo mount /boot
	fi
}

get_latest_kernel() {
	eselect --brief kernel list | tail -n1
}

get_selected_kernel() {
	readlink /usr/src/linux
}

get_running_kernel() {
	uname -r
}

get_running_config() {
	mount_boot_on_demand
	config=/boot/config-$(get_running_kernel)
	test -r "$config" || errx "Could not read configuration file: $config"
	echo "$config"
}

set_kernel() {
	kern=${1:-$(get_latest_kernel)}
	selected=$(get_selected_kernel)
	kerndir=/usr/src/$kern

	test -d "$kerndir" || errx "No such file or directory: $kerndir"

	if test "$kern" = "$selected"; then
		err "Warning: Latest kernel already selected: $kern"
		return
	fi

	printf "Selecting kernel: %s\n" "$kern"

	sudo eselect kernel set "$kern"
	sudo chown -R "$(id -un)":"$(id -gn)" "$kerndir"
}

config_kernel() {
	host=${1:-$(hostname)}

	test -d "$KERNFRAG" || errx "No such file or directory: $KERNFRAG"

	printf "Starting merge for %s...\n\n" "$host"

	awk -v prefix="$KERNFRAG/fragments/" '{print prefix $0}' \
		"$KERNFRAG/templates/base" "$KERNFRAG/templates/$host" | xargs scripts/kconfig/merge_config.sh
}

diff_kernel() {
	diff=${1:-$(get_running_config)}
	nvim -d .config "$diff"
}

build_kernel() {
	make -j"$(nproc)"
}

install_kernel() {
	mount_boot_on_demand
	sudo make install modules_install
}

post_install_kernel() {
	sudo emerge --ask=n @module-rebuild
}

clean_kernel() {
	sudo eclean-kernel -An3
}

test_diff() {
	dt=$(mktemp)
	sed -nE '/^(# )?CONFIG_.*/p' "$(get_running_config)" > "$dt"
	if ! sed -nE '/^(# )?CONFIG_.*/p' .config | diff -q "$dt" -; then
		diff_kernel
		printf "Continue? [y/N] "
		read -r response
		if test "$response" != "y"; then
			rm "$dt"
			return 1
		fi
	fi
	rm "$dt"
}

all() {
	set_kernel
	cd /usr/src/linux
	config_kernel
	test_diff
	build_kernel
	install_kernel
	post_install_kernel
	clean_kernel
}

if test $# -eq 0; then
	all
	exit 0
fi

cmd=$1
shift

case $cmd in
	set) set_kernel "$@";;
	config) config_kernel "$@";;
	diff) diff_kernel "$@";;
	build) build_kernel;;
	install) install_kernel;;
	postinst) post_install_kernel;;
	clean) clean_kernel;;
	all) all;;
esac