chore: import upstream snapshot with attribution

server/ast/call.go

@@ -0,0 +1,82 @@
+// Copyright 2023 Dolthub, Inc.
+//
+// Licensed under the Apache License, Version 2.0 (the "License");
+// you may not use this file except in compliance with the License.
+// You may obtain a copy of the License at
+//
+//     http://www.apache.org/licenses/LICENSE-2.0
+//
+// Unless required by applicable law or agreed to in writing, software
+// distributed under the License is distributed on an "AS IS" BASIS,
+// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+// See the License for the specific language governing permissions and
+// limitations under the License.
+
+package ast
+
+import (
+	"github.com/cockroachdb/errors"
+	vitess "github.com/dolthub/vitess/go/vt/sqlparser"
+
+	"github.com/dolthub/doltgresql/server/auth"
+
+	"github.com/dolthub/doltgresql/postgres/parser/sem/tree"
+	pgnodes "github.com/dolthub/doltgresql/server/node"
+)
+
+// nodeCall handles *tree.Call nodes.
+func nodeCall(ctx *Context, node *tree.Call) (vitess.Statement, error) {
+	if node == nil {
+		return nil, nil
+	}
+	if node.Procedure.Type != 0 {
+		return nil, errors.Errorf("procedure distinction is not yet supported")
+	}
+	if node.Procedure.Filter != nil {
+		return nil, errors.Errorf("procedure filters are not yet supported")
+	}
+	if node.Procedure.WindowDef != nil {
+		return nil, errors.Errorf("procedure window definitions are not yet supported")
+	}
+	if node.Procedure.AggType == tree.OrderedSetAgg {
+		return nil, errors.Errorf("procedure aggregation is not yet supported")
+	}
+	if len(node.Procedure.OrderBy) > 0 {
+		return nil, errors.Errorf("procedure ORDER BY is not yet supported")
+	}
+
+	ctx.Auth().PushAuthType(auth.AuthType_EXECUTE)
+	defer ctx.Auth().PopAuthType()
+
+	var qualifier vitess.TableIdent
+	var name vitess.ColIdent
+	switch funcRef := node.Procedure.Func.FunctionReference.(type) {
+	case *tree.FunctionDefinition:
+		name = vitess.NewColIdent(funcRef.Name)
+	case *tree.UnresolvedName:
+		if funcRef.NumParts > 2 {
+			return nil, errors.Errorf("referencing items outside the schema or database is not yet supported")
+		}
+		if funcRef.NumParts == 2 {
+			qualifier = vitess.NewTableIdent(funcRef.Parts[1])
+		}
+		name = vitess.NewColIdent(funcRef.Parts[0])
+	default:
+		return nil, errors.Errorf("unknown function reference")
+	}
+	exprs, err := nodeExprs(ctx, node.Procedure.Exprs)
+	if err != nil {
+		return nil, err
+	}
+	return vitess.InjectedStatement{
+		Statement: pgnodes.NewCall(qualifier.String(), name.String(), exprs),
+		Children:  exprs,
+		Auth: vitess.AuthInformation{
+			AuthType:    auth.AuthType_EXECUTE,
+			TargetType:  auth.AuthTargetType_FunctionIdentifiers,
+			TargetNames: []string{qualifier.String(), name.String()},
+			// TODO: need to get argument types separated by comma ( check routineArgTypesKey function )
+			Extra: "",
+		},
+	}, nil
+}