package parser
import (
"strconv"
"github.com/marzeq/quokka/shared"
"github.com/marzeq/quokka/tokeniser"
)
type (
Node interface{ GetLoc() shared.Location }
ExpressionNode interface {
Node
GetType() shared.Type
}
)
type RootNode struct {
Body []Node
Loc shared.Location
}
func (n RootNode) GetLoc() shared.Location { return n.Loc }
type IdentifierNode struct {
Name string
Next *IdentifierNode
Loc shared.Location
ExprType shared.Type
}
func (n IdentifierNode) GetLoc() shared.Location { return n.Loc }
func (n IdentifierNode) GetType() shared.Type { return n.ExprType }
func (n IdentifierNode) String() string {
if n.Next != nil {
return n.Name + "." + n.Next.String()
}
return n.Name
}
type ModuleAccessNode struct {
ModName string
Ident *IdentifierNode
Loc shared.Location
ExprType shared.Type
}
func (n ModuleAccessNode) GetLoc() shared.Location { return n.Loc }
func (n ModuleAccessNode) GetType() shared.Type { return n.ExprType }
func (n ModuleAccessNode) String() string {
if n.ModName == "" {
return n.Ident.String()
}
return n.ModName + ":" + n.Ident.String()
}
type TypeNode struct {
ModName string
Name string
PointerTo *TypeNode
ArrayTo *TypeNode
ArrayLen int
Loc shared.Location
}
func (n TypeNode) GetLoc() shared.Location { return n.Loc }
func (n TypeNode) String() string {
if n.ArrayTo != nil {
if n.ArrayLen > 0 {
return "[" + n.ArrayTo.String() + ", " + strconv.Itoa(n.ArrayLen) + "]"
} else {
return "[" + n.ArrayTo.String() + "]"
}
}
if n.PointerTo != nil {
return "*" + n.PointerTo.String()
}
if n.ModName != "" {
return n.ModName + ":" + n.Name
}
return n.Name
}
type BoolLiteralNode struct {
Value string
Loc shared.Location
ExprType shared.Type
}
func (n BoolLiteralNode) GetLoc() shared.Location { return n.Loc }
func (n BoolLiteralNode) GetType() shared.Type { return n.ExprType }
type IntegerLiteralNode struct {
Value string
Loc shared.Location
ExprType shared.Type
}
func (n IntegerLiteralNode) GetLoc() shared.Location { return n.Loc }
func (n IntegerLiteralNode) GetType() shared.Type { return n.ExprType }
type FloatLiteralNode struct {
Value string
Loc shared.Location
ExprType shared.Type
}
func (n FloatLiteralNode) GetLoc() shared.Location { return n.Loc }
func (n FloatLiteralNode) GetType() shared.Type { return n.ExprType }
type StringLiteralNode struct {
Value string
Loc shared.Location
ExprType shared.Type
}
func (n StringLiteralNode) GetLoc() shared.Location { return n.Loc }
func (n StringLiteralNode) GetType() shared.Type { return n.ExprType }
type CharLiteralNode struct {
Value byte
Loc shared.Location
ExprType shared.Type
}
func (n CharLiteralNode) GetLoc() shared.Location { return n.Loc }
func (n CharLiteralNode) GetType() shared.Type { return n.ExprType }
type NilLiteralNode struct {
Loc shared.Location
}
func (n NilLiteralNode) GetLoc() shared.Location { return n.Loc }
func (n NilLiteralNode) GetType() shared.Type {
return shared.Pointer{
To: shared.PRIMITIVE_VOID,
}
}
type StructLiteralNode struct {
Name *ModuleAccessNode
Fields []shared.Pair[string, ExpressionNode] // field name, value
Loc shared.Location
ExprType shared.Type
}
func (n StructLiteralNode) GetLoc() shared.Location { return n.Loc }
func (n StructLiteralNode) GetType() shared.Type { return n.ExprType }
type ArrayLiteralNode struct {
Elements []ExpressionNode
Loc shared.Location
ExprType shared.Type
}
func (n ArrayLiteralNode) GetLoc() shared.Location { return n.Loc }
func (n ArrayLiteralNode) GetType() shared.Type { return n.ExprType }
type FunctionCallNode struct {
Name *ModuleAccessNode
Args []ExpressionNode
Loc shared.Location
ExprType shared.Type
}
func (n FunctionCallNode) GetLoc() shared.Location { return n.Loc }
func (n FunctionCallNode) GetType() shared.Type { return n.ExprType }
type IfExprBranch struct {
Condition ExpressionNode
Node ExpressionNode
}
type IfExprNode struct {
IfBranch IfExprBranch
ElseIfBranches []IfExprBranch
ElseBranch ExpressionNode
Loc shared.Location
ExprType shared.Type
}
func (n IfExprNode) GetLoc() shared.Location { return n.Loc }
func (n IfExprNode) GetType() shared.Type { return n.ExprType }
type GivenExprNode struct {
Block *BlockNode
FinalExpr ExpressionNode
Loc shared.Location
ExprType shared.Type
}
func (n GivenExprNode) GetLoc() shared.Location { return n.Loc }
func (n GivenExprNode) GetType() shared.Type { return n.ExprType }
type UnaryOpType uint
const (
UNARY_OP_LOGICAL_NOT UnaryOpType = iota
UNARY_OP_NEGATE
UNARY_OP_REFERENCE
UNARY_OP_DEREFERENCE
UNARY_OP_ARRAY_LEN
)
func (u UnaryOpType) String() string {
switch u {
case UNARY_OP_LOGICAL_NOT:
return "not"
case UNARY_OP_NEGATE:
return "-"
case UNARY_OP_REFERENCE:
return "&"
case UNARY_OP_DEREFERENCE:
return "*"
case UNARY_OP_ARRAY_LEN:
return "[]"
default:
return "unknown"
}
}
type UnaryOpNode struct {
Op UnaryOpType
Operand ExpressionNode
Loc shared.Location
ExprType shared.Type
}
func (n UnaryOpNode) GetLoc() shared.Location { return n.Loc }
func (n UnaryOpNode) GetType() shared.Type { return n.ExprType }
type IndexExprNode struct {
Subject ExpressionNode
Index ExpressionNode
Loc shared.Location
ExprType shared.Type
}
func (n IndexExprNode) GetLoc() shared.Location { return n.Loc }
func (n IndexExprNode) GetType() shared.Type { return n.ExprType }
type BinaryOpType uint
const (
BINARY_OP_LOGICAL_OR BinaryOpType = iota
BINARY_OP_LOGICAL_AND
BINARY_OP_EQUAL
BINARY_OP_NOT_EQUAL
BINARY_OP_LESS
BINARY_OP_LESS_EQUAL
BINARY_OP_GREATER
BINARY_OP_GREATER_EQUAL
BINARY_OP_ADD
BINARY_OP_SUBTRACT
BINARY_OP_MULTIPLY
BINARY_OP_DIVIDE
BINARY_OP_MODULO
)
type BinaryOpNode struct {
Op BinaryOpType
Operand1 ExpressionNode
Operand2 ExpressionNode
Loc shared.Location
ExprType shared.Type
}
func (n BinaryOpNode) GetLoc() shared.Location { return n.Loc }
func (n BinaryOpNode) GetType() shared.Type { return n.ExprType }
type CastNode struct {
ToType *TypeNode
Operand ExpressionNode
Loc shared.Location
ExprType shared.Type
}
func (n CastNode) GetLoc() shared.Location { return n.Loc }
func (n CastNode) GetType() shared.Type { return n.ExprType }
type SizeOfNode struct {
Operand *TypeNode
Loc shared.Location
ExprType shared.Type
}
func (n SizeOfNode) GetLoc() shared.Location { return n.Loc }
func (n SizeOfNode) GetType() shared.Type { return n.ExprType }
type ImportNode struct {
Modules []string
Loc shared.Location
}
func (n ImportNode) GetLoc() shared.Location { return n.Loc }
type ModuleNode struct {
Name string
Loc shared.Location
}
func (n ModuleNode) GetLoc() shared.Location { return n.Loc }
type FunctionNodeType struct {
Name string
Type *TypeNode
Mutable bool
}
type FunctionDefNode struct {
Name string
Args []FunctionNodeType
RetType FunctionNodeType
Body Node
ExternFrom string
HasVariadic bool
Extern bool
Pub bool
Loc shared.Location
}
func (n FunctionDefNode) GetLoc() shared.Location { return n.Loc }
type StructField struct {
Name string
Type *TypeNode
}
type StructDefNode struct {
Name string
Fields []StructField
Pub bool
Loc shared.Location
}
func (n StructDefNode) GetLoc() shared.Location { return n.Loc }
type IfBranch struct {
Condition ExpressionNode
Node *BlockNode
}
type IfNode struct {
IfBranch IfBranch
ElseIfBranches []IfBranch
ElseBranch *BlockNode
Loc shared.Location
}
func (n IfNode) GetLoc() shared.Location { return n.Loc }
type ForNode struct {
ExprsOrStmts []Node
Body *BlockNode
Loc shared.Location
}
func (n ForNode) GetLoc() shared.Location { return n.Loc }
type ControlKeywordNode struct {
Keyword tokeniser.KeywordType
ReturnValue ExpressionNode // only for "return"
Loc shared.Location
}
func (n ControlKeywordNode) GetLoc() shared.Location { return n.Loc }
type DeclarationNode struct {
Name string
Mutable bool
Type *TypeNode
Value ExpressionNode
Loc shared.Location
}
func (n DeclarationNode) GetLoc() shared.Location { return n.Loc }
type AssignmentNode struct {
Assignee Node
Value ExpressionNode
Loc shared.Location
}
func (n AssignmentNode) GetLoc() shared.Location { return n.Loc }
type ArrayAssignmentNode struct {
Assignee ExpressionNode
Index ExpressionNode
Value ExpressionNode
Loc shared.Location
}
func (n ArrayAssignmentNode) GetLoc() shared.Location { return n.Loc }
type BlockNode struct {
Body []Node
Loc shared.Location
}
func (n BlockNode) GetLoc() shared.Location { return n.Loc }