1 files changed, 51 insertions, 0 deletions
diff --git a/vendor/gopkg.in/olivere/elastic.v5/search_queries_script.go b/vendor/gopkg.in/olivere/elastic.v5/search_queries_script.go
new file mode 100644
index 000000000..664555b3e
--- /dev/null
+++ b/vendor/gopkg.in/olivere/elastic.v5/search_queries_script.go
@@ -0,0 +1,51 @@
+// 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
+
+import "errors"
+
+// ScriptQuery allows to define scripts as filters.
+//
+// For details, see
+// https://www.elastic.co/guide/en/elasticsearch/reference/5.2/query-dsl-script-query.html
+type ScriptQuery struct {
+	script    *Script
+	queryName string
+}
+
+// NewScriptQuery creates and initializes a new ScriptQuery.
+func NewScriptQuery(script *Script) *ScriptQuery {
+	return &ScriptQuery{
+		script: script,
+	}
+}
+
+// QueryName sets the query name for the filter that can be used
+// when searching for matched_filters per hit
+func (q *ScriptQuery) QueryName(queryName string) *ScriptQuery {
+	q.queryName = queryName
+	return q
+}
+
+// Source returns JSON for the query.
+func (q *ScriptQuery) Source() (interface{}, error) {
+	if q.script == nil {
+		return nil, errors.New("ScriptQuery expected a script")
+	}
+	source := make(map[string]interface{})
+	params := make(map[string]interface{})
+	source["script"] = params
+
+	src, err := q.script.Source()
+	if err != nil {
+		return nil, err
+	}
+	params["script"] = src
+
+	if q.queryName != "" {
+		params["_name"] = q.queryName
+	}
+	return source, nil
+}