/* SPDX-License-Identifier: Apache-2.0
*
* The OpenSearch Contributors require contributions made to
* this file be licensed under the Apache-2.0 license or a
* compatible open source license.
*/
/*
* Modifications Copyright OpenSearch Contributors. See
* GitHub history for details.
*
*  Licensed to Elasticsearch B.V. under one or more contributor
*  license agreements. See the NOTICE file distributed with
*  this work for additional information regarding copyright
*  ownership. Elasticsearch B.V. licenses this file to you 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.
*/

using System;
using OpenSearch.OpenSearch.Xunit.XunitPlumbing;
using OpenSearch.Net;
using OpenSearch.Client;
using OpenSearch.Client.Specification.IndicesApi;
using Tests.Core.ManagedOpenSearch.Clusters;
using Tests.Domain;
using Tests.Framework.EndpointTests;
using Tests.Framework.EndpointTests.TestState;

namespace Tests.Indices.MappingManagement.PutMapping
{
	public class PutMappingApiTests
		: ApiIntegrationAgainstNewIndexTestBase
			<WritableCluster, PutMappingResponse, IPutMappingRequest, PutMappingDescriptor<Project>, PutMappingRequest<Project>>
	{
		public PutMappingApiTests(WritableCluster cluster, EndpointUsage usage) : base(cluster, usage) { }

		protected override bool ExpectIsValid => true;

		protected override object ExpectJson { get; } = new
		{
			properties = new
			{
				branches = new
				{
					fields = new
					{
						keyword = new
						{
							type = "keyword",
							ignore_above = 256
						}
					},
					type = "text"
				},
				curatedTags = new
				{
					properties = new
					{
						added = new { type = "date" },
						name = new { type = "text" }
					},
					type = "object"
				},
				type = new { type = "text" },
				dateString = new { type = "text" },
				description = new { type = "text" },
				join = new
				{
					relations = new { project = "commits" },
					type = "join"
				},
				lastActivity = new { type = "date" },
				leadDeveloper = new
				{
					properties = new
					{
						firstName = new { type = "text" },
						gender = new { type = "keyword" },
						id = new { type = "long" },
						ipAddress = new { type = "text" },
						jobTitle = new { type = "text" },
						lastName = new { type = "text" },
						location = new { type = "geo_point" },
						nickname = new { type = "text" },
						geoIp = new { type = "object" }
					},
					type = "object"
				},
				locationPoint = new
				{
					properties = new
					{
						lat = new { type = "double" },
						lon = new { type = "double" }
					},
					type = "object"
				},
				locationShape = new
				{
					type = "geo_shape"
				},
				metadata = new { type = "object" },
				name = new
				{
					index = false,
					type = "text"
				},
				numberOfCommits = new { type = "integer" },
				numberOfContributors = new { type = "integer" },
				sourceOnly = new { properties = new { }, type = "object" },
				startedOn = new { type = "date" },
				state = new { type = "keyword" },
				visibility = new { type = "keyword" },
				suggest = new { type = "completion" },
				ranges = new
				{
					properties = new
					{
						dates = new { type = "date_range" },
						doubles = new { type = "double_range" },
						floats = new { type = "float_range" },
						integers = new { type = "integer_range" },
						longs = new { type = "long_range" },
						ips = new { type = "ip_range" }
					},
					type = "object"
				},
				requiredBranches = new
				{
					type = "integer"
				},
				tags = new
				{
					properties = new
					{
						added = new { type = "date" },
						name = new { type = "text" }
					},
					type = "object"
				},
				rank = new
				{
					type = "rank_feature"
				},
				versionControl = new
				{
					type = "keyword"
				},
				vector = new
				{
					type = "knn_vector",
					dimension = 2,
					method = new {
						name = "hnsw",
						space_type = "l2",
						engine = "nmslib",
						parameters = new
						{
							ef_construction = 128,
							m = 24
						}
					}
				}
			}
		};

		protected override int ExpectStatusCode => 200;

		protected override Func<PutMappingDescriptor<Project>, IPutMappingRequest> Fluent => d => d
			.Index(CallIsolatedValue)
			.AutoMap()
			.Properties(prop => prop
				.Join(join => join
					.Name(p => p.Join)
					.Relations(relations => relations
						.Join<Project, CommitActivity>()
					)
				)
				.Object<Tag>(o => o
					.Name(p => p.CuratedTags)
					.AutoMap()
					.Properties(ps => ps
						.Text(t => t
							.Name(tag => tag.Name)
						)
					)
				)
				.Text(t => t.Name(p => p.Description))
				.Text(t => t.Name(p => p.DateString))
				.Text(t => t.Name(p => p.Type))
				.Text(s => s
					.Name(p => p.Name)
					.Index(false)
				)
				.Object<Developer>(o => o
					.Name(p => p.LeadDeveloper)
					.AutoMap()
					.Properties(ps => ps
						.Text(t => t.Name(dv => dv.FirstName))
						.Text(t => t.Name(dv => dv.IpAddress))
						.Text(t => t.Name(dv => dv.JobTitle))
						.Text(t => t.Name(dv => dv.LastName))
						.Text(t => t.Name(dv => dv.OnlineHandle))
						.Object<GeoIp>(t => t.Name(dv => dv.GeoIp))
					)
				)
				.Object<object>(o => o
					.Name(p => p.Metadata)
				)
				.Object<Tag>(o => o
					.AutoMap()
					.Name(p => p.Tags)
					.Properties(ps => ps
						.Text(t => t
							.Name(tag => tag.Name)
						)
					)
				)
				.RankFeature(rf => rf
					.Name(p => p.Rank)
				)
				.Keyword(k => k
					.Name(n => n.VersionControl)
				)
				.KnnVector(k => k
					.Name(p => p.Vector)
					.Dimension(2)
					.Method(m => m
						.Name("hnsw")
						.SpaceType("l2")
						.Engine("nmslib")
						.Parameters(p => p
							.Parameter("ef_construction", 128)
							.Parameter("m", 24)
						)
					)
				)
			);

		protected override HttpMethod HttpMethod => HttpMethod.PUT;

		protected override PutMappingRequest<Project> Initializer => new PutMappingRequest<Project>(CallIsolatedValue)
		{
			Properties = new Properties<Project>
			{
				{
					p => p.Join, new JoinProperty
					{
						Relations = new Relations
						{
							{ typeof(Project), typeof(CommitActivity) }
						}
					}
				},
				{
					p => p.Branches, new TextProperty
					{
						Fields = new Properties
						{
							{
								"keyword", new KeywordProperty
								{
									IgnoreAbove = 256
								}
							}
						}
					}
				},
				{
					p => p.CuratedTags, new ObjectProperty
					{
						Properties = new Properties<Tag>
						{
							{ p => p.Added, new DateProperty() },
							{ p => p.Name, new TextProperty() },
						}
					}
				},
				{ p => p.Description, new TextProperty() },
				{ p => p.DateString, new TextProperty() },
				{ p => p.Type, new TextProperty() },
				{ p => p.LastActivity, new DateProperty() },
				{
					p => p.LeadDeveloper, new ObjectProperty
					{
						Properties = new Properties<Developer>
						{
							{ p => p.FirstName, new TextProperty() },
							{ p => p.Gender, new KeywordProperty() },
							{ p => p.Id, new NumberProperty(NumberType.Long) },
							{ p => p.IpAddress, new TextProperty() },
							{ p => p.JobTitle, new TextProperty() },
							{ p => p.LastName, new TextProperty() },
							{ p => p.Location, new GeoPointProperty() },
							{ p => p.OnlineHandle, new TextProperty() },
							{ p => p.GeoIp, new ObjectProperty() },
						}
					}
				},
				{
					p => p.LocationPoint, new ObjectProperty
					{
						Properties = new Properties<SimpleGeoPoint>
						{
							{ p => p.Lat, new NumberProperty(NumberType.Double) },
							{ p => p.Lon, new NumberProperty(NumberType.Double) },
						}
					}
				},
				{ p => p.LocationShape, new GeoShapeProperty() },
				{ p => p.Metadata, new ObjectProperty() },
				{ p => p.Name, new TextProperty { Index = false } },
				{ p => p.NumberOfCommits, new NumberProperty(NumberType.Integer) },
				{ p => p.NumberOfContributors, new NumberProperty(NumberType.Integer) },
				{
					p => p.SourceOnly, new ObjectProperty()
					{
						Properties = new Properties()
					}
				},
				{ p => p.StartedOn, new DateProperty() },
				{ p => p.State, new KeywordProperty() },
				{ p => p.Visibility, new KeywordProperty() },
				{ p => p.Suggest, new CompletionProperty() },
				{
					p => p.Ranges, new ObjectProperty
					{
						Properties = new Properties<Ranges>
						{
							{ p => p.Dates, new DateRangeProperty() },
							{ p => p.Doubles, new DoubleRangeProperty() },
							{ p => p.Floats, new FloatRangeProperty() },
							{ p => p.Integers, new IntegerRangeProperty() },
							{ p => p.Longs, new LongRangeProperty() },
							{ p => p.Ips, new IpRangeProperty() },
						}
					}
				},
				{ p => p.RequiredBranches, new NumberProperty(NumberType.Integer) },
				{
					p => p.Tags, new ObjectProperty
					{
						Properties = new Properties<Tag>
						{
							{ p => p.Added, new DateProperty() },
							{ p => p.Name, new TextProperty() },
						}
					}
				},
				{ p => p.Rank, new RankFeatureProperty() },
				{ p => p.VersionControl, new KeywordProperty() },
				{ p => p.Vector, new KnnVectorProperty
				{
					Dimension = 2,
					Method = new KnnMethod
					{
						Name = "hnsw",
						SpaceType = "l2",
						Engine = "nmslib",
						Parameters = new KnnMethodParameters
						{
							{"ef_construction", 128},
							{"m", 24}
						}
					}
				} }
			}
		};

		protected override string UrlPath => $"/{CallIsolatedValue}/_mapping";

		protected override LazyResponses ClientUsage() => Calls(
			(client, f) => client.Map(f),
			(client, f) => client.MapAsync(f),
			(client, r) => client.Map(r),
			(client, r) => client.MapAsync(r)
		);
	}
}