<?php

/**
 * Sanitize the database after an sql-sync.
 *
 * We will transform all user email addresses to point at a test email address
 * at a test domain.
 */
function drush_sanitize_post_sql_sync($source = null, $destination = null) {
  $destination_settings = drush_sitealias_get_record($destination);
  
  $sanitize = drush_get_option(array('sanitize', 'destination-sanitize'), FALSE);
  $testemailaddress = drush_get_option(array('test-email', 'destination-test-email'), 'testaddr');
  $testdomain = drush_get_option(array('test-domain', 'destination-test-domain'), 'testdomain.org');
  
  if ($sanitize && $testdomain) {
    // We need a different sanitization query for Postgres and Mysql
    $db_driver = $destination_settings['databases']['default']['default']['driver'];
    if ($db_driver == 'pgsql') {
      $sanitize_query = "update users set mail = '" . $testemailaddress  . "+' || replace(mail, '@', '_') || '@" . $testdomain . "' where uid <> 0 and strpos(mail, '@" . $testdomain . "') = 0;";
    }
    else {
      $sanitize_query = "update users set mail = concat('" . $testemailaddress  . "+', replace(mail, '@', '_'), '@" . $testdomain . "') where uid <> 0 and instr(mail, '@" . $testdomain . "') = 0;";
    }
    
    drush_print("Sanitization sql:");
    drush_print($sanitize_query);
    if (!drush_get_context('DRUSH_SIMULATE')) {
      if (drush_confirm(dt('Would you like to sanitize the database of the destination site %dest?', array('%dest' => $destination)))) {
        $result = drush_do_site_command($destination_settings, "sql-query", array($sanitize_query) /*, array('s' => drush_get_context('DRUSH_SIMULATE')) */);
        drush_print($result['output']);
      }
    }
  }
}