import ply.lex as lex

reserved = {'int' : 'INT',
            'while' : 'WHILE'}

tokens = ['ID','NUMBER','PUBLICZEILE',
          'SYSTEMZEILE','GKAUF','GKZU',
          'ISTGLEICH','STRICHPUNKT',
          'GROESSER','MINUS','PLUS',
          'RKAUF','RKZU'] + list(reserved.values())                       

def t_PUBLICZEILE(t):
    r'public[^{]*'
    print(t)
    return t
    
def t_SYSTEMZEILE(t):
    r'System[^;]*;'
    print(t)
    # kein return, damit die Zeile vom Parser nicht gesehen wird

def t_GKAUF(t):
    r'\{'
    print(t)
    return t

def t_GKZU(t):
    r'\}'
    print(t)
    return t

def t_ISTGLEICH(t):
    r'='
    print(t)
    return(t)

def t_STRICHPUNKT(t):
    r';'
    print(t)
    return(t)

def t_GROESSER(t):
    r'\>'
    print(t)
    return(t)

def t_MINUS(t):
    r'\-'
    print(t)
    return(t)

def t_PLUS(t):
    r'\+'
    print(t)
    return(t)

def t_RKAUF(t):
    r'\('
    print(t)
    return t

def t_RKZU(t):
    r'\)'
    print(t)
    return t
    
def t_ID(t):
    r'[a-zA-Z_][a-zA-Z_0-9]*'
    t.type = reserved.get(t.value,'ID')
    print(t)
    return t                             

def t_NUMBER(t):
    r'\d+'
    t.value = int(t.value)
    print(t)
    return t

def t_newline(t):
    r'\n+'
    t.lexer.lineno = t.lexer.lineno + len(t.value)

t_ignore  = ' \t'

def t_error(t):
    print('Unerwartetes Zeichen:',t.value[0],',Zeile:',s.lineno)
    t.lexer.skip(1)

s = lex.lex()

import ply.yacc as yacc

def p_0(p):
    "start : PUBLICZEILE GKAUF PUBLICZEILE GKAUF kette GKZU GKZU"
    p[0] = ['S',p[1],p[2],p[3],p[4],p[5],p[6],p[7]]
    print('start :',p[1],p[2],p[3],p[4],p[5],p[6],p[7])

def p_1(p):
    'kette : befehl kette'
    p[0] = ['K',p[1],p[2]]
    print('kette :',p[1],p[2])

def p_2(p):
    'kette :'
    p[0] = ['K']
    print('kette :')

def p_3(p):
    'befehl : inc'
    p[0] = ['B',p[1]]
    print('befehl :',p[1])

def p_4(p):
    'befehl : dec'
    p[0] = ['B',p[1]]
    print('befehl :',p[1])

def p_5(p):
    'befehl : tst'
    p[0] = ['B',p[1]]
    print('befehl :',p[1])

def p_6(p):
    'befehl : zuw'
    p[0] = ['B',p[1]]
    print('befehl :',p[1])

def p_7(p):
    'zuw : INT ID ISTGLEICH NUMBER STRICHPUNKT'
    p[0] = ['Z',p[1],p[2],p[3],p[4],p[5]]
    print('zuw :',p[1],p[2],p[3],p[4],p[5])
    
def p_8(p):
    "inc : ID ISTGLEICH ID PLUS NUMBER STRICHPUNKT"
    p[0] = ['I',p[1],p[2],p[3],p[4],p[5],p[6]]
    if (p[1] != p[3]) or (p[5] != 1):
        print('Syntaxfehler in inc, Zeile:',s.lineno)
        raise SyntaxError
    print('inc :',p[1],p[2],p[3],p[4],p[5],p[6])
   
def p_9(p):
    "dec : ID ISTGLEICH ID MINUS NUMBER STRICHPUNKT"
    p[0] = ['D',p[1],p[2],p[3],p[4],p[5],p[6]]
    if (p[1] != p[3]) or (p[5] != 1):
        print('Syntaxfehler in dec, Zeile:',s.lineno)
        raise SyntaxError
    print('dec :',p[1],p[2],p[3],p[4],p[5],p[6])
    
def p_10(p):
    'tst : WHILE RKAUF ID GROESSER NUMBER RKZU GKAUF kette GKZU STRICHPUNKT'
    p[0] = ['T',p[1],p[2],p[3],p[4],p[5],p[6],p[7],p[8],p[9],p[10]]
    print('tst : ',p[1],p[2],p[3],p[4],p[5],p[6],p[7],p[8],p[9],p[10]) 
   
def p_error(p):
    print('Syntaxfehler in Zeile:',s.lineno)

pr = yacc.yacc()

import sys
try:
    dateiname = sys.argv[1]
except:
    dateiname = 'add1.java'
datei = open(dateiname,'r')
inhalt = datei.read()
datei.close()
ast = pr.parse(inhalt) # ast - abstrakter Syntax-Baum
print('\n',ast)