xmlsitemap.drush.inc 6.15 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13
<?php

/**
 * @file
 * Drush integration functions for the xmlsitemap module.
 *
 * @ingroup xmlsitemap
 */

/**
 * Implements hook_drush_command().
 */
function xmlsitemap_drush_command() {
14

15
  $items['xmlsitemap-regenerate'] = array(
16
    'description' => 'Regenerate the XML sitemap files.',
17
    'callback' => 'drush_xmlsitemap_regenerate',
18
    'drupal dependencies' => array('xmlsitemap'),
19
  );
20
  $items['xmlsitemap-rebuild'] = array(
21
    'description' => 'Dump and re-process all possible XML sitemap data, and then regenerate the files.',
22
    'callback' => 'drush_xmlsitemap_rebuild',
23
    'drupal dependencies' => array('xmlsitemap'),
24 25 26
    'options' => array(
      'types' => 'The types of links to rebuild, comma separated. If not provided, all link types will be used.',
    ),
27
  );
28 29
  $items['xmlsitemap-index'] = array(
    'description' => 'Process un-indexed XML sitemap links.',
30
    'callback' => 'drush_xmlsitemap_index',
31 32
    'drupal dependencies' => array('xmlsitemap'),
    'options' => array(
33
      'limit' => 'The limit of links of each type to process. Default value: ' . variable_get('xmlsitemap_batch_limit', XMLSITEMAP_BATCH_LIMIT),
34 35
    ),
  );
36 37 38
  $items['xmlsitemap-queue-rebuild'] = array(
    'description' => 'Dump and queues all possible XML sitemap data to be re-processed via the xmlsitemap_link_process queue. This command does not regenerate the sitemap files.',
    'options' => array(
39
      'types' => 'The types of links to queue for rebuild, comma separated. If not provided, all link types will be used.',
40 41 42
      'limit' => 'The number of links to be processed in each queue task.',
    ),
  );
43 44 45 46 47 48 49
  return $items;
}

/**
 * Regenerate the sitemap files from existing data.
 */
function drush_xmlsitemap_regenerate() {
50 51 52
  module_load_include('generate.inc', 'xmlsitemap');

  // Run the batch process.
53
  timer_start('xmlsitemap_regenerate');
54
  xmlsitemap_run_unprogressive_batch('xmlsitemap_regenerate_batch');
55 56 57 58

  $vars = array(
    '@timer' => timer_read('xmlsitemap_regenerate'),
    '@memory-peak' => format_size(memory_get_peak_usage(TRUE)),
59
  );
60 61 62 63 64 65 66
  drush_print(dt('XML sitemap files regenerated in @timer ms. Peak memory usage: @memory-peak.', $vars));
}

/**
 * Dump and rebuild all the sitemap data, then regenerate the files.
 */
function drush_xmlsitemap_rebuild() {
67
  module_load_include('generate.inc', 'xmlsitemap');
68

69
  // Build a list of rebuildable link types.
70
  $types = xmlsitemap_get_rebuildable_link_types();
Dave Reid's avatar
Dave Reid committed
71 72
  if ($option_types = drush_get_option('types', '')) {
    $option_types = explode(',', $option_types);
73 74 75 76 77 78
    if ($invalid_types = array_diff($option_types, $types)) {
      drush_set_error(dt('The following link types are invalid: @types', array('@types' => implode(', ', $invalid_types))));
    }
    $types = array_intersect($types, $option_types);
  }
  if (empty($types)) {
Dave Reid's avatar
Dave Reid committed
79
    return drush_set_error(dt('No link types are rebuildable.'));
80
  }
81 82

  // Run the batch process.
83
  timer_start('xmlsitemap_rebuild');
84
  xmlsitemap_run_unprogressive_batch('xmlsitemap_rebuild_batch', $types, TRUE);
85 86 87 88

  $vars = array(
    '@timer' => timer_read('xmlsitemap_rebuild'),
    '@memory-peak' => format_size(memory_get_peak_usage(TRUE)),
89
  );
90 91
  drush_print(dt('XML sitemap files rebuilt in @timer ms. Peak memory usage: @memory-peak.', $vars));
}
92 93 94 95 96

/**
 * Process un-indexed XML sitemap links.
 */
function drush_xmlsitemap_index() {
97
  $limit = (int) drush_get_option('limit', variable_get('xmlsitemap_batch_limit', XMLSITEMAP_BATCH_LIMIT));
98 99
  $count_before = db_query("SELECT COUNT(id) FROM {xmlsitemap}")->fetchField();

100
  module_invoke_all('xmlsitemap_index_links', $limit);
101 102 103 104 105 106 107 108
  $count_after = db_query("SELECT COUNT(id) FROM {xmlsitemap}")->fetchField();

  if ($count_after == $count_before) {
    drush_print(dt('No new XML sitemap links to index.'));
  }
  else {
    drush_print(dt('Indexed @count new XML sitemap links.', array('@count' => $count_after - $count_before)));
  }
109
}
110 111 112 113 114 115

/**
 * Dump and queue all the sitemap links to be rebuilt in a queue process.
 */
function drush_xmlsitemap_queue_rebuild() {
  module_load_include('generate.inc', 'xmlsitemap');
116 117

  $types = xmlsitemap_get_rebuildable_link_types();
Dave Reid's avatar
Dave Reid committed
118 119
  if ($option_types = drush_get_option('types', '')) {
    $option_types = explode(',', $option_types);
120 121 122 123 124 125
    if ($invalid_types = array_diff($option_types, $types)) {
      drush_set_error(dt('The following link types are invalid: @types', array('@types' => implode(', ', $invalid_types))));
    }
    $types = array_intersect($types, $option_types);
  }
  if (empty($types)) {
126 127 128
    return drush_set_error(dt('No link types are rebuildable.'));
  }

129
  xmlsitemap_rebuild_clear($types, TRUE);
130 131 132

  $link_count = 0;
  $chunk_count = 0;
133
  $chunk_size = (int) drush_get_option('limit', variable_get('xmlsitemap_batch_limit', XMLSITEMAP_BATCH_LIMIT));
134 135

  // @todo Figure out how to re-use this code with xmlsitemap_rebuild_batch_fetch()
136 137
  foreach ($types as $type) {
    $info = xmlsitemap_get_link_info($type);
138
    $query = new EntityFieldQuery();
139
    $query->entityCondition('entity_type', $type);
140 141 142
    $query->entityCondition('entity_id', 0, '>');
    $query->addTag('xmlsitemap_link_bundle_access');
    $query->addTag('xmlsitemap_rebuild');
143
    $query->addMetaData('entity', $type);
144
    $query->addMetaData('entity_info', $info);
145 146
    if ($bundles = xmlsitemap_get_link_type_enabled_bundles($type)) {
      $query->entityCondition('bundle', $bundles, 'IN');
147 148 149 150 151 152 153
    }
    else {
      // If no enabled bundle types, skip everything else.
      continue;
    }

    $results = $query->execute();
154 155
    if (!empty($results[$type])) {
      $ids = array_keys($results[$type]);
156 157 158 159
      $link_count += count($ids);
      $chunks = array_chunk($ids, $chunk_size);
      $chunk_count += count($chunks);
      foreach ($chunks as $chunk) {
160
        xmlsitemap_link_enqueue($type, $chunk);
161 162 163 164 165
      }
    }
  }

  if ($link_count) {
166 167 168 169 170
    drush_log(dt('Queued @link_count links for rebuild processing in the xmlsitemap_link_process (in @chunk_count chunks of up to @chunk_size links each).', array(
      '@link_count' => $link_count,
      '@chunk_count' => $chunk_count,
      '@chunk_size' => $chunk_size,
    )), 'success');
171 172 173 174
  }
  else {
    drush_log(dt('No links to queue for rebuild processing.'), 'ok');
  }
175
  variable_set('xmlsitemap_rebuild_needed', FALSE);
176
}