--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/popups_reference.module	Wed Jan 21 11:19:01 2009 +0100
@@ -0,0 +1,151 @@
+<?php
+// $Id: popups_reference.module,v 1.1.2.6 2009/01/18 22:40:33 starbow Exp $
+
+/**
+ * @file
+ * Modify the Node Reference widget to use a popup to add a new node.
+ */ 
+
+
+/**
+ * Implementation of hook_form_alter().
+ * 
+ * Modifies the nodereference setting form and the basic node form. 
+ */
+function popups_reference_form_alter(&$form, $form_state, $form_id) {
+  if ($form_id == 'content_field_edit_form') {
+    // Add a checkbox to the nodereference settings page.
+    $field_name = $form['#field']['field_name'];
+    $form['field']['show_add_link'] = array(
+    '#type' => 'checkbox',
+      '#default_value' => variable_get('popups_reference_show_add_link_'. $field_name, TRUE),
+      '#title' => t('Show the "Add New: Node Type" Popup links'),
+      '#description' => t("Activate Popups:Add and Reference behavior for this reference.")
+    );
+    $form['#submit'][] = '_popups_reference_manage_fields_submit';
+  }
+  elseif (isset($form['type'])) {
+    // Add the "Add New: Node Type" links.
+    $node = $form['#node'];
+    if ($form['type']['#value'] .'_node_form' == $form_id) {  
+      $fields = content_fields();
+      foreach ($form as $key => $item) {
+        if (is_array($item)) {
+          $type = $item['#type'];
+          if ($type == 'fieldset') { // Loop through all the subitems.
+            foreach ($form[$key] as $subkey => $subitem) {
+              popups_reference_alter_item($form[$key], $subkey, $subitem, $fields);
+            }
+          }
+          else {
+            popups_reference_alter_item($form, $key, $item, $fields);
+          }
+        }
+        
+      }
+    }
+  }
+}
+
+/**
+ * Implementation of hook_nodeapi().
+ * Add cookies with node info when a new node is created.
+ * These cookies will be found by the popups_reference behavior and used
+ *   to select the newly created node in the reference widget.
+ */ 
+function popups_reference_nodeapi($node, $op) {
+  if ($op == 'insert') {
+      $five = time()+300; // 5 minutes in the future.
+      setcookie("PopupRefNid", $node->nid, $five, '/'); 
+      setcookie("PopupRefTitle", $node->title, $five, '/');
+  }
+}
+
+/**
+ * Submit added to the the nodereference settings form.
+ * Set a variable for each nodereference field.
+ */
+function _popups_reference_manage_fields_submit($form, &$form_state) {
+  $field_name = $form['#field']['field_name'];
+  variable_set('popups_reference_show_add_link_'. $field_name, $form_state['values']['show_add_link']);
+}
+
+/**
+ * Run on every element in the basic node form.
+ * Wrap the enabled nodereference fields, and add the popup links.
+ *
+ * @param $form - the form (or fieldgroup).
+ * @param $key - form element name.
+ * @param $item - the form element array.
+ * @param $fields - all fields info.
+ */
+function popups_reference_alter_item(&$form, $key, $item, $fields) {  
+  $field_name = strstr($key, 'field_'); // Check if $key starts with 'field_';
+  if (isset($fields[$field_name]) && 
+      $fields[$field_name]['type'] == 'nodereference' &&
+      variable_get('popups_reference_show_add_link_'. $field_name, TRUE)) {
+    $type = $form['type']['#value'];
+    $field = content_fields($field_name, $form['type']['#value']);
+
+    $wrapper_id = 'popups-reference-' . _popups_reference_counter();
+    $links = _popups_reference_links($field, $type, $wrapper_id, $field['widget']['type']);
+    if ($links) {
+      // Put the nodereference widget and links in an wrapper.
+      // Makes it easy to find for Ahah targeting, and popups_reference behavior selecting.
+      $form[$key]['#prefix'] = '<div id="'. $wrapper_id .'">';
+      $form[$key]['#suffix'] = '<div>Add New: ' . implode(', ', $links) .'</div></div>';
+    }
+  }
+}
+
+/**
+ * Generates 'Add new...' link
+ * for each allowed content type
+ *
+ * @param $field
+ * @param $src_type - the type of base node.
+ * @param $wrapper_id - id for the wrapper around the node reference.
+ * @param $type - the type of the node being referenced.
+ * @return Array of html links.
+ */
+function _popups_reference_links($field, $src_type, $wrapper_id, $type) {
+  if ($type == 'nodereference_select' || $type == 'nodereference_buttons') { 
+    // Target the wrapper for replacing.
+    popups_add_popups(array('a.'.$wrapper_id=>array('targetSelectors'=>array('#'.$wrapper_id))));
+  }
+  else if ($type == 'nodereference_autocomplete') { 
+    // Don't replace the autocomplete when done.
+    popups_add_popups(array('a.'.$wrapper_id=>array('noUpdate'=>TRUE)));
+  }
+  else { // Unsupported type.
+    return;
+  }
+  $options = array(
+    'attributes' => array(
+      'class' => $wrapper_id . ' popups-reference', 
+      'rel' => $wrapper_id,
+  ),
+    'query' => array('destination' => 'node/add/' . str_replace('_', '-', $src_type)),  
+  );
+  $links = array();
+  $all_types = node_get_types();
+  foreach ($field['referenceable_types'] as $add_type => $value) {
+    if (!empty($value) && user_access("create $add_type content")) {
+      drupal_add_js(drupal_get_path('module', 'popups_reference') .'/popups_reference.js');
+      $path = 'node/add/' . str_replace('_', '-', $add_type);
+      $links[] = l("Add $add_type", $path, $options);
+    }
+  }
+  return $links;
+}
+
+/**
+ * A counter for generating unique element id's.
+ *
+ * @return int: next integer.
+ */
+function _popups_reference_counter() {
+  static $count = 0;
+  return $count++;
+}
+