def whois(bot, name): Report.trace('bot', 'Whois: ' + name) response = bot.sendChatMessage('/whois ' + name) responsetext = ''.join(x['text'] for x in response) match = re.search(r'<a[^>]*showplayer.php[^>]*><b[^>]*>([A-Za-z0-9_ ]+) \(#([0-9]+)\)</b></a>', responsetext) if match: player_name = match.group(1) player_id = match.group(2) Report.trace('bot', 'Whois resolved: Name: ' + str(player_name)) Report.trace('bot', 'Whois resolved: ID: ' + str(player_id)) return player_name, player_id elif 'Unknown Player: ' in responsetext: raise Error.Error("That player could not be found.", Error.USER_NOT_FOUND) else: Report.warning('bot', 'Unable to parse /whois response: ' + repr(responsetext)) raise Error.Error("Unable to parse /whois response.", Error.REQUEST_GENERIC)
def botProcessChat(context, **kwargs): returnCode = FilterManager.CONTINUE bot = kwargs['bot'] chat = kwargs['chat'] if chat['type'] in ['private']: # Initialize variables for response formatting user_name = str(chat['userName']) user_id = str(chat['userId']) current_time = time.time() exprlist = [] exprresults = [] channel = '' clan = alea.clan.Clan(0, '') target_name = '' target_id = '0' uneffectable = alea.util.Uneffectable('') msg = '' try: # Parse the abomination that our chat partner hath wrought request = alea.parser.aleabot_parse(chat['text']) if request[0] == 'rollrequest': # Handle a dice rolling request exprlist = request[1] channel = request[2] # Get the reply text that applies to this kind of roll request diceless = all(expr.classify_dice() == 0 for expr in exprlist) if channel == '': # Private rolling if diceless: msg = aleabot.config.get('rolltext_diceless_private') else: msg = aleabot.config.get('rolltext_private') else: # Public rolling if diceless: msg = aleabot.config.get('rolltext_diceless_public') else: msg = aleabot.config.get('rolltext_public') # Check if channel is allowed, and switch clan if needed if channel != '': if channel in aleabot.config.get('channels').split(): # Allowed public channel (e.g. /games) pass elif channel in aleabot.config.get('clanchannels').split(): # Allowed clan channel (e.g. /clan, /hobopolis, ...) aleabot.clanstate.set_session(bot.session) clan = aleabot.clanstate.player_clan(user_id) Report.info('bot', '%s asked me to roll in clan %s' % (user_name, clan.name())) if clan.id() == 0: Report.warning('bot', 'A player who is not in a clan asked me to roll in ' + channel) raise ClanlessPlayerError('clanless player') elif not aleabot.clanstate.have_whitelist(clan): Report.warning('bot', 'I do not have a whitelist in clan %s' % clan.name()) raise NeedWhitelistError('need whitelist') else: Report.info('bot', 'I have a whitelist in clan %s' % clan.name()) aleabot.clanstate.switch(clan) # Set timer to switch back to home clan aleabot.home_clan_timer = 0 else: raise ChannelDisallowedError(channel) # Apply time-based limits aleabot.rolllimiter.check(channel, user_id, clan.id(), current_time, aleabot.config) # Evaluate dice expressions exprresults = alea.expr.aleabot_eval(exprlist, channel != '', aleabot.rng, aleabot.config) # Update time-based roll limiter aleabot.rolllimiter.update(channel, user_id, clan.id(), current_time, aleabot.config) # Record roll and result in roll verification list state = bot.states['global'] rollverify_key = 'rollverify_' + user_id if rollverify_key not in state: state[rollverify_key] = [] state[rollverify_key].insert(0, ( current_time, [str(x) for x in exprlist], [str(x) for x in exprresults], str(channel), str(clan.name()))) rollverify_count = aleabot.config.get('rollverify_count') state[rollverify_key] = state[rollverify_key][0:rollverify_count] bot.writeState('global') elif request[0] == 'rollverifyrequest': # Handle a roll verification request try: target_name, target_id = resolve_target(bot, user_name, user_id, request[1]) # Get config settings rollverify_count = aleabot.config.get('rollverify_count') rollverify_header = aleabot.config.get('rollverify_header') rollverify_entry_private = aleabot.config.get('rollverify_entry_private') rollverify_entry_public = aleabot.config.get('rollverify_entry_public') rollverify_entry_clan = aleabot.config.get('rollverify_entry_clan') # Get saved list of rolls state = bot.states['global'] rollverify_key = 'rollverify_' + target_id rolls = state.get(rollverify_key, []) rolls = rolls[0:rollverify_count] # Build kmail rollverify_kmail = {} rollverify_kmail['userId'] = user_id rollverify_kmail['text'] = format_reply(rollverify_header, user_name=user_name, user_id=user_id, current_time=current_time, target_name=target_name, target_id=target_id) rollverify_kmail['text'] += '\n\n' for roll in rolls: if roll[4] != '': # clan != '' rollverify_entry = rollverify_entry_clan elif roll[3] != '': # channel != '' rollverify_entry = rollverify_entry_public else: rollverify_entry = rollverify_entry_private rollverify_kmail['text'] += format_reply(rollverify_entry, user_name=user_name, user_id=user_id, current_time=roll[0], target_name=target_name, target_id=target_id, exprlist=roll[1], exprresults=roll[2], channel=roll[3], clan=roll[4]) rollverify_kmail['text'] += '\n' # Try to send the kmail bot.sendKmail(rollverify_kmail) except Error.Error as err: if err.code == Error.USER_NOT_FOUND: msg = aleabot.config.get('error_rollverify_player_not_found') else: msg = aleabot.config.get('error_generic') elif request[0] == 'helprequest': # Handle a help request msg = aleabot.config.get('helptext') elif request[0] == 'hellorequest': # Handle a hello request msg = aleabot.config.get('hellotext') elif request[0] == 'thanksrequest': # Handle a thanks request msg = aleabot.config.get('thankstext') elif request[0] == 'timerequest': # Handle a time request msg = aleabot.config.get('timetext') elif request[0] == 'wangrequest': # Handle a wang request try: target_name, target_id = resolve_target(bot, user_name, user_id, request[1]) # Check limits # Use 'rollover' bot state which is cleared each rollover state = bot.states['rollover'] wang_sender_count_key = 'wang_sender_count_' + user_id wang_sender_count = state.get(wang_sender_count_key, 0) wang_target_count_key = 'wang_target_count_' + target_id wang_target_count = state.get(wang_target_count_key, 0) if target_id == str(bot.session.userId): msg = aleabot.config.get('error_wang_self') elif wang_sender_count >= aleabot.config.get('wang_sender_limit'): msg = aleabot.config.get('error_wang_sender_limit') elif wang_target_count >= aleabot.config.get('wang_target_limit'): msg = aleabot.config.get('error_wang_target_limit') elif target_id != user_id and ('dontwangme_' + target_id) in bot.states['global']: msg = aleabot.config.get('error_wang_notwanted') else: # Limits not reached yet. Slap! Report.info('bot', 'Slapping player with wang: ' + target_name) wangreq = CursePlayerRequest(bot.session, target_id, ITEM_ID_WANG) wangreq.doRequest() if target_id == user_id: msg = aleabot.config.get('wangtext_self') else: msg = aleabot.config.get('wangtext') # Increase limit counters state[wang_sender_count_key] = wang_sender_count + 1 state[wang_target_count_key] = wang_target_count + 1 bot.writeState('rollover') except Error.Error as err: if err.code == Error.ITEM_NOT_FOUND: msg = aleabot.config.get('error_wang_no_wangs') elif err.code == Error.USER_NOT_FOUND: msg = aleabot.config.get('error_wang_player_not_found') else: msg = aleabot.config.get('error_wang_generic') elif request[0] == 'arrowrequest': # Handle an arrow request try: target_name, target_id = resolve_target(bot, user_name, user_id, request[1]) # Check limits # Use 'rollover' bot state which is cleared each rollover state = bot.states['rollover'] arrow_sender_count_key = 'arrow_sender_count_' + user_id arrow_sender_count = state.get(arrow_sender_count_key, 0) if target_id == str(bot.session.userId): msg = aleabot.config.get('error_arrow_self') elif arrow_sender_count >= aleabot.config.get('arrow_sender_limit'): msg = aleabot.config.get('error_arrow_sender_limit') else: # Limits not reached yet. Fire! Report.info('bot', 'Firing arrow at player: ' + target_name) arrowreq = CursePlayerRequest(bot.session, target_id, ITEM_ID_ARROW) arrowreq.doRequest() if target_id == user_id: msg = aleabot.config.get('arrowtext_self') else: msg = aleabot.config.get('arrowtext') # Increase limit counters state[arrow_sender_count_key] = arrow_sender_count + 1 bot.writeState('rollover') except Error.Error as err: if err.code == Error.ITEM_NOT_FOUND: msg = aleabot.config.get('error_arrow_no_arrows') elif err.code == Error.USER_NOT_FOUND: msg = aleabot.config.get('error_arrow_player_not_found') elif err.code == Error.USER_IN_HARDCORE_RONIN: msg = aleabot.config.get('error_arrow_ronin') elif err.code == Error.ALREADY_COMPLETED: msg = aleabot.config.get('error_arrow_already_hit') else: msg = aleabot.config.get('error_arrow_generic') elif request[0] == 'uneffectrequest': # Handle an uneffect request uneffectable = request[1] if uneffectable.inputname() == '': msg = aleabot.config.get('error_uneffect_no_effect_given') elif uneffectable.count() == 0: msg = aleabot.config.get('error_uneffect_no_match') elif uneffectable.count() >= 2: msg = aleabot.config.get('error_uneffect_too_many_matches') else: # Exactly one effect matched effect_id = uneffectable.effect_ids()[0] print str(effect_id) uneffectreq = UneffectRequest(bot.session, effect_id) try: uneffectreq.doRequest() msg = aleabot.config.get('uneffecttext') except Error.Error as err: if err.code == Error.EFFECT_NOT_FOUND: msg = aleabot.config.get('error_uneffect_not_cursed') elif err.code == Error.ITEM_NOT_FOUND: msg = aleabot.config.get('error_uneffect_no_sgeea') else: msg = aleabot.config.get('error_uneffect_generic') elif request[0] == 'dontwangmerequest': # Handle a dontwangme request key = 'dontwangme_' + user_id bot.states['global'][key] = True bot.writeState('global') msg = aleabot.config.get('dontwangmetext') elif request[0] == 'allowwangrequest': # Handle an allowwang request key = 'dontwangme_' + user_id bot.states['global'].pop(key, None) bot.writeState('global') msg = aleabot.config.get('allowwangtext') except GenericAleabotError: msg = aleabot.config.get('error_generic') except ChannelDisallowedError: msg = aleabot.config.get('error_channel_disallowed') except ClanlessPlayerError: msg = aleabot.config.get('error_clanless_player') except NeedWhitelistError: msg = aleabot.config.get('error_need_whitelist') except alea.clan.ClanRequestError: msg = aleabot.config.get('error_clan_request') except alea.rolllimiter.PrivatePerPlayerRollLimitError: msg = aleabot.config.get('error_private_perplayer_limit') except alea.rolllimiter.PublicPerPlayerRollLimitError: msg = aleabot.config.get('error_public_perplayer_limit') except alea.rolllimiter.PublicPerChannelRollLimitError: msg = aleabot.config.get('error_public_perchannel_limit') except alea.expr.ResultCountExceededError: msg = aleabot.config.get('error_too_many_results') except alea.expr.DiceCountExceededError: msg = aleabot.config.get('error_expression_too_many_dice') except alea.expr.DicelessDisallowedError: msg = aleabot.config.get('error_diceless') except alea.expr.D1DisallowedError: msg = aleabot.config.get('error_d1') except alea.expr.AleabotEvalError: msg = aleabot.config.get('error_expression_eval') except alea.parser.AleabotSyntaxError: msg = aleabot.config.get('error_bad_syntax') # If not explicitly chatting to a public channel or a private # conversation, make sure we /msg the user who is talking to us if msg != '' and msg[0] != '/': msg = '/msg %p ' + msg # Format reply message msg = format_reply(msg, user_name=user_name, user_id=user_id, current_time=current_time, exprlist=exprlist, exprresults=exprresults, channel=channel, clan=clan, target_name=target_name, target_id=target_id, uneffectable=uneffectable) # Chat! if msg != '': response = bot.sendChatMessage(msg) response_text = "\n".join(x['text'] for x in response) # Handle chat errors if 'You cannot access that channel' in response_text: Report.warning('bot', 'Received error while chatting: ' + response_text) msg = aleabot.config.get('error_channel_inaccessible') msg = '/msg ' + user_id + ' ' + msg bot.sendChatMessage(msg) returnCode = FilterManager.FINISHED elif chat['type'] in ['notification:kmail']: Report.info('bot', 'Kmail notification received.') # Make sure to check for new kmails next cycle # (see botEndCycle handler) aleabot.kmail_check_timer = aleabot.config.get('time_to_sleep_kmail') elif chat['type'] in ['unknown']: # Handle some chat messages of type 'unknown' # Such as: whitelist changes, clan acceptance / rejection aleabot.clanstate.set_session(bot.session) if aleabot.clanstate.read_unknown_chat_message(chat['text']): Report.info('bot', 'Clan state is no longer valid, need to reload.') return returnCode
def botProcessKmail(context, **kwargs): returnCode = FilterManager.CONTINUE message = kwargs['kmail'] bot = kwargs['bot'] user_name = str(message['userName']) user_id = str(message['userId']) current_time = time.time() cmd = BotUtils.getKmailCommand(message) meat = message['meat'] items = message['items'] # Our response response = '' # Should items and meat be sent back? return_goodies = True # Should a candy heart be sent? send_heart = False # if 1 arrow was sent and the kmail is empty, interpret it as "arrow" if cmd == "" and len(items) == 1 and items[0]['id'] == ITEM_ID_ARROW and items[0]['quantity'] == 1 and meat == 0: cmd = 'arrow' if cmd == 'arrow': # Handle arrow request if len(items) == 1 and items[0]['id'] == ITEM_ID_ARROW and items[0]['quantity'] == 1 and meat == 0: # Everything is okay try: Report.info('bot', 'Firing arrow at player: ' + user_name) arrowreq = CursePlayerRequest(bot.session, user_id, ITEM_ID_ARROW) arrowreq.doRequest() return_goodies = False except Error.Error as err: if err.code == Error.ITEM_NOT_FOUND: response = aleabot.config.get('error_arrow_no_arrows') elif err.code == Error.USER_NOT_FOUND: response = aleabot.config.get('error_arrow_player_not_found') elif err.code == Error.USER_IN_HARDCORE_RONIN: response = aleabot.config.get('error_arrow_ronin') elif err.code == Error.ALREADY_COMPLETED: response = aleabot.config.get('error_arrow_already_hit') else: response = aleabot.config.get('error_arrow_generic') elif len(items) == 0 and meat == 0: Report.warning('bot', 'Arrow request without arrow from ' + user_name) response = aleabot.config.get('kmailtext_arrow_notattached') else: Report.warning('bot', 'Arrow request with extra items or meat from ' + user_name) response = aleabot.config.get('kmailtext_arrow_extraattached') elif cmd == 'donate' or cmd == 'donation': # Handle donation if len(items) == 0 and meat == 0: # Empty donation kmail? Report.warning('bot', 'Empty donation received from ' + user_name) response = aleabot.config.get('kmailtext_donate_empty') else: Report.info('bot', 'Donation received from ' + user_name) response = aleabot.config.get('kmailtext_donate_thanks') return_goodies = False send_heart = True else: # Handle unknown command Report.warning('bot', 'Unknown kmail command: ' + cmd) response = aleabot.config.get('kmailtext_unknown') # Send our response if response != '' or (return_goodies and (len(items) != 0 or meat != 0)): Report.info('bot', 'Responding to kmail') response_kmail = {} response_kmail['userId'] = message['userId'] response_kmail['text'] = format_reply(response + '\n\n' + aleabot.config.get('kmailtext_quote'), user_name=user_name, user_id=user_id, current_time=current_time) + '\n' + quote_kmail(message) if return_goodies: response_kmail['items'] = items response_kmail['meat'] = meat try: bot.sendKmail(response_kmail) except Error.Error as err: if err.code == Error.USER_IN_HARDCORE_RONIN: Report.error('bot', 'Tried to send items and meat back, but user is in Hardcore or Ronin!') response_kmail2 = {} response_kmail2['userId'] = message['userId'] response_kmail2['text'] = format_reply(response + '\n\n' + aleabot.config.get('kmailtext_quote_ronin'), user_name=user_name, user_id=user_id, curent_time=current_time) + '\n' + quote_kmail(message) try: bot.sendKmail(response_kmail2) except Error.Error as err2: Report.error('bot', 'Unexpected error while sending response_kmail2: ' + str(err2)) else: Report.error('bot', 'Unexpected error while sending response_kmail: ' + str(err)) # Send a candy heart if send_heart: try: Report.info('bot', 'Sending candy heart to player: ' + user_name) heartreq = CursePlayerRequest(bot.session, user_id, ITEM_ID_CANDYHEART) heartreq.requestData['texta'] = 'THANK' heartreq.requestData['textb'] = 'YOU' heartreq.doRequest() except Error.Error as err: Report.error('bot', 'Couldn\'t send candy heart: ' + str(err)) returnCode = FilterManager.FINISHED return returnCode