invite.controller.inc in Invite 7.4

File

includes/invite.controller.inc

<?php

/**
 * @file
 */

/**
 *
 */
class InviteController extends EntityAPIController {

  /**
   *
   */
  public function create(array $values = array()) {
    global $user;
    $values += array(
      'created' => REQUEST_TIME,
      'expiry' => REQUEST_TIME + variable_get('invite_default_expiry_time', 30) * 24 * 60 * 60,
      'uid' => $user->uid,
      'invitee' => 0,
      'status' => 1,
      'reg_code' => invite_generate_code(),
      'data' => array(),
    );
    return parent::create($values);
  }

  /**
   * @param Invite $entity
   * @param string $view_mode
   * @param null $langcode
   * @param array $content
   * @return mixed
   */
  public function buildContent($entity, $view_mode = 'full', $langcode = NULL, $content = array()) {
    $account = user_load($entity->uid);
    $content['author'] = array(
      '#prefix' => '<div>',
      '#suffix' => '</div>',
      '#markup' => t('Created by: <strong>!author</strong>', array(
        '!author' => theme('username', array(
          'account' => $account,
        )),
      )),
    );
    $content['status'] = array(
      '#prefix' => '<div>',
      '#suffix' => '</div>',
      '#markup' => t('Status: <strong>!status</strong>', array(
        '!status' => $entity
          ->status(TRUE),
      )),
    );
    return parent::buildContent($entity, $view_mode, $langcode, $content);
  }

}

/**
 *
 */
class InviteTypeController extends EntityAPIControllerExportable {

  /**
   *
   */
  public function create(array $values = array()) {
    $values += array(
      'label' => '',
      'description' => '',
      'data' => array(),
    );
    return parent::create($values);
  }

  /**
   * Save Invite Type.
   */
  public function save($entity, DatabaseTransaction $transaction = NULL) {
    parent::save($entity, $transaction);

    // Rebuild menu registry. We do not call menu_rebuild directly, but set
    // variable that indicates rebuild in the end.
    // @see _http://drupal.org/node/1399618
    variable_set('menu_rebuild_needed', TRUE);
  }

}

/**
 * UI controller for Task Type.
 */
class InviteTypeUIController extends EntityDefaultUIController {

  /**
   * Overrides hook_menu() defaults.
   */
  public function hook_menu() {
    $items = parent::hook_menu();
    $items[$this->path]['description'] = 'Manage Invite types.';
    return $items;
  }

}

/**
 * Invite class.
 */
class Invite extends Entity {

  /**
   * The user id of the profile owner.
   *
   * @var int
   */
  public $uid;

  /**
   *
   */
  public function type_details() {
    return invite_get_types($this->type);
  }

  /**
   * Returns the user owning this profile.
   */
  public function inviter() {
    return user_load($this->uid);
  }

  /**
   * Returns the user owning this profile.
   */
  public function invitee() {
    if (!empty($this->invitee)) {
      return user_load($this->invitee);
    }
    else {
      return FALSE;
    }
  }

  /**
   *
   */
  public function joinLink() {
    return url('invite/accept/' . $this->reg_code, array(
      'absolute' => TRUE,
    ));
  }

  /**
   *
   */
  public function sendInvite() {
    $result = FALSE;
    if (!empty($this->sendNotification)) {

      // Notify other modules.
      entity_get_controller('invite')
        ->invoke('send', $this);
      $result = TRUE;
    }
    else {
      $result = TRUE;
    }
    return $result;
  }

  /**
   * Sets a new user owning this profile.
   *
   * @param $account
   *   The user account object or the user account id (uid).
   */
  public function setUser($account) {
    $this->uid = is_object($account) ? $account->uid : $account;
  }

  /**
   *
   */
  public function __construct($values = array()) {
    if (isset($values['inviter'])) {
      $this
        ->setUser($values['inviter']);
      unset($values['inviter']);
    }
    if (!isset($values['label']) && isset($values['type']) && ($type = invite_get_types($values['type']))) {

      // Initialize the label with the type label, so newly created profiles
      // have that as interim label.
      $values['label'] = $type->label;
    }
    $this->sendNotification = TRUE;
    parent::__construct($values, 'invite');
  }

  /**
   *
   */
  public function status($readable = FALSE) {
    $options = invite_get_status_options();
    return $readable ? $options[$this->status] : $this->status;
  }

  /**
   *
   */
  protected function defaultLabel() {
    $entity = entity_metadata_wrapper('invite', $this);
    return t('Invitation #@number', array(
      '@number' => $this
        ->identifier(),
    ));
  }

  /**
   *
   */
  protected function defaultUri() {
    return array(
      'path' => 'invite/' . $this
        ->identifier(),
    );
  }

  /**
   *
   */
  public function save() {
    return parent::save();
  }

}

/**
 * Invite Type class.
 */
class InviteType extends Entity {
  public $type;
  public $label;
  public $weight = 0;

  /**
   * Array with the argument keys.
   *
   * This is used to allow creation of entity metadata properties based
   * on the argument keys.
   *
   * @see InviteMetadataController::entityPropertyInfo()
   *
   * @var array
   */
  public $argument_keys = array();
  public $argument = array();
  public $invite_sending_controller = array();

  /**
   *
   */
  public function __construct($values = array()) {
    parent::__construct($values, 'invite_type');
    if (empty($this->invite_sending_controller)) {
      $controllers = db_select('invite_sending_controller', 'isc')
        ->fields('isc')
        ->condition('type', array(
        $this->type,
      ))
        ->execute();
      $this->invite_sending_controller = array();
      foreach ($controllers as $controller) {
        $this->invite_sending_controller[$controller->name] = $controller->name;
      }
    }
  }

  /**
   *
   */
  public function hasSendingController($controller_name) {
    $result = FALSE;
    if (!empty($this->invite_sending_controller[$controller_name])) {
      $result = TRUE;
    }
    return $result;
  }

  /**
   *
   */
  public function isLocked() {
    return isset($this->status) && empty($this->is_new) && ($this->status & ENTITY_IN_CODE || $this->status & ENTITY_FIXED);
  }

  /**
   *
   */
  public function save() {
    parent::save();
    $controllers = array();

    // Remove old controller entries.
    db_delete('invite_sending_controller')
      ->condition('type', $this->type)
      ->execute();

    // Add controllers.
    foreach ($this->invite_sending_controller as $controller_name) {
      $module = !empty($this->module) ? $this->module : '';
      if (!in_array($controller_name, $controllers)) {
        $controllers[] = $controller_name;
        db_insert('invite_sending_controller')
          ->fields(array(
          'type' => $this->type,
          'module' => $module,
          'name' => $controller_name,
        ))
          ->execute();
      }
    }
  }

  /**
   *
   */
  public function delete() {
    parent::delete();
    db_delete('invite_sending_controller')
      ->condition('type', $this->type)
      ->execute();
  }

}

/**
 * Generates a unique tracking code.
 *
 * @return string
 *   An 10-digit unique tracking code.
 */
function invite_generate_code() {
  do {
    $reg_code = user_password(10);
    $result = (bool) db_query_range('SELECT reg_code FROM {invite} WHERE reg_code = :regcode', 0, 1, array(
      ':regcode' => $reg_code,
    ))
      ->fetchField();
  } while ($result);
  return $reg_code;
}

Functions

Name	Description
invite_generate_code	Generates a unique tracking code.

Classes

Name	Description
Invite	Invite class.
InviteController
InviteType	Invite Type class.
InviteTypeController
InviteTypeUIController	UI controller for Task Type.