bluebike script working with fzf search. tweaked docker script

bin/bluebikes/bb

@@ -0,0 +1,203 @@
+#!/usr/bin/env bash
+
+set -e
+
+usage() {
+    # basic usage statement
+    cat <<EOF
+Usage: bb [-s|f|h|r [option]]
+
+Queries Bluebikes API to get recent bike status
+
+Options:
+  -s, --search          pulls up a searchable fzf menu to update
+                          displayed station
+  -r, --rename[=NAME]   rename the currently selected station
+                          stored in data/alias.json
+                          if called without an arguement resets to default
+  -h, --help            display this message
+EOF
+}
+
+# constants
+timeout=60 # interval we request bike information at
+WORKINGDIR="$HOME/.dotfiles/bin/bluebikes"
+LOG=$(printf '%s/%s.log' "$WORKINGDIR/data" $(date +%m-%d))
+if [[ ! -f $LOG ]] ; then
+    touch $LOG
+fi
+
+# create directory on fresh installs
+if [[ ! -d "$WORKINGDIR/data/status" ]] ; then
+    mkdir -p "$WORKINGDIR/data/status"
+fi
+
+if [[ ! -e "$WORKINGDIR/data/alias.json" ]] ; then
+    starter_json=$(printf '{ "stations": [] }' | gojq '.')
+    echo "$starter_json" > "$WORKINGDIR/data/alias.json"
+fi
+
+# finding station id
+if [[ -e "$WORKINGDIR/data/.station" ]] ; then
+    # if file exists
+    source "$WORKINGDIR/data/.station"
+fi
+
+# setting time for expiration
+TIME=$(date +%s)
+# allows customizing search window or dropping in dmenu
+SEARCH="fzf-tmux --layout=reverse -p 50%,50% --border" # fzf window will float in tmux
+
+
+get_station_info() {
+    if [[ -z "$STATIONID" ]] ; then
+        printf 'Please set a station with bb -s' >&2
+        exit 1
+    fi
+    
+    # if $UPDATE cleans folder and returns most recent info for $STATION_ID
+    if $UPDATE ; then
+        printf '%s: %s\n' $(date +%H:%M:%S) "updated station status" >> $LOG
+        BB_STATUS=$(curl --silent -fL https://gbfs.bluebikes.com/gbfs/en/station_status.json | gojq '.data')
+        EXPIRATION=$(($TIME + $timeout)) # sets expiration to be in 120 seconds
+        echo "$BB_STATUS" > "$WORKINGDIR/data/status/$EXPIRATION.json"
+    else
+        BB_STATUS=$(cat "$WORKINGDIR/data/status/"*.json | gojq '.data') # getting status from file
+    fi
+    # get relevant station info and set vars
+    BIKES=$(echo "$BB_STATUS" | gojq --arg id $STATIONID '.stations[] | select( .station_id == $id ) | .num_bikes_available' 2>/dev/null)
+    DOCKS=$(echo "$BB_STATUS" | gojq --arg id $STATIONID '.stations[] | select( .station_id == $id ) | .num_docks_available' 2>/dev/null)
+    #echo "$BIKES" "$DOCKS"
+}
+
+check_update() {
+    # sets $UPDATE to bool
+    lst="$WORKINGDIR/data/status/*"
+    if ls $lst 1>/dev/null 2>&1 ; then
+        # file exists
+        printf '%s: %s\n' $(date +%H:%M:%S) "found status file" >> $LOG
+        OLD_EXPIRATION=$(ls "$WORKINGDIR/data/status" | tr -d '.json' | sort -n | tail -n 1) # gets latest file
+        if [[ $OLD_EXPIRATION -lt $TIME ]] ; then
+            # file is expired
+            printf '%s: %s\n' $(date +%H:%M:%S) "removing old status files" >> $LOG
+            rm "$WORKINGDIR/"data/status/*.json
+            UPDATE=true
+        fi
+    else
+        UPDATE=true
+    fi
+}
+
+
+update_station() {
+    # provides a gui to update the station to watch
+    printf '%s: %s\n' $(date +%H:%M:%S) "updating station to watch" >> $LOG
+    STATIONS=$(curl --silent -fL https://gbfs.bluebikes.com/gbfs/en/station_information.json | gojq '.data')
+    if [[ -z "$STATIONS" ]] ; then
+        printf 'Error retrieving station info! Check your connection' >&2
+        exit 1
+    fi
+
+    # all associated data for the specific ids
+    condensed=$(echo "$STATIONS" | gojq '[.stations[] | {name: .name, id: .station_id} ]') # [{ name:, id: }, ...]
+    # prompt user to search
+    station_list=$(echo "$condensed" | gojq '.[].name' | tr -d '"' )
+    new_station=$(echo "$station_list" | eval "$SEARCH")
+    if [[ -z "$new_station" ]] ; then
+        exit 0
+    fi
+    # number correlating to selected name
+    STATIONID=$(echo "$condensed"| gojq --arg name "$new_station" '.[] | select( .name == $name) | .id' | tr -d '"') # trim quotes
+    # setting data/.station file
+    output=$(printf 'STATIONID=%s\nSTATION_NAME="%s"\n' "$STATIONID" "$new_station")
+    echo "$output" > "$WORKINGDIR/data/.station"
+#    echo "$output"
+}
+
+
+output_status() {
+    # reads $OUTPUT and the Index value
+    get_station_info # sets $DOCKS $BIKES and $STATIONID
+    NAME="$STATION_NAME"
+    if [[ -e "$WORKINGDIR/data/alias.json" ]] ; then
+        alias=$(cat "$WORKINGDIR/data/alias.json" | gojq --arg id "$STATIONID" '.stations.[] | select(.id == $id) | .alias')
+        if [[ ! -z "$alias" ]] ; then
+            NAME="$alias"
+        fi
+    fi
+
+    printf '%d   %d   %s' "$DOCKS" "$BIKES" "$(echo "$NAME" | tr -d '"')"
+}
+
+set_station_alias() {
+    # sets $ALIAS for $STATION_ID
+
+    if [[ -z "$STATIONID" ]] ; then
+        # no station id
+        printf 'No Station set to change name for\nPlease run bb -s to set a station' >&2
+        exit 1
+    fi
+
+    aliases=$(cat "$WORKINGDIR/data/alias.json") # alias json
+
+    if [[ -z "$ALIAS" ]] ; then
+        # no arguement for rename, clearing
+        exists=$(echo "$aliases" | gojq --arg id "$STATIONID" '.stations.[] | select(.id == $id)') # will get an alias for the id
+        if [[ ! -z "$exists" ]] ; then
+            # alias exists wipe it
+            aliases=$(gojq --arg id "$STATIONID" '.stations |= [ .[] | select(.id != $id)]' <<<"$aliases")
+            echo "$aliases" > "$WORKINGDIR/data/alias.json"
+        fi
+    else
+        # rename to $ALIAS
+        exists=$(echo "$aliases" | gojq --arg id "$STATIONID" '.stations.[] | select(.id == $id)') # will get an alias for the id
+        if [[ -z "$exists" ]]; then
+            # create entry
+            entryjson=$(printf '{"name": "%s", "id": "%s", "alias": "%s"}' "$STATION_NAME" "$STATIONID" "$ALIAS" | gojq '.')
+            aliases=$(gojq --argjson entry "$entryjson" '.stations |= . + [$entry]' <<<"$aliases")
+        else
+            # update entry
+
+            aliases=$(echo "$aliases" | gojq --arg id "$STATIONID" --arg als "$ALIAS" '.stations |= [ .[] | select(.id == $id) | .alias |= $als]')
+
+        fi
+        echo "$aliases" > "$WORKINGDIR/data/alias.json"
+    fi
+}
+
+if [[ $# -eq 0 ]] ; then
+    # no args passed
+    check_update
+    output_status
+fi
+
+while [[ $# -gt 0 ]] ; do
+    # loop over args if present
+    case $1 in
+        -h | --help)
+            usage
+            exit 0
+            ;;
+        -s | --search)
+            update_station
+            ;;
+        -f | --force)
+            # force reload bike status
+            UPDATE=true
+            output_status
+            ;;
+        -r | --rename)
+            ALIAS="$2"
+            set_station_alias # arguement might be in $2
+            exit 0 # ensure we exit
+            ;;
+        *)
+            echo "Error: bb" >&2
+            usage
+            exit 1
+            ;;
+    esac
+    shift # shift over the args before loop
+done
+
+# any other case would throw an error in the while loop as an unrecognized arguement

bin/bluebikes/conf.json

@@ -1,24 +0,0 @@
-{
-  "stations": [
-    {
-      "name": "Mass Ave / Amherst",
-      "station_id": 67
-    },
-    {
-      "name": "Longfellow Elementary",
-      "station_id": 413
-    },
-    {
-        "name": "BCI",
-        "station_id": 80
-    },
-    {
-        "name": "Stata",
-        "station_id": 107
-    },
-    {
-        "name": "Grad Housing",
-        "station_id": 179
-    }
-  ]
-}

bin/bluebikes/get_status.sh

@@ -1,57 +0,0 @@
-#!/usr/bin/env bash
-
-# constants
-timeout=120 # in seconds
-refresh_rate=10 # display each station for 10 seconds
-WORKINGDIR="$HOME/.dotfiles/bin/bluebikes"
-
-if [[ ! -d "$WORKINGDIR/data" ]] ; then
-    mkdir -p "$WORKINGDIR/data"
-fi
-
-# getting time and existing information
-TIME=$(date +%s)
-OLD_EXPIRATION=$(ls "$WORKINGDIR/data" | cut -d '.' -f 1 2>/dev/null) # removes .json extension
-
-get_station_info() {
-    # cleans data folder and leaves only most recent copy
-    rm "$WORKINGDIR/data/"*.json 2>/dev/null
-    BB_STATUS=$(curl --silent -fL https://gbfs.bluebikes.com/gbfs/en/station_status.json)
-    EXPIRATION=$(($TIME + 120)) # sets expiration to be in 120 seconds
-    echo "$BB_STATUS" > "$WORKINGDIR/data/$EXPIRATION.json"
-    #echo $EXPIRATION
-}
-
-update_stations() {
-    # checks to see if we are out of date and sets $output to be station info
-    if [[ -z $OLD_EXPIRATION || $OLD_EXPIRATION -lt $EXPIRATION ]] ; then
-        # file doesn't exist or is more than a minute old
-        get_station_info
-    fi
-
-    STATIONS=$(cat "$WORKINGDIR/conf.json" | gojq '.stations') # stations we care about from config file
-    STATION_INFO=$(cat "$WORKINGDIR/"data/*.json 2>/dev/null) # all station information
-
-    # all associated data for the specific ids
-    info=$(echo "$STATION_INFO" | gojq --argjson js "$STATIONS" '[.data.stations[] | select( .station_id as $id | $js.[].station_id | tostring | . == $id )]')
-    # only available bikes/docks for the required ids
-    output=$(echo "$info" | gojq --argjson names "$STATIONS" '[ .[] | .station_id as $id | { bikes: .num_bikes_available, docks: .num_docks_available, station: $names.[] | select( .station_id | tostring | . == $id ) | .name }]')
-    
-#    echo "$output"
-}
-
-
-output_status() {
-    # reads $OUTPUT and the Index value
-    minute=$(date +%M | tr -d '0')
-    tot=$(echo "$STATIONS" | gojq '. | length') # display one per minute
-    index=$(( $minute % $tot ))
-    bikes=$(echo "$output" | gojq --arg indx $index '.[$indx | tonumber ].bikes')
-    docks=$(echo "$output" | gojq --arg indx $index '.[$indx | tonumber ].docks')
-    station=$(echo "$output" | gojq --arg indx $index '.[$indx | tonumber ].station' | tr -d '"')
-    printf '%d   %d   %s' "$docks" "$bikes" "$station"
-}
-
-update_stations # sets $output to be all the stations we are interested in
-output_status # echos correct station info based on index and time
-

bin/docker.sh

@@ -5,9 +5,9 @@ if [[ -z $EXISTS ]] ; then
     echo ""
 else
     DOCKER_STATUS=$(docker info)
-    CONTAINERS=$(echo "$DOCKER_STATUS" | awk '/Containers/ {print $2}')
+    #CONTAINERS=$(echo "$DOCKER_STATUS" | awk '/Containers/ {print $2}')
     RUNNING=$(echo "$DOCKER_STATUS" | awk '/Running/ {print $2}')
-    STOPPED=$(echo "$DOCKER_STATUS" | awk '/Stopped/ {print $2}')
+    #STOPPED=$(echo "$DOCKER_STATUS" | awk '/Stopped/ {print $2}')
     IMAGES=$(echo "$DOCKER_STATUS" | awk '/Images/ {print $2}')
-    echo -e "\ue7b0 $RUNNING \ufb99 $STOPPED \uf1c5 $IMAGES"
+    echo -e "\ue7b0 $RUNNING \uf1c5 $IMAGES"
 fi

bin/weather/main.go

@@ -35,7 +35,7 @@ const (
 )
 
 func (w *Weather) String() string {
-	return fmt.Sprintf(" %c %.1f%c %d%c in %s", w.Icon, w.Temp, DEGREES_F, w.Humidity, HUMIDITY, w.City) // Output as ICON ##.#*F ##% in City, ST where the rune is a degree sign
+	return fmt.Sprintf(" %c  %.1f%c %d%c in %s", w.Icon, w.Temp, DEGREES_F, w.Humidity, HUMIDITY, w.City) // Output as ICON ##.#*F ##% in City, ST where the rune is a degree sign
 }
 
 func main() {

tmux.conf

@@ -25,7 +25,7 @@ set -g status-interval 5
 set -g status-style "bg=black, fg=white"
 set -g status-right "#(~/.dotfiles/bin/docker.sh) | #(~/.tmux/plugins/tmux-mem-cpu-load/tmux-mem-cpu-load -p -i 1 -a 1)|#[default] %a %l:%M "
 set -g status-right-length 100
-set -g status-left "#(~/.dotfiles/bin/weather/weather) | #(~/.dotfiles/bin/bluebikes/get_status.sh) | #(~/.dotfiles/bin/water/water.sh) |"
+set -g status-left "#(~/.dotfiles/bin/weather/weather) | #(~/.dotfiles/bin/bluebikes/bb) | #(~/.dotfiles/bin/water/water.sh) |"
 set -g status-left-length 100
 
 # tmux auto start

zshrc

@@ -18,7 +18,7 @@ export GOBIN=$GOPATH/bin
 export PATH=$PATH:$GOBIN:$GOPATH:$GOROOT:$GOROOT/bin
 
 # adding scripts to path
-export PATH=$PATH:$HOME/.dotfiles/bin/startup:$HOME/.dotfiles/bin:$HOME/.dotfiles/bin/weather
+export PATH=$PATH:$HOME/.dotfiles/bin/startup:$HOME/.dotfiles/bin:$HOME/.dotfiles/bin/weather:$HOME/.dotfiles/bin/bluebikes
 
 # Set name of the theme to load --- if set to "random", it will
 # load a random theme each time oh-my-zsh is loaded, in which case,