Add percentage of max to tdu
[khome.git] / home / lib / login_functions.sh
index 7c569dc..8e25bc0 100644 (file)
+#
+
+d() {
+    local -r word=$(fzf < /usr/share/dict/words)
+    dict "$word"
+}
+
+shell_activity_report() {
+    # TODO: optional concrete number output
+    # TODO: optional combinations of granularities: hour, weekday, month, year
+    local group_by="$1"
+    case "$group_by" in
+        'mon') ;;
+        'dow') ;;
+        '') group_by='dow';;
+        *)
+            echo "Usage: $0 [mon|dow]" >&2
+            kill -INT $$
+    esac
+    history \
+    | awk -v group_by="$group_by" '
+        function date2dow(y, m, d,    _t, _i) {
+            # Contract:
+            #   y > 1752,  1 <= m <= 12.
+            # Source:
+            #   Sakamoto`s methods
+            #   https://en.wikipedia.org/wiki/Determination_of_the_day_of_the_week#Sakamoto%27s_methods
+            _t[ 0] = 0
+            _t[ 1] = 3
+            _t[ 2] = 2
+            _t[ 3] = 5
+            _t[ 4] = 0
+            _t[ 5] = 3
+            _t[ 6] = 5
+            _t[ 7] = 1
+            _t[ 8] = 4
+            _t[ 9] = 6
+            _t[10] = 2
+            _t[11] = 4
+            y -= m < 3
+            _i = int(y + y/4 - y/100 + y/400 + _t[m - 1] + d) % 7
+            _i = _i == 0 ? 7 : _i  # Make Sunday last
+            return _i
+
+        }
+
+        {
+            # NOTE: $2 & $3 are specific to oh-my-zsh history output
+            date = $2
+            time = $3
+            d_fields = split(date, d, "-")
+            t_fields = split(time, t, ":")
+            if (t_fields && d_fields) {
+                # +0 to coerce number from string
+                year  = d[1] + 0
+                month = d[2] + 0
+                day   = d[3] + 0
+                hour = t[1] + 0
+                dow = date2dow(year, month, day)
+                g = group_by == "mon" ? month : dow  # dow is default
+                c = count[g, hour]++
+            }
+            if (c > max)
+                max = c
+        }
+
+        END {
+            w[1] = "Monday"
+            w[2] = "Tuesday"
+            w[3] = "Wednesday"
+            w[4] = "Thursday"
+            w[5] = "Friday"
+            w[6] = "Saturday"
+            w[7] = "Sunday"
+
+            m[ 1] = "January"
+            m[ 2] = "February"
+            m[ 3] = "March"
+            m[ 4] = "April"
+            m[ 5] = "May"
+            m[ 6] = "June"
+            m[ 7] = "July"
+            m[ 8] = "August"
+            m[ 9] = "September"
+            m[10] = "October"
+            m[11] = "November"
+            m[12] = "December"
+
+            n = group_by == "mon" ? 12 : 7  # dow is default
+
+            for (gid = 1; gid <= n; gid++) {
+                group = group_by == "mon" ? m[gid] : w[gid]
+                printf "%s\n", group;
+                for (hour=0; hour<24; hour++) {
+                    c = count[gid, hour]
+                    printf "  %2d ", hour
+                    for (i = 1; i <= (c * 100) / max; i++)
+                        printf "|"
+                    printf "\n"
+                }
+            }
+        }'
+}
+
+top_commands() {
+    history \
+    | awk '
+        {
+            count[$4]++
+        }
+
+        END {
+            for (cmd in count)
+                print count[cmd], cmd
+        }' \
+    | sort -n -r -k 1 \
+    | head -50 \
+    | awk '
+        {
+            cmd[NR] = $2
+            c = count[NR] = $1 + 0  # + 0 to coerce number from string
+            if (c > max)
+                max = c
+        }
+
+        END {
+            for (i = 1; i <= NR; i++) {
+                c = count[i]
+                printf "%s %d ", cmd[i], c
+                scaled = (c * 100) / max
+                for (j = 1; j <= scaled; j++)
+                    printf "|"
+                printf "\n"
+            }
+        }' \
+    | column -t
+}
+
 # Top Disk-Using directories
 # TODO: Consider using numfmt instead of awk
 tdu() {
     du "$1" \
-    | sort -n -k 1 -r --parallel="$(nproc)" \
-    | head -50 \
     | awk '
         {
             size = $1
             path = $0
             sub("^" $1 "\t+", "", path)
-            gb = size / 1024 / 1024
-            printf("%f\t%s\n", gb, path)
-        }' \
-    | cut -c 1-115
+            paths[path] = size
+            if (size > max)
+                max = size
+        }
+
+        END {
+            for (path in paths) {
+                size = paths[path]
+                pct = 100 * (size / max)
+                gb = size / 1024 / 1024
+                printf("%6.2f %3d%% %s\n", gb, pct, path)
+            }
+        }
+    ' \
+    | sort -r -n -k 1 \
+    | head -50 \
+    | tac
+    # A slight optimization: head can exit before traversing the full input.
 }
 
 # Top Disk-Using Files
 tduf() {
     find "$1" -type f -printf '%s\t%p\0' \
-    | sort -z -n -k 1 -r \
-    | head -z -n 50 \
+    | sort -z -n -k 1 \
+    | tail -z -n 50 \
     | gawk -v RS='\0' '
         {
             size = $1
@@ -68,17 +218,28 @@ void_pkgs() {
 
 # Colorful man
 man() {
-    LESS_TERMCAP_md=$'\e[01;31m' \
+    # mb: begin blink
+    # md: begin bold
+    # me: end   bold, blink and underline
+    #
+    # so: begin standout (reverse video)
+    # se: end   standout
+    #
+    # us: begin underline
+    # ue: end   underline
+
+    LESS_TERMCAP_md=$'\e[01;30m' \
     LESS_TERMCAP_me=$'\e[0m' \
-    LESS_TERMCAP_se=$'\e[0m' \
     LESS_TERMCAP_so=$'\e[01;44;33m' \
+    LESS_TERMCAP_se=$'\e[0m' \
+    LESS_TERMCAP_us=$'\e[01;33m' \
     LESS_TERMCAP_ue=$'\e[0m' \
-    LESS_TERMCAP_us=$'\e[01;32m' \
     command man "$@"
 }
 
-experiment() {
-    cd "$(~/bin/experiment $@)" || exit 1
+# new experiment
+x() {
+    cd "$(~/bin/x $@)" || kill -INT $$
 }
 
 hump() {
@@ -86,19 +247,48 @@ hump() {
 }
 
 howto() {
-    cat "$(find  ~/Archives/Documents/HOWTOs -mindepth 1 -maxdepth 1 | sort | fzf)"
+    cat "$(find  ~/arc/doc/HOWTOs -mindepth 1 -maxdepth 1 | sort | fzf)"
+}
+
+_yt() {
+    local -r base_dir="$1"
+    local -r uri="$2"
+    local -r opts="$3"
+
+    local -r id=$(youtube-dlc --get-id "$uri")
+    local -r title=$(youtube-dlc --get-title "$uri" | sed 's/[^A-Za-z0-9._-]/_/g')
+    local -r dir="${base_dir}/${title}--${id}"
+
+    mkdir -p "$dir"
+    cd "$dir" || kill -INT $$
+    echo "$uri" > 'uri'
+    youtube-dlc $opts -c --write-description --write-info-json "$uri"
+}
+
+yt_audio() {
+    local -r uri="$1"
+    _yt "${DIR_YOUTUBE_AUDIO}/individual" "$uri" '-f 140'
+}
+
+yt_video() {
+    local -r uri="$1"
+    _yt "${DIR_YOUTUBE_VIDEO}/individual" "$uri"
 }
 
 gh_fetch_repos() {
-    curl "https://api.github.com/$1/$2/repos?page=1&per_page=10000"
+    local -r user_type="$1"
+    local -r user_name="$2"
+
+    curl "https://api.github.com/$user_type/$user_name/repos?page=1&per_page=10000"
 }
 
 gh_clone() {
-    gh_user_type="$1"
-    gh_user_name="$2"
-    gh_dir="${DIR_GITHUB}/${gh_user_name}"
+    local -r gh_user_type="$1"
+    local -r gh_user_name="$2"
+
+    local -r gh_dir="${DIR_GITHUB}/${gh_user_name}"
     mkdir -p "$gh_dir"
-    cd "$gh_dir" || exit 1
+    cd "$gh_dir" || kill -INT $$
     gh_fetch_repos "$gh_user_type" "$gh_user_name" \
     | jq --raw-output '.[] | select(.fork | not) | .git_url' \
     | parallel -j 25 \
@@ -117,21 +307,23 @@ gh_clone_repo() {
     gh_username=$(echo "$1" | awk -F / '"$1 == "https" && $3 == github.com" {print $4}')
     gh_dir="${DIR_GITHUB}/${gh_username}"
     mkdir -p "$gh_dir"
-    cd "$gh_dir" || exit 1
+    cd "$gh_dir" || kill -INT $$
     git clone "$1"
 }
 
 work_log_template() {
 cat << EOF
-$(date +%F)
+$(date '+%F %A')
 ==========
 
 Morning report
 --------------
 
-### Previous
+### Prev
+
+### Curr
 
-### Current
+### Next
 
 ### Blockers
 
@@ -142,41 +334,52 @@ EOF
 
 work_log() {
     mkdir -p "$DIR_WORK_LOG"
-    file_work_log_today="${DIR_WORK_LOG}/$(date +%F).md"
+    local -r file_work_log_today="${DIR_WORK_LOG}/$(date +%F).md"
     if [ ! -f "$file_work_log_today" ]
     then
         work_log_template > "$file_work_log_today"
     fi
-    vim "$file_work_log_today"
+    vim -c 'set spell' "$file_work_log_today"
 
 }
 
 note() {
     mkdir -p "$DIR_NOTES"
-    vim "$DIR_NOTES/$(date +'%Y_%m_%d--%H_%M_%S%z')--$1.md"
+    vim -c 'set spell' "$DIR_NOTES/$(date +'%Y_%m_%d--%H_%M_%S%z')--$1.md"
 }
 
 weather() {
-    curl "http://wttr.in/$WEATHER_LOCATION"
+    local _weather_location
+    case "$1" in
+        '') _weather_location="$WEATHER_LOCATION";;
+         *) _weather_location="$1"
+    esac
+    curl "http://wttr.in/$_weather_location?format=v2"
+}
+
+_bt_devs_infos() {
+    # grep's defintion of a line does not include \r, wile awk's does and
+    # which bluetoothctl outputs
+    awk '/^Device +/ {print $2}' \
+    | xargs -I% sh -c 'echo info % | bluetoothctl' \
+    | awk '/^Device |^\t[A-Z][A-Za-z0-9]+: /'
 }
 
 bt_devs_paired() {
-    bluetoothctl -- paired-devices \
-    | awk '{print $2}' \
-    | xargs bluetoothctl -- info
+    echo 'paired-devices' | bluetoothctl | _bt_devs_infos
 }
 
 bt_devs() {
-    bluetoothctl -- devices \
-    | awk '{print $2}' \
-    | xargs bluetoothctl -- info
+    echo 'devices' | bluetoothctl | _bt_devs_infos
 }
 
 run() {
-    stderr="$(mktemp)"
+    local -r stderr="$(mktemp)"
+
+    local code urgency
+
     $@ 2> >(tee "$stderr")
     code="$?"
-    urgency=''
     case "$code" in
         0) urgency='normal';;
         *) urgency='critical'
@@ -184,3 +387,324 @@ run() {
     notify-send -u "$urgency" "Job done: $code" "$(cat $stderr)"
     rm "$stderr"
 }
+
+bar_gauge() {
+    awk "$@" '
+        BEGIN {
+            # CLI options
+            width    = width    ? width    : 80
+            ch_left  = ch_left  ? ch_left  : "["
+            ch_right = ch_right ? ch_right : "]"
+            ch_blank = ch_blank ? ch_blank : "-"
+            ch_used  = ch_used  ? ch_used  : "|"
+            num      = num      ? 1        : 0
+            pct      = pct      ? 1        : 0
+        }
+
+        {
+            cur = $1
+            max = $2
+            lab = $3
+
+            cur_scaled = num_scale(cur, max, 1, width)
+
+            printf \
+                "%s%s%s%s", \
+                lab ? lab         " " : "", \
+                num ? cur "/" max " " : "", \
+                pct ? sprintf("%3.0f%% ", cur / max * 100) : "", \
+                ch_left
+            for (i=1; i<=width; i++) {
+                c = i <= cur_scaled ? ch_used : ch_blank
+                printf "%s", c
+            }
+            printf "%s\n", ch_right
+        }
+
+        function num_scale(src_cur, src_max, dst_min, dst_max) {
+            return dst_min + ((src_cur * (dst_max - dst_min)) / src_max)
+        }
+    '
+}
+
+flat_top_5() {
+    sort -n -k 1 -r \
+    | head -5 \
+    | awk '
+        {
+            cur  = $1
+            max  = $2
+            name = $3
+            pct  = cur / max * 100
+            printf "%s%s %.2f%%", sep, name, pct
+            sep = ",  "
+        }
+
+        END {printf "\n"}
+        '
+}
+
+internet_addr() {
+    curl --silent --show-error --max-time "${1:=1}" 'https://api.ipify.org' 2>&1
+}
+
+status_batt() {
+    case "$(uname)" in
+        'Linux')
+            if which upower > /dev/null
+            then
+                upower --dump \
+                | awk '
+                    /^Device:[ \t]+/ {
+                        device["path"] = $2
+                        next
+                    }
+
+                    /  battery/ && device["path"] {
+                        device["is_battery"] = 1
+                        next
+                    }
+
+                    /    percentage:/ && device["is_battery"] {
+                        device["battery_percentage"] = $2
+                        sub("%$", "", device["battery_percentage"])
+                        next
+                    }
+
+                    /^$/ {
+                        if (device["is_battery"] && device["path"] == "/org/freedesktop/UPower/devices/DisplayDevice")
+                            print device["battery_percentage"], 100, "batt"
+                        delete device
+                    }
+                '
+            fi
+        ;;
+    esac
+}
+
+indent() {
+    awk -v unit="$1" '{printf "%s%s\n", unit, $0}'
+}
+
+status() {
+    local -r indent_unit='    '
+
+    uname -srvmo
+    hostname | figlet
+    uptime
+
+    echo
+
+    echo 'accounting'
+
+    printf '%stmux\n%ssessions %d, clients %d\n' \
+        "$indent_unit" \
+        "${indent_unit}${indent_unit}" \
+        "$(tmux list-sessions 2> /dev/null | wc -l)" \
+        "$(tmux list-clients  2> /dev/null | wc -l)"
+
+    echo
+
+    printf '%sprocs by user\n' "${indent_unit}"
+    ps -eo user \
+    | awk '
+        NR > 1 {
+            count_by_user[$1]++
+            total++
+        }
+
+        END {
+            for (user in count_by_user)
+                print count_by_user[user], total, user
+        }
+        ' \
+    | flat_top_5 \
+    | indent "${indent_unit}${indent_unit}"
+
+    echo
+
+    echo 'resources'
+    (
+        free | awk '$1 == "Mem:" {print $3, $2, "mem"}'
+        df ~ | awk 'NR == 2 {print $3, $3 + $4, "disk"}'
+        status_batt
+    ) \
+    | bar_gauge -v width=60 -v pct=1 \
+    | column -t \
+    | indent "$indent_unit"
+
+    echo
+
+    printf '%smem by proc\n' "$indent_unit"
+    ps -eo rss,cmd \
+    | awk -v total="$(free | awk '$1 == "Mem:" {print $2; exit}')" '
+        NR > 1 {
+            rss = $1
+            cmd = $2
+            n = split(cmd, path, "/")  # _may_ be a path
+            proc = path[n]
+            by_proc[proc] += rss
+        }
+
+        END {
+            for (proc in by_proc)
+                print by_proc[proc], total, proc
+            }
+        ' \
+        | flat_top_5 \
+        | indent "${indent_unit}${indent_unit}"
+
+    echo
+
+    printf '%sthermal\n' "$indent_unit"
+    for _dir in /sys/class/thermal/thermal_zone*
+    do
+        printf '%s %.2f C\n' \
+            $(cat "$_dir"/type) \
+            $(( $(cat "$_dir"/temp) / 1000 ))
+    done \
+    | column -t \
+    | indent "${indent_unit}${indent_unit}"
+
+    echo 'net'
+    local -r internet_addr=$(internet_addr 0.5)
+    local -r internet_ptr=$(host -W 1 "$internet_addr" | awk 'NR == 1 {print $NF}' )
+
+    echo "${indent_unit}internet"
+    echo "${indent_unit}${indent_unit}$internet_addr  $internet_ptr"
+    echo "${indent_unit}if"
+    (ifconfig; iwconfig) 2> /dev/null \
+    | awk '
+        /^[^ ]/ {
+            device = $1
+            sub(":$", "", device)
+            if ($4 ~ "ESSID:") {
+                _essid = $4
+                sub("^ESSID:\"", "", _essid)
+                sub("\"$", "", _essid)
+                essid[device] = _essid
+            }
+            next
+        }
+
+        /^ / && $1 == "inet" {
+            address[device] = $2
+            next
+        }
+
+        /^ +Link Quality=[0-9]+\/[0-9]+ +Signal level=/ {
+            split($2, lq_parts_eq, "=")
+            split(lq_parts_eq[2], lq_parts_slash, "/")
+            cur = lq_parts_slash[1]
+            max = lq_parts_slash[2]
+            link[device] = cur / max * 100
+            next
+        }
+
+        END {
+            for (device in address)
+                if (device != "lo") {
+                    l = link[device]
+                    e = essid[device]
+                    l = l ? sprintf("%.0f%%", l) : "--"
+                    e = e ? e : "--"
+                    print device, address[device], e, l
+                }
+        }
+        ' \
+    | column -t \
+    | indent "${indent_unit}${indent_unit}"
+
+    # WARN: ensure: $USER ALL=(ALL) NOPASSWD:/bin/netstat
+
+    echo "${indent_unit}-->"
+
+    sudo -n netstat -tulnp \
+    | awk -v indent="${indent_unit}${indent_unit}" '
+        NR > 2 && ((/^tcp/ && proc = $7) || (/^udp/ && proc = $6)) {
+           protocol = $1
+           addr = $4
+           port = a[split(addr, a, ":")]
+           name = p[split(proc, p, "/")]
+           names[name] = 1
+           protocols[protocol] = 1
+           if (!seen[protocol, name, port]++)
+               ports[protocol, name, ++seen[protocol, name]] = port
+       }
+
+       END {
+           for (protocol in protocols) {
+               printf "%s%s\t", indent, toupper(protocol)
+               for (name in names) {
+                   if (n = seen[protocol, name]) {
+                       sep = ""
+                       printf "%s:", name
+                       for (i = 1; i <= n; i++) {
+                           printf "%s%d", sep, ports[protocol, name, i]
+                           sep = ","
+                       }
+                       printf "  "
+                   }
+               }
+               printf "\n"
+           }
+       }'
+
+    echo "${indent_unit}<->"
+
+    printf '%sTCP: ' "${indent_unit}${indent_unit}"
+    sudo -n netstat -tnp \
+    | awk 'NR > 2 && $6 == "ESTABLISHED" {print $7}' \
+    | awk -F/ '{print $2}' \
+    | sort -u \
+    | xargs \
+    | column -t
+
+    # TODO: iptables summary
+}
+
+ssh_invalid_attempts_from() {
+    awk '
+        /: Invalid user/ && $5 ~ /^sshd/ {
+            u=$8
+            addr=$10 == "port" ? $9 : $10
+            max++
+            curr[addr]++
+        }
+
+        END {
+            for (addr in curr)
+                if ((c = curr[addr]) > 1)
+                    print c, max, addr
+        }
+        ' \
+        /var/log/auth.log \
+        /var/log/auth.log.1 \
+    | sort -n -k 1 \
+    | bar_gauge -v width="$(stty size | awk '{print $2}')" -v num=1 -v ch_right=' ' -v ch_left=' ' -v ch_blank=' ' \
+    | column -t
+}
+
+loggers() {
+    awk '
+        {
+            split($5, prog, "[")
+            sub(":$", "", prog[1]) # if there were no [], than : will is left behind
+            print prog[1]
+        }' /var/log/syslog /var/log/syslog.1 \
+    | awk '
+        {
+            n = split($1, path, "/")  # prog may be in path form
+            prog = path[n]
+            total++
+            count[prog]++
+        }
+
+        END {
+            for (prog in count)
+                print count[prog], total, prog
+        }' \
+    | sort -n -k 1 \
+    | bar_gauge -v num=1 -v ch_right=' ' -v ch_left=' ' -v ch_blank=' ' \
+    | column -t
+}
This page took 0.055895 seconds and 4 git commands to generate.