AutoFixingLinterTrait.hack

/*
 *  Copyright (c) 2017-present, Facebook, Inc.
 *  All rights reserved.
 *
 *  This source code is licensed under the MIT license found in the
 *  LICENSE file in the root directory of this source tree.
 *
 */

namespace Facebook\HHAST;

use namespace Facebook\HHAST\__Private\{LSP, LSPLib};
use namespace HH\Lib\{C, Str};

trait AutoFixingLinterTrait<Terror as LintError>
  implements AutoFixingLinter<Terror> {
  protected function getTitleForFix(Terror $_error): string {
    return \get_class($this)
      |> Str\split($$, '\\')
      |> C\lastx($$)
      |> Str\strip_suffix($$, 'Linter')
      |> 'Fix '.$$.' Error';
  }

  public function getCodeActionForError(Terror $error): ?LSP\CodeAction {
    $fixed = $this->getFixedFile(varray[$error]);
    if ($fixed === null) {
      return null;
    }

    return shape(
      'title' => $this->getTitleForFix($error),
      'kind' => LSP\CodeActionKind::QUICK_FIX,
      'edit' => shape(
        'changes' => dict[
          'file://'.\realpath($this->getFile()->getPath()) =>
            LSPLib\create_textedits(
              $this->getFile()->getContents(),
              $fixed->getContents(),
            ),
        ],
      ),
    );
  }

}