wfNotification.php
3.53 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
<?php
class wfNotification {
const PRIORITY_LOW = 1500;
const PRIORITY_DEFAULT = 1000;
const PRIORITY_HIGH = 500;
protected $_id;
protected $_category;
protected $_priority;
protected $_ctime;
protected $_html;
protected $_links;
public static function notifications($since = 0) {
global $wpdb;
$rawNotifications = $wpdb->get_results($wpdb->prepare("SELECT * FROM {$wpdb->prefix}wfNotifications WHERE `new` = 1 AND `ctime` > %d ORDER BY `priority` ASC, `ctime` DESC", $since), ARRAY_A);
$notifications = array();
foreach ($rawNotifications as $raw) {
$notifications[] = new wfNotification($raw['id'], $raw['priority'], $raw['html'], $raw['category'], $raw['ctime'], json_decode($raw['links'], true), true);
}
return $notifications;
}
public static function getNotificationForID($id) {
global $wpdb;
$rawNotifications = $wpdb->get_results($wpdb->prepare("SELECT * FROM {$wpdb->prefix}wfNotifications WHERE `id` = %s ORDER BY `priority` ASC, `ctime` DESC", $id), ARRAY_A);
if (count($rawNotifications) == 1) {
$raw = $rawNotifications[0];
return new wfNotification($raw['id'], $raw['priority'], $raw['html'], $raw['category'], $raw['ctime'], json_decode($raw['links'], true), true);
}
return null;
}
public static function getNotificationForCategory($category, $requireNew = true) {
global $wpdb;
$rawNotifications = $wpdb->get_results($wpdb->prepare("SELECT * FROM {$wpdb->prefix}wfNotifications WHERE " . ($requireNew ? '`new` = 1 AND ' : '') . "`category` = %s ORDER BY `priority` ASC, `ctime` DESC LIMIT 1", $category), ARRAY_A);
if (count($rawNotifications) == 1) {
$raw = $rawNotifications[0];
return new wfNotification($raw['id'], $raw['priority'], $raw['html'], $raw['category'], $raw['ctime'], json_decode($raw['links'], true), true);
}
return null;
}
public function __construct($id, $priority, $html, $category = null, $ctime = null, $links = null, $memoryOnly = false) {
if ($id === null) {
$id = 'site-' . wfUtils::base32_encode(pack('I', wfConfig::atomicInc('lastNotificationID')));
}
if ($category === null) {
$category = '';
}
if ($ctime === null) {
$ctime = time();
}
if (!is_array($links)) {
$links = array();
}
$this->_id = $id;
$this->_category = $category;
$this->_priority = $priority;
$this->_ctime = $ctime;
$this->_html = $html;
$this->_links = $links;
global $wpdb;
if (!$memoryOnly) {
$linksJSON = json_encode($links);
if (!empty($category)) {
$existing = self::getNotificationForCategory($category);
if ($existing) {
$wpdb->query($wpdb->prepare("UPDATE {$wpdb->prefix}wfNotifications SET priority = %d, ctime = %d, html = %s, links = %s WHERE id = %s", $priority, $ctime, $html, $linksJSON, $existing->id));
return;
}
}
$wpdb->query($wpdb->prepare("INSERT IGNORE INTO {$wpdb->prefix}wfNotifications (id, category, priority, ctime, html, links) VALUES (%s, %s, %d, %d, %s, %s)", $id, $category, $priority, $ctime, $html, $linksJSON));
}
}
public function __get($key){
if ($key == 'id') { return $this->_id; }
else if ($key == 'category') { return $this->_category; }
else if ($key == 'priority') { return $this->_priority; }
else if ($key == 'ctime') { return $this->_ctime; }
else if ($key == 'html') { return $this->_html; }
else if ($key == 'links') { return $this->_links; }
throw new InvalidArgumentException();
}
public function markAsRead() {
global $wpdb;
$wpdb->query($wpdb->prepare("UPDATE {$wpdb->prefix}wfNotifications SET `new` = 0 WHERE `id` = %s", $this->_id));
}
}