GdprSqlMysql.php in General Data Protection Regulation 8

Namespace

Drupal\gdpr_dump\Sql

File

modules/gdpr_dump/src/Sql/GdprSqlMysql.php

<?php

namespace Drupal\gdpr_dump\Sql;

use Drupal\gdpr_dump\Form\SettingsForm;
use Drupal\gdpr_dump\Service\GdprSqlDump;
use Drush\Log\LogLevel;
use Drush\Sql\Sqlmysql;

/**
 * Class GdprSqlMysql.
 *
 * @package Drupal\gdpr_dump\Sql
 */
class GdprSqlMysql extends Sqlmysql {

  /**
   * The config for gdpr dump.
   *
   * @var \Drupal\Core\Config\ImmutableConfig
   */
  protected $gdprDumpConfig;

  /**
   * An array of tables to skip.
   *
   * @var array
   */
  protected $tablesToSkip = [];

  /**
   * An array of tables to be anonymized.
   *
   * @var array
   */
  protected $tablesToAnonymize = [];

  /**
   * {@inheritdoc}
   */
  public function __construct($db_spec = NULL) {
    parent::__construct($db_spec);
    $this->gdprDumpConfig = \Drupal::config(SettingsForm::GDPR_DUMP_CONF_KEY);
    $this->tablesToAnonymize = $this->gdprDumpConfig
      ->get('mapping');
    $this->tablesToSkip = \array_keys($this->gdprDumpConfig
      ->get('empty_tables'));
  }

  /**
   * 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.
   *
   * @return mixed
   *   Bool or nothing.
   */
  public function dump($file = '') {
    $file_suffix = '';
    $table_selection = $this
      ->get_expanded_table_selection();
    $file = $this
      ->dumpFile($file);

    // @todo: Cross-platform check.
    $cmd = '{ ';
    $cmd .= $this
      ->dumpCmd($table_selection);

    // Append the RENAME commands at the end.
    $cmd .= ' ; ' . $this
      ->createRenameCommands($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 (drush_op_system($cmd)) {
      return drush_set_error('DRUSH_SQL_DUMP_FAIL', 'Database dump failed');
    }
    if ($file) {
      drush_log(dt('Database dump saved to !path', [
        '!path' => $file,
      ]), LogLevel::SUCCESS);
      drush_backend_set_result($file);
    }
  }

  /**
   * Create table renames according to the GDPR config.
   *
   * @param array $tableSelection
   *   Supported keys: 'skip', 'structure', 'tables'.
   *
   * @return string
   *   The command.
   */
  protected function createRenameCommands(array $tableSelection) {
    $skipTables = \array_merge($tableSelection['skip'], $tableSelection['structure']);
    $skipTables = \array_flip($skipTables);
    $skipTables += $this->tablesToSkip;
    $command = '';
    foreach (\array_keys($this->tablesToAnonymize) as $table) {
      if (\array_key_exists($table, $skipTables)) {

        // Don't try to rename a table if it is excluded.
        continue;
      }
      $clone = GdprSqlDump::GDPR_TABLE_PREFIX . $table;
      $rename = "RENAME TABLE \\`{$clone}\\` TO \\`{$table}\\`;";
      if (drush_get_context('DRUSH_VERBOSE') || drush_get_context('DRUSH_SIMULATE')) {
        drush_print("Adding rename command: '{$rename}'", 0, STDERR);
      }
      $command .= " ( echo \"{$rename}\" ); ";
    }
    return $command;
  }

  /**
   * Build bash for dumping a database.
   *
   * @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($tableSelection) {
    $multipleCommands = FALSE;
    $skipTables = $tableSelection['skip'];
    $structureTables = $tableSelection['structure'];
    $structureTables = \array_merge($this->tablesToSkip, $structureTables);
    $tables = $tableSelection['tables'];
    $ignores = [];
    $skipTables = \array_merge($structureTables, $skipTables);

    // Skip tables with sensitive data.
    $skipTables = \array_merge(\array_keys($this->tablesToAnonymize), $skipTables);
    $dataOnly = drush_get_option('data-only');

    // The ordered-dump option is only supported by MySQL for now.
    // @todo add documentation once a hook for drush_get_option_help() is available.
    // @see drush_get_option_help() in drush.inc
    $orderedDump = drush_get_option('ordered-dump');
    $exec = 'mysqldump ';

    // Mysqldump wants 'databasename' instead of
    // 'database=databasename' for no good reason.
    $onlyDbName = \str_replace('--database=', ' ', $this
      ->creds());
    $exec .= $onlyDbName;

    // We had --skip-add-locks here for a while to help people with
    // insufficient permissions, but removed it because it slows down the
    // import a lot.  See http://drupal.org/node/1283978
    $extra = ' --no-autocommit --single-transaction --opt -Q';
    if (NULL !== $dataOnly) {
      $extra .= ' --no-create-info';
    }
    if (NULL !== $orderedDump) {
      $extra .= ' --skip-extended-insert --order-by-primary';
    }
    if ($option = drush_get_option('extra', $this->query_extra)) {
      $extra .= " {$option}";
    }
    $exec .= $extra;
    if (!empty($tables)) {
      $exec .= ' ' . \implode(' ', $tables);
    }
    else {

      // @todo: Maybe use --ignore-table={db.table1,db.table2,...} syntax.
      // Append the ignore-table options.
      foreach ($skipTables as $table) {
        $ignores[] = '--ignore-table=' . $this->db_spec['database'] . '.' . $table;
        $multipleCommands = TRUE;
      }
      $exec .= ' ' . \implode(' ', $ignores);

      // Run mysqldump again and append output
      // if we need some structure only tables.
      if (!empty($structureTables)) {
        $exec .= ' && mysqldump ' . $onlyDbName . " --no-data {$extra} " . \implode(' ', $structureTables);
        $multipleCommands = TRUE;
      }
    }
    return $multipleCommands ? "({$exec})" : $exec;
  }

}

Classes

Name	Description
GdprSqlMysql	Class GdprSqlMysql.