View source
<?php
namespace Drupal\commerce_product_bundle\Entity;
use Drupal\commerce_price\Price;
use Drupal\commerce_product\Entity\ProductInterface;
use Drupal\commerce_product\Entity\ProductVariationInterface;
use Drupal\Core\Entity\EntityStorageInterface;
use Drupal\Core\Entity\ContentEntityBase;
use Drupal\Core\Entity\EntityChangedTrait;
use Drupal\Core\Entity\EntityTypeInterface;
use Drupal\Core\Field\BaseFieldDefinition;
use Drupal\user\UserInterface;
class ProductBundleItem extends ContentEntityBase implements BundleItemInterface {
use EntityChangedTrait;
protected $bundle;
protected $activeQuantity;
protected $currentVariation;
public function getOwner() {
return $this
->get('uid')->entity;
}
public function setOwnerId($uid) {
$this
->set('uid', $uid);
return $this;
}
public function getOwnerId() {
return $this
->get('uid')->target_id;
}
public function getBundle() {
return $this
->get('bundle_id')->entity;
}
public function getBundleId() {
return $this
->get('bundle_id')->target_id;
}
public function getTitle() {
return $this
->get('title')->value;
}
public function setTitle($title) {
$this
->set('title', $title);
return $this;
}
public function getCreatedTime() {
return $this
->get('created')->value;
}
public function setCreatedTime($timestamp) {
$this
->set('created', $timestamp);
return $this;
}
public function setOwner(UserInterface $account) {
$this
->set('uid', $account
->id());
return $this;
}
public function isRequired() {
return $this
->get('required')->value;
}
public function setRequired($required) {
$this
->set('required', (bool) $required);
return $this;
}
public function getUnitPrice() {
if (!$this
->get('unit_price')
->isEmpty()) {
return $this
->get('unit_price')
->first()
->toPrice();
}
$variation = $this
->getCurrentVariation();
return $variation ? $variation
->getPrice() : NULL;
}
public function hasUnitPrice() {
return !$this
->get('unit_price')
->isEmpty();
}
public function setUnitPrice(Price $unit_price) {
$this
->set('unit_price', $unit_price);
return $this;
}
public function getQuantity() {
if (isset($this->activeQuantity)) {
return $this->activeQuantity;
}
return $this
->getMinimumQuantity();
}
public function setQuantity($quantity) {
$this->activeQuantity = (double) $quantity;
return $this;
}
public function setMinimumQuantity($minimum_quantity) {
$this
->set('min_quantity', (double) $minimum_quantity);
return $this;
}
public function setMaximumQuantity($maximum_quantity) {
$this
->set('max_quantity', (double) $maximum_quantity);
return $this;
}
public function hasProduct() {
return !$this
->get('product')
->isEmpty();
}
public function getProduct() {
if ($this
->hasProduct()) {
return $this
->get('product')
->referencedEntities()[0];
}
}
public function getProductId() {
return $this
->getProduct()
->id();
}
public function setProduct(ProductInterface $product) {
if ($this
->hasProduct()) {
$currentProductId = $this
->getProduct()
->id();
$newProductId = $product
->id();
if ($currentProductId !== $newProductId) {
$this
->set('variations', NULL);
}
}
$this
->set('product', $product);
return $this;
}
public function hasVariations() {
return !$this
->get('variations')
->isEmpty();
}
public function addVariation(ProductVariationInterface $variation) {
if ($this
->hasProduct() && $this
->hasVariations() && !$this
->hasVariation($variation)) {
$this
->assertSameProduct([
$variation,
]);
$this
->get('variations')
->appendItem($variation);
}
return $this;
}
protected function hasVariation(ProductVariationInterface $variation) {
return $this
->getVariationIndex($variation) !== FALSE;
}
protected function getVariationIndex(ProductVariationInterface $variation) {
$result = array_search($variation
->id(), $this
->getVariationIds() ?: []);
return $result;
}
public function getVariationIds() {
$variations = $this
->getVariations();
if (empty($variations)) {
return NULL;
}
return array_map(function ($variation) {
return $variation
->id();
}, $this
->getVariations());
}
public function removeVariation(ProductVariationInterface $variation) {
$index = $this
->getVariationIndex($variation);
if ($index !== FALSE) {
$this
->get('variations')
->offsetUnset($index);
}
return $this;
}
public function getDefaultVariation() {
if (!$this
->hasProduct()) {
return NULL;
}
foreach ($this
->getVariations() as $variation) {
if ($variation
->isPublished()) {
return $variation;
}
}
}
public function getVariations() {
if (!$this
->hasProduct()) {
return NULL;
}
$variations = $this
->get('variations')
->referencedEntities();
if (empty($variations)) {
return $this
->getEnabledVariations();
}
return $this
->ensureTranslations($variations);
}
protected function getEnabledVariations() {
$variationStorage = $this
->entityTypeManager()
->getStorage('commerce_product_variation');
return $variationStorage
->loadEnabled($this
->getProduct());
}
public function setVariations(array $variations) {
if (empty($variations)) {
return $this;
}
if ($this
->get('product')
->isEmpty()) {
$this
->setProduct($variations[0]
->getProduct());
}
$this
->assertSameProduct($variations);
$this
->set('variations', $variations);
return $this;
}
protected function assertSameProduct(array $variations) {
foreach ($variations as $variation) {
$shouldBeOfType = $this
->getProduct()
->id();
$isType = $variation
->getProductId();
if ($shouldBeOfType !== $isType) {
throw new \InvalidArgumentException('All variations of a bundle item must be from the same product.');
}
}
}
public function getCurrentVariation() {
return $this->currentVariation ?: $this
->getDefaultVariation();
}
public function setCurrentVariation(ProductVariationInterface $variation) {
$this
->assertSameProduct([
$variation,
]);
if (!$this
->hasVariation($variation)) {
throw new \InvalidArgumentException('Variation is not part of this product bundle.');
}
$this->currentVariation = $variation;
return $this;
}
public function getMinimumQuantity() {
return $this
->get('min_quantity')->value;
}
public function getMaximumQuantity() {
return $this
->get('max_quantity')->value;
}
public static function preCreate(EntityStorageInterface $storage_controller, array &$values) {
parent::preCreate($storage_controller, $values);
$values += [
'uid' => \Drupal::currentUser()
->id(),
];
}
public static function baseFieldDefinitions(EntityTypeInterface $entity_type) {
$fields = parent::baseFieldDefinitions($entity_type);
$fields['uid'] = BaseFieldDefinition::create('entity_reference')
->setLabel(t('Author'))
->setDescription(t('The user ID of author of the product bundle item entity.'))
->setSetting('target_type', 'user')
->setSetting('handler', 'default')
->setDefaultValueCallback('Drupal\\commerce_product_bundle\\Entity\\ProductBundleItem::getCurrentUserId')
->setDisplayOptions('view', [
'title' => 'hidden',
'type' => 'author',
'weight' => 0,
])
->setDisplayOptions('form', [
'type' => 'entity_reference_autocomplete',
'weight' => 5,
'settings' => [
'match_operator' => 'CONTAINS',
'size' => '60',
'autocomplete_type' => 'tags',
'match_limit' => 10,
'placeholder' => '',
],
])
->setDisplayConfigurable('form', TRUE)
->setDisplayConfigurable('view', TRUE);
$fields['title'] = BaseFieldDefinition::create('string')
->setLabel(t('Title'))
->setDescription(t('The title of the product bundle item entity.'))
->setRequired(TRUE)
->setTranslatable(TRUE)
->setSettings([
'max_length' => 128,
'text_processing' => 0,
])
->setDefaultValue('')
->setDisplayOptions('view', [
'label' => 'hidden',
'type' => 'string',
'weight' => -4,
])
->setDisplayOptions('form', [
'type' => 'string_textfield',
'weight' => -4,
])
->setDisplayConfigurable('form', TRUE)
->setDisplayConfigurable('view', TRUE);
$fields['required'] = BaseFieldDefinition::create('boolean')
->setLabel(t('Required?'))
->setDescription(t('A boolean indicating whether the product bundle item is required or optional.'))
->setSettings([
'on_label' => t('Yes, required'),
'off_label' => t('No, optional'),
])
->setDisplayOptions('form', [
'type' => 'boolean_checkbox',
'settings' => [
'display_label' => TRUE,
],
'weight' => 20,
])
->setDisplayConfigurable('form', TRUE)
->setDefaultValue(TRUE);
$fields['bundle_id'] = BaseFieldDefinition::create('entity_reference')
->setLabel(t('Product bundle'))
->setDescription(t('The parent product bundle.'))
->setSetting('target_type', 'commerce_product_bundle')
->setReadOnly(TRUE)
->setDisplayConfigurable('view', TRUE);
$fields['unit_price'] = BaseFieldDefinition::create('commerce_price')
->setLabel(t('Unit price'))
->setDescription(t('The unit price, if overridden, of the variation selected from this bundle item.'))
->setDisplayOptions('view', [
'label' => 'above',
'type' => 'commerce_price_default',
'weight' => 0,
])
->setDisplayOptions('form', [
'type' => 'commerce_price_default',
'weight' => 0,
])
->setDisplayConfigurable('form', TRUE)
->setDisplayConfigurable('view', TRUE);
$fields['product'] = BaseFieldDefinition::create('entity_reference')
->setLabel(t('Product'))
->setDescription(t('Reference to a product.'))
->setSetting('target_type', 'commerce_product')
->setSetting('handler', 'default')
->setCardinality(1)
->setRequired(TRUE)
->setDisplayOptions('view', [
'label' => 'hidden',
'type' => 'entity_reference_label',
'weight' => 0,
])
->setDisplayOptions('form', [
'type' => 'entity_reference_autocomplete',
'weight' => 5,
'settings' => [
'match_operator' => 'CONTAINS',
'size' => '60',
'autocomplete_type' => 'tags',
'match_limit' => 10,
'placeholder' => '',
],
])
->setDisplayConfigurable('form', TRUE)
->setDisplayConfigurable('view', TRUE);
$fields['min_quantity'] = BaseFieldDefinition::create('decimal')
->setLabel(t('Minimum Quantity'))
->setDescription(t('The minimum quantity.'))
->setSetting('unsigned', TRUE)
->setSetting('min', 0)
->setRequired(TRUE)
->setDefaultValue(1)
->setDisplayOptions('form', [
'type' => 'number',
'weight' => 1,
])
->setDisplayConfigurable('form', TRUE)
->setDisplayConfigurable('view', TRUE)
->addPropertyConstraints('value', [
'Range' => [
'min' => 0,
],
]);
$fields['max_quantity'] = BaseFieldDefinition::create('decimal')
->setLabel(t('Maximum Quantity'))
->setDescription(t('The maximum quantity.'))
->setSetting('unsigned', TRUE)
->setSetting('min', 1)
->setRequired(TRUE)
->setDefaultValue(1)
->setDisplayOptions('form', [
'type' => 'number',
'weight' => 1,
])
->setDisplayConfigurable('form', TRUE)
->setDisplayConfigurable('view', TRUE);
$fields['created'] = BaseFieldDefinition::create('created')
->setLabel(t('Created'))
->setDescription(t('The time that the entity was created.'));
$fields['changed'] = BaseFieldDefinition::create('changed')
->setLabel(t('Changed'))
->setDescription(t('The time that the entity was last edited.'));
return $fields;
}
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);
}
}
}
protected function ensureTranslations(array $entities) {
$langcode = $this
->language()
->getId();
foreach ($entities as $index => $entity) {
if ($entity
->hasTranslation($langcode)) {
$entities[$index] = $entity
->getTranslation($langcode);
}
}
return $entities;
}
public static function getCurrentUserId() {
return [
\Drupal::currentUser()
->id(),
];
}
}