Skip to content

Commit

Permalink
New segment github_notifications
Browse files Browse the repository at this point in the history
* add new util function "is_tmp_valid" to make handling
  of temp files based on last update reusable
* add new segment
* add jq requirement to README.md

fixes #431
  • Loading branch information
xx4h committed Jul 2, 2024
1 parent 29143d8 commit 4bb7c73
Show file tree
Hide file tree
Showing 3 changed files with 158 additions and 0 deletions.
1 change: 1 addition & 0 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -106,6 +106,7 @@ Some segments have their own requirements. If you enable them in your theme, mak
* `tmux_mem_cpu_load.sh`: [tmux-mem-cpu-load](https://github.com/thewtex/tmux-mem-cpu-load)
* `rainbarf.sh`: [rainbarf](https://github.com/creaktive/rainbarf)
* `weather.sh`: GNU `grep` with Perl regular expression enabled (FreeBSD specific), `jq` for yrno weather API.
* `github_notifications.sh`: `jq` for GitHub API.

## FreeBSD specific requirements
Preinstalled `grep` in FreeBSD doesn't support Perl regular expressions. Solution is rather simple -- you need to use `textproc/gnugrep` port instead. You also need to make sure, that it has support for PCRE and is compiled with `--enable-perl-regexp` flag.
Expand Down
30 changes: 30 additions & 0 deletions lib/util.sh
Original file line number Diff line number Diff line change
Expand Up @@ -12,3 +12,33 @@ is_flag_enabled() {
;;
esac
}

is_tmp_valid() {
local tmp_file="$1"
local tmp_validity="$2"

if [ -f "$tmp_file" ]; then
if shell_is_osx || shell_is_bsd; then
stat >/dev/null 2>&1 && is_gnu_stat=false || is_gnu_stat=true
if [ "$is_gnu_stat" == "true" ]; then
last_update=$(stat -c "%Y" "${tmp_file}")
else
last_update=$(stat -f "%m" "${tmp_file}")
fi
elif shell_is_linux || [ -z "$is_gnu_stat" ]; then
last_update=$(stat -c "%Y" "${tmp_file}")
fi

time_now=$(date +%s)
valid=$(echo "(${time_now}-${last_update}) < ${tmp_validity}" | bc)

if [ "$valid" -eq 1 ]; then
return 0
else
return 1
fi
else
return 1
fi

}
127 changes: 127 additions & 0 deletions segments/github_notifications.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,127 @@
# shellcheck shell=bash

# shellcheck source=lib/util.sh
source "${TMUX_POWERLINE_DIR_LIB}/util.sh"

TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SINCE="${TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SINCE:-$(date +%Y-%m-%dT00:00:00Z)}"
TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SINCE_ENABLE="${TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SINCE_ENABLE:-no}"
TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_PER_PAGE="${TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_PER_PAGE:-50}"
TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_MAX_PAGES="${TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_MAX_PAGES:-10}"
TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_UPDATE_INTERVAL="${TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_UPDATE_INTERVAL:-60}"
TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SUMMARIZE="${TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SUMMARIZE:-no}"
TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SYMBOL_MODE="${TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SYMBOL_MODE:-yes}"
TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_HIDE_NO_NOTIFICATIONS="${TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_HIDE_NO_NOTIFICATIONS:-yes}"

generate_segmentrc() {
read -r -d '' rccontents <<EORC
# Github token (https://github.com/settings/tokens) with at least "notifications" scope
export TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_TOKEN=""
# Include available notification reasons (https://docs.github.com/en/rest/activity/notifications?apiVersion=2022-11-28#about-notification-reasons),
# in the format "REASON:SEPARATOR"
# export TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_REASONS=("approval_requested:-󰴄 " "assign:-󰎔 " "author:-󰔗 " "comment:- " "ci_activity:-󰙨 " "invitation:- " "manual:-󱥃 " "mention:- " "review_requested:- " "security_alert:-󰒃 " "state_change:-󱇯 " "subscribed:- " "team_mention:- ")
# Or if you don't like so many symbols, try the abbreviation variant
# export TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_REASONS=("approval_requested:areq" "assign:as" "author:au" "comment:co" "ci_activity:ci" "invitation:in" "manual:ma" "mention:me" "review_requested:rreq" "security_alert:sec" "state_change:st" "subscribed:sub" "team_mention:team")
# Use symbol mode (ignored if you set TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_REASONS yourself)
# export TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SYMBOL_MODE="${TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SYMBOL_MODE}"
# Summarize all notifications
# export TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SUMMARIZE="${TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SUMMARIZE}"
# Hide if no notifications
# export TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_HIDE_NO_NOTIFICATIONS="${TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_HIDE_NO_NOTIFICATIONS}"
# Only show new notifications since date (default: today) (takes up to UPDATE_INTERVAL time to take effect)
# export TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SINCE="\$(date +%Y-%m-%dT00:00:00Z)"
# Enable show only notifications since date (takes up to UPDATE_INTERVAL time to take effect)
# export TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SINCE_ENABLE="${TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SINCE_ENABLE}"
# Maximum notifications to retreive per page (upstream github default per_page, 50)
# export TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_PER_PAGE="${TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_PER_PAGE}"
# Maximum pages to retreive
# export TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_MAX_PAGES="${TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_MAX_PAGES}"
# Update interval to pull latest state from github api
# export TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_UPDATE_INTERVAL="${TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_UPDATE_INTERVAL}"
EORC
echo "$rccontents"
}

run_segment() {
__process_settings
if ! type curl >/dev/null 2>&1; then
return 0
fi

if [ -z "$TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_TOKEN" ]; then
return 0
fi

local tmp_file
local api_query

tmp_file="${TMUX_POWERLINE_DIR_TEMPORARY}/github_notifications.stat"
tmp_headers_file="$TMUX_POWERLINE_DIR_TEMPORARY/github_notifications.headers"

if is_flag_enabled "$TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SINCE_ENABLE"; then
api_query="since=${TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SINCE}&"
fi
api_query="${api_query}per_page=${TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_PER_PAGE}"

if ! is_tmp_valid "$tmp_file" "$TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_UPDATE_INTERVAL"; then
read -r -d '' notifications < <(curl -s --url "https://api.github.com/notifications?${api_query}" -D "$tmp_headers_file" --header "Authorization: Bearer $TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_TOKEN")
if link=$(grep '^link: ' "$tmp_headers_file"); then
IFS=$',' read -r _ last <<<"$link"
last_page=$(echo "$last" | sed -r 's/.*page=([0-9]+)>; rel="last"/\1/' | tr -d '\r\n')
page=2
until [ "$page" -gt "$last_page" ] || [ "$page" -gt "$TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_MAX_PAGES" ]; do
read -r -d '' _tmp_notifications < <(curl -s --url "https://api.github.com/notifications?${api_query}&page=$page" --header "Authorization: Bearer $TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_TOKEN")
notifications=$(jq -s 'add' <(echo "$notifications") <(echo "$_tmp_notifications"))
((page++))
done
fi

all_count=0
result=""

for reason_entry in "${TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_REASONS[@]}"; do
count=0
IFS=$':' read -r reason separator <<<"$reason_entry"
count="$(echo "$notifications" | jq -c '.[] | select( .reason == "'"$reason"'" )' | jq -s 'length')"

if [ "$count" -eq 0 ] && is_flag_enabled "$TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_HIDE_NO_NOTIFICATIONS"; then
continue
fi

if is_flag_enabled "$TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SUMMARIZE"; then
all_count="$((all_count + count))"
else
if ! [[ "$separator" =~ ^-.*$ ]]; then
result="$result $separator:$count"
else
result="$result ${separator:1}$count"
fi
fi
done

if is_flag_enabled "$TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SUMMARIZE"; then
if [ "$all_count" -gt 0 ] || ! is_flag_enabled "$TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_HIDE_NO_NOTIFICATIONS"; then
echo "$all_count" >"$tmp_file"
else
echo -n >"$tmp_file"
fi
else
if [ -n "$result" ] || ! is_flag_enabled "$TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_HIDE_NO_NOTIFICATIONS"; then
echo "$result" >"$tmp_file"
else
echo -n >"$tmp_file"
fi
fi
fi

cat "$tmp_file"
}

__process_settings() {
if [ -z "$TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_REASONS" ]; then
if is_flag_enabled "$TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_SYMBOL_MODE"; then
export TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_REASONS=("approval_requested:-󰴄 " "assign:-󰎔 " "author:-󰔗 " "comment:- " "ci_activity:-󰙨 " "invitation:- " "manual:-󱥃 " "mention:- " "review_requested:- " "security_alert:-󰒃 " "state_change:-󱇯 " "subscribed:- " "team_mention:- ")
else
export TMUX_POWERLINE_SEG_GITHUB_NOTIFICATIONS_REASONS=("approval_requested:areq" "assign:as" "author:au" "comment:co" "ci_activity:ci" "invitation:in" "manual:ma" "mention:me" "review_requested:rreq" "security_alert:sec" "state_change:st" "subscribed:sub" "team_mention:team")
fi
fi
}

0 comments on commit 4bb7c73

Please sign in to comment.