80 lines
2.1 KiB
Python
80 lines
2.1 KiB
Python
# vim:fileencoding=utf-8:noet
|
||
from __future__ import (unicode_literals, division, absolute_import, print_function)
|
||
|
||
from powerline.renderer import Renderer
|
||
from powerline.colorscheme import ATTR_BOLD, ATTR_ITALIC, ATTR_UNDERLINE
|
||
|
||
|
||
def attrs_to_tmux_attrs(attrs):
|
||
if attrs is False:
|
||
return ['nobold', 'noitalics', 'nounderscore']
|
||
else:
|
||
ret = []
|
||
if attrs & ATTR_BOLD:
|
||
ret += ['bold']
|
||
else:
|
||
ret += ['nobold']
|
||
if attrs & ATTR_ITALIC:
|
||
ret += ['italics']
|
||
else:
|
||
ret += ['noitalics']
|
||
if attrs & ATTR_UNDERLINE:
|
||
ret += ['underscore']
|
||
else:
|
||
ret += ['nounderscore']
|
||
return ret
|
||
|
||
|
||
class TmuxRenderer(Renderer):
|
||
'''Powerline tmux segment renderer.'''
|
||
|
||
character_translations = Renderer.character_translations.copy()
|
||
character_translations[ord('#')] = '##[]'
|
||
|
||
def render(self, width=None, segment_info={}, **kwargs):
|
||
if width and segment_info:
|
||
width -= segment_info.get('width_adjust', 0)
|
||
if width < 10:
|
||
width = 10
|
||
return super(TmuxRenderer, self).render(width=width, segment_info=segment_info, **kwargs)
|
||
|
||
def hlstyle(self, fg=None, bg=None, attrs=None):
|
||
'''Highlight a segment.'''
|
||
# We don’t need to explicitly reset attributes, so skip those calls
|
||
if not attrs and not bg and not fg:
|
||
return ''
|
||
tmux_attrs = []
|
||
if fg is not None:
|
||
if fg is False or fg[0] is False:
|
||
tmux_attrs += ['fg=default']
|
||
else:
|
||
if self.term_truecolor and fg[1]:
|
||
tmux_attrs += ['fg=#{0:06x}'.format(int(fg[1]))]
|
||
else:
|
||
tmux_attrs += ['fg=colour' + str(fg[0])]
|
||
if bg is not None:
|
||
if bg is False or bg[0] is False:
|
||
tmux_attrs += ['bg=default']
|
||
else:
|
||
if self.term_truecolor and bg[1]:
|
||
tmux_attrs += ['bg=#{0:06x}'.format(int(bg[1]))]
|
||
else:
|
||
tmux_attrs += ['bg=colour' + str(bg[0])]
|
||
if attrs is not None:
|
||
tmux_attrs += attrs_to_tmux_attrs(attrs)
|
||
return '#[' + ','.join(tmux_attrs) + ']'
|
||
|
||
def get_segment_info(self, segment_info, mode):
|
||
r = self.segment_info.copy()
|
||
if segment_info:
|
||
r.update(segment_info)
|
||
if 'pane_id' in r:
|
||
varname = 'TMUX_PWD_' + str(r['pane_id'])
|
||
if varname in r['environ']:
|
||
r['getcwd'] = lambda: r['environ'][varname]
|
||
r['mode'] = mode
|
||
return r
|
||
|
||
|
||
renderer = TmuxRenderer
|