from dataclasses import dataclass
from enum import Enum, auto
from typing import List, Optional

from astree.type import Type
from astree.ident import Ident

Expr = None

# An identifier access expression.
#
#     a
AccessIdentifier = Ident

# An access expression.
AccessExpr = AccessIdentifier

# An assignment expression.
#
#     a = 10
@dataclass
class AssignExpr:
    _object: Expr
    value: Expr

# A binary arithmetic operator.
class BinarithmOp(Enum):
    BAND = '&'
    BOR = '|'
    DIV = '/'
    GT = '>'
    GTEQ = '>='
    LAND = '&&'
    LEQUAL = '=='
    LESS = '<'
    LESSEQ = '<='
    LOR = '||'
    MINUS = '-'
    NEQUAL = '!='
    PLUS = '+'
    TIMES = '*'

# A binary arithmetic expression.
#
#     a * b
@dataclass
class BinarithmExpr:
    op: BinarithmOp
    lvalue: Expr
    rvalue: Expr

# A break expression.
#
#     detener
@dataclass
class BreakExpr:
    pass

# A function call expression.
#
#     foo(bar)
@dataclass
class CallExpr:
    lvalue: Expr
    args: List[Expr]

# A compound expression.
#
#     {
#         foo;
#         bar;
#         // ...
#     }
@dataclass
class CompoundExpr:
    exprs: List[Expr]

# A continue expression.
#
#     continuar
@dataclass
class ContinueExpr:
    pass

# A scalar value.
Value = bool | str | int | type(None)

# An integer constant.
@dataclass
class NumberConstant:
    value: int

# A constant expression.
ConstantExpr = Value | NumberConstant

# A for loop.
#
#     porcada (entero a = 0; a < b; a++) {}
@dataclass
class ForExpr:
    bindings: Optional[Expr]
    cond: Expr
    afterthought: Optional[Expr]
    body: Expr

# An if or if..else expression.
#
#    si (a) { } sino { }
@dataclass
class IfExpr:
    cond: Expr
    tbranch: Expr
    fbranch: Optional[Expr]

# A print statement.
#
#     imprimir a
@dataclass
class PrintExpr:
    expr: Expr

# A read statement.
#
#     leer a
@dataclass
class ReadExpr:
    expr: AccessExpr

# A return statement.
#
#     retorna a
@dataclass
class ReturnExpr:
    expr: Optional[Expr]

# A while expression.
#
#     mientras (cond) { }
@dataclass
class WhileExpr:
    cond: Expr
    body: Expr

# A Javañol expression.
Expr = (AccessExpr | AssignExpr | BinarithmExpr | BreakExpr |
        CallExpr | ConstantExpr | ContinueExpr | ForExpr |
        IfExpr | CompoundExpr | PrintExpr | ReadExpr |
        ReturnExpr)