binary.go

package sqlparser

import (
	"github.com/viant/parsly"
	"github.com/viant/parsly/matcher"
	"github.com/viant/sqlparser/expr"
)

func parseBinaryExpr(cursor *parsly.Cursor, binary *expr.Binary) error {
	var err error
	if binary.X == nil {
		binary.X, err = expectOperand(cursor)
		if err != nil || binary.X == nil {
			return err
		}
	}
	//fmt.Printf("After op %v,: %s\n", binary.Op, cursor.Input[cursor.Pos:])
	pos := cursor.Pos
	if binary.Op == "" {
		match := cursor.MatchAfterOptional(whitespaceMatcher, betweenKeywordMatcher, binaryOperatorMatcher, logicalOperatorMatcher, placeholderMatcher)
		switch match.Code {
		case logicalOperator:
			if !matcher.IsWhiteSpace(cursor.Input[cursor.Pos]) {
				cursor.Pos = pos
				return nil
			}
			binary.Op = match.Text(cursor)
		case binaryOperator:
			binary.Op = match.Text(cursor)
		case betweenToken:
			binary.Op = match.Text(cursor)
			rng := &expr.Range{}
			if rng.Min, err = expectOperand(cursor); err != nil {
				return err
			}
			match := cursor.MatchAfterOptional(whitespaceMatcher, rangeOperatorMatcher)
			if match.Code != rangeOperator {
				return cursor.NewError(rangeOperatorMatcher)
			}
			if rng.Max, err = expectOperand(cursor); err != nil {
				return err
			}
			yExpr := &expr.Binary{X: rng}
			if err := parseBinaryExpr(cursor, yExpr); err != nil {
				return err
			}
			if yExpr.Y == nil {
				binary.Y = rng
			} else {
				binary.Y = yExpr
			}
			return nil
		case placeholderTokenCode:
			binary.Op = ""
			if binary.X == nil {
				binary.X = &expr.Placeholder{Name: match.Text(cursor)}
			} else {

				placeholder := &expr.Placeholder{Name: match.Text(cursor)}
				binary.Y = placeholder
				prevPos := cursor.Pos
				match = cursor.MatchAfterOptional(whitespaceMatcher, parenthesesMatcher, logicalOperatorMatcher, groupByMatcher, groupByMatcher, havingKeywordMatcher, orderByKeywordMatcher, windowMatcher, unionMatcher)
				switch match.Code {
				case logicalOperator:
					additionalExpr := &expr.Binary{X: binary.Y, Op: match.Text(cursor)}
					if err := parseBinaryExpr(cursor, additionalExpr); err != nil {
						return err
					}
					binary.Y = additionalExpr
				case parenthesesCode:
					placeholder.Name += match.Text(cursor)
				case groupByKeyword, havingKeyword, orderByKeyword, windowTokenCode, unionKeyword:
					cursor.Pos = prevPos
					return nil
				case parsly.Invalid:
					binary.Y = nil
					cursor.Pos = pos
					return nil
				}
			}
		default:
			return nil
		}
	}
	if binary.Y == nil {
		yExpr := &expr.Binary{}
		if err := parseBinaryExpr(cursor, yExpr); err != nil {
			return err
		}
		if yExpr.X != nil {
			binary.Y = yExpr
		}
		if yExpr.Op == "" && yExpr.Y == nil {
			binary.Y = yExpr.X
		}
	}
	return nil
}