piratepoet/lib/Models/Newsletter.php

<?php
namespace MailPoet\Models;

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

class Newsletter extends Model {
  public static $_table = MP_NEWSLETTERS_TABLE;

  function __construct() {
    parent::__construct();
  }

  function segments() {
    return $this->has_many_through(
      __NAMESPACE__.'\Segment',
      __NAMESPACE__.'\NewsletterSegment',
      'newsletter_id',
      'segment_id'
    );
  }

  static function search($orm, $search = '') {
    return $orm->where_like('subject', '%' . $search . '%');
  }

  static function filters() {
    $segments = Segment::orderByAsc('name')->findMany();
    $segment_list = array();
    $segment_list[] = array(
      'label' => __('All lists'),
      'value' => ''
    );
    foreach($segments as $segment) {
      $newsletters_count = $segment->newsletters()->count();
      if($newsletters_count > 0) {

        $segment_list[] = array(
          'label' => sprintf('%s (%d)', $segment->name, $newsletters_count),
          'value' => $segment->id()
        );
      }
    }

    $filters = array(
      array(
        'name' => 'segment',
        'options' => $segment_list
      )
    );

    return $filters;
  }

  static function filterBy($orm, $filters = null) {
    if(empty($filters)) {
      return $orm;
    }

    foreach($filters as $filter) {
      if($filter['name'] === 'segment') {

        $segment = Segment::findOne($filter['value']);
        if($segment !== false) {
          $orm = $orm
          ->select('model.*')
          ->select('newsletter_segment.id', 'newsletter_segment_id')
          ->join(
            MP_NEWSLETTER_SEGMENT_TABLE,
            'model.id = newsletter_segment.newsletter_id',
            'newsletter_segment'
          )
          ->where('newsletter_segment.segment_id', (int)$filter['value']);
        }
      }
    }
    return $orm;
  }

  static function groups() {
    return array(
      array(
        'name' => 'all',
        'label' => __('All'),
        'count' => Newsletter::count()
      )
    );
  }

  static function group($orm, $group = null) {
  }

  static function createOrUpdate($data = array()) {
    $newsletter = false;

    if(isset($data['id']) && (int) $data['id'] > 0) {
      $newsletter = self::findOne((int) $data['id']);
    }

    if($newsletter === false) {
      $newsletter = self::create();
      $newsletter->hydrate($data);
    } else {
      unset($data['id']);
      $newsletter->set($data);
    }

    $saved = $newsletter->save();

    if($saved === true) {
      return true;
    } else {
      $errors = $newsletter->getValidationErrors();
      if(!empty($errors)) {
        return $errors;
      }
    }
    return false;
  }

  static function trash($listing) {
    return $listing->getSelection()
      ->deleteMany();
  }
}