'Generate queue assignments', 'description' => 'Bulk add nodes into queues', 'page callback' => 'drupal_get_form', 'page arguments' => array('nodequeue_generate_form'), 'access callback' => 'user_access', 'access arguments' => array('manipulate all queues'), 'type' => MENU_LOCAL_TASK, ); return $items; } /** * Menu callback; Returns the nodequeue generate form. */ function nodequeue_generate_form() { $form['help'] = array( '#markup' => '

' . t('Select which queues shall be emptied and re-populated with new nodes.') . '

' ); $queues = nodequeue_load_queues(nodequeue_get_all_qids(25)); // Tableselect header. $header = array( 'name' => 'Queue name', 'max_nodes' => 'Max nodes', 'subqueues' => 'Subqueues', ); // Tableselect data. $data = array(); foreach ($queues as $queue) { $data[$queue->qid]['name'] = check_plain($queue->title); $data[$queue->qid]['max_nodes'] = $queue->size == 0 ? t('Infinite') : $queue->size; $data[$queue->qid]['subqueues'] = $queue->subqueues; } // Table select element. $form['nodequeues'] = array( '#type' => 'tableselect', '#header' => $header, '#options' => $data, '#empty' => t('There are no queues.'), ); $form['nodequeue_generate_nodes_limit'] = array( '#type' => 'textfield', '#title' => t('Nodes limit'), '#description' => t('How many nodes to insert in a queue. This value is only taken into consideration for infinite queues.'), '#size' => 3, '#default_value' => 10, ); $form['submit'] = array( '#type' => 'submit', '#value' => t('Generate'), ); return $form; } /** * Validate nodequeue generate form submission. * * Make sure that at least one queue was selected. */ function nodequeue_generate_form_validate($form, &$form_state) { $qids = array_keys(array_filter($form_state['values']['nodequeues'])); if (count($qids) < 1) { form_set_error('nodequeues', t('You must select a Queue.')); } } function nodequeue_generate_form_submit($form, &$form_state) { // Get qids of all nodequeues that need to be re-populated and repopulate them. $qids = array_keys(array_filter($form_state['values']['nodequeues'])); $nodes_limit = $form_state['values']['nodequeue_generate_nodes_limit']; nodequeue_generate_rehash(); nodequeue_generate_repopulate_queues($qids, $nodes_limit); } /** * Re-populates nodequeues with nodes. * * @param $qids Array of queues, that need to be repopulated. */ function nodequeue_generate_repopulate_queues($qids, $nodes_limit = 10) { // Remove existing nodes from queues. db_query("DELETE FROM {nodequeue_nodes} WHERE qid IN (:qids)", array(':qids' => $qids)); // Load all queues and their subqueues. $queues = nodequeue_load_queues($qids); $subqueues = nodequeue_load_subqueues_by_queue($qids); // Re-populate subqueues foreach ($qids as $qid) { $queue = nodequeue_load($qid); // Skip nodequeues that do not belong to any node types. if (!empty($queue->types)) { $limit = $queue->size ? $queue->size : $nodes_limit; $callback = $queue->owner . '_nodequeue_generate'; if (function_exists($callback)) { $callback($queue, $limit); } } } drupal_set_message(format_plural(count($qids), '1 queue populated', '@count queues populated.')); } /** * Rebuild all smartqueue_taxonomy queues. Useful after a data migration has wiped your terms. * When more smartqueue modules arrive, revisit this function. * * @param vids * An array of vocabulary ids. */ function nodequeue_generate_rehash() { // Delete existing smartqueue taxonomy subqueues db_query("DELETE ns FROM {nodequeue_subqueue} ns INNER JOIN {nodequeue_queue} nq ON ns.qid = nq.qid WHERE nq.owner = 'smartqueue_taxonomy'"); // Get all queues, owned by Smartqueue taxonomy. $qids = db_select('nodequeue_queue', 'nq') ->fields('nq', array('qid')) ->condition('owner', 'smartqueue_taxonomy') ->execute() ->fetchAll(); foreach ($qids as $qid) { $queue = nodequeue_load($qid->qid); $fields = explode('-', $queue->reference); $tids = array(); foreach ($fields as $field) { // Get all possible tids from this field. $query = db_select('field_data_' . $field, 'f'); $query ->condition('f.entity_type', 'node') ->condition('f.bundle', $queue->types, 'IN') ->condition('f.deleted', FALSE) ->addField('f', $field . '_tid', 'tid'); $query = $query->distinct(); $query = $query->execute(); $tids += $query->fetchAll(); } // Rehash for each tid. $nids = array(); foreach ($tids as $tid) { foreach ($fields as $field) { $query = db_select('field_data_' . $field, 'f') ->condition('f.entity_type', 'node') ->condition('f.bundle', $queue->types, 'IN') ->condition('f.deleted', FALSE) ->condition('f.' . $field . '_tid', $tid->tid) ->fields('f', array('entity_id')) ->range(0, 1) ->execute(); $nids += $query->fetchAll(); } foreach ($nids as $nid) { $node = node_load($nid->entity_id); nodequeue_api_subqueues($queue, $node); } } } } /** * Implements hook_nodequeue_generate() for owner 'nodequeue'. */ function nodequeue_nodequeue_generate($queue, $limit) { $subqueues = nodequeue_load_subqueues_by_queue($queue->qid); foreach ($subqueues as $subqueue) { $nodes = db_select('node', 'n') ->condition('n.status', NODE_PUBLISHED) ->condition('n.type', $queue->types, 'IN') ->orderRandom() ->fields('n', array('nid')) ->range(0, $limit) ->execute() ->fetchAll(); foreach ($nodes as $node) { nodequeue_subqueue_add($queue, $subqueue, $node->nid); } } } /** * Implements hook_nodequeue_generate() for owner 'smartqueue_taxonomy'. */ function smartqueue_taxonomy_nodequeue_generate($queue, $limit) { $subqueues = nodequeue_load_subqueues_by_queue($queue->qid); foreach ($subqueues as $subqueue) { $nodes = db_select('taxonomy_index', 'tn'); $nodes->join('node', 'n', 'n.nid=tn.nid'); $nodes->fields('n', array('nid')); $nodes->condition('n.status', NODE_PUBLISHED); $nodes->condition('n.type', $queue->types, 'IN'); $nodes->condition('tn.tid', $subqueue->reference); $nodes->orderRandom(); $nodes->range(0, $limit); $nodes = $nodes->execute(); $nodes = $nodes->fetchAll(); foreach ($nodes as $node) { nodequeue_subqueue_add($queue, $subqueue, $node->nid); } } }