home/lib/login_functions.sh

   1 #
   2
   3 ## open : string -> unit
   4 ##
   5 ## Fork xdg-open so we don't block current terminal session when opening
   6 ## things like pdf files. For example:
   7 ##
   8 ##    open book.pdf
   9 ##
  10 open() {
  11     (xdg-open "$1" &) &
  12 }
  13
  14 ## notify_done : unit -> unit
  15 notify_done() {
  16     local -r _status_code="$?"
  17     local -r _program="$1"
  18     local _timestamp
  19     _timestamp="$(timestamp)"
  20     local -r _msg="$_timestamp [$_program] done "
  21     if [[ "$_status_code" -eq 0 ]]
  22     then
  23         notify-send -u normal "$_msg OK: $_status_code"
  24     else
  25         notify-send -u critical "$_msg ERROR: $_status_code"
  26     fi
  27 }
  28
  29 ## p : string -> unit
  30 p() {
  31     awk \
  32         -v _s="$1" \
  33         '
  34             BEGIN {_s = tolower(_s)}
  35
  36             # TODO fzf/dmenu select instead of searching:
  37             /^[a-zA-Z]/ && tolower($1) ~ _s && NF >= 2 {
  38                 n++
  39                 s = $1
  40                 p = $NF
  41                 if (NF == 2) {
  42                     e = ""
  43                     u = ""
  44                 } else if (NF == 3) {
  45                     e = $2
  46                     u = ""
  47                 } else {
  48                     e = $2
  49                     u = $3
  50                 } # TODO What would NF > 4 mean?
  51
  52                 printf("%d [O] s:\"%s\", e:\"%s\", u:\"%s\"\n", n, s, e, u) > "/dev/stderr"
  53                 if (match(u, "@")) {
  54                     tmp = e
  55                     e = u
  56                     u = tmp
  57                     printf("%d [C] s:\"%s\", e:\"%s\", u:\"%s\"\n", n, s, e, u) > "/dev/stderr"
  58                 }
  59
  60                 printf "%s", p # XXX Intentionally avoiding newline in the result.
  61             }
  62
  63             END {
  64                 if (n == 1) {
  65                     exit 0
  66                 } else if (n == 0) {
  67                     printf "[ERROR] Found nothing.\n" > "/dev/stderr"
  68                     exit 1
  69                 } else if (n > 1) {
  70                     # TODO fzf-select which of the records the user (ahem, me) wants.
  71                     printf "[WARNING] Found more than one record.\n" > "/dev/stderr"
  72                     exit 0
  73                 }
  74             }
  75         ' \
  76         ~/._p/p \
  77         | xsel -i -b -t 30000
  78 }
  79
  80 ## web search
  81 ## ws : string -> unit
  82 ws() {
  83     local line search_string0 search_string
  84
  85     search_string0="$*"
  86     case "$search_string0" in
  87         '')
  88             while read -r line; do
  89                 search_string="${search_string} ${line}"
  90             done;;
  91          *)
  92             search_string="$search_string0";;
  93     esac
  94
  95     firefox --search "$search_string"
  96 }
  97
  98
  99 ## dictionary
 100 ## d : string -> string list
 101 d() {
 102     local -r word=$(fzf < /usr/share/dict/words)
 103     dict "$word"
 104 }
 105
 106 ## shell_activity_report : (mon | dow) -> string list
 107 shell_activity_report() {
 108     # TODO: optional concrete number output
 109     # TODO: optional combinations of granularities: hour, weekday, month, year
 110     local group_by="$1"
 111     case "$group_by" in
 112         'mon') ;;
 113         'dow') ;;
 114         '') group_by='dow';;
 115         *)
 116             echo "Usage: $0 [mon|dow]" >&2
 117             kill -INT $$
 118     esac
 119     history \
 120     | awk -v group_by="$group_by" '
 121         function date2dow(y, m, d,    _t, _i) {
 122             # Contract:
 123             #   y > 1752,  1 <= m <= 12.
 124             # Source:
 125             #   Sakamoto`s methods
 126             #   https://en.wikipedia.org/wiki/Determination_of_the_day_of_the_week#Sakamoto%27s_methods
 127             _t[ 0] = 0
 128             _t[ 1] = 3
 129             _t[ 2] = 2
 130             _t[ 3] = 5
 131             _t[ 4] = 0
 132             _t[ 5] = 3
 133             _t[ 6] = 5
 134             _t[ 7] = 1
 135             _t[ 8] = 4
 136             _t[ 9] = 6
 137             _t[10] = 2
 138             _t[11] = 4
 139             y -= m < 3
 140             _i = int(y + y/4 - y/100 + y/400 + _t[m - 1] + d) % 7
 141             _i = _i == 0 ? 7 : _i  # Make Sunday last
 142             return _i
 143
 144         }
 145
 146         {
 147             # NOTE: $2 & $3 are specific to oh-my-zsh history output
 148             date = $2
 149             time = $3
 150             d_fields = split(date, d, "-")
 151             t_fields = split(time, t, ":")
 152             if (t_fields && d_fields) {
 153                 # +0 to coerce number from string
 154                 year  = d[1] + 0
 155                 month = d[2] + 0
 156                 day   = d[3] + 0
 157                 hour = t[1] + 0
 158                 dow = date2dow(year, month, day)
 159                 g = group_by == "mon" ? month : dow  # dow is default
 160                 c = count[g, hour]++
 161             }
 162             if (c > max)
 163                 max = c
 164         }
 165
 166         END {
 167             w[1] = "Monday"
 168             w[2] = "Tuesday"
 169             w[3] = "Wednesday"
 170             w[4] = "Thursday"
 171             w[5] = "Friday"
 172             w[6] = "Saturday"
 173             w[7] = "Sunday"
 174
 175             m[ 1] = "January"
 176             m[ 2] = "February"
 177             m[ 3] = "March"
 178             m[ 4] = "April"
 179             m[ 5] = "May"
 180             m[ 6] = "June"
 181             m[ 7] = "July"
 182             m[ 8] = "August"
 183             m[ 9] = "September"
 184             m[10] = "October"
 185             m[11] = "November"
 186             m[12] = "December"
 187
 188             n = group_by == "mon" ? 12 : 7  # dow is default
 189
 190             for (gid = 1; gid <= n; gid++) {
 191                 group = group_by == "mon" ? m[gid] : w[gid]
 192                 printf "%s\n", group;
 193                 for (hour=0; hour<24; hour++) {
 194                     c = count[gid, hour]
 195                     printf "  %2d ", hour
 196                     for (i = 1; i <= (c * 100) / max; i++)
 197                         printf "|"
 198                     printf "\n"
 199                 }
 200             }
 201         }'
 202 }
 203
 204 ## top_commands : unit -> (command:string * count:number * bar:string) list
 205 top_commands() {
 206     history \
 207     | awk '
 208         {
 209             count[$4]++
 210         }
 211
 212         END {
 213             for (cmd in count)
 214                 print count[cmd], cmd
 215         }' \
 216     | sort -n -r -k 1 \
 217     | head -50 \
 218     | awk '
 219         {
 220             cmd[NR] = $2
 221             c = count[NR] = $1 + 0  # + 0 to coerce number from string
 222             if (c > max)
 223                 max = c
 224         }
 225
 226         END {
 227             for (i = 1; i <= NR; i++) {
 228                 c = count[i]
 229                 printf "%s %d ", cmd[i], c
 230                 scaled = (c * 100) / max
 231                 for (j = 1; j <= scaled; j++)
 232                     printf "|"
 233                 printf "\n"
 234             }
 235         }' \
 236     | column -t
 237 }
 238
 239 ## Top Disk-Using directories
 240 ## TODO: Consider using numfmt instead of awk
 241 ## tdu : path-string -> (size:number * directory:path-string) list
 242 tdu() {
 243     local -r root_path="$1"
 244
 245     du "$root_path" \
 246     | awk '
 247         {
 248             size = $1
 249             path = $0
 250             sub("^" $1 "\t+", "", path)
 251             paths[path] = size
 252             if (size > max)
 253                 max = size
 254         }
 255
 256         END {
 257             for (path in paths) {
 258                 size = paths[path]
 259                 pct = 100 * (size / max)
 260                 gb = size / 1024 / 1024
 261                 printf("%6.2f %3d%% %s\n", gb, pct, path)
 262             }
 263         }
 264     ' \
 265     | sort -r -n -k 1 \
 266     | head -50 \
 267     | tac
 268     # A slight optimization: head can exit before traversing the full input.
 269 }
 270
 271 ## Top Disk-Using Files
 272 ## tduf : path-string list -> (size:number * file:path-string) list
 273 tduf() {
 274     find "$@" -type f -printf '%s\t%p\0' \
 275     | sort -z -n -k 1 \
 276     | tail -z -n 50 \
 277     | gawk -v RS='\0' '
 278         {
 279             size = $1
 280             path = $0
 281             sub("^" $1 "\t+", "", path)
 282             gb = size / 1024 / 1024 / 1024
 283             printf("%f\t%s\n", gb, path)
 284         }'
 285 }
 286
 287 # Most-recently modified file system objects
 288 ## recent : ?(path-string list) -> path-string list
 289 recent() {
 290     # NOTES:
 291     # - %T+ is a GNU extension;
 292     # - gawk is able to split records on \0, while awk cannot.
 293     find "$@" -printf '%T@ %T+ %p\0' \
 294     | tee >(gawk -v RS='\0' 'END { printf("[INFO] Total found: %d\n", NR); }') \
 295     | sort -z -k 1 -n -r \
 296     | head -n "$(stty size | awk 'NR == 1 {print $1 - 5}')" -z \
 297     | gawk -v RS='\0' '
 298         {
 299             sub("^" $1 " +", "")  # Remove epoch time
 300             sub("+", " ")         # Blank-out the default separator
 301             sub("\\.[0-9]+", "")  # Remove fractional seconds
 302             print
 303         }'
 304 }
 305
 306 ## recent_dirs : ?(path-string list) -> path-string list
 307 recent_dirs() {
 308     recent "$@" -type d
 309 }
 310
 311 ## recent_files : ?(path-string list) -> path-string list
 312 recent_files() {
 313     recent "$@" -type f
 314 }
 315
 316 ## pa_def_sink : unit -> string
 317 pa_def_sink() {
 318     pactl info | awk '/^Default Sink:/ {print $3}'
 319 }
 320
 321 ## void_pkgs : ?(string) -> json
 322 void_pkgs() {
 323     curl "https://xq-api.voidlinux.org/v1/query/x86_64?q=$1" | jq '.data'
 324 }
 325
 326 ## Colorful man
 327 ## man : string -> string
 328 man() {
 329     # mb: begin blink
 330     # md: begin bold
 331     # me: end   bold, blink and underline
 332     #
 333     # so: begin standout (reverse video)
 334     # se: end   standout
 335     #
 336     # us: begin underline
 337     # ue: end   underline
 338
 339     LESS_TERMCAP_md=$'\e[01;30m' \
 340     LESS_TERMCAP_me=$'\e[0m' \
 341     LESS_TERMCAP_so=$'\e[01;44;33m' \
 342     LESS_TERMCAP_se=$'\e[0m' \
 343     LESS_TERMCAP_us=$'\e[01;33m' \
 344     LESS_TERMCAP_ue=$'\e[0m' \
 345     command man "$@"
 346 }
 347
 348 ## new experiment
 349 ## x : string list -> unit
 350 x() {
 351     cd "$(~/bin/x $@)" || kill -INT $$
 352 }
 353
 354 ## ocaml repl
 355 ## hump : unit -> unit
 356 hump() {
 357     ledit -l "$(stty size | awk '{print $2}')" ocaml $@
 358 }
 359
 360 ## search howtos
 361 ## howto : unit -> string
 362 howto() {
 363     cat "$(find  ~/arc/doc/HOWTOs -mindepth 1 -maxdepth 1 | sort | fzf)"
 364 }
 365
 366 _yt() {
 367     local -r base_dir="$1"
 368     local -r uri="$2"
 369     local -r opts="$3"
 370
 371     local -r yt=youtube-dl
 372     local -r id=$("$yt" --get-id "$uri")
 373     local -r title=$("$yt" --get-title "$uri" | sed 's/[^А-Яа-яA-Za-z0-9._-]/_/g')
 374     local -r dir="${base_dir}/${title}--${id}"
 375
 376     mkdir -p "$dir"
 377     cd "$dir" || kill -INT $$
 378     echo "$uri" > 'uri'
 379     "$yt" $opts -c --write-all-thumbnails --write-description --write-info-json "$uri"
 380 }
 381
 382 yt_audio() {
 383     local -r uri="$1"
 384     _yt "${DIR_YOUTUBE_AUDIO}/individual" "$uri" '-f 140'
 385 }
 386
 387 yt_video() {
 388     local -r uri="$1"
 389     _yt "${DIR_YOUTUBE_VIDEO}/individual" "$uri"
 390 }
 391
 392 gh_fetch_repos() {
 393     local -r user_type="$1"
 394     local -r user_name="$2"
 395
 396     curl "https://api.github.com/$user_type/$user_name/repos?page=1&per_page=10000"
 397 }
 398
 399 gh_clone() {
 400     local -r gh_user_type="$1"
 401     local -r gh_user_name="$2"
 402
 403     local -r gh_dir="${DIR_GITHUB}/${gh_user_name}"
 404     mkdir -p "$gh_dir"
 405     cd "$gh_dir" || kill -INT $$
 406     gh_fetch_repos "$gh_user_type" "$gh_user_name" \
 407     | jq --raw-output '.[] | select(.fork | not) | .clone_url' \
 408     | parallel -j 25 \
 409     git clone {}
 410 }
 411
 412 gh_clone_user() {
 413     gh_clone 'users' "$1"
 414 }
 415
 416 gh_clone_org() {
 417     gh_clone 'orgs' "$1"
 418 }
 419
 420 gh_clone_repo() {
 421     gh_username=$(echo "$1" | awk -F / '"$1 == "https" && $3 == github.com" {print $4}')
 422     gh_dir="${DIR_GITHUB}/${gh_username}"
 423     mkdir -p "$gh_dir"
 424     cd "$gh_dir" || kill -INT $$
 425     git clone "$1"
 426 }
 427
 428 bar() {
 429     local -r len="${1:-79}" # 1st arg or 79.
 430     local -r char="${2:--}" # 2nd arg or a dash.
 431     for _ in {1.."$len"}; do
 432         printf '%c' "$char";
 433     done
 434 }
 435
 436 daily_todo_file_template() {
 437 cat << EOF
 438 ===============================================================================
 439 $(date '+%F %A')
 440 ===============================================================================
 441
 442
 443 backlog
 444 -------------------------------------------------------------------------------
 445 EOF
 446 }
 447
 448 todo() {
 449     local date
 450     date="$(date +%F)"
 451     local -r dir="$DIR_TODO/daily"
 452     local -r file="$dir/$date.txt"
 453
 454     mkdir -p "$dir"
 455     if [ ! -f "$file" ]
 456     then
 457         daily_todo_file_template > "$file"
 458     fi
 459     cd "$DIR_TODO" && "$EDITOR" "$file"
 460 }
 461
 462 work_log_template() {
 463 cat << EOF
 464 $(date '+%F %A')
 465 ==========
 466
 467 Morning report
 468 --------------
 469
 470 ### Prev
 471
 472 ### Curr
 473
 474 ### Next
 475
 476 ### Blockers
 477
 478 Day's notes
 479 -----------
 480 EOF
 481 }
 482
 483 work_log() {
 484     mkdir -p "$DIR_WORK_LOG"
 485     local -r file_work_log_today="${DIR_WORK_LOG}/daily-$(date +%F).md"
 486     if [ ! -f "$file_work_log_today" ]
 487     then
 488         work_log_template > "$file_work_log_today"
 489     fi
 490     vim -c 'set spell' "$file_work_log_today"
 491
 492 }
 493
 494 note() {
 495     mkdir -p "$DIR_NOTES"
 496     vim -c 'set spell' "$DIR_NOTES/$(date +'%Y_%m_%d--%H_%M_%S%z')--$1.md"
 497 }
 498
 499 _bt_devs_infos() {
 500     # grep's defintion of a line does not include \r, wile awk's does and
 501     # which bluetoothctl outputs
 502     awk '/^Device +/ {print $2}' \
 503     | xargs -I% sh -c 'echo info % | bluetoothctl' \
 504     | awk '/^Device |^\t[A-Z][A-Za-z0-9]+: /'
 505 }
 506
 507 bt_devs_paired() {
 508     echo 'paired-devices' | bluetoothctl | _bt_devs_infos
 509 }
 510
 511 bt_devs() {
 512     echo 'devices' | bluetoothctl | _bt_devs_infos
 513 }
 514
 515 run() {
 516     local -r stderr="$(mktemp)"
 517
 518     local code urgency
 519
 520     $@ 2> >(tee "$stderr")
 521     code="$?"
 522     case "$code" in
 523         0) urgency='normal';;
 524         *) urgency='critical'
 525     esac
 526     notify-send -u "$urgency" "Job done: $code" "$(cat $stderr)"
 527     rm "$stderr"
 528 }
 529
 530 bar_gauge() {
 531     awk "$@" '
 532         BEGIN {
 533             # CLI options
 534             width    = width    ? width    : 80
 535             ch_left  = ch_left  ? ch_left  : "["
 536             ch_right = ch_right ? ch_right : "]"
 537             ch_blank = ch_blank ? ch_blank : "-"
 538             ch_used  = ch_used  ? ch_used  : "|"
 539             num      = num      ? 1        : 0
 540             pct      = pct      ? 1        : 0
 541         }
 542
 543         {
 544             cur = $1
 545             max = $2
 546             lab = $3
 547
 548             cur_scaled = num_scale(cur, max, 1, width)
 549
 550             printf \
 551                 "%s%s%s%s", \
 552                 lab ? lab         " " : "", \
 553                 num ? cur "/" max " " : "", \
 554                 pct ? sprintf("%3.0f%% ", cur / max * 100) : "", \
 555                 ch_left
 556             for (i=1; i<=width; i++) {
 557                 c = i <= cur_scaled ? ch_used : ch_blank
 558                 printf "%s", c
 559             }
 560             printf "%s\n", ch_right
 561         }
 562
 563         function num_scale(src_cur, src_max, dst_min, dst_max) {
 564             return dst_min + ((src_cur * (dst_max - dst_min)) / src_max)
 565         }
 566     '
 567 }
 568
 569 flat_top_5() {
 570     sort -n -k 1 -r \
 571     | head -5 \
 572     | awk '
 573         {
 574             cur  = $1
 575             max  = $2
 576             name = $3
 577             pct  = cur / max * 100
 578             printf "%s%s %.2f%%", sep, name, pct
 579             sep = ",  "
 580         }
 581
 582         END {printf "\n"}
 583         '
 584 }
 585
 586 internet_addr() {
 587     curl --silent --show-error --max-time "${1:=1}" 'https://api.ipify.org' 2>&1
 588 }
 589
 590 status_batt() {
 591     case "$(uname)" in
 592         'Linux')
 593             if which upower > /dev/null
 594             then
 595                 upower --dump \
 596                 | awk '
 597                     /^Device:[ \t]+/ {
 598                         device["path"] = $2
 599                         next
 600                     }
 601
 602                     /  battery/ && device["path"] {
 603                         device["is_battery"] = 1
 604                         next
 605                     }
 606
 607                     /    percentage:/ && device["is_battery"] {
 608                         device["battery_percentage"] = $2
 609                         sub("%$", "", device["battery_percentage"])
 610                         next
 611                     }
 612
 613                     /^$/ {
 614                         if (device["is_battery"] && device["path"] == "/org/freedesktop/UPower/devices/DisplayDevice")
 615                             print device["battery_percentage"], 100, "batt"
 616                         delete device
 617                     }
 618                 '
 619             fi
 620         ;;
 621     esac
 622 }
 623
 624 indent() {
 625     awk -v unit="$1" '{printf "%s%s\n", unit, $0}'
 626 }
 627
 628 status() {
 629     local -r indent_unit='    '
 630
 631     uname -srvmo
 632     hostname | figlet
 633     uptime
 634
 635     echo
 636
 637     echo 'accounting'
 638
 639     printf '%stmux\n%ssessions %d, clients %d\n' \
 640         "$indent_unit" \
 641         "${indent_unit}${indent_unit}" \
 642         "$(tmux list-sessions 2> /dev/null | wc -l)" \
 643         "$(tmux list-clients  2> /dev/null | wc -l)"
 644
 645     echo
 646
 647     printf '%sprocs by user\n' "${indent_unit}"
 648     ps -eo user \
 649     | awk '
 650         NR > 1 {
 651             count_by_user[$1]++
 652             total++
 653         }
 654
 655         END {
 656             for (user in count_by_user)
 657                 print count_by_user[user], total, user
 658         }
 659         ' \
 660     | flat_top_5 \
 661     | indent "${indent_unit}${indent_unit}"
 662
 663     echo
 664
 665     echo 'resources'
 666     (
 667         free | awk '$1 == "Mem:" {print $3, $2, "mem"}'
 668         df ~ | awk 'NR == 2 {print $3, $3 + $4, "disk"}'
 669         status_batt
 670     ) \
 671     | bar_gauge -v width=60 -v pct=1 \
 672     | column -t \
 673     | indent "$indent_unit"
 674
 675     echo
 676
 677     printf '%smem by proc\n' "$indent_unit"
 678     ps -eo rss,comm \
 679     | awk -v total="$(free | awk '$1 == "Mem:" {print $2; exit}')" '
 680         NR > 1 {
 681             rss = $1
 682             proc = $2
 683             by_proc[proc] += rss
 684         }
 685
 686         END {
 687             for (proc in by_proc)
 688                 print by_proc[proc], total, proc
 689             }
 690         ' \
 691         | flat_top_5 \
 692         | indent "${indent_unit}${indent_unit}"
 693
 694     echo
 695
 696     local _dir temp_input label_file label
 697
 698     printf '%sthermal\n' "$indent_unit"
 699     for _dir in /sys/class/hwmon/hwmon*; do
 700         cat "$_dir"/name
 701         find "$_dir"/ -name 'temp*_input' \
 702             | while read -r temp_input; do
 703                 label_file=${temp_input//_input/_label}
 704                 if [ -f "$label_file" ]; then
 705                     label=$(< "$label_file")
 706                 else
 707                     label=''
 708                 fi
 709                 awk -v label="$label" '{
 710                         if (label)
 711                             label = sprintf(" (%s)", label)
 712                             printf("%.2f°C%s\n", $1 / 1000, label)
 713                     }' \
 714                     "$temp_input"
 715             done \
 716             | sort \
 717             | indent "$indent_unit"
 718     done \
 719     | indent "${indent_unit}${indent_unit}"
 720
 721     echo 'net'
 722     #local -r internet_addr=$(internet_addr 0.5)
 723     #local -r internet_ptr=$(host -W 1 "$internet_addr" | awk 'NR == 1 {print $NF}' )
 724
 725     #echo "${indent_unit}internet"
 726     #echo "${indent_unit}${indent_unit}$internet_addr  $internet_ptr"
 727     echo "${indent_unit}if"
 728     (ifconfig; iwconfig) 2> /dev/null \
 729     | awk '
 730         /^[^ ]/ {
 731             device = $1
 732             sub(":$", "", device)
 733             if ($4 ~ "ESSID:") {
 734                 _essid = $4
 735                 sub("^ESSID:\"", "", _essid)
 736                 sub("\"$", "", _essid)
 737                 essid[device] = _essid
 738             }
 739             next
 740         }
 741
 742         /^ / && $1 == "inet" {
 743             address[device] = $2
 744             next
 745         }
 746
 747         /^ +Link Quality=[0-9]+\/[0-9]+ +Signal level=/ {
 748             split($2, lq_parts_eq, "=")
 749             split(lq_parts_eq[2], lq_parts_slash, "/")
 750             cur = lq_parts_slash[1]
 751             max = lq_parts_slash[2]
 752             link[device] = cur / max * 100
 753             next
 754         }
 755
 756         END {
 757             for (device in address)
 758                 if (device != "lo") {
 759                     l = link[device]
 760                     e = essid[device]
 761                     l = l ? sprintf("%.0f%%", l) : "--"
 762                     e = e ? e : "--"
 763                     print device, address[device], e, l
 764                 }
 765         }
 766         ' \
 767     | column -t \
 768     | indent "${indent_unit}${indent_unit}"
 769
 770     # WARN: ensure: $USER ALL=(ALL) NOPASSWD:/bin/netstat
 771
 772     echo "${indent_unit}-->"
 773
 774     sudo -n netstat -tulnp \
 775     | awk -v indent="${indent_unit}${indent_unit}" '
 776         NR > 2 && ((/^tcp/ && proc = $7) || (/^udp/ && proc = $6)) {
 777             protocol = $1
 778             addr = $4
 779             port = a[split(addr, a, ":")]
 780             name = p[split(proc, p, "/")]
 781             names[name] = 1
 782             protocols[protocol] = 1
 783             if (!seen[protocol, name, port]++)
 784                 ports[protocol, name, ++seen[protocol, name]] = port
 785         }
 786
 787         END {
 788             for (protocol in protocols) {
 789                 printf "%s%s\t", indent, toupper(protocol)
 790                 for (name in names) {
 791                     if (n = seen[protocol, name]) {
 792                         sep = ""
 793                         printf "%s:", name
 794                         for (i = 1; i <= n; i++) {
 795                             printf "%s%d", sep, ports[protocol, name, i]
 796                             sep = ","
 797                         }
 798                         printf "  "
 799                     }
 800                 }
 801                 printf "\n"
 802             }
 803         }'
 804
 805     echo "${indent_unit}<->"
 806
 807     printf '%sTCP\t' "${indent_unit}${indent_unit}"
 808     sudo -n netstat -tnp \
 809     | awk 'NR > 2 && $6 == "ESTABLISHED" {print $7}' \
 810     | awk '{sub("^[0-9]+/", ""); print}' \
 811     | sort -u \
 812     | xargs \
 813     | column -t
 814
 815     # TODO: iptables summary
 816 }
 817
 818 ssh_invalid_by_addr() {
 819     awk '
 820         /: Invalid user/ && $5 ~ /^sshd/ {
 821             addr=$10 == "port" ? $9 : $10
 822             max++
 823             by_addr[addr]++
 824         }
 825
 826         END {
 827             for (addr in by_addr)
 828                 if ((c = by_addr[addr]) > 1)
 829                     printf "%d %d %s\n", c, max, addr
 830         }
 831         ' \
 832         /var/log/auth.log \
 833         /var/log/auth.log.1 \
 834     | sort -n -k 1 \
 835     | bar_gauge -v width="$(stty size | awk '{print $2}')" -v num=1 -v ch_right=' ' -v ch_left=' ' -v ch_blank=' ' \
 836     | column -t
 837 }
 838
 839 ssh_invalid_by_day() {
 840     awk '
 841         BEGIN {
 842             m["Jan"] = "01"
 843             m["Feb"] = "02"
 844             m["Mar"] = "03"
 845             m["Apr"] = "04"
 846             m["May"] = "05"
 847             m["Jun"] = "06"
 848             m["Jul"] = "07"
 849             m["Aug"] = "08"
 850             m["Sep"] = "09"
 851             m["Oct"] = "10"
 852             m["Nov"] = "11"
 853             m["Dec"] = "12"
 854         }
 855
 856         /: Invalid user/ && $5 ~ /^sshd/ {
 857             day = m[$1] "-" $2
 858             max++
 859             by_day[day]++
 860         }
 861
 862         END {
 863             for (day in by_day)
 864                 if ((c = by_day[day]) > 1)
 865                     printf "%d %d %s\n", c, max, day
 866         }
 867         ' \
 868         /var/log/auth.log \
 869         /var/log/auth.log.1 \
 870     | sort -k 3 \
 871     | bar_gauge -v width="$(stty size | awk '{print $2}')" -v num=1 -v ch_right=' ' -v ch_left=' ' -v ch_blank=' ' \
 872     | column -t
 873 }
 874
 875 ssh_invalid_by_user() {
 876     awk '
 877         /: Invalid user/ && $5 ~ /^sshd/ {
 878             user=$8
 879             max++
 880             by_user[user]++
 881         }
 882
 883         END {
 884             for (user in by_user)
 885                 if ((c = by_user[user]) > 1)
 886                     printf "%d %d %s\n", c, max, user
 887         }
 888         ' \
 889         /var/log/auth.log \
 890         /var/log/auth.log.1 \
 891     | sort -n -k 1 \
 892     | bar_gauge -v width="$(stty size | awk '{print $2}')" -v num=1 -v ch_right=' ' -v ch_left=' ' -v ch_blank=' ' \
 893     | column -t
 894 }
 895
 896 loggers() {
 897     awk '
 898         {
 899             split($5, prog, "[")
 900             sub(":$", "", prog[1]) # if there were no [], than : will is left behind
 901             print prog[1]
 902         }' /var/log/syslog /var/log/syslog.1 \
 903     | awk '
 904         {
 905             n = split($1, path, "/")  # prog may be in path form
 906             prog = path[n]
 907             total++
 908             count[prog]++
 909         }
 910
 911         END {
 912             for (prog in count)
 913                 print count[prog], total, prog
 914         }' \
 915     | sort -n -k 1 \
 916     | bar_gauge -v num=1 -v ch_right=' ' -v ch_left=' ' -v ch_blank=' ' \
 917     | column -t
 918 }