<?php
/**
 * Automatic ajax validation
 * Drupal 6
 * @author BrendonC
 *
 *
 */


function zzzz_ajax_menu() {
  $items['admin/settings/ajax'] = array(
    'title' => t('Ajax Forms Settings'),
    'description' => t('Controls which forms should use AJAX submissions.'),
    'page callback' => 'drupal_get_form',
    'page arguments' => array('zzzz_ajax_admin'),
    'access arguments' => array('access administration pages'),
    'type' => MENU_NORMAL_ITEM,
  );
  return $items;
}

function zzzz_ajax_types() {
  return array_merge(
    zzzz_ajax_types_node(),
    zzzz_ajax_types_webform(),
    zzzz_ajax_types_module());
}

function zzzz_ajax_types_node() {
  $out = array();
  $types = node_get_types();
  foreach ($types as $k => $v) {
    $out[$k . '_node_form'] = ucwords('Content Type: ' . $v->name);
  }
  return $out;
}

function zzzz_ajax_types_module() {
  $out = array();
  $funcs = get_defined_functions();
  foreach ($funcs['user'] as $f) {
    if (preg_match("/^([a-zA-Z0-9_]+?)_submit$/", $f, $func_name)) {
      $out[$func_name[1]] = ucwords(str_replace('_', ' ', $func_name[1]));
    }
  }
  return $out;
}

function zzzz_ajax_types_webform() {
  $out = array();;
  $q =
    'SELECT ' .
    '  node.nid, ' .
    '  node.title ' .
    'FROM node ' .
    'WHERE node.type = "webform" ';
  $res = db_query($q);
  while ($row = db_fetch_array($res)) {
    $id = 'webform_client_form_' . (int) $row['nid'];
    $out[$id] = 'Webform: ' . $row['title'];
  }
  return $out;
}

function zzzz_ajax_admin() {
  $form = array();
  $form['ajax_types'] = array(
    '#type' => 'select',
    '#multiple' => TRUE,
    '#options' => zzzz_ajax_types(),
    '#title' => t('Forms to Use AJAX Submissions'),
    '#default_value' => variable_get('ajax_types', array()),
    '#size' => 20,
    '#description' => t(
      "Select the forms you wish to use AJAX submissions.")
  );
  return system_settings_form($form);
}

function zzzz_ajax_init() {
  drupal_add_js(drupal_get_path('module', 'zzzz_ajax') .
    '/Ajax.js', 'module');
}


function zzzz_ajax_istype($form_id) {
  return array_key_exists($form_id,
    variable_get('ajax_types', array()));
}


function zzzz_ajax_form_alter(&$form, $form_state, $form_id) {
  //watchdog('AJAX Forms', 'Form ID: %form_id', array('%form_id'=>$form_id));
  if (zzzz_ajax_isType($form_id)) {
    $found = FALSE;
    zzzz_ajax_setcommentform($form, $form_id);
    zzzz_ajax_setvalidator($form);
    zzzz_ajax_findsubmitter($form, $found);
    zzzz_ajax_setsubmitter($form, $found);
  }
  //print "<pre>";
  //print_r($form);
  //print_r("</pre>");
  return true;
}

function zzzz_ajax_setcommentform(&$form, $form_id) {
  if ($form_id === 'comment_form') {
    $form['submit'] = array(
      '#type' => 'submit',
      '#value' => t('Save'),
      '#weight' => 19
    );
  }
}


function zzzz_ajax_setvalidator(&$form) {
  $form['#validate'][] = 'zzzz_ajax_validator';
}

function zzzz_ajax_setsubmitter(&$form, $found) {
  if (!$found) {
    $form['#submit'][] = 'zzzz_ajax_submitter';
  }
}

function zzzz_ajax_findsubmitter(&$form, &$found) {
  foreach ($form as $form_key => $form_val) {
    if (is_array($form[$form_key])) {
      //submit or preview button
      if (
          (array_key_exists('#type', $form[$form_key]) &&
          ($form[$form_key]['#type'] === 'submit' ||
          $form[$form_key]['#type'] === 'button')) &&
          ($form_key === 'submit' || $form_key === 'preview')) {
        $form[$form_key]['#attributes'] =
          array('onclick' => 'return Ajax.go(this);');
        if (array_key_exists('#submit', $form[$form_key]) &&
            !empty($form[$form_key]['#submit'])) {
          $form[$form_key]['#submit'][] = 'zzzz_ajax_submitter';
          $found = TRUE;
        }
      }
      //nested
      else {
        zzzz_ajax_findsubmitter($form[$form_key], $found);
      }
    }
  }
  return true;
}

/**
 * Sometimes the redirect can be an array in the form of
 *   0 => path
 *   1 => query
 *   2 => fragment
 */
function zzzz_ajax_getredirect($redirect) {
  //watchdog('AJAX Forms', 'Redirect: !redirect',
  //  array('!redirect'=>gettype($redirect[1])));
  $args = array();
  $args['absolute'] = TRUE;
  if (is_array($redirect)) {
    if ($redirect[1] !== NULL) {
      $args['query'] = $redirect[1];
    }
    if ($redirect[2] !== NULL) {
      $args['fragment'] = $redirect[2];
    }
    $path = $redirect[0];
  }
  else {
    $path = $redirect;
  }
  return url($path, $args);
}

function zzzz_ajax_submitter(&$form, &$form_state) {
  $data = array();
  // Node Preview
  if (array_key_exists('node_preview', $form_state) &&
      $form_state['node_preview'] !== NULL) {
    $data['preview'] = $form_state['node_preview'];
  }
  // Go to redirection page
  if (array_key_exists('redirect', $form_state) &&
      $form_state['redirect'] !== NULL) {
    $data['redirect'] = $form_state['redirect'];
  }
  // Display messages internally without redirect
  else {
    $messages = drupal_get_messages(NULL, TRUE);
    if (array_key_exists('status', $messages)) {
      $data['messages_status'] = $messages['status'];
    }
    if (array_key_exists('warning', $messages)) {
      $data['messages_warning'] = $messages['warning'];
    }
  }
  $out = zzzz_ajax_build($data);
  zzzz_ajax_out($out);
}

function zzzz_ajax_validator(&$form, &$form_state) {
  if (array_key_exists('ajax', $_REQUEST)) {
    drupal_get_messages(NULL, TRUE);
    $data = zzzz_ajax_build(array(
      'messages_error' => form_get_errors()
    ));
    // Time to send messages
    if (!$data['status']) {
      zzzz_ajax_out($data);
    }
    // Special exception for the comment forms
    elseif ($form['form_id']['#value'] === 'comment_form' &&
        $form['#post']['op'] === t('Preview')) {
      $data['preview'] = zzzz_ajax_comment_form_preview($form, $form_state);
      zzzz_ajax_out($data);
    }
  }
}

// This function taken from comment_form_add_preview
function zzzz_ajax_comment_form_preview(&$form, &$form_state) {
  global $user;
  $edit = $form_state['values'];
  $output = '';
  $node = node_load($edit['nid']);
  _comment_form_submit($edit);
  $comment = (object)$edit;
  // Attach the user and time information.
  if (!empty($edit['author'])) {
    $account = user_load(array('name' => $edit['author']));
  }
  elseif ($user->uid && !isset($edit['is_anonymous'])) {
    $account = $user;
  }
  if (!empty($account)) {
    $comment->uid = $account->uid;
    $comment->name = check_plain($account->name);
  }
  elseif (empty($comment->name)) {
    $comment->name = variable_get('anonymous', t('Anonymous'));
  }
  $comment->timestamp = !empty($edit['timestamp']) ?
    $edit['timestamp'] : time();
  if ($edit['pid']) {
    $comment = db_fetch_object(db_query(
      'SELECT c.*, u.uid, u.name AS registered_name, ' .
      'u.signature, u.picture, u.data FROM {comments} c ' .
      'INNER JOIN {users} u ON c.uid = u.uid ' .
      'WHERE c.cid = %d AND c.status = %d',
      $edit['pid'], COMMENT_PUBLISHED));
    $comment = drupal_unpack($comment);
    $comment->name = $comment->uid ? $comment->registered_name : $comment->name;
  }
  else {
    $suffix = empty($form['#suffix']) ? '' : $form['#suffix'];
    $form['#suffix'] = $suffix . node_view($node);
    $edit['pid'] = 0;
  }
  $output  = '<div class="preview">';
  $output .= theme('comment_view', $comment, $node);
  $output .= '</div>';
  return $output;
}


function zzzz_ajax_out($data) {
  header("Content-Type: application/json; charset=UTF-8");
  print drupal_to_js($data);
  exit;
}

function zzzz_ajax_cleanid($field_id) {
  //taken from form_clean_id
  return str_replace(array('][', '_', ' '), '-', $field_id);
}

function zzzz_ajax_build($data) {
  $out = array(
    'status' => true,
    'messages_error' => array(),
    'messages_status' => array(),
    'messages_warning' => array(),
    'redirect' => NULL,
    'preview' => NULL
  );
  // MESSAGE:ERROR
  if (array_key_exists('messages_error', $data) &&
      $data['messages_error'] !== NULL) {
    $out['status'] = FALSE;
    foreach ($data['messages_error'] as $k => $v) {
      $out['messages_error'][] = array(
        'id' => zzzz_ajax_cleanid("edit-" . $k),
        'value' => $v
      );
    }
  }
  // MESSAGE:STATUS
  if (array_key_exists('messages_status', $data) &&
      $data['messages_status'] !== NULL) {
    foreach ($data['messages_status'] as $k => $v) {
      $out['messages_status'][] = array(
        'id' => (int)$k,
        'value' => $v
      );
    }
  }
  // MESSAGE:WARNING
  if (array_key_exists('messages_warning', $data) &&
      $data['messages_warning'] !== null) {
    foreach ($data['messages_warning'] as $k => $v) {
      $out['messages_warning'][] = array(
        'id' => (int)$k,
        'value' => $v
      );
    }
  }
  // Redirect
  if (array_key_exists('redirect', $data) && $data['redirect'] !== NULL) {
    $out['redirect'] = zzzz_ajax_getredirect($data['redirect']);
  }
  // Preview
  if (array_key_exists('preview', $data) && $data['preview'] !== NULL) {
    $out['preview'] = rawurlencode($data['preview']);
  }
  return $out;
}