<?php

require_once dirname(__DIR__, 3).DIRECTORY_SEPARATOR.'bootstrap/autoload.php';

use Blacklight\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);
} elseif ((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');
    }
} elseif ((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);
    }
} elseif ((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');
} elseif ((isset($argv[1]) && $argv[1] == 'test') && (isset($argv[2]) && $argv[2] == 'dump') && (isset($argv[3]) && file_exists($argv[3]))) {
    $arr = ['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);
    }
} elseif ((isset($argv[1]) && $argv[1] == 'test') && (isset($argv[2]) && $argv[2] == 'restore') && (isset($argv[3]) && file_exists($argv[3]))) {
    $arr = ['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');
} elseif ((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));
    }
} elseif ((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');
}