esclient

type AggregationResponse struct {
	Buckets []struct {
		Key string `json:"key"`
	} `json:"buckets"`
}

type BoolQuery struct {
	Term         map[string]string             `json:"term,omitempty"`
	Regexp       map[string]TermQuery          `json:"regexp,omitempty"`
	Nested       *NestedQuery                  `json:"nested,omitempty"`
	BoolQuery    map[BoolQueryType][]BoolQuery `json:"bool,omitempty"`
	RangeQueries map[string]RangeQuery         `json:"range,omitempty"`
	MatchQueries map[string]MatchQuery         `json:"match,omitempty"`
}

type BoolQueryType string

type BulkIndexResponse struct {
	ID     string `json:"_id"`
	Result string `json:"result"`
	Status int    `json:"status"`
	Error  struct {
		Type   string `json:"type"`
		Reason string `json:"reason"`
		Cause  struct {
			Type   string `json:"type"`
			Reason string `json:"reason"`
		} `json:"caused_by"`
	} `json:"error"`
}

type BulkResponse struct {
	Errors bool               `json:"errors"`
	Items  []BulkResponseItem `json:"items"`
}

type BulkResponseItem struct {
	Index BulkIndexResponse `json:"index"`
}

type ElasticsearchClient interface {
	// PutTemplate creates index template
	PutTemplate(ctx context.Context, name string, template io.Reader) error
	// Bulk submits a bulk request
	Bulk(ctx context.Context, bulkBody io.Reader) (*BulkResponse, error)
	// AddDataToBulkBuffer creates bulk item from data, index and typ and adds it to bulkBody
	AddDataToBulkBuffer(bulkBody *bytes.Buffer, data []byte, index, typ string)
	// Index indexes data into storage
	Index(ctx context.Context, body io.Reader, index, typ string) error

	// Search searches data via /_search
	Search(ctx context.Context, query SearchBody, size int, indices ...string) (*SearchResponse, error)
	// MultiSearch searches data via /_msearch
	MultiSearch(ctx context.Context, queries []SearchBody) (*MultiSearchResponse, error)

	// Major version returns major ES version
	MajorVersion() int
}

type Hit struct {
	Source *json.RawMessage `json:"_source"`
}

type Hits struct {
	Total int   `json:"total"`
	Hits  []Hit `json:"hits"`
}

type MatchQuery struct {
	Query string `json:"query"`
}

type MultiSearchResponse struct {
	Responses []SearchResponse `json:"responses"`
}

type NestedQuery struct {
	Path  string `json:"path"`
	Query Query  `json:"query"`
}

type Order string

type Query struct {
	Term         *Terms                        `json:"term,omitempty"`
	RangeQueries map[string]RangeQuery         `json:"range,omitempty"`
	BoolQuery    map[BoolQueryType][]BoolQuery `json:"bool,omitempty"`
}

type RangeQuery struct {
	GTE interface{} `json:"gte"`
	LTE interface{} `json:"lte"`
}

type SearchBody struct {
	// indices are not in body, the ES client puts them to request path
	Indices        []string           `json:"-"`
	Aggregations   json.RawMessage    `json:"aggs,omitempty"`
	Query          *Query             `json:"query,omitempty"`
	Sort           []map[string]Order `json:"sort,omitempty"`
	Size           int                `json:"size"`
	TerminateAfter int                `json:"terminate_after"`
	SearchAfter    []interface{}      `json:"search_after,omitempty"`
}

type SearchResponse struct {
	Hits  Hits                           `json:"hits"`
	Aggs  map[string]AggregationResponse `json:"aggregations,omitempty"`
	Error *SearchResponseError           `json:"error,omitempty"`
}

type SearchResponseError struct {
	json.RawMessage
}

type TermQuery struct {
	Value string `json:"value"`
}

type Terms map[string]TermQuery