/* 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 System.Collections.Generic;
using System.Linq;
using OpenSearch.OpenSearch.Xunit.XunitPlumbing;
using OpenSearch.Net;
using FluentAssertions;
using OpenSearch.Client;
using Newtonsoft.Json.Linq;
using Tests.Core.Client;
using Tests.Core.Extensions;
using Tests.Core.ManagedOpenSearch.Clusters;
using Tests.Core.Xunit;
using Tests.Domain;
using Tests.Domain.Extensions;
using Tests.Framework.DocumentationTests;
using Tests.Framework.EndpointTests;
using Tests.Framework.EndpointTests.TestState;
using static Tests.Domain.Helpers.TestValueHelper;

namespace Tests.Document.Single.Index
{
	public class IndexApiTests
		: ApiIntegrationTestBase<WritableCluster, IndexResponse, IIndexRequest<Project>, IndexDescriptor<Project>, IndexRequest<Project>>
	{
		public IndexApiTests(WritableCluster cluster, EndpointUsage usage) : base(cluster, usage) { }

		protected override bool ExpectIsValid => true;

		protected override object ExpectJson =>
			new
			{
				name = CallIsolatedValue,
				type = Project.TypeName,
				join = Document.Join.ToAnonymousObject(),
				state = "Stable",
				visibility = "Public",
				startedOn = FixedDate,
				lastActivity = FixedDate,
				numberOfContributors = 0,
				sourceOnly = Dependant(null, new { notWrittenByDefaultSerializer = "written" }),
				curatedTags = new[] { new { name = "x", added = FixedDate } },
			};

		protected override int ExpectStatusCode => 201;

		protected override Func<IndexDescriptor<Project>, IIndexRequest<Project>> Fluent => s => s
			.WaitForActiveShards("1")
			.OpType(OpType.Index)
			.Refresh(Refresh.True)
			.Routing("route");

		protected override HttpMethod HttpMethod => HttpMethod.PUT;
		protected override bool IncludeNullInExpected => false;

		protected override IndexRequest<Project> Initializer =>
			new IndexRequest<Project>(Document)
			{
				Refresh = Refresh.True,
				OpType = OpType.Index,
				WaitForActiveShards = "1",
				Routing = "route"
			};

		protected override string UrlPath
			=> $"/project/_doc/{CallIsolatedValue}?wait_for_active_shards=1&op_type=index&refresh=true&routing=route";

		private Project Document => new Project
		{
			State = StateOfBeing.Stable,
			Name = CallIsolatedValue,
			StartedOn = FixedDate,
			LastActivity = FixedDate,
			CuratedTags = new List<Tag> { new Tag { Name = "x", Added = FixedDate } },
			SourceOnly = TestClient.Configuration.Random.SourceSerializer ? new SourceOnlyObject() : null
		};

		protected override LazyResponses ClientUsage() => Calls(
			(client, f) => client.Index(Document, f),
			(client, f) => client.IndexAsync(Document, f),
			(client, r) => client.Index(r),
			(client, r) => client.IndexAsync(r)
		);

		protected override IndexDescriptor<Project> NewDescriptor() => new IndexDescriptor<Project>(Document);
	}

	public class IndexIntegrationTests : IntegrationDocumentationTestBase, IClusterFixture<WritableCluster>
	{
		public IndexIntegrationTests(WritableCluster cluster) : base(cluster) { }

		[I] public void OpTypeCreate()
		{
			var indexName = RandomString();
			var project = Project.Generator.Generate(1).First();
			var indexResult = Client.Index(project, f => f
				.Index(indexName)
				.OpType(OpType.Create)
			);
			indexResult.ShouldBeValid();
			indexResult.ApiCall.HttpStatusCode.Should().Be(201);
			indexResult.Result.Should().Be(Result.Created);
			indexResult.Index.Should().Be(indexName);
			if (Cluster.ClusterConfiguration.Version < "2.0.0")
				indexResult.Type.Should().Be("_doc");
			indexResult.Id.Should().Be(project.Name);

			indexResult = Client.Index(project, f => f
				.Index(indexName)
				.OpType(OpType.Create)
			);

			indexResult.ShouldNotBeValid();
			indexResult.ApiCall.HttpStatusCode.Should().Be(409);
			indexResult.ServerError.Should().NotBeNull();
			indexResult.ServerError.Error.Type.Should().Contain("conflict");
			indexResult.ServerError.Status.Should().Be(409);
		}

		[I]
		public void Index()
		{
			var indexName = RandomString();
			var commitActivity = CommitActivity.CommitActivities.First();
			var indexResult = Client.Index(commitActivity, f => f.Index(indexName));
			indexResult.ShouldBeValid();
			indexResult.ApiCall.HttpStatusCode.Should().Be(201);
			indexResult.Result.Should().Be(Result.Created);
			indexResult.Index.Should().Be(indexName);
			if (Cluster.ClusterConfiguration.Version < "2.0.0")
				indexResult.Type.Should().Be("_doc");
			indexResult.Id.Should().Be(commitActivity.Id);
			indexResult.Version.Should().Be(1);
			indexResult.Shards.Should().NotBeNull();
			indexResult.Shards.Total.Should().BeGreaterOrEqualTo(1);
			indexResult.Shards.Successful.Should().BeGreaterOrEqualTo(1);
			indexResult.SequenceNumber.Should().BeGreaterOrEqualTo(0);
			indexResult.PrimaryTerm.Should().BeGreaterThan(0);

			indexResult = Client.Index(commitActivity, f => f.Index(indexName));

			indexResult.ShouldBeValid();
			indexResult.ApiCall.HttpStatusCode.Should().Be(200);
			indexResult.Version.Should().Be(2);
			indexResult.Result.Should().Be(Result.Updated);
		}
	}

	[JsonNetSerializerOnly]
	public class IndexJObjectIntegrationTests : IntegrationDocumentationTestBase, IClusterFixture<WritableCluster>
	{
		public IndexJObjectIntegrationTests(WritableCluster cluster) : base(cluster) { }

		[I]
		public void Index()
		{
			var index = RandomString();
			var jObjects = Enumerable.Range(1, 1000)
				.Select(i =>
					new JObject
					{
						{ "id", i },
						{ "name", $"name {i}" },
						{ "value", Math.Pow(i, 2) },
						{ "date", new DateTime(2016, 1, 1) },
						{
							"child", new JObject
							{
								{ "child_name", $"child_name {i}{i}" },
								{ "child_value", 3 }
							}
						}
					});

			var jObject = jObjects.First();

			var indexResult = Client.Index(jObject, f => f
				.Index(index)
				.Id(jObject["id"].Value<int>())
			);

			indexResult.ShouldBeValid();
			indexResult.ApiCall.HttpStatusCode.Should().Be(201);
			indexResult.Result.Should().Be(Result.Created);
			indexResult.Index.Should().Be(index);
			if (Cluster.ClusterConfiguration.Version < "2.0.0")
				indexResult.Type.Should().Be("_doc");
			indexResult.Shards.Should().NotBeNull();
			indexResult.Shards.Total.Should().BeGreaterOrEqualTo(1);
			indexResult.Shards.Successful.Should().BeGreaterOrEqualTo(1);
			indexResult.SequenceNumber.Should().BeGreaterOrEqualTo(0);
			indexResult.PrimaryTerm.Should().BeGreaterThan(0);

			var bulkResponse = Client.Bulk(b => b
				.Index(index)
				.IndexMany(jObjects.Skip(1), (bi, d) => bi
					.Document(d)
					.Id(d["id"].Value<int>())
				)
			);

			foreach (var item in bulkResponse.Items)
			{
				item.IsValid.Should().BeTrue();
				item.Status.Should().Be(201);
				item.Shards.Should().NotBeNull();
				item.Shards.Total.Should().BeGreaterOrEqualTo(1);
				item.Shards.Successful.Should().BeGreaterOrEqualTo(1);
				item.SequenceNumber.Should().BeGreaterOrEqualTo(0);
				item.PrimaryTerm.Should().BeGreaterThan(0);
			}
		}
	}

	public class IndexAnonymousTypesIntegrationTests : IntegrationDocumentationTestBase, IClusterFixture<WritableCluster>
	{
		public IndexAnonymousTypesIntegrationTests(WritableCluster cluster) : base(cluster) { }

		[I] public void Index()
		{
			var index = RandomString();
			var anonymousType = new
			{
				name = "name",
				value = 3,
				date = new DateTime(2016, 1, 1),
				child = new
				{
					child_name = "child_name",
					child_value = 3
				}
			};

			var indexResult = Client.Index(anonymousType, f => f
				.Index(index)
				.Id(anonymousType.name)
			);

			indexResult.ShouldBeValid();
			indexResult.ApiCall.HttpStatusCode.Should().Be(201);
			indexResult.Result.Should().Be(Result.Created);
			indexResult.Index.Should().Be(index);
			if (Cluster.ClusterConfiguration.Version < "2.0.0")
				indexResult.Type.Should().StartWith("_doc");
			indexResult.Shards.Should().NotBeNull();
			indexResult.Shards.Total.Should().BeGreaterOrEqualTo(1);
			indexResult.Shards.Successful.Should().BeGreaterOrEqualTo(1);
			indexResult.SequenceNumber.Should().BeGreaterOrEqualTo(0);
			indexResult.PrimaryTerm.Should().BeGreaterThan(0);
		}
	}
}