OpenSQLManager/sys/db/drivers/sqlite.php

234 lines
4.8 KiB
PHP
Raw Normal View History

2012-01-26 16:09:05 -05:00
<?php
/**
* OpenSQLManager
*
* Free Database manager for Open Source Databases
*
* @author Timothy J. Warren
* @copyright Copyright (c) 2012
* @link https://github.com/aviat4ion/OpenSQLManager
* @license http://philsturgeon.co.uk/code/dbad-license
*/
2012-01-30 07:57:17 -05:00
// --------------------------------------------------------------------------
/**
* SQLite specific class
*
* @extends DB_PDO
*/
class SQLite extends DB_PDO {
2012-02-23 08:59:27 -05:00
protected $statement;
/**
* Open SQLite Database
*
* @param string $dsn
*/
public function __construct($dsn, $user=NULL, $pass=NULL)
2012-01-30 07:57:17 -05:00
{
// DSN is simply `sqlite:/path/to/db`
parent::__construct("sqlite:{$dsn}", $user, $pass);
2012-02-08 09:01:51 -05:00
$class = __CLASS__."_sql";
$this->sql = new $class;
2012-01-30 07:57:17 -05:00
}
// --------------------------------------------------------------------------
2012-01-26 16:09:05 -05:00
/**
* Empty a table
*
* @param string $table
*/
2012-02-21 11:45:42 -05:00
public function truncate($table)
{
// SQLite has a TRUNCATE optimization,
// but no support for the actual command.
2012-02-23 08:59:27 -05:00
$sql = 'DELETE FROM :table';
2012-02-13 13:29:31 -05:00
2012-02-23 08:59:27 -05:00
$this->prepare_query($sql, array(
':table' => $table
));
2012-02-23 20:10:13 -05:00
$this->statement->execute();
}
// --------------------------------------------------------------------------
/**
2012-02-02 19:07:26 -05:00
* List tables for the current database
*
* @return mixed
*/
2012-02-21 11:45:42 -05:00
public function get_tables()
{
2012-02-08 09:01:51 -05:00
$tables = array();
2012-02-13 13:29:31 -05:00
$sql = <<<SQL
2012-02-23 08:59:27 -05:00
SELECT "name", "sql"
FROM "sqlite_master"
WHERE "type"='table'
SQL;
2012-02-13 13:29:31 -05:00
$res = $this->query($sql);
2012-02-08 09:01:51 -05:00
$result = $res->fetchAll(PDO::FETCH_ASSOC);
foreach($result as $r)
{
$tables[$r['name']] = $r['sql'];
}
return $tables;
2012-02-06 16:56:16 -05:00
}
// --------------------------------------------------------------------------
2012-02-06 16:56:16 -05:00
/**
* List system tables for the current database
*
* @return array
*/
2012-02-21 11:45:42 -05:00
public function get_system_tables()
{
2012-02-13 13:46:53 -05:00
//SQLite only has the sqlite_master table
// that is of any importance.
return array('sqlite_master');
}
// --------------------------------------------------------------------------
/**
* Load a database for the current connection
*
* @param string $db
* @param string $name
*/
2012-02-21 11:45:42 -05:00
public function load_database($db, $name)
{
2012-02-28 22:07:13 -05:00
$sql = 'ATTACH DATABASE "'.$db.'" AS "'.$name.'"';
$this->query($sql);
}
// --------------------------------------------------------------------------
/**
* Unload a database from the current connection
*
* @param string $name
*/
2012-02-21 11:45:42 -05:00
public function unload_database($name)
{
2012-02-23 08:59:27 -05:00
$sql = 'DETACH DATABASE ":name"';
$this->prepare_query($sql, array(
':name' => $name,
));
2012-02-23 20:10:13 -05:00
$this->statement->execute();
}
// --------------------------------------------------------------------------
2012-02-06 16:56:16 -05:00
/**
* Return the number of rows returned for a SELECT query
*
* @return int
*/
2012-02-21 11:45:42 -05:00
public function num_rows()
2012-02-06 16:56:16 -05:00
{
2012-02-23 08:59:27 -05:00
return (isset($this->statement)) ? $this->statement->rowCount : FALSE;
}
// --------------------------------------------------------------------------
/**
* Create an SQL backup file for the current database's structure
*
* @return string
*/
public function backup_structure()
{
// Fairly easy for SQLite...just query the master table
$sql = 'SELECT "sql" FROM "sqlite_master"';
$res = $this->query($sql);
$result = $res->fetchAll(PDO::FETCH_ASSOC);
$sql_array = array();
foreach($result as $r)
{
$sql_array[] = $r['sql'];
}
$sql_structure = implode("\n\n", $sql_array);
return $sql_structure;
}
// --------------------------------------------------------------------------
/**
* Create an SQL backup file for the current database's data
*
* @param array $excluded
* @return string
*/
public function backup_data($excluded=array())
{
// Get a list of all the objects
$sql = 'SELECT "name" FROM "sqlite_master"';
if( ! empty($excluded))
{
$sql .= ' WHERE NOT IN("'.implode('","', $excluded).'")';
}
$res = $this->query($sql);
$result = $res->fetchAll(PDO::FETCH_ASSOC);
unset($res);
$output_sql = '';
// Get the data for each object
foreach($result as $r)
{
$sql = 'SELECT * FROM "'.$r['name'].'"';
$res = $this->query($sql);
$obj_res = $res->fetchAll(PDO::FETCH_ASSOC);
unset($res);
// Nab the column names by getting the keys of the first row
$columns = array_keys($obj_res[0]);
$insert_rows = array();
// Create the insert statements
foreach($obj_res as $row)
{
$row = array_values($row);
// Quote values as needed by type
for($i=0, $icount=count($row); $i<$icount; $i++)
{
$row[$i] = (is_numeric($row[$i])) ? $row[$i] : $this->quote($row[$i]);
}
$row_string = 'INSERT INTO "'.$r['name'].'" ("'.implode('","', $columns).'") VALUES ('.implode(',', $row).');';
unset($row);
$insert_rows[] = $row_string;
}
unset($obj_res);
$output_sql .= "\n\n".implode("\n", $insert_rows);
}
return $output_sql;
}
}
//End of sqlite.php