'smf_db_backup_table', 'db_optimize_table' => 'smf_db_optimize_table', 'db_insert_sql' => 'smf_db_insert_sql', 'db_table_sql' => 'smf_db_table_sql', 'db_list_tables' => 'smf_db_list_tables', 'db_get_backup' => 'smf_db_get_backup', 'db_get_version' => 'smf_db_get_version', ); } // Backup $table to $backup_table. function smf_db_backup_table($table, $backup_table) { global $smcFunc, $db_prefix; $table = str_replace('{db_prefix}', $db_prefix, $table); $result = $smcFunc['db_query']('', ' SELECT sql FROM sqlite_master WHERE type = {string:txttable} AND name = {string:table}', array( 'table' => $table, 'txttable' => 'table' ) ); list ($create) = $smcFunc['db_fetch_row']($result); $smcFunc['db_free_result']($result); $create = preg_split('/[\n\r]/', $create); $auto_inc = ''; // Remove the first line and check to see if the second one contain useless info. unset($create[0]); if (trim($create[1]) == '(') unset($create[1]); if (trim($create[count($create)]) == ')') unset($create[count($create)]); foreach ($create as $k => $l) { // Get the name of the auto_increment column. if (strpos($l, 'primary') || strpos($l, 'PRIMARY')) $auto_inc = trim($l); // Skip everything but keys... if ((strpos($l, 'KEY') !== false && strpos($l, 'PRIMARY KEY') === false) || strpos($l, $table) !== false || strpos(trim($l), 'PRIMARY KEY') === 0) unset($create[$k]); } if (!empty($create)) $create = '( ' . implode(' ', $create) . ')'; else $create = ''; // Is there an extra junk at the end? if (substr($create, -2, 1) == ',') $create = substr($create, 0, -2) . ')'; if (substr($create, -2) == '))') $create = substr($create, 0, -1); $smcFunc['db_query']('', ' DROP TABLE {raw:backup_table}', array( 'backup_table' => $backup_table, 'db_error_skip' => true, ) ); $request = $smcFunc['db_quote'](' CREATE TABLE {raw:backup_table} {raw:create}', array( 'backup_table' => $backup_table, 'create' => $create, )); $smcFunc['db_query']('', ' CREATE TABLE {raw:backup_table} {raw:create}', array( 'backup_table' => $backup_table, 'create' => $create, )); $request = $smcFunc['db_query']('', ' INSERT INTO {raw:backup_table} SELECT * FROM {raw:table}', array( 'backup_table' => $backup_table, 'table' => $table, )); return $request; } // Optimize a table - return data freed! function smf_db_optimize_table($table) { global $smcFunc, $db_prefix; $table = str_replace('{db_prefix}', $db_prefix, $table); $request = $smcFunc['db_query']('', ' VACUUM {raw:table}', array( 'table' => $table, ) ); if (!$request) return -1; $row = $smcFunc['db_fetch_assoc']($request); $smcFunc['db_free_result']($request); // The function returns nothing. return 0; } // List all the tables in the database. function smf_db_list_tables($db = false, $filter = false) { global $smcFunc; $filter = $filter == false ? '' : ' AND name LIKE \'' . str_replace("\_", "_", $filter) . '\''; $request = $smcFunc['db_query']('', ' SELECT name FROM sqlite_master WHERE type = {string:type} {raw:filter} ORDER BY name', array( 'type' => 'table', 'filter' => $filter, ) ); $tables = array(); while ($row = $smcFunc['db_fetch_row']($request)) $tables[] = $row[0]; $smcFunc['db_free_result']($request); return $tables; } // Get the content (INSERTs) for a table. function smf_db_insert_sql($tableName) { global $smcFunc, $db_prefix; $tableName = str_replace('{db_prefix}', $db_prefix, $tableName); // This will be handy... $crlf = "\r\n"; // Get everything from the table. $result = $smcFunc['db_query']('', ' SELECT * FROM {raw:table}', array( 'table' => $tableName, ) ); // The number of rows, just for record keeping and breaking INSERTs up. $num_rows = $smcFunc['db_num_rows']($result); if ($num_rows == 0) return ''; $fields = array_keys($smcFunc['db_fetch_assoc']($result)); // SQLite fetches an array so we need to filter out the numberic index for the columns. foreach ($fields as $key => $name) if (is_numeric($name)) unset($fields[$key]); $smcFunc['db_data_seek']($result, 0); // Start it off with the basic INSERT INTO. $data = 'BEGIN TRANSACTION;' . $crlf; // Loop through each row. while ($row = $smcFunc['db_fetch_row']($result)) { // Get the fields in this row... $field_list = array(); for ($j = 0; $j < $smcFunc['db_num_fields']($result); $j++) { // Try to figure out the type of each field. (NULL, number, or 'string'.) if (!isset($row[$j])) $field_list[] = 'NULL'; elseif (is_numeric($row[$j]) && (int) $row[$j] == $row[$j]) $field_list[] = $row[$j]; else $field_list[] = '\'' . $smcFunc['db_escape_string']($row[$j]) . '\''; } $data .= 'INSERT INTO ' . $tableName . ' (' . implode(', ', $fields) . ') VALUES (' . implode(', ', $field_list) . ');' . $crlf; } $smcFunc['db_free_result']($result); // Return an empty string if there were no rows. return $num_rows == 0 ? '' : $data . 'COMMIT;' . $crlf; } // Get the schema (CREATE) for a table. function smf_db_table_sql($tableName) { global $smcFunc, $db_prefix; $tableName = str_replace('{db_prefix}', $db_prefix, $tableName); // This will be needed... $crlf = "\r\n"; // Start the create table... $schema_create = ''; $index_create = ''; // Let's get the create statement directly from SQLite. $result = $smcFunc['db_query']('', ' SELECT sql FROM sqlite_master WHERE type = {string:type} AND name = {string:table_name}', array( 'type' => 'table', 'table_name' => $tableName, ) ); list ($schema_create) = $smcFunc['db_fetch_row']($result); $smcFunc['db_free_result']($result); // Now the indexes. $result = $smcFunc['db_query']('', ' SELECT sql FROM sqlite_master WHERE type = {string:type} AND tbl_name = {string:table_name}', array( 'type' => 'index', 'table_name' => $tableName, ) ); $indexes = array(); while ($row = $smcFunc['db_fetch_assoc']($result)) if (trim($row['sql']) != '') $indexes[] = $row['sql']; $smcFunc['db_free_result']($result); $index_create .= implode(';' . $crlf, $indexes); $schema_create = empty($indexes) ? rtrim($schema_create) : $schema_create . ';' . $crlf . $crlf; return $schema_create . $index_create; } // Get the version number. function smf_db_get_version() { return sqlite_libversion(); } // Simple return the database - and die! function smf_db_get_backup() { global $db_name; $db_file = substr($db_name, -3) === '.db' ? $db_name : $db_name . '.db'; // Add more info if zipped... $ext = ''; if (isset($_REQUEST['compress']) && function_exists('gzencode')) $ext = '.gz'; // Do the remaining headers. header('Content-Disposition: attachment; filename="' . $db_file . $ext . '"'); header('Cache-Control: private'); header('Connection: close'); // Literally dump the contents. Try reading the file first. if (@readfile($db_file) == null) echo file_get_contents($db_file); obExit(false); } ?>