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

internal/sql/ast/def_elem.go

@@ -0,0 +1,70 @@
+package ast
+
+import "github.com/sqlc-dev/sqlc/internal/sql/format"
+
+type DefElem struct {
+	Tag NodeTag[DefElem] `json:"tag"`
+
+	Defnamespace *string       `json:"defnamespace,omitempty"`
+	Defname      *string       `json:"defname,omitempty"`
+	Arg          Node          `json:"arg,omitempty"`
+	Defaction    DefElemAction `json:"defaction"`
+	Location     int           `json:"location"`
+}
+
+func (n *DefElem) Pos() int {
+	return n.Location
+}
+
+func (n *DefElem) Format(buf *TrackedBuffer, d format.Dialect) {
+	if n == nil {
+		return
+	}
+	if n.Defname != nil {
+		switch *n.Defname {
+		case "as":
+			buf.WriteString("AS ")
+			// AS clause contains function body which needs quoting
+			if l, ok := n.Arg.(*List); ok {
+				for i, item := range l.Items {
+					if i > 0 {
+						buf.WriteString(", ")
+					}
+					if s, ok := item.(*String); ok {
+						buf.WriteString("'")
+						buf.WriteString(s.Str)
+						buf.WriteString("'")
+					} else {
+						buf.astFormat(item, d)
+					}
+				}
+			} else {
+				buf.astFormat(n.Arg, d)
+			}
+		case "language":
+			buf.WriteString("LANGUAGE ")
+			buf.astFormat(n.Arg, d)
+		case "volatility":
+			// VOLATILE, STABLE, IMMUTABLE
+			buf.astFormat(n.Arg, d)
+		case "strict":
+			if s, ok := n.Arg.(*Boolean); ok && s.Boolval {
+				buf.WriteString("STRICT")
+			} else {
+				buf.WriteString("CALLED ON NULL INPUT")
+			}
+		case "security":
+			if s, ok := n.Arg.(*Boolean); ok && s.Boolval {
+				buf.WriteString("SECURITY DEFINER")
+			} else {
+				buf.WriteString("SECURITY INVOKER")
+			}
+		default:
+			buf.WriteString(*n.Defname)
+			if n.Arg != nil {
+				buf.WriteString(" ")
+				buf.astFormat(n.Arg, d)
+			}
+		}
+	}
+}