131 lines
3.4 KiB
Bash
Executable file
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
|