/home/arranoyd/www/wp-content/plugins/migrate-guru/callback/wings/db.php
<?php
if (!defined('ABSPATH')) exit;
if (!class_exists('BVDBCallback')) :
require_once dirname( __FILE__ ) . '/../streams.php';
class BVDBCallback extends BVCallbackBase {
public $db;
public $stream;
public $account;
public static $bvTables = array("fw_requests", "lp_requests", "ip_store");
public function __construct($callback_handler) {
$this->db = $callback_handler->db;
$this->account = $callback_handler->account;
$this->siteinfo = $callback_handler->siteinfo;
}
public function getLastID($pkeys, $end_row) {
$last_ids = array();
foreach($pkeys as $pk) {
$last_ids[$pk] = $end_row[$pk];
}
return $last_ids;
}
public function getTableData($table, $tname, $rcount, $offset, $limit, $bsize, $filter, $pkeys, $include_rows = false) {
$tinfo = array();
$rows_count = $this->db->rowsCount($table);
$result = array('count' => $rows_count);
if ($limit == 0) {
$limit = $rows_count;
}
$srows = 1;
while (($limit > 0) && ($srows > 0)) {
if ($bsize > $limit)
$bsize = $limit;
$rows = $this->db->getTableContent($table, '*', $filter, $bsize, $offset);
$srows = sizeof($rows);
$data = array();
$data["table_name"] = $tname;
$data["offset"] = $offset;
$data["size"] = $srows;
$serialized_rows = serialize($rows);
$data['md5'] = md5($serialized_rows);
$data['length'] = strlen($serialized_rows);
array_push($tinfo, $data);
if (!empty($pkeys) && $srows > 0) {
$end_row = end($rows);
$last_ids = $this->getLastID($pkeys, $end_row);
$data['last_ids'] = $last_ids;
$result['last_ids'] = $last_ids;
}
if ($include_rows) {
$data["rows"] = $rows;
$str = serialize($data);
$this->stream->writeStream($str);
}
$offset += $srows;
$limit -= $srows;
}
$result['size'] = $offset;
$result['tinfo'] = $tinfo;
return $result;
}
public function getCreateTableQueries($tables) {
$resp = array();
foreach($tables as $table) {
$tname = urldecode($table);
$resp[$tname] = array("create" => $this->db->showTableCreate($table));
}
return $resp;
}
public function checkTables($tables, $type) {
$resp = array();
foreach($tables as $table) {
$tname = urldecode($table);
$resp[$tname] = array("status" => $this->db->checkTable($table, $type));
}
return $resp;
}
public function describeTables($tables) {
$resp = array();
foreach($tables as $table) {
$tname = urldecode($table);
$resp[$tname] = array("description" => $this->db->describeTable($table));
}
return $resp;
}
public function checkTablesExist($tables) {
$resp = array();
foreach($tables as $table) {
$tname = urldecode($table);
$resp[$tname] = array("tblexists" => $this->db->isTablePresent($table));
}
return $resp;
}
public function getTablesRowCount($tables) {
$resp = array();
foreach($tables as $table) {
$tname = urldecode($table);
$resp[$tname] = array("count" => $this->db->rowsCount($table));
}
return $resp;
}
public function getTablesKeys($tables) {
$resp = array();
foreach($tables as $table) {
$tname = urldecode($table);
$resp[$tname] = array("keys" => $this->db->tableKeys($table));
}
return $resp;
}
public function process($request) {
$db = $this->db;
$params = $request->params;
$stream_init_info = BVStream::startStream($this->account, $request);
if (array_key_exists('stream', $stream_init_info)) {
$this->stream = $stream_init_info['stream'];
switch ($request->method) {
case "gettbls":
$resp = array("tables" => $db->showTables());
break;
case "tblstatus":
$resp = array("statuses" => $db->showTableStatus());
break;
case "tablekeys":
$table = urldecode($params['table']);
$resp = array("table_keys" => $db->tableKeys($table));
break;
case "describetable":
$table = urldecode($params['table']);
$resp = array("table_description" => $db->describeTable($table));
break;
case "checktable":
$table = urldecode($params['table']);
$type = urldecode($params['type']);
$resp = array("status" => $db->checkTable($table, $type));
break;
case "repairtable":
$table = urldecode($params['table']);
$resp = array("status" => $db->repairTable($table));
break;
case "gettcrt":
$table = urldecode($params['table']);
$resp = array("create" => $db->showTableCreate($table));
break;
case "tblskys":
$tables = $params['tables'];
$resp = $this->getTablesKeys($tables);
break;
case "getmlticrt":
$tables = $params['tables'];
$resp = $this->getCreateTableQueries($tables);
break;
case "desctbls":
$tables = $params['tables'];
$resp = $this->describeTables($tables);
break;
case "mltirwscount":
$tables = $params['tables'];
$resp = $this->getTablesRowCount($tables);
break;
case "chktabls":
$tables = $params['tables'];
$type = urldecode($params['type']);
$resp = $this->checkTables($tables, $type);
break;
case "chktablsxist":
$tables = $params['tables'];
$resp = $this->checkTablesExist($tables);
break;
case "getrowscount":
$table = urldecode($params['table']);
$resp = array("count" => $db->rowsCount($table));
break;
case "gettablecontent":
$result = array();
$table = urldecode($params['table']);
$fields = urldecode($params['fields']);
$filter = (array_key_exists('filter', $params)) ? urldecode($params['filter']) : "";
$limit = intval(urldecode($params['limit']));
$offset = intval(urldecode($params['offset']));
$pkeys = (array_key_exists('pkeys', $params)) ? $params['pkeys'] : array();
$result['timestamp'] = time();
$result['tablename'] = $table;
$rows = $db->getTableContent($table, $fields, $filter, $limit, $offset);
$srows = sizeof($rows);
if (!empty($pkeys) && $srows > 0) {
$end_row = end($rows);
$result['last_ids'] = $this->getLastID($pkeys, $end_row);
}
$result["rows"] = $rows;
$resp = $result;
break;
case "tableinfo":
$table = urldecode($params['table']);
$offset = intval(urldecode($params['offset']));
$limit = intval(urldecode($params['limit']));
$bsize = intval(urldecode($params['bsize']));
$filter = (array_key_exists('filter', $params)) ? urldecode($params['filter']) : "";
$rcount = intval(urldecode($params['rcount']));
$tname = urldecode($params['tname']);
$pkeys = (array_key_exists('pkeys', $params)) ? $params['pkeys'] : array();
$resp = $this->getTableData($table, $tname, $rcount, $offset, $limit, $bsize, $filter, $pkeys, false);
break;
case "getmulttables":
$result = array();
$tableParams = $params['table_params'];
$resp = array();
foreach($tableParams as $tableParam) {
$table = urldecode($tableParam['table']);
$tname = urldecode($tableParam['tname']);
$rcount = intval(urldecode($tableParam['rcount']));
$filter = (array_key_exists('filter', $tableParam)) ? urldecode($tableParam['filter']) : "";
$limit = intval(urldecode($tableParam['limit']));
$offset = intval(urldecode($tableParam['offset']));
$bsize = intval(urldecode($tableParam['bsize']));
$pkeys = (array_key_exists('pkeys', $tableParam)) ? $tableParam['pkeys'] : array();
$resp[$tname] = $this->getTableData($table, $tname, $rcount, $offset, $limit, $bsize, $filter, $pkeys, true);
}
break;
case "uploadrows":
$table = urldecode($params['table']);
$offset = intval(urldecode($params['offset']));
$limit = intval(urldecode($params['limit']));
$bsize = intval(urldecode($params['bsize']));
$filter = (array_key_exists('filter', $params)) ? urldecode($params['filter']) : "";
$rcount = intval(urldecode($params['rcount']));
$tname = urldecode($params['tname']);
$pkeys = (array_key_exists('pkeys', $params)) ? $params['pkeys'] : array();
$resp = $this->getTableData($table, $tname, $rcount, $offset, $limit, $bsize, $filter, $pkeys, true);
break;
case "tblexists":
$resp = array("tblexists" => $db->isTablePresent($params['table']));
break;
case "crttbl":
$usedbdelta = array_key_exists('usedbdelta', $params);
$resp = array("crttbl" => $db->createTable($params['query'], $params['table'], $usedbdelta));
break;
case "drptbl":
$resp = array("drptbl" => $db->dropBVTable($params['table']));
break;
case "trttbl":
$resp = array("trttbl" => $db->truncateBVTable($params['table']));
break;
case "altrtbl":
$resp = array("altrtbl" => $db->alterBVTable($params['query'], $params['query']));
break;
default:
$resp = false;
}
$end_stream_info = $this->stream->endStream();
if (!empty($end_stream_info) && is_array($resp)) {
$resp = array_merge($resp, $end_stream_info);
}
} else {
$resp = $stream_init_info;
}
return $resp;
}
}
endif;