#!/usr/bin/python
|
|
#
|
# Copyright 2017, The Android Open Source Project
|
#
|
# 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.
|
#
|
|
"""Script that is used by developers to run style checks on Kotlin files."""
|
|
import argparse
|
import errno
|
import os
|
import subprocess
|
import sys
|
|
MAIN_DIRECTORY = os.path.normpath(os.path.dirname(__file__))
|
KTLINT_JAR = os.path.join(MAIN_DIRECTORY, 'ktlint-android-all.jar')
|
FORMAT_MESSAGE = '''
|
**********************************************************************
|
To format run:
|
{}/ktlint.py --format --file {}
|
**********************************************************************
|
'''
|
|
|
def main(args=None):
|
parser = argparse.ArgumentParser()
|
parser.add_argument('--file', '-f', nargs='*')
|
parser.add_argument('--format', '-F', dest='format', action='store_true')
|
parser.add_argument('--noformat', dest='format', action='store_false')
|
parser.set_defaults(format=False)
|
args = parser.parse_args()
|
kt_files = [f for f in args.file if f.endswith('.kt') or f.endswith('.kts')]
|
ktlint_args = kt_files[:]
|
if args.format:
|
ktlint_args += ['-F']
|
if not ktlint_args:
|
sys.exit(0)
|
|
ktlint_args += ['--android']
|
|
ktlint_env = os.environ.copy()
|
ktlint_env['JAVA_CMD'] = 'java'
|
try:
|
check = subprocess.Popen(['java', '-jar', KTLINT_JAR] + ktlint_args,
|
stdout=subprocess.PIPE, env=ktlint_env)
|
stdout, _ = check.communicate()
|
if stdout:
|
print 'prebuilts/ktlint found errors in files you changed:'
|
print stdout
|
print FORMAT_MESSAGE.format(MAIN_DIRECTORY, ' '.join(kt_files))
|
sys.exit(1)
|
else:
|
sys.exit(0)
|
except OSError as e:
|
if e.errno == errno.ENOENT:
|
print 'Error running ktlint!'
|
sys.exit(1)
|
|
if __name__ == '__main__':
|
main()
|
