96 lines
2.0 KiB
Bash
Executable File
96 lines
2.0 KiB
Bash
Executable File
#!/bin/bash
|
|
# Credit from: https://github.com/alexedwards/go-reload/tree/aabe19d0a9935d1238763a4a35e71787854cd5f5
|
|
|
|
####################################################
|
|
# @kirsle's custom changes from upstream are below #
|
|
####################################################
|
|
|
|
function die() {
|
|
echo >&2 $1
|
|
exit 1
|
|
}
|
|
|
|
# Before we crash and burn, make sure necessary programs are installed!
|
|
command -v inotifywait || die "I need the inotifywait command (apt install inotify-tools)"
|
|
|
|
################################
|
|
# end @kirsle's custom changes #
|
|
################################
|
|
|
|
function monitor() {
|
|
if [ "$2" = "true" ]; then
|
|
# Watch all files in the specified directory
|
|
# Call the restart function when they are saved
|
|
inotifywait -q -m -r -e close_write -e moved_to $1 |
|
|
while read line; do
|
|
restart
|
|
done
|
|
else
|
|
# Watch all *.go files in the specified directory
|
|
# Call the restart function when they are saved
|
|
inotifywait -q -m -r -e close_write -e moved_to --exclude '[^g][^o]$' $1 |
|
|
while read line; do
|
|
restart
|
|
done
|
|
fi
|
|
}
|
|
|
|
# Terminate and rerun the main Go program
|
|
function restart {
|
|
if [ "$(pidof $PROCESS_NAME)" ]; then
|
|
killall -q -w -9 $PROCESS_NAME
|
|
fi
|
|
echo ">> Reloading..."
|
|
eval "go run $ARGS &"
|
|
}
|
|
|
|
# Make sure all background processes get terminated
|
|
function close {
|
|
killall -q -w -9 inotifywait
|
|
exit 0
|
|
}
|
|
|
|
trap close INT
|
|
echo "== Go-reload"
|
|
|
|
WATCH_ALL=false
|
|
while getopts ":a" opt; do
|
|
case $opt in
|
|
a)
|
|
WATCH_ALL=true
|
|
;;
|
|
\?)
|
|
echo "Invalid option: -$OPTARG" >&2
|
|
exit 0
|
|
;;
|
|
esac
|
|
done
|
|
|
|
shift "$((OPTIND - 1))"
|
|
|
|
FILE_NAME=$(basename $1)
|
|
PROCESS_NAME=${FILE_NAME%%.*}
|
|
|
|
ARGS=$@
|
|
|
|
# Start the main Go program
|
|
echo ">> Watching directories, CTRL+C to stop"
|
|
eval "go run $ARGS &"
|
|
|
|
# Monitor all /src directories on the GOPATH
|
|
OIFS="$IFS"
|
|
IFS=':'
|
|
for path in $GOPATH
|
|
do
|
|
monitor $path/src $WATCH_ALL &
|
|
done
|
|
IFS="$OIFS"
|
|
|
|
# If the current working directory isn't on the GOPATH, monitor it too
|
|
if [[ $PWD != "$GOPATH/"* ]]
|
|
then
|
|
monitor $PWD $WATCH_ALL
|
|
fi
|
|
|
|
wait
|