matrix-notifier/matrix-notifier
Bruno BELANYI 7b71469624 matrix-notifier: add 'auto' message type
This is useful to only ping me when the build is failing.
2021-06-12 19:19:26 +02:00

131 lines
3.4 KiB
Bash
Executable file

#!/usr/bin/env bash
print_err() {
printf "%s\n" "$1" >&2
}
usage() {
print_err "Usage: $0 [option]..."
print_err ""
print_err " -h, --help"
print_err " print this usage screen and exit"
print_err " -t, --type"
print_err " which message type should be sent must be one of"
print_err " 'text' or 'notice', or 'auto'."
print_err " The special value 'auto' defaults to 'notice', unless"
print_err " 'DRONE_BUILD_STATUS' indicates a failure."
}
# Blessed Stack Overflow
rawurlencode() {
(set +x
local string="${1}"
local strlen=${#string}
local encoded=""
local pos c o
for ((pos=0 ; pos<strlen; pos++)); do
c=${string:$pos:1}
case "$c" in
[-_.~a-zA-Z0-9] ) o="$c" ;;
* ) printf -v o '%%%02x' "'$c"
esac
encoded+="$o"
done
echo "$encoded")
}
default_drone_message() {
local msg="Build ${DRONE_BUILD_STATUS}"
msg="$msg ${DRONE_REPO_OWNER}/${DRONE_REPO_NAME}/${DRONE_BUILD_NUMBER}"
msg="$msg (${DRONE_BRANCH})"
printf '%s' "$msg"
}
ensure_variables() {
if [ "$DRONE" == "true" ] && [ -z "$MESSAGE" ]; then
MESSAGE="$(default_drone_message)"
fi
if [ -z "$USER" ] || [ -z "$PASS" ]; then
print_err "You must provide USER and PASS"
exit 1
elif [ -z "$ADDRESS" ] || [ -z "$ROOM" ]; then
print_err "You must provide ADDRESS and ROOM"
exit 1
elif [ -z "$MESSAGE" ]; then
print_err "You must provide MESSAGE"
exit 1
fi
}
send_message() {
local login_json
local token
local message_json
if [ "$(curl -XGET "$ADDRESS/_matrix/client/r0/login" 2>/dev/null |
jq 'any(.flows[].type; .== "m.login.password")')" != "true" ]; then
print_err "Login method not supported"
exit 1
fi
login_json="$(printf '%s\n%s' "$USER" "$PASS" |
jq -Rn '[inputs] | {type: "m.login.password", user: .[0], password: .[1]}')" &>/dev/null
token="$(curl -XPOST \
-d "$login_json" \
"$ADDRESS/_matrix/client/r0/login" 2>/dev/null| jq .access_token --raw-output)"
if [ -z "$token" ] || [ "$token" == "null" ]; then
print_err "Error during login"
exit 1
fi
message_json="$(printf '%s' "$MESSAGE" |
jq --raw-input --slurp "{msgtype: \"$MSG_TYPE\", body: .}")"
curl -XPOST \
-d "$message_json" \
"$ADDRESS/_matrix/client/r0/rooms/$(rawurlencode "$ROOM")/send/m.room.message?access_token=$(rawurlencode "$token")" 2>/dev/null
}
MSG_TYPE='m.notice'
while [ $# -gt 0 ]; do
opt="$1"
shift
case "$opt" in
-t|--type)
arg="$1"
shift
if [ -z "$arg" ]; then
print_err "Must give a message type when using '-t|--type'"
exit 1
elif [ "$arg" == "text" ] || [ "$arg" == "notice" ]; then
MSG_TYPE="m.$arg"
elif [ "$arg" == "auto" ]; then
MSG_TYPE="m.notice"
if [ "$DRONE_BUILD_STATUS" == "failure" ]; then
MSG_TYPE="m.text"
fi
else
print_err "Invalid message type '$arg'"
exit 1
fi
;;
-h|--help)
usage
exit
;;
*)
print_err "Unknown argument '$opt'"
usage
exit 1
;;
esac
done
ensure_variables
send_message