1#!/usr/bin/env python
  2
  3import sys, os, subprocess
  4
  5def escaped(s):
  6	return repr(s)[1:-1]
  7
  8def tput(term, name):
  9	try:
 10		return subprocess.check_output(['tput', '-T%s' % term, name]).decode()
 11	except subprocess.CalledProcessError as e:
 12		return e.output.decode()
 13
 14
 15def w(s):
 16	if s == None:
 17		return
 18	sys.stdout.write(s)
 19
 20terminals = {
 21	'xterm' : 'xterm',
 22	'rxvt-256color' : 'rxvt_256color',
 23	'rxvt-unicode' : 'rxvt_unicode',
 24	'linux' : 'linux',
 25	'Eterm' : 'eterm',
 26	'screen' : 'screen'
 27}
 28
 29keys = [
 30	"F1",		"kf1",
 31	"F2",		"kf2",
 32	"F3",		"kf3",
 33	"F4",		"kf4",
 34	"F5",		"kf5",
 35	"F6",		"kf6",
 36	"F7",		"kf7",
 37	"F8",		"kf8",
 38	"F9",		"kf9",
 39	"F10",		"kf10",
 40	"F11",		"kf11",
 41	"F12",		"kf12",
 42	"INSERT",	"kich1",
 43	"DELETE",	"kdch1",
 44	"HOME",		"khome",
 45	"END",		"kend",
 46	"PGUP",		"kpp",
 47	"PGDN",		"knp",
 48	"KEY_UP",	"kcuu1",
 49	"KEY_DOWN",	"kcud1",
 50	"KEY_LEFT",	"kcub1",
 51	"KEY_RIGHT",	"kcuf1"
 52]
 53
 54funcs = [
 55	"T_ENTER_CA",		"smcup",
 56	"T_EXIT_CA",		"rmcup",
 57	"T_SHOW_CURSOR",	"cnorm",
 58	"T_HIDE_CURSOR",	"civis",
 59	"T_CLEAR_SCREEN",	"clear",
 60	"T_SGR0",		"sgr0",
 61	"T_UNDERLINE",		"smul",
 62	"T_BOLD",		"bold",
 63	"T_BLINK",		"blink",
 64	"T_REVERSE",            "rev",
 65	"T_ENTER_KEYPAD",	"smkx",
 66	"T_EXIT_KEYPAD",	"rmkx"
 67]
 68
 69def iter_pairs(iterable):
 70	iterable = iter(iterable)
 71	while True:
 72		yield (next(iterable), next(iterable))
 73
 74def do_term(term, nick):
 75	w("// %s\n" % term)
 76	w("var %s_keys = []string{\n\t" % nick)
 77	for k, v in iter_pairs(keys):
 78		w('"')
 79		w(escaped(tput(term, v)))
 80		w('",')
 81	w("\n}\n")
 82	w("var %s_funcs = []string{\n\t" % nick)
 83	for k,v in iter_pairs(funcs):
 84		w('"')
 85		if v == "sgr":
 86			w("\\033[3%d;4%dm")
 87		elif v == "cup":
 88			w("\\033[%d;%dH")
 89		else:
 90			w(escaped(tput(term, v)))
 91		w('", ')
 92	w("\n}\n\n")
 93
 94def do_terms(d):
 95	w("var terms = []struct {\n")
 96	w("\tname  string\n")
 97	w("\tkeys  []string\n")
 98	w("\tfuncs []string\n")
 99	w("}{\n")
100	for k, v in d.items():
101		w('\t{"%s", %s_keys, %s_funcs},\n' % (k, v, v))
102	w("}\n\n")
103
104w("// +build !windows\n\npackage termbox\n\n")
105
106for k,v in terminals.items():
107	do_term(k, v)
108
109do_terms(terminals)
110