150 lines
4.8 KiB
Python
150 lines
4.8 KiB
Python
from helpers import *
|
|
from re import compile as reg_compile
|
|
from traceback import format_exc as print_traceback
|
|
|
|
|
|
mentions = open_json_file("mentio", {}) # contains a list of keywords for each player (uuid)
|
|
max_amount = 1000
|
|
arrow = colorify(u"&r&7\u2192&r")
|
|
colors_reg = reg_compile(u"\u00A7[\\da-fk-or]") # finds color codes
|
|
|
|
|
|
def saveMentions():
|
|
save_json_file("mentio", mentions)
|
|
|
|
|
|
@hook.event("player.AsyncPlayerChatEvent", "monitor")
|
|
def onChat(event):
|
|
if not event.isCancelled():
|
|
sender = event.getPlayer()
|
|
words = event.getMessage().split(" ")
|
|
recipients = event.getRecipients() # set of <Player>, may be a lazy or unmodifiable collection
|
|
|
|
for recipient in list(recipients):
|
|
recuid = uid(recipient)
|
|
|
|
if recuid in mentions:
|
|
keywords = mentions[uid(recipient)]
|
|
else:
|
|
# player
|
|
keywords = [recipient.getName().lower(), stripcolors(recipient.getDisplayName()).lower()]
|
|
|
|
rec_words = words[:] # copy
|
|
for index, word in enumerate(rec_words):
|
|
isMentioned = False
|
|
|
|
for keyword in keywords:
|
|
if keyword in word.lower(): # is the keyword mentioned
|
|
isMentioned = True
|
|
|
|
if isMentioned:
|
|
# join all color codes used upto this word
|
|
colors = "".join(colors_reg.findall("".join(words[:index+1])))
|
|
# highlight the word containing mention, then apply all previous color codes
|
|
rec_words[index] = colorify("&r&a&o") + stripcolors(word) + colorify("&r") + colors
|
|
# No need to
|
|
break
|
|
|
|
# player was mentioned
|
|
if rec_words != words:
|
|
try:
|
|
recipients.remove(recipient) # don't send original message
|
|
except:
|
|
# list might not be mutable, ignoring. Receiver will get the message twice
|
|
pass
|
|
message = " "+" ".join([sender.getDisplayName(), arrow] + rec_words)
|
|
msg(recipient, message, usecolor = False)
|
|
recipient.playSound(recipient.getLocation(), "mob.chicken.plop", 1, 2)
|
|
|
|
|
|
def get_keywords(player):
|
|
sender_id = uid(player)
|
|
if sender_id in mentions.keys():
|
|
keywords = mentions[sender_id]
|
|
else:
|
|
keywords = []
|
|
return keywords
|
|
|
|
|
|
def add_keyword(sender, args):
|
|
keywords = get_keywords(sender)
|
|
new_word = stripcolors(args[1].lower())
|
|
|
|
if (len(keywords) >= max_amount) and (max_amount >= 0):
|
|
msg(sender, "&cYou are already listening for %s words! Try &6/mentio del <word>" % max_amount)
|
|
return True
|
|
|
|
if len(new_word) > 20:
|
|
msg(sender, "&cThis word is longer than 20 characters. Pick a shorter one!")
|
|
return True
|
|
|
|
if new_word in keywords:
|
|
msg(sender, "&cYou are already listening for this word! Try &6/mentio list")
|
|
return True
|
|
|
|
keywords.append(new_word)
|
|
if keywords:
|
|
mentions[uid(sender)] = keywords
|
|
|
|
msg(sender, "&aYou are now listening for '&2%s&2'!" % new_word)
|
|
saveMentions()
|
|
return True
|
|
|
|
|
|
def del_keyword(sender, args):
|
|
keywords = get_keywords(sender)
|
|
del_word = stripcolors(args[1].lower())
|
|
|
|
if not keywords:
|
|
msg(sender, "&cYou are currently listening for no words! Try &6/mentio add <word>")
|
|
return
|
|
|
|
if del_word in keywords:
|
|
keywords.remove(del_word)
|
|
sender_id = uid(sender)
|
|
if keywords:
|
|
mentions[sender_id] = keywords
|
|
elif sender_id in mentions:
|
|
del mentions[sender_id]
|
|
saveMentions()
|
|
msg(sender, "&aYou are no longer listening for '&2%s&2'!" % del_word)
|
|
else:
|
|
msg(sender, "&cWe can't remove something that doesn't exist! Try &6/mentio list")
|
|
|
|
|
|
def show_help(player):
|
|
msg(player, "&a/mentio add <word>")
|
|
msg(player, "&a/mentio del <word>")
|
|
msg(player, "&a/mentio list")
|
|
|
|
|
|
@hook.command("mentio")
|
|
def onListenCommand(sender, command, label, args):
|
|
plugin_header(sender, "Mentio")
|
|
|
|
argnum = len(args)
|
|
if argnum:
|
|
cmd = args[0].lower()
|
|
|
|
# /mentio add <word>
|
|
if argnum == 2 and cmd == "add":
|
|
add_keyword(sender, args)
|
|
|
|
# /mentio del <word>
|
|
elif argnum == 2 and cmd == "del":
|
|
del_keyword(sender, args)
|
|
|
|
# /mentio list
|
|
elif argnum == 1 and cmd == "list":
|
|
msg(sender, "&aWords you're listening for:")
|
|
keywords = get_keywords(sender)
|
|
for word in keywords:
|
|
msg(sender, "&c- &3%s" % word)
|
|
if not keywords:
|
|
msg(sender, "&cYou are currently listening for no words! Try &6/mentio add <word>")
|
|
else:
|
|
show_help(sender)
|
|
else:
|
|
show_help(sender)
|
|
return True
|