1 files changed, 130 insertions, 0 deletions
diff --git a/vendor/gopkg.in/olivere/elastic.v5/suggester_context_geo.go b/vendor/gopkg.in/olivere/elastic.v5/suggester_context_geo.go
new file mode 100644
index 000000000..6815bfe73
--- /dev/null
+++ b/vendor/gopkg.in/olivere/elastic.v5/suggester_context_geo.go
@@ -0,0 +1,130 @@
+// Copyright 2012-present Oliver Eilhard. All rights reserved.
+// Use of this source code is governed by a MIT-license.
+// See http://olivere.mit-license.org/license.txt for details.
+
+package elastic
+
+// -- SuggesterGeoMapping --
+
+// SuggesterGeoMapping provides a mapping for a geolocation context in a suggester.
+// See https://www.elastic.co/guide/en/elasticsearch/reference/5.2/suggester-context.html#_geo_location_mapping.
+type SuggesterGeoMapping struct {
+	name             string
+	defaultLocations []*GeoPoint
+	precision        []string
+	neighbors        *bool
+	fieldName        string
+}
+
+// NewSuggesterGeoMapping creates a new SuggesterGeoMapping.
+func NewSuggesterGeoMapping(name string) *SuggesterGeoMapping {
+	return &SuggesterGeoMapping{
+		name: name,
+	}
+}
+
+func (q *SuggesterGeoMapping) DefaultLocations(locations ...*GeoPoint) *SuggesterGeoMapping {
+	q.defaultLocations = append(q.defaultLocations, locations...)
+	return q
+}
+
+func (q *SuggesterGeoMapping) Precision(precision ...string) *SuggesterGeoMapping {
+	q.precision = append(q.precision, precision...)
+	return q
+}
+
+func (q *SuggesterGeoMapping) Neighbors(neighbors bool) *SuggesterGeoMapping {
+	q.neighbors = &neighbors
+	return q
+}
+
+func (q *SuggesterGeoMapping) FieldName(fieldName string) *SuggesterGeoMapping {
+	q.fieldName = fieldName
+	return q
+}
+
+// Source returns a map that will be used to serialize the context query as JSON.
+func (q *SuggesterGeoMapping) Source() (interface{}, error) {
+	source := make(map[string]interface{})
+
+	x := make(map[string]interface{})
+	source[q.name] = x
+
+	x["type"] = "geo"
+
+	if len(q.precision) > 0 {
+		x["precision"] = q.precision
+	}
+	if q.neighbors != nil {
+		x["neighbors"] = *q.neighbors
+	}
+
+	switch len(q.defaultLocations) {
+	case 0:
+	case 1:
+		x["default"] = q.defaultLocations[0].Source()
+	default:
+		var arr []interface{}
+		for _, p := range q.defaultLocations {
+			arr = append(arr, p.Source())
+		}
+		x["default"] = arr
+	}
+
+	if q.fieldName != "" {
+		x["path"] = q.fieldName
+	}
+	return source, nil
+}
+
+// -- SuggesterGeoQuery --
+
+// SuggesterGeoQuery provides querying a geolocation context in a suggester.
+// See https://www.elastic.co/guide/en/elasticsearch/reference/5.2/suggester-context.html#_geo_location_query
+type SuggesterGeoQuery struct {
+	name      string
+	location  *GeoPoint
+	precision []string
+}
+
+// NewSuggesterGeoQuery creates a new SuggesterGeoQuery.
+func NewSuggesterGeoQuery(name string, location *GeoPoint) *SuggesterGeoQuery {
+	return &SuggesterGeoQuery{
+		name:      name,
+		location:  location,
+		precision: make([]string, 0),
+	}
+}
+
+func (q *SuggesterGeoQuery) Precision(precision ...string) *SuggesterGeoQuery {
+	q.precision = append(q.precision, precision...)
+	return q
+}
+
+// Source returns a map that will be used to serialize the context query as JSON.
+func (q *SuggesterGeoQuery) Source() (interface{}, error) {
+	source := make(map[string]interface{})
+
+	if len(q.precision) == 0 {
+		if q.location != nil {
+			source[q.name] = q.location.Source()
+		}
+	} else {
+		x := make(map[string]interface{})
+		source[q.name] = x
+
+		if q.location != nil {
+			x["value"] = q.location.Source()
+		}
+
+		switch len(q.precision) {
+		case 0:
+		case 1:
+			x["precision"] = q.precision[0]
+		default:
+			x["precision"] = q.precision
+		}
+	}
+
+	return source, nil
+}