FileMaster

Edit File: WooCommerceUsedCouponCode.php

<?php declare(strict_types = 1);

namespace MailPoet\Segments\DynamicSegments\Filters;

if (!defined('ABSPATH')) exit;

use MailPoet\Entities\DynamicSegmentFilterData;
use MailPoet\Entities\DynamicSegmentFilterEntity;
use MailPoet\Segments\DynamicSegments\Exceptions\InvalidFilterException;
use MailPoet\WooCommerce\Helper;
use MailPoetVendor\Doctrine\DBAL\ArrayParameterType;
use MailPoetVendor\Doctrine\DBAL\Query\QueryBuilder;

class WooCommerceUsedCouponCode implements Filter {
  const ACTION = 'usedCouponCode';

const COUPON_CODE_IDS_KEY = 'coupon_code_ids';

/** @var WooFilterHelper */
  private $wooFilterHelper;

/** @var FilterHelper */
  private $filterHelper;

/** @var Helper */
  private $wooHelper;

public function __construct(
    WooFilterHelper $wooFilterHelper,
    Helper $wooHelper,
    FilterHelper $filterHelper
  ) {
    $this->wooFilterHelper = $wooFilterHelper;
    $this->filterHelper = $filterHelper;
    $this->wooHelper = $wooHelper;
  }

public function apply(QueryBuilder $queryBuilder, DynamicSegmentFilterEntity $filter): QueryBuilder {
    $filterData = $filter->getFilterData();
    $this->validateFilterData((array)$filterData->getData());

$operator = $filterData->getParam('operator');

switch ($operator) {
      case DynamicSegmentFilterData::OPERATOR_ALL:
        $this->applyForAllOperator($queryBuilder, $filter);
        break;
      case DynamicSegmentFilterData::OPERATOR_ANY:
        $this->applyForAnyOperator($queryBuilder, $filter);
        break;
      case DynamicSegmentFilterData::OPERATOR_NONE:
        $subQuery = $this->filterHelper->getNewSubscribersQueryBuilder();
        $this->applyForAnyOperator($subQuery, $filter);
        $subscribersTable = $this->filterHelper->getSubscribersTable();
        $queryBuilder->andWhere($queryBuilder->expr()->notIn("$subscribersTable.id", $this->filterHelper->getInterpolatedSQL($subQuery)));
        break;
    }

return $queryBuilder;
  }

private function applyForAnyOperator(QueryBuilder $queryBuilder, DynamicSegmentFilterEntity $filter): void {
    $filterData = $filter->getFilterData();
    $couponIds = (array)$filterData->getParam(self::COUPON_CODE_IDS_KEY);
    $isAllTime = $filterData->getParam('timeframe') === DynamicSegmentFilterData::TIMEFRAME_ALL_TIME;

$orderStatsAlias = $this->wooFilterHelper->applyOrderStatusFilter($queryBuilder);

if (!$isAllTime) {
      /** @var int $days */
      $days = $filterData->getParam('days');
      $date = $this->filterHelper->getDateNDaysAgo(intval($days));
      $dateParam = $this->filterHelper->getUniqueParameterName('date');
      $queryBuilder->andWhere("$orderStatsAlias.date_created >= :$dateParam")
        ->setParameter($dateParam, $date->toDateTimeString());
    }

$queryBuilder->innerJoin(
      $orderStatsAlias,
      $this->filterHelper->getPrefixedTable('wc_order_coupon_lookup'),
      'couponLookup',
      "$orderStatsAlias.order_id = couponLookup.order_id"
    );
    $couponCodeIdsParam = $this->filterHelper->getUniqueParameterName('couponCodeIds');
    $queryBuilder
      ->andWhere("couponLookup.coupon_id IN (:$couponCodeIdsParam)")
      ->setParameter($couponCodeIdsParam, $couponIds, ArrayParameterType::INTEGER);
  }

private function applyForAllOperator(QueryBuilder $queryBuilder, DynamicSegmentFilterEntity $filter): void {
    $this->applyForAnyOperator($queryBuilder, $filter);

$filterData = $filter->getFilterData();
    $couponIds = (array)$filterData->getParam(self::COUPON_CODE_IDS_KEY);
    $queryBuilder->groupBy('inner_subscriber_id')
      ->having("COUNT(DISTINCT couponLookup.coupon_id) = " . count(array_unique($couponIds)));
  }

public function validateFilterData(array $data): void {
    $this->filterHelper->validateDaysPeriodData($data);
    $couponCodeIds = $data[self::COUPON_CODE_IDS_KEY] ?? [];
    if (count($couponCodeIds) === 0) {
      throw new InvalidFilterException('Missing coupon code IDs', InvalidFilterException::MISSING_VALUE);
    }

$operator = $data['operator'] ?? null;

if (
      !in_array($operator, [
      DynamicSegmentFilterData::OPERATOR_ANY,
      DynamicSegmentFilterData::OPERATOR_ALL,
      DynamicSegmentFilterData::OPERATOR_NONE,
      ])
    ) {
      throw new InvalidFilterException('Missing operator', InvalidFilterException::MISSING_OPERATOR);
    }
  }

public function getLookupData(DynamicSegmentFilterData $filterData): array {
    $lookupData = ['coupons' => []];
    if (!$this->wooHelper->isWooCommerceActive()) {
      return $lookupData;
    }
    $couponIds = $filterData->getArrayParam(self::COUPON_CODE_IDS_KEY);
    foreach ($couponIds as $couponId) {
      $couponCode = $this->wooHelper->wcGetCouponCodeById((int)$couponId);
      if (!empty($couponCode)) {
        $lookupData['coupons'][$couponId] = $couponCode;
      }
    }
    return $lookupData;
  }
}

Back