piratepoet/lib/Subscription/Pages.php

<?php
namespace MailPoet\Subscription;

use \MailPoet\Models\Subscriber;
use \MailPoet\Models\CustomField;
use \MailPoet\Models\Setting;
use \MailPoet\Models\Segment;
use \MailPoet\Util\Helpers;

class Pages {
  const DEMO_EMAIL = 'demo@mailpoet.com';

  function __construct() {

  }

  function init() {
    if(isset($_GET['mailpoet_page'])) {
      add_filter('wp_title', array($this,'setWindowTitle'));
      add_filter('the_title', array($this,'setPageTitle'));
      add_filter('the_content', array($this,'setPageContent'));
    }
  }

  function setWindowTitle() {
    // TODO
  }

  function setPageTitle($title = null) {
    $subscriber = $this->getSubscriber();

    switch($this->getAction()) {
      case 'confirm':
        return $this->getConfirmTitle($subscriber);
      break;

      case 'edit':
        return $this->getEditTitle($subscriber);
      break;

      case 'unsubscribe':
        if($subscriber !== false && $subscriber->email !== self::DEMO_EMAIL) {
          if($subscriber->status !== Subscriber::STATUS_UNSUBSCRIBED) {
            $subscriber->status = Subscriber::STATUS_UNSUBSCRIBED;
            $subscriber->save();
          }
        }
        return $this->getUnsubscribeTitle();
      break;
    }
    return $title;
  }

  function setPageContent($page_content = '[mailpoet_page]') {
    $content = '';
    $subscriber = $this->getSubscriber();

    switch($this->getAction()) {
      case 'confirm':
        $content = $this->getConfirmContent($subscriber);
      break;
      case 'edit':
        $content = $this->getEditContent($subscriber);
      break;
      case 'unsubscribe':
        $content = $this->getUnsubscribeContent($subscriber);
      break;
    }
    return str_replace('[mailpoet_page]', $content, $page_content);
  }

  private function getConfirmTitle($subscriber) {
    if($subscriber === false) {
      $title = __('Your confirmation link expired, please subscribe again.');
    } else {
      if($subscriber->email === self::DEMO_EMAIL) {
        $segment_names = array('demo 1', 'demo 2');
      } else {
        if($subscriber->status !== Subscriber::STATUS_SUBSCRIBED) {
          $subscriber->status = Subscriber::STATUS_SUBSCRIBED;
          $subscriber->save();
        }

        $segments = $subscriber->segments()->findMany();

        $segment_names = array_map(function($segment) {
          return $segment->name;
        }, $segments);
      }

      if(empty($segment_names)) {
        $title = __("You've subscribed!");
      } else {
        $title = sprintf(
          __("You've subscribed to: %s"),
          join(', ', $segment_names)
        );
      }
    }
    return $title;
  }

  private function getEditTitle($subscriber) {
    if($subscriber !== false) {
      return sprintf(
        __('Edit your subscriber profile: %s'),
        $subscriber->email
      );
    }
  }

  private function getUnsubscribeTitle() {
    return __("You've unsubscribed!");
  }


  private function getConfirmContent($subscriber) {
    if($subscriber !== false) {
      return __("Yup, we've added you to our list. You'll hear from us shortly.");
    }
  }

  private function getEditContent($subscriber) {
    if($subscriber !== false) {
      $subscriber = $subscriber
      ->withCustomFields()
      ->withSubscriptions();

      $custom_fields = array_map(function($custom_field) use($subscriber) {
        $custom_field->id = 'cf_'.$custom_field->id;
        $custom_field = $custom_field->asArray();
        $custom_field['params']['value'] = $subscriber->{$custom_field['id']};
        return $custom_field;
      }, CustomField::findMany());

      $segment_ids = Setting::getValue('subscription.segments', array());
      if(!empty($segment_ids)) {
        $segments = Segment::getPublic()
          ->whereIn('id', $segment_ids)
          ->findMany();
      } else {
        $segments = Segment::getPublic()->findMany();
      }

      $subscribed_segment_ids = Helpers::arrayColumn(
        $subscriber->subscriptions, 'id'
      );

      $segments = array_map(function($segment) use($subscribed_segment_ids) {
        return array(
          'id' => $segment->id,
          'name' => $segment->name,
          'is_checked' => in_array($segment->id, $subscribed_segment_ids)
        );
      }, $segments);

      $fields = array(
        array(
          'id' => 'email',
          'type' => 'text',
          'params' => array(
            'label' => __('Email'),
            'required' => true,
            'value' => $subscriber->email
          )
        ),
        array(
          'id' => 'first_name',
          'type' => 'text',
          'params' => array(
            'label' => __('First name'),
            'value' => $subscriber->first_name
          )
        ),
        array(
          'id' => 'last_name',
          'type' => 'text',
          'params' => array(
            'label' => __('Last name'),
            'value' => $subscriber->last_name
          )
        ),
        array(
          'id' => 'status',
          'type' => 'select',
          'params' => array(
            'label' => __('Status'),
            'values' => array(
              array(
                'value' => array(
                  Subscriber::STATUS_SUBSCRIBED => __('Subscribed')
                ),
                'is_checked' => (
                  $subscriber->status === Subscriber::STATUS_SUBSCRIBED
                )
              ),
              array(
                'value' => array(
                  Subscriber::STATUS_UNSUBSCRIBED => __('Unsubscribed')
                ),
                'is_checked' => (
                  $subscriber->status === Subscriber::STATUS_UNSUBSCRIBED
                )
              ),
              array(
                'value' => array(
                  Subscriber::STATUS_UNCONFIRMED => __('Unconfirmed')
                ),
                'is_checked' => (
                  $subscriber->status === Subscriber::STATUS_UNCONFIRMED
                )
              )
            )
          )
        )
      );

      $form = array_merge(
        $fields,
        $custom_fields,
        array(
          array(
            'id' => 'segment',
            'type' => 'segment',
            'params' => array(
              'label' => __('Your lists'),
              'values' => $segments
            )
          ),
          array(
            'id' => 'submit',
            'type' => 'submit',
            'params' => array(
              'label' => __('Subscribe!')
            )
          )
        )
      );

      $content = \MailPoet\Form\Renderer::renderBlocks($form);
    }
  }

  private function getUnsubscribeContent($subscriber) {
    $content = '<p>'.__("Great, you'll never hear from us again!").'</p>';
    if($subscriber !== false) {
      $content .= '<p><strong>'.
        str_replace(
          array('[link]', '[/link]'),
          array('<a href="'.$subscriber->getConfirmationUrl().'">', '</a>'),
          __('You made a mistake? [link]Undo unsubscribe.[/link]')
        ).
      '</strong></p>';
    }
    return $content;
  }

  private function getSubscriber() {
    $token = (isset($_GET['mailpoet_token']))
      ? $_GET['mailpoet_token']
      : null;
    $email = (isset($_GET['mailpoet_email']))
      ? $_GET['mailpoet_email']
      : null;

    if(empty($token) || empty($email) || $email === self::DEMO_EMAIL) {
      $subscriber = Subscriber::create();
      $subscriber->email = self::DEMO_EMAIL;
      return $subscriber;
    }

    if(md5(AUTH_KEY.$email) === $token) {
      $subscriber = Subscriber::findOne($email);
      if($subscriber !== false) {
        return $subscriber;
      }
    }
    return false;
  }

  private function getAction() {
    return (isset($_GET['mailpoet_action']))
      ? $_GET['mailpoet_action']
      : null;
  }
}