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
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
<?php
class EchoUserLocator {
/**
* Return all users watching the event title.
*
* The echo job queue must be enabled to prevent timeouts submitting to
* heavily watched pages when this is used.
*
* @param EchoEvent $event
* @param int $batchSize
* @return User[]
*/
public static function locateUsersWatchingTitle( EchoEvent $event, $batchSize = 500 ) {
$title = $event->getTitle();
if ( !$title ) {
return [];
}
$it = new BatchRowIterator(
wfGetDB( DB_REPLICA, 'watchlist' ),
/* $table = */ 'watchlist',
/* $primaryKeys = */ [ 'wl_user' ],
$batchSize
);
$it->addConditions( [
'wl_namespace' => $title->getNamespace(),
'wl_title' => $title->getDBkey(),
] );
// flatten the result into a stream of rows
$it = new RecursiveIteratorIterator( $it );
// add callback to convert user id to user objects
$it = new EchoCallbackIterator( $it, function ( $row ) {
return User::newFromId( $row->wl_user );
} );
return $it;
}
/**
* If the event occured on the talk page of a registered
* user return that user.
*
* @param EchoEvent $event
* @return User[]
*/
public static function locateTalkPageOwner( EchoEvent $event ) {
$title = $event->getTitle();
if ( !$title || $title->getNamespace() !== NS_USER_TALK ) {
return [];
}
$user = User::newFromName( $title->getDBkey() );
if ( $user && !$user->isAnon() ) {
return [ $user->getId() => $user ];
} else {
return [];
}
}
/**
* Return the event agent
*
* @param EchoEvent $event
* @return User[]
*/
public static function locateEventAgent( EchoEvent $event ) {
$agent = $event->getAgent();
if ( $agent && !$agent->isAnon() ) {
return [ $agent->getId() => $agent ];
} else {
return [];
}
}
/**
* Return the user that created the first revision of the
* associated title.
*
* @param EchoEvent $event
* @return User[]
*/
public static function locateArticleCreator( EchoEvent $event ) {
$title = $event->getTitle();
if ( !$title || $title->getArticleID() <= 0 ) {
return [];
}
$dbr = wfGetDB( DB_REPLICA );
$revQuery = Revision::getQueryInfo();
$res = $dbr->selectRow(
$revQuery['tables'],
[ 'rev_user' => $revQuery['fields']['rev_user'] ],
[ 'rev_page' => $title->getArticleID() ],
__METHOD__,
[ 'LIMIT' => 1, 'ORDER BY' => 'rev_timestamp, rev_id' ],
$revQuery['joins']
);
if ( !$res || !$res->rev_user ) {
return [];
}
$user = User::newFromId( $res->rev_user );
if ( $user ) {
return [ $user->getId() => $user ];
} else {
return [];
}
}
/**
* Fetch user ids from the event extra data. Requires additional
* parameter. Example $wgEchoNotifications parameter:
*
* 'user-locator' => array( array( 'event-extra', 'mentions' ) ),
*
* The above will look in the 'mentions' parameter for a user id or
* array of user ids. It will return all these users as notification
* targets.
*
* @param EchoEvent $event
* @param string[] $keys one or more keys to check for user ids
* @return User[]
*/
public static function locateFromEventExtra( EchoEvent $event, array $keys ) {
$users = [];
foreach ( $keys as $key ) {
$userIds = $event->getExtraParam( $key );
if ( !$userIds ) {
continue;
} elseif ( !is_array( $userIds ) ) {
$userIds = [ $userIds ];
}
foreach ( $userIds as $userId ) {
// we shouldn't receive User instances, but allow
// it for backward compatability
if ( $userId instanceof User ) {
if ( $userId->isAnon() ) {
continue;
}
$user = $userId;
} else {
$user = User::newFromId( $userId );
}
$users[$user->getId()] = $user;
}
}
return $users;
}
}
|
GitWeb