feat: parse continue and break

glox/ast/stmt.go

@@ -11,7 +11,7 @@ type StmtVisitor interface {
  VisitBlockStmt(*BlockStmt) any
  VisitIfStmt(*IfStmt) any
  VisitWhile(*WhileStmt) any
- VisitBreak(*Break) any
+ VisitBranch(*BranchStmt) any
 }
 
 type Statement interface {
@@ -94,14 +94,15 @@ func (stmt *WhileStmt) Accept(v StmtVisitor) any {
  return v.VisitWhile(stmt)
 }
 
-type Break struct {
+// BranchStmt statement represent `break` or `continue`
+type BranchStmt struct {
  Token token.Token
 }
 
-func NewBreakStmt(tok token.Token) *Break {
- return &Break{Token: tok}
+func NewBranch(tok token.Token) *BranchStmt {
+ return &BranchStmt{Token: tok}
 }
 
-func (b *Break) Accept(v StmtVisitor) any {
- return v.VisitBreak(b)
+func (b *BranchStmt) Accept(v StmtVisitor) any {
+ return v.VisitBranch(b)
 }

glox/token/token.go

@@ -42,8 +42,8 @@ const (
  FUNCTION = "FUNCTION"
  FOR = "FOR"
  OR = "OR"
- BREAK = "BREAK"
- CONTINUE = "CONTINUE"
+ BREAK = "BREAK"
+ CONTINUE = "CONTINUE"
  NIL = "NIL"
  PRINT = "PRINT"
  RETURN = "RETURN"
@@ -67,25 +67,25 @@ type Token struct {
 }
 
 var keywords = map[string]TokenType{
- "and": AND,
- "class": CLASS,
- "else": ELSE,
- "false": FALSE,
- "true": TRUE,
- "for": FOR,
- "fn": FUNCTION,
- "if": IF,
- "nil": NIL,
- "or": OR,
- "print": PRINT,
- "return": RETURN,
- "super": SUPER,
- "this": THIS,
- "let": LET,
- "while": WHILE,
- "var": LET,
- "break": BREAK,
- "continue": CONTINUE,
+ "and":  AND,
+ "class":  CLASS,
+ "else":  ELSE,
+ "false":  FALSE,
+ "true":  TRUE,
+ "for":  FOR,
+ "fn":  FUNCTION,
+ "if":  IF,
+ "nil":  NIL,
+ "or":  OR,
+ "print":  PRINT,
+ "return":  RETURN,
+ "super":  SUPER,
+ "this":  THIS,
+ "let":  LET,
+ "while":  WHILE,
+ "var":  LET,
+ "break":  BREAK,
+ "continue": CONTINUE,
 }
 
 func LookupIdentifier(keyword string) TokenType {
@@ -96,3 +96,7 @@ func LookupIdentifier(keyword string) TokenType {
 
  return IDENTIFIER
 }
+
+func IsLoopController(tok TokenType) bool {
+ return tok == CONTINUE || tok == BREAK
+}