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

1 files changed, 45 insertions, 0 deletions

diff --git a/build/plugins/code_generator.py b/build/plugins/code_generator.py
new file mode 100644
index 0000000000..c1dc792201
--- /dev/null
+++ b/build/plugins/code_generator.py
@@ -0,0 +1,45 @@
+import re
+import os
+
+import ymake
+
+pattern = re.compile(r"#include\s*[<\"](?P<INDUCED>[^>\"]+)[>\"]|(?:@|{@)\s*(?:import|include|from)\s*[\"'](?P<INCLUDE>[^\"']+)[\"']")
+
+
+class CodeGeneratorTemplateParser(object):
+    def __init__(self, path, unit):
+        self._path = path
+        retargeted = os.path.join(unit.path(), os.path.relpath(path, unit.resolve(unit.path())))
+        with open(path, 'rb') as f:
+            includes, induced = CodeGeneratorTemplateParser.parse_includes(f.readlines())
+        self._includes = unit.resolve_include([retargeted] + includes) if includes else []
+        self._induced = unit.resolve_include([retargeted] + induced) if induced else []
+
+    @staticmethod
+    def parse_includes(lines):
+        includes = []
+        induced = []
+
+        for line in lines:
+            for match in pattern.finditer(line):
+                type = match.lastgroup
+                if type == 'INCLUDE':
+                    includes.append(match.group(type))
+                elif type == 'INDUCED':
+                    induced.append(match.group(type))
+                else:
+                    raise Exception("Unexpected match! Perhaps it is a result of an error in pattern.")
+        return (includes, induced)
+
+    def includes(self):
+        return self._includes
+
+    def induced_deps(self):
+        return {
+            'h+cpp': self._induced
+        }
+
+
+def init():
+    ymake.addparser('markettemplate', CodeGeneratorTemplateParser)
+    ymake.addparser('macro', CodeGeneratorTemplateParser)