/Raven.Database/Indexing/AnonymousObjectToLuceneDocumentConverter.cs

//-----------------------------------------------------------------------

// <copyright file="AnonymousObjectToLuceneDocumentConverter.cs" company="Hibernating Rhinos LTD">

//     Copyright (c) Hibernating Rhinos LTD. All rights reserved.

// </copyright>

//-----------------------------------------------------------------------

using System;

using System.Collections;

using System.Collections.Generic;

using System.ComponentModel;

using System.Globalization;

using System.IO;

using System.Linq;

using Lucene.Net.Documents;

using Raven.Abstractions.Extensions;

using Raven.Database.Linq;

using Raven.Imports.Newtonsoft.Json;

using Raven.Imports.Newtonsoft.Json.Linq;

using Raven.Abstractions;

using Raven.Abstractions.Data;

using Raven.Abstractions.Indexing;

using Raven.Abstractions.Linq;

using Raven.Database.Extensions;

using Raven.Json.Linq;



namespace Raven.Database.Indexing

{

	public class AnonymousObjectToLuceneDocumentConverter

	{

		private readonly AbstractViewGenerator viewGenerator;

		private readonly DocumentDatabase database;

		private readonly IndexDefinition indexDefinition;

		private readonly List<int> multipleItemsSameFieldCount = new List<int>();

		private readonly Dictionary<FieldCacheKey, Field> fieldsCache = new Dictionary<FieldCacheKey, Field>();

		private readonly Dictionary<FieldCacheKey, NumericField> numericFieldsCache = new Dictionary<FieldCacheKey, NumericField>();



		public AnonymousObjectToLuceneDocumentConverter(DocumentDatabase database, IndexDefinition indexDefinition, AbstractViewGenerator viewGenerator)

		{

			this.database = database;

			this.indexDefinition = indexDefinition;

			this.viewGenerator = viewGenerator;

		}



		public IEnumerable<AbstractField> Index(object val, PropertyDescriptorCollection properties, Field.Store defaultStorage)

		{

			return from property in properties.Cast<PropertyDescriptor>()

				   where property.Name != Constants.DocumentIdFieldName

				   from field in CreateFields(property.Name, property.GetValue(val), defaultStorage)

				   select field;

		}



		public IEnumerable<AbstractField> Index(RavenJObject document, Field.Store defaultStorage)

		{

			return from property in document

				   where property.Key != Constants.DocumentIdFieldName

				   from field in CreateFields(property.Key, GetPropertyValue(property.Value), defaultStorage)

				   select field;

		}



		private static object GetPropertyValue(RavenJToken property)

		{

			switch (property.Type)

			{

				case JTokenType.Array:

				case JTokenType.Object:

					return property.ToString(Formatting.None);

				default:

					return property.Value<object>();

			}

		}



		/// <summary>

		/// This method generate the fields for indexing documents in lucene from the values.

		/// Given a name and a value, it has the following behavior:

		/// * If the value is enumerable, index all the items in the enumerable under the same field name

		/// * If the value is null, create a single field with the supplied name with the unanalyzed value 'NULL_VALUE'

		/// * If the value is string or was set to not analyzed, create a single field with the supplied name

		/// * If the value is date, create a single field with millisecond precision with the supplied name

		/// * If the value is numeric (int, long, double, decimal, or float) will create two fields:

		///		1. with the supplied name, containing the numeric value as an unanalyzed string - useful for direct queries

		///		2. with the name: name +'_Range', containing the numeric value in a form that allows range queries

		/// </summary>

        public IEnumerable<AbstractField> CreateFields(string name, object value, Field.Store defaultStorage, bool nestedArray = false, Field.TermVector defaultTermVector = Field.TermVector.NO, Field.Index? analyzed = null)

		{

			if (string.IsNullOrWhiteSpace(name))

				throw new ArgumentException("Field must be not null, not empty and cannot contain whitespace", "name");



			if (char.IsLetter(name[0]) == false && name[0] != '_')

			{

				name = "_" + name;

			}



			if (viewGenerator.IsSpatialField(name))

				return viewGenerator.GetSpatialField(name).CreateIndexableFields(value);



            return CreateRegularFields(name, value, defaultStorage, nestedArray, defaultTermVector, analyzed);

		}



		private IEnumerable<AbstractField> CreateRegularFields(string name, object value, Field.Store defaultStorage, bool nestedArray = false, Field.TermVector defaultTermVector = Field.TermVector.NO, Field.Index? analyzed = null)

		{

            var fieldIndexingOptions = analyzed ?? indexDefinition.GetIndex(name, null);

			var storage = indexDefinition.GetStorage(name, defaultStorage);

			var termVector = indexDefinition.GetTermVector(name, defaultTermVector);



			if (fieldIndexingOptions == Field.Index.NO && storage == Field.Store.NO && termVector == Field.TermVector.NO)

			{

				yield break;

			}



			if (fieldIndexingOptions == Field.Index.NO && storage == Field.Store.NO)

			{

				fieldIndexingOptions = Field.Index.ANALYZED; // we have some sort of term vector, forcing index to be analyzed, then.

			}



			if (value == null)

			{

				yield return CreateFieldWithCaching(name, Constants.NullValue, storage,

								 Field.Index.NOT_ANALYZED_NO_NORMS, Field.TermVector.NO);

				yield break;

			}

			var attachmentFoIndexing = value as AttachmentForIndexing;

			if (attachmentFoIndexing != null)

			{

				if (database == null)

					throw new InvalidOperationException(

						"Cannot use attachment for indexing if the database parameter is null. This is probably a RavenDB bug");



				var attachment = database.GetStatic(attachmentFoIndexing.Key);

				if (attachment == null)

				{

					yield break;

				}



				var fieldWithCaching = CreateFieldWithCaching(name, string.Empty, Field.Store.NO, fieldIndexingOptions, termVector);

				

				if (database.TransactionalStorage.IsAlreadyInBatch)

				{

					var streamReader = new StreamReader(attachment.Data());

					fieldWithCaching.SetValue(streamReader);

				}

				else

				{

					// we are not in batch operation so we have to create it be able to read attachment's data

					database.TransactionalStorage.Batch(accessor =>

					{

						var streamReader = new StreamReader(attachment.Data());

						// we have to read it into memory because we after exiting the batch an attachment's data stream will be closed

						fieldWithCaching.SetValue(streamReader.ReadToEnd()); 

					});

				}



				yield return fieldWithCaching;

				yield break;

			}

			if (Equals(value, string.Empty))

			{

				yield return CreateFieldWithCaching(name, Constants.EmptyString, storage,

							 Field.Index.NOT_ANALYZED_NO_NORMS, Field.TermVector.NO);

				yield break;

			}

			var dynamicNullObject = value as DynamicNullObject;

			if (ReferenceEquals(dynamicNullObject, null) == false)

			{

				if (dynamicNullObject.IsExplicitNull)

				{

					var sortOptions = indexDefinition.GetSortOption(name);

				    if (sortOptions == null || sortOptions.Value == SortOptions.None || sortOptions.Value == SortOptions.String ||

				        sortOptions.Value == SortOptions.StringVal || sortOptions.Value == SortOptions.Custom)

				    {

                        yield return CreateFieldWithCaching(name, Constants.NullValue, storage,

                                                      Field.Index.NOT_ANALYZED_NO_NORMS, Field.TermVector.NO);

				    }



				    foreach (var field in CreateNumericFieldWithCaching(name, GetNullValueForSorting(sortOptions), storage, termVector))

                        yield return field;

                   

                 }

				yield break;

			}

			var boostedValue = value as BoostedValue;

			if (boostedValue != null)

			{

				foreach (var field in CreateFields(name, boostedValue.Value, storage, false, termVector))

				{

					field.Boost = boostedValue.Boost;

					field.OmitNorms = false;

					yield return field;

				}

				yield break;

			}





			var abstractField = value as AbstractField;

			if (abstractField != null)

			{

				yield return abstractField;

				yield break;

			}

			var bytes = value as byte[];

			if (bytes != null)

			{

				yield return CreateBinaryFieldWithCaching(name, bytes, storage, fieldIndexingOptions, termVector);

				yield break;

			}



			var itemsToIndex = value as IEnumerable;

			if (itemsToIndex != null && ShouldTreatAsEnumerable(itemsToIndex))

			{

                int count = 1;



                if (nestedArray == false)

                    yield return new Field(name + "_IsArray", "true", storage, Field.Index.NOT_ANALYZED_NO_NORMS, Field.TermVector.NO);



				foreach (var itemToIndex in itemsToIndex)

				{

				    if (!CanCreateFieldsForNestedArray(itemToIndex, fieldIndexingOptions))

				        continue;



				    multipleItemsSameFieldCount.Add(count++);

				    foreach (var field in CreateFields(name, itemToIndex, storage, nestedArray: true, defaultTermVector: defaultTermVector, analyzed: analyzed))

				        yield return field;



				    multipleItemsSameFieldCount.RemoveAt(multipleItemsSameFieldCount.Count - 1);

				}



			    yield break;

			}



			if (Equals(fieldIndexingOptions, Field.Index.NOT_ANALYZED) ||

				Equals(fieldIndexingOptions, Field.Index.NOT_ANALYZED_NO_NORMS))// explicitly not analyzed

			{

				// date time, time span and date time offset have the same structure fo analyzed and not analyzed.

				if (!(value is DateTime) && !(value is DateTimeOffset) && !(value is TimeSpan))

				{

					yield return CreateFieldWithCaching(name, value.ToString(), storage,

														indexDefinition.GetIndex(name, Field.Index.NOT_ANALYZED_NO_NORMS), termVector);

					yield break;

				}

			}

			if (value is string)

			{

				var index = indexDefinition.GetIndex(name, Field.Index.ANALYZED);

				yield return CreateFieldWithCaching(name, value.ToString(), storage, index, termVector);

				yield break;

			}



			if (value is TimeSpan)

			{

				var val = (TimeSpan)value;

				yield return CreateFieldWithCaching(name, val.ToString("c", CultureInfo.InvariantCulture), storage,

						   indexDefinition.GetIndex(name, Field.Index.NOT_ANALYZED_NO_NORMS), termVector);

			}

			else if (value is DateTime)

			{

				var val = (DateTime)value;

				var dateAsString = val.ToString(Default.DateTimeFormatsToWrite, CultureInfo.InvariantCulture);

				if (val.Kind == DateTimeKind.Utc)

					dateAsString += "Z";

				yield return CreateFieldWithCaching(name, dateAsString, storage,

						   indexDefinition.GetIndex(name, Field.Index.NOT_ANALYZED_NO_NORMS), termVector);

			}

			else if (value is DateTimeOffset)

			{

				var val = (DateTimeOffset)value;



				string dtoStr;

				if (Equals(fieldIndexingOptions, Field.Index.NOT_ANALYZED) || Equals(fieldIndexingOptions, Field.Index.NOT_ANALYZED_NO_NORMS))

				{

					dtoStr = val.ToString(Default.DateTimeOffsetFormatsToWrite, CultureInfo.InvariantCulture);

				}

				else

				{

					dtoStr = val.UtcDateTime.ToString(Default.DateTimeFormatsToWrite, CultureInfo.InvariantCulture) + "Z";

				}

				yield return CreateFieldWithCaching(name, dtoStr, storage,

						   indexDefinition.GetIndex(name, Field.Index.NOT_ANALYZED_NO_NORMS), termVector);

			}

			else if (value is bool)

			{

				yield return new Field(name, ((bool)value) ? "true" : "false", storage,

							  indexDefinition.GetIndex(name, Field.Index.NOT_ANALYZED_NO_NORMS), termVector);



			}

			else if (value is double)

			{

				var d = (double)value;

				yield return CreateFieldWithCaching(name, d.ToString("r", CultureInfo.InvariantCulture), storage,

							   indexDefinition.GetIndex(name, Field.Index.NOT_ANALYZED_NO_NORMS), termVector);

			}

			else if (value is decimal)

			{

				var d = (decimal)value;

				var s = d.ToString(CultureInfo.InvariantCulture);

				if (s.Contains('.'))

				{

					s = s.TrimEnd('0');

					if (s.EndsWith("."))

						s = s.Substring(0, s.Length - 1);

				}

				yield return CreateFieldWithCaching(name, s, storage,

									   indexDefinition.GetIndex(name, Field.Index.NOT_ANALYZED_NO_NORMS), termVector);

			}

			else if (value is IConvertible) // we need this to store numbers in invariant format, so JSON could read them

			{

				var convert = ((IConvertible)value);

				yield return CreateFieldWithCaching(name, convert.ToString(CultureInfo.InvariantCulture), storage,

									   indexDefinition.GetIndex(name, Field.Index.NOT_ANALYZED_NO_NORMS), termVector);

			}

			else if (value is IDynamicJsonObject)

			{

				var inner = ((IDynamicJsonObject)value).Inner;

				yield return CreateFieldWithCaching(name + "_ConvertToJson", "true", Field.Store.YES, Field.Index.NOT_ANALYZED_NO_NORMS, Field.TermVector.NO);

				yield return CreateFieldWithCaching(name, inner.ToString(Formatting.None), storage,

									   indexDefinition.GetIndex(name, Field.Index.NOT_ANALYZED_NO_NORMS), termVector);

			}

			else

			{

				var jsonVal = RavenJToken.FromObject(value).ToString(Formatting.None);

				if (jsonVal.StartsWith("{") || jsonVal.StartsWith("["))

					yield return CreateFieldWithCaching(name + "_ConvertToJson", "true", Field.Store.YES, Field.Index.NOT_ANALYZED_NO_NORMS, Field.TermVector.NO);

				else if (jsonVal.StartsWith("\"") && jsonVal.EndsWith("\"") && jsonVal.Length > 1)

					jsonVal = jsonVal.Substring(1, jsonVal.Length - 2);

				yield return CreateFieldWithCaching(name, jsonVal, storage,

									   indexDefinition.GetIndex(name, Field.Index.NOT_ANALYZED_NO_NORMS), termVector);

			}





			foreach (var numericField in CreateNumericFieldWithCaching(name, value, storage, termVector))

				yield return numericField;

		}



	    private static object GetNullValueForSorting(SortOptions? sortOptions)

	    {

	        switch (sortOptions)

	        {

	            case SortOptions.Short:

	            case SortOptions.Int:

	                return int.MinValue;

	            case SortOptions.Double:

	                return double.MinValue;

	                break;

	            case SortOptions.Float:

	                return float.MinValue;



// ReSharper disable RedundantCaseLabel

	            case SortOptions.Long:



	            // to be able to sort on timestamps

	            case SortOptions.String:

	            case SortOptions.StringVal:

	            case SortOptions.None:

	            case SortOptions.Custom:

// ReSharper restore RedundantCaseLabel

	            default:

	                return long.MinValue;

	        }

	    }



	    private IEnumerable<AbstractField> CreateNumericFieldWithCaching(string name, object value,

			Field.Store defaultStorage, Field.TermVector termVector)

		{



			var fieldName = name + "_Range";

			var storage = indexDefinition.GetStorage(name, defaultStorage);

			var cacheKey = new FieldCacheKey(name, null, storage, termVector, multipleItemsSameFieldCount.ToArray());

			NumericField numericField;

			if (numericFieldsCache.TryGetValue(cacheKey, out numericField) == false)

			{

				numericFieldsCache[cacheKey] = numericField = new NumericField(fieldName, storage, true);

			}



			if (value is TimeSpan)

			{

				yield return numericField.SetLongValue(((TimeSpan)value).Ticks);

			}

			else if (value is int)

			{

				if (indexDefinition.GetSortOption(name) == SortOptions.Long)

					yield return numericField.SetLongValue((int)value);

                else if (indexDefinition.GetSortOption(name) == SortOptions.Float)

                    yield return numericField.SetFloatValue((int)value);

                else if (indexDefinition.GetSortOption(name) == SortOptions.Double)

					yield return numericField.SetDoubleValue((int)value);

				else

					yield return numericField.SetIntValue((int)value);

			}

			else if (value is long)

			{

				if (indexDefinition.GetSortOption(name) == SortOptions.Double)

					yield return numericField.SetDoubleValue((long)value);

                else if (indexDefinition.GetSortOption(name) == SortOptions.Float)

                    yield return numericField.SetFloatValue((long)value);

                else if (indexDefinition.GetSortOption(name) == SortOptions.Int)

                    yield return numericField.SetIntValue(Convert.ToInt32((long)value));

                else

					yield return numericField.SetLongValue((long)value);

			}

			else if (value is decimal)

			{

                if (indexDefinition.GetSortOption(name) == SortOptions.Float)

                    yield return numericField.SetFloatValue(Convert.ToSingle((decimal)value));

                else if (indexDefinition.GetSortOption(name) == SortOptions.Int)

                    yield return numericField.SetIntValue(Convert.ToInt32((decimal)value));

                else if (indexDefinition.GetSortOption(name) == SortOptions.Long)

                    yield return numericField.SetLongValue(Convert.ToInt64((decimal) value));

                else

                    yield return numericField.SetDoubleValue((double)(decimal)value);

			}

			else if (value is float)

			{

				if (indexDefinition.GetSortOption(name) == SortOptions.Double)

					yield return numericField.SetDoubleValue((float)value);

                else if (indexDefinition.GetSortOption(name) == SortOptions.Int)

                    yield return numericField.SetIntValue(Convert.ToInt32((float)value));

                else if (indexDefinition.GetSortOption(name) == SortOptions.Long)

                    yield return numericField.SetLongValue(Convert.ToInt64((float)value));

                else

					yield return numericField.SetFloatValue((float)value);

			}

			else if (value is double)

			{

                if (indexDefinition.GetSortOption(name) == SortOptions.Float)

                    yield return numericField.SetFloatValue(Convert.ToSingle((double)value));

                else if (indexDefinition.GetSortOption(name) == SortOptions.Int)

                    yield return numericField.SetIntValue(Convert.ToInt32((double)value));

                else if (indexDefinition.GetSortOption(name) == SortOptions.Long)

                    yield return numericField.SetLongValue(Convert.ToInt64((double)value));

                else

    			    yield return numericField.SetDoubleValue((double)value);

			}

		}



		public static bool ShouldTreatAsEnumerable(object itemsToIndex)

		{

			if (itemsToIndex == null)

				return false;



			if (itemsToIndex is DynamicJsonObject)

				return false;



			if (itemsToIndex is string)

				return false;



			if (itemsToIndex is RavenJObject)

				return false;



			if (itemsToIndex is IDictionary)

				return false;



			return true;

		}



		private Field CreateBinaryFieldWithCaching(string name, byte[] value, Field.Store store, Field.Index index, Field.TermVector termVector)

		{

			if (value.Length > 1024)

				throw new ArgumentException("Binary values must be smaller than 1Kb");



			var cacheKey = new FieldCacheKey(name, null, store, termVector, multipleItemsSameFieldCount.ToArray());

			Field field;

			var stringWriter = new StringWriter();

			JsonExtensions.CreateDefaultJsonSerializer().Serialize(stringWriter, value);

			var sb = stringWriter.GetStringBuilder();

			sb.Remove(0, 1); // remove prefix "

			sb.Remove(sb.Length - 1, 1); // remove postfix "

			var val = sb.ToString();



			if (fieldsCache.TryGetValue(cacheKey, out field) == false)

			{

				fieldsCache[cacheKey] = field = new Field(name, val, store, index, termVector);

			}

			field.SetValue(val);

			field.Boost = 1;

			field.OmitNorms = true;

			return field;

		}



		public class FieldCacheKey

		{

			private readonly string name;

			private readonly Field.Index? index;

			private readonly Field.Store store;

			private readonly Field.TermVector termVector;

			private readonly int[] multipleItemsSameField;



			public FieldCacheKey(string name, Field.Index? index, Field.Store store, Field.TermVector termVector, int[] multipleItemsSameField)

			{

				this.name = name;

				this.index = index;

				this.store = store;

				this.termVector = termVector;

				this.multipleItemsSameField = multipleItemsSameField;

			}





			protected bool Equals(FieldCacheKey other)

			{

				return string.Equals(name, other.name) &&

					Equals(index, other.index) &&

					Equals(store, other.store) &&

					Equals(termVector, other.termVector) &&

					multipleItemsSameField.SequenceEqual(other.multipleItemsSameField);

			}



			public override bool Equals(object obj)

			{

				if (ReferenceEquals(null, obj)) return false;

				if (ReferenceEquals(this, obj)) return true;

				if (obj.GetType() != typeof(FieldCacheKey)) return false;

				return Equals((FieldCacheKey)obj);

			}



			public override int GetHashCode()

			{

				unchecked

				{

					int hashCode = (name != null ? name.GetHashCode() : 0);

					hashCode = (hashCode * 397) ^ (index != null ? index.GetHashCode() : 0);

					hashCode = (hashCode * 397) ^ store.GetHashCode();

					hashCode = (hashCode * 397) ^ termVector.GetHashCode();

					hashCode = multipleItemsSameField.Aggregate(hashCode, (h, x) => h * 397 ^ x);

					return hashCode;

				}

			}

		}



		private Field CreateFieldWithCaching(string name, string value, Field.Store store, Field.Index index, Field.TermVector termVector)

		{

			var cacheKey = new FieldCacheKey(name, index, store, termVector, multipleItemsSameFieldCount.ToArray());

			Field field;



			if (fieldsCache.TryGetValue(cacheKey, out field) == false)

				fieldsCache[cacheKey] = field = new Field(name, value, store, index, termVector);



			field.SetValue(value);

			field.Boost = 1;

			field.OmitNorms = true;

			return field;

		}



		private bool CanCreateFieldsForNestedArray(object value, Field.Index fieldIndexingOptions)

		{

			if (!fieldIndexingOptions.IsAnalyzed())

			{

				return true;

			}



			if (value == null || value is DynamicNullObject)

			{

				return false;

			}



			return true;

		}

	}

}