Snap for 9010110 from 75418db586a698ad4ae11ada2acee6a9d881fcec to tm-qpr1-release
Change-Id: I548beee1599d27e05d820986e06cf8f2721443f8
diff --git a/generate_includes_file.py b/generate_includes_file.py
new file mode 100755
index 0000000..be9ebff
--- /dev/null
+++ b/generate_includes_file.py
@@ -0,0 +1,95 @@
+#!/usr/bin/env python3
+
+#
+# Copyright 2022, 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 to generate an include file that can be passed to ktfmt.py.
+
+The include file is generated from one or more folders containing one or more
+Kotlin files. The generated include file will exclude all Kotlin files that
+currently don't follow ktfmt format. This way, we can easily start formatting
+all new files in a project/directory.
+"""
+
+import argparse
+import os
+import subprocess
+import sys
+
+
+def main():
+ parser = argparse.ArgumentParser(
+ 'Generate an include file that can be passed to ktfmt.py')
+ parser.add_argument(
+ '--output', '-o', required=True, help='The output include file.')
+ parser.add_argument(
+ 'files',
+ nargs='*',
+ help='The files or directories that should be included.')
+ args = parser.parse_args()
+
+ # Add a line preprended with '+' for included files/folders.
+ with open(args.output, 'w+') as out:
+ includes = args.files
+ includes.sort()
+ for include in includes:
+ out.write('+' + include + '\n')
+
+ # Retrieve all Kotlin files.
+ kt_files = []
+ for file in args.files:
+ if os.path.isdir(file):
+ for root, dirs, files in os.walk(file):
+ for f in files:
+ if is_kotlin_file(f):
+ kt_files += [os.path.join(root, f)]
+
+ if is_kotlin_file(file):
+ kt_files += [file]
+
+ # Check all files with ktfmt.
+ ktfmt_args = ['--kotlinlang-style', '--set-exit-if-changed', '--dry-run'
+ ] + kt_files
+ dir = os.path.dirname(__file__)
+ ktfmt_jar = os.path.normpath(
+ os.path.join(dir,
+ '../../prebuilts/build-tools/common/framework/ktfmt.jar'))
+
+ ktlint_env = os.environ.copy()
+ ktlint_env['JAVA_CMD'] = 'java'
+ try:
+ process = subprocess.Popen(
+ ['java', '-jar', ktfmt_jar] + ktfmt_args,
+ stdout=subprocess.PIPE,
+ env=ktlint_env)
+
+ # Add a line prepended with '-' for all files that are not correctly.
+ # formatted.
+ stdout, _ = process.communicate()
+ incorrect_files = stdout.decode('utf-8').splitlines()
+ incorrect_files.sort()
+ for file in incorrect_files:
+ out.write('-' + file + '\n')
+ except OSError as e:
+ print('Error running ktfmt')
+ sys.exit(1)
+
+
+def is_kotlin_file(name):
+ return name.endswith('.kt') or name.endswith('.kts')
+
+
+if __name__ == '__main__':
+ main()
diff --git a/ktfmt.py b/ktfmt.py
index 9cfbd13..6f115f7 100755
--- a/ktfmt.py
+++ b/ktfmt.py
@@ -36,7 +36,7 @@
'--includes_file',
'-i',
default='',
- help='The file containing the Kotlin files and directories that should be formatted/checked.'
+ help='The file containing the Kotlin files and directories that should be included/excluded, generated using generate_includes_file.py.'
)
parser.add_argument(
'files',
@@ -68,17 +68,20 @@
f = open(includes_file, 'r')
lines = f.read().splitlines()
+ included = [line[1:] for line in lines if line.startswith('+')]
included_files = set()
included_dirs = []
- for line in lines:
+ for line in included:
if is_kotlin_file(line):
included_files.add(line)
else:
included_dirs += [line]
+ excluded_files = [line[1:] for line in lines if line.startswith('-')]
+
kt_files = [
- kt_file for kt_file in kt_files
- if kt_file in included_files or is_included(kt_file, included_dirs)
+ kt_file for kt_file in kt_files if kt_file not in excluded_files and
+ (kt_file in included_files or is_included(kt_file, included_dirs))
]
# No need to start ktfmt if there are no files to check/format.