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