114 lines
4.6 KiB
Python
114 lines
4.6 KiB
Python
import time
|
|
|
|
from core.chat_blob import ChatBlob
|
|
from core.command_param_types import Any, Int, Const, Options, Item
|
|
from core.decorators import instance, command
|
|
|
|
|
|
@instance()
|
|
class WantsController:
|
|
def inject(self, registry):
|
|
self.db = registry.get_instance("db")
|
|
self.text = registry.get_instance("text")
|
|
self.account_service = registry.get_instance("account_service")
|
|
|
|
def start(self):
|
|
self.db.exec("CREATE TABLE IF NOT EXISTS wants ("
|
|
"id INT PRIMARY KEY AUTO_INCREMENT, "
|
|
"char_id INT NOT NULL, "
|
|
"want TEXT NOT NULL,"
|
|
"created_at INT NOT NULL,"
|
|
"INDEX `char_id` (`char_id`))")
|
|
|
|
@command(command="wants", params=[], access_level="member",
|
|
description="Show your wants")
|
|
def wants_list_cmd(self, request):
|
|
alts = self.account_service.get_alts(request.sender.char_id)
|
|
|
|
cnt = 0
|
|
blob = ""
|
|
for alt in alts:
|
|
data = self.db.query("SELECT * FROM wants WHERE char_id = ? ORDER BY created_at DESC", [alt.char_id])
|
|
alt_cnt = len(data)
|
|
cnt += alt_cnt
|
|
|
|
if alt_cnt:
|
|
for row in data:
|
|
blob += "%s %s\n\n" % (row.want, self.text.make_tellcmd("Remove", "wants remove %d" % row.id))
|
|
|
|
return ChatBlob("Wants for %s (%d)" % (alts[0].name, cnt), blob)
|
|
|
|
@command(command="wants", params=[Const("add"), Any("item")], access_level="member",
|
|
description="Add a want")
|
|
def wants_add_cmd(self, request, _, want):
|
|
self.db.exec("INSERT INTO wants (char_id, want, created_at) "
|
|
"VALUES (?, ?, ?)",
|
|
[request.sender.char_id, want, int(time.time())])
|
|
|
|
return "Want added successfully."
|
|
|
|
@command(command="wants", params=[Options(["rem", "remove"]), Int("want_id")], access_level="member",
|
|
description="Remove a want")
|
|
def wants_remove_cmd(self, request, _, want_id):
|
|
want = self.db.query_single("SELECT n.*, p.name FROM wants n "
|
|
"LEFT JOIN player p ON n.char_id = p.char_id "
|
|
"WHERE n.id = ?",
|
|
[want_id])
|
|
|
|
if not want:
|
|
return f"Could not find want with ID <highlight>{want_id:d}</highlight>."
|
|
|
|
if self.account_service.get_main(request.sender.char_id).char_id \
|
|
!= self.account_service.get_main(want.char_id).char_id:
|
|
return f"You must be a confirmed alt of <highlight>{want.name}</highlight> to remove this want."
|
|
|
|
self.db.exec("DELETE FROM wants WHERE id = ?", [want_id])
|
|
|
|
return f"Want with ID <highlight>{want_id:d}</highlight> deleted successfully."
|
|
|
|
@command(command="wants", params=[Const("search"), Item("item")], access_level="member",
|
|
description="Search wants by itemref")
|
|
def wants_search_itemref_cmd(self, request, _, item):
|
|
return self.search_wants(item.name)
|
|
|
|
@command(command="wants", params=[Const("search"), Any("name")], access_level="member",
|
|
description="Search wants by name")
|
|
def wants_search_name_cmd(self, request, _, wants_search):
|
|
return self.search_wants(wants_search)
|
|
|
|
def search_wants(self, wants_search):
|
|
wants = self.db.query("SELECT w.char_id, w.want, p.name FROM wants w "
|
|
"LEFT JOIN player p ON w.char_id = p.char_id "
|
|
"WHERE want LIKE ?",
|
|
[f"%{wants_search}%"])
|
|
|
|
blob = ""
|
|
for want in wants:
|
|
alts = self.account_service.get_alts(want.char_id)
|
|
main_name = alts[0].name
|
|
blob += f"<header2>{main_name}</header2>\n{want.want}\n\n"
|
|
|
|
return ChatBlob("Search Results (%d)" % len(wants), blob)
|
|
|
|
@command(command="wants", params=[Const("list")], access_level="member",
|
|
description="Shows all wants")
|
|
def wants_all_cmd(self, request, _):
|
|
sql = "SELECT w.*, p.name FROM wants w \
|
|
LEFT JOIN alts a ON w.char_id = a.char_id \
|
|
LEFT JOIN alts a2 ON (a2.group_id = a.group_id AND a2.status = 2) \
|
|
LEFT JOIN player p ON p.char_id = COALESCE(a2.char_id, w.char_id) \
|
|
ORDER BY p.name"
|
|
|
|
data = self.db.query(sql)
|
|
|
|
blob = ""
|
|
current_main_name = ""
|
|
for want in data:
|
|
if want.name != current_main_name:
|
|
blob += f"\n<header2>{want.name}</header2>\n"
|
|
current_main_name = want.name
|
|
|
|
blob += want.want + "\n"
|
|
|
|
return ChatBlob(f"Wants List ({len(data):d})", blob)
|