<?php
// $Id: image_attach.module,v 1.40 2008/06/16 18:44:37 drewish Exp $

/**
 * @file image_attach.module
 */

define('IMAGE_ATTACH_HIDDEN', 'hidden');

/**
 * Implementation of hook_help().
 */
function image_attach_help($path, $arg) {
  switch ($path) {
    case 'admin/settings/modules#description':
      return t('Allows easy attaching of image nodes to other content types.');
  }
}

/**
 * Implementation of hook_menu()
 */
function image_attach_menu() {
  $items = array();

  $items['image_attach'] = array(
    'title' => 'Image Attachment View',
    'page callback' => 'image_attach_view_image',
    'access arguments' => array('access content'),
    'type' => MENU_CALLBACK,
  );
  $items['admin/settings/image/image_attach'] = array(
    'title' => 'Image attach',
    'description' => 'Enable image attach for content',
    'page callback' => 'drupal_get_form',
    'page arguments' => array('image_attach_admin_settings'),
    'access arguments' => array('administer site configuration'),
    'type' => MENU_LOCAL_TASK,
  );
  return $items;
}


function image_attach_admin_settings() {
  $form = array();

  $form['image_attach_existing'] = array(
    '#type' => 'radios',
    '#title' => t('Attach existing images'),
    '#default_value' => variable_get('image_attach_existing', 1),
    '#options' => array(0 => 'Disabled', 1 => 'Enabled'),
    '#description' => t('When enabled, will allow existing image nodes to be attached instead of uploading new images.')
  );

  return system_settings_form($form);
}

/**
 * Implementation of hook_block().
 */
function image_attach_block($op = 'list', $delta = 0, $edit = array()) {
  switch ($op) {
    case 'list':
      $blocks[0] = array(
        'info'       => t('Attached Images'),
        'status'     => TRUE,
        'weight'     => 0,
        'visibility' => 1,
        'pages'      => 'node/*',
      );
      return $blocks;

    case 'view':
      if ($delta == 0) {
        if (arg(0) == 'node' && is_numeric(arg(1))) {
          $node = node_load(arg(1));
          if (isset($node->iid) && $node->iid) {
            $output['subject'] = t('Attached Images');
            foreach ($node->iid as $this_image) {
              $image = node_load($this_image);
              if (node_access('view', $image)) {
                $img = image_display($image, variable_get('image_attach_block_0_size', IMAGE_THUMBNAIL));
                $content .= '<div class="attached-image">'.l($img, "node/$this_image", array('html' => TRUE)).'</div>';
              }
            }
            $output['content'] = $content;
            return $output;
          }
        }
      }
      break;

    case 'configure':
      if ($delta == 0) {
        $image_sizes = array();
        foreach (image_get_sizes() as $key => $size) {
          $image_sizes[$key] = $size['label'];
        }
        $form['image_attach_block_0_size'] = array(
          '#type' => 'select',
          '#title' => t('Image size'),
          '#default_value' => variable_get('image_attach_block_0_size', IMAGE_THUMBNAIL),
          '#options' => $image_sizes,
          '#description' => t("This determines the size of the image that appears in the block."),
        );
        return $form;
      }
      break;

    case 'save':
      if ($delta == 0) {
        variable_set('image_attach_block_0_size', $edit['image_attach_block_0_size']);
      }
      break;
  }
}



/**
 * implementation of hook_form_alter()
 */
function image_attach_form_alter(&$form, $form_state, $form_id) {
  // Content type settings form.
  if ($form_id == 'node_type_form' && $form['#node_type']->type != 'image') {
    _image_check_settings();

    $image_sizes = array(IMAGE_ATTACH_HIDDEN => t('<Hidden>'));
    foreach (image_get_sizes() as $key => $size) {
      $image_sizes[$key] = $size['label'];
    }
    
    $form['image_attach'] = array(
      '#type' => 'fieldset',
      '#title' => t('Image Attach settings'),
      '#collapsible' => TRUE,
      '#collapsed' => TRUE,
    );
    $form['image_attach']['image_attach'] = array(
      '#type' => 'radios',
      '#title' => t('Attach Images'),
      '#default_value' => variable_get('image_attach_'. $form['#node_type']->type, 0),
      '#options' => array(0 => t('Disabled'), 1 => t('Enabled')),
      '#description' => t('Should this node allows users to upload an image?'),
    );
    $form['image_attach']['image_attach_size_teaser'] = array(
      '#type' => 'select',
      '#title' => t('Teaser image size'),
      '#default_value' => variable_get('image_attach_size_teaser_'. $form['#node_type']->type, IMAGE_THUMBNAIL),
      '#options' => $image_sizes,
      '#description' => t("This determines the size of the image that appears when the node is displayed as a teaser. 'Hidden' will not show the image.")
    );
    $form['image_attach']['image_attach_weight_teaser'] = array(
      '#type' => 'weight',
      '#title' => t('Teaser image weight'),
      '#default_value' => variable_get('image_attach_weight_teaser_'. $form['#node_type']->type, 0),
      '#description' => t("This value determines the order of the image when displayed in the teaser."),
    );
    $form['image_attach']['image_attach_size_body'] = array(
      '#type' => 'select',
      '#title' => t('Full node image size'),
      '#default_value' => variable_get('image_attach_size_body_'. $form['#node_type']->type, IMAGE_THUMBNAIL),
      '#options' => $image_sizes,
      '#description' => t("This determines the size of the image that appears when the full node is displayed. 'Hidden' will not show the image.")
    );
    $form['image_attach']['image_attach_weight_body'] = array(
      '#type' => 'weight',
      '#title' => t('Full node image weight'),
      '#default_value' => variable_get('image_attach_weight_body_'. $form['#node_type']->type, 0),
      '#description' => t("This value determines the order of the image when displayed in the body."),
    );
  }
  // Node edit form.
  else if (isset($form['type']) && $form['type']['#value'] != 'image') {
    $type = $form['type']['#value'];
    // If enabled adjust the form.
    if ($form_id == $type .'_node_form' && variable_get('image_attach_'. $type, 0)) {
      $node = $form['#node'];
      _image_check_settings();
      $value = !empty($node->new_image) ? '#value' : '#default_value';
      $form['#attributes'] = array("enctype" => "multipart/form-data");

      $form['image_attach'] = array(
        '#type' => 'fieldset',
        '#title' => t('Attached Images'),
        '#collapsible' => TRUE,
        '#collapsed' => empty($node->iid),
      );
      // Add a custom submit handler so we can handle image creation on-the-fly
      $form['#validate'][] = 'image_attach_validate'; 

      if (!empty($node->iid)) {
        foreach ($node->iid as $iid) {
          $image = node_load($iid);
          $form['image_attach']['image_thumbnail'][] = array(
            '#type' => 'item',
            '#title' => t('Thumbnail'),
            '#value' => image_display($image, 'thumbnail')
          );
        }
      }
      if (variable_get('image_attach_existing', 1) && user_access('access content')) {
        $form['image_attach']['iid'] = array(
          '#type' => 'select',
          '#title' => t('Existing Image'),
          '#options' => _image_attach_get_image_nodes(),
          $value => empty($node->iid) ? NULL : $node->iid,
          '#description' => t('Choose an image already existing on the server if you do not upload a new one.'),
          '#multiple' => TRUE,
          '#size' => 6
        );
        $form['image_attach'][] = array(
          '#type' => 'item',
          '#value' => t('-or-'),
          '#attributes' => array('class' => 'either-choice')
        );
      }
      else {
        foreach ($node->iid as $iid) {
          $form['image_attach']['iid'][] = array(
            '#type' => 'hidden',
            $value => empty($iid) ? NULL : $iid,
          );
        }
      }
      $form['image_attach']['image'] = array(
        '#type' => 'file',
        '#title' => t('Upload Image')
      );
      $form['image_attach']['image_title'] = array(
        '#type' => 'textfield',
        '#title' => t('Image title'),
        $value => '',
        '#description' => t('The title the image will be shown with.')
      );
      //additional submit button which adds image and brings you back to node edit
      $form['image_attach']['image_attach_multiple'] = array(
        '#type' => 'submit',
        '#value' => t('Attach'),
        '#submit' => array('image_attach_image_add_submit'),
        '#validate' => array('image_attach_validate'),
      );
    }
  }
}

/**
 * This submit function adds a new image and returns you to the
 * node edit form directly afterwards, without creating the new node yet
 */
function image_attach_image_add_submit(&$form, &$form_state) {
  // Rebuild the attached image data
  if (isset($form_state['values']['iid'])) {
  	db_query("DELETE FROM {image_attach} WHERE nid=%d", $form['nid']['#value']);
    if (count($form_state['values']['iid'])) {
      $form_state['values']['iid'] = is_array($form_state['values']['iid']) ? $form_state['values']['iid'] : array($form_state['values']['iid']);
      foreach ($form_state['values']['iid'] as $iid) {
        db_query("INSERT INTO {image_attach} (nid, iid) VALUES (%d, %d)", $form['nid']['#value'], $iid);
      }
    }
  }
  
  // Rebuild the node edit form
  node_form_submit_build_node($form, &$form_state);
}

/**
 * Capture node form submission and immediately create an image if one has been
 * uploaded.
 * Note that the new image nodes are created even on preview. Taking several
 * attempts may create trash.
 */
function image_attach_validate(&$form, &$form_state) {
  $validators = array(
    'file_validate_is_image' => array(),
  );
  if ($file = file_save_upload('image', $validators)) {
    $image_title = $_POST['image_title'] ? $_POST['image_title'] : basename($file->filepath);
    // This func now does all the things to initialize an image properly
    $image = image_create_node_from($file->filepath, $image_title, '');
    if ($image && !form_get_errors()) {
      drupal_set_message(t("Created new image to attach to this node. !image_link", array('!image_link' => l($image_title, 'node/'. $image->nid) )));
      // This line changed to append to array of images instead of reset the single value
      $form_state['values']['iid'][$image->nid] = $image->nid;
    }
  }
}

/**
 * Implementation of hook_nodeapi().
 */
function image_attach_nodeapi(&$node, $op, $teaser, $page) {
  // Make sure that if an image is deleted it is detached from any nodes.
  if ($node->type == 'image') {
    switch ($op) {
      case 'delete':
        db_query("DELETE FROM {image_attach} WHERE iid=%d", $node->nid);
    }
    return;
  }
  else if (variable_get('image_attach_'. $node->type, 0) == 0) {
    return;
  }
  switch ($op) {
    case 'insert':
    case 'update':
      if (isset($node->iid)) {
        db_query("DELETE FROM {image_attach} WHERE nid=%d", $node->nid);
        if (count($node->iid)) {
          $node->iid = is_array($node->iid) ? $node->iid : array($node->iid);
          foreach ($node->iid as $iid) {
            db_query("INSERT INTO {image_attach} (nid, iid) VALUES (%d, %d)", $node->nid, $iid);
          }
        }
      }
      break;

    case 'delete':
      db_query("DELETE FROM {image_attach} WHERE nid=%d", $node->nid);
      break;

    case 'load':
      $res = (db_query("SELECT iid FROM {image_attach} WHERE nid=%d", $node->nid));
      $iids = array();
      while($iid = db_fetch_array($res) ){
        $iids[] = $iid['iid'];
      }
      return array('iid' => $iids);

    // Pass the body and teaser objects to the theme again to add the images
    case 'view':
      if (count($node->iid)) {
        $teaser_or_body = $teaser ? 'teaser' : 'body';
        $node->content['image_attach'] = array();
        if ($teaser_or_body == 'teaser') {
          $node->content['image_attach'][$node->iid[0]]['#value'] = theme("image_attach_{$teaser_or_body}", $node, $node->iid[0]);
          $node->content['image_attach'][$node->iid[0]]['#weight'] = variable_get("image_attach_weight_{$teaser_or_body}_{$node->type}", 0);
        } else {
          foreach ($node->iid as $iid) {
            $node->content['image_attach'][$iid]['#value'] = theme("image_attach_{$teaser_or_body}", $node, $iid);
            $node->content['image_attach'][$iid]['#weight'] = variable_get("image_attach_weight_{$teaser_or_body}_{$node->type}", 0);
          }
        }
      }
      break;

    case 'rss item':
      $ret = array();
      if (count($node->iid) && $image = node_load($node->iid)) {
        $info = image_get_info(file_create_path($image->images[IMAGE_PREVIEW]));
        $ret[] = array(
          'key' => 'enclosure',
          'attributes' => array(
            'url' => url("image/view/{$node->iid[0]}/". IMAGE_PREVIEW, array('absolute' => TRUE)),
            'length' => $info['file_size'],
            'type' => $info['mime_type'],
          )
        );
      }
      return $ret;
  }
}

/**
 * Fetch an array of all candidate referenced nodes, for use in presenting the selection form to the user.
 */
function _image_attach_get_image_nodes() {
  $result = db_query(db_rewrite_sql("SELECT n.nid, n.title FROM {node} n WHERE n.status=1 AND type='image' ORDER BY n.sticky DESC, n.title ASC"));
  $result_count = db_result(db_query("SELECT COUNT(*) FROM {node} n WHERE n.status=1 AND type='image'"));
  $rows = array(0 => t('None'));
  while ($node = db_fetch_object($result)) {
    $rows[$node->nid] = $node->title;
  }
  return $rows;
}

/**
 * Implementation of hook_theme() registry.
 **/
function image_attach_theme() {
  return array(
    'image_attach_teaser' => array(
      'arguments' => array('node' => new stdClass()),
    ),
    'image_attach_body' => array(
      'arguments' => array('node' => new stdClass()),
    ),
  );
}

/**
 * Theme the teaser.
 *
 * Override this in template.php to include a case statement if you want different node types to appear differently.
 * If you have additional image sizes you defined in image.module, you can use them by theming this function as well.
 */
function theme_image_attach_teaser($node, $iid) {
  $output = '';
  $img_size = variable_get('image_attach_size_teaser_'. $node->type, 'thumbnail');

  if ($img_size != IMAGE_ATTACH_HIDDEN) {
    drupal_add_css(drupal_get_path('module', 'image_attach') .'/image_attach.css');
  
    $image = node_load($iid);
    if (!node_access('view', $image)) {
      // If the image is restricted, don't show it as an attachment.
      return $output;
    }
    $info = image_get_info(file_create_path($image->images[$img_size]));

    $output .= '<div style="width: '. $info['width'] .'px" class="image-attach-teaser">';
    $output .= l(image_display($image, $img_size), "node/$iid", array('html' => TRUE));
    $output .= '</div>'."\n";
  }
  return $output;
}

/**
 * Theme the body
 */
function theme_image_attach_body($node, $iid) {
  $output = '';
  $img_size = variable_get('image_attach_size_body_'. $node->type, IMAGE_THUMBNAIL);

  if ($img_size != IMAGE_ATTACH_HIDDEN) {
    drupal_add_css(drupal_get_path('module', 'image_attach') .'/image_attach.css');

      $image = node_load($iid);
      if (!node_access('view', $image)) {
        // If the image is restricted, don't show it as an attachment.
        continue;
      }
      $info = image_get_info(file_create_path($image->images[$img_size]));

      $output .= '<div style="width: '. $info['width'] .'px" class="image-attach-body">';
      $output .= l(image_display($image, $img_size), "node/$iid", array('html' => TRUE));
      $output .= '</div>'."\n";
  }
  return $output;
}