0f4df23290
Some checks failed
kotlin / test (push) Has been skipped
python / test (push) Has been skipped
typescript / test (push) Has been skipped
go / build darwin/amd64 (push) Has been cancelled
go / build linux/amd64 (push) Has been cancelled
go / build windows/amd64 (push) Has been cancelled
go / build darwin/arm64 (push) Has been cancelled
go / build linux/arm64 (push) Has been cancelled
go / build windows/arm64 (push) Has been cancelled
go / test (push) Has been cancelled
go / vuln_check (push) Has been cancelled
89 lines
2.3 KiB
Go
89 lines
2.3 KiB
Go
package validate
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
|
|
"github.com/sqlc-dev/sqlc/internal/metadata"
|
|
"github.com/sqlc-dev/sqlc/internal/sql/ast"
|
|
"github.com/sqlc-dev/sqlc/internal/sql/astutils"
|
|
)
|
|
|
|
func validateCopyfrom(n ast.Node) error {
|
|
stmt, ok := n.(*ast.InsertStmt)
|
|
if !ok {
|
|
return errors.New(":copyfrom requires an INSERT INTO statement")
|
|
}
|
|
if stmt.OnConflictClause != nil {
|
|
return errors.New(":copyfrom is not compatible with ON CONFLICT")
|
|
}
|
|
if stmt.WithClause != nil {
|
|
return errors.New(":copyfrom is not compatible with WITH clauses")
|
|
}
|
|
if stmt.ReturningList != nil && len(stmt.ReturningList.Items) > 0 {
|
|
return errors.New(":copyfrom is not compatible with RETURNING")
|
|
}
|
|
sel, ok := stmt.SelectStmt.(*ast.SelectStmt)
|
|
if !ok {
|
|
return nil
|
|
}
|
|
if len(sel.FromClause.Items) > 0 {
|
|
return errors.New(":copyfrom is not compatible with INSERT INTO ... SELECT")
|
|
}
|
|
if sel.ValuesLists == nil || len(sel.ValuesLists.Items) != 1 {
|
|
return errors.New(":copyfrom requires exactly one example row to be inserted")
|
|
}
|
|
sublist, ok := sel.ValuesLists.Items[0].(*ast.List)
|
|
if !ok {
|
|
return nil
|
|
}
|
|
for _, v := range sublist.Items {
|
|
if _, ok := v.(*ast.ParamRef); !ok {
|
|
return errors.New(":copyfrom doesn't support non-parameter values")
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func validateBatch(n ast.Node) error {
|
|
args := astutils.Search(n, func(n ast.Node) bool {
|
|
_, ok := n.(*ast.ParamRef)
|
|
return ok
|
|
})
|
|
if len(args.Items) == 0 {
|
|
return errors.New(":batch* commands require parameters")
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func Cmd(n ast.Node, name, cmd string) error {
|
|
if cmd == metadata.CmdCopyFrom {
|
|
return validateCopyfrom(n)
|
|
}
|
|
if (cmd == metadata.CmdBatchExec || cmd == metadata.CmdBatchMany) || cmd == metadata.CmdBatchOne {
|
|
if err := validateBatch(n); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if !(cmd == metadata.CmdMany || cmd == metadata.CmdOne || cmd == metadata.CmdBatchMany || cmd == metadata.CmdBatchOne) {
|
|
return nil
|
|
}
|
|
var list *ast.List
|
|
switch stmt := n.(type) {
|
|
case *ast.SelectStmt:
|
|
return nil
|
|
case *ast.DeleteStmt:
|
|
list = stmt.ReturningList
|
|
case *ast.InsertStmt:
|
|
list = stmt.ReturningList
|
|
case *ast.UpdateStmt:
|
|
list = stmt.ReturningList
|
|
default:
|
|
return nil
|
|
}
|
|
if list == nil || len(list.Items) == 0 {
|
|
return fmt.Errorf("query %q specifies parameter %q without containing a RETURNING clause", name, cmd)
|
|
}
|
|
return nil
|
|
}
|