// ===========================================================================
// Vortrex's Roleplay Resource
// https://github.com/VortrexFTW/gtac_roleplay
// ===========================================================================
// FILE: bans.js
// DESC: Provides ban functions and usage
// TYPE: Server (JavaScript)
// ===========================================================================

// Ban Types
const AGRP_BANTYPE_NONE = 0;
const AGRP_BANTYPE_ACCOUNT = 1;
const AGRP_BANTYPE_SUBACCOUNT = 2;
const AGRP_BANTYPE_IPADDRESS = 3;
const AGRP_BANTYPE_SUBNET = 4;

// ===========================================================================

class BanData {
	constructor(dbAssoc = false) {
		this.databaseId = 0;
		this.type = AGRP_BANTYPE_NONE;
		this.detail = "";
		this.ipAddress = "";
		this.name = "";
		this.reason = "";

		if (dbAssoc) {
			this.databaseId = toInteger(dbAssoc["ban_id"]);
			this.type = dbAssoc["ban_type"];
			this.detail = toInteger(dbAssoc["ban_detail"]);
			this.ipAddress = toInteger(dbAssoc["ban_ip"]);
			this.reason = toInteger(dbAssoc["ban_reason"]);
		}
	}
}

// ===========================================================================

function initBanScript() {
	logToConsole(LOG_INFO, "[VRR.Ban]: Initializing ban script ...");
	logToConsole(LOG_INFO, "[VRR.Ban]: Ban script initialized!");
}

// ===========================================================================

function accountBanCommand(command, params, client) {
	if (areParamsEmpty(params)) {
		messagePlayerSyntax(client, getCommandSyntaxText(command));
		return false;
	}

	let splitParams = params.split(" ");
	let targetClient = getPlayerFromParams(getParam(params, " ", 1));
	let reason = splitParams.slice(1).join(" ");

	if (!targetClient) {
		messagePlayerError(client, "That player is not connected!")
		return false;
	}

	// Prevent banning admins with really high permissions
	if (doesPlayerHaveStaffPermission(targetClient, "ManageServer") || doesPlayerHaveStaffPermission(targetClient, "Developer")) {
		messagePlayerError(client, getLocaleString(client, "CantBanPlayer"));
		return false;
	}

	logToConsole(LOG_WARN, `[VRR.Ban]: ${getPlayerDisplayForConsole(targetClient)} (${getPlayerData(targetClient).accountData.name}) account was banned by ${getPlayerDisplayForConsole(client)}. Reason: ${reason}`);

	announceAdminAction(`PlayerAccountBanned`, `{ALTCOLOUR}${getPlayerName(targetClient)}{MAINCOLOUR}`);
	banAccount(getPlayerData(targetClient).accountData.databaseId, getPlayerData(client).accountData.databaseId, reason);

	getPlayerData(targetClient).customDisconnectReason = `Banned - ${reason}`;
	disconnectPlayer(targetClient);
}

// ===========================================================================

function subAccountBanCommand(command, params, client, fromDiscord) {
	if (areParamsEmpty(params)) {
		messagePlayerSyntax(client, getCommandSyntaxText(command));
		return false;
	}

	let splitParams = params.split(" ");
	let targetClient = getPlayerFromParams(getParam(params, " ", 1));
	let reason = splitParams.slice(1).join(" ");

	if (!targetClient) {
		messagePlayerError(client, "That player is not connected!")
		return false;
	}

	// Prevent banning admins with really high permissions
	if (doesPlayerHaveStaffPermission(targetClient, "ManageServer") || doesPlayerHaveStaffPermission(targetClient, "Developer")) {
		messagePlayerError(client, getLocaleString(client, "CantBanPlayer"));
		return false;
	}

	logToConsole(LOG_WARN, `[VRR.Ban]: ${getPlayerDisplayForConsole(targetClient)} (${getPlayerData(targetClient).accountData.name})'s subaccount was banned by ${getPlayerDisplayForConsole(client)}. Reason: ${reason}`);

	announceAdminAction(`PlayerCharacterBanned`, `{ALTCOLOUR}${getPlayerName(targetClient)}{MAINCOLOUR}`);
	banSubAccount(getPlayerData(targetClient).currentSubAccountData.databaseId, getPlayerData(client).accountData.databaseId, reason);

	getPlayerData(targetClient).customDisconnectReason = `Banned - ${reason}`;
	disconnectPlayer(targetClient);
}

// ===========================================================================

function ipBanCommand(command, params, client, fromDiscord) {
	if (areParamsEmpty(params)) {
		messagePlayerSyntax(client, getCommandSyntaxText(command));
		return false;
	}

	let splitParams = params.split(" ");
	let targetClient = getPlayerFromParams(getParam(params, " ", 1));
	let reason = splitParams.slice(1).join(" ");

	if (!targetClient) {
		messagePlayerError(client, "That player is not connected!")
		return false;
	}

	// Prevent banning admins with really high permissions
	if (doesPlayerHaveStaffPermission(targetClient, "ManageServer") || doesPlayerHaveStaffPermission(targetClient, "Developer")) {
		messagePlayerError(client, getLocaleString(client, "CantBanPlayer"));
		return false;
	}

	announceAdminAction(`PlayerIPBanned`, `{ALTCOLOUR}${getPlayerName(targetClient)}{MAINCOLOUR}`);
	banIPAddress(getPlayerIP(targetClient), getPlayerData(client).accountData.databaseId, reason);

	getPlayerData(targetClient).customDisconnectReason = `IP Banned - ${reason}`;
	serverBanIP(getPlayerIP(targetClient));
	disconnectPlayer(targetClient);
}

// ===========================================================================

function subNetBanCommand(command, params, client, fromDiscord) {
	if (areParamsEmpty(params)) {
		messagePlayerSyntax(client, getCommandSyntaxText(command));
		return false;
	}

	let splitParams = params.split(" ");
	let targetClient = getPlayerFromParams(getParam(params, " ", 1));
	let octetAmount = Number(getParam(params, " ", 2));
	let reason = splitParams.slice(2).join(" ");

	if (!targetClient) {
		messagePlayerError(client, "That player is not connected!")
		return false;
	}

	// Prevent banning admins with really high permissions
	if (doesPlayerHaveStaffPermission(targetClient, "ManageServer") || doesPlayerHaveStaffPermission(targetClient, "Developer")) {
		messagePlayerError(client, getLocaleString(client, "CantBanPlayer"));
		return false;
	}

	announceAdminAction(`PlayerSubNetBanned`, `{ALTCOLOUR}${getPlayerName(client)}{MAINCOLOUR}`);
	banSubNet(getPlayerIP(targetClient), getSubNet(getPlayerIP(targetClient), octetAmount), getPlayerData(client).accountData.databaseId, reason);

	getPlayerData(client).customDisconnectReason = `IP Subnet Banned - ${reason}`;
	serverBanIP(getPlayerIP(targetClient));
}

// ===========================================================================

function banAccount(accountId, adminAccountId, reason) {
	let dbConnection = connectToDatabase();
	if (dbConnection) {
		let safeReason = dbConnection.escapetoString(reason);
		let dbQuery = queryDatabase(dbConnection, `INSERT INTO ban_main (ban_type, ban_detail, ban_who_banned, ban_reason) VALUES (${AGRP_BANTYPE_ACCOUNT}, ${accountId}, ${adminAccountId}, '${safeReason}');`);
		freeDatabaseQuery(dbQuery);
		dbConnection.close();
		return true;
	}

	return false;
}

// ===========================================================================

function banSubAccount(subAccountId, adminAccountId, reason) {
	let dbConnection = connectToDatabase();
	if (dbConnection) {
		let safeReason = dbConnection.escapetoString(reason);
		let dbQuery = queryDatabase(dbConnection, `INSERT INTO ban_main (ban_type, ban_detail, ban_who_banned, ban_reason) VALUES (${AGRP_BANTYPE_SUBACCOUNT}, ${subAccountId}, ${adminAccountId}, '${safeReason}');`);
		freeDatabaseQuery(dbQuery);
		dbConnection.close();
		return true;
	}

	return false;
}

// ===========================================================================

function banIPAddress(ipAddress, adminAccountId, reason) {
	let dbConnection = connectToDatabase();
	if (dbConnection) {
		let safeReason = dbConnection.escapetoString(reason);
		let dbQuery = queryDatabase(dbConnection, `INSERT INTO ban_main (ban_type, ban_detail, ban_who_banned, ban_reason) VALUES (${AGRP_BANTYPE_IPADDRESS}, INET_ATON(${ipAddress}), ${adminAccountId}, '${safeReason}');`);
		freeDatabaseQuery(dbQuery);
		dbConnection.close();
		return true;
	}

	return false;
}

// ===========================================================================

function banSubNet(ipAddressStart, ipAddressEnd, adminAccountId, reason) {
	let dbConnection = connectToDatabase();
	if (dbConnection) {
		let safeReason = dbConnection.escapetoString(reason);
		let dbQuery = queryDatabase(dbConnection, `INSERT INTO ban_main (ban_type, ban_ip_start, ban_ip_end, ban_who_banned, ban_reason) VALUES (${AGRP_BANTYPE_SUBNET}, INET_ATON(${ipAddressStart}), INET_ATON(${ipAddressEnd}), ${adminAccountId}, '${safeReason}');`);
		freeDatabaseQuery(dbQuery);
		dbConnection.close();
		return true;
	}

	return false;
}

// ===========================================================================

function unbanAccount(accountId, adminAccountId) {
	let dbConnection = connectToDatabase();
	if (dbConnection) {
		let dbQuery = queryDatabase(dbConnection, `UPDATE ban_main SET ban_who_removed=${adminAccountId}, ban_removed=1 WHERE ban_type=${AGRP_BANTYPE_ACCOUNT} AND ban_detail=${accountId}`);
		freeDatabaseQuery(dbQuery);
		dbConnection.close();
		return true;
	}

	return false;
}

// ===========================================================================

function unbanSubAccount(subAccountId, adminAccountId) {
	let dbConnection = connectToDatabase();
	if (dbConnection) {
		let dbQuery = queryDatabase(dbConnection, `UPDATE ban_main SET ban_who_removed=${adminAccountId}, ban_removed=1 WHERE ban_type=${AGRP_BANTYPE_SUBACCOUNT} AND ban_detail=${subAccountId}`);
		freeDatabaseQuery(dbQuery);
		dbConnection.close();
		return true;
	}

	return false;
}

// ===========================================================================

function unbanIPAddress(ipAddress, adminAccountId) {
	let dbConnection = connectToDatabase();
	if (dbConnection) {
		let dbQuery = queryDatabase(dbConnection, `UPDATE ban_main SET ban_who_removed=${adminAccountId}, ban_removed=1 WHERE ban_type=${AGRP_BANTYPE_IPADDRESS} AND ban_detail=INET_ATON(${ipAddress})`);
		freeDatabaseQuery(dbQuery);
		dbConnection.close();
		return true;
	}

	return false;
}

// ===========================================================================

function unbanSubNet(ipAddressStart, ipAddressEnd, adminAccountId) {
	let dbConnection = connectToDatabase();
	if (dbConnection) {
		let dbQuery = queryDatabase(dbConnection, `UPDATE ban_main SET ban_who_removed=${adminAccountId}, ban_removed=1 WHERE ban_type=${AGRP_BANTYPE_SUBNET} AND ban_ip_start=INET_ATON(${ipAddressStart}) AND ban_ip_end=INET_ATON(${ipAddressEnd})`);
		freeDatabaseQuery(dbQuery);
		dbConnection.close();
		return true;
	}

	return false;
}

// ===========================================================================

function isAccountBanned(accountId) {
	let bans = getServerData().bans.filter(ban => ban.type === AGRP_BANTYPE_ACCOUNT && ban.detail === accountId);
	if (bans.length > 0) {
		return true;
	}

	return false;
}

// ===========================================================================

function isSubAccountBanned(subAccountId) {
	let bans = getServerData().bans.filter(ban => ban.type === AGRP_BANTYPE_SUBACCOUNT && ban.detail === subAccountId);
	if (bans.length > 0) {
		return true;
	}

	return false;
}

// ===========================================================================

function isIpAddressBanned(ipAddress) {
	let bans = getServerData().bans.filter(ban => ban.type === AGRP_BANTYPE_IPADDRESS && ban.detail === ipAddress);
	if (bans.length > 0) {
		return true;
	}

	return false;
}

// ===========================================================================