First stab at a badges typeahead

Summary: Ref T10702

Test Plan: Open a user profile, attempt to award an archived or previously awarded badge, badges dialog should provide a typeahead, and the suggestions should offer details about whether a badge is archived or already awarded.

Reviewers: epriestley, #blessed_reviewers

Reviewed By: epriestley, #blessed_reviewers

Subscribers: Korvin

Maniphest Tasks: T10702

Differential Revision: https://secure.phabricator.com/D15665

src/__phutil_library_map__.php

@@ -1880,6 +1880,7 @@ phutil_register_library_map(array(
     'PhabricatorBadgesController' => 'applications/badges/controller/PhabricatorBadgesController.php',
     'PhabricatorBadgesCreateCapability' => 'applications/badges/capability/PhabricatorBadgesCreateCapability.php',
     'PhabricatorBadgesDAO' => 'applications/badges/storage/PhabricatorBadgesDAO.php',
+    'PhabricatorBadgesDatasource' => 'applications/badges/typeahead/PhabricatorBadgesDatasource.php',
     'PhabricatorBadgesDefaultEditCapability' => 'applications/badges/capability/PhabricatorBadgesDefaultEditCapability.php',
     'PhabricatorBadgesEditConduitAPIMethod' => 'applications/badges/conduit/PhabricatorBadgesEditConduitAPIMethod.php',
     'PhabricatorBadgesEditController' => 'applications/badges/controller/PhabricatorBadgesEditController.php',
@@ -6274,6 +6275,7 @@ phutil_register_library_map(array(
     'PhabricatorBadgesController' => 'PhabricatorController',
     'PhabricatorBadgesCreateCapability' => 'PhabricatorPolicyCapability',
     'PhabricatorBadgesDAO' => 'PhabricatorLiskDAO',
+    'PhabricatorBadgesDatasource' => 'PhabricatorTypeaheadDatasource',
     'PhabricatorBadgesDefaultEditCapability' => 'PhabricatorPolicyCapability',
     'PhabricatorBadgesEditConduitAPIMethod' => 'PhabricatorEditEngineAPIMethod',
     'PhabricatorBadgesEditController' => 'PhabricatorBadgesController',

src/applications/badges/controller/PhabricatorBadgesAwardController.php

@@ -18,23 +18,24 @@ final class PhabricatorBadgesAwardController
     $view_uri = '/p/'.$user->getUsername();
 
     if ($request->isFormPost()) {
-      $xactions = array();
+      $badge_phids = $request->getArr('badgePHIDs');
-      $badge_phid = $request->getStr('badgePHID');
+      $badges = id(new PhabricatorBadgesQuery())
-      $badge = id(new PhabricatorBadgesQuery())
         ->setViewer($viewer)
-        ->withPHIDs(array($badge_phid))
+        ->withPHIDs($badge_phids)
         ->needRecipients(true)
         ->requireCapabilities(
           array(
             PhabricatorPolicyCapability::CAN_EDIT,
             PhabricatorPolicyCapability::CAN_VIEW,
           ))
-        ->executeOne();
+        ->execute();
-      if (!$badge) {
+      if (!$badges) {
         return new Aphront404Response();
       }
       $award_phids = array($user->getPHID());
 
+      foreach ($badges as $badge) {
+        $xactions = array();
         $xactions[] = id(new PhabricatorBadgesTransaction())
           ->setTransactionType(PhabricatorBadgesTransaction::TYPE_AWARD)
           ->setNewValue($award_phids);
@@ -45,33 +46,24 @@ final class PhabricatorBadgesAwardController
           ->setContinueOnNoEffect(true)
           ->setContinueOnMissingFields(true)
           ->applyTransactions($badge, $xactions);
+      }
 
       return id(new AphrontRedirectResponse())
         ->setURI($view_uri);
     }
 
-    $badges = id(new PhabricatorBadgesQuery())
-      ->setViewer($viewer)
-      ->withStatuses(array(
-        PhabricatorBadgesBadge::STATUS_ACTIVE,
-      ))
-      ->requireCapabilities(
-        array(
-          PhabricatorPolicyCapability::CAN_VIEW,
-          PhabricatorPolicyCapability::CAN_EDIT,
-        ))
-      ->execute();
-
-    $options = mpull($badges, 'getName', 'getPHID');
-    asort($options);
-
     $form = id(new AphrontFormView())
       ->setUser($viewer)
-      ->appendChild(
+      ->appendControl(
-        id(new AphrontFormSelectControl())
+        id(new AphrontFormTokenizerControl())
           ->setLabel(pht('Badge'))
-          ->setName('badgePHID')
+          ->setName('badgePHIDs')
-          ->setOptions($options));
+          ->setDatasource(
+            id(new PhabricatorBadgesDatasource())
+              ->setParameters(
+                array(
+                  'recipientPHID' => $user->getPHID(),
+                  ))));
 
     $dialog = $this->newDialog()
       ->setTitle(pht('Grant Badge'))

src/applications/badges/typeahead/PhabricatorBadgesDatasource.php

@@ -0,0 +1,69 @@
+<?php
+
+final class PhabricatorBadgesDatasource
+  extends PhabricatorTypeaheadDatasource {
+
+  public function getBrowseTitle() {
+    return pht('Browse Badges');
+  }
+
+  public function getPlaceholderText() {
+    return pht('Type a badge name...');
+  }
+
+  public function getDatasourceApplicationClass() {
+    return 'PhabricatorBadgesApplication';
+  }
+
+  public function loadResults() {
+    $viewer = $this->getViewer();
+    $raw_query = $this->getRawQuery();
+
+    $params = $this->getParameters();
+    $recipient_phid = $params['recipientPHID'];
+
+    $badges = id(new PhabricatorBadgesQuery())
+      ->setViewer($viewer)
+      ->requireCapabilities(
+        array(
+          PhabricatorPolicyCapability::CAN_VIEW,
+          PhabricatorPolicyCapability::CAN_EDIT,
+        ))
+      ->execute();
+
+    $awards = id(new PhabricatorBadgesAwardQuery())
+      ->setViewer($viewer)
+      ->withAwarderPHIDs(array($viewer->getPHID()))
+      ->withRecipientPHIDs(array($recipient_phid))
+      ->execute();
+    $awards = mpull($awards, null, 'getBadgePHID');
+
+    $results = array();
+    foreach ($badges as $badge) {
+      $closed = null;
+
+      $badge_awards = idx($awards, $badge->getPHID(), null);
+      if ($badge_awards) {
+        $closed = pht('Already awarded');
+      }
+
+      $status = $badge->getStatus();
+      if ($status === PhabricatorBadgesBadge::STATUS_ARCHIVED) {
+        $closed = pht('Archived');
+      }
+
+      $results[] = id(new PhabricatorTypeaheadResult())
+        ->setName($badge->getName())
+        ->setIcon($badge->getIcon())
+        ->setColor(
+          PhabricatorBadgesQuality::getQualityColor($badge->getQuality()))
+        ->setClosed($closed)
+        ->setPHID($badge->getPHID());
+    }
+
+    $results = $this->filterResultsAgainstTokens($results);
+
+    return $results;
+  }
+
+}