Save blame info to lint messages

Test Plan:
Applied the patch.
Looked at blame and plain blame of SVN and Git file.
Ran the lint saver.
Looked at lint messages list.
/diffusion/lint/

Reviewers: epriestley

Reviewed By: epriestley

CC: aran, Korvin

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

Author: vrana

resources/sql/patches/20130304.lintauthor.sql

@@ -0,0 +1,3 @@
+ALTER TABLE `{$NAMESPACE}_repository`.`repository_lintmessage`
+  ADD authorPHID varchar(64) COLLATE utf8_bin AFTER line,
+  ADD INDEX key_author (authorPHID);

scripts/repository/save_lint.php

@@ -39,6 +39,10 @@
       'default' => 256,
       'help' => "Number of paths passed to `arc` at once.",
     ),
+    array(
+      'name' => 'blame',
+      'help' => "Assign lint errors to authors who last modified the line.",
+    ),
   ));
 
 echo "Saving lint errors to database...\n";
@@ -48,6 +52,7 @@
   ->setArc($args->getArg('arc'))
   ->setSeverity($args->getArg('severity'))
   ->setChunkSize($args->getArg('chunk-size'))
+  ->setNeedsBlame($args->getArg('blame'))
   ->run('.');
 
 echo "\nProcessed {$count} files.\n";

src/applications/diffusion/DiffusionLintSaveRunner.php

@@ -5,13 +5,15 @@ final class DiffusionLintSaveRunner {
   private $severity = ArcanistLintSeverity::SEVERITY_ADVICE;
   private $all = false;
   private $chunkSize = 256;
+  private $needsBlame = false;
 
   private $svnRoot;
   private $lintCommit;
   private $branch;
   private $conn;
   private $deletes = array();
   private $inserts = array();
+  private $blame = array();
 
 
   public function setArc($path) {
@@ -34,6 +36,11 @@ public function setChunkSize($number) {
     return $this;
   }
 
+  public function setNeedsBlame($boolean) {
+    $this->needsBlame = $boolean;
+    return $this;
+  }
+
 
   public function run($dir) {
     $working_copy = ArcanistWorkingCopyIdentity::newFromPath($dir);
@@ -44,7 +51,7 @@ public function run($dir) {
       $svn_fetch = $api->getGitConfig('svn-remote.svn.fetch');
       list($this->svnRoot) = explode(':', $svn_fetch);
       if ($this->svnRoot != '') {
-        $this->svnRoot = '/' . $this->svnRoot;
+        $this->svnRoot = '/'.$this->svnRoot;
       }
     }
 
@@ -98,8 +105,6 @@ public function run($dir) {
       $all_files = $api->getAllFiles();
     }
 
-    $this->deletes = array();
-    $this->inserts = array();
     $count = 0;
 
     $files = array();
@@ -123,9 +128,16 @@ public function run($dir) {
 
     $this->runArcLint($files);
     $this->saveLintMessages();
-    $this->branch->setLintCommit($api->getUnderlyingWorkingCopyRevision());
+
+    $this->lintCommit = $api->getUnderlyingWorkingCopyRevision();
+    $this->branch->setLintCommit($this->lintCommit);
     $this->branch->save();
 
+    if ($this->blame) {
+      $this->blameAuthors();
+      $this->blame = array();
+    }
+
     return $count;
   }
 
@@ -156,22 +168,26 @@ private function runArcLint(array $files) {
           }
 
           foreach ($messages as $message) {
+            $line = idx($message, 'line', 0);
+
             $this->inserts[] = qsprintf(
               $this->conn,
               '(%d, %s, %d, %s, %s, %s, %s)',
               $this->branch->getID(),
               $this->svnRoot.'/'.$path,
-              idx($message, 'line', 0),
+              $line,
               idx($message, 'code', ''),
               idx($message, 'severity', ''),
               idx($message, 'name', ''),
               idx($message, 'description', ''));
+
+            if ($line && $this->needsBlame) {
+              $this->blame[$path][$line] = true;
+            }
           }
 
           if (count($this->deletes) >= 1024 || count($this->inserts) >= 256) {
-            $this->saveLintMessages($this->branch);
-            $this->deletes = array();
-            $this->inserts = array();
+            $this->saveLintMessages();
           }
         }
       }
@@ -205,6 +221,67 @@ private function saveLintMessages() {
     }
 
     $this->conn->saveTransaction();
+
+    $this->deletes = array();
+    $this->inserts = array();
+  }
+
+
+  private function blameAuthors() {
+    $repository = id(new PhabricatorRepository())->load(
+      $this->branch->getRepositoryID());
+
+    $queries = array();
+    $futures = array();
+    foreach ($this->blame as $path => $lines) {
+      $drequest = DiffusionRequest::newFromDictionary(array(
+        'repository' => $repository,
+        'branch' => $this->branch->getName(),
+        'path' => $path,
+        'commit' => $this->lintCommit,
+      ));
+      $query = DiffusionFileContentQuery::newFromDiffusionRequest($drequest)
+        ->setNeedsBlame(true);
+      $queries[$path] = $query;
+      $futures[$path] = $query->getFileContentFuture();
+    }
+
+    $authors = array();
+
+    foreach (Futures($futures)->limit(8) as $path => $future) {
+      $queries[$path]->loadFileContentFromFuture($future);
+      list(, $rev_list, $blame_dict) = $queries[$path]->getBlameData();
+      foreach (array_keys($this->blame[$path]) as $line) {
+        $commit_identifier = $rev_list[$line - 1];
+        $author = idx($blame_dict[$commit_identifier], 'authorPHID');
+        if ($author) {
+          $authors[$author][$path][] = $line;
+        }
+      }
+    }
+
+    if ($authors) {
+      $this->conn->openTransaction();
+
+      foreach ($authors as $author => $paths) {
+        $where = array();
+        foreach ($paths as $path => $lines) {
+          $where[] = qsprintf(
+            $this->conn,
+            '(path = %s AND line IN (%Ld))',
+            $this->svnRoot.'/'.$path,
+            $lines);
+        }
+        queryfx(
+          $this->conn,
+          'UPDATE %T SET authorPHID = %s WHERE %Q',
+          PhabricatorRepository::TABLE_LINTMESSAGE,
+          $author,
+          implode(' OR ', $where));
+      }
+
+      $this->conn->saveTransaction();
+    }
   }
 
 }

src/applications/diffusion/controller/DiffusionBrowseFileController.php

@@ -202,11 +202,7 @@ private function buildCorpus($selected,
         $rows = array();
         foreach ($text_list as $k => $line) {
           $rev = $rev_list[$k];
-          if (isset($blame_dict[$rev]['handle'])) {
-            $author = $blame_dict[$rev]['handle']->getName();
-          } else {
-            $author = $blame_dict[$rev]['author'];
-          }
+          $author = $blame_dict[$rev]['author'];
           $rows[] =
             sprintf("%-10s %-20s %s", substr($rev, 0, 7), $author, $line);
         }
@@ -430,11 +426,13 @@ private function buildDisplayRows(
     DiffusionFileContentQuery $file_query,
     $selected) {
 
+    $handles = array();
     if ($blame_dict) {
       $epoch_list  = ipull(ifilter($blame_dict, 'epoch'), 'epoch');
       $epoch_min   = min($epoch_list);
       $epoch_max   = max($epoch_list);
       $epoch_range = ($epoch_max - $epoch_min) + 1;
+      $handles     = $this->loadViewerHandles(ipull($blame_dict, 'authorPHID'));
     }
 
     $line_arr = array();
@@ -499,8 +497,9 @@ private function buildDisplayRows(
           $display_line['color'] = $color;
           $display_line['commit'] = $rev;
 
-          if (isset($blame['handle'])) {
-            $author_link = $blame['handle']->renderLink();
+          $author_phid = idx($blame, 'authorPHID');
+          if ($author_phid && $handles[$author_phid]) {
+            $author_link = $handles[$author_phid]->renderLink();
           } else {
             $author_link = phutil_tag(
               'span',

src/applications/diffusion/controller/DiffusionLintController.php

@@ -169,29 +169,30 @@ private function loadLintCodes(array $owner_phids) {
     }
 
     if ($owner_phids) {
+      $or = array();
+      $or[] = qsprintf($conn, 'authorPHID IN (%Ls)', $owner_phids);
+
+      $paths = array();
       $packages = id(new PhabricatorOwnersOwner())
         ->loadAllWhere('userPHID IN (%Ls)', $owner_phids);
-      if (!$packages) {
-        return array();
-      }
-
-      $paths = id(new PhabricatorOwnersPath())
-        ->loadAllWhere('packageID IN (%Ld)', array_keys($packages));
-      if (!$paths) {
-        return array();
+      if ($packages) {
+        $paths = id(new PhabricatorOwnersPath())->loadAllWhere(
+          'packageID IN (%Ld)',
+          mpull($packages, 'getPackageID'));
       }
 
-      $repositories = id(new PhabricatorRepository())->loadAllWhere(
-        'phid IN (%Ls)',
-        array_unique(mpull($paths, 'getRepositoryPHID')));
-      $repositories = mpull($repositories, 'getID', 'getPHID');
+      if ($paths) {
+        $repositories = id(new PhabricatorRepository())->loadAllWhere(
+          'phid IN (%Ls)',
+          array_unique(mpull($paths, 'getRepositoryPHID')));
+        $repositories = mpull($repositories, 'getID', 'getPHID');
 
-      $branches = id(new PhabricatorRepositoryBranch())->loadAllWhere(
-        'repositoryID IN (%Ld)',
-        $repositories);
-      $branches = mgroup($branches, 'getRepositoryID');
+        $branches = id(new PhabricatorRepositoryBranch())->loadAllWhere(
+          'repositoryID IN (%Ld)',
+          $repositories);
+        $branches = mgroup($branches, 'getRepositoryID');
+      }
 
-      $or = array();
       foreach ($paths as $path) {
         $branch = idx($branches, $repositories[$path->getRepositoryPHID()]);
         if ($branch) {
@@ -207,9 +208,6 @@ private function loadLintCodes(array $owner_phids) {
           }
         }
       }
-      if (!$or) {
-        return array();
-      }
       $where[] = '('.implode(' OR ', $or).')';
     }
 

src/applications/diffusion/controller/DiffusionLintDetailsController.php

@@ -11,6 +11,8 @@ public function processRequest() {
     $messages = $this->loadLintMessages($branch, $limit, $offset);
     $is_dir = (substr('/'.$drequest->getPath(), -1) == '/');
 
+    $authors = $this->loadViewerHandles(ipull($messages, 'authorPHID'));
+
     $rows = array();
     foreach ($messages as $message) {
       $path = hsprintf(
@@ -31,9 +33,15 @@ public function processRequest() {
         )),
         $message['line']);
 
+      $author = $message['authorPHID'];
+      if ($author && $authors[$author]) {
+        $author = $authors[$author]->renderLink();
+      }
+
       $rows[] = array(
         $path,
         $line,
+        $author,
         ArcanistLintSeverity::getStringForSeverity($message['severity']),
         $message['name'],
         $message['description'],
@@ -44,11 +52,12 @@ public function processRequest() {
       ->setHeaders(array(
         'Path',
         'Line',
+        'Author',
         'Severity',
         'Name',
         'Description',
       ))
-      ->setColumnClasses(array('', 'n', '', '', ''))
+      ->setColumnClasses(array('', 'n'))
       ->setColumnVisibility(array($is_dir));
 
     $content = array();

src/applications/diffusion/query/filecontent/DiffusionFileContentQuery.php

@@ -11,8 +11,11 @@ final public static function newFromDiffusionRequest(
     return parent::newQueryObject(__CLASS__, $request);
   }
 
-  final public function loadFileContent() {
-    $this->fileContent = $this->executeQuery();
+  abstract public function getFileContentFuture();
+  abstract protected function executeQueryFromFuture(Future $future);
+
+  final public function loadFileContentFromFuture(Future $future) {
+    $this->fileContent = $this->executeQueryFromFuture($future);
 
     $repository = $this->getRequest()->getRepository();
     $try_encoding = $repository->getDetail('encoding');
@@ -25,6 +28,14 @@ final public function loadFileContent() {
     return $this->fileContent;
   }
 
+  final protected function executeQuery() {
+    return $this->loadFileContentFromFuture($this->getFileContentFuture());
+  }
+
+  final public function loadFileContent() {
+    return $this->executeQuery();
+  }
+
   final public function getRawData() {
     return $this->fileContent->getCorpus();
   }
@@ -72,30 +83,22 @@ final public function getBlameData() {
           $commit->getEpoch();
       }
 
-      $commits_data = array();
       if ($commits) {
         $commits_data = id(new PhabricatorRepositoryCommitData())->loadAllWhere(
           'commitID IN (%Ls)',
           mpull($commits, 'getID'));
-      }
-
-      $phids = array();
-      foreach ($commits_data as $data) {
-        $phids[] = $data->getCommitDetail('authorPHID');
-      }
-
-      $loader = new PhabricatorObjectHandleData(array_unique($phids));
-      $loader->setViewer($this->viewer);
-      $handles = $loader->loadHandles();
 
-      foreach ($commits_data as $data) {
-        if ($data->getCommitDetail('authorPHID')) {
-          $commit_identifier =
-            $commits[$data->getCommitID()]->getCommitIdentifier();
-          $blame_dict[$commit_identifier]['handle'] =
-            $handles[$data->getCommitDetail('authorPHID')];
+        foreach ($commits_data as $data) {
+          $author_phid = $data->getCommitDetail('authorPHID');
+          if (!$author_phid) {
+            continue;
+          }
+          $commit = $commits[$data->getCommitID()];
+          $commit_identifier = $commit->getCommitIdentifier();
+          $blame_dict[$commit_identifier]['authorPHID'] = $author_phid;
         }
       }
+
    }
 
     return array($text_list, $rev_list, $blame_dict);