98 lines
2.7 KiB
Python
98 lines
2.7 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
|
|
import sys
|
|
import curses
|
|
|
|
|
|
class Word():
|
|
def __init__(self, target, x, y):
|
|
self.target = target
|
|
self.input = ""
|
|
self.x = x
|
|
self.y = y
|
|
|
|
|
|
def get_words(text, max_x, max_y):
|
|
curr_x = 0
|
|
curr_y = 0
|
|
|
|
words = []
|
|
for word in text.split():
|
|
if curr_x + len(word) > max_x:
|
|
curr_y += 1
|
|
curr_x = 0
|
|
|
|
if curr_y > max_y:
|
|
print("text too long, scrolling not implemented.")
|
|
exit(1)
|
|
|
|
words.append(Word(word, curr_x, curr_y))
|
|
curr_x += len(word) + 1
|
|
|
|
return words
|
|
|
|
|
|
def main_curses(stdscr, text):
|
|
curses.use_default_colors()
|
|
curses.init_pair(1, curses.COLOR_GREEN, -1)
|
|
curses.init_pair(2, curses.COLOR_WHITE, curses.COLOR_RED)
|
|
|
|
key = old_max_y = old_max_x = curr_word = 0
|
|
|
|
while True:
|
|
# rewrap words on screen size change
|
|
max_y, max_x = stdscr.getmaxyx()
|
|
if max_y != old_max_y or max_x != old_max_x:
|
|
words = get_words(text, max_x, max_y)
|
|
old_max_x = max_x
|
|
old_max_y = max_y
|
|
|
|
if key in (curses.KEY_BACKSPACE, 'KEY_BACKSPACE', '\b', '\x7f', 127): # fml
|
|
words[curr_word].input = words[curr_word].input[:-1]
|
|
elif key >= 32 and key <= 126:
|
|
words[curr_word].input += chr(key)
|
|
|
|
# increment current word if its completed
|
|
if words[curr_word].input == words[curr_word].target:
|
|
curr_word += 1
|
|
|
|
# text complete
|
|
if curr_word == len(words):
|
|
break
|
|
|
|
stdscr.clear()
|
|
|
|
for word in words:
|
|
if word.input == word.target:
|
|
stdscr.addstr(word.y, word.x, word.target, curses.A_DIM)
|
|
elif word == words[curr_word]:
|
|
stdscr.addstr(word.y, word.x, word.target, curses.A_UNDERLINE)
|
|
else:
|
|
stdscr.addstr(word.y, word.x, word.target)
|
|
|
|
for i, c in enumerate(words[curr_word].input):
|
|
if i >= len(words[curr_word].target) or words[curr_word].input[i] != words[curr_word].target[i]:
|
|
stdscr.addstr(words[curr_word].y, words[curr_word].x + i, c, curses.color_pair(2) | curses.A_BOLD)
|
|
else:
|
|
stdscr.addstr(words[curr_word].y, words[curr_word].x + i, c, curses.color_pair(1))
|
|
|
|
stdscr.move(words[curr_word].y, words[curr_word].x + len(words[curr_word].input))
|
|
|
|
key = stdscr.getch()
|
|
|
|
|
|
def main(argv):
|
|
from argparse import ArgumentParser
|
|
parser = ArgumentParser()
|
|
parser.add_argument("input")
|
|
args = parser.parse_args(argv[1:])
|
|
|
|
text = open(args.input).read()
|
|
|
|
curses.wrapper(main_curses, text)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main(sys.argv)
|