#!/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 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 {
            "source": attr_value.source,
            "ast": 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])