//  Copyright (c) 2014 Couchbase, Inc.
//  Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file
//  except in compliance with the License. You may obtain a copy of the License at
//    http://www.apache.org/licenses/LICENSE-2.0
//  Unless required by applicable law or agreed to in writing, software distributed under the
//  License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND,
//  either express or implied. See the License for the specific language governing permissions
//  and limitations under the License.
package main

import (
	"github.com/blevesearch/bleve"
)

const textFieldAnalyzer = "en"

func buildIndexMapping() *bleve.IndexMapping {

	nameMapping := bleve.NewDocumentMapping().
		AddFieldMapping(
		bleve.NewFieldMapping(
			"", "text", textFieldAnalyzer,
			true, true, true, true))

	descMapping := bleve.NewDocumentMapping().
		AddFieldMapping(
		bleve.NewFieldMapping(
			"", "text", textFieldAnalyzer,
			true, true, true, true)).
		AddFieldMapping(
		bleve.NewFieldMapping("descriptionLang", "text", "detect_lang",
			false, true, false, false))

	typeMapping := bleve.NewDocumentMapping().
		AddFieldMapping(
		bleve.NewFieldMapping(
			"", "text", "keyword",
			true, true, true, true))

	styleMapping := bleve.NewDocumentMapping().
		AddFieldMapping(
		bleve.NewFieldMapping(
			"", "text", "keyword",
			true, true, true, true))

	categoryMapping := bleve.NewDocumentMapping().
		AddFieldMapping(
		bleve.NewFieldMapping(
			"", "text", "keyword",
			true, true, true, true))

	beerMapping := bleve.NewDocumentMapping().
		AddSubDocumentMapping("name", nameMapping).
		AddSubDocumentMapping("description", descMapping).
		AddSubDocumentMapping("type", typeMapping).
		AddSubDocumentMapping("style", styleMapping).
		AddSubDocumentMapping("category", categoryMapping)

	breweryMapping := bleve.NewDocumentMapping().
		AddSubDocumentMapping("name", nameMapping).
		AddSubDocumentMapping("description", descMapping)

	indexMapping := bleve.NewIndexMapping().
		AddDocumentMapping("beer", beerMapping).
		AddDocumentMapping("brewery", breweryMapping)

	indexMapping.TypeField = "type"
	indexMapping.DefaultAnalyzer = textFieldAnalyzer

	return indexMapping
}