Add subscriber source

[MAILPOET-1377]
This commit is contained in:
Pavel Dohnal
2018-05-11 11:40:41 +01:00
parent 2627ea865a
commit 1af5802030
4 changed files with 82 additions and 0 deletions

View File

@@ -0,0 +1,37 @@
<?php
namespace MailPoet\Subscribers;
use MailPoet\Models\Subscriber;
class Source {
const FORM = 'form';
const IMPORTED = 'imported';
const ADMINISTRATOR = 'administrator';
const API = 'api';
const WORDPRESS_USER = 'wordpress_user';
const UNKNOWN = 'unknown';
private static $allowed_sources = array(
Source::FORM,
Source::IMPORTED,
Source::ADMINISTRATOR,
Source::API,
Source::WORDPRESS_USER,
Source::UNKNOWN,
);
static function setSource(Subscriber $subscriber, $source) {
if((isset($subscriber->source)) && ($subscriber->source !== Source::UNKNOWN)) {
// we don't want to override source
return $subscriber;
}
if(!in_array($source, Source::$allowed_sources)) {
throw new \InvalidArgumentException('Invalid source "' . $source . '""');
}
$subscriber->set('source', $source);
return $subscriber;
}
}