def expand(self, text):
"""
Looks at mud data and performs any highlights.
It returns the final text--even if there were no highlights.
@param text: the input text
@type text: string
@return: the finalized text--even if no highlights were expanded
@rtype: string
"""
if text:
faketext = ansi.filter_ansi(text)
textlist = ansi.split_ansi_from_text(text)
hlist = self._highlights.keys()
hlist.sort()
for mem in hlist:
miter = self._highlights[mem][2].finditer(faketext)
for m in miter:
# we need to loop for multiple instances of the highlight
begin, end = m.span()
hl = self._highlights[mem][1]
textlist = self.highlight(textlist, begin, end - begin, hl)
# here we sweep through the text string to update our current
# color and leftover color attributes
self._currcolor, self._colorleftover = ansi.figure_color(textlist, self._currcolor, self._colorleftover)
text = "".join(textlist)
return text
def group_member_health(self, match, colorline):
num = match.group("num")
name = match.group("name")
hp = match.group("hp")
position = match.group("position")
near = (match.group("near") == u'Д')
drain = (match.group("drained") == u'Д')
name = name.strip()
try:
self._group_list[name].update(position, near, len(hp), drain)
except KeyError:
exported.write_message(
u"ClerBot: WARN: could not find group member {0}".format(name))
# Detect stone curse
tokens = ansi.split_ansi_from_text(colorline)
tokens = list(map(str.strip, tokens))
try:
pos_index = tokens.index(position)
pos_color = tokens[pos_index - 1]
except ValueError:
pos_color = ""
#print u"At {0} the color is {1}, tokens being: {2}, position {3}.".format(name, pos_color.replace(chr(27), "^["), tokens, position)
if ansi.is_color_token(pos_color):
fg = pos_color
if ((ansi.STYLEMAP["light yellow"] == fg[2:-1])
or (ansi.STYLEMAP["yellow"] == fg[2:-1])):
self._group_list[name].set_stone_curse(True)
def expand(self, text):
"""
Looks at mud data and performs any gags.
It returns the final text--even if there were no gags.
@param text: the text to expand gags in
@type text: string
@return: the (un)adjusted text
@rtype: string
"""
if len(text) > 0:
# check for antigags first
for mem in self._antigags.values():
if mem.search(ansi.filter_ansi(text)):
return text
# check for gags
for mem in self._gags.values():
if mem.search(ansi.filter_ansi(text)):
tokens = ansi.split_ansi_from_text(text)
tokens = [m for m in tokens if ansi.is_color_token(m)]
return "".join(tokens)
return text
def _decode_colors(self, ses, default_attr, line, pretext=[]):
if self.unfinished_.has_key(ses):
(currentcolor, leftover) = self.unfinished_[ses]
else:
currentcolor = list(ansi.DEFAULT_COLOR)
leftover = ''
for single in line.splitlines(1):
current = []
tokens = ansi.split_ansi_from_text(leftover + single)
leftover = ''
lasttok = ''
for tok in tokens:
if ansi.is_color_token(tok):
currentcolor, leftover = ansi.figure_color([tok],
currentcolor,
leftover)
elif tok:
attr = default_attr
if currentcolor[ansi.PLACE_BOLD]:
attr |= curses.A_BOLD
if currentcolor[ansi.PLACE_UNDERLINE]:
attr |= curses.A_UNDERLINE
if currentcolor[ansi.PLACE_BLINK]:
attr |= curses.A_BLINK
if currentcolor[ansi.PLACE_REVERSE]:
attr |= curses.A_REVERSE
foreground = currentcolor[ansi.PLACE_FG] - 30
if 0 <= foreground and foreground <= 7:
attr += curses_fore(foreground)
background = currentcolor[ansi.PLACE_BG] - 40
if 0 <= background and background <= 7:
attr += curses_back(background)
lasttok = tok
current.append((tok, attr))
if current:
lines = self.lines_
current[:0] = pretext
if not lasttok.endswith("\n"): # it is a prompt
#
# Append newline to prompts coming from another sessions
if ses != exported.get_current_session():
current.append(("\n", curses.A_NORMAL))
else:
if self.cfg_compact_ and self.prompt_ == current:
#
# Remove the identical prompt from previous output buffer:
#
lines[self.prompt_index_:self.prompt_index_ +
1] = []
else:
self.prompt_ = current
self.prompt_index_ = len(lines)
self._append(current)
# eliminating empty lines:
elif current[0][0] != "\n" or not self.cfg_compact_:
self._append(current)
self.unfinished_[ses] = (currentcolor, leftover)
def expand(self, text):
"""
Looks at mud data and performs any gags.
It returns the final text--even if there were no gags.
@param text: the text to expand gags in
@type text: string
@return: the (un)adjusted text
@rtype: string
"""
if len(text) > 0:
# check for antigags first
for mem in self._antigags.values():
if mem.search(ansi.filter_ansi(text)):
return text
# check for gags
for mem in self._gags.values():
if mem.search(ansi.filter_ansi(text)):
tokens = ansi.split_ansi_from_text(text)
tokens = [m for m in tokens if ansi.is_color_token(m)]
return "".join(tokens)
return text
def _decode_colors(self, ses, default_attr, line, pretext=[]):
if self.unfinished_.has_key(ses):
(currentcolor, leftover) = self.unfinished_[ses]
else:
currentcolor = list(ansi.DEFAULT_COLOR)
leftover = ''
for single in line.splitlines(1):
current = []
tokens = ansi.split_ansi_from_text(leftover + single)
leftover = ''
lasttok = ''
for tok in tokens:
if ansi.is_color_token(tok):
currentcolor, leftover = ansi.figure_color([tok], currentcolor, leftover)
elif tok:
attr = default_attr
if currentcolor[ansi.PLACE_BOLD]:
attr |= curses.A_BOLD
if currentcolor[ansi.PLACE_UNDERLINE]:
attr |= curses.A_UNDERLINE
if currentcolor[ansi.PLACE_BLINK]:
attr |= curses.A_BLINK
if currentcolor[ansi.PLACE_REVERSE]:
attr |= curses.A_REVERSE
foreground = currentcolor[ansi.PLACE_FG] - 30
if 0 <= foreground and foreground <= 7:
attr += curses_fore(foreground)
background = currentcolor[ansi.PLACE_BG] - 40
if 0 <= background and background <= 7:
attr += curses_back(background)
lasttok = tok
current.append( (tok, attr) )
if current:
lines = self.lines_
current[:0] = pretext
if not lasttok.endswith("\n"): # it is a prompt
#
# Append newline to prompts coming from another sessions
if ses != exported.get_current_session():
current.append( ("\n", curses.A_NORMAL) )
else:
if self.cfg_compact_ and self.prompt_ == current:
#
# Remove the identical prompt from previous output buffer:
#
lines[self.prompt_index_:self.prompt_index_+1] = []
else:
self.prompt_ = current
self.prompt_index_ = len(lines)
self._append(current)
# eliminating empty lines:
elif current[0][0] != "\n" or not self.cfg_compact_:
self._append(current)
self.unfinished_[ses] = (currentcolor, leftover)
def buffer_write(msg, txtbuffer, currentcolor, unfinishedcolor):
"""
Handles writing messages to a Tk Text widget taking into accound
ANSI colors, message types, session scoping, and a variety of
other things.
@param msg: the ui.message.Message to write to the buffer
@type msg: ui.message.Message
@param txtbuffer: the Tk Text buffer to write to
@type txtbuffer: Text
@param currentcolor: the current color that we should start with
@type currentcolor: color (list of ints)
@param unfinishedcolor: the string of unfinished ANSI color stuff
that we'll prepend to the string we're printing
@type unfinishedcolor: string
@returns: the new color and unfinished color
@rtype: list of ints, string
"""
global myui
line = msg.data
ses = msg.session
if msg.type == message.ERROR:
if line.endswith("\n"):
line = "%s%s%s\n" % (ansi.get_color("b blue"), line[:-1], ansi.get_color("default"))
else:
line = "%s%s%s" % (ansi.get_color("b blue"), line[:-1], ansi.get_color("default"))
elif msg.type == message.USERDATA:
if myui._do_i_echo == 1:
if line.endswith("\n"):
line = "%s%s%s\n" % (ansi.get_color("b blue"), line[:-1], ansi.get_color("default"))
else:
line = "%s%s%s" % (ansi.get_color("b blue"), line[:-1], ansi.get_color("default"))
else:
# if echo is not on--we don't print this
return currentcolor, unfinishedcolor
elif msg.type == message.LTDATA:
if line.endswith("\n"):
line = "# %s\n" % line[:-1].replace("\n", "\n# ")
else:
line = "# %s" % line.replace("\n", "\n# ")
# now we go through and handle writing all the data
index = 0
start = 0
# we prepend the session name to the text if this is not the
# current session sending text and if the Message is session
# scoped.
if ses != None and ses != exported.get_current_session():
pretext = "[%s]" % ses.getName()
if line.endswith("\n"):
line = pretext + line[:-1].replace("\n", "\n" + pretext) + "\n"
else:
line = pretext + line.replace("\n", "\n" + pretext) + "\n"
# we remove all \\r stuff because it's icky
line = line.replace("\r", "")
tokens = ansi.split_ansi_from_text(line)
# each session has a saved current color for MUDDATA. we grab
# that current color--or use our default if we don't have one
# for the session yet. additionally, some sessions have an
# unfinished color as well--in case we got a part of an ansi
# color code in a mud message, and the other part is in another
# message.
if msg.type == message.MUDDATA:
color = currentcolor.get(ses, list(DEFAULT_COLOR))
leftover = unfinishedcolor.get(ses, "")
else:
color = list(DEFAULT_COLOR)
leftover = ""
for mem in tokens:
if ansi.is_color_token(mem):
color, leftover = ansi.figure_color([mem], color, leftover)
else:
format = []
fg = ""
bg = ""
# handle reverse
if color[ansi.PLACE_REVERSE] == 0:
if color[ansi.PLACE_FG] == -1:
fg = "37"
else:
fg = str(color[ansi.PLACE_FG])
if color[ansi.PLACE_BG] != -1:
bg = str(color[ansi.PLACE_BG])
else:
if color[ansi.PLACE_BG] == -1:
fg = "30"
else:
fg = str(color[ansi.PLACE_BG] - 10)
if color[ansi.PLACE_FG] == -1:
bg = "47"
else:
bg = str(color[ansi.PLACE_FG] + 10)
# handle bold
if color[ansi.PLACE_BOLD] == 1:
fg = "b" + fg
# handle underline
if color[ansi.PLACE_UNDERLINE] == 1:
format.append("u")
format.append(fg)
if bg:
format.append(bg)
# mem = add_special_characters(mem)
# insert the text using the formatting tuple we just generated
txtbuffer.insert("end", mem.decode(UNICODE_ENCODING), tuple(format))
return color, leftover
def buffer_write(msg, txtbuffer, currentcolor, unfinishedcolor):
"""
Handles writing messages to a Tk Text widget taking into accound
ANSI colors, message types, session scoping, and a variety of
other things.
@param msg: the ui.message.Message to write to the buffer
@type msg: ui.message.Message
@param txtbuffer: the Tk Text buffer to write to
@type txtbuffer: Text
@param currentcolor: the current color that we should start with
@type currentcolor: color (list of ints)
@param unfinishedcolor: the string of unfinished ANSI color stuff
that we'll prepend to the string we're printing
@type unfinishedcolor: string
@returns: the new color and unfinished color
@rtype: list of ints, string
"""
global myui
line = msg.data
ses = msg.session
if msg.type == message.ERROR:
if line.endswith("\n"):
line = "%s%s%s\n" % (ansi.get_color("b blue"), line[:-1],
ansi.get_color("default"))
else:
line = "%s%s%s" % (ansi.get_color("b blue"), line[:-1],
ansi.get_color("default"))
elif msg.type == message.USERDATA:
if myui._do_i_echo == 1:
if line.endswith("\n"):
line = "%s%s%s\n" % (ansi.get_color("b blue"), line[:-1],
ansi.get_color("default"))
else:
line = "%s%s%s" % (ansi.get_color("b blue"), line[:-1],
ansi.get_color("default"))
else:
# if echo is not on--we don't print this
return currentcolor, unfinishedcolor
elif msg.type == message.LTDATA:
if line.endswith("\n"):
line = "# %s\n" % line[:-1].replace("\n", "\n# ")
else:
line = "# %s" % line.replace("\n", "\n# ")
# now we go through and handle writing all the data
index = 0
start = 0
# we prepend the session name to the text if this is not the
# current session sending text and if the Message is session
# scoped.
if (ses != None and ses != exported.get_current_session()):
pretext = "[%s]" % ses.getName()
if line.endswith("\n"):
line = (pretext + line[:-1].replace("\n", "\n" + pretext) + "\n")
else:
line = pretext + line.replace("\n", "\n" + pretext) + "\n"
# we remove all \\r stuff because it's icky
line = line.replace("\r", "")
tokens = ansi.split_ansi_from_text(line)
# each session has a saved current color for MUDDATA. we grab
# that current color--or use our default if we don't have one
# for the session yet. additionally, some sessions have an
# unfinished color as well--in case we got a part of an ansi
# color code in a mud message, and the other part is in another
# message.
if msg.type == message.MUDDATA:
color = currentcolor.get(ses, list(DEFAULT_COLOR))
leftover = unfinishedcolor.get(ses, "")
else:
color = list(DEFAULT_COLOR)
leftover = ""
for mem in tokens:
if ansi.is_color_token(mem):
color, leftover = ansi.figure_color([mem], color, leftover)
else:
format = []
fg = ""
bg = ""
# handle reverse
if color[ansi.PLACE_REVERSE] == 0:
if color[ansi.PLACE_FG] == -1:
fg = "37"
else:
fg = str(color[ansi.PLACE_FG])
if color[ansi.PLACE_BG] != -1:
bg = str(color[ansi.PLACE_BG])
else:
if color[ansi.PLACE_BG] == -1:
fg = "30"
else:
fg = str(color[ansi.PLACE_BG] - 10)
if color[ansi.PLACE_FG] == -1:
bg = "47"
else:
bg = str(color[ansi.PLACE_FG] + 10)
# handle bold
if color[ansi.PLACE_BOLD] == 1:
fg = "b" + fg
# handle underline
if color[ansi.PLACE_UNDERLINE] == 1:
format.append("u")
format.append(fg)
if bg:
format.append(bg)
# insert the text using the formatting tuple we just generated
txtbuffer.insert('end', _decode(mem), tuple(format))
return color, leftover