php8.1: added enum (#12)

pkg/visitor/printer/printer.go

@@ -234,6 +234,28 @@ func (p *printer) StmtClass(n *ast.StmtClass) {
 	p.printToken(n.CloseCurlyBracketTkn, []byte("}"))
 }
 
+func (p *printer) StmtEnum(n *ast.StmtEnum) {
+	p.printList(n.AttrGroups)
+	p.printToken(n.EnumTkn, []byte("enum"))
+	p.printNode(n.Name)
+	p.printToken(n.ColonTkn, nil)
+	p.printNode(n.Type)
+	p.printToken(n.ImplementsTkn, p.ifNodeList(n.Implements, []byte("implements")))
+	p.printSeparatedList(n.Implements, n.ImplementsSeparatorTkns, []byte(","))
+	p.printToken(n.OpenCurlyBracketTkn, []byte("{"))
+	p.printList(n.Stmts)
+	p.printToken(n.CloseCurlyBracketTkn, []byte("}"))
+}
+
+func (p *printer) EnumCase(n *ast.EnumCase) {
+	p.printList(n.AttrGroups)
+	p.printToken(n.CaseTkn, []byte("case"))
+	p.printNode(n.Name)
+	p.printToken(n.EqualTkn, nil)
+	p.printNode(n.Expr)
+	p.printToken(n.SemiColonTkn, []byte(";"))
+}
+
 func (p *printer) StmtClassConstList(n *ast.StmtClassConstList) {
 	p.printList(n.AttrGroups)
 	p.printList(n.Modifiers)

pkg/visitor/printer/printer_php8_1_test.go

@@ -40,3 +40,37 @@ echo 0o10;
 echo 0O10;
 `)
 }
+
+func TestEnumPHP81(t *testing.T) {
+	tester.NewParserPrintTestSuite(t).UsePHP8().Run(`<?php
+enum A {
+  case B;
+  case B = 100;
+  case C = "aa";
+  case 
+	D;
+}
+
+enum A: int {
+  case B;
+  case B = 100;
+
+  #[Attribute]
+  case C = 100;
+
+  #[Attribute1]
+  #[Attribute2]
+  case D;
+}
+
+enum A implements B {
+  case C;
+  public function f(): string {}
+}
+
+enum A implements B, C, D {
+  case E;
+  public function f(): string {}
+}
+`)
+}