from smnp.ast.node.model import Node
from smnp.ast.parser import Parser
from smnp.token.type import TokenType


class Atom(Node):
    def __init__(self, pos):
        super().__init__(pos)
        self.children = [None]

    @property
    def value(self):
        return self[0]

    @value.setter
    def value(self, value):
        self[0] = value

    @classmethod
    def withValue(cls, value, pos):
        node = cls(pos)
        node.value = value
        return node


class IntegerLiteral(Atom):
    pass


class StringLiteral(Atom):
    pass


class NoteLiteral(Atom):
    pass


class BoolLiteral(Atom):
    pass


class TypeLiteral(Atom):
    pass


def IntegerParser(input):
    return Parser.terminal(TokenType.INTEGER, createNode=IntegerLiteral.withValue)(input)


def StringParser(input):
    return Parser.terminal(TokenType.STRING, createNode=StringLiteral.withValue)(input)


def NoteParser(input):
    return Parser.terminal(TokenType.NOTE, createNode=NoteLiteral.withValue)(input)


def BoolParser(input):
    return Parser.terminal(TokenType.BOOL, createNode=BoolLiteral.withValue)(input)


def TypeLiteralParser(input):
    return Parser.terminal(TokenType.TYPE, createNode=TypeLiteral.withValue)(input)


def LiteralParser(input):
    return Parser.oneOf(
        IntegerParser,
        StringParser,
        NoteParser,
        BoolParser,
        TypeLiteralParser,
        name="literal"
    )(input)


def AtomParser(input):
    from smnp.ast.node.identifier import IdentifierParser

    return Parser.oneOf(
        LiteralParser,
        IdentifierParser,
        name="atom"
    )(input)