Club Drupal

<?php

/**

 * Implementation of hook_drush_help().

 */

function views_bulk_operations_drush_help($section) {

  switch ($section) {

    case 'drush:vbo-list':

      return dt('List all Views Bulk Operations (VBO) views, along with the operations associated with each.');

    case 'drush:vbo-execute':

      return dt('Execute a bulk operation based on a Views Bulk Operations (VBO) view.');

  }

}

/**

 * Implementation of hook_drush_command().

 */

function views_bulk_operations_drush_command() {

  $items['vbo-list'] = array(

    'callback' => 'views_bulk_operations_drush_list',

    'description' => 'List all Views Bulk Operations (VBO) views, along with the operations associated with each.',

  );

  $items['vbo-execute'] = array(

    'callback' => 'views_bulk_operations_drush_execute',

    'description' => 'Execute a bulk operation based on a Views Bulk Operations (VBO) view.',

    'arguments' => array(

      'vid' => 'ID or name of the view to be executed.',

      'operation_id' => 'ID of the operation to be applied on the view results.',

      'type:[name=]value ...' => 'Parameters to be passed as view input filters, view arguments or operation arguments, where type is respectively {input, argument, operation}.',

    ),

    'examples' => array(

      '$ drush vbo-execute action::admin_content node_publish_action' =>

        'Publish nodes returned by view admin_content.',

      '$ drush vbo-execute 44 action::node_assign_owner_action operation:owner_uid=3' =>

        'Change node ownership on nodes returned by view #44, passing argument owner_uid=3 to the action.',

      '$ drush vbo-execute admin_content action::node_unpublish_action input:type=expense argument:3' =>

        'Unpublish nodes returned by view admin_content, filtering results of type expense and passing value 3 as first view argument.',

    ),

  );

  return $items;

}

/**

 * Implementation of 'vbo list' command.

 */

function views_bulk_operations_drush_list() {

  // Impersonate admin.

  global $user;

  $user = user_load(1);

  drupal_save_session(FALSE);

  // Find all VBO views and their associated operations.

  $rows = array(array(sprintf('%5s', dt('View ID')), dt('Name'), dt('Description'), dt('Operations')));

  foreach (views_get_all_views() as $name => $view) {

    $view->build();

    $vbo = _views_bulk_operations_get_field($view);

    if ($vbo) {

      $operations = array();

      foreach ($vbo->get_selected_operations() as $operation_id => $operation) {

        $operations[] = $operation->label() .' ('. $operation_id .')';

      }

      $operations[] = "---------------";

      $rows[] = array(

        sprintf('%5d', $view->vid),

        $view->name,

        $view->description,

        implode("\n", $operations),

      );

    }

  }

  drush_print_table($rows, TRUE);

}

/**

 * Implementation of 'vbo execute' command.

 */

function views_bulk_operations_drush_execute($vid = NULL, $operation_id = NULL) {

  // Parse arguments.

  if (is_null($vid)) {

    drush_set_error('VIEWS_BULK_OPERATIONS_MISSING_VID', dt('Please specify a view ID to execute.'));

    return;

  }

  if (is_null($operation_id)) {

    drush_set_error('VIEWS_BULK_OPERATIONS_MISSING_OPERATION', dt('Please specify an operation to execute.'));

    return;

  }

  $args = func_get_args();

  $view_exposed_input = array();

  $operation_arguments = array();

  $view_arguments = array();

  if (count($args) > 2) for ($i=2; $i<count($args); $i++) {

    $parts = array();

    if (FALSE === preg_match('/^(?<type>\w+):(?:(?<name>\w+)=)?(?<value>(.*?))$/', $args[$i], $parts)) {

      drush_set_error('VIEWS_BULK_OPERATIONS_INVALID_PARAMETER', dt('The parameter %arg should be of the form type:[name=]value where type in {input, argument, operation}.', array('%arg' => $args[$i])));

      return;

    }

    switch ($parts['type']) {

      case 'input':

        $view_exposed_input[$parts['name']] = $parts['value'];

        break;

      case 'operation':

        $operation_arguments[$parts['name']] = $parts['value'];

        break;

      case 'argument':

        $view_arguments[] = $parts['value'];

        break;

      default:

        drush_set_error('VIEWS_BULK_OPERATIONS_UNKNOWN_PARAMETER', dt('The parameter type %type is unknown. Please specify either input, argument or operation.', array('%type' => $parts['type'])));

        return;

    }

  }

  // Impersonate admin.

  global $user;

  $user = user_load(1);

  drupal_save_session(FALSE);

  // Load the view.

  $view = views_get_view($vid);

  if (!is_object($view)) {

    _views_bulk_operations_report_error('Could not find view %vid.', array('%vid' => $vid));

    return;

  }

  // Build the view, so that the VBO field can be found.

  $view->set_exposed_input($view_exposed_input);

  $view->set_arguments($view_arguments);

  $view->build();

  $view->query->set_limit(NULL); // reset the work done by the pager

  $view->query->set_offset(NULL);

  // Find the VBO field.

  $vbo = _views_bulk_operations_get_field($view);

  if (!$vbo) {

    _views_bulk_operations_report_error('Could not find a VBO field in view %vid.', array('%vid' => $vid));

    return;

  }

  $view->execute();

  // Find the selected operation.

  $operations = $vbo->get_selected_operations();

  if (!isset($operations[$operation_id])) {

    _views_bulk_operations_report_error('Could not find operation %operation_id in view %vid.', array('%operation_id' => $operation_id, '%vid' => $vid));

    return;

  }

  $operation = views_bulk_operations_get_operation($operation_id, $vbo->get_entity_type(), $vbo->get_operation_options($operation_id));

  if ($operation_arguments) {

    $operation->formOptions = $operation_arguments;

  }

  // Select all rows.

  $rows = array();

  $current = 1;

  foreach ($view->result as $row_index => $result) {

    $rows[$row_index] = array(

      'entity_id' => $result->{$vbo->real_field},

      'views_row' => array(),

      'position' => array(

        'current' => $current++,

        'total' => $view->total_rows,

      ),

    );

    // Some operations require full selected rows.

    if ($operation->needsRows()) {

      $rows[$row_index]['views_row'] = $result;

    }

  }

  // Enqueue the fetched rows.

  $queue_name = 'views_bulk_operations_active_queue_' . db_next_id();

  $options = array(

    'revision' => $vbo->revision,

    'entity_load_capacity' => $vbo->get_vbo_option('entity_load_capacity', 10),

  );

  views_bulk_operations_enqueue_rows($queue_name, $rows, $operation, $options);

  // Process the queue using Batch API.

  $batch = array(

    'file' => drupal_get_path('module', 'views_bulk_operations') . '/views_bulk_operations.module',

    'operations' => array(

      array('views_bulk_operations_active_queue_process', array($queue_name, $operation, $vbo->view->total_rows)),

    ),

    'finished' => '_views_bulk_operations_execute_finished',

    'title' => t('Performing %operation on the selected items...', array('%operation' => $operation->label())),

  );

  batch_set($batch);

  drush_backend_batch_process();

  // Looks like drush has no way to show messages set in subprocesses,

  // so all batch messages get lost. Setting a success message manually here.

  drush_log(dt('Performed "!operation" on @items.', array(

      '!operation' => $operation->label(),

      '@items' => format_plural($vbo->view->total_rows, '1 item', '@count items'),

    )), 'ok');

}