phorge-arcanist/src/workflow/lint/ArcanistLintWorkflow.php

<?php

/*
 * Copyright 2011 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.
 */

class ArcanistLintWorkflow extends ArcanistBaseWorkflow {

  const RESULT_OKAY       = 0;
  const RESULT_WARNINGS   = 1;
  const RESULT_ERRORS     = 2;
  const RESULT_SKIP       = 3;

  public function getCommandHelp() {
    return phutil_console_format(<<<EOTEXT
      **lint** [__options__] [__paths__] (svn)
      **lint** [__options__] [__commit_range__] (git)
          Supports: git, svn
          Run static analysis on changes to check for mistakes. If no files
          are specified, lint will be run on all files which have been modified.
EOTEXT
      );
  }

  public function getArguments() {
    return array(
      'lintall' => array(
        'help' =>
          "Show all lint warnings, not just those on changed lines."
      ),
      'summary' => array(
        'help' =>
          "Show lint warnings in a more compact format."
      ),
      'advice' => array(
        'help' =>
          "Show lint advice, not just warnings and errors."
      ),
      'engine' => array(
        'param' => 'classname',
        'help' =>
          "Override configured lint engine for this project."
      ),
      '*' => 'paths',
    );
  }

  public function requiresWorkingCopy() {
    return true;
  }

  public function run() {
    $working_copy = $this->getWorkingCopy();

    $engine = $this->getArgument('engine');
    if (!$engine) {
      $engine = $working_copy->getConfig('lint_engine');
    }

    $should_lint_all = $this->getArgument('lintall');

    $repository_api = null;
    if (!$should_lint_all) {
      try {
        $repository_api = ArcanistRepositoryAPI::newAPIFromWorkingCopyIdentity(
          $working_copy);
        $this->setRepositoryAPI($repository_api);
      } catch (ArcanistUsageException $ex) {
        throw new ArcanistUsageException(
          $ex->getMessage()."\n\n".
          "Use '--lintall' to ignore working copy changes when running lint.");
      }

      if ($repository_api instanceof ArcanistSubversionAPI) {
        $paths = $repository_api->getWorkingCopyStatus();
        $list  = new FileList($this->getArgument('paths'));
        foreach ($paths as $path => $flags) {
          if (!$list->contains($path)) {
            unset($paths[$path]);
          }
        }
      } else {
        $this->parseGitRelativeCommit(
          $repository_api,
          $this->getArgument('paths'));
        $paths = $repository_api->getWorkingCopyStatus();
      }

      foreach ($paths as $path => $flags) {
        if ($flags & ArcanistRepositoryAPI::FLAG_UNTRACKED) {
          unset($paths[$path]);
        }
      }

      $paths = array_keys($paths);

    } else {
      $paths = $this->getArgument('paths');
      if (empty($paths)) {
        throw new ArcanistUsageException(
          "You must specify one or more files to lint when using '--lintall'.");
      }
    }

    if (!$engine) {
      throw new ArcanistNoEngineException(
        "No lint engine configured for this project. Edit .arcconfig to ".
        "specify a lint engine.");
    }

    $ok = phutil_autoload_class($engine);
    if (!$ok) {
      throw new ArcanistUsageException(
        "Configured lint engine '{$engine}' could not be loaded.");
    }

    $engine = newv($engine, array());
    $engine->setWorkingCopy($working_copy);

    if ($this->getArgument('advice')) {
      $engine->setMinimumSeverity(ArcanistLintSeverity::SEVERITY_ADVICE);
    } else {
      $engine->setMinimumSeverity(ArcanistLintSeverity::SEVERITY_WARNING);
    }

    $engine->setPaths($paths);
    if (!$should_lint_all) {
      foreach ($paths as $path) {
        $engine->setPathChangedLines(
          $path,
          $this->getChangedLines($path, 'new'));
      }
    }

    $results = $engine->run();

    $apply_patches = true;
    $prompt_patches = true;
    $wrote_to_disk = false;

    $renderer = new ArcanistLintRenderer();
    if ($this->getArgument('summary')) {
      $renderer->setSummaryMode(true);
    }
    foreach ($results as $result) {
      if (!$result->getMessages()) {
        continue;
      }

      echo $renderer->renderLintResult($result);

      if ($apply_patches && $result->isPatchable()) {
        $patcher = ArcanistLintPatcher::newFromArcanistLintResult($result);
        $old = $patcher->getUnmodifiedFileContent();
        $new = $patcher->getModifiedFileContent();

        if ($prompt_patches) {
          $old_file = $result->getFilePathOnDisk();
          $new_file = new TempFile();
          Filesystem::writeFile($new_file, $new);

          // TODO: Improve the behavior here, make it more like
          // difference_render().
          passthru(csprintf("diff -u %s %s", $old_file, $new_file));

          $prompt = phutil_console_format(
            "Apply this patch to __%s__?",
            $result->getPath());
          if (!phutil_console_confirm($prompt, $default_no = false)) {
            continue;
          }
        }

        $patcher->writePatchToDisk();
        $wrote_to_disk = true;
      }
    }

    if ($wrote_to_disk && ($repository_api instanceof ArcanistGitAPI)) {
      $amend = phutil_console_confirm(
        "Amend HEAD with lint patches?",
        $default_no = false);
      if (!$amend) {
        throw new ArcanistUsageException("Resolve lint changes and rediff.");
      }
      execx(
        '(cd %s; git commit -a --amend -C HEAD)',
        $repository_api->getPath());
    }

    $result_code = self::RESULT_OKAY;
    foreach ($results as $result) {
      foreach ($result->getMessages() as $message) {
        if (!$message->isPatchApplied()) {
          if ($message->isError()) {
            $result_code = self::RESULT_ERRORS;
            break;
          } else if ($message->isWarning()) {
            $result_code = self::RESULT_WARNINGS;
          }
        }
      }
    }

    if (!$this->getParentWorkflow()) {
      if ($result_code == self::RESULT_OKAY) {
        echo phutil_console_format(
          "<bg:green>** OKAY **</bg> No lint warnings.\n");
      }
    }

    return $result_code;
  }
}