Lib.Krakatau.java.ast2.py Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of krakatau-lib Show documentation
Assembler, disassebmler, decompiler and compiler tools library for Java.
There is a newer version: 1.1
from . import ast
from .stringescape import escapeString as escape

class MethodDef(object):
    def __init__(self, class_, flags, name, retType, paramDecls, body):
        self.flagstr = flags + ' ' if flags else ''
        self.retType, self.paramDecls = retType, paramDecls
        self.body = body
        self.comment = None

        if name == '':
            self.isStaticInit, self.isConstructor = True, False
        elif name == '':
            self.isStaticInit, self.isConstructor = False, True
            self.name = ast.TypeName((class_.name, 0))
        else:
            self.isStaticInit, self.isConstructor = False, False
            self.name = escape(name)

    def print_(self):
        argstr = ', '.join(decl.print_() for decl in self.paramDecls)
        if self.isStaticInit:
            header = 'static'
        elif self.isConstructor:
            name = self.name.print_().rpartition('.')[-1]
            header = '{}{}({})'.format(self.flagstr, name, argstr)
        else:
            header = '{}{} {}({})'.format(self.flagstr, self.retType.print_(), self.name, argstr)

        if self.comment:
            header = '//{}\n{}'.format(self.comment, header)

        if self.body is None:
            return header + ';\n'
        else:
            return header + ' ' + self.body.print_()

class FieldDef(object):
    def __init__(self, flags, type_, name, expr=None):
        self.flagstr = flags + ' ' if flags else ''
        self.type_ = type_
        self.name = escape(name)
        self.expr = None if expr is None else ast.makeCastExpr(type_.tt, expr) 

    def print_(self):
        if self.expr is not None:
            return '{}{} {} = {};'.format(self.flagstr, self.type_.print_(), self.name, self.expr.print_()) 
        return '{}{} {};'.format(self.flagstr, self.type_.print_(), self.name)

class ClassDef(object):
    def __init__(self, flags, isInterface, name, superc, interfaces, fields, methods):
        self.flagstr = flags + ' ' if flags else ''
        self.isInterface = isInterface
        self.name = ast.TypeName((name,0))
        self.super = ast.TypeName((superc,0)) if superc is not None else None
        self.interfaces = [ast.TypeName((iname,0)) for iname in interfaces]
        self.fields = fields
        self.methods = methods
        if superc == 'java/lang/Object':
            self.super = None

    def print_(self):
        contents = ''
        if self.fields:
            contents = '\n'.join(x.print_() for x in self.fields)
        if self.methods:
            if contents:
                contents += '\n\n' #extra line to divide fields and methods
            contents += '\n\n'.join(x.print_() for x in self.methods)

        indented = ['    '+line for line in contents.splitlines()]
        name = self.name.print_().rpartition('.')[-1]
        defname = 'interface' if self.isInterface else 'class'
        header = '{}{} {}'.format(self.flagstr, defname, name)

        if self.super:
            header += ' extends ' + self.super.print_()
        if self.interfaces:
            if self.isInterface:
                assert(self.super is None)
                header += ' extends ' + ', '.join(x.print_() for x in self.interfaces)
            else:
                header += ' implements ' + ', '.join(x.print_() for x in self.interfaces)

        lines = [header + ' {\n'] + indented + ['}']
        return '\n'.join(lines)