// Copyright 2012-2015 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

// TopHitsAggregation keeps track of the most relevant document
// being aggregated. This aggregator is intended to be used as a
// sub aggregator, so that the top matching documents
// can be aggregated per bucket.
//
// It can effectively be used to group result sets by certain fields via
// a bucket aggregator. One or more bucket aggregators determines by
// which properties a result set get sliced into.
//
// See: http://www.elasticsearch.org/guide/en/elasticsearch/reference/current/search-aggregations-metrics-top-hits-aggregation.html
type TopHitsAggregation struct {
	searchSource *SearchSource
}

func NewTopHitsAggregation() TopHitsAggregation {
	a := TopHitsAggregation{
		searchSource: NewSearchSource(),
	}
	return a
}

func (a TopHitsAggregation) From(from int) TopHitsAggregation {
	a.searchSource = a.searchSource.From(from)
	return a
}

func (a TopHitsAggregation) Size(size int) TopHitsAggregation {
	a.searchSource = a.searchSource.Size(size)
	return a
}

func (a TopHitsAggregation) TrackScores(trackScores bool) TopHitsAggregation {
	a.searchSource = a.searchSource.TrackScores(trackScores)
	return a
}

func (a TopHitsAggregation) Explain(explain bool) TopHitsAggregation {
	a.searchSource = a.searchSource.Explain(explain)
	return a
}

func (a TopHitsAggregation) Version(version bool) TopHitsAggregation {
	a.searchSource = a.searchSource.Version(version)
	return a
}

func (a TopHitsAggregation) NoFields() TopHitsAggregation {
	a.searchSource = a.searchSource.NoFields()
	return a
}

func (a TopHitsAggregation) FetchSource(fetchSource bool) TopHitsAggregation {
	a.searchSource = a.searchSource.FetchSource(fetchSource)
	return a
}

func (a TopHitsAggregation) FetchSourceContext(fetchSourceContext *FetchSourceContext) TopHitsAggregation {
	a.searchSource = a.searchSource.FetchSourceContext(fetchSourceContext)
	return a
}

func (a TopHitsAggregation) FieldDataFields(fieldDataFields ...string) TopHitsAggregation {
	a.searchSource = a.searchSource.FieldDataFields(fieldDataFields...)
	return a
}

func (a TopHitsAggregation) FieldDataField(fieldDataField string) TopHitsAggregation {
	a.searchSource = a.searchSource.FieldDataField(fieldDataField)
	return a
}

func (a TopHitsAggregation) ScriptFields(scriptFields ...*ScriptField) TopHitsAggregation {
	a.searchSource = a.searchSource.ScriptFields(scriptFields...)
	return a
}

func (a TopHitsAggregation) ScriptField(scriptField *ScriptField) TopHitsAggregation {
	a.searchSource = a.searchSource.ScriptField(scriptField)
	return a
}

func (a TopHitsAggregation) PartialFields(partialFields ...*PartialField) TopHitsAggregation {
	a.searchSource = a.searchSource.PartialFields(partialFields...)
	return a
}

func (a TopHitsAggregation) PartialField(partialField *PartialField) TopHitsAggregation {
	a.searchSource = a.searchSource.PartialField(partialField)
	return a
}

func (a TopHitsAggregation) Sort(field string, ascending bool) TopHitsAggregation {
	a.searchSource = a.searchSource.Sort(field, ascending)
	return a
}

func (a TopHitsAggregation) SortWithInfo(info SortInfo) TopHitsAggregation {
	a.searchSource = a.searchSource.SortWithInfo(info)
	return a
}

func (a TopHitsAggregation) SortBy(sorter ...Sorter) TopHitsAggregation {
	a.searchSource = a.searchSource.SortBy(sorter...)
	return a
}

func (a TopHitsAggregation) Highlight(highlight *Highlight) TopHitsAggregation {
	a.searchSource = a.searchSource.Highlight(highlight)
	return a
}

func (a TopHitsAggregation) Highlighter() *Highlight {
	return a.searchSource.Highlighter()
}

func (a TopHitsAggregation) Source() interface{} {
	// Example:
	// {
	//   "aggs": {
	//       "top_tag_hits": {
	//           "top_hits": {
	//               "sort": [
	//                   {
	//                       "last_activity_date": {
	//                           "order": "desc"
	//                       }
	//                   }
	//               ],
	//               "_source": {
	//                   "include": [
	//                       "title"
	//                   ]
	//               },
	//               "size" : 1
	//           }
	//       }
	//   }
	// }
	// This method returns only the { "top_hits" : { ... } } part.

	source := make(map[string]interface{})
	source["top_hits"] = a.searchSource.Source()
	return source
}