horsegen: generate umas

horsegen/load.go

@@ -17,6 +17,9 @@ var characterAffinity2SQL string
 //go:embed character.affinity3.sql
 var characterAffinity3SQL string
 
+//go:embed uma.sql
+var umaSQL string
+
 //go:embed skill-group.sql
 var skillGroupSQL string
 
@@ -541,3 +544,70 @@ func SparkEffects(ctx context.Context, db *sqlitex.Pool) (map[int]map[int][]Spar
 	}
 	return r, nil
 }
+
+type Uma struct {
+	ID            int
+	CharacterID   int
+	Name          string
+	Variant       string
+	CharacterName string
+
+	Sprint, Mile, Medium, Long int
+	Front, Pace, Late, End     int
+	Turf, Dirt                 int
+
+	UniqueID                               int
+	Skill1, Skill2, Skill3                 int
+	SkillPL2, SkillPL3, SkillPL4, SkillPL5 int
+}
+
+func Umas(ctx context.Context, db *sqlitex.Pool) ([]Uma, error) {
+	conn, err := db.Take(ctx)
+	defer db.Put(conn)
+	if err != nil {
+		return nil, fmt.Errorf("couldn't get connection for umas: %w", err)
+	}
+	stmt, _, err := conn.PrepareTransient(umaSQL)
+	if err != nil {
+		return nil, fmt.Errorf("couldn't prepare statement for umas: %w", err)
+	}
+	defer stmt.Finalize()
+
+	var r []Uma
+	for {
+		ok, err := stmt.Step()
+		if err != nil {
+			return nil, fmt.Errorf("error stepping umas: %w", err)
+		}
+		if !ok {
+			break
+		}
+		uma := Uma{
+			ID:            stmt.ColumnInt(0),
+			CharacterID:   stmt.ColumnInt(1),
+			Name:          stmt.ColumnText(2),
+			Variant:       stmt.ColumnText(3),
+			CharacterName: stmt.ColumnText(4),
+			Sprint:        stmt.ColumnInt(5),
+			Mile:          stmt.ColumnInt(6),
+			Medium:        stmt.ColumnInt(7),
+			Long:          stmt.ColumnInt(8),
+			Front:         stmt.ColumnInt(9),
+			Pace:          stmt.ColumnInt(10),
+			Late:          stmt.ColumnInt(11),
+			End:           stmt.ColumnInt(12),
+			Turf:          stmt.ColumnInt(13),
+			Dirt:          stmt.ColumnInt(14),
+			UniqueID:      stmt.ColumnInt(15),
+			Skill1:        stmt.ColumnInt(16),
+			Skill2:        stmt.ColumnInt(17),
+			Skill3:        stmt.ColumnInt(18),
+			SkillPL2:      stmt.ColumnInt(19),
+			SkillPL3:      stmt.ColumnInt(20),
+			SkillPL4:      stmt.ColumnInt(21),
+			SkillPL5:      stmt.ColumnInt(22),
+		}
+		r = append(r, uma)
+	}
+	return r, nil
+}