checker.go

package kvql

func (e *BinaryOpExpr) Check(ctx *CheckCtx) error {
	if err := e.Left.Check(ctx); err != nil {
		return err
	}
	if err := e.Right.Check(ctx); err != nil {
		return err
	}
	e.tryRewriteExpr(ctx)
	switch e.Op {
	case And, Or:
		return e.checkWithAndOr(ctx)
	case Not:
		return NewSyntaxError(e.GetPos(), "Invalid operator !")
	case Add, Sub, Mul, Div:
		return e.checkWithMath(ctx)
	case In:
		return e.checkWithIn(ctx)
	case Between:
		return e.checkWithBetween(ctx)
	default:
		return e.checkWithCompares(ctx)
	}
}

func (e *BinaryOpExpr) tryRewriteExpr(ctx *CheckCtx) {
	switch lexp := e.Left.(type) {
	case *NameExpr:
		if nexpr, have := ctx.GetNamedExpr(lexp.Data); have {
			e.Left = &FieldReferenceExpr{
				Name:      lexp,
				FieldExpr: nexpr,
			}
		}
	}
	switch rexp := e.Right.(type) {
	case *NameExpr:
		if nexpr, have := ctx.GetNamedExpr(rexp.Data); have {
			e.Right = &FieldReferenceExpr{
				Name:      rexp,
				FieldExpr: nexpr,
			}
		}
	}
}

func (e *BinaryOpExpr) checkWithAndOr(ctx *CheckCtx) error {
	op := OperatorToString[e.Op]
	switch exp := e.Left.(type) {
	case *BinaryOpExpr, *FunctionCallExpr, *NotExpr, *FieldReferenceExpr:
		if e.Left.ReturnType() != TBOOL {
			return NewSyntaxError(e.Left.GetPos(), "%s operator has wrong type of left expression %s", op, exp)
		}
	default:
		return NewSyntaxError(e.Left.GetPos(), "%s operator with invalid left expression %s", op, exp)
	}

	switch exp := e.Right.(type) {
	case *BinaryOpExpr, *FunctionCallExpr, *NotExpr, *FieldReferenceExpr:
		if exp.ReturnType() != TBOOL {
			return NewSyntaxError(e.Right.GetPos(), "%s operator has wrong type of right expression %s", op, exp)
		}
	default:
		return NewSyntaxError(e.Right.GetPos(), "%s operator with invalid right expression %s", op, exp)
	}
	return nil
}

func (e *BinaryOpExpr) checkWithMath(ctx *CheckCtx) error {
	op := OperatorToString[e.Op]
	lstring := false
	rstring := false
	switch exp := e.Left.(type) {
	case *BinaryOpExpr, *FunctionCallExpr, *NumberExpr, *FloatExpr, *FieldReferenceExpr:
		if e.Left.ReturnType() != TNUMBER {
			if e.Left.ReturnType() == TSTR {
				lstring = true
			} else {
				return NewSyntaxError(e.Left.GetPos(), "%s operator has wrong type of left expression %s", op, exp)
			}
		}
	case *StringExpr, *FieldExpr, *FieldAccessExpr:
		lstring = true
	default:
		return NewSyntaxError(e.Left.GetPos(), "%s operator with invalid left expression %s", op, exp)
	}

	switch exp := e.Right.(type) {
	case *BinaryOpExpr, *FunctionCallExpr, *NumberExpr, *FloatExpr, *FieldReferenceExpr:
		if e.Right.ReturnType() != TNUMBER {
			if e.Right.ReturnType() == TSTR {
				rstring = true
			} else {
				return NewSyntaxError(e.Right.GetPos(), "%s operator has wrong type of right expression %s", op, exp)
			}
		}
	case *StringExpr, *FieldExpr, *FieldAccessExpr:
		rstring = true
	default:
		return NewSyntaxError(e.Right.GetPos(), "%s operator with invalid right expression %s", op, exp)
	}

	if op == "+" && lstring && rstring {
	} else {
		if lstring {
			return NewSyntaxError(e.Left.GetPos(), "%s operator with invalid left expression %s", op, e.Left)
		}
		if rstring {
			return NewSyntaxError(e.Right.GetPos(), "%s operator with invalid right expression %s", op, e.Left)
		}
	}
	if op == "/" {
		switch rval := e.Right.(type) {
		case *NumberExpr:
			if rval.Int == 0 {
				return NewSyntaxError(e.Right.GetPos(), "/ operator divide by zero")
			}
		case *FloatExpr:
			if rval.Float == 0.0 {
				return NewSyntaxError(e.Right.GetPos(), "/ operator divide by zero")
			}
		}
	}
	return nil
}

func (e *BinaryOpExpr) checkWithCompares(ctx *CheckCtx) error {
	var (
		numKeyFieldExpr   = 0
		numValueFieldExpr = 0
		numCallExpr       = 0
	)
	op := OperatorToString[e.Op]

	switch exp := e.Left.(type) {
	case *FieldExpr:
		switch exp.Field {
		case KeyKW:
			numKeyFieldExpr++
		case ValueKW:
			numValueFieldExpr++
		}
	case *FunctionCallExpr, *FieldReferenceExpr:
		numCallExpr++
	case *StringExpr, *BoolExpr, *NumberExpr, *FloatExpr, *BinaryOpExpr, *FieldAccessExpr:
	default:
		return NewSyntaxError(e.Left.GetPos(), "%s operator with invalid left expression", op)
	}

	switch exp := e.Right.(type) {
	case *FieldExpr:
		switch exp.Field {
		case KeyKW:
			numKeyFieldExpr++
		case ValueKW:
			numValueFieldExpr++
		}
	case *FunctionCallExpr, *FieldReferenceExpr:
		numCallExpr++
	case *StringExpr, *BoolExpr, *NumberExpr, *FloatExpr, *BinaryOpExpr, *FieldAccessExpr:
	default:
		return NewSyntaxError(e.Right.GetPos(), "%s operator with invalid right expression", op)
	}

	if numKeyFieldExpr == 2 || numValueFieldExpr == 2 {
		return NewSyntaxError(e.GetPos(), "%s operator with two same field", op)
	}

	ltype := e.Left.ReturnType()
	rtype := e.Right.ReturnType()
	if ltype != rtype {
		return NewSyntaxError(e.GetPos(), "%s operator left and right type not same", op)
	}
	switch e.Op {
	case Gt, Gte, Lt, Lte:
		if ltype != TNUMBER && ltype != TSTR {
			return NewSyntaxError(e.Left.GetPos(), "%s operator has wrong type of left expression", op)
		}
	case PrefixMatch, RegExpMatch:
		if ltype != TSTR {
			return NewSyntaxError(e.Left.GetPos(), "%s operator has wrong type of left expression", op)
		}
	}
	return nil
}

func (e *BinaryOpExpr) checkWithIn(ctx *CheckCtx) error {
	ltype := e.Left.ReturnType()
	switch r := e.Right.(type) {
	case *ListExpr:
		for _, expr := range r.List {
			if expr.ReturnType() != ltype {
				return NewSyntaxError(expr.GetPos(), "in operator element has wrong type")
			}
		}
	case *FunctionCallExpr, *FieldReferenceExpr:
		if r.ReturnType() != TLIST {
			return NewSyntaxError(r.GetPos(), "in operator element has wrong type")
		}
	default:
		return NewSyntaxError(e.Right.GetPos(), "in operator right expression must be list expression")
	}
	return nil
}

func (e *BinaryOpExpr) checkWithBetween(ctx *CheckCtx) error {
	ltype := e.Left.ReturnType()
	rlist, ok := e.Right.(*ListExpr)
	if !ok || len(rlist.List) != 2 {
		return NewSyntaxError(e.Right.GetPos(), "between operator invalid right expression")
	}

	switch ltype {
	case TSTR, TNUMBER:
	default:
		return NewSyntaxError(e.Left.GetPos(), "between operator only support string and number type")
	}

	lexpr := rlist.List[0]
	uexpr := rlist.List[1]
	if lexpr.ReturnType() != ltype || uexpr.ReturnType() != ltype {
		return NewSyntaxError(e.Right.GetPos(), "between operator right expression with wrong type")
	}
	return nil
}

func (e *FieldExpr) Check(ctx *CheckCtx) error {
	if e.Field == KeyKW && ctx.NotAllowKey {
		return NewSyntaxError(e.Pos, "not allow key keyword in expression")
	}
	if e.Field == ValueKW && ctx.NotAllowValue {
		return NewSyntaxError(e.Pos, "not allow value keyword in expression")
	}
	return nil
}

func (e *StringExpr) Check(ctx *CheckCtx) error {
	return nil
}

func (e *NotExpr) Check(ctx *CheckCtx) error {
	if e.Right.ReturnType() != TBOOL {
		return NewSyntaxError(e.Right.GetPos(), "! operator right expression has wrong type")
	}
	return nil
}

func (e *FunctionCallExpr) Check(ctx *CheckCtx) error {
	_, ok := e.Name.(*NameExpr)
	if !ok {
		return NewSyntaxError(e.Name.GetPos(), "Invalid function name")
	}
	if len(e.Args) > 0 {
		for _, a := range e.Args {
			if err := a.Check(ctx); err != nil {
				return err
			}
		}
	}
	return nil
}

func (e *NameExpr) Check(ctx *CheckCtx) error {
	return nil
}

func (e *FloatExpr) Check(ctx *CheckCtx) error {
	return nil
}

func (e *NumberExpr) Check(ctx *CheckCtx) error {
	return nil
}

func (e *BoolExpr) Check(ctx *CheckCtx) error {
	return nil
}

func (e *ListExpr) Check(ctx *CheckCtx) error {
	if len(e.List) == 0 {
		return NewSyntaxError(e.GetPos(), "Empty list")
	}
	if len(e.List) > 1 {
		ftype := e.List[0].ReturnType()
		for i, item := range e.List[1:] {
			if item.ReturnType() != ftype {
				return NewSyntaxError(item.GetPos(), "List %d item has wrong type", i)
			}
		}
	}
	return nil
}

func (e *FieldAccessExpr) Check(ctx *CheckCtx) error {
	_, leftIsFAE := e.Left.(*FieldAccessExpr)
	lrType := e.Left.ReturnType()
	switch lrType {
	case TJSON, TLIST:
	default:
		if leftIsFAE {
			// Support cascade field access such as:
			// json(value)['x']['y']
			return nil
		}
		return NewSyntaxError(e.Left.GetPos(), "Field access expression left require JSON or List type")
	}
	switch e.FieldName.(type) {
	case *StringExpr:
		if lrType == TJSON {
			return nil
		} else if leftIsFAE {
			// Support cascade array index access such as:
			// json(value)['list'][1]
			return nil
		}
	case *NumberExpr:
		if lrType == TLIST {
			return nil
		} else if leftIsFAE {
			// Support cascade array index access such as:
			// json(value)['list'][1]
			return nil
		}
	}
	return NewSyntaxError(e.FieldName.GetPos(), "Invalid field name")
}

func (e *FieldReferenceExpr) Check(ctx *CheckCtx) error {
	return nil
}