src/QueryCat.Backend/Parser/ProgramParserVisitor.Insert.cs - develop - krasninja/querycat

src/QueryCat.Backend/Parser/ProgramParserVisitor.Insert.cs
73 строки3 KB
Ivan Kozhin
Allow to specify simple expressions in FROM block
24 май 2024, 15:12
24 май 2024, 15:121894842
using QueryCat.Backend.Ast;
using QueryCat.Backend.Ast.Nodes;
using QueryCat.Backend.Ast.Nodes.Function;
using QueryCat.Backend.Ast.Nodes.Insert;
using QueryCat.Backend.Ast.Nodes.Select;
using QueryCat.Backend.Core.Types;

namespace QueryCat.Backend.Parser;

internal partial class ProgramParserVisitor
{
    /// <inheritdoc />
    public override IAstNode VisitInsertStatement(QueryCatParser.InsertStatementContext context)
        => new InsertStatementNode(
            new InsertNode(
                this.Visit<ExpressionNode>(context.insertToSource()),
                this.Visit<SelectQueryNode>(context.insertFromSource())
            )
            {
                ColumnsNode = this.VisitMaybe<InsertColumnsListNode>(context.insertColumnsList()),
            });

    #region From

    /// <inheritdoc />
    public override IAstNode VisitInsertSourceQuery(QueryCatParser.InsertSourceQueryContext context)
        => this.Visit<SelectQueryNode>(context.selectQueryExpression());

    /// <inheritdoc />
    public override IAstNode VisitInsertSourceTable(QueryCatParser.InsertSourceTableContext context)
    {
        return new SelectQuerySpecificationNode(new SelectColumnsListNode(new SelectColumnsSublistAll()))
        {
            TableExpressionNode = new SelectTableNode(
                new SelectTableReferenceListNode(this.Visit<SelectTableValuesNode>(context.selectTableValues()))
            )
        };
    }

    #endregion

    #region To

    /// <inheritdoc />
    public override IAstNode VisitInsertColumnsList(QueryCatParser.InsertColumnsListContext context)
        => new InsertColumnsListNode(
            this.Visit<IdentifierExpressionNode>(context.identifier()).Select(n => n.Name));

    /// <inheritdoc />
    public override IAstNode VisitInsertNoFormat(QueryCatParser.InsertNoFormatContext context)
        => this.Visit<FunctionCallNode>(context.functionCall());

    /// <inheritdoc />
    public override IAstNode VisitInsertWithFormat(QueryCatParser.InsertWithFormatContext context)
    {
        var readFunction = new FunctionCallNode("write");
        var uri = GetUnwrappedText(context.uri);
        readFunction.Arguments.Add(new FunctionCallArgumentNode("uri",
            new LiteralNode(new VariantValue(uri))));
        if (context.functionCall() != null)
        {
            var formatterFunctionCallNode = this.Visit<FunctionCallNode>(context.functionCall());
            readFunction.Arguments.Add(new FunctionCallArgumentNode("fmt", formatterFunctionCallNode));
        }
        return new FunctionCallNode(readFunction);
    }

    /// <inheritdoc />
    public override IAstNode VisitInsertFromVariable(QueryCatParser.InsertFromVariableContext context)
        => this.Visit<IdentifierExpressionNode>(context.identifier());

    #endregion
}