#!/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
}

colorize() {
    # sets $COLOR to value set in $COLORIZE
    #echo $COLORIZE
    case "$COLORIZE" in
        [0-2])
            clr='#[fg=color1]' # red
            ;;
        [3-6])
            clr='#[fg=color184]' # yellow
            ;;
        *)
            clr='#[fg=color34]' # green
            ;;
    esac
    COLOR="$clr$COLORIZE"
}
    

# constants
timeout=60 # interval we request bike information at
WORKINGDIR="$HOME/.dotfiles/bin/bluebikes"
LOG=$(printf '%s/%s.log' "$WORKINGDIR/data" $(date +%m-%d))

# create directory on fresh installs
if [[ ! -d "$WORKINGDIR/data/status" ]] ; then
    mkdir -p "$WORKINGDIR/data/status"
fi
# create log
if [[ ! -f $LOG ]] ; then
    touch $LOG
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

    COLORIZE="$BIKES"
    colorize
    echo "$BIKES $COLOR $clr $COLORIZE"
    BIKES="$COLOR"

    COLORIZE="$DOCKS"
    colorize
    DOCKS="$COLOR"
    echo "$DOCKS $COLOR $clr $COLORIZE"
    printf '%s   %s  #[default] %s' $BIKES $DOCKS "$(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