covid-19_stats.c.15m.sh

#!/bin/zsh 
# covid-19_stats.1c.15m.sh
# This script displays stats of US COVID-19 cases, with a submenu for 
# user-defineable States. Can also be configured to show the top n states.
# Created by: Wilson Goode
# Modified for Argos by: David Madison Hardaway, Jr.
# Pulled from GitHub: March 30, 2020
# Last update for Argos: April 5, 2020


# ==============================DEPENDENCIES================================= #
# This script requires Node.js/npm and the following CLI tool                 #
# Requires: https://github.com/ahmadawais/corona-cli                          #
# Install via npm:  `npm i -g corona-cli`                                     #
# =========================================================================== #


# ==============================CONFIGURATION================================ #
# Set these variables to configure the output to your                         #
# liking. Set the directory for your Argos Plugins / directory you           #
# want to keep a cache in.                                                    #
ARGOS_DIR=~/.config/argos/covid-argos

# Choose which states you want stats for. Any states you add here will        #
# be shown within the dropdown menu. Be sure to separate each state in        #
# its own parentheses.                                                        #
STATES=("North Carolina" "New York")

# ALTERNATIVE MODE: Instead of choosing states, you can choose to have        #
# the top n states. If TOP_N=true, shows N_STATES number of states with       #
# the most cases.                                                             #
TOP_N=false firstoccur=$LINENO #stores line number for later function
N_STATES=15
# =========================================================================== #


# ==============================SCRIPT BELOW================================= #
# This section modifies the call to grep to switch between specific states or
# TOP_N states.                                                     
MOD_GREP_STATES="#"
if [ "$TOP_N" = true ]; then
    MOD_GREP_A="-A$N_STATES"
else
    MOD_GREP_A=""
    for state in "${STATES[@]}"
        do 
            MOD_GREP_STATES="$MOD_GREP_STATES\\|$state"
        done
fi 

# These calls pull data for USA and then individual states, storing in a cache.
corona usa -x -m &>$ARGOS_DIR/.corona_usa_cache
corona states -x -m &>$ARGOS_DIR/.corona_states_cache

# Defining ANSI colors for output
RED='\033[01;31m'
GREEN='\033[01;32m'
YELLOW='\033[01;33m'
BLUE='\033[01;36m'
NONE='\033[0m'

# Top line for USA data
cat $ARGOS_DIR/.corona_usa_cache |
    grep "USA" |
    sed 's/\x1b\[[0-9;]*m//g' |
    sed -E 's/[[:space:]][[:space:]][[:space:]]*/;/g' |
    awk -v r=$RED -v y=$YELLOW -v g=$GREEN -v b=$BLUE -v n=$NONE -F';' \
	'{ printf "%15s %15s %15s %15s %15s |font=Courier\n",
            n$2, "😷 "b$3, g"("$4"▲)", n"💀 "r$5, y"("$6"▲)"}'
echo "---"

# Submenu for GMT clock time -- the time that the updates are based upon
echo -n "The current date and time is: " ; date -u +%c

# Submenu for States of Interest
cat $ARGOS_DIR/.corona_states_cache | 
    grep $MOD_GREP_A $MOD_GREP_STATES |
    sed 's/\x1b\[[0-9;]*m//g' |
    sed -E 's/[[:space:]][[:space:]][[:space:]]*/;/g ; 
        s/District Of Columbia/Washington, D.C./ ;
        s/United States Virgin Islands/US Virgin Islands/ ;
        s/Diamond Princess Cruise/Diamond Princess Cr./' |
    awk -v r=$RED -v y=$YELLOW -v g=$GREEN -v b=$BLUE -v n=$NONE -F';' \
        '{ printf "%-30s %20s %30s %20s %30s |font=Courier size=12\n",
            y$2, b$3, g"("$4"▲)", r$5, y"("$6"▲)"}'

# Submenu for ALTERNATIVE MODE selection:
echo "Would you like to view Top 15 States or chosen States?"
echo "--Top 15 States | terminal=false refresh=true \
    bash='sed -i '$firstoccur\s/TOP_N=false/TOP_N=true/' $0'"
echo "--Selected States | terminal=false refresh=true \
    bash='sed -i '$firstoccur\s/TOP_N=true/TOP_N=false/' $0'"
#EOF