newznab-tmux/lib/testing/DB/mysqldump_tables.php

<?php
require_once(dirname(__FILE__) . "/../../../bin/config.php");
require_once(WWW_DIR . "/lib/framework/db.php");


//	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 = " . DB_USER
				."\n"
				."password = " . DB_PASSWORD
				."\n[mysql]"
				."\n"
				."user = " . DB_USER
				."\n"
				."password = " . 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 = DB_HOST;
$dbport = DB_PORT;
$dbsocket = DB_SOCKET;
$dbuser = DB_USER;
$dbpass = DB_PASSWORD;
$dbname = DB_NAME;

if (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_' . 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_' . 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", "partrepair", "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", "partrepair", "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_' . 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_' . 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");
}