3 * StatusNet - the distributed open-source microblogging tool
4 * Copyright (C) 2010 StatusNet, Inc.
6 * This program is free software: you can redistribute it and/or modify
7 * it under the terms of the GNU Affero General Public License as published by
8 * the Free Software Foundation, either version 3 of the License, or
9 * (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU Affero General Public License for more details.
16 * You should have received a copy of the GNU Affero General Public License
17 * along with this program. If not, see <http://www.gnu.org/licenses/>.
21 * Class for activity streams
23 * Includes faves, notices, and subscriptions.
25 * We extend atomusernoticefeed since it does some nice setup for us.
28 class UserActivityStream extends AtomUserNoticeFeed
30 public $activities = array();
32 const OUTPUT_STRING = 1;
34 public $outputMode = self::OUTPUT_STRING;
39 * @param boolean $indent
40 * @param boolean $outputMode: UserActivityStream::OUTPUT_STRING to return a string,
41 * or UserActivityStream::OUTPUT_RAW to go to raw output.
42 * Raw output mode will attempt to stream, keeping less
43 * data in memory but will leave $this->activities incomplete.
45 function __construct($user, $indent = true, $outputMode = UserActivityStream::OUTPUT_STRING)
47 parent::__construct($user, null, $indent);
49 $this->outputMode = $outputMode;
50 if ($this->outputMode == self::OUTPUT_STRING) {
51 // String buffering? Grab all the notices now.
52 $notices = $this->getNotices();
53 } elseif ($this->outputMode == self::OUTPUT_RAW) {
54 // Raw output... need to restructure from the stringer init.
55 $this->xw = new XMLWriter();
56 $this->xw->openURI('php://output');
57 if(is_null($indent)) {
58 $indent = common_config('site', 'indent');
60 $this->xw->setIndent($indent);
62 // We'll fetch notices later.
65 throw new Exception('Invalid outputMode provided to ' . __METHOD__);
68 // Assume that everything but notices is feasible
69 // to pull at once and work with in memory...
71 $subscriptions = $this->getSubscriptions();
72 $subscribers = $this->getSubscribers();
73 $groups = $this->getGroups();
74 $faves = $this->getFaves();
76 $objs = array_merge($subscriptions, $subscribers, $groups, $faves, $notices);
78 $subscriptions = null;
83 unset($subscriptions);
88 // Sort by create date
90 usort($objs, 'UserActivityStream::compareObject');
92 // We'll keep these around for later, and interleave them into
93 // the output stream with the user's notices.
99 * Interleave the pre-sorted subs/groups/faves with the user's
100 * notices, all in reverse chron order.
102 function renderEntries($format=Feed::ATOM, $handle=null)
105 foreach ($this->objs as $obj) {
107 $act = $obj->asActivity();
108 } catch (Exception $e) {
109 common_log(LOG_ERR, $e->getMessage());
115 if ($this->outputMode == self::OUTPUT_RAW && $start != $end) {
116 // In raw mode, we haven't pre-fetched notices.
117 // Grab the chunks of notices between other activities.
119 $notices = $this->getNoticesBetween($start, $end);
120 foreach ($notices as $noticeAct) {
122 $nact = $noticeAct->asActivity();
123 if ($format == Feed::ATOM) {
124 $nact->outputTo($this, false, false);
126 fwrite($handle, json_encode($nact->asArray()));
128 } catch (Exception $e) {
129 common_log(LOG_ERR, $e->getMessage());
135 } catch (Exception $e) {
136 common_log(LOG_ERR, $e->getMessage());
144 if ($format == Feed::ATOM) {
145 // Only show the author sub-element if it's different from default user
146 $act->outputTo($this, false, ($act->actor->id != $this->user->uri));
148 fwrite($handle, json_encode($act->asArray()));
150 } catch (Exception $e) {
151 common_log(LOG_ERR, $e->getMessage());
160 if ($this->outputMode == self::OUTPUT_RAW) {
161 // Grab anything after the last pre-sorted activity.
163 $notices = $this->getNoticesBetween(0, $end);
164 foreach ($notices as $noticeAct) {
166 $nact = $noticeAct->asActivity();
167 if ($format == Feed::ATOM) {
168 $nact->outputTo($this, false, false);
170 fwrite($handle, json_encode($nact->asArray()));
172 } catch (Exception $e) {
173 common_log(LOG_ERR, $e->getMessage());
177 } catch (Exception $e) {
178 common_log(LOG_ERR, $e->getMessage());
183 function compareObject($a, $b)
185 $ac = strtotime((empty($a->created)) ? $a->modified : $a->created);
186 $bc = strtotime((empty($b->created)) ? $b->modified : $b->created);
188 return (($ac == $bc) ? 0 : (($ac < $bc) ? 1 : -1));
191 function getSubscriptions()
195 $sub = new Subscription();
197 $sub->subscriber = $this->user->id;
200 while ($sub->fetch()) {
201 if ($sub->subscribed != $this->user->id) {
202 $subs[] = clone($sub);
210 function getSubscribers()
214 $sub = new Subscription();
216 $sub->subscribed = $this->user->id;
219 while ($sub->fetch()) {
220 if ($sub->subscriber != $this->user->id) {
221 $subs[] = clone($sub);
235 $fave->user_id = $this->user->id;
238 while ($fave->fetch()) {
239 $faves[] = clone($fave);
248 * @param int $start unix timestamp for earliest
249 * @param int $end unix timestamp for latest
250 * @return array of Notice objects
252 function getNoticesBetween($start=0, $end=0)
256 $notice = new Notice();
258 $notice->profile_id = $this->user->id;
261 $tsstart = common_sql_date($start);
262 $notice->whereAdd("created >= '$tsstart'");
265 $tsend = common_sql_date($end);
266 $notice->whereAdd("created < '$tsend'");
269 $notice->orderBy('created DESC');
271 if ($notice->find()) {
272 while ($notice->fetch()) {
273 $notices[] = clone($notice);
280 function getNotices()
282 return $this->getNoticesBetween();
289 $gm = new Group_member();
291 $gm->profile_id = $this->user->id;
294 while ($gm->fetch()) {
295 $groups[] = clone($gm);
302 function writeJSON($handle)
304 require_once INSTALLDIR.'/lib/activitystreamjsondocument.php';
305 fwrite($handle, '{"items": [');
306 $this->renderEntries(Feed::JSON, $handle);
307 fwrite($handle, ']');