package ast

import "github.com/sqlc-dev/sqlc/internal/sql/format"

type WithClause struct {
	Tag NodeTag[WithClause] `json:"tag"`

	Ctes      *List `json:"ctes,omitempty"`
	Recursive bool  `json:"recursive"`
	Location  int   `json:"location"`
}

func (n *WithClause) Pos() int {
	return n.Location
}

func (n *WithClause) Format(buf *TrackedBuffer, d format.Dialect) {
	if n == nil {
		return
	}
	buf.WriteString("WITH ")
	if n.Recursive {
		buf.WriteString("RECURSIVE ")
	}
	buf.joinComma(n.Ctes, d)
}