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

internal/sql/ast/create_function_stmt.go

@@ -0,0 +1,47 @@
+package ast
+
+import "github.com/sqlc-dev/sqlc/internal/sql/format"
+
+type CreateFunctionStmt struct {
+	Tag NodeTag[CreateFunctionStmt] `json:"tag"`
+
+	Replace    bool      `json:"replace"`
+	Params     *List     `json:"params,omitempty"`
+	ReturnType *TypeName `json:"return_type,omitempty"`
+	Func       *FuncName `json:"func,omitempty"`
+	// TODO: Understand these two fields
+	Options    *List `json:"options,omitempty"`
+	WithClause *List `json:"with_clause,omitempty"`
+}
+
+func (n *CreateFunctionStmt) Pos() int {
+	return 0
+}
+
+func (n *CreateFunctionStmt) Format(buf *TrackedBuffer, d format.Dialect) {
+	if n == nil {
+		return
+	}
+	buf.WriteString("CREATE ")
+	if n.Replace {
+		buf.WriteString("OR REPLACE ")
+	}
+	buf.WriteString("FUNCTION ")
+	buf.astFormat(n.Func, d)
+	buf.WriteString("(")
+	if items(n.Params) {
+		buf.join(n.Params, d, ", ")
+	}
+	buf.WriteString(")")
+	if n.ReturnType != nil {
+		buf.WriteString(" RETURNS ")
+		buf.astFormat(n.ReturnType, d)
+	}
+	// Format options (AS, LANGUAGE, etc.)
+	if items(n.Options) {
+		for _, opt := range n.Options.Items {
+			buf.WriteString(" ")
+			buf.astFormat(opt, d)
+		}
+	}
+}