<?php

function BWPS_install() {
	global $wpdb;
	
	BWPS_checkVersions();
	
	$BWPSinstall = "CREATE TABLE " . BWPS_TABLE_D404 . " (
		`attempt_id` bigint(20) NOT NULL AUTO_INCREMENT,
		`computer_id` varchar(20),
		`attempt_date` int(10),
		`qstring` varchar(255),
		`referrer` varchar(255),
		PRIMARY KEY  (`attempt_id`)
		);";
		$vers['TABLE_D404'] = BWPS_VERSION_TABLE_D404;

	$BWPSinstall .= "CREATE TABLE " . BWPS_TABLE_LL . " (
		`attempt_id` bigint(20) NOT NULL AUTO_INCREMENT ,
		`attempt_date` int(10),
		`user_id` bigint(20),
		`computer_id` varchar(20),
		PRIMARY KEY  (`attempt_id`)
		);";		
		$vers['TABLE_LL'] = BWPS_VERSION_TABLE_LL;

	$BWPSinstall .= "CREATE TABLE " . BWPS_TABLE_LOCKOUTS . " (
		`lockout_id` bigint(20) NOT NULL AUTO_INCREMENT,
		`computer_id` varchar(20),
		`user_id` varchar(20),
		`lockout_date` int(10),
		`mode` int(5),
		PRIMARY KEY  (`lockout_id`)
		);";
		$vers['TABLE_LOCKOUTS'] = BWPS_VERSION_TABLE_LOCKOUTS;
			
	require_once(ABSPATH . 'wp-admin/includes/upgrade.php');
	dbDelta($BWPSinstall);
	
	$BWPS = new BWPS();
	
	$opts = $BWPS->getOptions();
		
	$htaccess = trailingslashit(ABSPATH).'.htaccess';
	
	if ($BWPS->can_write($htaccess)) {
		
		$BWPS->createhtaccess();
	
	}
	
	unset($opts);
}
	
function BWPS_uninstall() {
	global $wpdb;
	
	$BWPS = new BWPS();

	$htaccess = trailingslashit(ABSPATH).'.htaccess';
		
	if (!$BWPS->can_write($htaccess)) {
		echo "Unable to update htaccess rules";
	} else {
		
		$BWPS->remove_section($htaccess, 'Better WP Security');
		
		$del_d404 = 'DROP TABLE '. BWPS_TABLE_D404 . ';';
		$del_ll = 'DROP TABLE '. BWPS_TABLE_LL . ';';
		$del_lockouts = 'DROP TABLE '. BWPS_TABLE_LOCKOUTS . ';';
		$wpdb->query($del_d404);
		$wpdb->query($del_ll);
		$wpdb->query($del_lockouts);
	}
}

function BWPS_checkVersions() {
	if (get_option("BWPS_versions")) {
		$oldVers = unserialize(get_option("BWPS_versions"));
		$vers = BWPS_versions();
		if(count(array_diff($oldVers,$vers)) > 0) {
			delete_option('BWPS_versions');
			update_option('BWPS_update', '1');
		}
	} else {
		$vers = BWPS_versions();
		update_option('BWPS_update', '1');
	}
	update_option('BWPS_versions', serialize($vers));
}

function BWPS_versions() {
	$vers = array(
		'TABLE_D404' => '1',
		'TABLE_LL' => '0',
		'TABLE_LOCKOUTS' => '0',
		'AWAY' => '0',
		'BANVISITS' => '0',
		'TWEAKS' => '0',
		'HIDEBE' => '1',
		'LL' => '0',
		'HTACCESS' => '0',
		'IDETECT' => '3'
	);
	
	return $vers;
}

function BWPS_defaults() {
	$opts = array(
		"away_enable" => "0",
		"away_mode" => "0",
		"away_start" => "1",
		"away_end" => "1",
		"tweaks_removeGenerator" => "0",
		"tweaks_removeLoginMessages" => "0",
		"tweaks_randomVersion" => "0",
		"tweaks_themeUpdates" => "0",
		"tweaks_pluginUpdates" => "0",
		"tweaks_coreUpdates" => "0",
		"tweaks_removersd" => "0",
		"tweaks_removewlm" => "0",
		"tweaks_longurls" => "0",
		"tweaks_strongpass" => "0",
		"tweaks_strongpassrole" => "administrator",
		"htaccess_protectht" => "0",
		"htaccess_protectwpc" => "0",
		"htaccess_dirbrowse" => "0",
		"htaccess_request" => "0",
		"htaccess_qstring" => "0",
		"htaccess_protectreadme" => "0",
		"htaccess_protectinstall" => "0",
		"hidebe_enable" => "0",
		"hidebe_login_slug" => "login",
		"hidebe_admin_slug" => "admin",
		"hidebe_register_slug" => "register",
		"hidebe_canregister" => get_option('users_can_register'),
		"hidebe_key" => "",
		"ll_error_message" => "error",
		"ll_enable" => "0",
		"ll_maxattemptshost" => "5",
		"ll_maxattemptsuser" => "10",
		"ll_checkinterval" => "5",
		"ll_banperiod" => "60",
		"ll_denyaccess" => "1",
		"ll_emailnotify" => "1",
		"banvisits_enable" => "0",
		"banvisits_banlist" => "",
		"idetect_d404enable" => "0",
		"idetect_emailnotify" => "1",
		"idetect_checkint" => "300",
		"idetect_locount" => "20",
		"idetect_lolength" => "900",
		"idetect_error_message" => "error",
		"idetect_whitelist" => ""
	);
	
	return $opts;
}