horsegen: generate saddles

horsegen/gen.go

@@ -11,7 +11,7 @@ import (
 	"unicode"
 )
 
-//go:embed character.kk.template skill.kk.template character.go.template skill.go.template race.kk.template race.go.template
+//go:embed *.template
 var templates embed.FS
 
 // LoadTemplates sets up templates to render game data to source code.
@@ -108,6 +108,21 @@ func ExecRace(t *template.Template, region string, kk, g io.Writer, races []Race
 	return err
 }
 
+func ExecSaddle(t *template.Template, region string, kk, g io.Writer, saddles []Saddle) error {
+	data := struct {
+		Region  string
+		Saddles []Saddle
+	}{region, saddles}
+	var err error
+	if kk != nil {
+		err = errors.Join(err, t.ExecuteTemplate(kk, "koka-saddle", &data))
+	}
+	if g != nil {
+		err = errors.Join(err, t.ExecuteTemplate(g, "go-saddle", &data))
+	}
+	return err
+}
+
 const wordSeps = " ,!?/-+();#○☆♡'=♪∀ﾟ∴"
 
 var (

horsegen/load.go

@@ -26,6 +26,9 @@ var skillSQL string
 //go:embed race.sql
 var raceSQL string
 
+//go:embed saddle.sql
+var saddleSQL string
+
 type (
 	Character  struct{}
 	SkillGroup struct{}
@@ -359,3 +362,46 @@ func Races(ctx context.Context, db *sqlitex.Pool) ([]Race, error) {
 	}
 	return r, nil
 }
+
+type Saddle struct {
+	ID        int
+	Name      string
+	Races     [3]int
+	Type      int
+	Primary   int
+	Alternate int
+}
+
+func Saddles(ctx context.Context, db *sqlitex.Pool) ([]Saddle, error) {
+	conn, err := db.Take(ctx)
+	defer db.Put(conn)
+	if err != nil {
+		return nil, fmt.Errorf("couldn't get connection for saddles: %w", err)
+	}
+	stmt, _, err := conn.PrepareTransient(saddleSQL)
+	if err != nil {
+		return nil, fmt.Errorf("couldn't prepare statement for saddles: %w", err)
+	}
+	defer stmt.Finalize()
+
+	var r []Saddle
+	for {
+		ok, err := stmt.Step()
+		if err != nil {
+			return nil, fmt.Errorf("error stepping saddles: %w", err)
+		}
+		if !ok {
+			break
+		}
+		s := Saddle{
+			ID:        stmt.ColumnInt(0),
+			Name:      stmt.ColumnText(1),
+			Races:     [3]int{stmt.ColumnInt(2), stmt.ColumnInt(3), stmt.ColumnInt(4)},
+			Type:      stmt.ColumnInt(5),
+			Primary:   stmt.ColumnInt(6),
+			Alternate: stmt.ColumnInt(7),
+		}
+		r = append(r, s)
+	}
+	return r, nil
+}

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)

horsegen/saddle.go.template

@@ -0,0 +1,27 @@
+{{- define "go-saddle" -}}
+package {{ $.Region }}
+
+// Automatically generated with horsegen; DO NOT EDIT
+
+import . "git.sunturtle.xyz/zephyr/horse/horse"
+
+const (
+	{{- range $s := $.Saddles }}
+	Saddle{{ goenum $s.Name }}{{ if $s.Alternate }}Alt{{ $s.Alternate }}{{ end }} SaddleID = {{ $s.ID }} // {{ $s.Name }}
+	{{- end }}
+)
+
+var AllSaddles = map[SaddleID]Saddle{
+	{{- range $s := $.Saddles }}
+	Saddle{{ goenum $s.Name }}{{ if $s.Alternate }}Alt{{ $s.Alternate }}{{ end }}: {
+		ID: {{ $s.ID }},
+		Name: {{ printf "%q" $s.Name }}{{ if $s.Alternate }} + " (Alternate {{ $s.Alternate }})"{{ end }},
+		Races: []RaceID{ {{- range $id := $s.Races }}{{ if $id }}{{ $id }}, {{ end }}{{ end -}} },
+		Type: SaddleType{{ if eq $s.Type 0 }}Honor{{ else if eq $s.Type 1 }}G3{{ else if eq $s.Type 2 }}G2{{ else if eq $s.Type 3 }}G1{{ else }}??? $s.Type={{ $s.Type }}{{ end }},
+		{{- if $s.Alternate }}
+		Primary: {{ $s.Primary }},
+		{{- end }}
+	},
+	{{- end }}
+}
+{{ end }}

horsegen/saddle.kk.template

@@ -0,0 +1,69 @@
+{{- define "koka-saddle" -}}
+module horse/{{ $.Region }}/saddle
+
+// Automatically generated with horsegen; DO NOT EDIT
+
+import horse/game-id
+pub import horse/race
+pub import horse/{{ $.Region }}/race
+
+// Enumeration of all saddles for type-safe programming.
+pub type saddle
+  {{- range $s := $.Saddles }}
+  {{ kkenum $s.Name }}{{ if $s.Alternate }}-Alt{{ $s.Alternate }}{{ end }}
+  {{- end }}
+
+// Get the saddle ID for a saddle.
+pub fun saddle-id(s: saddle): saddle-id
+  match s
+    {{- range $s := $.Saddles }}
+    {{ kkenum $s.Name }}{{ if $s.Alternate }}-Alt{{ $s.Alternate }}{{ end }} -> Saddle-id({{ $s.ID }})
+    {{- end }}
+  
+// List of all saddles in ID order for easy iterating.
+pub val all = [
+  {{- range $s := $.Saddles }}
+  {{ kkenum $s.Name }}{{ if $s.Alternate }}-Alt{{ $s.Alternate }}{{ end }},
+  {{- end }}
+]
+
+// Get the name for a saddle.
+// Alternate versions of saddles have an indication of such in their names.
+// If no saddle matches the ID, the result contains the numeric ID.
+pub fun show(s: saddle-id): string
+  match s.game-id
+    {{- range $s := $.Saddles }}
+    {{ $s.ID }} -> {{ printf "%q" $s.Name }}{{ if $s.Alternate }} ++ " (Alternate {{ $s.Alternate }})"{{ end }}
+    {{- end }}
+    x -> "saddle " ++ x.show
+
+// Get the list of races that entitle a horse to a saddle.
+// If no saddle matches the ID, the result is the empty list.
+pub fun races(s: saddle-id): list<race-id>
+  match s.game-id
+    {{- range $s := $.Saddles }}
+    {{ $s.ID }} -> [{{ range $id := $s.Races }}{{ if $id }}Race-id({{ $id }}), {{ end }}{{ end }}]
+    {{- end }}
+    _ -> []
+
+// Get a saddle's type.
+// If no saddle matches the ID, the result is Honor.
+pub fun saddle-type(s: saddle-id): saddle-type
+  match s.game-id
+    {{- range $s := $.Saddles }}
+    {{ $s.ID }} -> {{ if eq $s.Type 0 }}Honor{{ else if eq $s.Type 1 }}G3-Win{{ else if eq $s.Type 2 }}G2-Win{{ else if eq $s.Type 3 }}G1-Win{{ else }}??? $s.Type={{ $s.Type }}{{ end }}
+    {{- end }}
+    _ -> Honor
+
+// Get the primary ID for a saddle.
+// For saddles which are the primary version, or if no saddle matches the given ID,
+// the result is the input.
+pub fun primary(s: saddle-id): saddle-id
+  match s.game-id
+    {{- range $s := $.Saddles }}
+    {{- if $s.Alternate }}
+    {{ $s.ID }} -> Saddle-id({{ $s.Primary }})
+    {{- end }}
+    {{- end }}
+    _ -> s
+{{ end }}

horsegen/saddle.sql

@@ -0,0 +1,20 @@
+WITH saddle_names AS (
+	SELECT "index" AS id, "text" AS name
+	FROM text_data
+	WHERE category = 111
+)
+SELECT
+	s.id,
+	n.name,
+	ri1.id AS race1,
+	IFNULL(ri2.id, 0) AS race2,
+	IFNULL(ri3.id, 0) AS race3,
+	s.win_saddle_type,
+	MIN(s.id) OVER (PARTITION BY n.name) AS "primary",
+	ROW_NUMBER() OVER (PARTITION BY n.name ORDER BY s.id) - 1 AS "alternate"
+FROM single_mode_wins_saddle s
+	JOIN race_instance ri1 ON s.race_instance_id_1 = ri1.id
+	LEFT JOIN race_instance ri2 ON s.race_instance_id_2 = ri2.id
+	LEFT JOIN race_instance ri3 ON s.race_instance_id_3 = ri3.id
+	LEFT JOIN saddle_names n ON s.id = n.id
+ORDER BY s.id