#!/bin/sh
set -f

words="the be of and a to in he have it that for they i with as not on she at by this we you do but from or which one would all will there say who make when can more if no man out other so what time up go about than into could state only new year some take come these know see use get like then first any work now may such give over think most even find day also after way many must look before great back through long where much should well people down own just because good each those feel seem how high too place little world very still nation hand old life tell write become here show house both between need mean call develop under last right move thing general school never same another begin while number part turn real leave might want point form off child few small since against ask late home interest large person end open public follow during present without again hold govern around possible head consider word program problem however lead system set order eye plan run keep face fact group play stand increase early course change help line"

text=""

test_type="time"
test_length="30"

# test if date supports %N
NANOSECONDS=true
_s=
case "$(date +%N)" in 
    ""|"%N") NANOSECONDS=false;;
esac

gettime () {
    $NANOSECONDS && date +%s%N \
        || printf "%s000000000" "$(date +%s)" 2>/dev/null
}

tty_init () {
    printf ""
    export SAVED_TTY_SETTINGS=$(stty -g)
    stty raw -echo
    trap typr_cleanup 1 2 3 6

    printf "[6 q"
}

tty_cleanup () {
    stty $SAVED_TTY_SETTINGS
    printf "[?25h"
}

tty_readc () {
    stty -echo -icanon min 1 time 0
    s="$(dd bs=1 count=1 of=/dev/stdout 2>/dev/null)"
    stty -icanon min 0 time 0
    [ "$s" = "" ]  && {
        s="$s$(dd bs=1 count=2 of=/dev/stdout 2>/dev/null)"
    }
    printf "%s" "$s"
}

typr_draw_text () {
    y="${areay}" startcol=${areax}

    printf "[?25l"
    printf "%s\n" "$text" | while IFS= read -r line; do
        printf "%s" "[${y};${startcol}H${line}"
        y=$((y+1))
    done
    printf "[%d;%dH[?25h" "${areay}" "${startcol}"

}

typr_get_time () {
    time_ns="$1"

    time_ms=$(((time_ns/1000000)%1000))
    time_seconds=$(((time_ns/1000000000)%60))
    time_minutes=$((time_ns/60000000000))

    printf "%02d:%02d.%03d" "$time_minutes" "$time_seconds" "$time_ms"
}

typr_draw_time () {
    [ -z "$start" ] && return 1

    now="$(gettime)"
    time_ns=$((now-start))
    case "$test_type" in
        "time")
            time_ns=$((1000000000*test_length - time_ns))
            ;;
    esac

    [ "$time_ns" -lt "0" ] && {
        [ "$(((time_ns / 500000000) % 2))" = "0" ] \
            && printf "[$((areay - 1));${areax}H%s" "         " \
            && return 1
        time_ns=0
    }

    printf "[$((areay - 1));${areax}H%s" "$(typr_get_time "$time_ns")"
}

typr_calculate_raw_wpm () {
    time_ns="$1"
    printf "%s" "$((${#entered_text}*60000000000/(time_ns*5)))"
}

typr_calculate_acc () {
    [ "$total_kp" != "0" ] && printf "%s" "$(((100*correct_kp)/total_kp))" || printf "%s" "0"
}


typr_show_results () {
    printf "[?25l"

    now="$(gettime)"
    time_ns=$((now-start))

    #wc="$(set -- $text ; printf "%s" "$#")"
    raw_wpm="$(typr_calculate_raw_wpm "$time_ns")"
    acc="$(typr_calculate_acc)"
    wpm="$(((acc*raw_wpm) / 100))"

    printf "[%s;${areax}H%s" \
        "${areay}" "wpm" \
        "$((areay+1))" "$wpm" \
        "$((areay+2))" "acc" \
        "$((areay+3))" "${acc}%" \
        "$((areay+4))" "time" \
        "$((areay+5))" "$(typr_get_time "$time_ns")" \
        "$((areay+6))" "raw" \
        "$((areay+7))" "$raw_wpm"
}

typr_wrap_text () {
    t="$text" line="" lt=""
    while [ "$t" ] ; do
        lt="$ct"
        ct="${t%"${t#?}"}"
        t="${t#?}"

        [ "$lt" = " " ] && {
            next_word=${t%% *}
            [ "${#line}" -gt "$((areax - ${#next_word}))" ] && {
                printf "%s\n" "${line}"
                line=""
            }
        }
        line="${line}${ct}"
    done
    printf "%s\n" "${line}"
}

typr_generate_text () {
    case "$test_type" in
        "words")
            wordcount="$test_length"
            ;;
        "time")
            # TODO auto add lines as test progresses
            wordcount="100"
            ;;
    esac

    text="$(printf "%s " $(printf "%s\n" $words | shuf -n $wordcount))"
    text="${text% }"
    text="$(typr_wrap_text)"
}

typr_draw_loop () {
    while true; do
        typr_draw_time
    done
}

typr_start_timer () {
    start="$(gettime)"
    typr_draw_loop &
    draw_pid="$!"
    export start draw_pid
}

typr_update_acc () {
    t="$current_line" e="$entered_line"
    while [ "$e" ] ; do
        ct="${t%"${t#?}"}"
        t="${t#?}"

        ce="${e%"${e#?}"}"
        e="${e#?}"

        [ "$ct" = "$ce" ] && correct_kp=$((correct_kp+1))
        total_kp=$((total_kp+1))
    done
    export correct_kp total_kp
}

typr_redraw_line () {
    color=""
    line=$((areay + current_line_no - 1))
    startcol=${areax}

    draw="[${line};${startcol}H${color}"

    i=0
    t="$current_line"
    e="$entered_line"
    while [ "$t" ] ; do
        ct="${t%"${t#?}"}"
        t="${t#?}"

        ce="${e%"${e#?}"}"
        e="${e#?}"

        case "$ce" in
            "$ct")
                newcolor=""
                ;;
            "")
                newcolor=""
                ;;
            *)
                newcolor=""
            ;;
        esac

        [ "$color" != "$newcolor" ] && {
            color="$newcolor"
            draw="${draw}$newcolor"
        }


        [ "$color" = "" ] && {
            case "$ce" in " ") draw="${draw}_" ;; *) draw="${draw}$ce"; esac
        } || {
            draw="${draw}$ct"
        }
    done
    printf "%s[${line};$((${#entered_line} + startcol))H[?25h" "$draw"
}

typr_new_line () {
    [ -n  "$entered_text" ] && entered_text="$entered_text
"
    entered_text="${entered_text}${entered_line}"
    typr_update_acc

    entered_line=""

    current_line_no=$((current_line_no+1))
    current_line="$(typr_get_text_line $current_line_no)"
    typr_redraw_line
}

typr_add_letter () {
    c="$1"
    [ -z "$start" ] && typr_start_timer

    entered_line="$entered_line$c"

    typr_redraw_line

    [ "${#entered_line}" = "${#current_line}" ] && {
        typr_new_line
    }

}

typr_del_letter () {
    [ -z "$entered_line" ] && prev=true || prev=false

    entered_line="${entered_line%?}"
    typr_redraw_line

    $prev && [ "$current_line_no" -gt "1" ] || return 0

    # move back one line in entered
    current_line_no=$((current_line_no-1))
    current_line="$(typr_get_text_line $current_line_no)"

    entered_line="$(typr_get_text_line $current_line_no "$entered_text")"
    entered_line="${entered_line%?}"
    entered_text="$(typr_remove_last_line "$entered_text")"
    typr_redraw_line
}

# removes the last line, ignoring trailing whitespace
#
typr_remove_last_line() {
    t="${1%
}"
    printf "%s" "${1%
}" | while IFS= read -r line; do
        printf "%s\n" "$line"
    done
}

typr_get_text_line() {
    n="$1"
    t="${2:-$text}"
    i=0
    case "$n" in
        ""|*[!0-9]*) return 1;;
    esac

    printf "%s\n" "$t" | while IFS= read -r line; do
        i=$((i+1))
        [ "$i" = "$n" ] && printf "%s\n" "$line"
    done
}

typr_main () {
    entered_text=""
    entered_line=""
    start=

    total_kp=0
    correct_kp=0

    current_line_no=1
    current_line="$(typr_get_text_line $current_line_no)"

    export correct_kp total_kp entered_text entered_line start current_line_no current_line

    printf ""
    typr_draw_text
    while true; do
        c="$(tty_readc)"
        case "$c" in
            ''|'') break;;
            '')
                typr_del_letter
                ;;
                " "|A|B|C|D|E|F|G|H|I|J|K|L|M|N|O|P|Q|R|S|T|U|V|W|X|Y|Z|a|b|c|d|e|f|g|h|i|j|k|l|m|n|o|p|q|r|s|t|u|v|w|x|y|z|0|1|2|3|4|5|6|7|8|9|,|.|\?|\"|\'|!|-)
                    typr_add_letter "$c"
                ;;
            '	')
                kill "$draw_pid"
                typr_generate_text
                typr_main
                return
                ;;
        esac

        case "$test_type" in
            "words")
                [ "$((${#entered_text}+${#entered_line}))" = "${#text}" ] && break
                ;;
            "time")
                now="$(gettime)"
                time_ns=$((now-start))
                [ "$time_ns" -gt "$((test_length*1000000000))" ] && break
                ;;
        esac
    done
    kill "$draw_pid"

    entered_line="$entered_line "
    typr_new_line
    typr_show_results

    while true; do
        case "$(tty_readc)" in
            ''|''|q) break;;
            '	')
                typr_generate_text
                typr_main
                return
                ;;
        esac
    done
}

typr_init () {
    set -- $(stty size)
    cols="$2"
    lines="$1"

    areax=$((cols / 3))
    areay=$((lines / 3))
    tty_init
    typr_generate_text
    typr_main
    tty_cleanup
}

typr_usage () {
    printf "%s\n" "$0 [-h] [-m time|words] [-d duration/words]" 1>&2
    exit 1
}


while getopts "m:d:h" opt; do
    case "$opt" in
        h)
            typr_usage ;;
        m)
            case "$OPTARG" in
                "time"|"words") test_type="$OPTARG" ;;
                *) typr_usage ;;
            esac
        ;;
        d)
            test_length="$OPTARG"
        ;;
    esac
done

shift $((OPTIND-1))

typr_init