1 files changed, 98 insertions, 0 deletions
diff --git a/ipc/ipdl/ipdl/__init__.py b/ipc/ipdl/ipdl/__init__.py
new file mode 100644
index 0000000000..50ceb4f953
--- /dev/null
+++ b/ipc/ipdl/ipdl/__init__.py
@@ -0,0 +1,98 @@
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+__all__ = [
+    "gencxx",
+    "genipdl",
+    "parse",
+    "typecheck",
+    "writeifmodified",
+    "checkSyncMessage",
+    "checkFixedSyncMessages",
+]
+
+import os
+import sys
+from io import StringIO
+
+from ipdl.cgen import IPDLCodeGen
+from ipdl.lower import LowerToCxx, msgenums
+from ipdl.parser import Parser, ParseError
+from ipdl.type import TypeCheck
+from ipdl.checker import checkSyncMessage, checkFixedSyncMessages
+
+from ipdl.cxx.cgen import CxxCodeGen
+
+
+def parse(specstring, filename="/stdin", includedirs=[], errout=sys.stderr):
+    """Return an IPDL AST if parsing was successful.  Print errors to |errout|
+    if it is not."""
+    # The file type and name are later enforced by the type checker.
+    # This is just a hint to the parser.
+    prefix, ext = os.path.splitext(filename)
+    name = os.path.basename(prefix)
+    if ext == ".ipdlh":
+        type = "header"
+    else:
+        type = "protocol"
+
+    try:
+        return Parser(type, name).parse(
+            specstring, os.path.abspath(filename), includedirs
+        )
+    except ParseError as p:
+        print(p, file=errout)
+        return None
+
+
+def typecheck(ast, errout=sys.stderr):
+    """Return True iff |ast| is well typed.  Print errors to |errout| if
+    it is not."""
+    return TypeCheck().check(ast, errout)
+
+
+def gencxx(ipdlfilename, ast, outheadersdir, outcppdir, segmentcapacitydict):
+    headers, cpps = LowerToCxx().lower(ast, segmentcapacitydict)
+
+    def resolveHeader(hdr):
+        return [
+            hdr,
+            os.path.join(
+                outheadersdir, *([ns.name for ns in ast.namespaces] + [hdr.name])
+            ),
+        ]
+
+    def resolveCpp(cpp):
+        return [cpp, os.path.join(outcppdir, cpp.name)]
+
+    for ast, filename in [resolveHeader(hdr) for hdr in headers] + [
+        resolveCpp(cpp) for cpp in cpps
+    ]:
+        tempfile = StringIO()
+        CxxCodeGen(tempfile).cgen(ast)
+        writeifmodified(tempfile.getvalue(), filename)
+
+
+def genipdl(ast, outdir):
+    return IPDLCodeGen().cgen(ast)
+
+
+def genmsgenum(ast):
+    return msgenums(ast.protocol, pretty=True)
+
+
+def writeifmodified(contents, file):
+    contents = contents.encode("utf-8")
+    dir = os.path.dirname(file)
+    os.path.exists(dir) or os.makedirs(dir)
+
+    oldcontents = None
+    if os.path.exists(file):
+        fd = open(file, "rb")
+        oldcontents = fd.read()
+        fd.close()
+    if oldcontents != contents:
+        fd = open(file, "wb")
+        fd.write(contents)
+        fd.close()