snapshot of https://github.com/sqlc-dev/sqlc.git (main, 2026-09-16; blobs > 131072B omitted)

internal/core/cast.go

@@ -0,0 +1,69 @@
+package core
+
+import (
+	"context"
+	"fmt"
+
+	"github.com/sqlc-dev/sqlc/internal/core/catalogdb"
+)
+
+type CastSpec struct {
+	SourceTypeOID int64
+	TargetTypeOID int64
+	ProcOID       int64
+	Context       string
+	DialectOID    int64
+}
+
+func (c *Catalog) CreateCast(cs CastSpec) error {
+	if cs.Context == "" {
+		cs.Context = "e"
+	}
+	err := c.q.CreateCast(context.Background(), catalogdb.CreateCastParams{
+		SourceTypeOid: cs.SourceTypeOID,
+		TargetTypeOid: cs.TargetTypeOID,
+		ProcOid:       nullInt64(cs.ProcOID),
+		Context:       cs.Context,
+		DialectOid:    nullInt64(cs.DialectOID),
+	})
+	if err != nil {
+		return fmt.Errorf("create cast %d->%d: %w", cs.SourceTypeOID, cs.TargetTypeOID, err)
+	}
+	return nil
+}
+
+func (c *Catalog) FindCast(src, tgt int64) (CastSpec, bool, error) {
+	row, err := c.q.FindCast(context.Background(), catalogdb.FindCastParams{
+		SourceTypeOid: src,
+		TargetTypeOid: tgt,
+	})
+	if err != nil {
+		return CastSpec{}, false, nil
+	}
+	return CastSpec{
+		SourceTypeOID: row.SourceTypeOid,
+		TargetTypeOID: row.TargetTypeOid,
+		ProcOID:       orZero(row.ProcOid),
+		Context:       row.Context,
+		DialectOID:    orZero(row.DialectOid),
+	}, true, nil
+}
+
+func (c *Catalog) CastAllowed(src, tgt int64, ctx string) (bool, error) {
+	if src == tgt {
+		return true, nil
+	}
+	cs, ok, err := c.FindCast(src, tgt)
+	if err != nil || !ok {
+		return false, err
+	}
+	switch ctx {
+	case "i":
+		return cs.Context == "i", nil
+	case "a":
+		return cs.Context == "i" || cs.Context == "a", nil
+	case "e":
+		return true, nil
+	}
+	return false, fmt.Errorf("unknown cast context %q", ctx)
+}