provision.context.inc 3.66 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13
<?php

/**
 * @file Provision named context base classes.
 */

/**
 * Store and access context objects by alias name.
 *
 * @param $name
 *   A Drush alias name, including leading @.
 * @param $_root_object
 *   Internal use only, set default object returned by d().
14 15 16
 * @param $allow_creation
 *   Defaults to TRUE. Allows creating a new context object with the specified
 *   $name.
17 18
 *
 * @return
ergonlogic's avatar
ergonlogic committed
19
 *   provision_Context object (by reference) or NULL if it can't be loaded and
20
 *   $allow_creation == FALSE.
21
 */
22
function & d($name = NULL, $_root_object = FALSE, $allow_creation = TRUE) {
23 24 25
  static $instances = null;
  static $default_instance = '@self';

26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44
  /*
   * add safety check so that we don't get caught running d() before drush is
   * fully initialized. otherwise, the proper services hooks may not be
   * properly declared and found by drush, so the object cache ($instances)
   * would be initialised with incomplete data.
   *
   * it is possible that more initialisation we need happens after
   * DRUSH_BOOTSTRAP_NONE, but this is already better than nothing.
   *
   * if something is weird with d(), it may be that it is being called too
   * early and that check is failing to detect that, so watch out, it took two
   * senior Aegir developpers three full days of headbanging on keyboards to
   * figure that stuff out.
   */
  if (drush_get_context('DRUSH_BOOTSTRAP_PHASE') == DRUSH_BOOTSTRAP_NONE) {
    $ret = drush_set_error('DRUSH_BOOTSTRAPPING', 'drush is still bootstrapping, d() should be ran only within a hook or a function');
    return $ret;
  }

45 46 47 48 49 50 51 52 53
  if (is_object($name)) {
    return $name;
  }

  if ($name == 'all') {
    return $instances;
  }

  if (is_null($name)) {
54
    $name = $default_instance;
55
  }
omega8cc's avatar
omega8cc committed
56

57 58 59
  if ($_root_object) {
    $default_instance = $name;
  }
omega8cc's avatar
omega8cc committed
60

61
  $name = provision_normalise_context_name($name);
62

63 64 65 66 67 68 69 70
  if (isset($instances[$name])) {
    return $instances[$name];
  }
  else {
    $instances[$name] = provision_context_factory($name, $allow_creation);
    if (!is_null($instances[$name])) {
      $instances[$name]->method_invoke('init');
      $instances[$name]->type_invoke('init');
71
    }
72
    return $instances[$name];
73 74 75 76 77 78 79
  }
}

/**
 * Simple access layer for drush_sitealias_get_record.
 *
 * Everytime sitealiases are fetched a lot of processing happens, but if the
omega8cc's avatar
omega8cc committed
80
 * file doesnt exist yet there's a whole lot of unnecesary stuff happening.
81 82 83 84 85
 *
 * We cache the result locally here.
 */
function provision_sitealias_get_record($name) {
  static $cache = array();
omega8cc's avatar
omega8cc committed
86

87
  $name = provision_normalise_context_name($name);
88 89 90 91 92 93 94 95 96

  if (!isset($cache[$name])) {
    $cache[$name] = drush_sitealias_get_record($name);
  }

  return $cache[$name];
}

/**
97 98 99 100 101 102 103 104 105 106
 * Create a new context object.
 *
 * @param $name
 *   The name of the context object to instantiate.
 * @param $allow_creation
 *   Defaults to TRUE. Allows creating a new context object with the specified
 *   $name.
 *
 * @return
 *   An instance of the specified context, or NULL if it could not be loaded.
107
 */
108
function provision_context_factory($name, $allow_creation = TRUE) {
109
  // the default type, can also be 'platform' or 'site'
110 111 112
  $type = 'server';

  $record = provision_sitealias_get_record($name);
113
  if (!$allow_creation && empty($record)) {
114
    drush_set_error('PROVISION_MISSING_CONTEXT', dt('Could not find provision alias named: @name' , array('@name' => $name)));
115 116
    return NULL;
  }
117
  $options = array_merge(drush_get_context('stdin'), drush_get_context('options'), drush_get_context('cli'));
118 119 120 121 122 123 124

  if (isset($record['context_type'])) {
    $type = $record['context_type'];
  }
  elseif (isset($options['context_type'])) {
    $type = $options['context_type'];
  }
125
  $classname = "Provision_Context_{$type}";
126 127 128

  return new $classname($name);
}