views.drush.inc

<?php


/**
 * @file
 * Drush integration of views.
 *
 * * drush cache-clear views - Clears the views specific caches.
 * * views-revert - Drush command to revert views overridden in the system.
 */

/**
 * Implement 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');
    }
}

/**
 * Implement hook_drush_command().
 */
function views_drush_command() {
    $items = array();
    $items['views-revert'] = array(
        'callback' => 'views_drush_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',
        ),
        '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.',
        ),
    );
    $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_drush_revert_views() {
    $views = views_get_all_views();
    $i = 0;
    // The provided views names specified in the command.
    $viewnames = _convert_csv_to_array(func_get_args());
    // Find all overridden views.
    foreach ($views as $view) {
        if ($view->disabled) {
            continue;
        }
        if ($view->type == dt('Overridden')) {
            $overridden[$view->name] = $view->name;
        }
    }
    // Return early if there are no overridden views in the system.
    if (empty($overridden)) {
        return drush_set_error(dt('There are no overridden views in the system.'));
    }
    // If the user specified in the command the views to be overridden.
    if (!empty($viewnames)) {
        foreach ($viewnames as $key => $viewname) {
            $is_overridden = key_exists($viewname, $overridden);
            // Check if the provided view name is in the system
            if ($viewname && !key_exists($viewname, $views)) {
                drush_set_error(dt("'@viewname' view is not present in the system.", array(
                    '@viewname' => $viewname,
                )));
            }
            elseif (!$is_overridden) {
                drush_set_error(dt("The view specified '@viewname' is not overridden.", array(
                    '@viewname' => $viewname,
                )));
            }
            elseif ($is_overridden) {
                views_drush_revert_view($views[$viewname]);
                $i++;
            }
            else {
                drush_set_error(dt("The view specified '@viewname' is not provided in code, and thus cannot be reverted.", array(
                    '@viewname' => $viewname,
                )));
            }
        }
    }
    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_drush_revert_allviews($views);
            }
            else {
                views_drush_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 $views
 * All views in the system as provided by views_get_all_views().
 */
function views_drush_revert_allviews($views) {
    $i = 0;
    foreach ($views as $view) {
        if ($view->disabled) {
            continue;
        }
        if ($view->type == t('Overridden')) {
            views_drush_revert_view($view);
            $i++;
        }
    }
    return $i;
}

/**
 * Reverts a specified view
 * @param $view
 * The view object to be reverted
 *
 * 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...
 */
function views_drush_revert_view($view) {
    // check anyway just in case
    if ($view->type == t('Overridden')) {
        // Revert the view.
        $view->delete();
        // Clear its cache.
        views_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,
        )));
    }
}

/**
 * Callback function for views-list command.
 */
function drush_views_list() {
    // Initialize stuf
    $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,
    )));
}
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 $viewnames: array of viewnames to process
* @param $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');
            menu_rebuild();
            drush_log(dt("Menu cache was cleared"), 'ok');
        }
    }
}

/**
 * Adds a cache clear option for views.
 */
function views_drush_cache_clear(&$types) {
    $types['views'] = 'views_invalidate_cache';
}