add library/cpp/actors, ymake build to ydb oss export

1 files changed, 112 insertions, 0 deletions

diff --git a/build/scripts/create_jcoverage_report.py b/build/scripts/create_jcoverage_report.py
new file mode 100644
index 0000000000..45083ff4f7
--- /dev/null
+++ b/build/scripts/create_jcoverage_report.py
@@ -0,0 +1,112 @@
+import argparse
+import tarfile
+import zipfile
+import os
+import sys
+import time
+import subprocess
+
+
+def mkdir_p(path):
+    try:
+        os.makedirs(path)
+    except OSError:
+        pass
+
+
+class Timer(object):
+
+    def __init__(self):
+        self.start = time.time()
+
+    def step(self, msg):
+        sys.stderr.write("{} ({}s)\n".format(msg, int(time.time() - self.start)))
+        self.start = time.time()
+
+
+def main(source, output, java, prefix_filter, exclude_filter, jars_list, output_format, tar_output, agent_disposition, runners_paths):
+    timer = Timer()
+    reports_dir = 'jacoco_reports_dir'
+    mkdir_p(reports_dir)
+    with tarfile.open(source) as tf:
+        tf.extractall(reports_dir)
+    timer.step("Coverage data extracted")
+    reports = [os.path.join(reports_dir, fname) for fname in os.listdir(reports_dir)]
+
+    with open(jars_list) as f:
+        jars = f.read().strip().split()
+    if jars and runners_paths:
+        for r in runners_paths:
+            try:
+                jars.remove(r)
+            except ValueError:
+                pass
+
+    src_dir = 'sources_dir'
+    cls_dir = 'classes_dir'
+
+    mkdir_p(src_dir)
+    mkdir_p(cls_dir)
+
+    for jar in jars:
+        if jar.endswith('devtools-jacoco-agent.jar'):
+            agent_disposition = jar
+
+        # Skip java contrib - it's irrelevant coverage
+        if jar.startswith('contrib/java'):
+            continue
+
+        with zipfile.ZipFile(jar) as jf:
+            for entry in jf.infolist():
+                if entry.filename.endswith('.java'):
+                    dest = src_dir
+
+                elif entry.filename.endswith('.class'):
+                    dest = cls_dir
+
+                else:
+                    continue
+
+                entry.filename = entry.filename.encode('utf-8')
+                jf.extract(entry, dest)
+    timer.step("Jar files extracted")
+
+    if not agent_disposition:
+        print>>sys.stderr, 'Can\'t find jacoco agent. Will not generate html report for java coverage.'
+
+    if tar_output:
+        report_dir = 'java.report.temp'
+    else:
+        report_dir = output
+    mkdir_p(report_dir)
+
+    if agent_disposition:
+        agent_cmd = [java, '-jar', agent_disposition, src_dir, cls_dir, prefix_filter or '.', exclude_filter or '__no_exclude__', report_dir, output_format]
+        agent_cmd += reports
+        subprocess.check_call(agent_cmd)
+        timer.step("Jacoco finished")
+
+    if tar_output:
+        with tarfile.open(output, 'w') as outf:
+            outf.add(report_dir, arcname='.')
+
+
+if __name__ == '__main__':
+    if 'LC_ALL' in os.environ:
+        if os.environ['LC_ALL'] == 'C':
+            os.environ['LC_ALL'] = 'en_GB.UTF-8'
+
+    parser = argparse.ArgumentParser()
+
+    parser.add_argument('--source', action='store')
+    parser.add_argument('--output', action='store')
+    parser.add_argument('--java', action='store')
+    parser.add_argument('--prefix-filter', action='store')
+    parser.add_argument('--exclude-filter', action='store')
+    parser.add_argument('--jars-list', action='store')
+    parser.add_argument('--output-format', action='store', default="html")
+    parser.add_argument('--raw-output', dest='tar_output', action='store_false', default=True)
+    parser.add_argument('--agent-disposition', action='store')
+    parser.add_argument('--runner-path', dest='runners_paths', action='append', default=[])
+    args = parser.parse_args()
+    main(**vars(args))