<?php

/**
 * @file
 * Drush integration for Views.
 *
 * Useful commands:
 * - drush cache-clear views - Clears the views specific caches.
 * - views-revert - Drush command to revert views overridden in the system.
 */

/**
 * Implements hook_drush_help().
 */
function views_drush_help($section) {
  switch ($section) {
    case 'drush:views-revert':
      $help = dt('Reverts views in the drupal installation that have been overriden. ');
      $help .= dt('If no view names are specified, you will be presented with a list of overridden views to choose from. ');
      $help .= dt('To revert all views, do not specify any view names, and choose the option "All" from the options presented.');
      return $help;

    case 'drush:views-list':
      return dt('Show a list of available views with information about them.');

    case 'drush:views-enable':
      return dt('Enable the specified views. Follow the command with a space delimited list of view names');

    case 'drush:views-disable':
      return dt('Disable the specified views. Follow the command with a space delimited list of view names');
  }
}

/**
 * Implements hook_drush_command().
 */
function views_drush_command() {
  $items = array();

  $items['views-revert'] = array(
    'callback' => 'views_revert_views',
    'drupal dependencies' => array('views'),
    'description' => 'Revert overridden views to their default state. Make sure to backup first.',
    'arguments' => array(
      'views' => 'A space delimited list of view names.',
    ),
    'bootstrap' => DRUSH_BOOTSTRAP_DRUPAL_FULL,
    'aliases' => array('vr'),
    'options' => array(
      'all' => 'If provided, all views will be reverted.',
    ),
    'examples' => array(
      'drush vr archive' => 'Reverts the "archive" view.',
      'drush rln archive frontpage' => 'Reverts the "archive" and "frontpage" view.',
      'drush vr' => 'Will present you with a list of overridden views to choose from, and an option to revert all overridden views.',
      'drush vr --all' => 'Will revert all overridden views.',
    ),
  );
  $items['views-dev'] = array(
    'callback' => 'views_development_settings',
    'drupal dependencies' => array('views'),
    'description' => 'Set the Views settings to more developer-oriented values.',
    'bootstrap' => DRUSH_BOOTSTRAP_DRUPAL_FULL,
    'aliases' => array('vd'),
  );

  $items['views-list'] = array(
    'drupal dependencies' => array('views'),
    'description' => 'Get a list of all views in the system.',
    'bootstrap' => DRUSH_BOOTSTRAP_DRUPAL_FULL,
    'aliases' => array('vl'),
    'options' => array(
      'name' => 'String contained in view\'s name by which filter the results.',
      'tags' => 'A comma-separated list of views tags by which to filter the results.',
      'status' => 'Status of the views by which to filter the results. Choices: enabled, disabled.',
      'type' => 'Type of the views by which to filter the results. Choices: normal, default or overridden.',
    ),
    'examples' => array(
      'drush vl' => 'Show a list of all available views.',
      'drush vl --name=blog' => 'Show a list of views which names contain "blog".',
      'drush vl --tags=tag1,tag2' => 'Show a list of views tagged with "tag1" or "tag2".',
      'drush vl --status=enabled' => 'Show a list of enabled views.',
      'drush vl --type=overridden' => 'Show a list of overridden views.',
    ),
  );
  $items['views-analyze'] = array(
    'drupal dependencies' => array('views', 'views_ui'),
    'description' => 'Get a list of all Views analyze warnings',
    'bootstrap' => DRUSH_BOOTSTRAP_DRUPAL_FULL,
    'aliases' => array('va'),
  );
  $items['views-enable'] = array(
    'drupal dependencies' => array('views'),
    'description' => 'Enable the specified views.',
    'arguments' => array(
      'views' => 'A space delimited list of view names.',
    ),
    'aliases' => array('ven'),
    'examples' => array(
      'drush ven frontpage taxonomy_term' => 'Enable the frontpage and taxonomy_term views.',
    ),
  );
  $items['views-disable'] = array(
    'drupal dependencies' => array('views'),
    'description' => 'Disable the specified views.',
    'arguments' => array(
      'views' => 'A space delimited list of view names.',
    ),
    'aliases' => array('vdis'),
    'examples' => array(
      'drush vdis frontpage taxonomy_term' => 'Disable the frontpage and taxonomy_term views.',
    ),
  );

  return $items;
}

/**
 * Callback function for views-revert command.
 */
function views_revert_views() {
  $args = func_get_args();
  // The provided views names specified in the command.
  $viewnames = _convert_csv_to_array($args);
  $views = views_get_all_views();
  $i = 0;

  // Find all overridden views.
  foreach ($views as $view) {
    if ($view->disabled) {
      continue;
    }
    if ($view->type == dt('Overridden')) {
      $overridden[$view->name] = $view->name;
    }
  }

  // If there are no overridden views in the system, report it.
  if (empty($overridden)) {
    drush_log(dt('There are no overridden views in the system.'), 'ok');
  }

  // If the user provided the "--all" option, revert all views.
  if (drush_get_option('all')) {
    $i = views_revert_allviews($views);
  }

  // If the user specified a list of views on the CLI, revert those.
  elseif (!empty($viewnames)) {
    foreach ($viewnames as $key => $viewname) {
      $is_overridden = array_key_exists($viewname, $overridden);

      // Check if the provided view name is in the system.
      if ($viewname && !array_key_exists($viewname, $views)) {
        drush_set_error(dt("'@viewname' view is not present in the system.", array('@viewname' => $viewname)));
      }
      // Check if the provided view is overridden.
      elseif (!$is_overridden) {
        drush_set_error(dt("The view specified '@viewname' is not overridden.", array('@viewname' => $viewname)));
      }
      // If the view is overriden, revert it.
      elseif ($is_overridden) {
        views_revert_view($views[$viewname]);
        $i++;
      }
      // We should never get here but well.
      else {
        drush_set_error(dt(
          "The view specified '@viewname' is not provided in code, and thus cannot be reverted.",
          array('@viewname' => $viewname)
        ));
      }
    }
  }

  // The user neither selected the "--all" option, nor provided a list of views
  // to revert. Prompt the user.
  else {
    // List of choices for the user.
    $overridden['all'] = dt('Revert all overridden views');
    // Add a choice at the end.
    $choice = drush_choice($overridden, 'Enter a number to choose which view to revert.', '!key');
    // Prompt the user.
    if ($choice !== FALSE) {
      // Revert all views option.
      if ($choice == 'all') {
        $i = views_revert_allviews($views);
      }
      // Else the user specified a single view.
      else {
        views_revert_view($views[$choice]);
        $i++;
      }
    }

  }

  // Final results output.
  if ($i == 0) {
    drush_log(dt('No views were reverted.'), 'ok');
  }
  else {
    drush_log(dt('Reverted a total of @count views.', array('@count' => $i)), 'ok');
  }
}

/**
 * Reverts all views.
 *
 * @param array $views
 *   All views in the system as provided by views_get_all_views().
 */
function views_revert_allviews($views) {
  $i = 0;
  foreach ($views as $view) {
    if ($view->disabled) {
      continue;
    }

    if ($view->type == t('Overridden')) {
      views_revert_view($view);
      $i++;
    }
  }
  return $i;
}

/**
 * Revert a specified view.
 *
 * Checks on wether or not the view is overridden is handled in
 * views_revert_views_revert(). We perform a check here anyway in case someone
 * somehow calls this function on their own...
 *
 * @param object $view
 *   The view object to be reverted.
 */
function views_revert_view($view) {
  // Check anyway just in case.
  if ($view->type == t('Overridden')) {
    // Revert the view.
    $view->delete();
    // Clear its cache.
    ctools_include('object-cache');
    ctools_object_cache_clear('view', $view->name);
    // Give feedback.
    $message = dt("Reverted the view '@viewname'", array('@viewname' => $view->name));
    drush_log($message, 'success');
    // Reverted one more view.
  }
  else {
    drush_set_error(dt("The view '@viewname' is not overridden.", array('@viewname' => $view->name)));
  }
}

/**
 * Change the settings to a more developer oriented value.
 */
function views_development_settings() {
  variable_set('views_ui_show_listing_filters', TRUE);
  variable_set('views_ui_show_master_display', TRUE);
  variable_set('views_ui_show_advanced_column', TRUE);
  variable_set('views_ui_always_live_preview', FALSE);
  variable_set('views_ui_always_live_preview_button', TRUE);
  variable_set('views_ui_show_preview_information', TRUE);
  variable_set('views_ui_show_sql_query', TRUE);
  variable_set('views_ui_show_performance_statistics', TRUE);
  variable_set('views_show_additional_queries', TRUE);
  variable_set('views_devel_output', TRUE);
  variable_set('views_devel_region', 'message');
  variable_set('views_ui_display_embed', TRUE);
  $message = dt("Setup the new views settings.");
  drush_log($message, 'success');
}


/**
 * Callback function for views-list command.
 */
function drush_views_list() {
  // Initialize stuff.
  $rows = array();
  $disabled_views = array();
  $enabled_views = array();
  $overridden = 0;
  $indb = 0;
  $incode = 0;
  $disabled = 0;
  $total = 0;

  $views = views_get_all_views();

  // Get the --name option.
  // @todo Take into account the case off a comma-separated list of names.
  $name = drush_get_option_list('name');
  $with_name = !empty($name) ? TRUE : FALSE;

  // Get the --tags option.
  $tags = drush_get_option_list('tags');
  $with_tags = !empty($tags) ? TRUE : FALSE;

  // Get the --status option. Store user input appart to reuse it after.
  $status_opt = drush_get_option_list('status');
  // Use the same logic than $view->disabled.
  if (in_array('disabled', $status_opt)) {
    $status = TRUE;
    $with_status = TRUE;
  }
  elseif (in_array('enabled', $status_opt)) {
    $status = FALSE;
    $with_status = TRUE;
  }
  else {
    $status = NULL;
    // Wrong or empty --status option.
    $with_status = FALSE;
  }

  // Get the --type option.
  $type = drush_get_option_list('type');
  // Use the same logic than $view->type.
  $with_type = FALSE;
  if (in_array('normal', $type) || in_array('default', $type)|| in_array('overridden', $type)) {
    $with_type = TRUE;
  }

  // Set the table headers.
  $header = array(
    dt('Machine name'),
    dt('Description'),
    dt('Type'),
    dt('Status'),
    dt('Tag'),
  );

  // Setup a row for each view.
  foreach ($views as $id => $view) {
    // If options were specified, check that first mismatch push the loop to
    // the next view.
    if ($with_tags && !in_array($view->tag, $tags)) {
      continue;
    }
    if ($with_status && !$view->disabled == $status) {
      continue;
    }
    if ($with_type && strtolower($view->type) !== $type[0]) {
      continue;
    }
    if ($with_name && !stristr($view->name, $name[0])) {
      continue;
    }

    $row = array();
    // Each row entry should be in the same order as the header.
    $row[] = $view->name;
    $row[] = $view->description;
    $row[] = $view->type;
    $row[] = $view->disabled ? dt('Disabled') : dt('Enabled');
    $row[] = $view->tag;

    // Place the row in the appropiate array so we can have disabled views at
    // the bottom.
    if ($view->disabled) {
      $disabled_views[] = $row;
    }
    else {
      $enabled_views[] = $row;
    }
    unset($row);

    // Gather some statistics.
    switch ($view->type) {
      case dt('Normal'):
        $indb++;
        break;

      case dt('Overridden'):
        $overridden++;
        break;

      case dt('Default'):
        $incode++;
        break;
    }
    $total++;
  }

  $disabled = count($disabled_views);

  // Sort alphabeticaly.
  asort($disabled_views);
  asort($enabled_views);

  // If options were used.
  $summary = "";
  if ($with_name || $with_tags || $with_status || $with_type) {
    $summary = "Views";

    if ($with_name) {
      $summary .= " named $name[0]";
    }

    if ($with_tags) {
      $tags = implode(" or ", $tags);
      $summary .= " tagged $tags";
    }

    if ($with_status) {
      $status_opt = implode("", $status_opt);
      $summary .= " which status is '$status_opt'";
    }

    if ($with_type) {
      $type = ucfirst($type[0]);
      $summary .= " of type '$type'";
    }
  }

  if (!empty($summary)) {
    drush_print($summary . "\n");
  }

  // Print all rows as a table.
  if ($total > 0) {
    $rows = array_merge($enabled_views, $disabled_views);
    // Put the headers as first row.
    array_unshift($rows, $header);

    drush_print_table($rows, TRUE);
  }

  // Print the statistics messages.
  drush_print(dt("A total of @total views were found in this Drupal installation:", array('@total' => $total)));
  drush_print(dt("  @indb views reside only in the database", array('@indb' => $indb)));
  drush_print(dt("  @over views are overridden", array('@over' => $overridden)));
  drush_print(dt("  @incode views are in their default state", array('@incode' => $incode)));
  drush_print(dt("  @dis views are disabled\n", array('@dis' => $disabled)));
}

/**
 * Analyze all installed views.
 */
function drush_views_analyze() {
  views_include('analyze');
  $messages_count = 0;
  $total = 0;

  foreach (views_get_all_views() as $view_name => $view) {
    $total++;
    if ($messages = views_analyze_view($view)) {
      drush_print($view_name);
      foreach ($messages as $message) {
        $messages_count++;
        drush_print($message['type'] . ': ' . $message['message'], 2);
      }
    }
  }
  drush_log(dt('A total of @total views were analyzed and @messages problems were found.', array('@total' => $total, '@messages' => $messages_count)), 'ok');
}

/**
 * Enables views.
 */
function drush_views_enable() {
  $viewnames = _convert_csv_to_array(func_get_args());
  // Return early if no view names were specified.
  if (empty($viewnames)) {
    return drush_set_error(dt('Please specify a space delimited list of view names to enable'));
  }
  _views_drush_changestatus($viewnames, FALSE);
}

/**
 * Disables views.
 */
function drush_views_disable() {
  $viewnames = _convert_csv_to_array(func_get_args());
  // Return early if no view names were specified.
  if (empty($viewnames)) {
    return drush_set_error(dt('Please specify a space delimited list of view names to disable'));
  }
  _views_drush_changestatus($viewnames, TRUE);
}

/**
 * Helper function to enable / disable views.
 *
 * @param array $viewnames
 *   Names of the views to process.
 * @param bool $status
 *   TRUE to disable or FALSE to enable the view.
 */
function _views_drush_changestatus($viewnames = array(), $status = NULL) {
  if ($status !== NULL && !empty($viewnames)) {
    $changed = FALSE;
    $processed = $status ? dt('disabled') : dt('enabled');
    $views_status = variable_get('views_defaults', array());

    foreach ($viewnames as $key => $viewname) {
      if ($views_status[$viewname] !== $status) {
        $views_status[$viewname] = $status;
        $changed = TRUE;
        drush_log(dt("The view '!name' has been !processed", array('!name' => $viewname, '!processed' => $processed)), 'success');
      }
      else {
        drush_set_error(dt("The view '!name' is already !processed", array('!name' => $viewname, '!processed' => $processed)));
      }
    }
    // If we made changes to views status, save them and clear caches.
    if ($changed) {
      variable_set('views_defaults', $views_status);
      views_invalidate_cache();
      drush_log(dt("Views cache was cleared"), 'ok');
      drush_log(dt("Menu cache is set to be rebuilt on the next request."), 'ok');
    }
  }
}

/**
 * Adds a cache clear option for views.
 *
 * @param array $types
 *   The list of cache types that are available.
 */
function views_drush_cache_clear(&$types) {
  $types['views'] = 'views_invalidate_cache';
}
