333 lines
		
	
	
		
			9.2 KiB
		
	
	
	
		
			PHP
		
	
	
			
		
		
	
	
			333 lines
		
	
	
		
			9.2 KiB
		
	
	
	
		
			PHP
		
	
	
<?php
 | 
						|
// $Id$
 | 
						|
 | 
						|
/**
 | 
						|
 * @file
 | 
						|
 * Defines an entity type.
 | 
						|
 */
 | 
						|
 | 
						|
/**
 | 
						|
 * Implements hook_entity_info().
 | 
						|
 */
 | 
						|
function field_test_entity_info() {
 | 
						|
  $bundles = variable_get('field_test_bundles', array('test_bundle' => array('label' => 'Test Bundle')));
 | 
						|
  $test_entity_modes = array(
 | 
						|
    'full' => array(
 | 
						|
      'label' => t('Full object'),
 | 
						|
    ),
 | 
						|
    'teaser' => array(
 | 
						|
      'label' => t('Teaser'),
 | 
						|
    ),
 | 
						|
  );
 | 
						|
 | 
						|
  return array(
 | 
						|
    'test_entity' => array(
 | 
						|
      'name' => t('Test Entity'),
 | 
						|
      'entity keys' => array(
 | 
						|
        'id' => 'ftid',
 | 
						|
        'revision' => 'ftvid',
 | 
						|
        'bundle' => 'fttype',
 | 
						|
      ),
 | 
						|
      'cacheable' => FALSE,
 | 
						|
      'bundles' => $bundles,
 | 
						|
      'fieldable' => TRUE,
 | 
						|
      'view modes' => $test_entity_modes,
 | 
						|
    ),
 | 
						|
    // This entity type doesn't get form handling for now...
 | 
						|
    'test_cacheable_entity' => array(
 | 
						|
      'name' => t('Test Entity, cacheable'),
 | 
						|
      'entity keys' => array(
 | 
						|
        'id' => 'ftid',
 | 
						|
        'revision' => 'ftvid',
 | 
						|
        'bundle' => 'fttype',
 | 
						|
      ),
 | 
						|
      'fieldable' => TRUE,
 | 
						|
      'cacheable' => TRUE,
 | 
						|
      'bundles' => $bundles,
 | 
						|
      'view modes' => $test_entity_modes,
 | 
						|
    ),
 | 
						|
  );
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Implements hook_entity_info_alter().
 | 
						|
 */
 | 
						|
function field_test_entity_info_alter(&$entity_info) {
 | 
						|
  // Enable/disable field_test as a translation handler.
 | 
						|
  foreach (field_test_entity_info_translatable() as $entity_type => $translatable) {
 | 
						|
    $entity_info[$entity_type]['translation']['field_test'] = $translatable;
 | 
						|
  }
 | 
						|
  // Disable locale as a translation handler.
 | 
						|
  foreach ($entity_info as $entity_type => $info) {
 | 
						|
    $entity_info[$entity_type]['translation']['locale'] = FALSE;
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Helper function to enable entity translations.
 | 
						|
 */
 | 
						|
function field_test_entity_info_translatable($entity_type = NULL, $translatable = NULL) {
 | 
						|
  drupal_static_reset('field_has_translation_handler');
 | 
						|
  $stored_value = &drupal_static(__FUNCTION__, array());
 | 
						|
  if (isset($entity_type)) {
 | 
						|
    $stored_value[$entity_type] = $translatable;
 | 
						|
    entity_info_cache_clear();
 | 
						|
  }
 | 
						|
  return $stored_value;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Creates a new bundle for test_entity entities.
 | 
						|
 *
 | 
						|
 * @param $bundle
 | 
						|
 *   The machine-readable name of the bundle.
 | 
						|
 * @param $text
 | 
						|
 *   The human-readable name of the bundle. If none is provided, the machine
 | 
						|
 *   name will be used.
 | 
						|
 */
 | 
						|
function field_test_create_bundle($bundle, $text = NULL) {
 | 
						|
  $bundles = variable_get('field_test_bundles', array('test_bundle' => array('label' => 'Test Bundle')));
 | 
						|
  $bundles += array($bundle => array('label' => $text ? $text : $bundle));
 | 
						|
  variable_set('field_test_bundles', $bundles);
 | 
						|
 | 
						|
  $info = field_test_entity_info();
 | 
						|
  foreach ($info as $type => $type_info) {
 | 
						|
    field_attach_create_bundle($type, $bundle);
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Renames a bundle for test_entity entities.
 | 
						|
 *
 | 
						|
 * @param $bundle_old
 | 
						|
 *   The machine-readable name of the bundle to rename.
 | 
						|
 * @param $bundle_new
 | 
						|
 *   The new machine-readable name of the bundle.
 | 
						|
 */
 | 
						|
function field_test_rename_bundle($bundle_old, $bundle_new) {
 | 
						|
  $bundles = variable_get('field_test_bundles', array('test_bundle' => array('label' => 'Test Bundle')));
 | 
						|
  $bundles[$bundle_new] = $bundles[$bundle_old];
 | 
						|
  unset($bundles[$bundle_old]);
 | 
						|
  variable_set('field_test_bundles', $bundles);
 | 
						|
 | 
						|
  $info = field_test_entity_info();
 | 
						|
  foreach ($info as $type => $type_info) {
 | 
						|
    field_attach_rename_bundle($type, $bundle_old, $bundle_new);
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Deletes a bundle for test_entity objects.
 | 
						|
 *
 | 
						|
 * @param $bundle
 | 
						|
 *   The machine-readable name of the bundle to delete.
 | 
						|
 */
 | 
						|
function field_test_delete_bundle($bundle) {
 | 
						|
  $bundles = variable_get('field_test_bundles', array('test_bundle' => array('label' => 'Test Bundle')));
 | 
						|
  unset($bundles[$bundle]);
 | 
						|
  variable_set('field_test_bundles', $bundles);
 | 
						|
 | 
						|
  $info = field_test_entity_info();
 | 
						|
  foreach ($info as $type => $type_info) {
 | 
						|
    field_attach_delete_bundle($type, $bundle);
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Creates a basic test_entity entity.
 | 
						|
 */
 | 
						|
function field_test_create_stub_entity($id = 1, $vid = 1, $bundle = 'test_bundle') {
 | 
						|
  $entity = new stdClass();
 | 
						|
  // Only set id and vid properties if they don't come as NULL (creation form).
 | 
						|
  if (isset($id)) {
 | 
						|
    $entity->ftid = $id;
 | 
						|
  }
 | 
						|
  if (isset($vid)) {
 | 
						|
    $entity->ftvid = $vid;
 | 
						|
  }
 | 
						|
  $entity->fttype = $bundle;
 | 
						|
 | 
						|
  return $entity;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Loads a test_entity.
 | 
						|
 *
 | 
						|
 * @param $ftid
 | 
						|
 *   The id of the entity to load.
 | 
						|
 * @param $ftvid
 | 
						|
 *   (Optional) The revision id of the entity to load. If not specified, the
 | 
						|
 *   current revision will be used.
 | 
						|
 * @return
 | 
						|
 *   The loaded entity.
 | 
						|
 */
 | 
						|
function field_test_entity_test_load($ftid, $ftvid = NULL) {
 | 
						|
  // Load basic strucure.
 | 
						|
  $query = db_select('test_entity', 'fte', array())
 | 
						|
    ->fields('fte')
 | 
						|
    ->condition('ftid', $ftid);
 | 
						|
  if ($ftvid) {
 | 
						|
    $query->condition('ftvid', $ftvid);
 | 
						|
  }
 | 
						|
  $entities = $query->execute()->fetchAllAssoc('ftid');
 | 
						|
 | 
						|
  // Attach fields.
 | 
						|
  if ($ftvid) {
 | 
						|
    field_attach_load_revision('test_entity', $entities);
 | 
						|
  }
 | 
						|
  else {
 | 
						|
    field_attach_load('test_entity', $entities);
 | 
						|
  }
 | 
						|
 | 
						|
  return $entities[$ftid];
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Saves a test_entity.
 | 
						|
 *
 | 
						|
 * A new entity is created if $entity->ftid and $entity->is_new are both empty.
 | 
						|
 * A new revision is created if $entity->revision is not empty.
 | 
						|
 *
 | 
						|
 * @param $entity
 | 
						|
 *   The entity to save.
 | 
						|
 */
 | 
						|
function field_test_entity_save(&$entity) {
 | 
						|
  field_attach_presave('test_entity', $entity);
 | 
						|
 | 
						|
  if (!isset($entity->is_new)) {
 | 
						|
    $entity->is_new = empty($entity->ftid);
 | 
						|
  }
 | 
						|
 | 
						|
  if (!$entity->is_new && !empty($entity->revision)) {
 | 
						|
    $entity->old_ftvid = $entity->ftvid;
 | 
						|
    unset($entity->ftvid);
 | 
						|
  }
 | 
						|
 | 
						|
  $update_entity = TRUE;
 | 
						|
  if ($entity->is_new) {
 | 
						|
    drupal_write_record('test_entity', $entity);
 | 
						|
    drupal_write_record('test_entity_revision', $entity);
 | 
						|
    $op = 'insert';
 | 
						|
  }
 | 
						|
  else {
 | 
						|
    drupal_write_record('test_entity', $entity, 'ftid');
 | 
						|
    if (!empty($entity->revision)) {
 | 
						|
      drupal_write_record('test_entity_revision', $entity);
 | 
						|
    }
 | 
						|
    else {
 | 
						|
      drupal_write_record('test_entity_revision', $entity, 'ftvid');
 | 
						|
      $update_entity = FALSE;
 | 
						|
    }
 | 
						|
    $op = 'update';
 | 
						|
  }
 | 
						|
  if ($update_entity) {
 | 
						|
    db_update('test_entity')
 | 
						|
      ->fields(array('ftvid' => $entity->ftvid))
 | 
						|
      ->condition('ftid', $entity->ftid)
 | 
						|
      ->execute();
 | 
						|
  }
 | 
						|
 | 
						|
  // Save fields.
 | 
						|
  $function = "field_attach_$op";
 | 
						|
  $function('test_entity', $entity);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Menu callback: displays the 'Add new test_entity' form.
 | 
						|
 */
 | 
						|
function field_test_entity_add($fttype) {
 | 
						|
  $fttype = str_replace('-', '_', $fttype);
 | 
						|
  $entity = (object)array('fttype' => $fttype);
 | 
						|
  drupal_set_title(t('Create test_entity @bundle', array('@bundle' => $fttype)), PASS_THROUGH);
 | 
						|
  return drupal_get_form('field_test_entity_form', $entity, TRUE);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Menu callback: displays the 'Edit exiisting test_entity' form.
 | 
						|
 */
 | 
						|
function field_test_entity_edit($entity) {
 | 
						|
  drupal_set_title(t('test_entity @ftid revision @ftvid', array('@ftid' => $entity->ftid, '@ftvid' => $entity->ftvid)), PASS_THROUGH);
 | 
						|
  return drupal_get_form('field_test_entity_form', $entity);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Test_entity form.
 | 
						|
 */
 | 
						|
function field_test_entity_form($form, &$form_state, $entity, $add = FALSE) {
 | 
						|
  if (isset($form_state['test_entity'])) {
 | 
						|
    $entity = $form_state['test_entity'] + (array)$entity;
 | 
						|
  }
 | 
						|
  $entity = (object)$entity;
 | 
						|
 | 
						|
  foreach (array('ftid', 'ftvid', 'fttype') as $key) {
 | 
						|
    $form[$key] = array(
 | 
						|
      '#type' => 'value',
 | 
						|
      '#value' => isset($entity->$key) ? $entity->$key : NULL,
 | 
						|
    );
 | 
						|
  }
 | 
						|
 | 
						|
  // Add field widgets.
 | 
						|
  $form['#builder_function'] = 'field_test_entity_form_submit_builder';
 | 
						|
  field_attach_form('test_entity', $entity, $form, $form_state);
 | 
						|
 | 
						|
  if (!$add) {
 | 
						|
    $form['revision'] = array(
 | 
						|
      '#access' => user_access('administer field_test content'),
 | 
						|
      '#type' => 'checkbox',
 | 
						|
      '#title' => t('Create new revision'),
 | 
						|
      '#default_value' => FALSE,
 | 
						|
      '#weight' => 100,
 | 
						|
    );
 | 
						|
  }
 | 
						|
  $form['submit'] = array(
 | 
						|
    '#type' => 'submit',
 | 
						|
    '#value' => t('Save'),
 | 
						|
    '#weight' => 101,
 | 
						|
  );
 | 
						|
 | 
						|
  return $form;
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Validate handler for field_test_entity_form().
 | 
						|
 */
 | 
						|
function field_test_entity_form_validate($form, &$form_state) {
 | 
						|
  $entity = field_test_create_stub_entity($form_state['values']['ftid'], $form_state['values']['ftvid'], $form_state['values']['fttype']);
 | 
						|
  field_attach_form_validate('test_entity', $entity, $form, $form_state);
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Submit handler for field_test_entity_form().
 | 
						|
 */
 | 
						|
function field_test_entity_form_submit($form, &$form_state) {
 | 
						|
  $entity = field_test_entity_form_submit_builder($form, $form_state);
 | 
						|
  $insert = empty($entity->ftid);
 | 
						|
  field_test_entity_save($entity);
 | 
						|
 | 
						|
  $message = $insert ? t('test_entity @id has been created.', array('@id' => $entity->ftid)) : t('test_entity @id has been updated.', array('@id' => $entity->ftid));
 | 
						|
  drupal_set_message($message);
 | 
						|
 | 
						|
  if ($entity->ftid) {
 | 
						|
    unset($form_state['rebuild']);
 | 
						|
    $form_state['redirect'] = 'test-entity/' . $entity->ftid . '/edit';
 | 
						|
  }
 | 
						|
  else {
 | 
						|
    // Error on save.
 | 
						|
    drupal_set_message(t('The entity could not be saved.'), 'error');
 | 
						|
  }
 | 
						|
}
 | 
						|
 | 
						|
/**
 | 
						|
 * Builds a test_entity from submitted form values.
 | 
						|
 */
 | 
						|
function field_test_entity_form_submit_builder($form, &$form_state) {
 | 
						|
  $entity = field_test_create_stub_entity($form_state['values']['ftid'], $form_state['values']['ftvid'], $form_state['values']['fttype']);
 | 
						|
  $entity->revision = !empty($form_state['values']['revision']);
 | 
						|
  field_attach_submit('test_entity', $entity, $form, $form_state);
 | 
						|
 | 
						|
  $form_state['test_entity'] = (array)$entity;
 | 
						|
  $form_state['rebuild'] = TRUE;
 | 
						|
 | 
						|
  return $entity;
 | 
						|
}
 |