git.r.bdr.sh - rbdr/dotfiles/blame_incremental - vim/autoload/conque

... / ...

Commit	Line	Data
	1	" FILE: plugin/conque_term.vim {{{
	2	"
	3	" AUTHOR: Nico Raffo <nicoraffo@gmail.com>
	4	" MODIFIED: 2010-05-27
	5	" VERSION: 1.1, for Vim 7.0
	6	" LICENSE:
	7	" Conque - pty interaction in Vim
	8	" Copyright (C) 2009-2010 Nico Raffo
	9	"
	10	" MIT License
	11	"
	12	" Permission is hereby granted, free of charge, to any person obtaining a copy
	13	" of this software and associated documentation files (the "Software"), to deal
	14	" in the Software without restriction, including without limitation the rights
	15	" to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
	16	" copies of the Software, and to permit persons to whom the Software is
	17	" furnished to do so, subject to the following conditions:
	18	"
	19	" The above copyright notice and this permission notice shall be included in
	20	" all copies or substantial portions of the Software.
	21	"
	22	" THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
	23	" IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
	24	" FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
	25	" AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
	26	" LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
	27	" OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
	28	" THE SOFTWARE.
	29	" }}}
	30
	31
	32	" **********************************************************************************************************
	33	" ** VIM FUNCTIONS *************************************************************************************
	34	" **********************************************************************************************************
	35
	36	" launch conque
	37	function! conque_term#open(...) "{{{
	38	let command = get(a:000, 0, '')
	39	let hooks = get(a:000, 1, [])
	40
	41	" bare minimum validation
	42	if has('python') != 1
	43	echohl WarningMsg \| echomsg "Conque requires the Python interface to be installed" \| echohl None
	44	return 0
	45	endif
	46	if empty(command)
	47	echohl WarningMsg \| echomsg "No command found" \| echohl None
	48	return 0
	49	else
	50	let l:cargs = split(command, '\s')
	51	if !executable(l:cargs[0])
	52	echohl WarningMsg \| echomsg "Not an executable: " . l:cargs[0] \| echohl None
	53	return 0
	54	endif
	55	endif
	56
	57	" set buffer window options
	58	let g:ConqueTerm_BufName = substitute(command, ' ', '\\ ', 'g') . "\\ -\\ " . g:ConqueTerm_Idx
	59	call conque_term#set_buffer_settings(command, hooks)
	60	let b:ConqueTerm_Var = 'ConqueTerm_' . g:ConqueTerm_Idx
	61	let g:ConqueTerm_Var = 'ConqueTerm_' . g:ConqueTerm_Idx
	62	let g:ConqueTerm_Idx += 1
	63
	64	" open command
	65	try
	66	let l:config = '{"color":' . string(g:ConqueTerm_Color) . ',"TERM":"' . g:ConqueTerm_TERM . '"}'
	67	execute 'python ' . b:ConqueTerm_Var . ' = Conque()'
	68	execute "python " . b:ConqueTerm_Var . ".open('" . conque_term#python_escape(command) . "', " . l:config . ")"
	69	catch
	70	echohl WarningMsg \| echomsg "Unable to open command: " . command \| echohl None
	71	return 0
	72	endtry
	73
	74	" set buffer mappings and auto commands
	75	call conque_term#set_mappings('start')
	76
	77	startinsert!
	78	return 1
	79	endfunction "}}}
	80
	81	" set buffer options
	82	function! conque_term#set_buffer_settings(command, pre_hooks) "{{{
	83
	84	" optional hooks to execute, e.g. 'split'
	85	for h in a:pre_hooks
	86	sil exe h
	87	endfor
	88	sil exe "edit " . g:ConqueTerm_BufName
	89
	90	" buffer settings
	91	setlocal nocompatible " conque won't work in compatible mode
	92	setlocal nopaste " conque won't work in paste mode
	93	setlocal buftype=nofile " this buffer is not a file, you can't save it
	94	setlocal nonumber " hide line numbers
	95	setlocal foldcolumn=0 " reasonable left margin
	96	setlocal nowrap " default to no wrap (esp with MySQL)
	97	setlocal noswapfile " don't bother creating a .swp file
	98	setlocal updatetime=50 " trigger cursorhold event after 50ms / XXX - global
	99	setlocal scrolloff=0 " don't use buffer lines. it makes the 'clear' command not work as expected
	100	setlocal sidescrolloff=0 " don't use buffer lines. it makes the 'clear' command not work as expected
	101	setlocal sidescroll=1 " don't use buffer lines. it makes the 'clear' command not work as expected
	102	setlocal foldmethod=manual " don't fold on {{{}}} and stuff
	103	setlocal bufhidden=hide " when buffer is no longer displayed, don't wipe it out
	104	setfiletype conque_term " useful
	105	sil exe "setlocal syntax=" . g:ConqueTerm_Syntax
	106
	107	endfunction " }}}
	108
	109	" set key mappings and auto commands
	110	function! conque_term#set_mappings(action) "{{{
	111
	112	" set action
	113	if a:action == 'toggle'
	114	if exists('b:conque_on') && b:conque_on == 1
	115	let l:action = 'stop'
	116	echohl WarningMsg \| echomsg "Terminal is paused" \| echohl None
	117	else
	118	let l:action = 'start'
	119	echohl WarningMsg \| echomsg "Terminal is resumed" \| echohl None
	120	endif
	121	else
	122	let l:action = a:action
	123	endif
	124
	125	" if mappings are being removed, add 'un'
	126	let map_modifier = 'nore'
	127	if l:action == 'stop'
	128	let map_modifier = 'un'
	129	endif
	130
	131	" remove all auto commands
	132	if l:action == 'stop'
	133	execute 'autocmd! ' . b:ConqueTerm_Var
	134
	135	else
	136	execute 'augroup ' . b:ConqueTerm_Var
	137
	138	" handle unexpected closing of shell, passes HUP to parent and all child processes
	139	execute 'autocmd ' . b:ConqueTerm_Var . ' BufUnload <buffer> python ' . b:ConqueTerm_Var . '.proc.signal(1)'
	140
	141	" check for resized/scrolled buffer when entering buffer
	142	execute 'autocmd ' . b:ConqueTerm_Var . ' BufEnter <buffer> python ' . b:ConqueTerm_Var . '.update_window_size()'
	143	execute 'autocmd ' . b:ConqueTerm_Var . ' VimResized python ' . b:ConqueTerm_Var . '.update_window_size()'
	144
	145	" set/reset updatetime on entering/exiting buffer
	146	autocmd BufEnter <buffer> set updatetime=50
	147	autocmd BufLeave <buffer> set updatetime=2000
	148
	149	" check for resized/scrolled buffer when entering insert mode
	150	" XXX - messed up since we enter insert mode at each updatetime
	151	"execute 'autocmd InsertEnter <buffer> python ' . b:ConqueTerm_Var . '.screen.align()'
	152
	153	" read more output when this isn't the current buffer
	154	if g:ConqueTerm_ReadUnfocused == 1
	155	execute 'autocmd ' . b:ConqueTerm_Var . ' CursorHold * call conque_term#read_all()'
	156	endif
	157
	158	" poll for more output
	159	sil execute 'autocmd ' . b:ConqueTerm_Var . ' CursorHoldI <buffer> python ' . b:ConqueTerm_Var . '.auto_read()'
	160	endif
	161
	162	" use F22 key to get more input
	163	if l:action == 'start'
	164	sil exe 'i' . map_modifier . 'map <silent> <buffer> <expr> <F22> "\<left>\<right>"'
	165	sil exe 'i' . map_modifier . 'map <silent> <buffer> <expr> <F23> "\<right>\<left>"'
	166	else
	167	sil exe 'i' . map_modifier . 'map <silent> <buffer> <expr> <F22>'
	168	sil exe 'i' . map_modifier . 'map <silent> <buffer> <expr> <F23>'
	169	endif
	170
	171	" map ASCII 1-31
	172	for c in range(1, 31)
	173	" <Esc>
	174	if c == 27
	175	continue
	176	endif
	177	if l:action == 'start'
	178	sil exe 'i' . map_modifier . 'map <silent> <buffer> <C-' . nr2char(64 + c) . '> <C-o>:python ' . b:ConqueTerm_Var . '.write(chr(' . c . '))<CR>'
	179	else
	180	sil exe 'i' . map_modifier . 'map <silent> <buffer> <C-' . nr2char(64 + c) . '>'
	181	endif
	182	endfor
	183	if l:action == 'start'
	184	sil exe 'n' . map_modifier . 'map <silent> <buffer> <C-c> <C-o>:python ' . b:ConqueTerm_Var . '.write(chr(3))<CR>'
	185	else
	186	sil exe 'n' . map_modifier . 'map <silent> <buffer> <C-c>'
	187	endif
	188
	189	" leave insert mode
	190	if !exists('g:ConqueTerm_EscKey') \|\| g:ConqueTerm_EscKey == '<Esc>'
	191	" use <Esc><Esc> to send <Esc> to terminal
	192	if l:action == 'start'
	193	sil exe 'i' . map_modifier . 'map <silent> <buffer> <Esc><Esc> <C-o>:python ' . b:ConqueTerm_Var . '.write(chr(27))<CR>'
	194	else
	195	sil exe 'i' . map_modifier . 'map <silent> <buffer> <Esc><Esc>'
	196	endif
	197	else
	198	" use <Esc> to send <Esc> to terminal
	199	if l:action == 'start'
	200	sil exe 'i' . map_modifier . 'map <silent> <buffer> ' . g:ConqueTerm_EscKey . ' <Esc>'
	201	sil exe 'i' . map_modifier . 'map <silent> <buffer> <Esc> <C-o>:python ' . b:ConqueTerm_Var . '.write(chr(27))<CR>'
	202	else
	203	sil exe 'i' . map_modifier . 'map <silent> <buffer> ' . g:ConqueTerm_EscKey
	204	sil exe 'i' . map_modifier . 'map <silent> <buffer> <Esc>'
	205	endif
	206	endif
	207
	208	" Map <C-w> in insert mode
	209	if exists('g:ConqueTerm_CWInsert') && g:ConqueTerm_CWInsert == 1
	210	inoremap <silent> <buffer> <C-w>j <Esc><C-w>j
	211	inoremap <silent> <buffer> <C-w>k <Esc><C-w>k
	212	inoremap <silent> <buffer> <C-w>h <Esc><C-w>h
	213	inoremap <silent> <buffer> <C-w>l <Esc><C-w>l
	214	inoremap <silent> <buffer> <C-w>w <Esc><C-w>w
	215	endif
	216
	217	" map ASCII 33-127
	218	for i in range(33, 127)
	219	" <Bar>
	220	if i == 124
	221	if l:action == 'start'
	222	sil exe "i" . map_modifier . "map <silent> <buffer> <Bar> <C-o>:python " . b:ConqueTerm_Var . ".write(chr(124))<CR>"
	223	else
	224	sil exe "i" . map_modifier . "map <silent> <buffer> <Bar>"
	225	endif
	226	continue
	227	endif
	228	if l:action == 'start'
	229	sil exe "i" . map_modifier . "map <silent> <buffer> " . nr2char(i) . " <C-o>:python " . b:ConqueTerm_Var . ".write(chr(" . i . "))<CR>"
	230	else
	231	sil exe "i" . map_modifier . "map <silent> <buffer> " . nr2char(i)
	232	endif
	233	endfor
	234
	235	" map ASCII 128-255
	236	for i in range(128, 255)
	237	if l:action == 'start'
	238	sil exe "i" . map_modifier . "map <silent> <buffer> " . nr2char(i) . " <C-o>:python " . b:ConqueTerm_Var . ".write('" . nr2char(i) . "')<CR>"
	239	else
	240	sil exe "i" . map_modifier . "map <silent> <buffer> " . nr2char(i)
	241	endif
	242	endfor
	243
	244	" Special cases
	245	if l:action == 'start'
	246	sil exe 'i' . map_modifier . 'map <silent> <buffer> <BS> <C-o>:python ' . b:ConqueTerm_Var . '.write(u"\u0008")<CR>'
	247	sil exe 'i' . map_modifier . 'map <silent> <buffer> <Space> <C-o>:python ' . b:ConqueTerm_Var . '.write(" ")<CR>'
	248	sil exe 'i' . map_modifier . 'map <silent> <buffer> <Up> <C-o>:python ' . b:ConqueTerm_Var . '.write(u"\u001b[A")<CR>'
	249	sil exe 'i' . map_modifier . 'map <silent> <buffer> <Down> <C-o>:python ' . b:ConqueTerm_Var . '.write(u"\u001b[B")<CR>'
	250	sil exe 'i' . map_modifier . 'map <silent> <buffer> <Right> <C-o>:python ' . b:ConqueTerm_Var . '.write(u"\u001b[C")<CR>'
	251	sil exe 'i' . map_modifier . 'map <silent> <buffer> <Left> <C-o>:python ' . b:ConqueTerm_Var . '.write(u"\u001b[D")<CR>'
	252	else
	253	sil exe 'i' . map_modifier . 'map <silent> <buffer> <BS>'
	254	sil exe 'i' . map_modifier . 'map <silent> <buffer> <Space>'
	255	sil exe 'i' . map_modifier . 'map <silent> <buffer> <Up>'
	256	sil exe 'i' . map_modifier . 'map <silent> <buffer> <Down>'
	257	sil exe 'i' . map_modifier . 'map <silent> <buffer> <Right>'
	258	sil exe 'i' . map_modifier . 'map <silent> <buffer> <Left>'
	259	endif
	260
	261	" send selected text into conque
	262	if l:action == 'start'
	263	sil exe 'v' . map_modifier . 'map <silent> <F9> :<C-u>call conque_term#send_selected(visualmode())<CR>'
	264	else
	265	sil exe 'v' . map_modifier . 'map <silent> <F9>'
	266	endif
	267
	268	" remap paste keys
	269	if l:action == 'start'
	270	sil exe 'n' . map_modifier . 'map <silent> <buffer> p :python ' . b:ConqueTerm_Var . '.write(vim.eval("@@"))<CR>a'
	271	sil exe 'n' . map_modifier . 'map <silent> <buffer> P :python ' . b:ConqueTerm_Var . '.write(vim.eval("@@"))<CR>a'
	272	sil exe 'n' . map_modifier . 'map <silent> <buffer> ]p :python ' . b:ConqueTerm_Var . '.write(vim.eval("@@"))<CR>a'
	273	sil exe 'n' . map_modifier . 'map <silent> <buffer> [p :python ' . b:ConqueTerm_Var . '.write(vim.eval("@@"))<CR>a'
	274	else
	275	sil exe 'n' . map_modifier . 'map <silent> <buffer> p'
	276	sil exe 'n' . map_modifier . 'map <silent> <buffer> P'
	277	sil exe 'n' . map_modifier . 'map <silent> <buffer> ]p'
	278	sil exe 'n' . map_modifier . 'map <silent> <buffer> [p'
	279	endif
	280	if has('gui_running')
	281	if l:action == 'start'
	282	sil exe 'i' . map_modifier . 'map <buffer> <S-Insert> <Esc>:<C-u>python ' . b:ConqueTerm_Var . ".write(vim.eval('@+'))<CR>a"
	283	else
	284	sil exe 'i' . map_modifier . 'map <buffer> <S-Insert>'
	285	endif
	286	endif
	287
	288	" disable other normal mode keys which insert text
	289	if l:action == 'start'
	290	sil exe 'n' . map_modifier . 'map <silent> <buffer> r :echo "Replace mode disabled in shell."<CR>'
	291	sil exe 'n' . map_modifier . 'map <silent> <buffer> R :echo "Replace mode disabled in shell."<CR>'
	292	sil exe 'n' . map_modifier . 'map <silent> <buffer> c :echo "Change mode disabled in shell."<CR>'
	293	sil exe 'n' . map_modifier . 'map <silent> <buffer> C :echo "Change mode disabled in shell."<CR>'
	294	sil exe 'n' . map_modifier . 'map <silent> <buffer> s :echo "Change mode disabled in shell."<CR>'
	295	sil exe 'n' . map_modifier . 'map <silent> <buffer> S :echo "Change mode disabled in shell."<CR>'
	296	else
	297	sil exe 'n' . map_modifier . 'map <silent> <buffer> r'
	298	sil exe 'n' . map_modifier . 'map <silent> <buffer> R'
	299	sil exe 'n' . map_modifier . 'map <silent> <buffer> c'
	300	sil exe 'n' . map_modifier . 'map <silent> <buffer> C'
	301	sil exe 'n' . map_modifier . 'map <silent> <buffer> s'
	302	sil exe 'n' . map_modifier . 'map <silent> <buffer> S'
	303	endif
	304
	305	" set conque as on or off
	306	if l:action == 'start'
	307	let b:conque_on = 1
	308	else
	309	let b:conque_on = 0
	310	endif
	311
	312	" map command to start stop the shell
	313	if a:action == 'start'
	314	nnoremap <F5> :<C-u>call conque_term#set_mappings('toggle')<CR>
	315	endif
	316
	317	endfunction " }}}
	318
	319
	320	" send selected text from another buffer
	321	function! conque_term#send_selected(type) "{{{
	322	let reg_save = @@
	323
	324	" save user's sb settings
	325	let sb_save = &switchbuf
	326	set switchbuf=usetab
	327
	328	" yank current selection
	329	sil exe "normal! `<" . a:type . "`>y"
	330
	331	" format yanked text
	332	let @@ = substitute(@@, '^[\r\n]*', '', '')
	333	let @@ = substitute(@@, '[\r\n]*$', '', '')
	334
	335	" execute yanked text
	336	sil exe ":sb " . g:ConqueTerm_BufName
	337	sil exe 'python ' . g:ConqueTerm_Var . '.paste_selection()'
	338
	339	" reset original values
	340	let @@ = reg_save
	341	sil exe 'set switchbuf=' . sb_save
	342
	343	" scroll buffer left
	344	startinsert!
	345	normal 0zH
	346	endfunction "}}}
	347
	348	" read from all known conque buffers
	349	function! conque_term#read_all() "{{{
	350	" don't run this if we're in a conque buffer
	351	if exists('b:ConqueTerm_Var')
	352	return
	353	endif
	354
	355	try
	356	for i in range(1, g:ConqueTerm_Idx - 1)
	357	execute 'python ConqueTerm_' . string(i) . '.read(1)'
	358	endfor
	359	catch
	360	" probably a deleted buffer
	361	endtry
	362
	363	" restart updatetime
	364	call feedkeys("f\e")
	365	endfunction "}}}
	366
	367	" util function to add enough \s to pass a string to python
	368	function! conque_term#python_escape(input) "{{{
	369	let l:cleaned = a:input
	370	let l:cleaned = substitute(l:cleaned, '\\', '\\\\', 'g')
	371	let l:cleaned = substitute(l:cleaned, '\n', '\\n', 'g')
	372	let l:cleaned = substitute(l:cleaned, '\r', '\\r', 'g')
	373	let l:cleaned = substitute(l:cleaned, "'", "\\\\'", 'g')
	374	return l:cleaned
	375	endfunction "}}}
	376
	377	" **********************************************************************************************************
	378	" ** PYTHON ********************************************************************************************
	379	" **********************************************************************************************************
	380
	381	if has('python')
	382
	383	python << EOF
	384
	385	import vim, re, time, math
	386
	387	# CONFIG CONSTANTS {{{
	388
	389	CONQUE_CTL = {
	390	7:'bel', # bell
	391	8:'bs', # backspace
	392	9:'tab', # tab
	393	10:'nl', # new line
	394	13:'cr' # carriage return
	395	}
	396	# 11 : 'vt', # vertical tab
	397	# 12 : 'ff', # form feed
	398	# 14 : 'so', # shift out
	399	# 15 : 'si' # shift in
	400
	401	# Escape sequences
	402	CONQUE_ESCAPE = {
	403	'm':'font',
	404	'J':'clear_screen',
	405	'K':'clear_line',
	406	'@':'add_spaces',
	407	'A':'cursor_up',
	408	'B':'cursor_down',
	409	'C':'cursor_right',
	410	'D':'cursor_left',
	411	'G':'cursor_to_column',
	412	'H':'cursor',
	413	'P':'delete_chars',
	414	'f':'cursor',
	415	'g':'tab_clear',
	416	'r':'set_coords',
	417	'h':'set',
	418	'l':'reset'
	419	}
	420	# 'L':'insert_lines',
	421	# 'M':'delete_lines',
	422	# 'd':'cusor_vpos',
	423
	424	# Alternate escape sequences, no [
	425	CONQUE_ESCAPE_PLAIN = {
	426	'D':'scroll_up',
	427	'E':'next_line',
	428	'H':'set_tab',
	429	'M':'scroll_down'
	430	}
	431	# 'N':'single_shift_2',
	432	# 'O':'single_shift_3',
	433	# '=':'alternate_keypad',
	434	# '>':'numeric_keypad',
	435	# '7':'save_cursor',
	436	# '8':'restore_cursor',
	437
	438	# Uber alternate escape sequences, with # or ?
	439	CONQUE_ESCAPE_QUESTION = {
	440	'1h':'new_line_mode',
	441	'3h':'132_cols',
	442	'4h':'smooth_scrolling',
	443	'5h':'reverse_video',
	444	'6h':'relative_origin',
	445	'7h':'set_auto_wrap',
	446	'8h':'set_auto_repeat',
	447	'9h':'set_interlacing_mode',
	448	'1l':'set_cursor_key',
	449	'2l':'set_vt52',
	450	'3l':'80_cols',
	451	'4l':'set_jump_scrolling',
	452	'5l':'normal_video',
	453	'6l':'absolute_origin',
	454	'7l':'reset_auto_wrap',
	455	'8l':'reset_auto_repeat',
	456	'9l':'reset_interlacing_mode'
	457	}
	458
	459	CONQUE_ESCAPE_HASH = {
	460	'8':'screen_alignment_test'
	461	}
	462	# '3':'double_height_top',
	463	# '4':'double_height_bottom',
	464	# '5':'single_height_single_width',
	465	# '6':'single_height_double_width',
	466
	467	# Font codes {{{
	468	CONQUE_FONT = {
	469	0: {'description':'Normal (default)', 'attributes': {'cterm':'NONE','ctermfg':'NONE','ctermbg':'NONE','gui':'NONE','guifg':'NONE','guibg':'NONE'}, 'normal':True},
	470	1: {'description':'Bold', 'attributes': {'cterm':'BOLD','gui':'BOLD'}, 'normal':False},
	471	4: {'description':'Underlined', 'attributes': {'cterm':'UNDERLINE','gui':'UNDERLINE'}, 'normal':False},
	472	5: {'description':'Blink (appears as Bold)', 'attributes': {'cterm':'BOLD','gui':'BOLD'}, 'normal':False},
	473	7: {'description':'Inverse', 'attributes': {'cterm':'REVERSE','gui':'REVERSE'}, 'normal':False},
	474	8: {'description':'Invisible (hidden)', 'attributes': {'ctermfg':'0','ctermbg':'0','guifg':'#000000','guibg':'#000000'}, 'normal':False},
	475	22: {'description':'Normal (neither bold nor faint)', 'attributes': {'cterm':'NONE','gui':'NONE'}, 'normal':True},
	476	24: {'description':'Not underlined', 'attributes': {'cterm':'NONE','gui':'NONE'}, 'normal':True},
	477	25: {'description':'Steady (not blinking)', 'attributes': {'cterm':'NONE','gui':'NONE'}, 'normal':True},
	478	27: {'description':'Positive (not inverse)', 'attributes': {'cterm':'NONE','gui':'NONE'}, 'normal':True},
	479	28: {'description':'Visible (not hidden)', 'attributes': {'ctermfg':'NONE','ctermbg':'NONE','guifg':'NONE','guibg':'NONE'}, 'normal':True},
	480	30: {'description':'Set foreground color to Black', 'attributes': {'ctermfg':'16','guifg':'#000000'}, 'normal':False},
	481	31: {'description':'Set foreground color to Red', 'attributes': {'ctermfg':'1','guifg':'#ff0000'}, 'normal':False},
	482	32: {'description':'Set foreground color to Green', 'attributes': {'ctermfg':'2','guifg':'#00ff00'}, 'normal':False},
	483	33: {'description':'Set foreground color to Yellow', 'attributes': {'ctermfg':'3','guifg':'#ffff00'}, 'normal':False},
	484	34: {'description':'Set foreground color to Blue', 'attributes': {'ctermfg':'4','guifg':'#0000ff'}, 'normal':False},
	485	35: {'description':'Set foreground color to Magenta', 'attributes': {'ctermfg':'5','guifg':'#990099'}, 'normal':False},
	486	36: {'description':'Set foreground color to Cyan', 'attributes': {'ctermfg':'6','guifg':'#009999'}, 'normal':False},
	487	37: {'description':'Set foreground color to White', 'attributes': {'ctermfg':'7','guifg':'#ffffff'}, 'normal':False},
	488	39: {'description':'Set foreground color to default (original)', 'attributes': {'ctermfg':'NONE','guifg':'NONE'}, 'normal':True},
	489	40: {'description':'Set background color to Black', 'attributes': {'ctermbg':'16','guibg':'#000000'}, 'normal':False},
	490	41: {'description':'Set background color to Red', 'attributes': {'ctermbg':'1','guibg':'#ff0000'}, 'normal':False},
	491	42: {'description':'Set background color to Green', 'attributes': {'ctermbg':'2','guibg':'#00ff00'}, 'normal':False},
	492	43: {'description':'Set background color to Yellow', 'attributes': {'ctermbg':'3','guibg':'#ffff00'}, 'normal':False},
	493	44: {'description':'Set background color to Blue', 'attributes': {'ctermbg':'4','guibg':'#0000ff'}, 'normal':False},
	494	45: {'description':'Set background color to Magenta', 'attributes': {'ctermbg':'5','guibg':'#990099'}, 'normal':False},
	495	46: {'description':'Set background color to Cyan', 'attributes': {'ctermbg':'6','guibg':'#009999'}, 'normal':False},
	496	47: {'description':'Set background color to White', 'attributes': {'ctermbg':'7','guibg':'#ffffff'}, 'normal':False},
	497	49: {'description':'Set background color to default (original).', 'attributes': {'ctermbg':'NONE','guibg':'NONE'}, 'normal':True},
	498	90: {'description':'Set foreground color to Black', 'attributes': {'ctermfg':'8','guifg':'#000000'}, 'normal':False},
	499	91: {'description':'Set foreground color to Red', 'attributes': {'ctermfg':'9','guifg':'#ff0000'}, 'normal':False},
	500	92: {'description':'Set foreground color to Green', 'attributes': {'ctermfg':'10','guifg':'#00ff00'}, 'normal':False},
	501	93: {'description':'Set foreground color to Yellow', 'attributes': {'ctermfg':'11','guifg':'#ffff00'}, 'normal':False},
	502	94: {'description':'Set foreground color to Blue', 'attributes': {'ctermfg':'12','guifg':'#0000ff'}, 'normal':False},
	503	95: {'description':'Set foreground color to Magenta', 'attributes': {'ctermfg':'13','guifg':'#990099'}, 'normal':False},
	504	96: {'description':'Set foreground color to Cyan', 'attributes': {'ctermfg':'14','guifg':'#009999'}, 'normal':False},
	505	97: {'description':'Set foreground color to White', 'attributes': {'ctermfg':'15','guifg':'#ffffff'}, 'normal':False},
	506	100: {'description':'Set background color to Black', 'attributes': {'ctermbg':'8','guibg':'#000000'}, 'normal':False},
	507	101: {'description':'Set background color to Red', 'attributes': {'ctermbg':'9','guibg':'#ff0000'}, 'normal':False},
	508	102: {'description':'Set background color to Green', 'attributes': {'ctermbg':'10','guibg':'#00ff00'}, 'normal':False},
	509	103: {'description':'Set background color to Yellow', 'attributes': {'ctermbg':'11','guibg':'#ffff00'}, 'normal':False},
	510	104: {'description':'Set background color to Blue', 'attributes': {'ctermbg':'12','guibg':'#0000ff'}, 'normal':False},
	511	105: {'description':'Set background color to Magenta', 'attributes': {'ctermbg':'13','guibg':'#990099'}, 'normal':False},
	512	106: {'description':'Set background color to Cyan', 'attributes': {'ctermbg':'14','guibg':'#009999'}, 'normal':False},
	513	107: {'description':'Set background color to White', 'attributes': {'ctermbg':'15','guibg':'#ffffff'}, 'normal':False}
	514	}
	515	# }}}
	516
	517	# regular expression matching (almost) all control sequences
	518	CONQUE_SEQ_REGEX = re.compile(ur"(\u001b\[?\??#?[0-9;][a-zA-Z@]\|\u001b\][0-9];.?\u0007\|[\u0007-\u000f])", re.UNICODE)
	519	CONQUE_SEQ_REGEX_CTL = re.compile(ur"^[\u0007-\u000f]$", re.UNICODE)
	520	CONQUE_SEQ_REGEX_CSI = re.compile(ur"^\u001b\[", re.UNICODE)
	521	CONQUE_SEQ_REGEX_TITLE = re.compile(ur"^\u001b\]", re.UNICODE)
	522	CONQUE_SEQ_REGEX_HASH = re.compile(ur"^\u001b#", re.UNICODE)
	523	CONQUE_SEQ_REGEX_ESC = re.compile(ur"^\u001b", re.UNICODE)
	524
	525	# match table output
	526	CONQUE_TABLE_OUTPUT = re.compile("^\s\\|\s.\s\\|\s$\|^\s\+[=+-]+\+\s*$")
	527
	528	# }}}
	529
	530	###################################################################################################
	531	class Conque:
	532
	533	# CLASS PROPERTIES {{{
	534
	535	# screen object
	536	screen = None
	537
	538	# subprocess object
	539	proc = None
	540
	541	# terminal dimensions and scrolling region
	542	columns = 80 # same as $COLUMNS
	543	lines = 24 # same as $LINES
	544	working_columns = 80 # can be changed by CSI ? 3 l/h
	545	working_lines = 24 # can be changed by CSI r
	546
	547	# top/bottom of the scroll region
	548	top = 1 # relative to top of screen
	549	bottom = 24 # relative to top of screen
	550
	551	# cursor position
	552	l = 1 # current cursor line
	553	c = 1 # current cursor column
	554
	555	# autowrap mode
	556	autowrap = True
	557
	558	# absolute coordinate mode
	559	absolute_coords = True
	560
	561	# tabstop positions
	562	tabstops = []
	563
	564	# enable colors
	565	enable_colors = True
	566
	567	# color changes
	568	color_changes = {}
	569
	570	# color history
	571	color_history = {}
	572
	573	# don't wrap table output
	574	unwrap_tables = True
	575
	576	# wrap CUF/CUB around line breaks
	577	wrap_cursor = False
	578
	579	# }}}
	580
	581	# constructor
	582	def __init__(self): # {{{
	583	self.screen = ConqueScreen()
	584	# }}}
	585
	586	# start program and initialize this instance
	587	def open(self, command, options): # {{{
	588
	589	# int vars
	590	self.columns = vim.current.window.width
	591	self.lines = vim.current.window.height
	592	self.working_columns = vim.current.window.width
	593	self.working_lines = vim.current.window.height
	594	self.bottom = vim.current.window.height
	595
	596	# init color
	597	self.enable_colors = options['color']
	598
	599	# init tabstops
	600	self.init_tabstops()
	601
	602	# open command
	603	self.proc = ConqueSubprocess()
	604	self.proc.open(command, { 'TERM' : options['TERM'], 'CONQUE' : '1', 'LINES' : str(self.lines), 'COLUMNS' : str(self.columns)})
	605	# }}}
	606
	607	# write to pty
	608	def write(self, input): # {{{
	609
	610
	611	# check if window size has changed
	612	self.update_window_size()
	613
	614	# write and read
	615	self.proc.write(input)
	616	self.read(1)
	617	# }}}
	618
	619	# read from pty, and update buffer
	620	def read(self, timeout = 1): # {{{
	621	# read from subprocess
	622	output = self.proc.read(timeout)
	623	# and strip null chars
	624	output = output.replace(chr(0), '')
	625
	626	if output == '':
	627	return
	628
	629
	630
	631
	632
	633	chunks = CONQUE_SEQ_REGEX.split(output)
	634
	635
	636
	637
	638
	639	# don't go through all the csi regex if length is one (no matches)
	640	if len(chunks) == 1:
	641
	642	self.plain_text(chunks[0])
	643
	644	else:
	645	for s in chunks:
	646	if s == '':
	647	continue
	648
	649
	650
	651
	652
	653
	654	# Check for control character match {{{
	655	if CONQUE_SEQ_REGEX_CTL.match(s[0]):
	656
	657	nr = ord(s[0])
	658	if nr in CONQUE_CTL:
	659	getattr(self, 'ctl_' + CONQUE_CTL[nr])()
	660	else:
	661
	662	pass
	663	# }}}
	664
	665	# check for escape sequence match {{{
	666	elif CONQUE_SEQ_REGEX_CSI.match(s):
	667
	668	if s[-1] in CONQUE_ESCAPE:
	669	csi = self.parse_csi(s[2:])
	670
	671	getattr(self, 'csi_' + CONQUE_ESCAPE[s[-1]])(csi)
	672	else:
	673
	674	pass
	675	# }}}
	676
	677	# check for title match {{{
	678	elif CONQUE_SEQ_REGEX_TITLE.match(s):
	679
	680	self.change_title(s[2], s[4:-1])
	681	# }}}
	682
	683	# check for hash match {{{
	684	elif CONQUE_SEQ_REGEX_HASH.match(s):
	685
	686	if s[-1] in CONQUE_ESCAPE_HASH:
	687	getattr(self, 'hash_' + CONQUE_ESCAPE_HASH[s[-1]])()
	688	else:
	689
	690	pass
	691	# }}}
	692
	693	# check for other escape match {{{
	694	elif CONQUE_SEQ_REGEX_ESC.match(s):
	695
	696	if s[-1] in CONQUE_ESCAPE_PLAIN:
	697	getattr(self, 'esc_' + CONQUE_ESCAPE_PLAIN[s[-1]])()
	698	else:
	699
	700	pass
	701	# }}}
	702
	703	# else process plain text {{{
	704	else:
	705	self.plain_text(s)
	706	# }}}
	707
	708	# set cursor position
	709	self.screen.set_cursor(self.l, self.c)
	710
	711	vim.command('redraw')
	712
	713
	714	# }}}
	715
	716	# for polling
	717	def auto_read(self): # {{{
	718	self.read(1)
	719	if self.c == 1:
	720	vim.command('call feedkeys("\<F23>", "t")')
	721	else:
	722	vim.command('call feedkeys("\<F22>", "t")')
	723	self.screen.set_cursor(self.l, self.c)
	724	# }}}
	725
	726	###############################################################################################
	727	# Plain text # {{{
	728
	729	def plain_text(self, input):
	730
	731	current_line = self.screen[self.l]
	732
	733	if len(current_line) < self.working_columns:
	734	current_line = current_line + ' ' * (self.c - len(current_line))
	735
	736	# if line is wider than screen
	737	if self.c + len(input) - 1 > self.working_columns:
	738	# Table formatting hack
	739	if self.unwrap_tables and CONQUE_TABLE_OUTPUT.match(input):
	740	self.screen[self.l] = current_line[ : self.c - 1] + input + current_line[ self.c + len(input) - 1 : ]
	741	self.apply_color(self.c, self.c + len(input))
	742	self.c += len(input)
	743	return
	744
	745	diff = self.c + len(input) - self.working_columns - 1
	746	# if autowrap is enabled
	747	if self.autowrap:
	748	self.screen[self.l] = current_line[ : self.c - 1] + input[ : -1 * diff ]
	749	self.apply_color(self.c, self.working_columns)
	750	self.ctl_nl()
	751	self.ctl_cr()
	752	remaining = input[ -1 * diff : ]
	753
	754	self.plain_text(remaining)
	755	else:
	756	self.screen[self.l] = current_line[ : self.c - 1] + input[ : -1 * diff - 1 ] + input[-1]
	757	self.apply_color(self.c, self.working_columns)
	758	self.c = self.working_columns
	759
	760	# no autowrap
	761	else:
	762	self.screen[self.l] = current_line[ : self.c - 1] + input + current_line[ self.c + len(input) - 1 : ]
	763	self.apply_color(self.c, self.c + len(input))
	764	self.c += len(input)
	765
	766	def apply_color(self, start, end):
	767
	768
	769	# stop here if coloration is disabled
	770	if not self.enable_colors:
	771	return
	772
	773	real_line = self.screen.get_real_line(self.l)
	774
	775	# check for previous overlapping coloration
	776
	777	to_del = []
	778	if self.color_history.has_key(real_line):
	779	for i in range(len(self.color_history[real_line])):
	780	syn = self.color_history[real_line][i]
	781
	782	if syn['start'] >= start and syn['start'] < end:
	783
	784	vim.command('syn clear ' + syn['name'])
	785	to_del.append(i)
	786	# outside
	787	if syn['end'] > end:
	788
	789	self.exec_highlight(real_line, end, syn['end'], syn['highlight'])
	790	elif syn['end'] > start and syn['end'] <= end:
	791
	792	vim.command('syn clear ' + syn['name'])
	793	to_del.append(i)
	794	# outside
	795	if syn['start'] < start:
	796
	797	self.exec_highlight(real_line, syn['start'], start, syn['highlight'])
	798
	799	if len(to_del) > 0:
	800	to_del.reverse()
	801	for di in to_del:
	802	del self.color_history[real_line][di]
	803
	804	# if there are no new colors
	805	if len(self.color_changes) == 0:
	806	return
	807
	808	highlight = ''
	809	for attr in self.color_changes.keys():
	810	highlight = highlight + ' ' + attr + '=' + self.color_changes[attr]
	811
	812	# execute the highlight
	813	self.exec_highlight(real_line, start, end, highlight)
	814
	815	def exec_highlight(self, real_line, start, end, highlight):
	816	unique_key = str(self.proc.pid)
	817
	818	syntax_name = 'EscapeSequenceAt_' + unique_key + '_' + str(self.l) + '_' + str(start) + '_' + str(len(self.color_history) + 1)
	819	syntax_options = ' contains=ALLBUT,ConqueString,MySQLString,MySQLKeyword oneline'
	820	syntax_region = 'syntax match ' + syntax_name + ' /\%' + str(real_line) + 'l\%>' + str(start - 1) + 'c.*\%<' + str(end + 1) + 'c/' + syntax_options
	821	syntax_highlight = 'highlight ' + syntax_name + highlight
	822
	823	vim.command(syntax_region)
	824	vim.command(syntax_highlight)
	825
	826	# add syntax name to history
	827	if not self.color_history.has_key(real_line):
	828	self.color_history[real_line] = []
	829
	830	self.color_history[real_line].append({'name':syntax_name, 'start':start, 'end':end, 'highlight':highlight})
	831
	832	# }}}
	833
	834	###############################################################################################
	835	# Control functions {{{
	836
	837	def ctl_nl(self):
	838	# if we're in a scrolling region, scroll instead of moving cursor down
	839	if self.lines != self.working_lines and self.l == self.bottom:
	840	del self.screen[self.top]
	841	self.screen.insert(self.bottom, '')
	842	elif self.l == self.bottom:
	843	self.screen.append('')
	844	else:
	845	self.l += 1
	846
	847	self.color_changes = {}
	848
	849	def ctl_cr(self):
	850	self.c = 1
	851
	852	self.color_changes = {}
	853
	854	def ctl_bs(self):
	855	if self.c > 1:
	856	self.c += -1
	857
	858	def ctl_bel(self):
	859	print 'BELL'
	860
	861	def ctl_tab(self):
	862	# default tabstop location
	863	ts = self.working_columns
	864
	865	# check set tabstops
	866	for i in range(self.c, len(self.tabstops)):
	867	if self.tabstops[i]:
	868	ts = i + 1
	869	break
	870
	871
	872
	873	self.c = ts
	874
	875	# }}}
	876
	877	###############################################################################################
	878	# CSI functions {{{
	879
	880	def csi_font(self, csi): # {{{
	881	if not self.enable_colors:
	882	return
	883
	884	# defaults to 0
	885	if len(csi['vals']) == 0:
	886	csi['vals'] = [0]
	887
	888	# 256 xterm color foreground
	889	if len(csi['vals']) == 3 and csi['vals'][0] == 38 and csi['vals'][1] == 5:
	890	self.color_changes['ctermfg'] = str(csi['vals'][2])
	891	self.color_changes['guifg'] = '#' + self.xterm_to_rgb(csi['vals'][2])
	892
	893	# 256 xterm color background
	894	elif len(csi['vals']) == 3 and csi['vals'][0] == 48 and csi['vals'][1] == 5:
	895	self.color_changes['ctermbg'] = str(csi['vals'][2])
	896	self.color_changes['guibg'] = '#' + self.xterm_to_rgb(csi['vals'][2])
	897
	898	# 16 colors
	899	else:
	900	for val in csi['vals']:
	901	if CONQUE_FONT.has_key(val):
	902
	903	# ignore starting normal colors
	904	if CONQUE_FONT[val]['normal'] and len(self.color_changes) == 0:
	905
	906	continue
	907	# clear color changes
	908	elif CONQUE_FONT[val]['normal']:
	909
	910	self.color_changes = {}
	911	# save these color attributes for next plain_text() call
	912	else:
	913
	914	for attr in CONQUE_FONT[val]['attributes'].keys():
	915	if self.color_changes.has_key(attr) and (attr == 'cterm' or attr == 'gui'):
	916	self.color_changes[attr] += ',' + CONQUE_FONT[val]['attributes'][attr]
	917	else:
	918	self.color_changes[attr] = CONQUE_FONT[val]['attributes'][attr]
	919	# }}}
	920
	921	def csi_clear_line(self, csi): # {{{
	922
	923
	924	# this escape defaults to 0
	925	if len(csi['vals']) == 0:
	926	csi['val'] = 0
	927
	928
	929
	930
	931	# 0 means cursor right
	932	if csi['val'] == 0:
	933	self.screen[self.l] = self.screen[self.l][0 : self.c - 1]
	934
	935	# 1 means cursor left
	936	elif csi['val'] == 1:
	937	self.screen[self.l] = ' ' * (self.c) + self.screen[self.l][self.c : ]
	938
	939	# clear entire line
	940	elif csi['val'] == 2:
	941	self.screen[self.l] = ''
	942
	943	# clear colors
	944	if csi['val'] == 2 or (csi['val'] == 0 and self.c == 1):
	945	real_line = self.screen.get_real_line(self.l)
	946	if self.color_history.has_key(real_line):
	947	for syn in self.color_history[real_line]:
	948	vim.command('syn clear ' + syn['name'])
	949
	950
	951
	952	# }}}
	953
	954	def csi_cursor_right(self, csi): # {{{
	955	# we use 1 even if escape explicitly specifies 0
	956	if csi['val'] == 0:
	957	csi['val'] = 1
	958
	959
	960
	961
	962	if self.wrap_cursor and self.c + csi['val'] > self.working_columns:
	963	self.l += int(math.floor( (self.c + csi['val']) / self.working_columns ))
	964	self.c = (self.c + csi['val']) % self.working_columns
	965	return
	966
	967	self.c = self.bound(self.c + csi['val'], 1, self.working_columns)
	968	# }}}
	969
	970	def csi_cursor_left(self, csi): # {{{
	971	# we use 1 even if escape explicitly specifies 0
	972	if csi['val'] == 0:
	973	csi['val'] = 1
	974
	975	if self.wrap_cursor and csi['val'] >= self.c:
	976	self.l += int(math.floor( (self.c - csi['val']) / self.working_columns ))
	977	self.c = self.working_columns - (csi['val'] - self.c) % self.working_columns
	978	return
	979
	980	self.c = self.bound(self.c - csi['val'], 1, self.working_columns)
	981	# }}}
	982
	983	def csi_cursor_to_column(self, csi): # {{{
	984	self.c = self.bound(csi['val'], 1, self.working_columns)
	985	# }}}
	986
	987	def csi_cursor_up(self, csi): # {{{
	988	self.l = self.bound(self.l - csi['val'], self.top, self.bottom)
	989
	990	self.color_changes = {}
	991	# }}}
	992
	993	def csi_cursor_down(self, csi): # {{{
	994	self.l = self.bound(self.l + csi['val'], self.top, self.bottom)
	995
	996	self.color_changes = {}
	997	# }}}
	998
	999	def csi_clear_screen(self, csi): # {{{
	1000	# default to 0
	1001	if len(csi['vals']) == 0:
	1002	csi['val'] = 0
	1003
	1004	# 2 == clear entire screen
	1005	if csi['val'] == 2:
	1006	self.l = 1
	1007	self.c = 1
	1008	self.screen.clear()
	1009
	1010	# 0 == clear down
	1011	elif csi['val'] == 0:
	1012	for l in range(self.bound(self.l + 1, 1, self.lines), self.lines + 1):
	1013	self.screen[l] = ''
	1014
	1015	# clear end of current line
	1016	self.csi_clear_line(self.parse_csi('K'))
	1017
	1018	# 1 == clear up
	1019	elif csi['val'] == 1:
	1020	for l in range(1, self.bound(self.l, 1, self.lines + 1)):
	1021	self.screen[l] = ''
	1022
	1023	# clear beginning of current line
	1024	self.csi_clear_line(self.parse_csi('1K'))
	1025
	1026	# clear coloration
	1027	if csi['val'] == 2 or csi['val'] == 0:
	1028	real_line = self.screen.get_real_line(self.l)
	1029	for line in self.color_history.keys():
	1030	if line >= real_line:
	1031	for syn in self.color_history[line]:
	1032	vim.command('syn clear ' + syn['name'])
	1033
	1034	self.color_changes = {}
	1035	# }}}
	1036
	1037	def csi_delete_chars(self, csi): # {{{
	1038	self.screen[self.l] = self.screen[self.l][ : self.c ] + self.screen[self.l][ self.c + csi['val'] : ]
	1039	# }}}
	1040
	1041	def csi_add_spaces(self, csi): # {{{
	1042	self.screen[self.l] = self.screen[self.l][ : self.c - 1] + ' ' * csi['val'] + self.screen[self.l][self.c : ]
	1043	# }}}
	1044
	1045	def csi_cursor(self, csi): # {{{
	1046	if len(csi['vals']) == 2:
	1047	new_line = csi['vals'][0]
	1048	new_col = csi['vals'][1]
	1049	else:
	1050	new_line = 1
	1051	new_col = 1
	1052
	1053	if self.absolute_coords:
	1054	self.l = self.bound(new_line, 1, self.lines)
	1055	else:
	1056	self.l = self.bound(self.top + new_line - 1, self.top, self.bottom)
	1057
	1058	self.c = self.bound(new_col, 1, self.working_columns)
	1059	if self.c > len(self.screen[self.l]):
	1060	self.screen[self.l] = self.screen[self.l] + ' ' * (self.c - len(self.screen[self.l]))
	1061
	1062	# }}}
	1063
	1064	def csi_set_coords(self, csi): # {{{
	1065	if len(csi['vals']) == 2:
	1066	new_start = csi['vals'][0]
	1067	new_end = csi['vals'][1]
	1068	else:
	1069	new_start = 1
	1070	new_end = vim.current.window.height
	1071
	1072	self.top = new_start
	1073	self.bottom = new_end
	1074	self.working_lines = new_end - new_start + 1
	1075
	1076	# if cursor is outside scrolling region, reset it
	1077	if self.l < self.top:
	1078	self.l = self.top
	1079	elif self.l > self.bottom:
	1080	self.l = self.bottom
	1081
	1082	self.color_changes = {}
	1083	# }}}
	1084
	1085	def csi_tab_clear(self, csi): # {{{
	1086	# this escape defaults to 0
	1087	if len(csi['vals']) == 0:
	1088	csi['val'] = 0
	1089
	1090
	1091
	1092	if csi['val'] == 0:
	1093	self.tabstops[self.c - 1] = False
	1094	elif csi['val'] == 3:
	1095	for i in range(0, self.columns + 1):
	1096	self.tabstops[i] = False
	1097	# }}}
	1098
	1099	def csi_set(self, csi): # {{{
	1100	# 132 cols
	1101	if csi['val'] == 3:
	1102	self.csi_clear_screen(self.parse_csi('2J'))
	1103	self.working_columns = 132
	1104
	1105	# relative_origin
	1106	elif csi['val'] == 6:
	1107	self.absolute_coords = False
	1108
	1109	# set auto wrap
	1110	elif csi['val'] == 7:
	1111	self.autowrap = True
	1112
	1113
	1114	self.color_changes = {}
	1115	# }}}
	1116
	1117	def csi_reset(self, csi): # {{{
	1118	# 80 cols
	1119	if csi['val'] == 3:
	1120	self.csi_clear_screen(self.parse_csi('2J'))
	1121	self.working_columns = 80
	1122
	1123	# absolute origin
	1124	elif csi['val'] == 6:
	1125	self.absolute_coords = True
	1126
	1127	# reset auto wrap
	1128	elif csi['val'] == 7:
	1129	self.autowrap = False
	1130
	1131
	1132	self.color_changes = {}
	1133	# }}}
	1134
	1135	# }}}
	1136
	1137	###############################################################################################
	1138	# ESC functions {{{
	1139
	1140	def esc_scroll_up(self): # {{{
	1141	self.ctl_nl()
	1142
	1143	self.color_changes = {}
	1144	# }}}
	1145
	1146	def esc_next_line(self): # {{{
	1147	self.ctl_nl()
	1148	self.c = 1
	1149	# }}}
	1150
	1151	def esc_set_tab(self): # {{{
	1152
	1153	if self.c <= len(self.tabstops):
	1154	self.tabstops[self.c - 1] = True
	1155	# }}}
	1156
	1157	def esc_scroll_down(self): # {{{
	1158	if self.l == self.top:
	1159	del self.screen[self.bottom]
	1160	self.screen.insert(self.top, '')
	1161	else:
	1162	self.l += -1
	1163
	1164	self.color_changes = {}
	1165	# }}}
	1166
	1167	# }}}
	1168
	1169	###############################################################################################
	1170	# HASH functions {{{
	1171
	1172	def hash_screen_alignment_test(self): # {{{
	1173	self.csi_clear_screen(self.parse_csi('2J'))
	1174	self.working_lines = self.lines
	1175	for l in range(1, self.lines + 1):
	1176	self.screen[l] = 'E' * self.working_columns
	1177	# }}}
	1178
	1179	# }}}
	1180
	1181	###############################################################################################
	1182	# Random stuff {{{
	1183
	1184	def change_title(self, key, val):
	1185
	1186
	1187	if key == '0' or key == '2':
	1188
	1189	vim.command('setlocal statusline=' + re.escape(val))
	1190
	1191	def paste(self):
	1192	self.write(vim.eval('@@'))
	1193	self.read(50)
	1194
	1195	def paste_selection(self):
	1196	self.write(vim.eval('@@'))
	1197
	1198	def update_window_size(self):
	1199	# resize if needed
	1200	if vim.current.window.width != self.columns or vim.current.window.height != self.lines:
	1201
	1202	# reset all window size attributes to default
	1203	self.columns = vim.current.window.width
	1204	self.lines = vim.current.window.height
	1205	self.working_columns = vim.current.window.width
	1206	self.working_lines = vim.current.window.height
	1207	self.bottom = vim.current.window.height
	1208
	1209	# reset screen object attributes
	1210	self.l = self.screen.reset_size(self.l)
	1211
	1212	# reset tabstops
	1213	self.init_tabstops()
	1214
	1215
	1216
	1217	# signal process that screen size has changed
	1218	self.proc.window_resize(self.lines, self.columns)
	1219
	1220	def init_tabstops(self):
	1221	for i in range(0, self.columns + 1):
	1222	if i % 8 == 0:
	1223	self.tabstops.append(True)
	1224	else:
	1225	self.tabstops.append(False)
	1226
	1227	# }}}
	1228
	1229	###############################################################################################
	1230	# Utility {{{
	1231
	1232	def parse_csi(self, s): # {{{
	1233	attr = { 'key' : s[-1], 'flag' : '', 'val' : 1, 'vals' : [] }
	1234
	1235	if len(s) == 1:
	1236	return attr
	1237
	1238	full = s[0:-1]
	1239
	1240	if full[0] == '?':
	1241	full = full[1:]
	1242	attr['flag'] = '?'
	1243
	1244	if full != '':
	1245	vals = full.split(';')
	1246	for val in vals:
	1247
	1248	val = re.sub("\D", "", val)
	1249
	1250	if val != '':
	1251	attr['vals'].append(int(val))
	1252
	1253	if len(attr['vals']) == 1:
	1254	attr['val'] = int(attr['vals'][0])
	1255
	1256	return attr
	1257	# }}}
	1258
	1259	def bound(self, val, min, max): # {{{
	1260	if val > max:
	1261	return max
	1262
	1263	if val < min:
	1264	return min
	1265
	1266	return val
	1267	# }}}
	1268
	1269	def xterm_to_rgb(self, color_code): # {{{
	1270	if color_code < 16:
	1271	ascii_colors = ['000000', 'CD0000', '00CD00', 'CDCD00', '0000EE', 'CD00CD', '00CDCD', 'E5E5E5',
	1272	'7F7F7F', 'FF0000', '00FF00', 'FFFF00', '5C5CFF', 'FF00FF', '00FFFF', 'FFFFFF']
	1273	return ascii_colors[color_code]
	1274
	1275	elif color_code < 232:
	1276	cc = int(color_code) - 16
	1277
	1278	p1 = "%02x" % (math.floor(cc / 36) * (255/5))
	1279	p2 = "%02x" % (math.floor((cc % 36) / 6) * (255/5))
	1280	p3 = "%02x" % (math.floor(cc % 6) * (255/5))
	1281
	1282	return p1 + p2 + p3
	1283	else:
	1284	grey_tone = "%02x" % math.floor((255/24) * (color_code - 232))
	1285	return grey_tone + grey_tone + grey_tone
	1286	# }}}
	1287
	1288	# }}}
	1289
	1290
	1291	import os, signal, pty, tty, select, fcntl, termios, struct
	1292
	1293	###################################################################################################
	1294	class ConqueSubprocess:
	1295
	1296	# process id
	1297	pid = 0
	1298
	1299	# stdout+stderr file descriptor
	1300	fd = None
	1301
	1302	# constructor
	1303	def __init__(self): # {{{
	1304	self.pid = 0
	1305	# }}}
	1306
	1307	# create the pty or whatever (whatever == windows)
	1308	def open(self, command, env = {}): # {{{
	1309	command_arr = command.split()
	1310	executable = command_arr[0]
	1311	args = command_arr
	1312
	1313	try:
	1314	self.pid, self.fd = pty.fork()
	1315
	1316	except:
	1317	pass
	1318
	1319
	1320	# child proc, replace with command after altering terminal attributes
	1321	if self.pid == 0:
	1322
	1323	# set requested environment variables
	1324	for k in env.keys():
	1325	os.environ[k] = env[k]
	1326
	1327	# set some attributes
	1328	try:
	1329	attrs = tty.tcgetattr(1)
	1330	attrs[0] = attrs[0] ^ tty.IGNBRK
	1331	attrs[0] = attrs[0] \| tty.BRKINT \| tty.IXANY \| tty.IMAXBEL
	1332	attrs[2] = attrs[2] \| tty.HUPCL
	1333	attrs[3] = attrs[3] \| tty.ICANON \| tty.ECHO \| tty.ISIG \| tty.ECHOKE
	1334	attrs[6][tty.VMIN] = 1
	1335	attrs[6][tty.VTIME] = 0
	1336	tty.tcsetattr(1, tty.TCSANOW, attrs)
	1337	except:
	1338	pass
	1339
	1340	os.execvp(executable, args)
	1341
	1342	# else master, do nothing
	1343	else:
	1344	pass
	1345
	1346	# }}}
	1347
	1348	# read from pty
	1349	# XXX - select.poll() doesn't work in OS X!!!!!!!
	1350	def read(self, timeout = 1): # {{{
	1351
	1352	output = ''
	1353	read_timeout = float(timeout) / 1000
	1354
	1355	try:
	1356	# what, no do/while?
	1357	while 1:
	1358	s_read, s_write, s_error = select.select( [ self.fd ], [], [], read_timeout)
	1359
	1360	lines = ''
	1361	for s_fd in s_read:
	1362	try:
	1363	lines = os.read( self.fd, 32 )
	1364	except:
	1365	pass
	1366	output = output + lines
	1367
	1368	if lines == '':
	1369	break
	1370	except:
	1371	pass
	1372
	1373	return output
	1374	# }}}
	1375
	1376	# I guess this one's not bad
	1377	def write(self, input): # {{{
	1378	try:
	1379	os.write(self.fd, input)
	1380	except:
	1381	pass
	1382	# }}}
	1383
	1384	# signal process
	1385	def signal(self, signum): # {{{
	1386	try:
	1387	os.kill(self.pid, signum)
	1388	except:
	1389	pass
	1390	# }}}
	1391
	1392	# get process status
	1393	def get_status(self): #{{{
	1394
	1395	p_status = True
	1396
	1397	try:
	1398	if os.waitpid( self.pid, os.WNOHANG )[0]:
	1399	p_status = False
	1400	except:
	1401	p_status = False
	1402
	1403	return p_status
	1404
	1405	# }}}
	1406
	1407	# update window size in kernel, then send SIGWINCH to fg process
	1408	def window_resize(self, lines, columns): # {{{
	1409	try:
	1410	fcntl.ioctl(self.fd, termios.TIOCSWINSZ, struct.pack("HHHH", lines, columns, 0, 0))
	1411	os.kill(self.pid, signal.SIGWINCH)
	1412	except:
	1413	pass
	1414
	1415	# }}}
	1416
	1417
	1418	###################################################################################################
	1419	# ConqueScreen is an extention of the vim.current.buffer object
	1420	# It restricts the working indices of the buffer object to the scroll region which pty is expecting
	1421	# It also uses 1-based indexes, to match escape sequence commands
	1422	#
	1423	# E.g.:
	1424	# s = ConqueScreen()
	1425	# ...
	1426	# s[5] = 'Set 5th line in terminal to this line'
	1427	# s.append('Add new line to terminal')
	1428	# s[5] = 'Since previous append() command scrolled the terminal down, this is a different line than first cb[5] call'
	1429	#
	1430
	1431	import vim
	1432
	1433	class ConqueScreen(object):
	1434
	1435	# CLASS PROPERTIES {{{
	1436
	1437	# the buffer
	1438	buffer = None
	1439
	1440	# screen and scrolling regions
	1441	screen_top = 1
	1442
	1443	# screen width
	1444	screen_width = 80
	1445	screen_height = 80
	1446
	1447	# }}}
	1448
	1449	def __init__(self): # {{{
	1450	self.buffer = vim.current.buffer
	1451
	1452	self.screen_top = 1
	1453	self.screen_width = vim.current.window.width
	1454	self.screen_height = vim.current.window.height
	1455	# }}}
	1456
	1457	###############################################################################################
	1458	# List overload {{{
	1459	def __len__(self): # {{{
	1460	return len(self.buffer)
	1461	# }}}
	1462
	1463	def __getitem__(self, key): # {{{
	1464	real_line = self.get_real_idx(key)
	1465
	1466	# if line is past buffer end, add lines to buffer
	1467	if real_line >= len(self.buffer):
	1468	for i in range(len(self.buffer), real_line + 1):
	1469	self.append(' ' * self.screen_width)
	1470
	1471	return self.buffer[ real_line ]
	1472	# }}}
	1473
	1474	def __setitem__(self, key, value): # {{{
	1475	real_line = self.get_real_idx(key)
	1476
	1477	# if line is past end of screen, append
	1478	if real_line == len(self.buffer):
	1479	self.buffer.append(value)
	1480	else:
	1481	self.buffer[ real_line ] = value
	1482	# }}}
	1483
	1484	def __delitem__(self, key): # {{{
	1485	del self.buffer[ self.screen_top + key - 2 ]
	1486	# }}}
	1487
	1488	def append(self, value): # {{{
	1489	if len(self.buffer) > self.screen_top + self.screen_height - 1:
	1490	self.buffer[len(self.buffer) - 1] = value
	1491	else:
	1492	self.buffer.append(value)
	1493
	1494	if len(self.buffer) > self.screen_top + self.screen_height - 1:
	1495	self.screen_top += 1
	1496	if vim.current.buffer.number == self.buffer.number:
	1497	vim.command('normal G')
	1498	# }}}
	1499
	1500	def insert(self, line, value): # {{{
	1501
	1502	l = self.screen_top + line - 2
	1503	self.buffer[l:l] = [ value ]
	1504
	1505	# }}}
	1506	# }}}
	1507
	1508	###############################################################################################
	1509	# Util {{{
	1510	def get_top(self): # {{{
	1511	return self.screen_top
	1512	# }}}
	1513
	1514	def get_real_idx(self, line): # {{{
	1515	return (self.screen_top + line - 2)
	1516	# }}}
	1517
	1518	def get_real_line(self, line): # {{{
	1519	return (self.screen_top + line - 1)
	1520	# }}}
	1521
	1522	def set_screen_width(self, width): # {{{
	1523	self.screen_width = width
	1524	# }}}
	1525
	1526	# }}}
	1527
	1528	###############################################################################################
	1529	def clear(self): # {{{
	1530	self.buffer.append(' ')
	1531	vim.command('normal Gzt')
	1532	self.screen_top = len(self.buffer)
	1533	# }}}
	1534
	1535	def set_cursor(self, line, column): # {{{
	1536	# figure out line
	1537	real_line = self.screen_top + line - 1
	1538	if real_line > len(self.buffer):
	1539	for l in range(len(self.buffer) - 1, real_line):
	1540	self.buffer.append('')
	1541
	1542	# figure out column
	1543	real_column = column
	1544	if len(self.buffer[real_line - 1]) < real_column:
	1545	self.buffer[real_line - 1] = self.buffer[real_line - 1] + ' ' * (real_column - len(self.buffer[real_line - 1]))
	1546
	1547	# python version is occasionally grumpy
	1548	try:
	1549	vim.current.window.cursor = (real_line, real_column - 1)
	1550	except:
	1551	vim.command('call cursor(' + str(real_line) + ', ' + str(real_column) + ')')
	1552	# }}}
	1553
	1554	def reset_size(self, line): # {{{
	1555
	1556
	1557
	1558
	1559	# save cursor line number
	1560	real_line = self.screen_top + line
	1561
	1562	# reset screen size
	1563	self.screen_width = vim.current.window.width
	1564	self.screen_height = vim.current.window.height
	1565	self.screen_top = len(self.buffer) - vim.current.window.height + 1
	1566	if self.screen_top < 1:
	1567	self.screen_top = 1
	1568
	1569
	1570	# align bottom of buffer to bottom of screen
	1571	vim.command('normal ' + str(self.screen_height) + 'kG')
	1572
	1573	# return new relative line number
	1574	return (real_line - self.screen_top)
	1575	# }}}
	1576
	1577	def scroll_to_bottom(self): # {{{
	1578	vim.current.window.cursor = (len(self.buffer) - 1, 1)
	1579	# }}}
	1580
	1581	def align(self): # {{{
	1582	# align bottom of buffer to bottom of screen
	1583	vim.command('normal ' + str(self.screen_height) + 'kG')
	1584	# }}}
	1585
	1586
	1587	EOF
	1588
	1589	endif
	1590