git.r.bdr.sh - rbdr/dotfiles/blob - vim/tmp/conque/autoload/conque_term.vim

   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