Add a wrapper script for ktlint.
Test: locally
Change-Id: I1beb3a73e41b03532a865f4546ef60c666ce36ff
diff --git a/ktlint.py b/ktlint.py
new file mode 100755
index 0000000..c78b6a9
--- /dev/null
+++ b/ktlint.py
@@ -0,0 +1,57 @@
+#!/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')
+
+
+def main(args=None):
+ parser = argparse.ArgumentParser()
+ parser.add_argument('--file', '-f', nargs='+')
+ args = parser.parse_args()
+ kotlin_files = [f for f in args.file if f.endswith('.kt')]
+ if not kotlin_files:
+ sys.exit(0)
+
+ ktlint_env = os.environ.copy()
+ ktlint_env['JAVA_CMD'] = 'java'
+ try:
+ check = subprocess.Popen(['java', '-jar', KTLINT_JAR] + kotlin_files,
+ stdout=subprocess.PIPE, env=ktlint_env)
+ stdout, _ = check.communicate()
+ if stdout:
+ print 'prebuilts/ktlint found errors in files you changed:'
+ print stdout
+ 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()
