- Fixes custom field shortcode matching logic

This commit is contained in:
Vlad
2016-06-01 09:48:01 -04:00
parent a208104fc8
commit 61987a204e

View File

@ -1,8 +1,8 @@
<?php <?php
namespace MailPoet\Newsletter\Shortcodes\Categories; namespace MailPoet\Newsletter\Shortcodes\Categories;
use MailPoet\Models\SubscriberCustomField;
use MailPoet\Models\Subscriber as SubscriberModel; use MailPoet\Models\Subscriber as SubscriberModel;
use MailPoet\Models\SubscriberCustomField;
require_once(ABSPATH . 'wp-includes/pluggable.php'); require_once(ABSPATH . 'wp-includes/pluggable.php');
@ -31,17 +31,19 @@ class Subscriber {
return $default_value; return $default_value;
break; break;
case 'count': case 'count':
return SubscriberModel::filter('subscribed')->count(); return SubscriberModel::filter('subscribed')
->count();
break; break;
case preg_match('/cf_(\d+)/', $action, $custom_field) ? true : false: default:
if(empty($subscriber['id'])) return false; if(preg_match('/cf_(\d+)/', $action, $custom_field) &&
!empty($subscriber['id'])
) {
$custom_field = SubscriberCustomField $custom_field = SubscriberCustomField
::where('subscriber_id', $subscriber['id']) ::where('subscriber_id', $subscriber['id'])
->where('custom_field_id', $custom_field[1]) ->where('custom_field_id', $custom_field[1])
->findOne(); ->findOne();
return ($custom_field) ? $custom_field->value : false; return ($custom_field) ? $custom_field->value : false;
break; }
default:
return false; return false;
break; break;
} }