No OneTemporary
Actions

Size

71 KB

Referenced Files

None

Subscribers

None

View Options

	diff --git a/src/applications/metamta/storage/mail/PhabricatorMetaMTAMail.php b/src/applications/metamta/storage/mail/PhabricatorMetaMTAMail.php
	index 5c67c3cf3f..ae845745ac 100644
	--- a/src/applications/metamta/storage/mail/PhabricatorMetaMTAMail.php
	+++ b/src/applications/metamta/storage/mail/PhabricatorMetaMTAMail.php
	@@ -1,689 +1,736 @@
	<?php

	/*
	* Copyright 2012 Facebook, Inc.
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/

	/**
	* See #394445 for an explanation of why this thing even exists.
	*/
	final class PhabricatorMetaMTAMail extends PhabricatorMetaMTADAO {

	const STATUS_QUEUE = 'queued';
	const STATUS_SENT = 'sent';
	const STATUS_FAIL = 'fail';
	const STATUS_VOID = 'void';

	const MAX_RETRIES = 250;
	const RETRY_DELAY = 5;

	protected $parameters;
	protected $status;
	protected $message;
	protected $retryCount;
	protected $nextRetry;
	protected $relatedPHID;

	public function __construct() {

	$this->status = self::STATUS_QUEUE;
	$this->retryCount = 0;
	$this->nextRetry = time();
	$this->parameters = array();

	parent::__construct();
	}

	public function getConfiguration() {
	return array(
	self::CONFIG_SERIALIZATION => array(
	'parameters' => self::SERIALIZATION_JSON,
	),
	) + parent::getConfiguration();
	}

	protected function setParam($param, $value) {
	$this->parameters[$param] = $value;
	return $this;
	}

	protected function getParam($param) {
	return idx($this->parameters, $param);
	}

	/**
	* Set tags (@{class:MetaMTANotificationType} constants) which identify the
	* content of this mail in a general way. These tags are used to allow users
	* to opt out of receiving certain types of mail, like updates when a task's
	* projects change.
	*
	* @param list<const> List of @{class:MetaMTANotificationType} constants.
	* @return this
	*/
	public function setMailTags(array $tags) {
	$this->setParam('mailtags', $tags);
	return $this;
	}

	/**
	* In Gmail, conversations will be broken if you reply to a thread and the
	* server sends back a response without referencing your Message-ID, even if
	* it references a Message-ID earlier in the thread. To avoid this, use the
	* parent email's message ID explicitly if it's available. This overwrites the
	* "In-Reply-To" and "References" headers we would otherwise generate. This
	* needs to be set whenever an action is triggered by an email message. See
	* T251 for more details.
	*
	* @param string The "Message-ID" of the email which precedes this one.
	* @return this
	*/
	public function setParentMessageID($id) {
	$this->setParam('parent-message-id', $id);
	return $this;
	}

	public function getParentMessageID() {
	return $this->getParam('parent-message-id');
	}

	public function getSubject() {
	return $this->getParam('subject');
	}

	public function addTos(array $phids) {
	$phids = array_unique($phids);
	$this->setParam('to', $phids);
	return $this;
	}

	public function addRawTos(array $raw_email) {
	$this->setParam('raw-to', $raw_email);
	return $this;
	}

	public function addCCs(array $phids) {
	$phids = array_unique($phids);
	$this->setParam('cc', $phids);
	return $this;
	}

	public function addHeader($name, $value) {
	$this->parameters['headers'][$name] = $value;
	return $this;
	}

	public function addAttachment(PhabricatorMetaMTAAttachment $attachment) {
	$this->parameters['attachments'][] = $attachment;
	return $this;
	}

	public function getAttachments() {
	return $this->getParam('attachments');
	}

	public function setAttachments(array $attachments) {
	assert_instances_of($attachments, 'PhabricatorMetaMTAAttachment');
	$this->setParam('attachments', $attachments);
	return $this;
	}

	public function setFrom($from) {
	$this->setParam('from', $from);
	return $this;
	}

	public function setReplyTo($reply_to) {
	$this->setParam('reply-to', $reply_to);
	return $this;
	}

	public function setSubject($subject) {
	$this->setParam('subject', $subject);
	return $this;
	}

	public function setVarySubject($subject) {
	$this->setParam('vary-subject', $subject);
	return $this;
	}

	public function setBody($body) {
	$this->setParam('body', $body);
	return $this;
	}

	public function getBody() {
	return $this->getParam('body');
	}

	public function setIsHTML($html) {
	$this->setParam('is-html', $html);
	return $this;
	}

	public function getSimulatedFailureCount() {
	return nonempty($this->getParam('simulated-failures'), 0);
	}

	public function setSimulatedFailureCount($count) {
	$this->setParam('simulated-failures', $count);
	return $this;
	}

	public function getWorkerTaskID() {
	return $this->getParam('worker-task');
	}

	public function setWorkerTaskID($id) {
	$this->setParam('worker-task', $id);
	return $this;
	}

	/**
	* Flag that this is an auto-generated bulk message and should have bulk
	* headers added to it if appropriate. Broadly, this means some flavor of
	* "Precedence: bulk" or similar, but is implementation and configuration
	* dependent.
	*
	* @param bool True if the mail is automated bulk mail.
	* @return this
	*/
	public function setIsBulk($is_bulk) {
	$this->setParam('is-bulk', $is_bulk);
	return $this;
	}

	/**
	* Use this method to set an ID used for message threading. MetaMTA will
	* set appropriate headers (Message-ID, In-Reply-To, References and
	* Thread-Index) based on the capabilities of the underlying mailer.
	*
	* @param string Unique identifier, appropriate for use in a Message-ID,
	* In-Reply-To or References headers.
	* @param bool If true, indicates this is the first message in the thread.
	* @return this
	*/
	public function setThreadID($thread_id, $is_first_message = false) {
	$this->setParam('thread-id', $thread_id);
	$this->setParam('is-first-message', $is_first_message);
	return $this;
	}

	/**
	* Save a newly created mail to the database and attempt to send it
	* immediately if the server is configured for immediate sends. When
	* applications generate new mail they should generally use this method to
	* deliver it. If the server doesn't use immediate sends, this has the same
	* effect as calling save(): the mail will eventually be delivered by the
	* MetaMTA daemon.
	*
	* @return this
	*/
	public function saveAndSend() {
	$ret = null;

	if (PhabricatorEnv::getEnvConfig('metamta.send-immediately')) {
	$ret = $this->sendNow();
	} else {
	$ret = $this->save();
	}

	return $ret;
	}

	protected function didWriteData() {
	parent::didWriteData();

	if (!$this->getWorkerTaskID()) {
	$mailer_task = new PhabricatorWorkerTask();
	$mailer_task->setTaskClass('PhabricatorMetaMTAWorker');
	$mailer_task->setData($this->getID());
	$mailer_task->save();
	$this->setWorkerTaskID($mailer_task->getID());
	$this->save();
	}
	}


	public function buildDefaultMailer() {
	return PhabricatorEnv::newObjectFromConfig('metamta.mail-adapter');
	}

	/**
	* Attempt to deliver an email immediately, in this process.
	*
	* @param bool Try to deliver this email even if it has already been
	* delivered or is in backoff after a failed delivery attempt.
	* @param PhabricatorMailImplementationAdapter Use a specific mail adapter,
	* instead of the default.
	*
	* @return void
	*/
	public function sendNow(
	$force_send = false,
	PhabricatorMailImplementationAdapter $mailer = null) {

	if ($mailer === null) {
	$mailer = $this->buildDefaultMailer();
	}

	if (!$force_send) {
	if ($this->getStatus() != self::STATUS_QUEUE) {
	throw new Exception("Trying to send an already-sent mail!");
	}

	if (time() < $this->getNextRetry()) {
	throw new Exception("Trying to send an email before next retry!");
	}
	}

	try {
	$parameters = $this->parameters;
	$phids = array();
	foreach ($parameters as $key => $value) {
	switch ($key) {
	case 'from':
	case 'to':
	case 'cc':
	if (!is_array($value)) {
	$value = array($value);
	}
	foreach (array_filter($value) as $phid) {
	- $phids[] = $phid;
	+ $type = phid_get_type($phid);
	+ $phids[$phid] = $type;
	}
	break;
	}
	}

	- $handles = id(new PhabricatorObjectHandleData($phids))
	- ->loadHandles();
	+ $this->loadEmailAndNameDataFromPHIDs($phids);

	$exclude = array();

	$params = $this->parameters;
	$default = PhabricatorEnv::getEnvConfig('metamta.default-address');
	if (empty($params['from'])) {
	$mailer->setFrom($default);
	} else {
	$from = $params['from'];

	// If the user has set their preferences to not send them email about
	// things they do, exclude them from being on To or Cc.
	$from_user = id(new PhabricatorUser())->loadOneWhere(
	'phid = %s',
	$from);
	if ($from_user) {
	$pref_key = PhabricatorUserPreferences::PREFERENCE_NO_SELF_MAIL;
	$exclude_self = $from_user
	->loadPreferences()
	->getPreference($pref_key);
	if ($exclude_self) {
	$exclude[$from] = true;
	}
	}

	if (!PhabricatorEnv::getEnvConfig('metamta.can-send-as-user')) {
	- $handle = $handles[$from];
	if (empty($params['reply-to'])) {
	- $params['reply-to'] = $handle->getEmail();
	- $params['reply-to-name'] = $handle->getFullName();
	+ $params['reply-to'] = $phids[$from]['email'];
	+ $params['reply-to-name'] = $phids[$from]['name'];
	}
	$mailer->setFrom(
	$default,
	- $handle->getFullName());
	+ $phids[$from]['name']);
	unset($params['from']);
	}
	}

	$is_first = idx($params, 'is-first-message');
	unset($params['is-first-message']);

	$is_threaded = (bool)idx($params, 'thread-id');

	$reply_to_name = idx($params, 'reply-to-name', '');
	unset($params['reply-to-name']);

	$add_cc = array();
	$add_to = array();

	foreach ($params as $key => $value) {
	switch ($key) {
	case 'from':
	- $mailer->setFrom($handles[$value]->getEmail());
	+ $mailer->setFrom($phids[$from]['email']);
	break;
	case 'reply-to':
	$mailer->addReplyTo($value, $reply_to_name);
	break;
	case 'to':
	- $emails = $this->getDeliverableEmailsFromHandles(
	- $value,
	- $handles,
	- $exclude);
	- if ($emails) {
	- $add_to = array_merge($add_to, $emails);
	+ $to_emails = $this->filterSendable($value, $phids, $exclude);
	+ if ($to_emails) {
	+ $add_to = array_merge($add_to, $to_emails);
	}
	break;
	case 'raw-to':
	$add_to = array_merge($add_to, $value);
	break;
	case 'cc':
	- $emails = $this->getDeliverableEmailsFromHandles(
	- $value,
	- $handles,
	- $exclude);
	- if ($emails) {
	- $add_cc = $emails;
	+ $cc_emails = $this->filterSendable($value, $phids, $exclude);
	+ if ($cc_emails) {
	+ $add_cc = $cc_emails;
	}
	break;
	case 'headers':
	foreach ($value as $header_key => $header_value) {
	// NOTE: If we have \n in a header, SES rejects the email.
	$header_value = str_replace("\n", " ", $header_value);

	$mailer->addHeader($header_key, $header_value);
	}
	break;
	case 'attachments':
	foreach ($value as $attachment) {
	$mailer->addAttachment(
	$attachment->getData(),
	$attachment->getFilename(),
	$attachment->getMimeType()
	);
	}
	break;
	case 'body':
	$mailer->setBody($value);
	break;
	case 'subject':
	// Only try to use preferences if everything is multiplexed, so we
	// get consistent behavior.
	$use_prefs = self::shouldMultiplexAllMail();

	$prefs = null;
	if ($use_prefs) {
	$to = idx($params, 'to', array());
	$user = id(new PhabricatorUser())->loadOneWhere(
	'phid = %s',
	head($to));
	if ($user) {
	$prefs = $user->loadPreferences();
	}
	}

	$alt_subject = idx($params, 'vary-subject');
	if ($alt_subject) {
	$use_subject = PhabricatorEnv::getEnvConfig(
	'metamta.vary-subjects');

	if ($prefs) {
	$use_subject = $prefs->getPreference(
	PhabricatorUserPreferences::PREFERENCE_VARY_SUBJECT,
	$use_subject);
	}

	if ($use_subject) {
	$value = $alt_subject;
	}
	}

	if ($is_threaded) {
	$add_re = PhabricatorEnv::getEnvConfig('metamta.re-prefix');

	if ($prefs) {
	$add_re = $prefs->getPreference(
	PhabricatorUserPreferences::PREFERENCE_RE_PREFIX,
	$add_re);
	}

	if ($add_re) {
	$value = 'Re: '.$value;
	}
	}

	$mailer->setSubject($value);
	break;
	case 'is-html':
	if ($value) {
	$mailer->setIsHTML(true);
	}
	break;
	case 'is-bulk':
	if ($value) {
	if (PhabricatorEnv::getEnvConfig('metamta.precedence-bulk')) {
	$mailer->addHeader('Precedence', 'bulk');
	}
	}
	break;
	case 'thread-id':

	// NOTE: Gmail freaks out about In-Reply-To and References which
	// aren't in the form "<string@domain.tld>"; this is also required
	// by RFC 2822, although some clients are more liberal in what they
	// accept.
	$domain = PhabricatorEnv::getEnvConfig('metamta.domain');
	$value = '<'.$value.'@'.$domain.'>';

	if ($is_first && $mailer->supportsMessageIDHeader()) {
	$mailer->addHeader('Message-ID', $value);
	} else {
	$in_reply_to = $value;
	$references = array($value);
	$parent_id = $this->getParentMessageID();
	if ($parent_id) {
	$in_reply_to = $parent_id;
	// By RFC 2822, the most immediate parent should appear last
	// in the "References" header, so this order is intentional.
	$references[] = $parent_id;
	}
	$references = implode(' ', $references);
	$mailer->addHeader('In-Reply-To', $in_reply_to);
	$mailer->addHeader('References', $references);
	}
	$thread_index = $this->generateThreadIndex($value, $is_first);
	$mailer->addHeader('Thread-Index', $thread_index);
	break;
	case 'mailtags':
	// Handled below.
	break;
	case 'vary-subject':
	// Handled above.
	break;
	default:
	// Just discard.
	}
	}

	$mailer->addHeader('X-Phabricator-Sent-This-Message', 'Yes');
	$mailer->addHeader('X-Mail-Transport-Agent', 'MetaMTA');

	// Some clients respect this to suppress OOF and other auto-responses.
	$mailer->addHeader('X-Auto-Response-Suppress', 'All');

	// If the message has mailtags, filter out any recipients who don't want
	// to receive this type of mail.
	$mailtags = $this->getParam('mailtags');
	if ($mailtags && ($add_to \|\| $add_cc)) {

	$tag_header = array();
	foreach ($mailtags as $mailtag) {
	$tag_header[] = '<'.$mailtag.'>';
	}
	$tag_header = implode(', ', $tag_header);
	$mailer->addHeader('X-Phabricator-Mail-Tags', $tag_header);

	$exclude = array();

	$all_recipients = array_merge(
	array_keys($add_to),
	array_keys($add_cc));

	$all_prefs = id(new PhabricatorUserPreferences())->loadAllWhere(
	'userPHID in (%Ls)',
	$all_recipients);
	$all_prefs = mpull($all_prefs, null, 'getUserPHID');

	foreach ($all_recipients as $recipient) {
	$prefs = idx($all_prefs, $recipient);
	if (!$prefs) {
	continue;
	}

	$user_mailtags = $prefs->getPreference(
	PhabricatorUserPreferences::PREFERENCE_MAILTAGS,
	array());

	// The user must have elected to receive mail for at least one
	// of the mailtags.
	$send = false;
	foreach ($mailtags as $tag) {
	if (idx($user_mailtags, $tag, true)) {
	$send = true;
	break;
	}
	}

	if (!$send) {
	$exclude[$recipient] = true;
	}
	}

	$add_to = array_diff_key($add_to, $exclude);
	$add_cc = array_diff_key($add_cc, $exclude);
	}


	if ($add_to) {
	$mailer->addTos($add_to);
	if ($add_cc) {
	$mailer->addCCs($add_cc);
	}
	} else if ($add_cc) {
	// If we have CC addresses but no "to" address, promote the CCs to
	// "to".
	$mailer->addTos($add_cc);
	} else {
	$this->setStatus(self::STATUS_VOID);
	$this->setMessage(
	"Message has no valid recipients: all To/CC are disabled or ".
	"configured not to receive this mail.");
	return $this->save();
	}

	} catch (Exception $ex) {
	$this->setStatus(self::STATUS_FAIL);
	$this->setMessage($ex->getMessage());
	return $this->save();
	}

	if ($this->getRetryCount() < $this->getSimulatedFailureCount()) {
	$ok = false;
	$error = 'Simulated failure.';
	} else {
	try {
	$ok = $mailer->send();
	$error = null;
	} catch (Exception $ex) {
	$ok = false;
	$error = $ex->getMessage()."\n".$ex->getTraceAsString();
	}
	}

	if (!$ok) {
	$this->setMessage($error);
	if ($this->getRetryCount() > self::MAX_RETRIES) {
	$this->setStatus(self::STATUS_FAIL);
	} else {
	$this->setRetryCount($this->getRetryCount() + 1);
	$next_retry = time() + ($this->getRetryCount() * self::RETRY_DELAY);
	$this->setNextRetry($next_retry);
	}
	} else {
	$this->setStatus(self::STATUS_SENT);
	}

	return $this->save();
	}

	public static function getReadableStatus($status_code) {
	static $readable = array(
	self::STATUS_QUEUE => "Queued for Delivery",
	self::STATUS_FAIL => "Delivery Failed",
	self::STATUS_SENT => "Sent",
	self::STATUS_VOID => "Void",
	);
	$status_code = coalesce($status_code, '?');
	return idx($readable, $status_code, $status_code);
	}

	private function generateThreadIndex($seed, $is_first_mail) {
	// When threading, Outlook ignores the 'References' and 'In-Reply-To'
	// headers that most clients use. Instead, it uses a custom 'Thread-Index'
	// header. The format of this header is something like this (from
	// camel-exchange-folder.c in Evolution Exchange):

	/* A new post to a folder gets a 27-byte-long thread index. (The value
	* is apparently unique but meaningless.) Each reply to a post gets a
	* 32-byte-long thread index whose first 27 bytes are the same as the
	* parent's thread index. Each reply to any of those gets a
	* 37-byte-long thread index, etc. The Thread-Index header contains a
	* base64 representation of this value.
	*/

	// The specific implementation uses a 27-byte header for the first email
	// a recipient receives, and a random 5-byte suffix (32 bytes total)
	// thereafter. This means that all the replies are (incorrectly) siblings,
	// but it would be very difficult to keep track of the entire tree and this
	// gets us reasonable client behavior.

	$base = substr(md5($seed), 0, 27);
	if (!$is_first_mail) {
	// Not totally sure, but it seems like outlook orders replies by
	// thread-index rather than timestamp, so to get these to show up in the
	// right order we use the time as the last 4 bytes.
	$base .= ' '.pack('N', time());
	}

	return base64_encode($base);
	}

	- private function getDeliverableEmailsFromHandles(
	- array $phids,
	- array $handles,
	- array $exclude) {
	- assert_instances_of($handles, 'PhabricatorObjectHandle');
	-
	- $emails = array();
	- foreach ($phids as $phid) {
	- if ($handles[$phid]->isDisabled()) {
	- continue;
	+ private function loadEmailAndNameDataFromPHIDs(array &$phids) {
	+ $users = array();
	+ $mlsts = array();
	+ // first iteration - group by types to do data fetches
	+ foreach ($phids as $phid => $type) {
	+ switch ($type) {
	+ case PhabricatorPHIDConstants::PHID_TYPE_USER:
	+ $users[] = $phid;
	+ break;
	+ case PhabricatorPHIDConstants::PHID_TYPE_MLST:
	+ $mlsts[] = $phid;
	+ break;
	}
	- if (!$handles[$phid]->isComplete()) {
	- continue;
	+ }
	+ $user_emails = array();
	+ if ($users) {
	+ $user_emails = id(new PhabricatorUserEmail())->loadAllWhere(
	+ 'userPHID IN (%Ls) AND isPrimary = 1', $users);
	+ $users = id(new PhabricatorUser())->loadAllWhere(
	+ 'phid IN (%Ls)', $users);
	+ $user_emails = mpull($user_emails, null, 'getUserPHID');
	+ $users = mpull($users, null, 'getPHID');
	+ }
	+ if ($mlsts) {
	+ $mlsts = id(new PhabricatorMetaMTAMailingList())->loadAllWhere(
	+ 'phid IN (%Ls)', $mlsts);
	+ $mlsts = mpull($mlsts, null, 'getPHID');
	+ }
	+
	+ // second iteration - create entries for each phid
	+ $default = PhabricatorEnv::getEnvConfig('metamta.default-address');
	+ foreach ($phids as $phid => &$value) {
	+ $name = '';
	+ $email = $default;
	+ $is_mailable = false;
	+ switch ($value) {
	+ case PhabricatorPHIDConstants::PHID_TYPE_USER:
	+ $user = $users[$phid];
	+ if ($user) {
	+ $name = $user->getFullName();
	+ $is_mailable = !$user->getIsDisabled()
	+ && !$user->getIsSystemAgent();
	+ }
	+ $email = $user_emails[$phid] ?
	+ $user_emails[$phid]->getAddress() :
	+ $default;
	+ break;
	+ case PhabricatorPHIDConstants::PHID_TYPE_MLST:
	+ $mlst = $mlsts[$phid];
	+ if ($mlst) {
	+ $name = $mlst->getName();
	+ $email = $mlst->getEmail();
	+ $is_mailable = true;
	+ }
	+ break;
	}
	+ $value = array(
	+ 'name' => $name,
	+ 'email' => $email,
	+ 'mailable' => $is_mailable,
	+ );
	+ }
	+ }
	+
	+ private function filterSendable($value, $phids, $exclude) {
	+ $result = array();
	+ foreach ($value as $phid) {
	if (isset($exclude[$phid])) {
	continue;
	}
	- $emails[$phid] = $handles[$phid]->getEmail();
	+ if (isset($phids[$phid]) && $phids[$phid]['mailable']) {
	+ $result[$phid] = $phids[$phid]['email'];
	+ }
	}
	-
	- return $emails;
	+ return $result;
	}

	public static function shouldMultiplexAllMail() {
	return PhabricatorEnv::getEnvConfig('metamta.one-mail-per-recipient');
	}

	}
	diff --git a/src/applications/metamta/storage/mail/__init__.php b/src/applications/metamta/storage/mail/__init__.php
	index 7df205fc8c..2470c3d0ac 100644
	--- a/src/applications/metamta/storage/mail/__init__.php
	+++ b/src/applications/metamta/storage/mail/__init__.php
	@@ -1,20 +1,23 @@
	<?php
	/**
	* This file is automatically generated. Lint this module to rebuild it.
	* @generated
	*/



	phutil_require_module('phabricator', 'applications/metamta/storage/base');
	+phutil_require_module('phabricator', 'applications/metamta/storage/mailinglist');
	+phutil_require_module('phabricator', 'applications/people/storage/email');
	phutil_require_module('phabricator', 'applications/people/storage/preferences');
	phutil_require_module('phabricator', 'applications/people/storage/user');
	-phutil_require_module('phabricator', 'applications/phid/handle/data');
	+phutil_require_module('phabricator', 'applications/phid/constants');
	+phutil_require_module('phabricator', 'applications/phid/utils');
	phutil_require_module('phabricator', 'infrastructure/daemon/workers/storage/task');
	phutil_require_module('phabricator', 'infrastructure/env');

	phutil_require_module('phutil', 'utils');


	phutil_require_source('PhabricatorMetaMTAAttachment.php');
	phutil_require_source('PhabricatorMetaMTAMail.php');
	diff --git a/src/applications/people/storage/user/PhabricatorUser.php b/src/applications/people/storage/user/PhabricatorUser.php
	index 0ce125d13e..639b984a33 100644
	--- a/src/applications/people/storage/user/PhabricatorUser.php
	+++ b/src/applications/people/storage/user/PhabricatorUser.php
	@@ -1,596 +1,600 @@
	<?php

	/*
	* Copyright 2012 Facebook, Inc.
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/

	final class PhabricatorUser extends PhabricatorUserDAO {

	const SESSION_TABLE = 'phabricator_session';
	const NAMETOKEN_TABLE = 'user_nametoken';

	protected $phid;
	protected $userName;
	protected $realName;
	protected $sex;
	protected $passwordSalt;
	protected $passwordHash;
	protected $profileImagePHID;
	protected $timezoneIdentifier = '';

	protected $consoleEnabled = 0;
	protected $consoleVisible = 0;
	protected $consoleTab = '';

	protected $conduitCertificate;

	protected $isSystemAgent = 0;
	protected $isAdmin = 0;
	protected $isDisabled = 0;

	private $preferences = null;
	private $primaryEmail;

	protected function readField($field) {
	switch ($field) {
	case 'timezoneIdentifier':
	// If the user hasn't set one, guess the server's time.
	return nonempty(
	$this->timezoneIdentifier,
	date_default_timezone_get());
	// Make sure these return booleans.
	case 'isAdmin':
	return (bool)$this->isAdmin;
	case 'isDisabled':
	return (bool)$this->isDisabled;
	case 'isSystemAgent':
	return (bool)$this->isSystemAgent;
	default:
	return parent::readField($field);
	}
	}

	public function getConfiguration() {
	return array(
	self::CONFIG_AUX_PHID => true,
	self::CONFIG_PARTIAL_OBJECTS => true,
	) + parent::getConfiguration();
	}

	public function generatePHID() {
	return PhabricatorPHID::generateNewPHID(
	PhabricatorPHIDConstants::PHID_TYPE_USER);
	}

	public function setPassword($password) {
	if (!$this->getPHID()) {
	throw new Exception(
	"You can not set a password for an unsaved user because their PHID ".
	"is a salt component in the password hash.");
	}

	if (!strlen($password)) {
	$this->setPasswordHash('');
	} else {
	$this->setPasswordSalt(md5(mt_rand()));
	$hash = $this->hashPassword($password);
	$this->setPasswordHash($hash);
	}
	return $this;
	}

	public function isLoggedIn() {
	return !($this->getPHID() === null);
	}

	public function save() {
	if (!$this->getConduitCertificate()) {
	$this->setConduitCertificate($this->generateConduitCertificate());
	}
	$result = parent::save();

	$this->updateNameTokens();
	PhabricatorSearchUserIndexer::indexUser($this);

	return $result;
	}

	private function generateConduitCertificate() {
	return Filesystem::readRandomCharacters(255);
	}

	public function comparePassword($password) {
	if (!strlen($password)) {
	return false;
	}
	if (!strlen($this->getPasswordHash())) {
	return false;
	}
	$password = $this->hashPassword($password);
	return ($password === $this->getPasswordHash());
	}

	private function hashPassword($password) {
	$password = $this->getUsername().
	$password.
	$this->getPHID().
	$this->getPasswordSalt();
	for ($ii = 0; $ii < 1000; $ii++) {
	$password = md5($password);
	}
	return $password;
	}

	const CSRF_CYCLE_FREQUENCY = 3600;
	const CSRF_TOKEN_LENGTH = 16;

	const EMAIL_CYCLE_FREQUENCY = 86400;
	const EMAIL_TOKEN_LENGTH = 24;

	public function getCSRFToken($offset = 0) {
	return $this->generateToken(
	time() + (self::CSRF_CYCLE_FREQUENCY * $offset),
	self::CSRF_CYCLE_FREQUENCY,
	PhabricatorEnv::getEnvConfig('phabricator.csrf-key'),
	self::CSRF_TOKEN_LENGTH);
	}

	public function validateCSRFToken($token) {

	if (!$this->getPHID()) {
	return true;
	}

	// When the user posts a form, we check that it contains a valid CSRF token.
	// Tokens cycle each hour (every CSRF_CYLCE_FREQUENCY seconds) and we accept
	// either the current token, the next token (users can submit a "future"
	// token if you have two web frontends that have some clock skew) or any of
	// the last 6 tokens. This means that pages are valid for up to 7 hours.
	// There is also some Javascript which periodically refreshes the CSRF
	// tokens on each page, so theoretically pages should be valid indefinitely.
	// However, this code may fail to run (if the user loses their internet
	// connection, or there's a JS problem, or they don't have JS enabled).
	// Choosing the size of the window in which we accept old CSRF tokens is
	// an issue of balancing concerns between security and usability. We could
	// choose a very narrow (e.g., 1-hour) window to reduce vulnerability to
	// attacks using captured CSRF tokens, but it's also more likely that real
	// users will be affected by this, e.g. if they close their laptop for an
	// hour, open it back up, and try to submit a form before the CSRF refresh
	// can kick in. Since the user experience of submitting a form with expired
	// CSRF is often quite bad (you basically lose data, or it's a big pain to
	// recover at least) and I believe we gain little additional protection
	// by keeping the window very short (the overwhelming value here is in
	// preventing blind attacks, and most attacks which can capture CSRF tokens
	// can also just capture authentication information [sniffing networks]
	// or act as the user [xss]) the 7 hour default seems like a reasonable
	// balance. Other major platforms have much longer CSRF token lifetimes,
	// like Rails (session duration) and Django (forever), which suggests this
	// is a reasonable analysis.
	$csrf_window = 6;

	for ($ii = -$csrf_window; $ii <= 1; $ii++) {
	$valid = $this->getCSRFToken($ii);
	if ($token == $valid) {
	return true;
	}
	}

	return false;
	}

	private function generateToken($epoch, $frequency, $key, $len) {
	$time_block = floor($epoch / $frequency);
	$vec = $this->getPHID().$this->getPasswordHash().$key.$time_block;
	return substr(PhabricatorHash::digest($vec), 0, $len);
	}

	/**
	* Issue a new session key to this user. Phabricator supports different
	* types of sessions (like "web" and "conduit") and each session type may
	* have multiple concurrent sessions (this allows a user to be logged in on
	* multiple browsers at the same time, for instance).
	*
	* Note that this method is transport-agnostic and does not set cookies or
	* issue other types of tokens, it ONLY generates a new session key.
	*
	* You can configure the maximum number of concurrent sessions for various
	* session types in the Phabricator configuration.
	*
	* @param string Session type, like "web".
	* @return string Newly generated session key.
	*/
	public function establishSession($session_type) {
	$conn_w = $this->establishConnection('w');

	if (strpos($session_type, '-') !== false) {
	throw new Exception("Session type must not contain hyphen ('-')!");
	}

	// We allow multiple sessions of the same type, so when a caller requests
	// a new session of type "web", we give them the first available session in
	// "web-1", "web-2", ..., "web-N", up to some configurable limit. If none
	// of these sessions is available, we overwrite the oldest session and
	// reissue a new one in its place.

	$session_limit = 1;
	switch ($session_type) {
	case 'web':
	$session_limit = PhabricatorEnv::getEnvConfig('auth.sessions.web');
	break;
	case 'conduit':
	$session_limit = PhabricatorEnv::getEnvConfig('auth.sessions.conduit');
	break;
	default:
	throw new Exception("Unknown session type '{$session_type}'!");
	}

	$session_limit = (int)$session_limit;
	if ($session_limit <= 0) {
	throw new Exception(
	"Session limit for '{$session_type}' must be at least 1!");
	}

	// NOTE: Session establishment is sensitive to race conditions, as when
	// piping `arc` to `arc`:
	//
	// arc export ... \| arc paste ...
	//
	// To avoid this, we overwrite an old session only if it hasn't been
	// re-established since we read it.

	// Consume entropy to generate a new session key, forestalling the eventual
	// heat death of the universe.
	$session_key = Filesystem::readRandomCharacters(40);

	// Load all the currently active sessions.
	$sessions = queryfx_all(
	$conn_w,
	'SELECT type, sessionKey, sessionStart FROM %T
	WHERE userPHID = %s AND type LIKE %>',
	PhabricatorUser::SESSION_TABLE,
	$this->getPHID(),
	$session_type.'-');
	$sessions = ipull($sessions, null, 'type');
	$sessions = isort($sessions, 'sessionStart');

	$existing_sessions = array_keys($sessions);

	// UNGUARDED WRITES: Logging-in users don't have CSRF stuff yet.
	$unguarded = AphrontWriteGuard::beginScopedUnguardedWrites();

	$retries = 0;
	while (true) {


	// Choose which 'type' we'll actually establish, i.e. what number we're
	// going to append to the basic session type. To do this, just check all
	// the numbers sequentially until we find an available session.
	$establish_type = null;
	for ($ii = 1; $ii <= $session_limit; $ii++) {
	$try_type = $session_type.'-'.$ii;
	if (!in_array($try_type, $existing_sessions)) {
	$establish_type = $try_type;
	$expect_key = $session_key;
	$existing_sessions[] = $try_type;

	// Ensure the row exists so we can issue an update below. We don't
	// care if we race here or not.
	queryfx(
	$conn_w,
	'INSERT IGNORE INTO %T (userPHID, type, sessionKey, sessionStart)
	VALUES (%s, %s, %s, 0)',
	self::SESSION_TABLE,
	$this->getPHID(),
	$establish_type,
	$session_key);
	break;
	}
	}

	// If we didn't find an available session, choose the oldest session and
	// overwrite it.
	if (!$establish_type) {
	$oldest = reset($sessions);
	$establish_type = $oldest['type'];
	$expect_key = $oldest['sessionKey'];
	}

	// This is so that we'll only overwrite the session if it hasn't been
	// refreshed since we read it. If it has, the session key will be
	// different and we know we're racing other processes. Whichever one
	// won gets the session, we go back and try again.

	queryfx(
	$conn_w,
	'UPDATE %T SET sessionKey = %s, sessionStart = UNIX_TIMESTAMP()
	WHERE userPHID = %s AND type = %s AND sessionKey = %s',
	self::SESSION_TABLE,
	$session_key,
	$this->getPHID(),
	$establish_type,
	$expect_key);

	if ($conn_w->getAffectedRows()) {
	// The update worked, so the session is valid.
	break;
	} else {
	// We know this just got grabbed, so don't try it again.
	unset($sessions[$establish_type]);
	}

	if (++$retries > $session_limit) {
	throw new Exception("Failed to establish a session!");
	}
	}

	$log = PhabricatorUserLog::newLog(
	$this,
	$this,
	PhabricatorUserLog::ACTION_LOGIN);
	$log->setDetails(
	array(
	'session_type' => $session_type,
	'session_issued' => $establish_type,
	));
	$log->setSession($session_key);
	$log->save();

	return $session_key;
	}

	public function destroySession($session_key) {
	$conn_w = $this->establishConnection('w');
	queryfx(
	$conn_w,
	'DELETE FROM %T WHERE userPHID = %s AND sessionKey = %s',
	self::SESSION_TABLE,
	$this->getPHID(),
	$session_key);
	}

	private function generateEmailToken(
	PhabricatorUserEmail $email,
	$offset = 0) {

	$key = implode(
	'-',
	array(
	PhabricatorEnv::getEnvConfig('phabricator.csrf-key'),
	$this->getPHID(),
	$email->getVerificationCode(),
	));

	return $this->generateToken(
	time() + ($offset * self::EMAIL_CYCLE_FREQUENCY),
	self::EMAIL_CYCLE_FREQUENCY,
	$key,
	self::EMAIL_TOKEN_LENGTH);
	}

	public function validateEmailToken(
	PhabricatorUserEmail $email,
	$token) {
	for ($ii = -1; $ii <= 1; $ii++) {
	$valid = $this->generateEmailToken($email, $ii);
	if ($token == $valid) {
	return true;
	}
	}
	return false;
	}

	public function getEmailLoginURI(PhabricatorUserEmail $email = null) {
	if (!$email) {
	$email = $this->loadPrimaryEmail();
	if (!$email) {
	throw new Exception("User has no primary email!");
	}
	}
	$token = $this->generateEmailToken($email);
	$uri = PhabricatorEnv::getProductionURI('/login/etoken/'.$token.'/');
	$uri = new PhutilURI($uri);
	return $uri->alter('email', $email->getAddress());
	}

	public function loadPrimaryEmailAddress() {
	$email = $this->loadPrimaryEmail();
	if (!$email) {
	throw new Exception("User has no primary email address!");
	}
	return $email->getAddress();
	}

	public function loadPrimaryEmail() {
	return id(new PhabricatorUserEmail())->loadOneWhere(
	'userPHID = %s AND isPrimary = %d',
	$this->getPHID(),
	1);
	}

	public function attachPrimaryEmail(PhabricatorUserEmail $email) {
	$this->primaryEmail = $email;
	return $this;
	}

	public function getPrimaryEmail() {
	if ($this->primaryEmail === null) {
	throw new Exception(
	"Call attachPrimaryEmail() before getPrimaryEmail()!");
	}
	return $this->primaryEmail;
	}

	public function loadPreferences() {
	if ($this->preferences) {
	return $this->preferences;
	}

	$preferences = id(new PhabricatorUserPreferences())->loadOneWhere(
	'userPHID = %s',
	$this->getPHID());

	if (!$preferences) {
	$preferences = new PhabricatorUserPreferences();
	$preferences->setUserPHID($this->getPHID());

	$default_dict = array(
	PhabricatorUserPreferences::PREFERENCE_TITLES => 'glyph',
	PhabricatorUserPreferences::PREFERENCE_EDITOR => '',
	PhabricatorUserPreferences::PREFERENCE_MONOSPACED => '');

	$preferences->setPreferences($default_dict);
	}

	$this->preferences = $preferences;
	return $preferences;
	}

	public function loadEditorLink($path, $line, $callsign) {
	$editor = $this->loadPreferences()->getPreference(
	PhabricatorUserPreferences::PREFERENCE_EDITOR);
	if ($editor) {
	return strtr($editor, array(
	'%%' => '%',
	'%f' => phutil_escape_uri($path),
	'%l' => phutil_escape_uri($line),
	'%r' => phutil_escape_uri($callsign),
	));
	}
	}

	private static function tokenizeName($name) {
	if (function_exists('mb_strtolower')) {
	$name = mb_strtolower($name, 'UTF-8');
	} else {
	$name = strtolower($name);
	}
	$name = trim($name);
	if (!strlen($name)) {
	return array();
	}
	return preg_split('/\s+/', $name);
	}

	/**
	* Populate the nametoken table, which used to fetch typeahead results. When
	* a user types "linc", we want to match "Abraham Lincoln" from on-demand
	* typeahead sources. To do this, we need a separate table of name fragments.
	*/
	public function updateNameTokens() {
	$tokens = array_merge(
	self::tokenizeName($this->getRealName()),
	self::tokenizeName($this->getUserName()));
	$tokens = array_unique($tokens);
	$table = self::NAMETOKEN_TABLE;
	$conn_w = $this->establishConnection('w');

	$sql = array();
	foreach ($tokens as $token) {
	$sql[] = qsprintf(
	$conn_w,
	'(%d, %s)',
	$this->getID(),
	$token);
	}

	queryfx(
	$conn_w,
	'DELETE FROM %T WHERE userID = %d',
	$table,
	$this->getID());
	if ($sql) {
	queryfx(
	$conn_w,
	'INSERT INTO %T (userID, token) VALUES %Q',
	$table,
	implode(', ', $sql));
	}
	}

	public function sendWelcomeEmail(PhabricatorUser $admin) {
	$admin_username = $admin->getUserName();
	$admin_realname = $admin->getRealName();
	$user_username = $this->getUserName();
	$is_serious = PhabricatorEnv::getEnvConfig('phabricator.serious-business');

	$base_uri = PhabricatorEnv::getProductionURI('/');

	$uri = $this->getEmailLoginURI();
	$body = <<<EOBODY
	Welcome to Phabricator!

	{$admin_username} ({$admin_realname}) has created an account for you.

	Username: {$user_username}

	To login to Phabricator, follow this link and set a password:

	{$uri}

	After you have set a password, you can login in the future by going here:

	{$base_uri}

	EOBODY;

	if (!$is_serious) {
	$body .= <<<EOBODY

	Love,
	Phabricator

	EOBODY;
	}

	$mail = id(new PhabricatorMetaMTAMail())
	->addTos(array($this->getPHID()))
	->setSubject('[Phabricator] Welcome to Phabricator')
	->setBody($body)
	->setFrom($admin->getPHID())
	->saveAndSend();
	}

	public static function validateUsername($username) {
	return (bool)preg_match('/^[a-zA-Z0-9]+$/', $username);
	}

	public static function getDefaultProfileImageURI() {
	return celerity_get_resource_uri('/rsrc/image/avatar.png');
	}

	public function loadProfileImageURI() {
	$src_phid = $this->getProfileImagePHID();

	$file = id(new PhabricatorFile())->loadOneWhere('phid = %s', $src_phid);
	if ($file) {
	return $file->getBestURI();
	}

	return self::getDefaultProfileImageURI();
	}

	+ public function getFullName() {
	+ return $this->getUsername().' ('.$this->getRealName().')';
	+ }
	+
	public static function loadOneWithEmailAddress($address) {
	$email = id(new PhabricatorUserEmail())->loadOneWhere(
	'address = %s',
	$address);
	if (!$email) {
	return null;
	}
	return id(new PhabricatorUser())->loadOneWhere(
	'phid = %s',
	$email->getUserPHID());
	}

	}
	diff --git a/src/applications/phid/handle/PhabricatorObjectHandle.php b/src/applications/phid/handle/PhabricatorObjectHandle.php
	index 56fa21e2d4..fb28c92bc9 100644
	--- a/src/applications/phid/handle/PhabricatorObjectHandle.php
	+++ b/src/applications/phid/handle/PhabricatorObjectHandle.php
	@@ -1,235 +1,226 @@
	<?php

	/*
	* Copyright 2012 Facebook, Inc.
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/

	final class PhabricatorObjectHandle {

	private $uri;
	private $phid;
	private $type;
	private $name;
	private $email;
	private $fullName;
	private $imageURI;
	private $timestamp;
	private $alternateID;
	private $status = PhabricatorObjectHandleStatus::STATUS_OPEN;
	private $complete;
	private $disabled;

	public function setURI($uri) {
	$this->uri = $uri;
	return $this;
	}

	public function getURI() {
	return $this->uri;
	}

	public function setPHID($phid) {
	$this->phid = $phid;
	return $this;
	}

	public function getPHID() {
	return $this->phid;
	}

	public function setName($name) {
	$this->name = $name;
	return $this;
	}

	public function getName() {
	return $this->name;
	}

	public function setStatus($status) {
	$this->status = $status;
	return $this;
	}

	public function getStatus() {
	return $this->status;
	}

	public function setFullName($full_name) {
	$this->fullName = $full_name;
	return $this;
	}

	public function getFullName() {
	if ($this->fullName !== null) {
	return $this->fullName;
	}
	return $this->getName();
	}

	public function setType($type) {
	$this->type = $type;
	return $this;
	}

	public function getType() {
	return $this->type;
	}

	- public function setEmail($email) {
	- $this->email = $email;
	- return $this;
	- }
	-
	- public function getEmail() {
	- return $this->email;
	- }
	-
	public function setImageURI($uri) {
	$this->imageURI = $uri;
	return $this;
	}

	public function getImageURI() {
	return $this->imageURI;
	}

	public function setTimestamp($timestamp) {
	$this->timestamp = $timestamp;
	return $this;
	}

	public function getTimestamp() {
	return $this->timestamp;
	}

	public function setAlternateID($alternate_id) {
	$this->alternateID = $alternate_id;
	return $this;
	}

	public function getAlternateID() {
	return $this->alternateID;
	}

	public function getTypeName() {
	static $map = array(
	PhabricatorPHIDConstants::PHID_TYPE_USER => 'User',
	PhabricatorPHIDConstants::PHID_TYPE_TASK => 'Task',
	PhabricatorPHIDConstants::PHID_TYPE_DREV => 'Revision',
	PhabricatorPHIDConstants::PHID_TYPE_CMIT => 'Commit',
	PhabricatorPHIDConstants::PHID_TYPE_WIKI => 'Phriction',
	);

	return idx($map, $this->getType());
	}


	/**
	* Set whether or not the underlying object is complete. See
	* @{method:isComplete} for an explanation of what it means to be complete.
	*
	* @param bool True if the handle represents a complete object.
	* @return this
	*/
	public function setComplete($complete) {
	$this->complete = $complete;
	return $this;
	}


	/**
	* Determine if the handle represents an object which was completely loaded
	* (i.e., the underlying object exists) vs an object which could not be
	* completely loaded (e.g., the type or data for the PHID could not be
	* identified or located).
	*
	* Basically, @{class:PhabricatorObjectHandleData} gives you back a handle for
	* any PHID you give it, but it gives you a complete handle only for valid
	* PHIDs.
	*
	* @return bool True if the handle represents a complete object.
	*/
	public function isComplete() {
	return $this->complete;
	}


	/**
	* Set whether or not the underlying object is disabled. See
	* @{method:isDisabled} for an explanation of what it means to be disabled.
	*
	* @param bool True if the handle represents a disabled object.
	* @return this
	*/
	public function setDisabled($disabled) {
	$this->disabled = $disabled;
	return $this;
	}


	/**
	* Determine if the handle represents an object which has been disabled --
	* for example, disabled users, archived projects, etc. These objects are
	* complete and exist, but should be excluded from some system interactions
	* (for instance, they usually should not appear in typeaheads, and should
	* not have mail/notifications delivered to or about them).
	*
	* @return bool True if the handle represents a disabled object.
	*/
	public function isDisabled() {
	return $this->disabled;
	}


	public function renderLink() {
	$name = $this->getLinkName();
	$class = null;
	$title = null;

	if ($this->status != PhabricatorObjectHandleStatus::STATUS_OPEN) {
	$class .= ' handle-status-'.$this->status;
	$title = $this->status;
	}

	if ($this->disabled) {
	$class .= ' handle-disabled';
	$title = 'disabled'; // Overwrite status.
	}

	return phutil_render_tag(
	'a',
	array(
	'href' => $this->getURI(),
	'class' => $class,
	'title' => $title,
	),
	phutil_escape_html($name));
	}

	public function getLinkName() {
	switch ($this->getType()) {
	case PhabricatorPHIDConstants::PHID_TYPE_USER:
	case PhabricatorPHIDConstants::PHID_TYPE_CMIT:
	$name = $this->getName();
	break;
	default:
	$name = $this->getFullName();
	break;
	}
	return $name;
	}

	}
	diff --git a/src/applications/phid/handle/data/PhabricatorObjectHandleData.php b/src/applications/phid/handle/data/PhabricatorObjectHandleData.php
	index 8fa322a56d..ae87809767 100644
	--- a/src/applications/phid/handle/data/PhabricatorObjectHandleData.php
	+++ b/src/applications/phid/handle/data/PhabricatorObjectHandleData.php
	@@ -1,500 +1,487 @@
	<?php

	/*
	* Copyright 2012 Facebook, Inc.
	*
	* Licensed under the Apache License, Version 2.0 (the "License");
	* you may not use this file except in compliance with the License.
	* You may obtain a copy of the License at
	*
	* http://www.apache.org/licenses/LICENSE-2.0
	*
	* Unless required by applicable law or agreed to in writing, software
	* distributed under the License is distributed on an "AS IS" BASIS,
	* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
	* See the License for the specific language governing permissions and
	* limitations under the License.
	*/

	final class PhabricatorObjectHandleData {

	private $phids;

	public function __construct(array $phids) {
	$this->phids = array_unique($phids);
	}

	public static function loadOneHandle($phid) {
	$handles = id(new PhabricatorObjectHandleData(array($phid)))->loadHandles();
	return $handles[$phid];
	}

	public function loadObjects() {
	$types = array();
	foreach ($this->phids as $phid) {
	$type = phid_get_type($phid);
	$types[$type][] = $phid;
	}

	$objects = array();
	foreach ($types as $type => $phids) {
	switch ($type) {
	case PhabricatorPHIDConstants::PHID_TYPE_USER:
	$user_dao = newv('PhabricatorUser', array());
	$users = $user_dao->loadAllWhere(
	'phid in (%Ls)',
	$phids);
	foreach ($users as $user) {
	$objects[$user->getPHID()] = $user;
	}
	break;
	case PhabricatorPHIDConstants::PHID_TYPE_CMIT:
	$commit_dao = newv('PhabricatorRepositoryCommit', array());
	$commits = $commit_dao->loadAllWhere(
	'phid IN (%Ls)',
	$phids);
	$commit_data = array();
	if ($commits) {
	$data_dao = newv('PhabricatorRepositoryCommitData', array());
	$commit_data = $data_dao->loadAllWhere(
	'commitID IN (%Ld)',
	mpull($commits, 'getID'));
	$commit_data = mpull($commit_data, null, 'getCommitID');
	}
	foreach ($commits as $commit) {
	$data = idx($commit_data, $commit->getID());
	if ($data) {
	$commit->attachCommitData($data);
	$objects[$commit->getPHID()] = $commit;
	} else {
	// If we couldn't load the commit data, just act as though we
	// couldn't load the object at all so we don't load half an object.
	}
	}
	break;
	case PhabricatorPHIDConstants::PHID_TYPE_TASK:
	$task_dao = newv('ManiphestTask', array());
	$tasks = $task_dao->loadAllWhere(
	'phid IN (%Ls)',
	$phids);
	foreach ($tasks as $task) {
	$objects[$task->getPHID()] = $task;
	}
	break;
	case PhabricatorPHIDConstants::PHID_TYPE_DREV:
	$revision_dao = newv('DifferentialRevision', array());
	$revisions = $revision_dao->loadAllWhere(
	'phid IN (%Ls)',
	$phids);
	foreach ($revisions as $revision) {
	$objects[$revision->getPHID()] = $revision;
	}
	break;
	}
	}

	return $objects;
	}

	public function loadHandles() {

	$types = phid_group_by_type($this->phids);

	$handles = array();

	$external_loaders = PhabricatorEnv::getEnvConfig('phid.external-loaders');

	foreach ($types as $type => $phids) {
	switch ($type) {
	case PhabricatorPHIDConstants::PHID_TYPE_MAGIC:
	// Black magic!
	foreach ($phids as $phid) {
	$handle = new PhabricatorObjectHandle();
	$handle->setPHID($phid);
	$handle->setType($type);
	switch ($phid) {
	case ManiphestTaskOwner::OWNER_UP_FOR_GRABS:
	$handle->setName('Up For Grabs');
	$handle->setFullName('upforgrabs (Up For Grabs)');
	$handle->setComplete(true);
	break;
	case ManiphestTaskOwner::PROJECT_NO_PROJECT:
	$handle->setName('No Project');
	$handle->setFullName('noproject (No Project)');
	$handle->setComplete(true);
	break;
	default:
	$handle->setName('Foul Magicks');
	break;
	}
	$handles[$phid] = $handle;
	}
	break;
	case PhabricatorPHIDConstants::PHID_TYPE_USER:
	$class = 'PhabricatorUser';
	PhutilSymbolLoader::loadClass($class);
	$object = newv($class, array());

	$users = $object->loadAllWhere('phid IN (%Ls)', $phids);
	$users = mpull($users, null, 'getPHID');

	$image_phids = mpull($users, 'getProfileImagePHID');
	$image_phids = array_unique(array_filter($image_phids));

	$images = array();
	if ($image_phids) {
	$images = id(new PhabricatorFile())->loadAllWhere(
	'phid IN (%Ls)',
	$image_phids);
	$images = mpull($images, 'getBestURI', 'getPHID');
	}

	- // TODO: This probably should not be part of Handles anymore, only
	- // MetaMTA actually uses it.
	- $emails = id(new PhabricatorUserEmail())->loadAllWhere(
	- 'userPHID IN (%Ls) AND isPrimary = 1',
	- $phids);
	- $emails = mpull($emails, 'getAddress', 'getUserPHID');
	-
	- $statuses = id(new PhabricatorUserStatus())->loadCurrentStatuses(
	- $phids);
	-
	foreach ($phids as $phid) {
	$handle = new PhabricatorObjectHandle();
	$handle->setPHID($phid);
	$handle->setType($type);
	if (empty($users[$phid])) {
	$handle->setName('Unknown User');
	} else {
	$user = $users[$phid];
	$handle->setName($user->getUsername());
	$handle->setURI('/p/'.$user->getUsername().'/');
	- $handle->setEmail(idx($emails, $phid));
	$handle->setFullName(
	$user->getUsername().' ('.$user->getRealName().')');
	$handle->setAlternateID($user->getID());
	$handle->setComplete(true);
	if (isset($statuses[$phid])) {
	$handle->setStatus($statuses[$phid]->getTextStatus());
	}
	- $handle->setDisabled($user->getIsDisabled() \|\|
	- $user->getIsSystemAgent());
	+ $handle->setDisabled($user->getIsDisabled());

	$img_uri = idx($images, $user->getProfileImagePHID());
	if ($img_uri) {
	$handle->setImageURI($img_uri);
	} else {
	$handle->setImageURI(
	PhabricatorUser::getDefaultProfileImageURI());
	}
	}
	$handles[$phid] = $handle;
	}
	break;
	case PhabricatorPHIDConstants::PHID_TYPE_MLST:
	$class = 'PhabricatorMetaMTAMailingList';

	PhutilSymbolLoader::loadClass($class);
	$object = newv($class, array());

	$lists = $object->loadAllWhere('phid IN (%Ls)', $phids);
	$lists = mpull($lists, null, 'getPHID');

	foreach ($phids as $phid) {
	$handle = new PhabricatorObjectHandle();
	$handle->setPHID($phid);
	$handle->setType($type);
	if (empty($lists[$phid])) {
	$handle->setName('Unknown Mailing List');
	} else {
	$list = $lists[$phid];
	- $handle->setEmail($list->getEmail());
	$handle->setName($list->getName());
	$handle->setURI($list->getURI());
	$handle->setFullName($list->getName());
	$handle->setComplete(true);
	}
	$handles[$phid] = $handle;
	}
	break;
	case PhabricatorPHIDConstants::PHID_TYPE_DREV:
	$class = 'DifferentialRevision';
	PhutilSymbolLoader::loadClass($class);
	$object = newv($class, array());

	$revs = $object->loadAllWhere('phid in (%Ls)', $phids);
	$revs = mpull($revs, null, 'getPHID');

	foreach ($phids as $phid) {
	$handle = new PhabricatorObjectHandle();
	$handle->setPHID($phid);
	$handle->setType($type);
	if (empty($revs[$phid])) {
	$handle->setName('Unknown Revision');
	} else {
	$rev = $revs[$phid];
	$handle->setName($rev->getTitle());
	$handle->setURI('/D'.$rev->getID());
	$handle->setFullName('D'.$rev->getID().': '.$rev->getTitle());
	$handle->setComplete(true);

	$status = $rev->getStatus();
	if (($status == ArcanistDifferentialRevisionStatus::CLOSED) \|\|
	($status == ArcanistDifferentialRevisionStatus::ABANDONED)) {
	$closed = PhabricatorObjectHandleStatus::STATUS_CLOSED;
	$handle->setStatus($closed);
	}

	}
	$handles[$phid] = $handle;
	}
	break;
	case PhabricatorPHIDConstants::PHID_TYPE_CMIT:
	$class = 'PhabricatorRepositoryCommit';
	PhutilSymbolLoader::loadClass($class);
	$object = newv($class, array());

	$commits = $object->loadAllWhere('phid in (%Ls)', $phids);
	$commits = mpull($commits, null, 'getPHID');

	$repository_ids = array();
	$callsigns = array();
	if ($commits) {
	$repository_ids = mpull($commits, 'getRepositoryID');
	$repositories = id(new PhabricatorRepository())->loadAllWhere(
	'id in (%Ld)', array_unique($repository_ids));
	$callsigns = mpull($repositories, 'getCallsign');
	}

	foreach ($phids as $phid) {
	$handle = new PhabricatorObjectHandle();
	$handle->setPHID($phid);
	$handle->setType($type);
	if (empty($commits[$phid]) \|\|
	!isset($callsigns[$repository_ids[$phid]])) {
	$handle->setName('Unknown Commit');
	} else {
	$commit = $commits[$phid];
	$callsign = $callsigns[$repository_ids[$phid]];
	$repository = $repositories[$repository_ids[$phid]];
	$commit_identifier = $commit->getCommitIdentifier();

	// In case where the repository for the commit was deleted,
	// we don't have have info about the repository anymore.
	if ($repository) {
	$name = $repository->formatCommitName($commit_identifier);
	$handle->setName($name);
	} else {
	$handle->setName('Commit '.'r'.$callsign.$commit_identifier);
	}

	$handle->setURI('/r'.$callsign.$commit_identifier);
	$handle->setFullName('r'.$callsign.$commit_identifier);
	$handle->setTimestamp($commit->getEpoch());
	$handle->setComplete(true);
	}
	$handles[$phid] = $handle;
	}
	break;
	case PhabricatorPHIDConstants::PHID_TYPE_TASK:
	$class = 'ManiphestTask';
	PhutilSymbolLoader::loadClass($class);
	$object = newv($class, array());

	$tasks = $object->loadAllWhere('phid in (%Ls)', $phids);
	$tasks = mpull($tasks, null, 'getPHID');

	foreach ($phids as $phid) {
	$handle = new PhabricatorObjectHandle();
	$handle->setPHID($phid);
	$handle->setType($type);
	if (empty($tasks[$phid])) {
	$handle->setName('Unknown Revision');
	} else {
	$task = $tasks[$phid];
	$handle->setName($task->getTitle());
	$handle->setURI('/T'.$task->getID());
	$handle->setFullName('T'.$task->getID().': '.$task->getTitle());
	$handle->setComplete(true);
	$handle->setAlternateID($task->getID());
	if ($task->getStatus() != ManiphestTaskStatus::STATUS_OPEN) {
	$closed = PhabricatorObjectHandleStatus::STATUS_CLOSED;
	$handle->setStatus($closed);
	}
	}
	$handles[$phid] = $handle;
	}
	break;
	case PhabricatorPHIDConstants::PHID_TYPE_FILE:
	$class = 'PhabricatorFile';
	PhutilSymbolLoader::loadClass($class);
	$object = newv($class, array());

	$files = $object->loadAllWhere('phid IN (%Ls)', $phids);
	$files = mpull($files, null, 'getPHID');

	foreach ($phids as $phid) {
	$handle = new PhabricatorObjectHandle();
	$handle->setPHID($phid);
	$handle->setType($type);
	if (empty($files[$phid])) {
	$handle->setName('Unknown File');
	} else {
	$file = $files[$phid];
	$handle->setName($file->getName());
	$handle->setURI($file->getBestURI());
	$handle->setComplete(true);
	}
	$handles[$phid] = $handle;
	}
	break;
	case PhabricatorPHIDConstants::PHID_TYPE_PROJ:
	$class = 'PhabricatorProject';
	PhutilSymbolLoader::loadClass($class);
	$object = newv($class, array());

	$projects = $object->loadAllWhere('phid IN (%Ls)', $phids);
	$projects = mpull($projects, null, 'getPHID');

	foreach ($phids as $phid) {
	$handle = new PhabricatorObjectHandle();
	$handle->setPHID($phid);
	$handle->setType($type);
	if (empty($projects[$phid])) {
	$handle->setName('Unknown Project');
	} else {
	$project = $projects[$phid];
	$handle->setName($project->getName());
	$handle->setURI('/project/view/'.$project->getID().'/');
	$handle->setComplete(true);
	}
	$handles[$phid] = $handle;
	}
	break;
	case PhabricatorPHIDConstants::PHID_TYPE_REPO:
	$class = 'PhabricatorRepository';
	PhutilSymbolLoader::loadClass($class);
	$object = newv($class, array());

	$repositories = $object->loadAllWhere('phid in (%Ls)', $phids);
	$repositories = mpull($repositories, null, 'getPHID');

	foreach ($phids as $phid) {
	$handle = new PhabricatorObjectHandle();
	$handle->setPHID($phid);
	$handle->setType($type);
	if (empty($repositories[$phid])) {
	$handle->setName('Unknown Repository');
	} else {
	$repository = $repositories[$phid];
	$handle->setName($repository->getCallsign());
	$handle->setURI('/diffusion/'.$repository->getCallsign().'/');
	$handle->setComplete(true);
	}
	$handles[$phid] = $handle;
	}
	break;
	case PhabricatorPHIDConstants::PHID_TYPE_OPKG:
	$class = 'PhabricatorOwnersPackage';
	PhutilSymbolLoader::loadClass($class);
	$object = newv($class, array());

	$packages = $object->loadAllWhere('phid in (%Ls)', $phids);
	$packages = mpull($packages, null, 'getPHID');

	foreach ($phids as $phid) {
	$handle = new PhabricatorObjectHandle();
	$handle->setPHID($phid);
	$handle->setType($type);
	if (empty($packages[$phid])) {
	$handle->setName('Unknown Package');
	} else {
	$package = $packages[$phid];
	$handle->setName($package->getName());
	$handle->setURI('/owners/package/'.$package->getID().'/');
	$handle->setComplete(true);
	}
	$handles[$phid] = $handle;
	}
	break;
	case PhabricatorPHIDConstants::PHID_TYPE_APRJ:
	$project_dao = newv('PhabricatorRepositoryArcanistProject', array());

	$projects = $project_dao->loadAllWhere(
	'phid IN (%Ls)',
	$phids);
	$projects = mpull($projects, null, 'getPHID');
	foreach ($phids as $phid) {
	$handle = new PhabricatorObjectHandle();
	$handle->setPHID($phid);
	$handle->setType($type);
	if (empty($projects[$phid])) {
	$handle->setName('Unknown Arcanist Project');
	} else {
	$project = $projects[$phid];
	$handle->setName($project->getName());
	$handle->setComplete(true);
	}
	$handles[$phid] = $handle;
	}
	break;
	case PhabricatorPHIDConstants::PHID_TYPE_WIKI:
	$document_dao = newv('PhrictionDocument', array());
	$content_dao = newv('PhrictionContent', array());

	$conn = $document_dao->establishConnection('r');
	$documents = queryfx_all(
	$conn,
	'SELECT * FROM %T document JOIN %T content
	ON document.contentID = content.id
	WHERE document.phid IN (%Ls)',
	$document_dao->getTableName(),
	$content_dao->getTableName(),
	$phids);
	$documents = ipull($documents, null, 'phid');

	foreach ($phids as $phid) {
	$handle = new PhabricatorObjectHandle();
	$handle->setPHID($phid);
	$handle->setType($type);
	if (empty($documents[$phid])) {
	$handle->setName('Unknown Document');
	} else {
	$info = $documents[$phid];
	$handle->setName($info['title']);
	$handle->setURI(PhrictionDocument::getSlugURI($info['slug']));
	$handle->setComplete(true);
	}
	$handles[$phid] = $handle;
	}
	break;
	default:
	$loader = null;
	if (isset($external_loaders[$type])) {
	$loader = $external_loaders[$type];
	} else if (isset($external_loaders['*'])) {
	$loader = $external_loaders['*'];
	}

	if ($loader) {
	PhutilSymbolLoader::loadClass($loader);
	$object = newv($loader, array());
	$handles += $object->loadHandles($phids);
	break;
	}

	foreach ($phids as $phid) {
	$handle = new PhabricatorObjectHandle();
	$handle->setType($type);
	$handle->setPHID($phid);
	$handle->setName('Unknown Object');
	$handle->setFullName('An Unknown Object');
	$handles[$phid] = $handle;
	}
	break;
	}
	}

	return $handles;
	}
	}
	diff --git a/src/applications/phid/handle/data/__init__.php b/src/applications/phid/handle/data/__init__.php
	index d2622bba45..309d3dc001 100644
	--- a/src/applications/phid/handle/data/__init__.php
	+++ b/src/applications/phid/handle/data/__init__.php
	@@ -1,30 +1,29 @@
	<?php
	/**
	* This file is automatically generated. Lint this module to rebuild it.
	* @generated
	*/



	phutil_require_module('arcanist', 'differential/constants/revisionstatus');

	phutil_require_module('phabricator', 'applications/files/storage/file');
	phutil_require_module('phabricator', 'applications/maniphest/constants/owner');
	phutil_require_module('phabricator', 'applications/maniphest/constants/status');
	-phutil_require_module('phabricator', 'applications/people/storage/email');
	phutil_require_module('phabricator', 'applications/people/storage/user');
	phutil_require_module('phabricator', 'applications/people/storage/userstatus');
	phutil_require_module('phabricator', 'applications/phid/constants');
	phutil_require_module('phabricator', 'applications/phid/handle');
	phutil_require_module('phabricator', 'applications/phid/handle/const/status');
	phutil_require_module('phabricator', 'applications/phid/utils');
	phutil_require_module('phabricator', 'applications/phriction/storage/document');
	phutil_require_module('phabricator', 'applications/repository/storage/repository');
	phutil_require_module('phabricator', 'infrastructure/env');
	phutil_require_module('phabricator', 'storage/queryfx');

	phutil_require_module('phutil', 'symbols');
	phutil_require_module('phutil', 'utils');


	phutil_require_source('PhabricatorObjectHandleData.php');

File Metadata

Mime Type: text/x-diff
Expires: Tue, Apr 29, 4:57 PM (23 h, 9 m)
Storage Engine: blob
Storage Format: Raw Data
Storage Handle: 108456
Default Alt Text: (71 KB)

No OneTemporaryActions

View Options

File Metadata

Event Timeline

No OneTemporary
Actions