<?php
require_once dirname(__DIR__, 3) . DIRECTORY_SEPARATOR . 'bootstrap.php';

use nntmux\db\DB;


//	This script can dump all tables or just collections/binaries/parts/partrepair/groups.

$pdo = new DB();

$exportopts = "";

//determine mysql platform Percona or Other
$mysqlplatform = exec('mysqladmin version | grep "Percona"');
if (strlen($mysqlplatform) > 0) {
	//Percona only has --innodb-optimize-keys
	$exportopts = "--opt --innodb-optimize-keys --complete-insert --skip-quick";
} else {
	//generic (or unknown) instance of MySQL
	$exportopts = "--opt --complete-insert --skip-quick";
}


function newname($filename)
{
	rename($filename, dirname($filename)."/".basename($filename,".gz")."_".date("Y_m_d_His", filemtime($filename)).".gz");
}

function builddefaultsfile()
{
	//generate file contents
	$filetext = "[mysqldump]"
				."\n"
				."user = " . env('DB_USER')
				."\n"
				."password = " . env('DB_PASSWORD')
				."\n[mysql]"
				."\n"
				."user = " . env('DB_USER')
				."\n"
				."password = " . env('DB_PASSWORD');

	$filehandle = fopen("mysql-defaults.txt", "w+");
	if(!$filehandle) {
		exit("Unable to write mysql defaults file! Exiting");
	} else {
		fwrite($filehandle, $filetext);
		fclose($filehandle);
		chmod("mysql-defaults.txt", 0600);
	}
}

$dbhost = env('DB_HOST');
$dbport = env('DB_PORT');
$dbsocket = env('DB_SOCKET');
$dbuser = env('DB_USER');
$dbpass = env('DB_PASSWORD');
$dbname = env(env('DB_NAME'));

if (env('DB_SOCKET') !== '') {
	$use = "-S $dbsocket";
} else {
	$use = "-P$dbport";
}

//generate defaults file used to store database login information so it is not in cleartext in ps command for mysqldump
builddefaultsfile();

if((isset($argv[1]) && $argv[1] == "db") && (isset($argv[2]) && $argv[2] == "dump") && (isset($argv[3]) && file_exists($argv[3]))) {
	$filename = $argv[3]."/".$dbname.".gz";
	echo $pdo->log->header("Dumping $dbname.");
	if (file_exists($filename)) {
		newname($filename);
	}
	$command = "mysqldump --defaults-file=mysql-defaults.txt $exportopts -h$dbhost $use "."$dbname | gzip -9 > $filename";
	system($command);
} else if((isset($argv[1]) && $argv[1] == "db") && (isset($argv[2]) && $argv[2] == "restore") && (isset($argv[3]) && file_exists($argv[3]))) {
	$filename = $argv[3]."/".$dbname.".gz";
	if (file_exists($filename)) {
		echo $pdo->log->header("Restoring $dbname.");
		$command = "zcat < $filename | mysql --defaults-file=mysql-defaults.txt -h$dbhost $use $dbname";
		$pdo->queryExec("SET FOREIGN_KEY_CHECKS=0");
		system($command);
		$pdo->queryExec("SET FOREIGN_KEY_CHECKS=1");
	}
} else if((isset($argv[1]) && $argv[1] == "all") && (isset($argv[2]) && $argv[2] == "dump") && (isset($argv[3]) && file_exists($argv[3]))) {
	$sql = "SHOW tables";
	$tables = $pdo->query($sql);
	foreach($tables as $row) {
		$tbl = $row['Tables_in_'. env('DB_NAME')];
		$filename = $argv[3]."/".$tbl.".gz";
		echo $pdo->log->header("Dumping $tbl.");
		if (file_exists($filename)) {
			newname($filename);
		}
		$command = "mysqldump --defaults-file=mysql-defaults.txt $exportopts -h$dbhost $use "."$dbname $tbl | gzip -9 > $filename";
		system($command);
	}
} else if((isset($argv[1]) && $argv[1] == "all") && (isset($argv[2]) && $argv[2] == "restore") && (isset($argv[3]) && file_exists($argv[3]))) {
	$sql = "SHOW tables";
	$tables = $pdo->query($sql);
	$pdo->queryExec("SET FOREIGN_KEY_CHECKS=0");
	foreach($tables as $row) {
		$tbl = $row['Tables_in_'.env('DB_NAME')];
		$filename = $argv[3]."/".$tbl.".gz";
		if (file_exists($filename)) {
			echo $pdo->log->header("Restoring $tbl.");
			$command = "zcat < $filename | mysql --defaults-file=mysql-defaults.txt -h$dbhost $use $dbname";
			system($command);
		}
	}
	$pdo->queryExec("SET FOREIGN_KEY_CHECKS=1");
} else if((isset($argv[1]) && $argv[1] == "test") && (isset($argv[2]) && $argv[2] == "dump") && (isset($argv[3]) && file_exists($argv[3]))) {
	$arr = array("parts", "binaries", "missed_parts", "groups");
	foreach ($arr as &$tbl) {
		$filename = $argv[3]."/".$tbl.".gz";
		echo $pdo->log->header("Dumping $tbl..");
		if (file_exists($filename)) {
			newname($filename);
		}
		$command = "mysqldump --defaults-file=mysql-defaults.txt $exportopts -h$dbhost $use "."$dbname $tbl | gzip -9 > $filename";
		system($command);
	}
} else if((isset($argv[1]) && $argv[1] == "test") && (isset($argv[2]) && $argv[2] == "restore") && (isset($argv[3]) && file_exists($argv[3]))) {
	$arr = array("parts", "binaries", "missed_parts", "groups");
	$pdo->queryExec("SET FOREIGN_KEY_CHECKS=0");
	foreach ($arr as &$tbl) {
		$filename = $argv[3]."/".$tbl.".gz";
		if (file_exists($filename)) {
			echo $pdo->log->header("Restoring $tbl.");
			$command = "zcat < $filename | mysql --defaults-file=mysql-defaults.txt -h$dbhost $use $dbname";
			system($command);
		}
	}
	$pdo->queryExec("SET FOREIGN_KEY_CHECKS=1");
} else if((isset($argv[1]) && $argv[1] == "all") && (isset($argv[2]) && $argv[2] == "outfile") && (isset($argv[3]) && file_exists($argv[3]))) {
	$sql = "SHOW tables";
	$tables = $pdo->query($sql);
	foreach($tables as $row) {
		$tbl = $row['Tables_in_'.env('DB_NAME')];
		$filename = $argv[3].$tbl.".csv";
		echo $pdo->log->header("Dumping $tbl.");
		if (file_exists($filename)) {
			newname($filename);
		}
		$pdo->queryDirect(sprintf("SELECT * INTO OUTFILE %s FROM %s", $pdo->escapeString($filename), $tbl));
	}
} else if((isset($argv[1]) && $argv[1] == "all") && (isset($argv[2]) && $argv[2] == "infile") && (isset($argv[3]) && is_dir($argv[3]))) {
	$sql = "SHOW tables";
	$tables = $pdo->query($sql);
	$pdo->queryExec("SET FOREIGN_KEY_CHECKS=0");
	foreach($tables as $row) {
		$tbl = $row['Tables_in_'.env('DB_NAME')];
		$filename = $argv[3].$tbl.".csv";
		if (file_exists($filename)) {
			echo $pdo->log->header("Restoring $tbl.");
			$pdo->queryExec(sprintf("LOAD DATA INFILE %s INTO TABLE %s", $pdo->escapeString($filename), $tbl));
		}
	}
	$pdo->queryExec("SET FOREIGN_KEY_CHECKS=1");
} else {
	passthru("clear");
	echo $pdo->log->error("\nThis script can dump/restore all tables, compressed or OUTFILE/INFILE, or just collections/binaries/parts.\n\n"
	. "**Single File\n"
	. "php $argv[0] db dump /path/to/save/to              ...: To dump the database.\n"
	. "php $argv[0] db restore /path/to/restore/from      ...: To restore the database.\n\n"
	. "**Individual Table Files\n"
	. "php $argv[0] all dump /path/to/save/to             ...: To dump all tables.\n"
	. "php $argv[0] all restore /path/to/restore/from     ...: To restore all tables.\n\n"
	. "**Three Tables (collections, binaries, parts)\n"
	. "php $argv[0] test dump /path/to/save/to            ...: To dump binaries and parts tables.\n"
	. "php $argv[0] test restore /path/to/restore/from    ...: To restore binaries and parts tables.\n\n"
	. "**Individal Files - OUTFILE/INFILE - No schema\n"
	. "**MySQL MUST have write permissions to this path\n"
	. "php $argv[0] all outfile /path/to/save/to          ...: To dump all tables, using OUTFILE.\n"
	. "php $argv[0] all infile /path/to/restore/from      ...: To restore all tables, using INFILE.\n\n");
}

if(file_exists("mysql-defaults.txt")) {
	@unlink("mysql-defaults.txt");
}