1 files changed, 98 insertions, 0 deletions
diff --git a/ast2json/rpyc2json.py b/ast2json/rpyc2json.py
new file mode 100755
index 0000000..e57ed83
--- /dev/null
+++ b/ast2json/rpyc2json.py
@@ -0,0 +1,98 @@
+#!/usr/bin/env python3
+
+# Copyright (c) 2012 Yuri K. Schlesner
+#
+# Permission is hereby granted, free of charge, to any person obtaining a copy
+# of this software and associated documentation files (the "Software"), to deal
+# in the Software without restriction, including without limitation the rights
+# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+# copies of the Software, and to permit persons to whom the Software is
+# furnished to do so, subject to the following conditions:
+#
+# The above copyright notice and this permission notice shall be included in
+# all copies or substantial portions of the Software.
+#
+# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+# SOFTWARE.
+
+import ast2json
+import glob
+import itertools
+import simplejson as json
+import optparse
+import os.path
+import pickle
+import sys
+import zlib
+
+class Dummy:
+    def record_pycode(self, *args, **kwargs):
+        return
+    all_pycode = []
+
+# Needed for pickle to read the AST
+import renpy
+import renpy.object
+import renpy.game
+renpy.game.script = Dummy()
+import renpy.ast
+import renpy.atl
+
+def pretty_print_ast(out_file, ast):
+    json.dump(rast2json(ast), out_file, separators=(',', ':'))
+
+def node2json(node):
+    to_return = {}
+    to_return['_type'] = node.__class__.__name__
+    for attr in node.__slots__:
+        to_return[attr] = get_value(getattr(node, attr))
+
+    return to_return
+
+def rast2json(ast):
+    return list(map(node2json, ast))
+
+def get_value(attr_value):
+    if attr_value is None:
+        return attr_value
+    if isinstance(attr_value, (int, str, float, complex, bool)):
+        return attr_value
+    if isinstance(attr_value, list) or isinstance(attr_value, tuple):
+        return [get_value(x) for x in attr_value]
+    if isinstance(attr_value, renpy.ast.Node):
+        return node2json(attr_value)
+    if isinstance(attr_value, renpy.ast.PyCode):
+        return ast2json.str2json(attr_value.source)
+    if isinstance(attr_value, renpy.ast.ArgumentInfo):
+        return list(map(lambda x: getattr(attr_value, x), ["arguments", "extrapos", "extrakw"]))
+    if isinstance(attr_value, renpy.atl.RawBlock):
+        return 'ATL not implemented'
+
+    raise Exception("I don't know how to decompile '%s' of type '%s'!" % (attr_value, type(attr_value)))
+
+def read_ast_from_file(in_file):
+    raw_contents = zlib.decompress(in_file.read())
+    _, stmts = pickle.loads(raw_contents)
+    return stmts
+
+def decompile_rpyc(input_filename, out_filename):
+
+    print("Decompiling %s to %s..." % (input_filename, out_filename))
+
+    with open(input_filename, 'rb') as in_file:
+        ast = read_ast_from_file(in_file)
+
+    with open(out_filename, 'w', encoding='utf-8') as out_file:
+        pretty_print_ast(out_file, ast)
+
+if __name__ == "__main__":
+
+    if len(sys.argv) != 3: raise Exception()
+
+    decompile_rpyc(sys.argv[1], sys.argv[2])
+