View source
<?php
namespace Drupal\push_notifications\Entity;
use Drupal\Core\Entity\EntityStorageInterface;
use Drupal\Core\Entity\EntityTypeInterface;
use Drupal\Core\Field\BaseFieldDefinition;
use Drupal\push_notifications\PushNotificationInterface;
use Drupal\Core\Entity\ContentEntityBase;
use Drupal\user\UserInterface;
class PushNotification extends ContentEntityBase implements PushNotificationInterface {
public static function preCreate(EntityStorageInterface $storage_controller, array &$values) {
parent::preCreate($storage_controller, $values);
$values += array(
'user_id' => \Drupal::currentUser()
->id(),
);
}
public function preSave(EntityStorageInterface $storage) {
parent::preSave($storage);
foreach (array_keys($this
->getTranslationLanguages()) as $langcode) {
$translation = $this
->getTranslation($langcode);
if (!$translation
->getOwner()) {
$translation
->setOwnerId(0);
}
}
}
public function getTitle() {
return $this
->get('title')->value;
}
public function getMessage() {
return $this
->get('message')->value;
}
public function isPushed() {
return (bool) $this
->getEntityKey('status');
}
public function setPushed($pushed) {
$this
->set('status', $pushed ? NODE_PUBLISHED : NODE_NOT_PUBLISHED);
return $this;
}
public function getOwner() {
return $this
->get('user_id')->entity;
}
public function getOwnerId() {
return $this
->get('user_id')->target_id;
}
public function getCreated() {
return $this
->get('created')->value;
}
public function getTokenId() {
return $this
->get('tid')->target_id;
}
public function getCreatedTime($type = 'short') {
return \Drupal::service('date.formatter')
->format($this
->getCreated(), $type);
}
public function setOwnerId($uid) {
$this
->set('user_id', $uid);
return $this;
}
public function setOwner(UserInterface $account) {
$this
->set('user_id', $account
->id());
return $this;
}
public static function baseFieldDefinitions(EntityTypeInterface $entity_type) {
$fields['id'] = BaseFieldDefinition::create('integer')
->setLabel(t('ID'))
->setDescription(t('The ID of the Push Notification entity.'))
->setReadOnly(TRUE);
$fields['uuid'] = BaseFieldDefinition::create('uuid')
->setLabel(t('UUID'))
->setDescription(t('The UUID of the Push Notifications.'))
->setReadOnly(TRUE);
$fields['user_id'] = BaseFieldDefinition::create('entity_reference')
->setLabel(t('Authored By'))
->setDescription(t('The user who created the Push Notification entity'))
->setSetting('target_type', 'user')
->setSetting('handler', 'default')
->setDisplayOptions('view', array(
'label' => 'above',
'type' => 'author',
'weight' => 50,
))
->setDisplayOptions('form', array(
'type' => 'entity_reference_autocomplete',
'settings' => array(
'match_operator' => 'CONTAINS',
'size' => 60,
'placeholder' => '',
),
'weight' => 50,
))
->setDisplayConfigurable('form', TRUE)
->setDisplayConfigurable('view', TRUE)
->setRequired(TRUE);
$fields['title'] = BaseFieldDefinition::create('string')
->setLabel(t('Title'))
->setDescription(t('The title of the Push Notification entity.'))
->setTranslatable(TRUE)
->setRequired(TRUE)
->setSettings(array(
'default_value' => '',
'max_length' => 255,
'text_processing' => 0,
))
->setDisplayOptions('view', array(
'label' => 'above',
'type' => 'string',
'weight' => 1,
))
->setDisplayOptions('form', array(
'type' => 'string_textfield',
'weight' => 1,
))
->setDisplayConfigurable('form', TRUE)
->setDisplayConfigurable('view', TRUE);
$fields['message'] = BaseFieldDefinition::create('string')
->setLabel(t('Message'))
->setDescription(t('The message of the Push Notification entity.'))
->setTranslatable(TRUE)
->setRequired(TRUE)
->setSettings(array(
'default_value' => '',
'max_length' => 255,
'text_processing' => 0,
))
->setDisplayOptions('view', array(
'label' => 'above',
'type' => 'string',
'weight' => 2,
))
->setDisplayOptions('form', array(
'type' => 'string_textfield',
'weight' => 2,
))
->setDisplayConfigurable('form', TRUE)
->setDisplayConfigurable('view', TRUE);
$fields['langcode'] = BaseFieldDefinition::create('language')
->setLabel(t('Language code'))
->setDescription(t('The language code of the entity.'));
$fields['created'] = BaseFieldDefinition::create('created')
->setLabel(t('Created'))
->setDescription(t('The time that the entity was created.'))
->setTranslatable(TRUE);
$fields['status'] = BaseFieldDefinition::create('boolean')
->setLabel(t('Sent status'))
->setDescription(t('A boolean indicating whether the push_notification has been sent.'))
->setTranslatable(TRUE)
->setDefaultValue(FALSE);
return $fields;
}
}