horsegen: generate saddles

horsegen/main.go

@@ -45,12 +45,13 @@ func main() {
 
 	eg, ctx := errgroup.WithContext(pctx)
 	var (
-		charas []NamedID[Character]
-		pairs  []AffinityRelation
-		trios  []AffinityRelation
-		sg     []NamedID[SkillGroup]
-		skills []Skill
-		races  []Race
+		charas  []NamedID[Character]
+		pairs   []AffinityRelation
+		trios   []AffinityRelation
+		sg      []NamedID[SkillGroup]
+		skills  []Skill
+		races   []Race
+		saddles []Saddle
 	)
 	eg.Go(func() error {
 		slog.Info("get characters")
@@ -88,6 +89,12 @@ func main() {
 		races = r
 		return err
 	})
+	eg.Go(func() error {
+		slog.Info("get saddles")
+		s, err := Saddles(ctx, db)
+		saddles = s
+		return err
+	})
 	if err := eg.Wait(); err != nil {
 		slog.Error("load", slog.Any("err", err))
 		os.Exit(1)
@@ -135,6 +142,18 @@ func main() {
 		slog.Info("write races")
 		return ExecRace(t, region, kf, gf, races)
 	})
+	eg.Go(func() error {
+		kf, err := os.Create(filepath.Join(out, region, "saddle.kk"))
+		if err != nil {
+			return err
+		}
+		gf, err := os.Create(filepath.Join(out, region, "saddle.go"))
+		if err != nil {
+			return err
+		}
+		slog.Info("write saddles")
+		return ExecSaddle(t, region, kf, gf, saddles)
+	})
 	if err := eg.Wait(); err != nil {
 		slog.Error("generate", slog.Any("err", err))
 		os.Exit(1)