Fix Kotlin coverage

ref:486ae0399a478a28300588f7f4d5b8b99619235a

2 files changed, 21 insertions, 8 deletions

diff --git a/build/scripts/java_pack_to_file.py b/build/scripts/java_pack_to_file.py
index 8d2aeb93fda..c8ab7c311b4 100644
--- a/build/scripts/java_pack_to_file.py
+++ b/build/scripts/java_pack_to_file.py
@@ -2,7 +2,8 @@ import os
 import re
 import optparse
 
-PACKAGE_REGEX = re.compile(r'^\s*package\s+(.*?);', flags=re.MULTILINE | re.DOTALL)
+JAVA_PACKAGE_REGEX = re.compile(r'^\s*package\s+(.*?);', flags=re.MULTILINE | re.DOTALL)
+KOTLIN_PACKAGE_REGEX = re.compile(r'^\s*package\s+(.*?)^', flags=re.MULTILINE | re.DOTALL)
 
 
 def parse_args():
@@ -14,9 +15,15 @@ def parse_args():
 
 def get_package_name(filename):
     with open(filename) as afile:
-        match = PACKAGE_REGEX.search(afile.read())
-        if match:
-            return match.group(1).replace('\n\t ', '').replace('.', '/')
+        content = afile.read()
+        if filename.endswith(".kt"):
+            match = KOTLIN_PACKAGE_REGEX.search(content)
+            if match:
+                return match.group(1).strip().replace('.', '/')
+        else:
+            match = JAVA_PACKAGE_REGEX.search(content)
+            if match:
+                return match.group(1).replace('\n\t ', '').replace('.', '/')
     return ''
 
 
diff --git a/build/scripts/make_java_srclists.py b/build/scripts/make_java_srclists.py
index f1dc807f7df..65174bafd7a 100644
--- a/build/scripts/make_java_srclists.py
+++ b/build/scripts/make_java_srclists.py
@@ -10,6 +10,12 @@ def writelines(f, rng):
     f.writelines(item + '\n' for item in rng)
 
 
+def add_rel_src_to_coverage(coverage, src, source_root):
+    rel = os.path.relpath(src, source_root)
+    if not rel.startswith('..' + os.path.sep):
+        coverage.append(rel)
+
+
 def main():
     args = pcf.get_args(sys.argv[1:])
     parser = argparse.ArgumentParser()
@@ -60,11 +66,11 @@ def main():
             java.append(src)
             kotlin.append(src)
             if args.coverage and args.source_root:
-                rel = os.path.relpath(src, args.source_root)
-                if not rel.startswith('..' + os.path.sep):
-                    coverage.append(rel)
+                add_rel_src_to_coverage(coverage, src, args.source_root)
         elif args.kotlin and src.endswith(".kt"):
             kotlin.append(src)
+            if args.coverage and args.source_root:
+                add_rel_src_to_coverage(coverage, src, args.source_root)
         elif args.groovy and src.endswith(".groovy"):
             groovy.append(src)
         else:
@@ -119,4 +125,4 @@ def main():
 
 
 if __name__ == '__main__':
-    sys.exit(main())
\ No newline at end of file
+    sys.exit(main())