254 lines
9.5 KiB
Python
254 lines
9.5 KiB
Python
from datetime import datetime
|
|
|
|
from django.db import connection
|
|
|
|
from django_coreprotect.models import CoWorld, GUIResult
|
|
from django_coreprotect.utils import safe_int, checkbox, time_to_form
|
|
|
|
|
|
class GUIForm:
|
|
|
|
def __init__(self):
|
|
self.block_break = False
|
|
self.block_place = False
|
|
self.chat = False
|
|
self.chest_use = False
|
|
self.command = False
|
|
self.interact = False
|
|
self.login_logout = False
|
|
self.sign_place = False
|
|
self.worlds = []
|
|
self.ignore_environment = False
|
|
self.oldest_first = False
|
|
self.page = ""
|
|
self.page_size = ""
|
|
self.start = ""
|
|
self.players = ""
|
|
self.x = ""
|
|
self.y = ""
|
|
self.z = ""
|
|
self.radius = ""
|
|
self.blocks = ""
|
|
self.date_from = ""
|
|
self.date_to = ""
|
|
|
|
|
|
def gui_data(request):
|
|
request_data = request.GET
|
|
form = GUIForm()
|
|
worlds = CoWorld.objects.all()
|
|
|
|
# Actions
|
|
form.block_break = checkbox(request_data["block_break"]) if "block_break" in request_data else False
|
|
form.block_place = checkbox(request_data["block_place"]) if "block_place" in request_data else False
|
|
form.chat = checkbox(request_data["chat"]) if "chat" in request_data else False
|
|
form.chest_use = checkbox(request_data["chest_use"]) if "chest_use" in request_data else False
|
|
form.command = checkbox(request_data["command"]) if "command" in request_data else False
|
|
form.interact = checkbox(request_data["interact"]) if "interact" in request_data else False
|
|
form.login_logout = checkbox(request_data["login_logout"]) if "login_logout" in request_data else False
|
|
form.sign_place = checkbox(request_data["sign_place"]) if "sign_place" in request_data else False
|
|
|
|
# Permissions
|
|
if not request.user.has_perm("django_coreprotect.gui_extra"):
|
|
form.chat = False
|
|
form.command = False
|
|
|
|
# Worlds
|
|
request_worlds = request_data.getlist("world")
|
|
for world in worlds:
|
|
world_id = "world_{}".format(world.id)
|
|
w = {
|
|
"id": str(world.id),
|
|
"world_id": world_id,
|
|
"checked": True if str(world.id) in request_worlds else False,
|
|
"name": world.world
|
|
}
|
|
form.worlds.append(w)
|
|
|
|
# Options
|
|
form.ignore_environment = checkbox(request_data["ignore_environment"]) if "ignore_environment" in request_data else False
|
|
form.oldest_first = checkbox(request_data["oldest_first"]) if "oldest_first" in request_data else False
|
|
|
|
# Limit Results
|
|
form.page = request_data["page"] if "page" in request_data else "0"
|
|
form.page_size = request_data["page_size"] if "page_size" in request_data else "20"
|
|
if safe_int(form.page_size) == 0 or safe_int(form.page_size) > 1000:
|
|
form.page_size = "1000"
|
|
form.start = (safe_int(form.page) * safe_int(form.page_size))
|
|
|
|
# Players
|
|
form.players = request_data["players"] if "players" in request_data else ""
|
|
|
|
# Coordinates
|
|
form.x = request_data["x"] if "x" in request_data else ""
|
|
form.y = request_data["y"] if "y" in request_data else ""
|
|
form.z = request_data["z"] if "z" in request_data else ""
|
|
form.radius = request_data["radius"] if "radius" in request_data else ""
|
|
|
|
# Blocks
|
|
form.blocks = request_data["blocks"] if "blocks" in request_data else ""
|
|
|
|
# Date and Time
|
|
form.date_from = time_to_form(request_data["date_from"]) if "date_from" in request_data else ""
|
|
if form.date_from:
|
|
form.date_from = form.date_from.timestamp()
|
|
form.date_to = time_to_form(request_data["date_to"]) if "date_to" in request_data else ""
|
|
if form.date_to:
|
|
form.date_to = form.date_to.timestamp()
|
|
|
|
return form
|
|
|
|
|
|
def gui_results(form):
|
|
queries = []
|
|
ignore_environment = " AND cu.user NOT LIKE '#%%' " if form.ignore_environment else ""
|
|
oldest_first = " ASC " if form.oldest_first else " DESC "
|
|
|
|
coords = []
|
|
radius = form.radius if form.radius else "0"
|
|
if form.x:
|
|
coords.append(" x BETWEEN {0} - {1} AND {0} + {1}".format(form.x, radius))
|
|
if form.y:
|
|
coords.append(" y BETWEEN {0} - {1} AND {0} + {1}".format(form.y, radius))
|
|
if form.z:
|
|
coords.append(" z BETWEEN {0} - {1} AND {0} + {1}".format(form.z, radius))
|
|
coords_clause = ""
|
|
if coords:
|
|
coords_clause = " AND ({})".format(" AND ".join(coords))
|
|
|
|
players = []
|
|
players_clause = ""
|
|
if form.players:
|
|
for player in form.players.split(","):
|
|
players.append(player.strip())
|
|
players_clause = " AND ({})".format(" OR ".join(["cu.user LIKE '{}%%'".format(p) for p in players]))
|
|
|
|
blocks = []
|
|
blocks_clause = ""
|
|
chat_clause = ""
|
|
command_clause = ""
|
|
if form.blocks:
|
|
for block in form.blocks.split(","):
|
|
blocks.append(block.strip())
|
|
blocks_clause = " AND ({})".format(" OR ".join(["cmm.material LIKE 'minecraft:{}%%'".format(b) for b in blocks]))
|
|
chat_clause = " AND ({})".format(" OR ".join(["cc.message LIKE '%%{}%%'".format(b) for b in blocks]))
|
|
command_clause = " AND ({})".format(" OR ".join(["cc.message LIKE '%%{}%%'".format(b) for b in blocks]))
|
|
|
|
worlds_clause = ""
|
|
worlds = [world["id"] for world in form.worlds if world["checked"]]
|
|
if len(worlds):
|
|
worlds_clause = " AND cw.rowid IN ({})".format(",".join(worlds))
|
|
|
|
time_clause = ""
|
|
if form.date_from or form.date_to:
|
|
df, dt = form.date_from, form.date_to
|
|
if form.date_from and not form.date_to:
|
|
dt = datetime.now().timestamp()
|
|
if form.date_to and not form.date_from:
|
|
df = datetime.now().timestamp()
|
|
time_clause = " GROUP BY unix HAVING unix BETWEEN {} AND {} ".format(df, dt)
|
|
|
|
# Block Break, Block Place, and Interact
|
|
block_actions = []
|
|
if form.block_break:
|
|
block_actions.append("0")
|
|
if form.block_place:
|
|
block_actions.append("1")
|
|
if form.interact:
|
|
block_actions.append("2")
|
|
if len(block_actions):
|
|
queries.append('''SELECT
|
|
0 AS id, "block" AS type, cb.time AS unix, cu.user AS player, cb.action, cmm.material AS data, cb.x, cb.y, cb.z, cw.world
|
|
FROM co_block cb
|
|
JOIN co_user cu ON cb.user = cu.rowid
|
|
JOIN co_material_map cmm ON cb.type = cmm.rowid
|
|
JOIN co_world cw ON cb.wid = cw.rowid
|
|
WHERE cb.action IN ({action})
|
|
{ignore_environment}
|
|
{players}
|
|
{coords}
|
|
{blocks}
|
|
{worlds}
|
|
{time}
|
|
'''.format(action=",".join(block_actions), ignore_environment=ignore_environment, players=players_clause,
|
|
coords=coords_clause, blocks=blocks_clause, worlds=worlds_clause, time=time_clause))
|
|
|
|
# Chat
|
|
if form.chat:
|
|
queries.append('''SELECT
|
|
0 AS id, "chat" AS type, cc.time AS unix, cu.user AS player, "" AS action, cc.message AS data, "" AS x, "" AS y, "" AS z, "" AS world
|
|
FROM co_chat cc
|
|
JOIN co_user cu ON cc.user = cu.rowid
|
|
WHERE 1 = 1
|
|
{players}
|
|
{chat}
|
|
{time}
|
|
'''.format(players=players_clause, chat=chat_clause, time=time_clause))
|
|
|
|
# Chest Use
|
|
if form.chest_use:
|
|
concat = 'CONCAT(cc.amount, " ", cmm.material)'
|
|
if connection.vendor == "sqlite":
|
|
concat = 'cc.amount || " " || cmm.material'
|
|
queries.append('''SELECT
|
|
0 AS id, "container" AS type, cc.time AS unix, cu.user AS player, cc.action, {concat} AS data, cc.x, cc.y, cc.z, cw.world
|
|
FROM co_container cc
|
|
JOIN co_user cu ON cc.user = cu.rowid
|
|
JOIN co_material_map cmm ON cc.type = cmm.rowid
|
|
JOIN co_world cw ON cc.wid = cw.rowid
|
|
WHERE 1 = 1
|
|
{worlds}
|
|
{players}
|
|
{coords}
|
|
{blocks}
|
|
{time}
|
|
'''.format(concat=concat, worlds=worlds_clause, players=players_clause, coords=coords_clause, blocks=blocks_clause, time=time_clause))
|
|
|
|
# Commands
|
|
if form.command:
|
|
queries.append('''SELECT
|
|
0 AS id, "command" AS type, cc.time AS unix, cu.user AS player, "" AS action, cc.message AS data, "" AS x, "" AS y, "" AS z, "" AS world
|
|
FROM co_command cc
|
|
JOIN co_user cu ON cc.user = cu.rowid
|
|
WHERE 1 = 1
|
|
{players}
|
|
{command}
|
|
{coords}
|
|
{time}
|
|
'''.format(players=players_clause, command=command_clause, coords=coords_clause, time=time_clause))
|
|
|
|
# Login/Logout
|
|
if form.login_logout:
|
|
queries.append('''SELECT
|
|
0 AS id, "session" AS type, cs.time AS unix, cu.user AS player, cs.action, "" AS data, cs.x, cs.y, cs.z, cw.world
|
|
FROM co_session cs
|
|
JOIN co_user cu ON cs.user = cu.rowid
|
|
JOIN co_world cw ON cs.wid = cw.rowid
|
|
WHERE 1 = 1
|
|
{worlds}
|
|
{players}
|
|
{coords}
|
|
{time}
|
|
'''.format(worlds=worlds_clause, players=players_clause, coords=coords_clause, time=time_clause))
|
|
|
|
# Sign Place
|
|
if form.sign_place:
|
|
queries.append('''SELECT
|
|
0 AS id, "sign" AS type, cs.time AS unix, cu.user AS player, "" AS action, cs.line_1 || "|" || cs.line_2 || "|" || cs.line_3 || "|" || cs.line_4 AS data, cs.x, cs.y, cs.z, cw.world
|
|
FROM co_sign cs
|
|
JOIN co_user cu ON cs.user = cu.rowid
|
|
JOIN co_world cw ON cs.wid = cw.rowid
|
|
WHERE 1 = 1
|
|
{worlds}
|
|
{players}
|
|
{coords}
|
|
{time}
|
|
'''.format(worlds=worlds_clause, players=players_clause, coords=coords_clause, time=time_clause))
|
|
|
|
query = " UNION ALL ".join(queries)
|
|
if query:
|
|
query += " ORDER BY unix {} LIMIT {}, {}".format(oldest_first, form.start, form.page_size)
|
|
if query:
|
|
return GUIResult.objects.raw(query)
|
|
return []
|