horsegen: generate sparks

horsegen/load.go

@@ -32,6 +32,12 @@ var saddleSQL string
 //go:embed scenario.sql
 var scenarioSQL string
 
+//go:embed spark.sql
+var sparkSQL string
+
+//go:embed spark-effect.sql
+var sparkEffectSQL string
+
 type (
 	Character  struct{}
 	SkillGroup struct{}
@@ -447,3 +453,88 @@ func Scenarios(ctx context.Context, db *sqlitex.Pool) ([]Scenario, error) {
 	}
 	return r, nil
 }
+
+type Spark struct {
+	ID          int
+	Name        string
+	Description string
+	Group       int
+	Rarity      int
+	Type        int
+}
+
+type SparkEffect struct {
+	Target int
+	Value1 int
+	Value2 int
+}
+
+func Sparks(ctx context.Context, db *sqlitex.Pool) ([]Spark, error) {
+	conn, err := db.Take(ctx)
+	defer db.Put(conn)
+	if err != nil {
+		return nil, fmt.Errorf("couldn't get connection for sparks: %w", err)
+	}
+	stmt, _, err := conn.PrepareTransient(sparkSQL)
+	if err != nil {
+		return nil, fmt.Errorf("couldn't prepare statement for sparks: %w", err)
+	}
+	defer stmt.Finalize()
+
+	var r []Spark
+	for {
+		ok, err := stmt.Step()
+		if err != nil {
+			return nil, fmt.Errorf("error stepping sparks: %w", err)
+		}
+		if !ok {
+			break
+		}
+		s := Spark{
+			ID:          stmt.ColumnInt(0),
+			Name:        stmt.ColumnText(1),
+			Description: stmt.ColumnText(2),
+			Group:       stmt.ColumnInt(3),
+			Rarity:      stmt.ColumnInt(4),
+			Type:        stmt.ColumnInt(5),
+		}
+		r = append(r, s)
+	}
+	return r, nil
+}
+
+func SparkEffects(ctx context.Context, db *sqlitex.Pool) (map[int]map[int][]SparkEffect, error) {
+	conn, err := db.Take(ctx)
+	defer db.Put(conn)
+	if err != nil {
+		return nil, fmt.Errorf("couldn't get connection for spark effects: %w", err)
+	}
+	stmt, _, err := conn.PrepareTransient(sparkEffectSQL)
+	if err != nil {
+		return nil, fmt.Errorf("couldn't prepare statement for spark effects: %w", err)
+	}
+	defer stmt.Finalize()
+
+	r := make(map[int]map[int][]SparkEffect)
+	for {
+		ok, err := stmt.Step()
+		if err != nil {
+			return nil, fmt.Errorf("error stepping spark effects: %w", err)
+		}
+		if !ok {
+			break
+		}
+		group := stmt.ColumnInt(0)
+		eff := stmt.ColumnInt(1)
+		s := SparkEffect{
+			Target: stmt.ColumnInt(2),
+			Value1: stmt.ColumnInt(3),
+			Value2: stmt.ColumnInt(4),
+		}
+		if r[group] == nil {
+			r[group] = make(map[int][]SparkEffect)
+		}
+		r[group][eff] = append(r[group][eff], s)
+	}
+	return r, nil
+}