<?php

require_once "SQL/sql.php";
require_once "SQL/settings.php";

class sql_db
{
	public $name = "SQLDB";
	public $author = "Valware";
	public $version = "1.0";
	public $description = "SQL database backend";
	public $email = "v.a.pond@outlook.com";

	function __construct()
	{
		Hook::func(HOOKTYPE_USER_LOOKUP, 'sql_db::get_user');
		Hook::func(HOOKTYPE_USERMETA_ADD, 'sql_db::add_usermeta');
		Hook::func(HOOKTYPE_USERMETA_DEL, 'sql_db::del_usermeta');
		Hook::func(HOOKTYPE_USERMETA_GET, 'sql_db::get_usermeta');
		Hook::func(HOOKTYPE_USER_CREATE, 'sql_db::user_create');
		Hook::func(HOOKTYPE_GET_USER_LIST, 'sql_db::get_user_list');
		Hook::func(HOOKTYPE_USER_DELETE, 'sql_db::user_delete');
		Hook::func(HOOKTYPE_EDIT_USER, 'sql_db::edit_core');
		Hook::func(HOOKTYPE_PRE_OVERVIEW_CARD, 'sql_db::add_pre_overview_card');
		Hook::func(HOOKTYPE_UPGRADE, 'sql_db::create_tables'); // handles upgrades too ;)
		AuthModLoaded::$status = 1;
	}

	public static function add_pre_overview_card($empty)
	{
		if (defined('SQL_DEFAULT_USER'))
			Message::Fail("Warning: SQL_DEFAULT_USER is set in config.php. You should remove that item now, as it is only used during installation.");
		if (defined('DEFAULT_USER'))
			Message::Fail("Warning: DEFAULT_USER is set in config.php. You should remove that item now, as it is only used during installation.");
	}

	/**
	 * Delete all the tables (of us) in the SQLdb
	 * @return void
	 */
	public static function delete_tables()
	{
		$conn = sqlnew();
		foreach(["users","user_meta", "settings", "fail2ban"] as $table_name)
			$conn->query("DROP TABLE IF EXISTS " . get_config("mysql::table_prefix") . $table_name);
	}

	/**
	 * Create the tables we'll be using in the SQLdb
	 * @return void
	 */
	public static function create_tables()
	{
		$conn = sqlnew();
		$conn->query("CREATE TABLE IF NOT EXISTS " . get_config("mysql::table_prefix") . "users (
			user_id int AUTO_INCREMENT NOT NULL,
			user_name VARCHAR(255) NOT NULL,
			user_pass VARCHAR(255) NOT NULL,
			user_email VARCHAR(255),
			user_fname VARCHAR(255),
			user_lname VARCHAR(255),
			user_bio VARCHAR(255),
			created VARCHAR(255),
			PRIMARY KEY (user_id)
		)");

		/**
		 * Patch for beta users
		 * This adds the email column to existing tables without it
		*/
		$columns = $conn->query("SHOW COLUMNS FROM " . get_config("mysql::table_prefix") . "users");
		$column_names = array();
		$c = $columns->fetchAll();

		foreach($c as $column) {
			$column_names[] = $column['Field'];
		}
		$column_exists = in_array("user_email", $column_names);
		if (!$column_exists) {
			$conn->query("ALTER TABLE " . get_config("mysql::table_prefix") . "users ADD COLUMN user_email varchar(255)");
		}

		/**
		 * Another patch for beta users
		 * This changes the size of the meta_value so we can store more
		 */
		
		$conn->query("CREATE TABLE IF NOT EXISTS " . get_config("mysql::table_prefix") . "user_meta (
			meta_id int AUTO_INCREMENT NOT NULL,
			user_id int NOT NULL,
			meta_key VARCHAR(255) NOT NULL,
			meta_value VARCHAR(255),
			PRIMARY KEY (meta_id),
			CONSTRAINT meta_key_user_id UNIQUE(meta_key,user_id)
		)");
		$conn->query("CREATE TABLE IF NOT EXISTS " . get_config("mysql::table_prefix") . "settings (
			id int AUTO_INCREMENT NOT NULL,
			setting_key VARCHAR(255) NOT NULL,
			setting_value VARCHAR(5000),
			PRIMARY KEY (id),
			UNIQUE(setting_key)
		)");
		$conn->query("CREATE TABLE IF NOT EXISTS " . get_config("mysql::table_prefix") . "fail2ban (
			id int AUTO_INCREMENT NOT NULL,
			ip VARCHAR(255) NOT NULL,
			count VARCHAR(255),
			PRIMARY KEY (id)
		)");

		/* Upgrades: */
		/* - user_meta: set charset and size */
		$c = [];
		if (($columns = $conn->query("SHOW COLUMNS FROM ".get_config("mysql::table_prefix")."user_meta")))
			$c = $columns->fetchAll();
		if (!empty($c))
			$conn->query("ALTER TABLE ".get_config("mysql::table_prefix")."user_meta CHANGE `meta_value` `meta_value` VARCHAR(5000) CHARACTER SET utf8mb3 COLLATE utf8mb3_bin NULL DEFAULT NULL");

		/* - settings: add UNIQUE(setting_key) */
		$c = [];
		if (($columns = $conn->query("SHOW INDEXES FROM ".get_config("mysql::table_prefix")."settings WHERE Key_name='setting_key'")))
			$c = $columns->fetchAll();
		if (empty($c))
		{
			$conn->query("ALTER TABLE " . get_config("mysql::table_prefix") . "settings ADD CONSTRAINT setting_key UNIQUE(setting_key)");
		}
		else
			$conn->query("ALTER TABLE ".get_config("mysql::table_prefix")."settings CHANGE setting_value setting_value VARCHAR(5000) CHARACTER SET utf8mb3 COLLATE utf8mb3_bin NULL DEFAULT NULL");
		
		/* - user_meta: add UNIQUE(meta_key,user_id) */
		$c = [];
		if (($columns = $conn->query("SHOW INDEXES FROM ".get_config("mysql::table_prefix")."user_meta WHERE Key_name='meta_key_user_id'")))
			$c = $columns->fetchAll();
		if (empty($c))
			$conn->query("ALTER TABLE " . get_config("mysql::table_prefix") . "user_meta ADD CONSTRAINT meta_key_user_id UNIQUE(meta_key,user_id)");

		/* make sure everything went well */
		$tables = ["users", "user_meta", "fail2ban", "settings"];
		$errors = 0; // counter
		$error_messages = "";
		foreach($tables as $table)
		{
			$prefix = get_config("sql::table_prefix");
			$sql = "SHOW TABLES LIKE '$prefix%'"; // SQL query to check if table exists

			$result = $conn->query($sql);
			if ($result->rowCount())
			{   /* great! */ }

			else {
				$errors++;
				strcat($error_messages,"Table '$prefix$table' was not created successfully.<br>");
			}
		}
		if (!$errors)
		{
			if (defined('DEFAULT_USER')) // we've got a default account
			{
				$lkup = new PanelUser(DEFAULT_USER['username']);

				if (!$lkup->id) // doesn't exist, add it with full privileges
				{
					$user = [];
					$user['user_name'] = DEFAULT_USER['username'];
					$user['user_pass'] = DEFAULT_USER['password'];
					$user['err'] = "";
					create_new_user($user);
				}
				$lkup = new PanelUser(DEFAULT_USER['username']);
				if (!user_can($lkup, PERMISSION_MANAGE_USERS))
					$lkup->add_permission(PERMISSION_MANAGE_USERS);
			}
			return true;
		}
		else
			return false;
	
	}

	/* We convert $u with a full user as an object ;D*/
	public static function get_user(&$u)
	{
		$id = $u['id'];
		$name = $u['name'];
		$conn = sqlnew();

		if ($id)
		{
			$prep = $conn->prepare("SELECT * FROM " . get_config("mysql::table_prefix") . "users WHERE user_id = :id LIMIT 1");
			$prep->execute(["id" => strtolower($id)]);
		}
		elseif ($name)
		{
			$prep = $conn->prepare("SELECT * FROM " . get_config("mysql::table_prefix") . "users WHERE LOWER(user_name) = :name LIMIT 1");
			$prep->execute(["name" => strtolower($name)]);
		}
		$data = NULL;
		$obj = (object) [];
		if ($prep)
			$data = $prep->fetchAll();
		if (isset($data[0]) && $data = $data[0])
		{
			$obj->id = $data['user_id'];
			$obj->username = $data['user_name'];
			$obj->passhash = $data['user_pass'];
			$obj->first_name = $data['user_fname'] ?? NULL;
			$obj->last_name = $data['user_lname'] ?? NULL;
			$obj->created = $data['created'];
			$obj->bio = $data['user_bio'];
			$obj->email = $data['user_email'];
			$obj->user_meta = (new PanelUser_Meta($obj->id))->list;
		}
		$u['object'] = $obj;
	}

	public static function get_usermeta(&$u)
	{
		$list = &$u['meta'];
		$id = $u['id'];
		$conn = sqlnew();
		if (isset($id))
		{
			$prep = $conn->prepare("SELECT * FROM " . get_config("mysql::table_prefix") . "user_meta WHERE user_id = :id");
			$prep->execute(["id" => $id]);
		}
		foreach ($prep->fetchAll() as $row)
		{
			$list[$row['meta_key']] = $row['meta_value'];
		}
	}

	public static function add_usermeta(&$meta)
	{
		$meta = $meta['meta'];
		$conn = sqlnew();
		/* check if it exists first, update it if it does */
		$query = "SELECT * FROM " . get_config("mysql::table_prefix") . "user_meta WHERE user_id = :id AND meta_key = :key";
		$stmt = $conn->prepare($query);
		$stmt->execute(["id" => $meta['id'], "key" => $meta['key']]);
		if ($stmt->rowCount()) // it exists, update instead of insert
		{
			$query = "UPDATE " . get_config("mysql::table_prefix") . "user_meta SET meta_value = :value WHERE user_id = :id AND meta_key = :key";
			$stmt = $conn->prepare($query);
			$stmt->execute($meta);
			if ($stmt->rowCount())
				return true;
			return false;
		}

		else
		{
			$query = "INSERT INTO " . get_config("mysql::table_prefix") . "user_meta (user_id, meta_key, meta_value) VALUES (:id, :key, :value)";
			$stmt = $conn->prepare($query);
			$stmt->execute($meta);
			if ($stmt->rowCount())
				return true;
			return false;
		}
	}
	public static function del_usermeta(&$meta)
	{
		$conn = sqlnew();
		$query = "DELETE FROM " . get_config("mysql::table_prefix") . "user_meta WHERE user_id = :id AND meta_key = :key";
		$stmt = $conn->prepare($query);
		$stmt->execute($meta['meta']);
		if ($stmt->rowCount())
			return true;
		return false;
	}
	public static function user_create(&$u)
	{
		$username = $u['user_name'];
		$first_name = $u['fname'] ?? NULL;
		$last_name = $u['lname'] ?? NULL;
		$password = $u['user_pass'] ?? NULL;
		$user_bio = $u['user_bio'] ?? NULL;
		$user_email = $u['user_email'] ?? NULL;
		$conn = sqlnew();
		$prep = $conn->prepare("INSERT INTO " . get_config("mysql::table_prefix") . "users (user_name, user_pass, user_fname, user_lname, user_bio, user_email, created) VALUES (:name, :pass, :fname, :lname, :user_bio, :user_email, :created)");
		$prep->execute(["name" => $username, "pass" => $password, "fname" => $first_name, "lname" => $last_name, "user_bio" => $user_bio, "user_email" => $user_email, "created" => date("Y-m-d H:i:s")]);
		if ($prep->rowCount())
			$u['success'] = true;
		else
			$u['errmsg'][] = "Could not add user";
	}

	public static function get_user_list(&$list)
	{
		$conn = sqlnew();
		$result = $conn->query("SELECT user_id FROM " . get_config("mysql::table_prefix") . "users");
		if (!$result) // impossible
		{
			die("Something went wrong.");
		}
		$userlist = [];
		while($row =  $result->fetch())
		{
			$userlist[] = new PanelUser(NULL, $row['user_id']);
		}
		if (!empty($userlist))
			$list = $userlist;
		
	}
	public static function user_delete(&$u)
	{
		$user = $u['user'];
		$query = "DELETE FROM " . get_config("mysql::table_prefix") . "users WHERE user_id = :id";
		$conn = sqlnew();
		$stmt = $conn->prepare($query);
		$stmt->execute(["id" => $user->id]);
		$deleted = $stmt->rowCount();
		if ($deleted)
		{
			$u['info'][] = "Successfully deleted user \"$user->username\"";
			$u['boolint'] =  1;
		} else {
			$u['info'][] = "Unknown error";
			$u['boolint'] = 0;
		}
	}

	public static function edit_core($arr)
	{
		$conn = sqlnew();
		$user = $arr['user'];
		$info = $arr['info'];
		foreach($info as $key => $val)
		{
			$value = NULL;
			if (!$val || !strlen($val) || BadPtr($val))
				continue;
			if (!strcmp($key,"update_fname") && $val != $user->first_name)
			{
				$value = "user_fname";
				$valuestr = "first name";
			}
			elseif (!strcmp($key,"update_lname") && $val != $user->last_name)
			{
				$value = "user_lname";
				$valuestr = "last name";
			}
			elseif (!strcmp($key,"update_bio") && $val != $user->bio)
			{
				$value = "user_bio";
				$valuestr = "bio";
			}
			elseif (!strcmp($key,"update_pass") || !strcmp($key,"update_pass_conf"))
			{
				$value = "user_pass";
				$valuestr = "password";
			}
			elseif(!strcmp($key,"update_email") && $val != $user->email)
			{
				$value = "user_email";
				$valuestr = "email address";
			}
			
			if (!$value)
				continue;
			$query = "UPDATE " . get_config("mysql::table_prefix") . "users SET $value=:value WHERE user_id = :id";
			$stmt = $conn->prepare($query);
			$stmt->execute(["value" => $val, "id" => $user->id]);

			if (!$stmt->rowCount() && $stmt->errorInfo()[0] != "00000")
				Message::Fail("Could not update $valuestr for $user->username: ".$stmt->errorInfo()[0]." (CODE: ".$stmt->errorCode().")");

			else
				Message::Success("Successfully updated the $valuestr for $user->username");
		}
	}
}


function security_check()
{
	$ip = $_SERVER['REMOTE_ADDR'];
	if (dnsbl_check($ip))
		return true;

	else if (fail2ban_check($ip))
	{

	}
}

function dnsbl_check($ip)
{
	$dnsbl_lookup = get_config("dnsbl");
	if (!$dnsbl_lookup)
		return;

	// clear variable just in case
	$listed = NULL;

	// if the IP was not given because you're an idiot, stop processing
	if (!$ip) { return; }
	
	// get the first two segments of the IPv4	
	$because = split($ip, ".");   // why you
	$you = $because[1]; 		  // gotta play
	$want = $because[2];		 // that song
	$to = $you.".".$want.".";	// so loud?
	
	// exempt local connections because sometimes they get a false positive
	if ($to == "192.168." || $to == "127.0.") { return NULL; }
	
	// you spin my IP right round, right round, to check the records baby, right round-round-round
	$reverse_ip = glue(array_reverse(split($ip, ".")), ".");
	
	// checkem
	foreach ($dnsbl_lookup as $host) {
		
		//if it was listed
		if (checkdnsrr($reverse_ip . "." . $host . ".", "A")) {
			
			//take note
			$listed = $host;
		}
	}

	// if it was safe, return NOTHING
	if (!$listed) {
		return NULL;
	}
	
	// else, you guessed it, return where it was listed
	else {
		return $listed;
	}
}

function fail2ban_check($ip)
{

}