28 lines
744 B
Go
28 lines
744 B
Go
package lang
|
|
|
|
import "fmt"
|
|
|
|
func (p *parser) parseForEach() (Stmt, error) {
|
|
if _, err := p.expect(tokenLParen, "expected '(' after 'for'"); err != nil {
|
|
return nil, err
|
|
}
|
|
name, err := p.expect(tokenIdent, "expected iteration variable")
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if _, err := p.expect(tokenIn, "expected 'in' after iteration variable"); err != nil {
|
|
return nil, err
|
|
}
|
|
source, err := p.parseExpr(0)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if _, err := p.expect(tokenRParen, "expected ')' after iteration source"); err != nil {
|
|
return nil, err
|
|
}
|
|
body, err := p.parseBlock()
|
|
if err != nil {
|
|
return nil, fmt.Errorf("invalid for body: %w", err)
|
|
}
|
|
return ForEachStmt{Name: name.lexeme, Source: source, Body: body}, nil
|
|
}
|