Initial QSfera import

Server/vendor/github.com/CiscoM31/godata/select_parser.go

@@ -0,0 +1,97 @@
+package godata
+
+import (
+	"context"
+	"errors"
+	"strings"
+)
+
+type SelectItem struct {
+	Segments []*Token
+}
+
+func ParseSelectString(ctx context.Context, sel string) (*GoDataSelectQuery, error) {
+	return GlobalExpressionParser.ParseSelectString(ctx, sel)
+}
+
+func (p *ExpressionParser) ParseSelectString(ctx context.Context, sel string) (*GoDataSelectQuery, error) {
+	items := strings.Split(sel, ",")
+
+	result := []*SelectItem{}
+
+	for _, item := range items {
+		item = strings.TrimSpace(item)
+
+		cfg, hasComplianceConfig := ctx.Value(odataCompliance).(OdataComplianceConfig)
+		if !hasComplianceConfig {
+			// Strict ODATA compliance by default.
+			cfg = ComplianceStrict
+		}
+
+		if len(strings.TrimSpace(item)) == 0 && cfg&ComplianceIgnoreInvalidComma == 0 {
+			return nil, BadRequestError("Extra comma in $select.")
+		}
+
+		if _, err := p.tokenizer.Tokenize(ctx, item); err != nil {
+			switch e := err.(type) {
+			case *GoDataError:
+				return nil, &GoDataError{
+					ResponseCode: e.ResponseCode,
+					Message:      "Invalid $select value",
+					Cause:        e,
+				}
+			default:
+				return nil, &GoDataError{
+					ResponseCode: 500,
+					Message:      "Invalid $select value",
+					Cause:        e,
+				}
+			}
+		} else {
+			segments := []*Token{}
+			for _, val := range strings.Split(item, "/") {
+				segments = append(segments, &Token{Value: val})
+			}
+			result = append(result, &SelectItem{segments})
+		}
+	}
+
+	return &GoDataSelectQuery{result, sel}, nil
+}
+
+func SemanticizeSelectQuery(sel *GoDataSelectQuery, service *GoDataService, entity *GoDataEntityType) error {
+	if sel == nil {
+		return nil
+	}
+
+	newItems := []*SelectItem{}
+
+	// replace wildcards with every property of the entity
+	for _, item := range sel.SelectItems {
+		// TODO: allow multiple path segments
+		if len(item.Segments) > 1 {
+			return NotImplementedError("Multiple path segments in select clauses are not yet supported.")
+		}
+
+		if item.Segments[0].Value == "*" {
+			for _, prop := range service.PropertyLookup[entity] {
+				newItems = append(newItems, &SelectItem{[]*Token{{Value: prop.Name}}})
+			}
+		} else {
+			newItems = append(newItems, item)
+		}
+	}
+
+	sel.SelectItems = newItems
+
+	for _, item := range sel.SelectItems {
+		if prop, ok := service.PropertyLookup[entity][item.Segments[0].Value]; ok {
+			item.Segments[0].SemanticType = SemanticTypeProperty
+			item.Segments[0].SemanticReference = prop
+		} else {
+			return errors.New("Entity " + entity.Name + " has no property " + item.Segments[0].Value)
+		}
+	}
+
+	return nil
+}