Homec4science
Diffusion Phabricator (feature/better-testing) Browse

phab/src/applications/phrequent/controller/PhrequentTrackController.php
d0128afa2908feature/better-testing

Last Change
Raw File

PhrequentTrackController.php
View Options

<?php
final class PhrequentTrackController
extends PhrequentController {
private $verb;
private $phid;
public function willProcessRequest(array $data) {
$this->phid = $data['phid'];
$this->verb = $data['verb'];
}
public function processRequest() {
$request = $this->getRequest();
$user = $request->getUser();
$phid = $this->phid;
$handle = id(new PhabricatorHandleQuery())
->setViewer($user)
->withPHIDs(array($phid))
->executeOne();
if (!$this->isStartingTracking() &&
!$this->isStoppingTracking()) {
throw new Exception('Unrecognized verb: '.$this->verb);
}
switch ($this->verb) {
case 'start':
$button_text = pht('Start Tracking');
$title_text = pht('Start Tracking Time');
$inner_text = pht('What time did you start working?');
$action_text = pht('Start Timer');
$label_text = pht('Start Time');
break;
case 'stop':
$button_text = pht('Stop Tracking');
$title_text = pht('Stop Tracking Time');
$inner_text = pht('What time did you stop working?');
$action_text = pht('Stop Timer');
$label_text = pht('Stop Time');
break;
}
$epoch_control = id(new AphrontFormDateControl())
->setUser($user)
->setName('epoch')
->setLabel($action_text)
->setValue(time());
$err = array();
if ($request->isDialogFormPost()) {
$timestamp = $epoch_control->readValueFromRequest($request);
$note = $request->getStr('note');
if (!$epoch_control->isValid() || $timestamp > time()) {
$err[] = pht('Invalid date, please enter a valid non-future date');
}
if (!$err) {
if ($this->isStartingTracking()) {
$this->startTracking($user, $this->phid, $timestamp);
} else if ($this->isStoppingTracking()) {
$this->stopTracking($user, $this->phid, $timestamp, $note);
}
return id(new AphrontRedirectResponse());
}
}
$dialog = $this->newDialog()
->setTitle($title_text)
->setWidth(AphrontDialogView::WIDTH_FORM);
if ($err) {
$dialog->setErrors($err);
}
$form = new PHUIFormLayoutView();
$form
->appendChild(hsprintf(
'<p>%s</p><br />', $inner_text));
$form->appendChild($epoch_control);
if ($this->isStoppingTracking()) {
$form
->appendChild(
id(new AphrontFormTextControl())
->setLabel(pht('Note'))
->setName('note'));
}
$dialog->appendChild($form);
$dialog->addCancelButton($handle->getURI());
$dialog->addSubmitButton($action_text);
return $dialog;
}
private function isStartingTracking() {
return $this->verb === 'start';
}
private function isStoppingTracking() {
return $this->verb === 'stop';
}
private function startTracking($user, $phid, $timestamp) {
$usertime = new PhrequentUserTime();
$usertime->setDateStarted($timestamp);
$usertime->setUserPHID($user->getPHID());
$usertime->setObjectPHID($phid);
$usertime->save();
}
private function stopTracking($user, $phid, $timestamp, $note) {
if (!PhrequentUserTimeQuery::isUserTrackingObject($user, $phid)) {
// Don't do anything, it's not being tracked.
return;
}
$usertime_dao = new PhrequentUserTime();
$conn = $usertime_dao->establishConnection('r');
queryfx(
$conn,
'UPDATE %T usertime '.
'SET usertime.dateEnded = %d, '.
'usertime.note = %s '.
'WHERE usertime.userPHID = %s '.
'AND usertime.objectPHID = %s '.
'AND usertime.dateEnded IS NULL '.
'ORDER BY usertime.dateStarted, usertime.id DESC '.
'LIMIT 1',
$usertime_dao->getTableName(),
$timestamp,
$note,
$user->getPHID(),
$phid);
}
}
c4science · Help