81 lines
1.7 KiB
Go
81 lines
1.7 KiB
Go
package parser
|
|
|
|
import "fmt"
|
|
|
|
// Node represents any node in the AST
|
|
type Node interface {
|
|
String() string
|
|
}
|
|
|
|
// Statement represents statement nodes
|
|
type Statement interface {
|
|
Node
|
|
statementNode()
|
|
}
|
|
|
|
// Expression represents expression nodes
|
|
type Expression interface {
|
|
Node
|
|
expressionNode()
|
|
}
|
|
|
|
// Program represents the root of the AST
|
|
type Program struct {
|
|
Statements []Statement
|
|
}
|
|
|
|
func (p *Program) String() string {
|
|
var result string
|
|
for _, stmt := range p.Statements {
|
|
result += stmt.String() + "\n"
|
|
}
|
|
return result
|
|
}
|
|
|
|
// AssignStatement represents variable assignment
|
|
type AssignStatement struct {
|
|
Name *Identifier
|
|
Value Expression
|
|
}
|
|
|
|
func (as *AssignStatement) statementNode() {}
|
|
func (as *AssignStatement) String() string {
|
|
return fmt.Sprintf("%s = %s", as.Name.String(), as.Value.String())
|
|
}
|
|
|
|
// Identifier represents identifiers
|
|
type Identifier struct {
|
|
Value string
|
|
}
|
|
|
|
func (i *Identifier) expressionNode() {}
|
|
func (i *Identifier) String() string { return i.Value }
|
|
|
|
// NumberLiteral represents numeric literals
|
|
type NumberLiteral struct {
|
|
Value float64
|
|
}
|
|
|
|
func (nl *NumberLiteral) expressionNode() {}
|
|
func (nl *NumberLiteral) String() string { return fmt.Sprintf("%.2f", nl.Value) }
|
|
|
|
// StringLiteral represents string literals
|
|
type StringLiteral struct {
|
|
Value string
|
|
}
|
|
|
|
func (sl *StringLiteral) expressionNode() {}
|
|
func (sl *StringLiteral) String() string { return fmt.Sprintf(`"%s"`, sl.Value) }
|
|
|
|
// InfixExpression represents binary operations
|
|
type InfixExpression struct {
|
|
Left Expression
|
|
Operator string
|
|
Right Expression
|
|
}
|
|
|
|
func (ie *InfixExpression) expressionNode() {}
|
|
func (ie *InfixExpression) String() string {
|
|
return fmt.Sprintf("(%s %s %s)", ie.Left.String(), ie.Operator, ie.Right.String())
|
|
}
|