#!/usr/bin/env python3
# Copyright 2021 The Chromium Authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.
"""Redirects to the version of rustfmt present in the Chrome tree.

Rust binaries are pulled down from Google Cloud Storage whenever you sync
Chrome. This script knows how to locate those tools, assuming the script is
invoked from inside a Chromium checkout."""

import gclient_paths
import os
import subprocess
import sys


class NotFoundError(Exception):
  """A file could not be found."""

  def __init__(self, e):
    Exception.__init__(
        self, 'Problem while looking for rustfmt in Chromium source tree:\n'
        '%s' % e)


def FindRustfmtToolInChromiumTree():
  """Return a path to the rustfmt executable, or die trying."""
  chromium_src_path = gclient_paths.GetPrimarySolutionPath()
  if not chromium_src_path:
    raise NotFoundError(
        'Could not find checkout in any parent of the current path.\n'
        'Set CHROMIUM_BUILDTOOLS_PATH to use outside of a chromium checkout.')

  # TODO(lukasza): Deduplicate the is-Rust-supported and find-Rust-binaries code
  # by somehow sharing the `rust_prefix` variable from //build/config/rust.gni
  tool_path = os.path.join(chromium_src_path, 'third_party',
                           'android_rust_toolchain', 'toolchain', 'bin',
                           'rustfmt' + gclient_paths.GetExeSuffix())
  if not os.path.exists(tool_path):
    raise NotFoundError('File does not exist: %s' % tool_path)
  return tool_path


def IsRustfmtSupported():
  try:
    FindRustfmtToolInChromiumTree()
    return True
  except NotFoundError:
    return False


def main(args):
  try:
    tool = FindRustfmtToolInChromiumTree()
  except NotFoundError as e:
    sys.stderr.write("%s\n" % str(e))
    return 1

  # Add some visibility to --help showing where the tool lives, since this
  # redirection can be a little opaque.
  help_syntax = ('-h', '--help', '-help', '-help-list', '--help-list')
  if any(match in args for match in help_syntax):
    print('\nDepot tools redirects you to the rustfmt at:\n    %s\n' % tool)

  return subprocess.call([tool] + args)


if __name__ == '__main__':
  try:
    sys.exit(main(sys.argv[1:]))
  except KeyboardInterrupt:
    sys.stderr.write('interrupted\n')
    sys.exit(1)
