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 }
 154
 155 # Top Disk-Using Files
 156 tduf() {
 157     find "$1" -type f -printf '%s\t%p\0' \
 158     | sort -z -n -k 1 \
 159     | tail -z -n 50 \
 160     | gawk -v RS='\0' '
 161         {
 162             size = $1
 163             path = $0
 164             sub("^" $1 "\t+", "", path)
 165             gb = size / 1024 / 1024 / 1024
 166             printf("%f\t%s\n", gb, path)
 167         }'
 168 }
 169
 170 # Most-recently modified file system objects
 171 recent() {
 172     # NOTES:
 173     # - intentionally not quoting the parameters, so that some can be ignored
 174     #   if not passed, rather than be passed to find as an empty string;
 175     # - %T+ is a GNU extension;
 176     # - gawk is able to split records on \0, while awk cannot.
 177     find $@ -printf '%T@ %T+ %p\0' \
 178     | tee >(gawk -v RS='\0' 'END { printf("[INFO] Total found: %d\n", NR); }') \
 179     | sort -z -k 1 -n -r \
 180     | head -n "$(stty size | awk 'NR == 1 {print $1 - 5}')" -z \
 181     | gawk -v RS='\0' '
 182         {
 183             sub("^" $1 " +", "")  # Remove epoch time
 184             sub("+", " ")         # Blank-out the default separator
 185             sub("\\.[0-9]+", "")  # Remove fractional seconds
 186             print
 187         }'
 188 }
 189
 190 recent_dirs() {
 191     recent "$1" -type d
 192 }
 193
 194 recent_files() {
 195     recent "$1" -type f
 196 }
 197
 198 pa_def_sink() {
 199     pactl info | awk '/^Default Sink:/ {print $3}'
 200 }
 201
 202 void_pkgs() {
 203     curl "https://xq-api.voidlinux.org/v1/query/x86_64?q=$1" | jq '.data'
 204 }
 205
 206 # Colorful man
 207 man() {
 208     # mb: begin blink
 209     # md: begin bold
 210     # me: end   bold, blink and underline
 211     #
 212     # so: begin standout (reverse video)
 213     # se: end   standout
 214     #
 215     # us: begin underline
 216     # ue: end   underline
 217
 218     LESS_TERMCAP_md=$'\e[01;30m' \
 219     LESS_TERMCAP_me=$'\e[0m' \
 220     LESS_TERMCAP_so=$'\e[01;44;33m' \
 221     LESS_TERMCAP_se=$'\e[0m' \
 222     LESS_TERMCAP_us=$'\e[01;33m' \
 223     LESS_TERMCAP_ue=$'\e[0m' \
 224     command man "$@"
 225 }
 226
 227 # new experiment
 228 x() {
 229     cd "$(~/bin/x $@)" || kill -INT $$
 230 }
 231
 232 hump() {
 233     ledit -l "$(stty size | awk '{print $2}')" ocaml $@
 234 }
 235
 236 howto() {
 237     cat "$(find  ~/arc/doc/HOWTOs -mindepth 1 -maxdepth 1 | sort | fzf)"
 238 }
 239
 240 _yt() {
 241     local -r base_dir="$1"
 242     local -r uri="$2"
 243     local -r opts="$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" "$uri" '-f 140'
 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     local -r gh_user_type="$1"
 271     local -r gh_user_name="$2"
 272
 273     local -r gh_dir="${DIR_GITHUB}/${gh_user_name}"
 274     mkdir -p "$gh_dir"
 275     cd "$gh_dir" || kill -INT $$
 276     gh_fetch_repos "$gh_user_type" "$gh_user_name" \
 277     | jq --raw-output '.[] | select(.fork | not) | .git_url' \
 278     | parallel -j 25 \
 279     git clone {}
 280 }
 281
 282 gh_clone_user() {
 283     gh_clone 'users' "$1"
 284 }
 285
 286 gh_clone_org() {
 287     gh_clone 'orgs' "$1"
 288 }
 289
 290 gh_clone_repo() {
 291     gh_username=$(echo "$1" | awk -F / '"$1 == "https" && $3 == github.com" {print $4}')
 292     gh_dir="${DIR_GITHUB}/${gh_username}"
 293     mkdir -p "$gh_dir"
 294     cd "$gh_dir" || kill -INT $$
 295     git clone "$1"
 296 }
 297
 298 work_log_template() {
 299 cat << EOF
 300 $(date '+%F %A')
 301 ==========
 302
 303 Morning report
 304 --------------
 305
 306 ### Prev
 307
 308 ### Curr
 309
 310 ### Next
 311
 312 ### Blockers
 313
 314 Day's notes
 315 -----------
 316 EOF
 317 }
 318
 319 work_log() {
 320     mkdir -p "$DIR_WORK_LOG"
 321     local -r file_work_log_today="${DIR_WORK_LOG}/$(date +%F).md"
 322     if [ ! -f "$file_work_log_today" ]
 323     then
 324         work_log_template > "$file_work_log_today"
 325     fi
 326     vim -c 'set spell' "$file_work_log_today"
 327
 328 }
 329
 330 note() {
 331     mkdir -p "$DIR_NOTES"
 332     vim -c 'set spell' "$DIR_NOTES/$(date +'%Y_%m_%d--%H_%M_%S%z')--$1.md"
 333 }
 334
 335 weather() {
 336     local _weather_location
 337     case "$1" in
 338         '') _weather_location="$WEATHER_LOCATION";;
 339          *) _weather_location="$1"
 340     esac
 341     curl "http://wttr.in/$_weather_location?format=v2"
 342 }
 343
 344 _bt_devs_infos() {
 345     # grep's defintion of a line does not include \r, wile awk's does and
 346     # which bluetoothctl outputs
 347     awk '/^Device +/ {print $2}' \
 348     | xargs -I% sh -c 'echo info % | bluetoothctl' \
 349     | awk '/^Device |^\t[A-Z][A-Za-z0-9]+: /'
 350 }
 351
 352 bt_devs_paired() {
 353     echo 'paired-devices' | bluetoothctl | _bt_devs_infos
 354 }
 355
 356 bt_devs() {
 357     echo 'devices' | bluetoothctl | _bt_devs_infos
 358 }
 359
 360 run() {
 361     local -r stderr="$(mktemp)"
 362
 363     local code urgency
 364
 365     $@ 2> >(tee "$stderr")
 366     code="$?"
 367     case "$code" in
 368         0) urgency='normal';;
 369         *) urgency='critical'
 370     esac
 371     notify-send -u "$urgency" "Job done: $code" "$(cat $stderr)"
 372     rm "$stderr"
 373 }
 374
 375 bar_gauge() {
 376     awk "$@" '
 377         BEGIN {
 378             # CLI options
 379             width    = width    ? width    : 80
 380             ch_left  = ch_left  ? ch_left  : "["
 381             ch_right = ch_right ? ch_right : "]"
 382             ch_blank = ch_blank ? ch_blank : "-"
 383             ch_used  = ch_used  ? ch_used  : "|"
 384             num      = num      ? 1        : 0
 385             pct      = pct      ? 1        : 0
 386         }
 387
 388         {
 389             cur = $1
 390             max = $2
 391             lab = $3
 392
 393             cur_scaled = num_scale(cur, max, 1, width)
 394
 395             printf \
 396                 "%s%s%s%s", \
 397                 lab ? lab         " " : "", \
 398                 num ? cur "/" max " " : "", \
 399                 pct ? sprintf("%3.0f%% ", cur / max * 100) : "", \
 400                 ch_left
 401             for (i=1; i<=width; i++) {
 402                 c = i <= cur_scaled ? ch_used : ch_blank
 403                 printf "%s", c
 404             }
 405             printf "%s\n", ch_right
 406         }
 407
 408         function num_scale(src_cur, src_max, dst_min, dst_max) {
 409             return dst_min + ((src_cur * (dst_max - dst_min)) / src_max)
 410         }
 411     '
 412 }
 413
 414 flat_top_5() {
 415     sort -n -k 1 -r \
 416     | head -5 \
 417     | awk '
 418         {
 419             cur  = $1
 420             max  = $2
 421             name = $3
 422             pct  = cur / max * 100
 423             printf "%s%s %.2f%%", sep, name, pct
 424             sep = ",  "
 425         }
 426
 427         END {printf "\n"}
 428         '
 429 }
 430
 431 internet_addr() {
 432     curl --silent --show-error --max-time "${1:=1}" 'https://api.ipify.org' 2>&1
 433 }
 434
 435 status_batt() {
 436     case "$(uname)" in
 437         'Linux')
 438             if which upower > /dev/null
 439             then
 440                 upower --dump \
 441                 | awk '
 442                     /^Device:[ \t]+/ {
 443                         device["path"] = $2
 444                         next
 445                     }
 446
 447                     /  battery/ && device["path"] {
 448                         device["is_battery"] = 1
 449                         next
 450                     }
 451
 452                     /    percentage:/ && device["is_battery"] {
 453                         device["battery_percentage"] = $2
 454                         sub("%$", "", device["battery_percentage"])
 455                         next
 456                     }
 457
 458                     /^$/ {
 459                         if (device["is_battery"] && device["path"] == "/org/freedesktop/UPower/devices/DisplayDevice")
 460                             print device["battery_percentage"], 100, "batt"
 461                         delete device
 462                     }
 463                 '
 464             fi
 465         ;;
 466     esac
 467 }
 468
 469 indent() {
 470     awk -v unit="$1" '{printf "%s%s\n", unit, $0}'
 471 }
 472
 473 status() {
 474     local -r indent_unit='    '
 475
 476     uname -srvmo
 477     hostname | figlet
 478     uptime
 479
 480     echo
 481
 482     echo 'accounting'
 483
 484     printf '%stmux\n%ssessions %d, clients %d\n' \
 485         "$indent_unit" \
 486         "${indent_unit}${indent_unit}" \
 487         "$(tmux list-sessions 2> /dev/null | wc -l)" \
 488         "$(tmux list-clients  2> /dev/null | wc -l)"
 489
 490     echo
 491
 492     printf '%sprocs by user\n' "${indent_unit}"
 493     ps -eo user \
 494     | awk '
 495         NR > 1 {
 496             count_by_user[$1]++
 497             total++
 498         }
 499
 500         END {
 501             for (user in count_by_user)
 502                 print count_by_user[user], total, user
 503         }
 504         ' \
 505     | flat_top_5 \
 506     | indent "${indent_unit}${indent_unit}"
 507
 508     echo
 509
 510     echo 'resources'
 511     (
 512         free | awk '$1 == "Mem:" {print $3, $2, "mem"}'
 513         df ~ | awk 'NR == 2 {print $3, $3 + $4, "disk"}'
 514         status_batt
 515     ) \
 516     | bar_gauge -v width=60 -v pct=1 \
 517     | column -t \
 518     | indent "$indent_unit"
 519
 520     echo
 521
 522     printf '%smem by proc\n' "$indent_unit"
 523     ps -eo rss,cmd \
 524     | awk -v total="$(free | awk '$1 == "Mem:" {print $2; exit}')" '
 525         NR > 1 {
 526             rss = $1
 527             cmd = $2
 528             n = split(cmd, path, "/")  # _may_ be a path
 529             proc = path[n]
 530             by_proc[proc] += rss
 531         }
 532
 533         END {
 534             for (proc in by_proc)
 535                 print by_proc[proc], total, proc
 536             }
 537         ' \
 538         | flat_top_5 \
 539         | indent "${indent_unit}${indent_unit}"
 540
 541     echo
 542
 543     printf '%sthermal\n' "$indent_unit"
 544     for _dir in /sys/class/thermal/thermal_zone*
 545     do
 546         printf '%s %.2f C\n' \
 547             $(cat "$_dir"/type) \
 548             $(( $(cat "$_dir"/temp) / 1000 ))
 549     done \
 550     | column -t \
 551     | indent "${indent_unit}${indent_unit}"
 552
 553     echo 'net'
 554     local -r internet_addr=$(internet_addr 0.5)
 555     local -r internet_ptr=$(host -W 1 "$internet_addr" | awk 'NR == 1 {print $NF}' )
 556
 557     echo "${indent_unit}internet"
 558     echo "${indent_unit}${indent_unit}$internet_addr  $internet_ptr"
 559     echo "${indent_unit}if"
 560     (ifconfig; iwconfig) 2> /dev/null \
 561     | awk '
 562         /^[^ ]/ {
 563             device = $1
 564             sub(":$", "", device)
 565             if ($4 ~ "ESSID:") {
 566                 _essid = $4
 567                 sub("^ESSID:\"", "", _essid)
 568                 sub("\"$", "", _essid)
 569                 essid[device] = _essid
 570             }
 571             next
 572         }
 573
 574         /^ / && $1 == "inet" {
 575             address[device] = $2
 576             next
 577         }
 578
 579         /^ +Link Quality=[0-9]+\/[0-9]+ +Signal level=/ {
 580             split($2, lq_parts_eq, "=")
 581             split(lq_parts_eq[2], lq_parts_slash, "/")
 582             cur = lq_parts_slash[1]
 583             max = lq_parts_slash[2]
 584             link[device] = cur / max * 100
 585             next
 586         }
 587
 588         END {
 589             for (device in address)
 590                 if (device != "lo") {
 591                     l = link[device]
 592                     e = essid[device]
 593                     l = l ? sprintf("%.0f%%", l) : "--"
 594                     e = e ? e : "--"
 595                     print device, address[device], e, l
 596                 }
 597         }
 598         ' \
 599     | column -t \
 600     | indent "${indent_unit}${indent_unit}"
 601
 602     # WARN: ensure: $USER ALL=(ALL) NOPASSWD:/bin/netstat
 603
 604     echo "${indent_unit}-->"
 605
 606     sudo -n netstat -tulnp \
 607     | awk -v indent="${indent_unit}${indent_unit}" '
 608         NR > 2 && ((/^tcp/ && proc = $7) || (/^udp/ && proc = $6)) {
 609             protocol = $1
 610             addr = $4
 611             port = a[split(addr, a, ":")]
 612             name = p[split(proc, p, "/")]
 613             names[name] = 1
 614             protocols[protocol] = 1
 615             if (!seen[protocol, name, port]++)
 616                 ports[protocol, name, ++seen[protocol, name]] = port
 617         }
 618
 619         END {
 620             for (protocol in protocols) {
 621                 printf "%s%s\t", indent, toupper(protocol)
 622                 for (name in names) {
 623                     if (n = seen[protocol, name]) {
 624                         sep = ""
 625                         printf "%s:", name
 626                         for (i = 1; i <= n; i++) {
 627                             printf "%s%d", sep, ports[protocol, name, i]
 628                             sep = ","
 629                         }
 630                         printf "  "
 631                     }
 632                 }
 633                 printf "\n"
 634             }
 635         }'
 636
 637     echo "${indent_unit}<->"
 638
 639     printf '%sTCP: ' "${indent_unit}${indent_unit}"
 640     sudo -n netstat -tnp \
 641     | awk 'NR > 2 && $6 == "ESTABLISHED" {print $7}' \
 642     | awk -F/ '{print $2}' \
 643     | sort -u \
 644     | xargs \
 645     | column -t
 646
 647     # TODO: iptables summary
 648 }
 649
 650 ssh_invalid_attempts_from() {
 651     awk '
 652         /: Invalid user/ && $5 ~ /^sshd/ {
 653             u=$8
 654             addr=$10 == "port" ? $9 : $10
 655             max++
 656             curr[addr]++
 657         }
 658
 659         END {
 660             for (addr in curr)
 661                 if ((c = curr[addr]) > 1)
 662                     print c, max, addr
 663         }
 664         ' \
 665         /var/log/auth.log \
 666         /var/log/auth.log.1 \
 667     | sort -n -k 1 \
 668     | bar_gauge -v width="$(stty size | awk '{print $2}')" -v num=1 -v ch_right=' ' -v ch_left=' ' -v ch_blank=' ' \
 669     | column -t
 670 }
 671
 672 loggers() {
 673     awk '
 674         {
 675             split($5, prog, "[")
 676             sub(":$", "", prog[1]) # if there were no [], than : will is left behind
 677             print prog[1]
 678         }' /var/log/syslog /var/log/syslog.1 \
 679     | awk '
 680         {
 681             n = split($1, path, "/")  # prog may be in path form
 682             prog = path[n]
 683             total++
 684             count[prog]++
 685         }
 686
 687         END {
 688             for (prog in count)
 689                 print count[prog], total, prog
 690         }' \
 691     | sort -n -k 1 \
 692     | bar_gauge -v num=1 -v ch_right=' ' -v ch_left=' ' -v ch_blank=' ' \
 693     | column -t
 694 }