array('id' => $batch['id']))); $js_setting = array( 'batch' => array( 'errorMessage' => $current_set['error_message'] .'
'. $batch['error_message'], 'initMessage' => $current_set['init_message'], 'uri' => $url, ), ); drupal_add_js($js_setting, 'setting'); drupal_add_js('misc/batch.js', 'core', 'header', FALSE, TRUE); $output = '
'; return $output; } /** * Do one pass of execution and inform back the browser about progression. */ function _batch_do() { // HTTP POST required if ($_SERVER['REQUEST_METHOD'] != 'POST') { drupal_set_message(t('HTTP POST is required.'), 'error'); drupal_set_title(t('Error')); return ''; } list($percentage, $message) = _batch_process(); drupal_set_header('Content-Type: text/plain; charset=utf-8'); print drupal_to_js(array('status' => TRUE, 'percentage' => $percentage, 'message' => $message)); exit(); } /** * Batch processing page without JavaScript support. */ function _batch_progress_page_nojs() { $batch =& batch_get(); $current_set = _batch_current_set(); drupal_set_title($current_set['title']); $new_op = 'do_nojs'; if (!isset($batch['running'])) { // This is the first page so return some output immediately. $percentage = 0; $message = $current_set['init_message']; $batch['running'] = TRUE; } else { // This is one of the later requests: do some processing first. // Error handling: if PHP dies due to a fatal error (e.g. non-existant // function), it will output whatever is in the output buffer, // followed by the error message. ob_start(); $fallback = $current_set['error_message'] .'
'. $batch['error_message']; $fallback = theme('maintenance_page', $fallback, FALSE); // We strip the end of the page using a marker in the template, so any // additional HTML output by PHP shows up inside the page rather than // below it. While this causes invalid HTML, the same would be true if // we didn't, as content is not allowed to appear after anyway. list($fallback) = explode('', $fallback); print $fallback; list($percentage, $message) = _batch_process($batch); if ($percentage == 100) { $new_op = 'finished'; } // Processing successful; remove fallback. ob_end_clean(); } $url = url($batch['url'], array('query' => array('id' => $batch['id'], 'op' => $new_op))); drupal_set_html_head(''); $output = theme('progress_bar', $percentage, $message); return $output; } /** * Advance batch processing for 1 second (or process the whole batch if it * was not set for progressive execution). */ function _batch_process() { $batch =& batch_get(); $current_set =& _batch_current_set(); while (!$current_set['success']) { $finished = 1; $task_message = ''; if ((list($function, $args) = reset($current_set['operations'])) && function_exists($function)) { // Build the 'batch context' array, execute the function call, and retrieve the user message. $batch_context = array('sandbox' => &$current_set['sandbox'], 'results' => &$current_set['results'], 'finished' => &$finished, 'message' => &$task_message); call_user_func_array($function, array_merge($args, array(&$batch_context))); } if ($finished == 1) { // Make sure this step isn't counted double. $finished = 0; // Remove the operation, and clear the sandbox to reduce the stored data. array_shift($current_set['operations']); $current_set['sandbox'] = array(); } // Make sure we display progress information about a batch set that // actually has operations, and not about a 'control' set (form submit // handler). $remaining = count($current_set['operations']); $progress_message = $current_set['progress_message']; $total = $current_set['total']; // If the batch set is completed, browse through the remaining sets // until we find one that acually has operations. while (empty($current_set['operations']) && ($current_set['success'] = TRUE) && _batch_next_set()) { $current_set =& _batch_current_set(); } // Progressive mode : stop after 1 second if ($batch['progressive'] && timer_read('page') > 1000) { break; } } if ($batch['progressive']) { $current = $total - $remaining + $finished; $percentage = $total ? floor($current / $total * 100) : 100; $values = array( '@remaining' => $remaining, '@total' => $total, '@current' => floor($current), '@percentage' => $percentage, ); $message = strtr($progress_message, $values) .'
'; $message.= $task_message ? $task_message : ' '; return array($percentage, $message); } else { return _batch_finished(); } } /** * Retrieve the batch set being currently processed. */ function &_batch_current_set() { $batch =& batch_get(); return $batch['sets'][$batch['current_set']]; } /** * Move execution to the next batch set if any, executing the stored * form _submit handlers along the way (thus possibly inserting * additional batch sets) */ function _batch_next_set() { $batch =& batch_get(); if (isset($batch['sets'][$batch['current_set'] + 1])) { $batch['current_set']++; $current_set =& _batch_current_set(); if (isset($current_set['form_submit']) && ($function = $current_set['form_submit']) && function_exists($function)) { // We use our stored copies of $form and $form_state, to account for // possible alteration by the submit handlers. $function($batch['form_state']['values'], $batch['form'], $batch['form_state']); } return TRUE; } } /** * End the batch : * Call the 'finished' callbacks to allow custom handling of results, * and resolve page redirection. */ function _batch_finished() { $batch =& batch_get(); // Execute the 'finished' callbacks. foreach($batch['sets'] as $key => $batch_set) { if (isset($batch_set['finished']) && function_exists($batch_set['finished'])) { $batch_set['finished']($batch_set['success'], $batch_set['results'], $batch_set['operations']); } } // Cleanup the batch table and unset the global $batch variable. db_query("DELETE FROM {batch} WHERE bid = %d", $batch['id']); $_batch = $batch; $batch = NULL; // Redirect if needed. if ($_batch['progressive']) { if (isset($_batch['destination'])) { $_REQUEST['destination'] = $_batch['destination']; } // Use $_batch['form_state']['redirect'], or $_batch['redirect'], or $_batch['source_page']. if (isset($_batch['form_state']['redirect'])) { $redirect = $_batch['form_state']['redirect']; } elseif (isset($_batch['redirect'])) { $redirect = $_batch['redirect']; } else { $redirect = $_batch['source_page']; } // Let drupal_redirect_form handle redirection logic. $form = isset($batch['form']) ? $batch['form'] : array(); if (empty($_batch['form_state']['rebuild']) && empty($_batch['form_state']['storage'])) { drupal_redirect_form($form, $redirect); } // We get here if $form['#redirect'] was FALSE, or if // Save the final $form_state value, Redirect to the originating page. $_SESSION['batch_form_state'] = $_batch['form_state']; drupal_goto($_batch['source_page']); } } /** * Store tha batch data for next request, or clear the table if the batch is finished. */ function _batch_shutdown() { if ($batch = batch_get()) { db_query("UPDATE {batch} SET batch = '%s' WHERE bid = %d", serialize($batch), $batch['id']); } }