<?php

/**

 * @file

 * Helper module for the Batch API tests.

 */

/**

 * Implement hook_menu().

 */

function batch_test_menu() {

  $items = array();

  $items['batch-test'] = array(

    'title' => 'Batch test',

    'page callback' => 'drupal_get_form',

    'page arguments' => array('batch_test_simple_form'),

    'access callback' => TRUE,

  );

  // Simple form: one submit handler, setting a batch.

  $items['batch-test/simple'] = array(

    'title' => 'Simple',

    'type' => MENU_DEFAULT_LOCAL_TASK,

    'weight' => 0,

  );

  // Multistep form: two steps, each setting a batch.

  $items['batch-test/multistep'] = array(

    'title' => 'Multistep',

    'page callback' => 'drupal_get_form',

    'page arguments' => array('batch_test_multistep_form'),

    'access callback' => TRUE,

    'type' => MENU_LOCAL_TASK,

    'weight' => 1,

  );

  // Chained form: four submit handlers, several of which set a batch.

  $items['batch-test/chained'] = array(

    'title' => 'Chained',

    'page callback' => 'drupal_get_form',

    'page arguments' => array('batch_test_chained_form'),

    'access callback' => TRUE,

    'type' => MENU_LOCAL_TASK,

    'weight' => 2,

  );

  // Programmatic form: the page submits the 'Chained' form through

  // drupal_form_submit().

  $items['batch-test/programmatic'] = array(

    'title' => 'Programmatic',

    'page callback' => 'batch_test_programmatic',

    'access callback' => TRUE,

    'type' => MENU_LOCAL_TASK,

    'weight' => 3,

  );

  // No form: fire a batch simply by accessing a page.

  $items['batch-test/no-form'] = array(

    'title' => 'Simple page',

    'page callback' => 'batch_test_no_form',

    'access callback' => TRUE,

    'type' => MENU_LOCAL_TASK,

    'weight' => 4,

  );

  // No form: fire a batch; return > 100% complete

  $items['batch-test/large-percentage'] = array(

    'title' => 'Simple page with batch over 100% complete',

    'page callback' => 'batch_test_large_percentage',

    'access callback' => TRUE,

    'type' => MENU_LOCAL_TASK,

    'weight' => 5,

  );

  // Tests programmatic form submission within a batch operation.

  $items['batch-test/nested-programmatic'] = array(

    'title' => 'Nested programmatic',

    'page callback' => 'batch_test_nested_drupal_form_submit',

    'access callback' => TRUE,

    'type' => MENU_LOCAL_TASK,

    'weight' => 6,

  );

  // Landing page to test redirects.

  $items['batch-test/redirect'] = array(

    'title' => 'Redirect',

    'page callback' => 'batch_test_redirect_page',

    'access callback' => TRUE,

    'type' => MENU_LOCAL_TASK,

    'weight' => 7,

  );

  // This item lives under 'admin' so that the page uses the admin theme.

  $items['admin/batch-test/test-theme'] = array(

    'page callback' => 'batch_test_theme_batch',

    'access callback' => TRUE,

    'type' => MENU_CALLBACK,

  );

  return $items;

}

/**

 * Simple form.

 */

function batch_test_simple_form() {

  $form['batch'] = array(

    '#type' => 'select',

    '#title' => 'Choose batch',

    '#options' => array(

      'batch_0' => 'batch 0',

      'batch_1' => 'batch 1',

      'batch_2' => 'batch 2',

      'batch_3' => 'batch 3',

      'batch_4' => 'batch 4',

    ),

  );

  $form['submit'] = array(

    '#type' => 'submit',

    '#value' => 'Submit',

  );

  return $form;

}

/**

 * Submit handler for the simple form.

 */

function batch_test_simple_form_submit($form, &$form_state) {

  batch_test_stack(NULL, TRUE);

  $function = '_batch_test_' . $form_state['values']['batch'];

  batch_set($function());

  $form_state['redirect'] = 'batch-test/redirect';

}

/**

 * Multistep form.

 */

function batch_test_multistep_form($form, &$form_state) {

  if (empty($form_state['storage']['step'])) {

    $form_state['storage']['step'] = 1;

  }

  $form['step_display'] = array(

    '#markup' => 'step ' . $form_state['storage']['step'] . '<br/>',

  );

  $form['submit'] = array(

    '#type' => 'submit',

    '#value' => 'Submit',

  );

  return $form;

}

/**

 * Submit handler for the multistep form.

 */

function batch_test_multistep_form_submit($form, &$form_state) {

  batch_test_stack(NULL, TRUE);

  switch ($form_state['storage']['step']) {

    case 1:

      batch_set(_batch_test_batch_1());

      break;

    case 2:

      batch_set(_batch_test_batch_2());

      break;

  }

  if ($form_state['storage']['step'] < 2) {

    $form_state['storage']['step']++;

    $form_state['rebuild'] = TRUE;

  }

  // This will only be effective on the last step.

  $form_state['redirect'] = 'batch-test/redirect';

}

/**

 * Form with chained submit callbacks.

 */

function batch_test_chained_form() {

  // This value is used to test that $form_state persists through batched

  // submit handlers.

  $form['value'] = array(

    '#type' => 'textfield',

    '#title' => 'Value',

    '#default_value' => 1,

  );

  $form['submit'] = array(

    '#type' => 'submit',

    '#value' => 'Submit',

  );

  $form['#submit'] = array(

    'batch_test_chained_form_submit_1',

    'batch_test_chained_form_submit_2',

    'batch_test_chained_form_submit_3',

    'batch_test_chained_form_submit_4',

  );

  return $form;

}

/**

 * Submit handler #1 for the chained form.

 */

function batch_test_chained_form_submit_1($form, &$form_state) {

  batch_test_stack(NULL, TRUE);

  batch_test_stack('submit handler 1');

  batch_test_stack('value = ' . $form_state['values']['value']);

  $form_state['values']['value']++;

  batch_set(_batch_test_batch_1());

  // This redirect should not be taken into account.

  $form_state['redirect'] = 'should/be/discarded';

}

/**

 * Submit handler #2 for the chained form.

 */

function batch_test_chained_form_submit_2($form, &$form_state) {

  batch_test_stack('submit handler 2');

  batch_test_stack('value = ' . $form_state['values']['value']);

  $form_state['values']['value']++;

  batch_set(_batch_test_batch_2());

  // This redirect should not be taken into account.

  $form_state['redirect'] = 'should/be/discarded';

}

/**

 * Submit handler #3 for the chained form.

 */

function batch_test_chained_form_submit_3($form, &$form_state) {

  batch_test_stack('submit handler 3');

  batch_test_stack('value = ' . $form_state['values']['value']);

  $form_state['values']['value']++;

  // This redirect should not be taken into account.

  $form_state['redirect'] = 'should/be/discarded';

}

/**

 * Submit handler #4 for the chained form.

 */

function batch_test_chained_form_submit_4($form, &$form_state) {

  batch_test_stack('submit handler 4');

  batch_test_stack('value = ' . $form_state['values']['value']);

  $form_state['values']['value']++;

  batch_set(_batch_test_batch_3());

  // This is the redirect that should prevail.

  $form_state['redirect'] = 'batch-test/redirect';

}

/**

 * Menu callback: programmatically submits the 'Chained' form.

 */

function batch_test_programmatic($value = 1) {

  $form_state = array(

    'values' => array('value' => $value)

  );

  drupal_form_submit('batch_test_chained_form', $form_state);

  return 'Got out of a programmatic batched form.';

}

/**

 * Menu callback: programmatically submits a form within a batch.

 */

function batch_test_nested_drupal_form_submit($value = 1) {

  // Set the batch and process it.

  $batch['operations'] = array(

    array('_batch_test_nested_drupal_form_submit_callback', array($value)),

  );

  batch_set($batch);

  batch_process('batch-test/redirect');

}

/**

 * Batch operation: submits form_test_mock_form using drupal_form_submit().

 */

function _batch_test_nested_drupal_form_submit_callback($value) {

  $state['values']['test_value'] = $value;

  drupal_form_submit('batch_test_mock_form', $state);

}

/**

 * A simple form with a textfield and submit button.

 */

function batch_test_mock_form($form, $form_state) {

  $form['test_value'] = array(

    '#type' => 'textfield',

  );

  $form['submit'] = array(

    '#type' => 'submit',

    '#value' => t('Submit'),

  );

  return $form;

}

/**

 * Submit handler for the batch_test_mock form.

 */

function batch_test_mock_form_submit($form, &$form_state) {

  batch_test_stack('mock form submitted with value = ' . $form_state['values']['test_value']);

}

/**

 * Menu callback: fire a batch process without a form submission.

 */

function batch_test_no_form() {

  batch_test_stack(NULL, TRUE);

  batch_set(_batch_test_batch_1());

  batch_process('batch-test/redirect');

}

/**

 * Menu callback: fire a batch process without a form submission.

 */

function batch_test_large_percentage() {

  batch_test_stack(NULL, TRUE);

  batch_set(_batch_test_batch_5());

  batch_process('batch-test/redirect');

}

/**

 * Menu callback: successful redirection.

 */

function batch_test_redirect_page() {

  return 'Redirection successful.';

}

/**

 * Batch 0: no operation.

 */

function _batch_test_batch_0() {

  $batch = array(

    'operations' => array(),

    'finished' => '_batch_test_finished_0',

    'file' => drupal_get_path('module', 'batch_test'). '/batch_test.callbacks.inc',

  );

  return $batch;

}

/**

 * Batch 1: repeats a simple operation.

 *

 * Operations: op 1 from 1 to 10.

 */

function _batch_test_batch_1() {

  // Ensure the batch takes at least two iterations.

  $total = 10;

  $sleep = (1000000 / $total) * 2;

  $operations = array();

  for ($i = 1; $i <= $total; $i++) {

    $operations[] = array('_batch_test_callback_1', array($i, $sleep));

  }

  $batch = array(

    'operations' => $operations,

    'finished' => '_batch_test_finished_1',

    'file' => drupal_get_path('module', 'batch_test'). '/batch_test.callbacks.inc',

  );

  return $batch;

}

/**

 * Batch 2: single multistep operation.

 *

 * Operations: op 2 from 1 to 10.

 */

function _batch_test_batch_2() {

  // Ensure the batch takes at least two iterations.

  $total = 10;

  $sleep = (1000000 / $total) * 2;

  $operations = array(

    array('_batch_test_callback_2', array(1, $total, $sleep)),

  );

  $batch = array(

    'operations' => $operations,

    'finished' => '_batch_test_finished_2',

    'file' => drupal_get_path('module', 'batch_test') . '/batch_test.callbacks.inc',

  );

  return $batch;

}

/**

 * Batch 3: both single and multistep operations.

 *

 * Operations:

 * - op 1 from 1 to 5,

 * - op 2 from 1 to 5,

 * - op 1 from 6 to 10,

 * - op 2 from 6 to 10.

 */

function _batch_test_batch_3() {

  // Ensure the batch takes at least two iterations.

  $total = 10;

  $sleep = (1000000 / $total) * 2;

  $operations = array();

  for ($i = 1; $i <= round($total / 2); $i++) {

    $operations[] = array('_batch_test_callback_1', array($i, $sleep));

  }

  $operations[] = array('_batch_test_callback_2', array(1, $total / 2, $sleep));

  for ($i = round($total / 2) + 1; $i <= $total; $i++) {

    $operations[] = array('_batch_test_callback_1', array($i, $sleep));

  }

  $operations[] = array('_batch_test_callback_2', array(6, $total / 2, $sleep));

  $batch = array(

    'operations' => $operations,

    'finished' => '_batch_test_finished_3',

    'file' => drupal_get_path('module', 'batch_test') . '/batch_test.callbacks.inc',

  );

  return $batch;

}

/**

 * Batch 4: batch within a batch.

 *

 * Operations:

 * - op 1 from 1 to 5,

 * - set batch 2 (op 2 from 1 to 10, should run at the end)

 * - op 1 from 6 to 10,

 */

function _batch_test_batch_4() {

  // Ensure the batch takes at least two iterations.

  $total = 10;

  $sleep = (1000000 / $total) * 2;

  $operations = array();

  for ($i = 1; $i <= round($total / 2); $i++) {

    $operations[] = array('_batch_test_callback_1', array($i, $sleep));

  }

  $operations[] = array('_batch_test_nested_batch_callback', array());

  for ($i = round($total / 2) + 1; $i <= $total; $i++) {

    $operations[] = array('_batch_test_callback_1', array($i, $sleep));

  }

  $batch = array(

    'operations' => $operations,

    'finished' => '_batch_test_finished_4',

    'file' => drupal_get_path('module', 'batch_test') . '/batch_test.callbacks.inc',

  );

  return $batch;

}

/**

 * Batch 5: repeats a simple operation.

 *

 * Operations: op 1 from 1 to 10.

 */

function _batch_test_batch_5() {

  // Ensure the batch takes at least two iterations.

  $total = 10;

  $sleep = (1000000 / $total) * 2;

  $operations = array();

  for ($i = 1; $i <= $total; $i++) {

    $operations[] = array('_batch_test_callback_5', array($i, $sleep));

  }

  $batch = array(

    'operations' => $operations,

    'finished' => '_batch_test_finished_5',

    'file' => drupal_get_path('module', 'batch_test'). '/batch_test.callbacks.inc',

  );

  return $batch;

}

/**

 * Menu callback: run a batch for testing theme used on the progress page.

 */

function batch_test_theme_batch() {

  batch_test_stack(NULL, TRUE);

  $batch = array(

    'operations' => array(

      array('_batch_test_theme_callback', array()),

    ),

  );

  batch_set($batch);

  batch_process('batch-test/redirect');

}

/**

 * Batch callback function for testing the theme used on the progress page.

 */

function _batch_test_theme_callback() {

  // Because drupalGet() steps through the full progressive batch before

  // returning control to the test function, we cannot test that the correct

  // theme is being used on the batch processing page by viewing that page

  // directly. Instead, we save the theme being used in a variable here, so

  // that it can be loaded and inspected in the thread running the test.

  global $theme;

  batch_test_stack($theme);

}

/**

 * Helper function: store or retrieve traced execution data.

 */

function batch_test_stack($data = NULL, $reset = FALSE) {

  if ($reset) {

    variable_del('batch_test_stack');

  }

  if (!isset($data)) {

    return variable_get('batch_test_stack', array());

  }

  $stack = variable_get('batch_test_stack', array());

  $stack[] = $data;

  variable_set('batch_test_stack', $stack);

}