Agreement.php 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139
  1. <?php
  2. /**
  3. * Copyright © Magento, Inc. All rights reserved.
  4. * See COPYING.txt for license details.
  5. */
  6. namespace Magento\CheckoutAgreements\Model\ResourceModel;
  7. /**
  8. * Resource Model for Checkout Agreement
  9. */
  10. class Agreement extends \Magento\Framework\Model\ResourceModel\Db\AbstractDb
  11. {
  12. /**
  13. * @var \Magento\Framework\Filter\FilterManager
  14. */
  15. protected $filterManager;
  16. /**
  17. * @param \Magento\Framework\Model\ResourceModel\Db\Context $context
  18. * @param \Magento\Framework\Filter\FilterManager $filterManager
  19. * @param string $connectionName
  20. * @codeCoverageIgnore
  21. */
  22. public function __construct(
  23. \Magento\Framework\Model\ResourceModel\Db\Context $context,
  24. \Magento\Framework\Filter\FilterManager $filterManager,
  25. $connectionName = null
  26. ) {
  27. $this->filterManager = $filterManager;
  28. parent::__construct($context, $connectionName);
  29. }
  30. /**
  31. * Model initialization
  32. *
  33. * @return void
  34. * @codeCoverageIgnore
  35. */
  36. protected function _construct()
  37. {
  38. $this->_init('checkout_agreement', 'agreement_id');
  39. }
  40. /**
  41. * Method to run before save
  42. *
  43. * @param \Magento\Framework\Model\AbstractModel $object
  44. * @return $this
  45. */
  46. protected function _beforeSave(\Magento\Framework\Model\AbstractModel $object)
  47. {
  48. // format height
  49. $height = $object->getContentHeight();
  50. $height = $this->filterManager->stripTags($height);
  51. if (!$height) {
  52. $height = '';
  53. }
  54. if ($height && preg_match('/[0-9]$/', $height)) {
  55. $height .= 'px';
  56. }
  57. $object->setContentHeight($height);
  58. return parent::_beforeSave($object);
  59. }
  60. /**
  61. * Method to run after save
  62. *
  63. * @param \Magento\Framework\Model\AbstractModel $object
  64. * @return $this
  65. */
  66. protected function _afterSave(\Magento\Framework\Model\AbstractModel $object)
  67. {
  68. $this->getConnection()->delete(
  69. $this->getTable('checkout_agreement_store'),
  70. ['agreement_id = ?' => $object->getId()]
  71. );
  72. foreach ((array)$object->getData('stores') as $storeId) {
  73. $storeArray = [
  74. 'agreement_id' => $object->getId(),
  75. 'store_id' => $storeId
  76. ];
  77. $this->getConnection()->insert($this->getTable('checkout_agreement_store'), $storeArray);
  78. }
  79. return parent::_afterSave($object);
  80. }
  81. /**
  82. * Method to run after load
  83. *
  84. * @param \Magento\Framework\Model\AbstractModel $object
  85. * @return $this
  86. */
  87. protected function _afterLoad(\Magento\Framework\Model\AbstractModel $object)
  88. {
  89. $select = $this->getConnection()
  90. ->select()
  91. ->from($this->getTable('checkout_agreement_store'), ['store_id'])
  92. ->where('agreement_id = :agreement_id');
  93. $stores = $this->getConnection()->fetchCol($select, [':agreement_id' => $object->getId()]);
  94. if ($stores) {
  95. $object->setData('stores', $stores);
  96. }
  97. return parent::_afterLoad($object);
  98. }
  99. /**
  100. * Get load select
  101. *
  102. * @param string $field
  103. * @param mixed $value
  104. * @param \Magento\Framework\Model\AbstractModel $object
  105. * @return \Magento\Framework\DB\Select
  106. */
  107. protected function _getLoadSelect($field, $value, $object)
  108. {
  109. $select = parent::_getLoadSelect($field, $value, $object);
  110. if ($object->getStoreId()) {
  111. $select->join(
  112. ['cps' => $this->getTable('checkout_agreement_store')],
  113. $this->getMainTable() . '.agreement_id = cps.agreement_id'
  114. )->where(
  115. 'is_active=1'
  116. )->where(
  117. 'cps.store_id IN (0, ?)',
  118. $object->getStoreId()
  119. )->order(
  120. 'store_id DESC'
  121. )->limit(
  122. 1
  123. );
  124. }
  125. return $select;
  126. }
  127. }