mirror of https://github.com/hashicorp/consul
132 lines
2.7 KiB
Go
132 lines
2.7 KiB
Go
package bexpr
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"strings"
|
|
)
|
|
|
|
// TODO - Probably should make most of what is in here un-exported
|
|
|
|
//go:generate pigeon -o grammar.go -optimize-parser grammar.peg
|
|
//go:generate goimports -w grammar.go
|
|
|
|
type Expression interface {
|
|
ExpressionDump(w io.Writer, indent string, level int)
|
|
}
|
|
|
|
type UnaryOperator int
|
|
|
|
const (
|
|
UnaryOpNot UnaryOperator = iota
|
|
)
|
|
|
|
func (op UnaryOperator) String() string {
|
|
switch op {
|
|
case UnaryOpNot:
|
|
return "Not"
|
|
default:
|
|
return "UNKNOWN"
|
|
}
|
|
}
|
|
|
|
type BinaryOperator int
|
|
|
|
const (
|
|
BinaryOpAnd BinaryOperator = iota
|
|
BinaryOpOr
|
|
)
|
|
|
|
func (op BinaryOperator) String() string {
|
|
switch op {
|
|
case BinaryOpAnd:
|
|
return "And"
|
|
case BinaryOpOr:
|
|
return "Or"
|
|
default:
|
|
return "UNKNOWN"
|
|
}
|
|
}
|
|
|
|
type MatchOperator int
|
|
|
|
const (
|
|
MatchEqual MatchOperator = iota
|
|
MatchNotEqual
|
|
MatchIn
|
|
MatchNotIn
|
|
MatchIsEmpty
|
|
MatchIsNotEmpty
|
|
)
|
|
|
|
func (op MatchOperator) String() string {
|
|
switch op {
|
|
case MatchEqual:
|
|
return "Equal"
|
|
case MatchNotEqual:
|
|
return "Not Equal"
|
|
case MatchIn:
|
|
return "In"
|
|
case MatchNotIn:
|
|
return "Not In"
|
|
case MatchIsEmpty:
|
|
return "Is Empty"
|
|
case MatchIsNotEmpty:
|
|
return "Is Not Empty"
|
|
default:
|
|
return "UNKNOWN"
|
|
}
|
|
}
|
|
|
|
type MatchValue struct {
|
|
Raw string
|
|
Converted interface{}
|
|
}
|
|
|
|
type UnaryExpression struct {
|
|
Operator UnaryOperator
|
|
Operand Expression
|
|
}
|
|
|
|
type BinaryExpression struct {
|
|
Left Expression
|
|
Operator BinaryOperator
|
|
Right Expression
|
|
}
|
|
|
|
type Selector []string
|
|
|
|
func (sel Selector) String() string {
|
|
return strings.Join([]string(sel), ".")
|
|
}
|
|
|
|
type MatchExpression struct {
|
|
Selector Selector
|
|
Operator MatchOperator
|
|
Value *MatchValue
|
|
}
|
|
|
|
func (expr *UnaryExpression) ExpressionDump(w io.Writer, indent string, level int) {
|
|
localIndent := strings.Repeat(indent, level)
|
|
fmt.Fprintf(w, "%s%s {\n", localIndent, expr.Operator.String())
|
|
expr.Operand.ExpressionDump(w, indent, level+1)
|
|
fmt.Fprintf(w, "%s}\n", localIndent)
|
|
}
|
|
|
|
func (expr *BinaryExpression) ExpressionDump(w io.Writer, indent string, level int) {
|
|
localIndent := strings.Repeat(indent, level)
|
|
fmt.Fprintf(w, "%s%s {\n", localIndent, expr.Operator.String())
|
|
expr.Left.ExpressionDump(w, indent, level+1)
|
|
expr.Right.ExpressionDump(w, indent, level+1)
|
|
fmt.Fprintf(w, "%s}\n", localIndent)
|
|
}
|
|
|
|
func (expr *MatchExpression) ExpressionDump(w io.Writer, indent string, level int) {
|
|
switch expr.Operator {
|
|
case MatchEqual, MatchNotEqual, MatchIn, MatchNotIn:
|
|
fmt.Fprintf(w, "%[1]s%[3]s {\n%[2]sSelector: %[4]v\n%[2]sValue: %[5]q\n%[1]s}\n", strings.Repeat(indent, level), strings.Repeat(indent, level+1), expr.Operator.String(), expr.Selector, expr.Value.Raw)
|
|
default:
|
|
fmt.Fprintf(w, "%[1]s%[3]s {\n%[2]sSelector: %[4]v\n%[1]s}\n", strings.Repeat(indent, level), strings.Repeat(indent, level+1), expr.Operator.String(), expr.Selector)
|
|
}
|
|
}
|