Refactor
* fix session name quoting for long strings including spaces * exit plugin if no sessions are running * echo to stderr * debug to named tmpfile
This commit is contained in:
dgengtek
parent
9a8e30f196
commit
4b6fd87db3
1 changed files with 59 additions and 61 deletions
|
|
@ -1,12 +1,13 @@
|
||||||
#!/bin/bash
|
#!/usr/bin/env bash
|
||||||
|
set -eu
|
||||||
|
|
||||||
POLL_INTERVAL=10 # seconds
|
readonly DEBUG=0
|
||||||
|
readonly POLL_INTERVAL=10 # seconds
|
||||||
BUCKET_ID="aw-watcher-tmux"
|
readonly BUCKET_ID="aw-watcher-tmux"
|
||||||
HOST="localhost"
|
readonly HOST="localhost"
|
||||||
PORT="5600"
|
readonly PORT="5600"
|
||||||
API_URL="http://$HOST:$PORT/api"
|
readonly API_URL="http://$HOST:$PORT/api"
|
||||||
PULSETIME="120.0"
|
readonly PULSETIME="120.0"
|
||||||
|
|
||||||
######
|
######
|
||||||
# Related documentation:
|
# Related documentation:
|
||||||
|
|
@ -15,40 +16,38 @@ PULSETIME="120.0"
|
||||||
#
|
#
|
||||||
#
|
#
|
||||||
|
|
||||||
|
TMP_FILE=$(mktemp "${TMP:-/tmp}/${BUCKET_ID}.XXXXXX")
|
||||||
|
|
||||||
|
if (($DEBUG)); then
|
||||||
|
echo "$TMP_FILE"
|
||||||
|
set -x
|
||||||
|
exec 1>>"$TMP_FILE"
|
||||||
|
exec 2>&1
|
||||||
|
fi
|
||||||
|
|
||||||
### FUNCTIONS
|
### FUNCTIONS
|
||||||
|
|
||||||
DEBUG=0
|
|
||||||
TMP_FILE=$(mktemp)
|
|
||||||
echo $TMP_FILE
|
|
||||||
|
|
||||||
init_bucket() {
|
init_bucket() {
|
||||||
HTTP_CODE=$(curl -X GET "${API_URL}/0/buckets/$BUCKET_ID" -H "accept: application/json" -s -o /dev/null -w %{http_code})
|
HTTP_CODE=$(curl -X GET "${API_URL}/0/buckets/$BUCKET_ID" -H "accept: application/json" -s -o /dev/null -w '%{http_code}')
|
||||||
if (( $HTTP_CODE == 404 )) # not found
|
if (( $HTTP_CODE == 404 )); then # not found
|
||||||
then
|
|
||||||
JSON="{\"client\":\"$BUCKET_ID\",\"type\":\"tmux.sessions\",\"hostname\":\"$(hostname)\"}"
|
JSON="{\"client\":\"$BUCKET_ID\",\"type\":\"tmux.sessions\",\"hostname\":\"$(hostname)\"}"
|
||||||
HTTP_CODE=$(curl -X POST "${API_URL}/0/buckets/$BUCKET_ID" -H "accept: application/json" -H "Content-Type: application/json" -d "$JSON" -s -o /dev/null -w %{http_code})
|
HTTP_CODE=$(curl -X POST "${API_URL}/0/buckets/$BUCKET_ID" -H "accept: application/json" -H "Content-Type: application/json" -d "$JSON" -s -o /dev/null -w '%{http_code}')
|
||||||
if (( $HTTP_CODE != 200 ))
|
if (( $HTTP_CODE != 200 )); then
|
||||||
then
|
echo "ERROR creating bucket" >&2
|
||||||
echo "ERROR creating bucket"
|
exit 1
|
||||||
exit -1
|
|
||||||
fi
|
fi
|
||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
log_to_bucket() {
|
log_to_bucket() {
|
||||||
sess=$1
|
session=$1
|
||||||
DATA=$(tmux display -t $sess -p "{\"title\":\"#{session_name}\",\"session_name\":\"#{session_name}\",\"window_name\":\"#{window_name}\",\"pane_title\":\"#{pane_title}\",\"pane_current_command\":\"#{pane_current_command}\",\"pane_current_path\":\"#{pane_current_path}\"}");
|
DATA=$(tmux display -t "$session" -p "{\"title\":\"#{session_name}\",\"session_name\":\"#{session_name}\",\"window_name\":\"#{window_name}\",\"pane_title\":\"#{pane_title}\",\"pane_current_command\":\"#{pane_current_command}\",\"pane_current_path\":\"#{pane_current_path}\"}");
|
||||||
PAYLOAD="{\"timestamp\":\"$(date -Is)\",\"duration\":0,\"data\":$DATA}"
|
PAYLOAD="{\"timestamp\":\"$(date -Is)\",\"duration\":0,\"data\":$DATA}"
|
||||||
echo "$PAYLOAD"
|
(($DEBUG)) && echo "$PAYLOAD" >&2
|
||||||
HTTP_CODE=$(curl -X POST "${API_URL}/0/buckets/$BUCKET_ID/heartbeat?pulsetime=$PULSETIME" -H "accept: application/json" -H "Content-Type: application/json" -d "$PAYLOAD" -s -o $TMP_FILE -w %{http_code})
|
HTTP_CODE=$(curl -X POST "${API_URL}/0/buckets/$BUCKET_ID/heartbeat?pulsetime=$PULSETIME" -H "accept: application/json" -H "Content-Type: application/json" -d "$PAYLOAD" -s -o "$TMP_FILE" -w '%{http_code}')
|
||||||
if (( $HTTP_CODE != 200 )); then
|
if (( $HTTP_CODE != 200 )); then
|
||||||
echo "Request failed"
|
echo "Request failed - $HTTP_CODE" >&2
|
||||||
cat $TMP_FILE
|
return 1
|
||||||
fi
|
|
||||||
|
|
||||||
if [[ "$DEBUG" -eq "1" ]]; then
|
|
||||||
cat $TMP_FILE
|
|
||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
@ -59,37 +58,36 @@ declare -A act_last
|
||||||
declare -A act_current
|
declare -A act_current
|
||||||
|
|
||||||
init_bucket
|
init_bucket
|
||||||
|
OLDIFS=$IFS
|
||||||
|
|
||||||
while [ true ]
|
while :; do
|
||||||
do
|
if ! sessions=$(tmux list-sessions -F '#{session_name}'); then
|
||||||
#clear
|
echo "tmux list-sessions ERROR: $?" >&2
|
||||||
sessions=$(tmux list-sessions | awk '{print $1}')
|
exit 1
|
||||||
if (( $? != 0 )); then
|
|
||||||
echo "tmux list-sessions ERROR: $?"
|
|
||||||
fi
|
fi
|
||||||
if (( $? == 0 )); then
|
if [[ -z "$sessions" ]]; then
|
||||||
LAST_IFS=$IFS
|
echo "sessions empty" >&2
|
||||||
IFS='
|
continue
|
||||||
'
|
|
||||||
for sess in ${sessions}; do
|
|
||||||
act_time=$(tmux display -t $sess -p '#{session_activity}')
|
|
||||||
if [[ ! -v "act_last[$sess]" ]]; then
|
|
||||||
act_last[$sess]='0'
|
|
||||||
fi
|
fi
|
||||||
if (( $act_time > ${act_last[$sess]} )); then
|
|
||||||
# echo "###> "$sess' '$(date -Iseconds)' '$act_time' '$act_last[$sess] ## >> tmux-sess-act.log
|
IFS=$'\n'
|
||||||
log_to_bucket $sess
|
for session in ${sessions}; do
|
||||||
|
act_time=$(tmux display -t "$session" -p '#{session_activity}')
|
||||||
|
if [[ ! -v "act_last[$session]" ]]; then
|
||||||
|
act_last[$session]='0'
|
||||||
fi
|
fi
|
||||||
act_current[$sess]=$act_time
|
if (( $act_time > ${act_last[$session]} )); then
|
||||||
|
# echo "###> "$session' '$(date -Iseconds)' '$act_time' '$act_last[$session] ## >> tmux-session-act.log
|
||||||
|
log_to_bucket "$session"
|
||||||
|
fi
|
||||||
|
act_current[$session]=$act_time
|
||||||
done
|
done
|
||||||
IFS=$LAST_IFS
|
IFS=$OLDIFS
|
||||||
# copy arrays
|
# copy arrays
|
||||||
unset R
|
|
||||||
declare -A act_last
|
declare -A act_last
|
||||||
for sess in "${!act_current[@]}"; do
|
for session in "${!act_current[@]}"; do
|
||||||
act_last[$sess]=${act_current[$sess]}
|
act_last[$session]=${act_current[$session]}
|
||||||
done
|
done
|
||||||
fi
|
|
||||||
|
|
||||||
sleep $POLL_INTERVAL
|
sleep $POLL_INTERVAL
|
||||||
done
|
done
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue