from core.buddy_service import BuddyService from core.chat_blob import ChatBlob from core.command_alias_service import CommandAliasService from core.command_param_types import Const, Options, Character from core.db import DB from core.decorators import instance, command from core.logger import Logger from core.lookup.pork_service import PorkService from core.setting_service import SettingService from core.text import Text from core.tyrbot import Tyrbot from core.util import Util from modules.core.accounting.services.account_service import AccountService from modules.core.discord.discord_controller import DiscordController @instance() class PreferenceController: def inject(self, registry): self.logger = Logger(__name__) self.bot: Tyrbot = registry.get_instance("bot") self.db: DB = registry.get_instance("db") self.text: Text = registry.get_instance("text") self.util: Util = registry.get_instance("util") self.account_service: AccountService = registry.get_instance("account_service") self.pork: PorkService = registry.get_instance("pork_service") self.org_pork: PorkService = registry.get_instance("org_pork_service") self.command_alias_service: CommandAliasService = registry.get_instance("command_alias_service") self.buddy_service: BuddyService = registry.get_instance("buddy_service") self.setting_service: SettingService = registry.get_instance("setting_service") self.discord: DiscordController = registry.get_instance("discord_controller") self.job_scheduler = registry.get_instance("job_scheduler") def start(self): self.command_alias_service.add_alias("prefs", "preferences") self.command_alias_service.add_alias("autoinvite", "preferences set autoinvite") @command(command="preferences", params=[], description="View your Preferences", access_level="member") def show_prefs(self, request): account = self.account_service.get_account(request.sender.char_id) if not account: return "You dont have any preferences you could manage...." return ChatBlob("Your Preferences", self.get_pref_view_full(account)) @command(command="preferences", params=[Const("set"), Options(["news", "raidinvite", "subtilespam", "autoinvite", "raidspam"]), Options(["on", "off", "yes", "no"])], description="Change your Preferences", access_level="member") def set_prefs(self, request, _, pref: str, value): account = self.account_service.get_account(request.sender.char_id) if not account: return "You dont have any preferences you could manage...." self.set_pref(request.sender.char_id, pref, 0 if value in ["off", "no"] else 1) return f"Your {pref.capitalize()} preference has been set to {value}." @command(command="prefadmin", params=[Const("set"), Character("main"), Options(["news", "raidinvite", "subtilespam", "autoinvite", "raidspam"]), Options(["on", "off", "yes", "no"])], description="Change your Preferences", sub_command='mdf', access_level="admin") def set_pref_admin(self, _, _1, main, pref: str, value): self.set_pref(main.char_id, pref, 0 if value in ["off", "no"] else 1) return f"{main.name}'s {pref.capitalize()} " \ f"preference has been set to {value}." def get_prefs(self, char_id): return self.account_service.get_account(char_id) def set_pref(self, char_id, pref, value): pref = "subtile_spam" if pref == "subtilespam" \ else "raid_invite" if pref == "raidinvite" \ else "news_spam" if pref == "news" \ else "auto_invite" if pref == "autoinvite" \ else 'raid_spam' if pref == "raidspam" \ else "" self.db.exec(f"UPDATE account set {pref}={value} where char_id=(SELECT main from account where char_id=?)", [char_id]) def get_pref_view_small(self, prefs): return f"\n" \ f" └ [{self._make_cmd('news', prefs.news_spam)}] News - " \ f"Autoinvite [{self._make_cmd('autoinvite', prefs.auto_invite)}], \n" \ f" └ [{self._make_cmd('raidinvite', prefs.raid_invite)}] Raidinvite - " \ f"Massmessage [{self._make_cmd('raidspam', prefs.raid_invite)}], \n" \ f" └ [{self._make_cmd('subtilespam', prefs.subtile_spam)}] Subtilespam" def get_pref_view_full(self, prefs): return f"\n" \ f"└ [ {self._make_cmd('news', prefs.news_spam)} ] Do you want your News on Logon? \n" \ f"└ [ {self._make_cmd('autoinvite', prefs.auto_invite)} ] Do you want to receive autoinvites? \n" \ f"└ [ {self._make_cmd('raidinvite', prefs.raid_invite)} ] Do you want to receive raidinvites? \n" \ f"└ [ {self._make_cmd('raidspam', prefs.raid_invite)} ] Do you want to receive massmessages? \n" \ f"└ [ {self._make_cmd('subtilespam', prefs.subtile_spam)} ] Do you want a subtile invite spam? \n" def _make_cmd(self, pref, value): # --- # [ ON | OFF ] if value == 1: return f"YES | {self.text.make_chatcmd('NO', f'/tell preferences set {pref} off')}" else: return f"{self.text.make_chatcmd('YES', f'/tell preferences set {pref} on')} | NO"