bluebike script working with fzf search. tweaked docker script
Keegan
3 years ago
parent
0b9c8429b8
commit
6a67b47397
@ -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
|
||||
@ -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
|
||||
}
|
||||
]
|
||||
}
|
||||
@ -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
|
||||
|
||||
Binary file not shown.
Loading…
Reference in New Issue