<?php

namespace MailPoet\Subscribers\ImportExport\Export;

use MailPoet\Models\Subscriber;
use MailPoetVendor\Idiorm\ORM;

/**
 * Gets batches of subscribers for export.
 */
abstract class SubscribersGetter {

  protected $segmentsIds;
  protected $batchSize;
  protected $offset;
  protected $finished;

  public function __construct($segmentsIds, $batchSize) {
    $this->segmentsIds = $segmentsIds;
    $this->batchSize = $batchSize;
    $this->reset();
  }

  public function reset() {
    $this->offset = 0;
    $this->finished = false;
  }

  /**
   * Initialize the query by selecting fields and ignoring trashed subscribers.
   *
   * @return ORM
   */
  protected function select() {
    return Subscriber::selectMany(
      'first_name',
      'last_name',
      'email',
      'subscribed_ip',
      [
        'global_status' => Subscriber::$_table . '.status',
      ]
    )
    ->filter('filterWithCustomFieldsForExport')
    ->groupBy(Subscriber::$_table . '.id')
    ->whereNull(Subscriber::$_table . '.deleted_at');
  }

  /**
   * Filters the subscribers query based on the segments, offset and batch size.
   *
   * @param  ORM $subscribers
   * @return array
   */
  abstract protected function filter($subscribers);

  /**
   * Gets the next batch of subscribers or `false` if no more!
   */
  public function get() {
    if ($this->finished) {
      return false;
    }

    $subscribers = $this->select();
    $subscribers = $this->filter($subscribers);

    $this->offset += $this->batchSize;

    if (count($subscribers) < $this->batchSize) {
      $this->finished = true;
    }

    return $subscribers;
  }
}