home/lib/login_functions.sh

   1 #
   2
   3 d() {
   4     local -r word=$(fzf < /usr/share/dict/words)
   5     dict "$word"
   6 }
   7
   8 shell_activity_report() {
   9     # TODO: optional concrete number output
  10     # TODO: optional combinations of granularities: hour, weekday, month, year
  11     local group_by="$1"
  12     case "$group_by" in
  13         'mon') ;;
  14         'dow') ;;
  15         '') group_by='dow';;
  16         *)
  17             echo "Usage: $0 [mon|dow]" >&2
  18             kill -INT $$
  19     esac
  20     history \
  21     | awk -v group_by="$group_by" '
  22         function date2dow(y, m, d,    _t, _i) {
  23             # Contract:
  24             #   y > 1752,  1 <= m <= 12.
  25             # Source:
  26             #   Sakamoto`s methods
  27             #   https://en.wikipedia.org/wiki/Determination_of_the_day_of_the_week#Sakamoto%27s_methods
  28             _t[ 0] = 0
  29             _t[ 1] = 3
  30             _t[ 2] = 2
  31             _t[ 3] = 5
  32             _t[ 4] = 0
  33             _t[ 5] = 3
  34             _t[ 6] = 5
  35             _t[ 7] = 1
  36             _t[ 8] = 4
  37             _t[ 9] = 6
  38             _t[10] = 2
  39             _t[11] = 4
  40             y -= m < 3
  41             _i = int(y + y/4 - y/100 + y/400 + _t[m - 1] + d) % 7
  42             _i = _i == 0 ? 7 : _i  # Make Sunday last
  43             return _i
  44
  45         }
  46
  47         {
  48             # NOTE: $2 & $3 are specific to oh-my-zsh history output
  49             date = $2
  50             time = $3
  51             d_fields = split(date, d, "-")
  52             t_fields = split(time, t, ":")
  53             if (t_fields && d_fields) {
  54                 # +0 to coerce number from string
  55                 year  = d[1] + 0
  56                 month = d[2] + 0
  57                 day   = d[3] + 0
  58                 hour = t[1] + 0
  59                 dow = date2dow(year, month, day)
  60                 g = group_by == "mon" ? month : dow  # dow is default
  61                 c = count[g, hour]++
  62             }
  63             if (c > max)
  64                 max = c
  65         }
  66
  67         END {
  68             w[1] = "Monday"
  69             w[2] = "Tuesday"
  70             w[3] = "Wednesday"
  71             w[4] = "Thursday"
  72             w[5] = "Friday"
  73             w[6] = "Saturday"
  74             w[7] = "Sunday"
  75
  76             m[ 1] = "January"
  77             m[ 2] = "February"
  78             m[ 3] = "March"
  79             m[ 4] = "April"
  80             m[ 5] = "May"
  81             m[ 6] = "June"
  82             m[ 7] = "July"
  83             m[ 8] = "August"
  84             m[ 9] = "September"
  85             m[10] = "October"
  86             m[11] = "November"
  87             m[12] = "December"
  88
  89             n = group_by == "mon" ? 12 : 7  # dow is default
  90
  91             for (gid = 1; gid <= n; gid++) {
  92                 group = group_by == "mon" ? m[gid] : w[gid]
  93                 printf "%s\n", group;
  94                 for (hour=0; hour<24; hour++) {
  95                     c = count[gid, hour]
  96                     printf "  %2d ", hour
  97                     for (i = 1; i <= (c * 100) / max; i++)
  98                         printf "|"
  99                     printf "\n"
 100                 }
 101             }
 102         }'
 103 }
 104
 105 top_commands() {
 106     history \
 107     | awk '
 108         {
 109             count[$4]++
 110         }
 111
 112         END {
 113             for (cmd in count)
 114                 print count[cmd], cmd
 115         }' \
 116     | sort -n -r -k 1 \
 117     | head -50 \
 118     | awk '
 119         {
 120             cmd[NR] = $2
 121             c = count[NR] = $1 + 0  # + 0 to coerce number from string
 122             if (c > max)
 123                 max = c
 124         }
 125
 126         END {
 127             for (i = 1; i <= NR; i++) {
 128                 c = count[i]
 129                 printf "%s %d ", cmd[i], c
 130                 scaled = (c * 100) / max
 131                 for (j = 1; j <= scaled; j++)
 132                     printf "|"
 133                 printf "\n"
 134             }
 135         }' \
 136     | column -t
 137 }
 138
 139 # Top Disk-Using directories
 140 # TODO: Consider using numfmt instead of awk
 141 tdu() {
 142     du "$1" \
 143     | sort -n -k 1 \
 144     | tail -50 \
 145     | awk '
 146         {
 147             size = $1
 148             path = $0
 149             sub("^" $1 "\t+", "", path)
 150             gb = size / 1024 / 1024
 151             printf("%f\t%s\n", gb, path)
 152         }' \
 153     | cut -c 1-115
 154 }
 155
 156 # Top Disk-Using Files
 157 tduf() {
 158     find "$1" -type f -printf '%s\t%p\0' \
 159     | sort -z -n -k 1 \
 160     | tail -z -n 50 \
 161     | gawk -v RS='\0' '
 162         {
 163             size = $1
 164             path = $0
 165             sub("^" $1 "\t+", "", path)
 166             gb = size / 1024 / 1024 / 1024
 167             printf("%f\t%s\n", gb, path)
 168         }'
 169 }
 170
 171 # Most-recently modified file system objects
 172 recent() {
 173     # NOTES:
 174     # - intentionally not quoting the parameters, so that some can be ignored
 175     #   if not passed, rather than be passed to find as an empty string;
 176     # - %T+ is a GNU extension;
 177     # - gawk is able to split records on \0, while awk cannot.
 178     find $@ -printf '%T@ %T+ %p\0' \
 179     | tee >(gawk -v RS='\0' 'END { printf("[INFO] Total found: %d\n", NR); }') \
 180     | sort -z -k 1 -n -r \
 181     | head -n "$(stty size | awk 'NR == 1 {print $1 - 5}')" -z \
 182     | gawk -v RS='\0' '
 183         {
 184             sub("^" $1 " +", "")  # Remove epoch time
 185             sub("+", " ")         # Blank-out the default separator
 186             sub("\\.[0-9]+", "")  # Remove fractional seconds
 187             print
 188         }'
 189 }
 190
 191 recent_dirs() {
 192     recent "$1" -type d
 193 }
 194
 195 recent_files() {
 196     recent "$1" -type f
 197 }
 198
 199 pa_def_sink() {
 200     pactl info | awk '/^Default Sink:/ {print $3}'
 201 }
 202
 203 void_pkgs() {
 204     curl "https://xq-api.voidlinux.org/v1/query/x86_64?q=$1" | jq '.data'
 205 }
 206
 207 # Colorful man
 208 man() {
 209     # mb: begin blink
 210     # md: begin bold
 211     # me: end   bold, blink and underline
 212     #
 213     # so: begin standout (reverse video)
 214     # se: end   standout
 215     #
 216     # us: begin underline
 217     # ue: end   underline
 218
 219     LESS_TERMCAP_md=$'\e[01;30m' \
 220     LESS_TERMCAP_me=$'\e[0m' \
 221     LESS_TERMCAP_so=$'\e[01;44;33m' \
 222     LESS_TERMCAP_se=$'\e[0m' \
 223     LESS_TERMCAP_us=$'\e[01;33m' \
 224     LESS_TERMCAP_ue=$'\e[0m' \
 225     command man "$@"
 226 }
 227
 228 experiment() {
 229     cd "$(~/bin/experiment $@)" || kill -INT $$
 230 }
 231
 232 hump() {
 233     ledit -l "$(stty size | awk '{print $2}')" ocaml $@
 234 }
 235
 236 howto() {
 237     cat "$(find  ~/Archives/Documents/HOWTOs -mindepth 1 -maxdepth 1 | sort | fzf)"
 238 }
 239
 240 _yt() {
 241     local -r base_dir="$1"
 242     local -r opts="$2"
 243     local -r uri="$3"
 244
 245     local -r id=$(youtube-dlc --get-id "$uri")
 246     local -r title=$(youtube-dlc --get-title "$uri" | sed 's/[^A-Za-z0-9._-]/_/g')
 247     local -r dir="${base_dir}/${title}--${id}"
 248
 249     mkdir -p "$dir"
 250     cd "$dir" || kill -INT $$
 251     echo "$uri" > 'uri'
 252     youtube-dlc $opts -c --write-description --write-info-json "$uri"
 253 }
 254
 255 yt_audio() {
 256     local -r uri="$1"
 257     _yt "${DIR_YOUTUBE_AUDIO}/individual" '-f 140' "$uri"
 258 }
 259
 260 yt_video() {
 261     local -r uri="$1"
 262     _yt "${DIR_YOUTUBE_VIDEO}/individual" "$uri"
 263 }
 264
 265 gh_fetch_repos() {
 266     curl "https://api.github.com/$1/$2/repos?page=1&per_page=10000"
 267 }
 268
 269 gh_clone() {
 270     gh_user_type="$1"
 271     gh_user_name="$2"
 272     gh_dir="${DIR_GITHUB}/${gh_user_name}"
 273     mkdir -p "$gh_dir"
 274     cd "$gh_dir" || kill -INT $$
 275     gh_fetch_repos "$gh_user_type" "$gh_user_name" \
 276     | jq --raw-output '.[] | select(.fork | not) | .git_url' \
 277     | parallel -j 25 \
 278     git clone {}
 279 }
 280
 281 gh_clone_user() {
 282     gh_clone 'users' "$1"
 283 }
 284
 285 gh_clone_org() {
 286     gh_clone 'orgs' "$1"
 287 }
 288
 289 gh_clone_repo() {
 290     gh_username=$(echo "$1" | awk -F / '"$1 == "https" && $3 == github.com" {print $4}')
 291     gh_dir="${DIR_GITHUB}/${gh_username}"
 292     mkdir -p "$gh_dir"
 293     cd "$gh_dir" || kill -INT $$
 294     git clone "$1"
 295 }
 296
 297 work_log_template() {
 298 cat << EOF
 299 $(date '+%F %A')
 300 ==========
 301
 302 Morning report
 303 --------------
 304
 305 ### Prev
 306
 307 ### Curr
 308
 309 ### Next
 310
 311 ### Blockers
 312
 313 Day's notes
 314 -----------
 315 EOF
 316 }
 317
 318 work_log() {
 319     mkdir -p "$DIR_WORK_LOG"
 320     file_work_log_today="${DIR_WORK_LOG}/$(date +%F).md"
 321     if [ ! -f "$file_work_log_today" ]
 322     then
 323         work_log_template > "$file_work_log_today"
 324     fi
 325     vim -c 'set spell' "$file_work_log_today"
 326
 327 }
 328
 329 note() {
 330     mkdir -p "$DIR_NOTES"
 331     vim -c 'set spell' "$DIR_NOTES/$(date +'%Y_%m_%d--%H_%M_%S%z')--$1.md"
 332 }
 333
 334 weather() {
 335     local _weather_location
 336     case "$1" in
 337         '') _weather_location="$WEATHER_LOCATION";;
 338          *) _weather_location="$1"
 339     esac
 340     curl "http://wttr.in/$_weather_location?format=v2"
 341 }
 342
 343 bt_devs_paired() {
 344     bluetoothctl -- paired-devices \
 345     | awk '{print $2}' \
 346     | xargs bluetoothctl -- info
 347 }
 348
 349 bt_devs() {
 350     bluetoothctl -- devices \
 351     | awk '{print $2}' \
 352     | xargs bluetoothctl -- info
 353 }
 354
 355 run() {
 356     stderr="$(mktemp)"
 357     $@ 2> >(tee "$stderr")
 358     code="$?"
 359     urgency=''
 360     case "$code" in
 361         0) urgency='normal';;
 362         *) urgency='critical'
 363     esac
 364     notify-send -u "$urgency" "Job done: $code" "$(cat $stderr)"
 365     rm "$stderr"
 366 }
 367
 368 bar_gauge() {
 369     local -r width="$1"
 370
 371     awk -v width="$width" '
 372         {
 373             used  = $1
 374             total = $2
 375
 376             u = num_scale(used, total, 1, width)
 377
 378             printf "["
 379             for (i=1; i<=width; i++) {
 380                 c = i <= u ? "|" : " "
 381                 printf "%c", c
 382             }
 383             printf "]\n"
 384         }
 385
 386         function num_scale(src_cur, src_max, dst_min, dst_max) {
 387             return dst_min + ((src_cur * (dst_max - dst_min)) / src_max)
 388         }
 389     '
 390 }
 391
 392 motd() {
 393     uname -srvmo
 394     hostname | figlet
 395     uptime
 396
 397     echo
 398
 399     printf 'mem  '
 400     free \
 401     | awk '$1 == "Mem:" {total=$2; used=$3; print used, total}' \
 402     | bar_gauge 73
 403
 404     printf 'disk '
 405     df ~ \
 406     | awk 'NR == 2 {used=$3; avail=$4; total=used+avail; print used, total}' \
 407     | bar_gauge 73
 408
 409     case "$(uname)" in
 410         'Linux')
 411             printf 'batt '
 412             upower --dump \
 413             | awk '
 414                 /^Device:[ \t]+/ {
 415                     device["path"] = $2
 416                     next
 417                 }
 418
 419                 /  battery/ && device["path"] {
 420                     device["is_battery"] = 1
 421                     next
 422                 }
 423
 424                 /    percentage:/ && device["is_battery"] {
 425                     device["battery_percentage"] = $2
 426                     sub("%$", "", device["battery_percentage"])
 427                     next
 428                 }
 429
 430                 /^$/ {
 431                     if (device["is_battery"] && device["path"] == "/org/freedesktop/UPower/devices/DisplayDevice")
 432                         print device["battery_percentage"], 100
 433                     delete device
 434                 }
 435             ' \
 436             | bar_gauge 73
 437         ;;
 438     esac
 439
 440     echo
 441
 442     (ifconfig; iwconfig) 2> /dev/null \
 443     | awk '
 444         /^[^ ]/ {
 445             device = $1
 446             sub(":$", "", device)
 447             if ($4 ~ "ESSID:") {
 448                 _essid = $4
 449                 sub("^ESSID:\"", "", _essid)
 450                 sub("\"$", "", _essid)
 451                 essid[device] = _essid
 452             }
 453             next
 454         }
 455
 456         /^ / && $1 == "inet" {
 457             address[device] = $2
 458             next
 459         }
 460
 461         /^ +Link Quality=[0-9]+\/[0-9]+ +Signal level=/ {
 462             split($2, lq_parts_eq, "=")
 463             split(lq_parts_eq[2], lq_parts_slash, "/")
 464             cur = lq_parts_slash[1]
 465             max = lq_parts_slash[2]
 466             link[device] = cur / max * 100
 467             next
 468         }
 469
 470         END {
 471             for (device in address)
 472                 if (device != "lo") {
 473                     l = link[device]
 474                     e = essid[device]
 475                     l = l ? l : "--"
 476                     e = e ? e : "--"
 477                     print device, address[device], e, l
 478                 }
 479         }
 480         ' \
 481     | column -t
 482
 483     #echo
 484     # TODO: netstat summary
 485     # WARN: ensure: $USER ALL=(ALL) NOPASSWD:/bin/netstat
 486     #sudo -n netstat -tulpn | awk '/^udp/ && !first++ {printf "\n"} 1'
 487 }