relatedPHIDs = $phids; return $this; } public function setSubscribedPHIDs(array $phids) { $this->subscribedPHIDs = $phids; return $this; } public function setPrimaryObjectPHID($phid) { $this->primaryObjectPHID = $phid; return $this; } public function setStoryType($story_type) { $this->storyType = $story_type; return $this; } public function setStoryData(array $data) { $this->storyData = $data; return $this; } public function setStoryTime($time) { $this->storyTime = $time; return $this; } public function setStoryAuthorPHID($phid) { $this->storyAuthorPHID = $phid; return $this; } public function publish() { if (!$this->relatedPHIDs) { throw new Exception("There are no PHIDs related to this story!"); } if (!$this->storyType) { throw new Exception("Call setStoryType() before publishing!"); } $chrono_key = $this->generateChronologicalKey(); $story = new PhabricatorFeedStoryData(); $story->setStoryType($this->storyType); $story->setStoryData($this->storyData); $story->setAuthorPHID($this->storyAuthorPHID); $story->setChronologicalKey($chrono_key); $story->save(); $ref = new PhabricatorFeedStoryReference(); $sql = array(); $conn = $ref->establishConnection('w'); foreach (array_unique($this->relatedPHIDs) as $phid) { $sql[] = qsprintf( $conn, '(%s, %s)', $phid, $chrono_key); } queryfx( $conn, 'INSERT INTO %T (objectPHID, chronologicalKey) VALUES %Q', $ref->getTableName(), implode(', ', $sql)); if (PhabricatorEnv::getEnvConfig('notification.enabled')) { $this->insertNotifications($chrono_key); $this->sendNotification($chrono_key); } return $story; } private function insertNotifications($chrono_key) { if (!$this->subscribedPHIDs) { return; } if (!$this->primaryObjectPHID) { throw new Exception( "You must call setPrimaryObjectPHID() if you setSubscribedPHIDs()!"); } $notif = new PhabricatorFeedStoryNotification(); $sql = array(); $conn = $notif->establishConnection('w'); foreach (array_unique($this->subscribedPHIDs) as $user_phid) { $sql[] = qsprintf( $conn, '(%s, %s, %s, %d)', $this->primaryObjectPHID, $user_phid, $chrono_key, 0); } queryfx( $conn, 'INSERT INTO %T (primaryObjectPHID, userPHID, chronologicalKey, hasViewed) VALUES %Q', $notif->getTableName(), implode(', ', $sql)); } private function sendNotification($chrono_key) { $aphlict_url = 'http://127.0.0.1:22281/push?'; //TODO: make configurable $future = new HTTPFuture($aphlict_url, array( "key" => (string)$chrono_key, // TODO: fix. \r\n appears to be appended to the final value here. // this is a temporary workaround "nothing" => "", )); $future->setMethod('POST'); $future->resolve(); } /** * We generate a unique chronological key for each story type because we want * to be able to page through the stream with a cursor (i.e., select stories * after ID = X) so we can efficiently perform filtering after selecting data, * and multiple stories with the same ID make this cumbersome without putting * a bunch of logic in the client. We could use the primary key, but that * would prevent publishing stories which happened in the past. Since it's * potentially useful to do that (e.g., if you're importing another data * source) build a unique key for each story which has chronological ordering. * * @return string A unique, time-ordered key which identifies the story. */ private function generateChronologicalKey() { // Use the epoch timestamp for the upper 32 bits of the key. Default to // the current time if the story doesn't have an explicit timestamp. $time = nonempty($this->storyTime, time()); // Generate a random number for the lower 32 bits of the key. $rand = head(unpack('L', Filesystem::readRandomBytes(4))); // On 32-bit machines, we have to get creative. if (PHP_INT_SIZE < 8) { // We're on a 32-bit machine. if (function_exists('bcadd')) { // Try to use the 'bc' extension. return bcadd(bcmul($time, bcpow(2, 32)), $rand); } else { // Do the math in MySQL. TODO: If we formalize a bc dependency, get // rid of this. $conn_r = id(new PhabricatorFeedStoryData())->establishConnection('r'); $result = queryfx_one( $conn_r, 'SELECT (%d << 32) + %d as N', $time, $rand); return $result['N']; } } else { // This is a 64 bit machine, so we can just do the math. return ($time << 32) + $rand; } } }