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
def highlight(self, textlist, place, memlength, hl):
"""
Takes a bunch of stuff and applies the highlight involved.
It's messy.
@param textlist: the list of strings representing the incoming
text--this is usually text interspersed with ansi color tokens.
@type textlist: list of strings
@param place: the point in the text (skipping over ansi color stuff)
that marks the beginning of the highlight
@type place: int
@param memlength: the length of the string to be highlighted
@type memlength: int
@param hl: the highlight to apply
@type hl: string
@returns: the newly adjusted textlist
@rtype: list of strings
"""
# first we find the place to stick the highlight thingy.
i = 0
for i in range(0, len(textlist)):
if not ansi.is_color_token(textlist[i]):
if place > len(textlist[i]):
place -= len(textlist[i])
else:
break
newlist = textlist[:i]
newlist.append(textlist[i][:place])
newcolor = ansi.figure_color(newlist, self._currcolor)[0]
newlist.append(hl)
# if the string to highlight begins and ends in the
# same token we deal with that and eject
if len(textlist[i][place:]) >= memlength:
newlist.append(textlist[i][place:place + memlength])
newlist.append(chr(27) + "[0m")
color = ansi.convert_tuple_to_ansi(newcolor)
if color:
newlist.append(color)
newlist.append(textlist[i][place + memlength:])
for mem in textlist[i+1:]:
newlist.append(mem)
return newlist
newlist.append(textlist[i][place:])
# now we have to find the end of the highlight
memlength -= len(textlist[i][place:])
j = i+1
for j in range(i+1, len(textlist)):
if not ansi.is_color_token(textlist[j]):
if memlength > len(textlist[j]):
memlength -= len(textlist[j])
newlist.append(textlist[j])
else:
break
else:
newcolor = ansi.figure_color([textlist[j]], newcolor, '')[0]
newlist.append(textlist[j][:memlength])
newlist.append(chr(27) + "[0m")
color = ansi.convert_tuple_to_ansi(newcolor)
if color:
newlist.append(color)
newlist.append(textlist[j][memlength:])
for mem in textlist[j+1:]:
newlist.append(mem)
return newlist
