add ymake export to ydb

1 files changed, 41 insertions, 0 deletions

diff --git a/build/plugins/ssqls.py b/build/plugins/ssqls.py
new file mode 100644
index 0000000000..35461851de
--- /dev/null
+++ b/build/plugins/ssqls.py
@@ -0,0 +1,41 @@
+from os.path import splitext
+
+import ymake
+from _common import resolve_includes
+
+
+class SSQLSParser(object):
+    def __init__(self, path, unit):
+        s = unit.resolve_arc_path(path)
+        assert s.startswith('$S/') and s.endswith('.ssqls'), s
+        h = '$B/' + s[3:-6] + '.h'
+
+        import xml.etree.cElementTree as ET
+
+        try:
+            doc = ET.parse(path)
+        except ET.ParseError as e:
+            unit.message(['error', 'malformed XML {}: {}'.format(path, e)])
+            doc = ET.Element('DbObject')
+        xmls, headers = self.parse_doc(doc)
+        self._includes = resolve_includes(unit, s, xmls)
+        self._induced = {'cpp': [h], 'h': resolve_includes(unit, h, headers)}
+
+    @staticmethod
+    def parse_doc(doc):
+        paths = lambda nodes: filter(None, (e.get('path') for e in nodes))
+        includes = doc.findall('include')
+        ancestors = paths(doc.findall('ancestors/ancestor'))
+        headers = [e.text.strip('<>""') for e in includes]
+        headers += [splitext(s)[0] + '.h' for s in ancestors]
+        return paths(includes) + ancestors, headers
+
+    def includes(self):
+        return self._includes
+
+    def induced_deps(self):
+        return self._induced
+
+
+def init():
+    ymake.addparser('ssqls', SSQLSParser)