horsegen: generate umas

horsegen/main.go

@@ -55,6 +55,7 @@ func main() {
 		scens    []Scenario
 		sparks   []Spark
 		sparkeff map[int]map[int][]SparkEffect
+		umas     []Uma
 	)
 	eg.Go(func() error {
 		slog.Info("get characters")
@@ -116,6 +117,12 @@ func main() {
 		sparkeff = r
 		return err
 	})
+	eg.Go(func() error {
+		slog.Info("get umas")
+		r, err := Umas(ctx, db)
+		umas = r
+		return err
+	})
 	if err := eg.Wait(); err != nil {
 		slog.Error("load", slog.Any("err", err))
 		os.Exit(1)
@@ -199,6 +206,18 @@ func main() {
 		slog.Info("write sparks")
 		return ExecSparks(t, region, kf, gf, sparks, sparkeff)
 	})
+	eg.Go(func() error {
+		kf, err := os.Create(filepath.Join(out, region, "uma.kk"))
+		if err != nil {
+			return err
+		}
+		gf, err := os.Create(filepath.Join(out, region, "uma.go"))
+		if err != nil {
+			return err
+		}
+		slog.Info("write umas")
+		return ExecUmas(t, region, kf, gf, umas)
+	})
 	if err := eg.Wait(); err != nil {
 		slog.Error("generate", slog.Any("err", err))
 		os.Exit(1)