mirror of
https://github.com/SpiceBot/SpiceBot.git
synced 2026-11-10 21:50:59 -05:00
221 lines
8.5 KiB
Python
221 lines
8.5 KiB
Python
# coding=utf-8
|
|
|
|
from __future__ import unicode_literals, absolute_import, division, print_function
|
|
|
|
import sopel
|
|
import sopel.module
|
|
|
|
import os
|
|
|
|
from sopel_modules.SpiceBot_Events.System import botevents
|
|
from sopel_modules.SpiceBot_SBTools import bot_logging
|
|
|
|
import spicemanip
|
|
|
|
|
|
def setup(bot):
|
|
bot_logging(bot, 'SpiceBot_CommandsQuery', "Starting setup procedure")
|
|
botevents.startup_add([botevents.BOT_COMMANDSQUERY])
|
|
|
|
if 'SpiceBot_CommandsQuery' not in bot.memory:
|
|
bot.memory['SpiceBot_CommandsQuery'] = {"counts": 0, "commands": {}, "nickrules": []}
|
|
|
|
for comtype in ['module', 'nickname', 'rule']:
|
|
if comtype not in bot.memory['SpiceBot_CommandsQuery']['commands'].keys():
|
|
bot.memory['SpiceBot_CommandsQuery']['commands'][comtype] = dict()
|
|
|
|
filepathlisting = []
|
|
|
|
# main Modules directory
|
|
main_dir = os.path.dirname(os.path.abspath(sopel.__file__))
|
|
modules_dir = os.path.join(main_dir, 'modules')
|
|
filepathlisting.append(modules_dir)
|
|
|
|
# Home Directory
|
|
home_modules_dir = os.path.join(bot.config.homedir, 'modules')
|
|
if os.path.isdir(home_modules_dir):
|
|
filepathlisting.append(home_modules_dir)
|
|
|
|
# pypi installed
|
|
try:
|
|
import sopel_modules
|
|
for plugin_dir in set(sopel_modules.__path__):
|
|
for pathname in os.listdir(plugin_dir):
|
|
pypi_modules_dir = os.path.join(plugin_dir, pathname)
|
|
filepathlisting.append(pypi_modules_dir)
|
|
except Exception as e:
|
|
bot_logging(bot, 'SpiceBot_CommandsQuery', "sopel_modules not loaded :" + str(e))
|
|
|
|
# Extra directories
|
|
filepathlist = []
|
|
for directory in bot.config.core.extra:
|
|
filepathlisting.append(directory)
|
|
|
|
for directory in filepathlisting:
|
|
for pathname in os.listdir(directory):
|
|
path = os.path.join(directory, pathname)
|
|
if (os.path.isfile(path) and path.endswith('.py') and not path.startswith('_')):
|
|
if pathname not in ["SpiceBot_dummycommand.py"]:
|
|
filepathlist.append(str(path))
|
|
|
|
# CoreTasks
|
|
ct_path = os.path.join(main_dir, 'coretasks.py')
|
|
filepathlist.append(ct_path)
|
|
|
|
for modulefile in filepathlist:
|
|
|
|
module_file_lines = []
|
|
module_file = open(modulefile, 'r')
|
|
lines = module_file.readlines()
|
|
for line in lines:
|
|
module_file_lines.append(line)
|
|
module_file.close()
|
|
|
|
dict_from_file = dict()
|
|
|
|
detected_lines = []
|
|
for line in module_file_lines:
|
|
|
|
if str(line).startswith("@"):
|
|
line = str(line)[1:]
|
|
|
|
if str(line).startswith(tuple(["commands", "module.commands", "sopel.module.commands"])):
|
|
line = str(line).split("commands")[-1]
|
|
line = "commands" + line
|
|
detected_lines.append(line)
|
|
elif str(line).startswith(tuple(["nickname_commands", "module.nickname_commands", "sopel.module.nickname_commands"])):
|
|
line = str(line).split("nickname_commands")[-1]
|
|
line = "nickname_commands" + line
|
|
detected_lines.append(line)
|
|
elif str(line).startswith(tuple(["rule", "module.rule", "sopel.module.rule"])):
|
|
line = str(line).split("rule")[-1]
|
|
line = "rule" + line
|
|
else:
|
|
line = None
|
|
|
|
if line:
|
|
detected_lines.append(line)
|
|
|
|
if len(detected_lines):
|
|
|
|
filelinelist = []
|
|
currentsuccesslines = 0
|
|
for detected_line in detected_lines:
|
|
|
|
# Commands
|
|
if str(detected_line).startswith("commands"):
|
|
comtype = "module"
|
|
validcoms = eval(str(detected_line).split("commands")[-1])
|
|
elif str(detected_line).startswith("nickname_commands"):
|
|
comtype = "nickname"
|
|
validcoms = eval(str(detected_line).split("nickname_commands")[-1])
|
|
elif str(detected_line).startswith("rule"):
|
|
comtype = "rule"
|
|
validcoms = eval(str(detected_line).split("rule")[-1])
|
|
|
|
if isinstance(validcoms, tuple):
|
|
validcoms = list(validcoms)
|
|
else:
|
|
validcoms = [validcoms]
|
|
for regexcom in ["(.*)", '^\?(.*)']:
|
|
if regexcom in validcoms:
|
|
while regexcom in validcoms:
|
|
validcoms.remove(regexcom)
|
|
|
|
if len(validcoms):
|
|
validcomdict = {"comtype": comtype, "validcoms": validcoms}
|
|
filelinelist.append(validcomdict)
|
|
currentsuccesslines += 1
|
|
|
|
if currentsuccesslines:
|
|
bot.memory['SpiceBot_CommandsQuery']['counts'] += 1
|
|
|
|
if len(filelinelist):
|
|
for atlinefound in filelinelist:
|
|
|
|
dict_from_file = dict()
|
|
|
|
comtype = atlinefound["comtype"]
|
|
validcoms = atlinefound["validcoms"]
|
|
|
|
# default command to filename
|
|
if "validcoms" not in dict_from_file.keys():
|
|
dict_from_file["validcoms"] = validcoms
|
|
|
|
maincom = dict_from_file["validcoms"][0]
|
|
if len(dict_from_file["validcoms"]) > 1:
|
|
comaliases = spicemanip.main(dict_from_file["validcoms"], '2+', 'list')
|
|
else:
|
|
comaliases = []
|
|
|
|
bot.memory['SpiceBot_CommandsQuery']['commands'][comtype][maincom] = dict_from_file
|
|
for comalias in comaliases:
|
|
if comalias not in bot.memory['SpiceBot_CommandsQuery']['commands'][comtype].keys():
|
|
bot.memory['SpiceBot_CommandsQuery']['commands'][comtype][comalias] = {"aliasfor": maincom}
|
|
|
|
for comtype in ['module', 'nickname', 'rule']:
|
|
bot_logging(bot, 'SpiceBot_CommandsQuery', "Found " + str(len(bot.memory['SpiceBot_CommandsQuery']['commands'][comtype].keys())) + " " + comtype + " commands.", True)
|
|
|
|
for command in bot.memory['SpiceBot_CommandsQuery']['commands']['rule'].keys():
|
|
if command.startswith("$nickname"):
|
|
command = command.split("$nickname")[-1]
|
|
if command not in bot.memory['SpiceBot_CommandsQuery']['nickrules']:
|
|
bot.memory['SpiceBot_CommandsQuery']['nickrules'].append(command)
|
|
|
|
botevents.trigger(botevents.BOT_COMMANDSQUERY, "SpiceBot_CommandsQuery")
|
|
|
|
|
|
def shutdown(bot):
|
|
if "SpiceBot_CommandsQuery" in bot.memory:
|
|
del bot.memory["SpiceBot_CommandsQuery"]
|
|
|
|
|
|
@sopel.module.event(botevents.BOT_LOADED)
|
|
@sopel.module.rule('.*')
|
|
def bot_events_complete(bot, trigger):
|
|
botevents.recieved(trigger)
|
|
|
|
for comtype in bot.memory['SpiceBot_CommandsQuery']['commands'].keys():
|
|
if comtype not in ['module', 'nickname', 'rule']:
|
|
bot_logging(bot, 'SpiceBot_CommandsQuery', "Found " + str(len(bot.memory['SpiceBot_CommandsQuery']['commands'][comtype].keys())) + " " + comtype + " commands.", True)
|
|
|
|
|
|
def commandsquery_register(bot, command_type, validcoms, aliasfor=None):
|
|
|
|
if 'SpiceBot_CommandsQuery' not in bot.memory:
|
|
bot.memory['SpiceBot_CommandsQuery'] = {"counts": 0, "commands": {}, "nickrules": []}
|
|
|
|
if not isinstance(validcoms, list):
|
|
validcoms = [validcoms]
|
|
|
|
if command_type not in bot.memory['SpiceBot_CommandsQuery']['commands'].keys():
|
|
bot.memory['SpiceBot_CommandsQuery']['commands'][command_type] = dict()
|
|
bot.memory['SpiceBot_CommandsQuery']['counts'] += 1
|
|
|
|
dict_from_file = dict()
|
|
|
|
# default command to filename
|
|
if "validcoms" not in dict_from_file.keys():
|
|
dict_from_file["validcoms"] = validcoms
|
|
|
|
if not aliasfor:
|
|
|
|
maincom = dict_from_file["validcoms"][0]
|
|
if len(dict_from_file["validcoms"]) > 1:
|
|
comaliases = spicemanip.main(dict_from_file["validcoms"], '2+', 'list')
|
|
else:
|
|
comaliases = []
|
|
bot.memory['SpiceBot_CommandsQuery']['commands'][command_type][maincom] = dict_from_file
|
|
else:
|
|
comaliases = validcoms
|
|
|
|
for comalias in comaliases:
|
|
if comalias not in bot.memory['SpiceBot_CommandsQuery']['commands'][command_type].keys():
|
|
bot.memory['SpiceBot_CommandsQuery']['commands'][command_type][comalias] = {"aliasfor": aliasfor}
|
|
|
|
|
|
@sopel.module.event(botevents.BOT_COMMANDSQUERY)
|
|
@sopel.module.rule('.*')
|
|
def bot_events_setup(bot, trigger):
|
|
botevents.recieved(trigger)
|