from pyparsing import *

alpha = Word(alphas)
publicKeyword = Keyword("public")
protectedKeyword = Keyword("protected")
number = Word(nums)
string = "\"" + Word(alphanums) + "\""
rawcode = "'" + Word(alphanums, max=4, min=4) + "'" # rawcodes solamente pueden ser de 4 caracteres de largo
variableType = alpha
operator = Word("-+*/", max=1)
relationalOperator = (Word("<>!", max=1) + Optional("=")) ^ Word("==") # !=, ==, <, >, etc
logicalOperator = Word("&&") ^ Word("||") # si falta el operador not, pero es mas facil agregarlo directamente en la expresion boleana
values = Forward()
typedArgument = variableType - alpha + Optional("=" - values) # argumentos en los cuales se debe especificar el tipo (por ejemplo: <unit> caster)
typedArguments = typedArgument - ZeroOrMore("," - typedArgument)
arguments = values - ZeroOrMore("," - values) # argumentos en donde no se debe especificar el tipo
functionCall = alpha - "(" - ZeroOrMore(arguments) - ")"
methodCall = alpha - "." - functionCall
anonymousFunction = Forward()
value = (functionCall ^ anonymousFunction ^ alpha ^ number ^ string ^ rawcode)
values <<= value - ZeroOrMore(operator - value)
variableDeclaration = variableType - alpha - Optional("=" - OneOrMore(values))
setVariable = Optional(alpha + ".") - alpha - Optional(operator) - "=" - OneOrMore(values)
blockCode = Forward()
booleanExpression = Optional("!") + value + Optional(relationalOperator + value)
booleanExpresions = booleanExpression - ZeroOrMore(logicalOperator - booleanExpression)
returnDefinition = Keyword("return") - (OneOrMore(values) ^ ZeroOrMore(booleanExpresions) ^ Empty())
elseifDefinition = Keyword("elseif") - "(" - OneOrMore(booleanExpression) - ")" - blockCode
elseDefinition = Keyword("else") - blockCode
ifDefinition = Keyword("if") - "(" - OneOrMore(booleanExpression) - ")" - blockCode - ZeroOrMore(elseifDefinition) + Optional(elseDefinition)
expression = ((functionCall ^ returnDefinition ^ variableDeclaration ^ setVariable ^ methodCall) - ";") ^ ifDefinition
blockCode <<= "{" - Group(ZeroOrMore(expression)) - "}"
functionDefinition = Optional(publicKeyword) + variableType - alpha - "(" - ZeroOrMore(typedArguments) - ")" - blockCode
anonymousFunction <<= variableType - "(" - ZeroOrMore(typedArguments) - ")" - blockCode
staticKeyword = Keyword("static")
methodDefinition = Optional(publicKeyword ^ protectedKeyword) + Optional(staticKeyword) + variableType - alpha - "(" - ZeroOrMore(typedArguments) - ")" - blockCode
propertyDefinition = Optional(publicKeyword ^ protectedKeyword) + Optional(staticKeyword) + variableDeclaration
classBody = "{" - Group(ZeroOrMore(methodDefinition) & ZeroOrMore(propertyDefinition - ";")) - "}"
classDefinition = Optional(publicKeyword) + Keyword("class") - alpha - classBody
globalVariableDefinition = Optional(publicKeyword) + variableDeclaration
libraryBody = Group(ZeroOrMore(functionDefinition) & ZeroOrMore(globalVariableDefinition - ";") & ZeroOrMore(classDefinition))
requirement = alpha
requirements = requirement - ZeroOrMore("," - requirement)
grammar = Keyword("library") - alpha - Optional(Keyword("requires") - OneOrMore(requirements)) - "{" - libraryBody - "}"

# cargamos y leemos el archivo de prueba
testFile = open("example.txt", "r")
code = testFile.read()
testFile.close()

# y vemos que sale <.<
parsed = grammar.parseString(code, parseAll=True)

print code
print parsed

# luego de "desmenuzar" el codigo, hay que ir parte por parte
# para pasar el codigo a jass puro
# en esta parte, solamente mostramos el nombre de la libreria
print "El nombre de la libreria es: ", parsed[1]