provision/db_server/restore.provision.inc

<?php

function drush_provision_mysql_provision_restore_validate() {
  provision_db_connect();
}

function drush_provision_mysql_pre_provision_restore($url = NULL) {
  $db_type   = drush_set_option('db_type', drush_get_option('db_type'));
  $db_host   = drush_set_option('db_host', drush_get_option('db_host'));
  $db_passwd = drush_set_option('db_passwd', provision_password());
  $db_name   = drush_set_option('db_name', _provision_mysql_suggest_db_name($url));
  $db_user   = drush_set_option('db_user', $db_name);

  _provision_mysql_new_site_db($db_name, $db_user, $db_passwd);
}

function drush_provision_mysql_provision_restore($url) {
  _provision_mysql_import_dump(
    drush_get_option('sites_path') .'/'. $url .'/database.sql',
    drush_get_option('db_name'), drush_get_option('db_user'),
    drush_get_option('db_passwd'), drush_get_option('db_host')
  );
}


function drush_provision_mysql_pre_provision_restore_rollback($url = NULL) {
  _provision_mysql_destroy_site_db(drush_get_option('db_name'), drush_get_option('db_user'), drush_get_option('db_passwd'));

  $keys = array('db_name', 'db_passwd', 'db_user', 'db_host');

  //Restore the original database credentials of the site.
  // They were never truly lost, but the options in the 'process' context
  // were overriding them. By unsetting the temporary options, the originals
  // are automatically restored.
  foreach ($keys as $key) {
    drush_unset_option($key, 'process');
  }
}

// Rollback doesn't apply here yet. Unless we trigger a restore of the first dump
// made. Which could go on infinitely if something is really long.
function drush_provision_mysql_post_provision_restore($url = NULL) {
  provision_path('unlink', drush_get_option('sites_path') . '/' . $url .'/database.sql', TRUE,
    dt("Removed dump file @path after restoring from it"),
    dt("Could not remove dump file @path"), 'DRUSH_PERM_ERROR');

  $db_grant_host = _provision_mysql_grant_host(
      drush_get_option('db_host', '', 'site'), 
      drush_get_option('web_ip', null, 'site'), 
      drush_get_option('web_host', null, 'site'));
  // We have now completed successfully, remove the old database.
  _provision_mysql_destroy_site_db(
    drush_get_option('db_name', null, 'site'), 
    drush_get_option('db_user', null, 'site'), 
    drush_get_option('db_passwd', null, 'site'), 
    $db_grant_host);

  // The new database credentials will be saved against the site now.
  drush_set_option('db_name', drush_get_option('db_name'), 'site');
  drush_set_option('db_type', drush_get_option('db_type'), 'site');
  drush_set_option('db_user', drush_get_option('db_user'), 'site');
  drush_set_option('db_host', drush_get_option('db_host'), 'site');
  drush_set_option('db_passwd', drush_get_option('db_passwd'), 'site');
}

1	<?php
2
3	function drush_provision_mysql_provision_restore_validate() {
4	provision_db_connect();
5	}
6
7	function drush_provision_mysql_pre_provision_restore($url = NULL) {
8	$db_type = drush_set_option('db_type', drush_get_option('db_type'));
9	$db_host = drush_set_option('db_host', drush_get_option('db_host'));
10	$db_passwd = drush_set_option('db_passwd', provision_password());
11	$db_name = drush_set_option('db_name', _provision_mysql_suggest_db_name($url));
12	$db_user = drush_set_option('db_user', $db_name);
13
14	_provision_mysql_new_site_db($db_name, $db_user, $db_passwd);
15	}
16
17	function drush_provision_mysql_provision_restore($url) {
18	_provision_mysql_import_dump(
19	drush_get_option('sites_path') .'/'. $url .'/database.sql',
20	drush_get_option('db_name'), drush_get_option('db_user'),
21	drush_get_option('db_passwd'), drush_get_option('db_host')
22	);
23	}
24
25
26	function drush_provision_mysql_pre_provision_restore_rollback($url = NULL) {
27	_provision_mysql_destroy_site_db(drush_get_option('db_name'), drush_get_option('db_user'), drush_get_option('db_passwd'));
28
29	$keys = array('db_name', 'db_passwd', 'db_user', 'db_host');
30
31	//Restore the original database credentials of the site.
32	// They were never truly lost, but the options in the 'process' context
33	// were overriding them. By unsetting the temporary options, the originals
34	// are automatically restored.
35	foreach ($keys as $key) {
36	drush_unset_option($key, 'process');
37	}
38	}
39
40	// Rollback doesn't apply here yet. Unless we trigger a restore of the first dump
41	// made. Which could go on infinitely if something is really long.
42	function drush_provision_mysql_post_provision_restore($url = NULL) {
43	provision_path('unlink', drush_get_option('sites_path') . '/' . $url .'/database.sql', TRUE,
44	dt("Removed dump file @path after restoring from it"),
45	dt("Could not remove dump file @path"), 'DRUSH_PERM_ERROR');
46
47	$db_grant_host = _provision_mysql_grant_host(
48	drush_get_option('db_host', '', 'site'),
49	drush_get_option('web_ip', null, 'site'),
50	drush_get_option('web_host', null, 'site'));
51	// We have now completed successfully, remove the old database.
52	_provision_mysql_destroy_site_db(
53	drush_get_option('db_name', null, 'site'),
54	drush_get_option('db_user', null, 'site'),
55	drush_get_option('db_passwd', null, 'site'),
56	$db_grant_host);
57
58	// The new database credentials will be saved against the site now.
59	drush_set_option('db_name', drush_get_option('db_name'), 'site');
60	drush_set_option('db_type', drush_get_option('db_type'), 'site');
61	drush_set_option('db_user', drush_get_option('db_user'), 'site');
62	drush_set_option('db_host', drush_get_option('db_host'), 'site');
63	drush_set_option('db_passwd', drush_get_option('db_passwd'), 'site');
64	}
65