[backend/golang] MatchS

Author: lemon-mint

_out/main.go

@@ -107,6 +107,48 @@ func (e Speices) Match(
 	}
 }
 
+func (e Speices) MatchS(s struct {
+	onHuman      func()
+	onElf        func()
+	onOrc        func()
+	onDwarf      func()
+	onGnome      func()
+	onHalfling   func()
+	onHalfElf    func()
+	onHalfOrc    func()
+	onDragonborn func()
+	onTiefling   func()
+	onGnoll      func()
+	onGoblin     func()
+}) {
+	switch e {
+	case Speices_Human:
+		s.onHuman()
+	case Speices_Elf:
+		s.onElf()
+	case Speices_Orc:
+		s.onOrc()
+	case Speices_Dwarf:
+		s.onDwarf()
+	case Speices_Gnome:
+		s.onGnome()
+	case Speices_Halfling:
+		s.onHalfling()
+	case Speices_HalfElf:
+		s.onHalfElf()
+	case Speices_HalfOrc:
+		s.onHalfOrc()
+	case Speices_Dragonborn:
+		s.onDragonborn()
+	case Speices_Tiefling:
+		s.onTiefling()
+	case Speices_Gnoll:
+		s.onGnoll()
+	case Speices_Goblin:
+		s.onGoblin()
+	}
+}
+
 type ItemType uint8
 
 const (
@@ -142,6 +184,21 @@ func (e ItemType) Match(
 	}
 }
 
+func (e ItemType) MatchS(s struct {
+	onWeapon func()
+	onArmor  func()
+	onPotion func()
+}) {
+	switch e {
+	case ItemType_Weapon:
+		s.onWeapon()
+	case ItemType_Armor:
+		s.onArmor()
+	case ItemType_Potion:
+		s.onPotion()
+	}
+}
+
 type Coordinate []byte
 
 func (s Coordinate) X() int64 {

compile/backend/golang/enums.go

@@ -38,5 +38,18 @@ func writeEnums(w io.Writer, i *ir.IR) {
 		}
 		fmt.Fprintf(w, "}\n")
 		fmt.Fprintf(w, "}\n\n")
+
+		fmt.Fprintf(w, "func (e %s) MatchS(s struct {\n", NameConv(e.Name))
+		for _, o := range e.Options {
+			fmt.Fprintf(w, "on%s func()\n", NameConv(o))
+		}
+		fmt.Fprintf(w, "}) {\n")
+		fmt.Fprintf(w, "switch e {\n")
+		for _, o := range e.Options {
+			fmt.Fprintf(w, "case %s_%s:\n", NameConv(e.Name), NameConv(o))
+			fmt.Fprintf(w, "s.on%s()\n", NameConv(o))
+		}
+		fmt.Fprintf(w, "}\n")
+		fmt.Fprintf(w, "}\n\n")
 	}
 }