+++ /dev/null
-<?php
-
-namespace Drush\Sql;
-
-use Drupal\Core\Database\Database;
-use Drush\Log\LogLevel;
-use Webmozart\PathUtil\Path;
-
-class SqlBase {
-
- // An Drupal style array containing specs for connecting to database.
- public $db_spec;
-
- // Default code appended to sql-query connections.
- public $query_extra = '';
-
- // The way you pass a sql file when issueing a query.
- public $query_file = '<';
-
- /**
- * Typically, SqlBase objects are contructed via drush_sql_get_class().
- */
- public function __construct($db_spec = NULL) {
- $this->db_spec = $db_spec;
- }
-
- /*
- * Get the current $db_spec.
- */
- public function db_spec() {
- return $this->db_spec;
- }
-
- /**
- * The unix command used to connect to the database.
- * @return string
- */
- public function command() {}
-
- /**
- * A string for connecting to a database.
- *
- * @param bool $hide_password
- * If TRUE, DBMS should try to hide password from process list.
- * On mysql, that means using --defaults-extra-file to supply the user+password.
- *
- * @return string
- */
- public function connect($hide_password = TRUE) {
- return trim($this->command() . ' ' . $this->creds($hide_password) . ' ' . drush_get_option('extra', $this->query_extra));
- }
-
-
- /*
- * Execute a SQL dump and return the path to the resulting dump file.
- *
- * @param string|bool @file
- * The path where the dump file should be stored. If TRUE, generate a path
- * based on usual backup directory and current date.
- */
- public function dump($file = '') {
- $file_suffix = '';
- $table_selection = $this->get_expanded_table_selection();
- $file = $this->dumpFile($file);
- $cmd = $this->dumpCmd($table_selection);
- // Gzip the output from dump command(s) if requested.
- if (drush_get_option('gzip')) {
- $cmd .= ' | gzip -f';
- $file_suffix .= '.gz';
- }
- if ($file) {
- $file .= $file_suffix;
- $cmd .= ' > ' . drush_escapeshellarg($file);
- }
-
- // Avoid the php memory of the $output array in drush_shell_exec().
- if (!$return = drush_op_system($cmd)) {
- if ($file) {
- drush_log(dt('Database dump saved to !path', array('!path' => $file)), LogLevel::SUCCESS);
- drush_backend_set_result($file);
- }
- }
- else {
- return drush_set_error('DRUSH_SQL_DUMP_FAIL', 'Database dump failed');
- }
- }
-
- /*
- * Build bash for dumping a database.
- *
- * @param array $table_selection
- * Supported keys: 'skip', 'structure', 'tables'.
- * @return string
- * One or more mysqldump/pg_dump/sqlite3/etc statements that are ready for executing.
- * If multiple statements are needed, enclose in parenthesis.
- */
- public function dumpCmd($table_selection) {}
-
- /*
- * Generate a path to an output file for a SQL dump when needed.
- *
- * @param string|bool @file
- * If TRUE, generate a path based on usual backup directory and current date.
- * Otherwise, just return the path that was provided.
- */
- public function dumpFile($file) {
- $database = $this->db_spec['database'];
-
- // $file is passed in to us usually via --result-file. If the user
- // has set $options['result-file'] = TRUE, then we
- // will generate an SQL dump file in the same backup
- // directory that pm-updatecode uses.
- if ($file) {
- if ($file === TRUE) {
- // User did not pass a specific value for --result-file. Make one.
- $backup = drush_include_engine('version_control', 'backup');
- $backup_dir = $backup->prepare_backup_dir($database);
- if (empty($backup_dir)) {
- $backup_dir = drush_find_tmp();
- }
- $file = Path::join($backup_dir, '@DATABASE_@DATE.sql');
- }
- $file = str_replace(array('@DATABASE', '@DATE'), array($database, gmdate('Ymd_His')), $file);
- }
- return $file;
- }
-
- /**
- * Execute a SQL query.
- *
- * Note: This is an API function. Try to avoid using drush_get_option() and instead
- * pass params in. If you don't want to query results to print during --debug then
- * provide a $result_file whose value can be drush_bit_bucket().
- *
- * @param string $query
- * The SQL to be executed. Should be NULL if $input_file is provided.
- * @param string $input_file
- * A path to a file containing the SQL to be executed.
- * @param string $result_file
- * A path to save query results to. Can be drush_bit_bucket() if desired.
- *
- * @return
- * TRUE on success, FALSE on failure
- */
- public function query($query, $input_file = NULL, $result_file = '') {
- $input_file_original = $input_file;
- if ($input_file && drush_file_is_tarball($input_file)) {
- if (drush_shell_exec('gzip -d %s', $input_file)) {
- $input_file = trim($input_file, '.gz');
- }
- else {
- return drush_set_error(dt('Failed to decompress input file.'));
- }
- }
-
- // Save $query to a tmp file if needed. We will redirect it in.
- if (!$input_file) {
- $query = $this->query_prefix($query);
- $query = $this->query_format($query);
- $input_file = drush_save_data_to_temp_file($query);
- }
-
- $parts = array(
- $this->command(),
- $this->creds(),
- $this->silent(), // This removes column header and various helpful things in mysql.
- drush_get_option('extra', $this->query_extra),
- $this->query_file,
- drush_escapeshellarg($input_file),
- );
- $exec = implode(' ', $parts);
-
- if ($result_file) {
- $exec .= ' > '. drush_escapeshellarg($result_file);
- }
-
- // In --verbose mode, drush_shell_exec() will show the call to mysql/psql/sqlite,
- // but the sql query itself is stored in a temp file and not displayed.
- // We show the query when --debug is used and this function created the temp file.
- if ((drush_get_context('DRUSH_DEBUG') || drush_get_context('DRUSH_SIMULATE')) && empty($input_file_original)) {
- drush_log('sql-query: ' . $query, LogLevel::NOTICE);
- }
-
- $success = drush_shell_exec($exec);
-
- if ($success && drush_get_option('file-delete')) {
- drush_op('drush_delete_dir', $input_file);
- }
-
- return $success;
- }
-
- /*
- * A string to add to the command when queries should not print their results.
- */
- public function silent() {}
-
-
- public function query_prefix($query) {
- // Inject table prefixes as needed.
- if (drush_has_boostrapped(DRUSH_BOOTSTRAP_DRUPAL_DATABASE)) {
- // Enable prefix processing which can be dangerous so off by default. See http://drupal.org/node/1219850.
- if (drush_get_option('db-prefix')) {
- if (drush_drupal_major_version() >= 7) {
- $query = Database::getConnection()->prefixTables($query);
- }
- else {
- $query = db_prefix_tables($query);
- }
- }
- }
- return $query;
- }
-
-
- public function query_format($query) {
- return $query;
- }
-
- /**
- * Drop specified database.
- *
- * @param array $tables
- * An array of table names
- * @return boolean
- * True if successful, FALSE if failed.
- */
- public function drop($tables) {
- $return = TRUE;
- if ($tables) {
- $sql = 'DROP TABLE '. implode(', ', $tables);
- $return = $this->query($sql);
- }
- return $return;
- }
-
- /**
- * Build a SQL string for dropping and creating a database.
- *
- * @param string dbname
- * The database name.
- * @param boolean $quoted
- * Quote the database name. Mysql uses backticks to quote which can cause problems
- * in a Windows shell. Set TRUE if the CREATE is not running on the bash command line.
- */
- public function createdb_sql($dbname, $quoted = FALSE) {}
-
- /**
- * Create a new database.
- *
- * @param boolean $quoted
- * Quote the database name. Mysql uses backticks to quote which can cause problems
- * in a Windows shell. Set TRUE if the CREATE is not running on the bash command line.
- * @return boolean
- * True if successful, FALSE otherwise.
- */
- public function createdb($quoted = FALSE) {
- $dbname = $this->db_spec['database'];
- $sql = $this->createdb_sql($dbname, $quoted);
- // Adjust connection to allow for superuser creds if provided.
- $this->su();
- return $this->query($sql);
- }
-
- /**
- * Drop all tables (if DB exists) or CREATE target database.
- *
- * return boolean
- * TRUE or FALSE depending on success.
- */
- public function drop_or_create() {
- if ($this->db_exists()) {
- return $this->drop($this->listTables());
- }
- else {
- return $this->createdb();
- }
- }
-
- /*
- * Determine if the specified DB already exists.
- *
- * @return bool
- */
- public function db_exists() {}
-
- public function delete() {}
-
- /**
- * Build a fragment connection parameters.
- *
- * @param bool $hide_password
- * If TRUE, DBMS should try to hide password from process list.
- * On mysql, that means using --defaults-extra-file to supply the user+password.
- * @return string
- */
- public function creds($hide_password = TRUE) {}
-
- /**
- * The active database driver.
- * @return string
- */
- public function scheme() {
- return $this->db_spec['driver'];
- }
-
- /**
- * Get a list of all table names and expand input that may contain
- * wildcards (`*`) if necessary so that the array returned only contains valid
- * table names i.e. actual tables that exist, without a wildcard.
- *
- * @return array
- * An array of tables with each table name in the appropriate
- * element of the array.
- */
- public function get_expanded_table_selection() {
- $table_selection = drush_sql_get_table_selection();
- // Get the existing table names in the specified database.
- $db_tables = $this->listTables();
- if (isset($table_selection['skip'])) {
- $table_selection['skip'] = _drush_sql_expand_and_filter_tables($table_selection['skip'], $db_tables);
- }
- if (isset($table_selection['structure'])) {
- $table_selection['structure'] = _drush_sql_expand_and_filter_tables($table_selection['structure'], $db_tables);
- }
- if (isset($table_selection['tables'])) {
- $table_selection['tables'] = _drush_sql_expand_and_filter_tables($table_selection['tables'], $db_tables);
- }
- return $table_selection;
- }
-
- /**
- * Extract the name of all existing tables in the given database.
- *
- * @return array
- * An array of table names which exist in the current database.
- */
- public function listTables() {}
-
- /*
- * Helper method to turn associative array into options with values.
- *
- * @return string
- * A bash fragment.
- */
- public function params_to_options($parameters) {
- // Turn each parameter into a valid parameter string.
- $parameter_strings = array();
- foreach ($parameters as $key => $value) {
- // Only escape the values, not the keys or the rest of the string.
- $value = drush_escapeshellarg($value);
- $parameter_strings[] = "--$key=$value";
- }
-
- // Join the parameters and return.
- return implode(' ', $parameter_strings);
- }
-
- /**
- * Adjust DB connection with superuser credentials if provided.
- *
- * The options 'db-su' and 'db-su-pw' will be retreived from the
- * specified site alias record, if it exists and contains those items.
- * If it does not, they will be fetched via drush_get_option.
- *
- * Note that in the context of sql-sync, the site alias record will
- * be taken from the target alias (e.g. `drush sql-sync @source @target`),
- * which will be overlayed with any options that begin with 'target-';
- * therefore, the commandline options 'target-db-su' and 'target-db-su-pw'
- * may also affect the operation of this function.
- *
- * @return null
- */
- public function su() {
- $create_db_target = $this->db_spec;
-
- $create_db_target['database'] = '';
- $db_superuser = drush_get_option('db-su');
- if (isset($db_superuser)) {
- $create_db_target['username'] = $db_superuser;
- }
- $db_su_pw = drush_get_option('db-su-pw');
- // If --db-su-pw is not provided and --db-su is, default to empty password.
- // This way db cli command will take password from .my.cnf or .pgpass.
- if (!empty($db_su_pw)) {
- $create_db_target['password'] = $db_su_pw;
- }
- elseif (isset($db_superuser)) {
- unset($create_db_target['password']);
- }
- $this->db_spec = $create_db_target;
- }
-}