/* 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.IO;
using OpenSearch.OpenSearch.Xunit.XunitPlumbing;
using OpenSearch.Net;
using FluentAssertions;
using OpenSearch.Client;
using OpenSearch.Client.Specification.IngestApi;
using Tests.Core.Extensions;
using Tests.Core.ManagedOpenSearch.Clusters;
using Tests.Framework.EndpointTests;
using Tests.Framework.EndpointTests.TestState;

namespace Tests.Document.Single.Index
{
	public class TestDocument
	{
		static TestDocument()
		{
			using (var stream = typeof(TestDocument).Assembly.GetManifestResourceStream("Tests.Document.Single.Index.Attachment_Test_Document.pdf"))
			{
				using (var memoryStream = new MemoryStream())
				{
					stream.CopyTo(memoryStream);
					TestPdfDocument = Convert.ToBase64String(memoryStream.ToArray());
				}
			}
		}

		// Base 64 encoded version of Attachment_Test_Document.pdf
		public static string TestPdfDocument { get; }
	}

	public class IngestedAttachment
	{
		public Attachment Attachment { get; set; }
		public string Content { get; set; }
		public int Id { get; set; }
	}

	public class IndexIngestAttachmentApiTests
		: ApiIntegrationTestBase<IntrusiveOperationCluster, IndexResponse,
			IIndexRequest<IngestedAttachment>,
			IndexDescriptor<IngestedAttachment>,
			IndexRequest<IngestedAttachment>>
	{
		public IndexIngestAttachmentApiTests(IntrusiveOperationCluster cluster, EndpointUsage usage) : base(cluster, usage) { }

		protected override bool ExpectIsValid => true;
		protected override object ExpectJson => new { id = 1, content = Content };
		protected override int ExpectStatusCode => 201;

		protected override Func<IndexDescriptor<IngestedAttachment>, IIndexRequest<IngestedAttachment>> Fluent => s => s
			.Index(CallIsolatedValue)
			.Refresh(Refresh.True)
			.Pipeline(PipelineId);

		protected override HttpMethod HttpMethod => HttpMethod.PUT;

		protected override IndexRequest<IngestedAttachment> Initializer =>
			new IndexRequest<IngestedAttachment>(CallIsolatedValue, Id.From(Document))
			{
				Refresh = Refresh.True,
				Pipeline = PipelineId,
				Document = Document
			};

		protected override bool SupportsDeserialization => false;

		protected override string UrlPath
			=> $"/{CallIsolatedValue}/_doc/1?refresh=true&pipeline={PipelineId}";

		private static string Content => TestDocument.TestPdfDocument;

		private IngestedAttachment Document => new IngestedAttachment
		{
			Id = 1,
			Content = Content
		};

		private static string PipelineId { get; } = "pipeline-" + RandomString();

		protected override void IntegrationSetup(IOpenSearchClient client, CallUniqueValues values)
		{
			foreach (var value in values)
			{
				var index = value.Value;

				client.Indices.Create(index, c => c
					.Map<IngestedAttachment>(mm => mm
						.Properties(p => p
							.Text(s => s
								.Name(f => f.Content)
							)
							.Object<Attachment>(o => o
								.Name(f => f.Attachment)
							)
						)
					)
				);
			}

			client.Ingest.PutPipeline(new PutPipelineRequest(PipelineId)
			{
				Description = "Attachment pipeline test",
				Processors = new List<IProcessor>
				{
					new AttachmentProcessor
					{
						Field = "content",
						TargetField = "attachment"
					}
				}
			});
		}

		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<IngestedAttachment> NewDescriptor() => new IndexDescriptor<IngestedAttachment>(Document);

		protected override void ExpectResponse(IndexResponse response)
		{
			response.ShouldBeValid();

			var getResponse = Client.Get<IngestedAttachment>(response.Id, g => g.Index(CallIsolatedValue));

			getResponse.ShouldBeValid();
			getResponse.Source.Should().NotBeNull();

			getResponse.Source.Attachment.Should().NotBeNull();

			var attachment = getResponse.Source.Attachment;

			attachment.Title.Should().Be("Attachment Test Document");
			attachment.Keywords.Should().Be("osc,test,document");
			attachment.Date.Should().Be(new DateTime(2016, 12, 08, 3, 5, 13, DateTimeKind.Utc));
			attachment.ContentType.Should().Be("application/pdf");
			attachment.Author.Should().Be("Russ Cam");
			attachment.Language.Should().Be("fr");
			attachment.ContentLength.Should().Be(96);
			attachment.Content.Should().Contain("mapper-attachment support");
		}
	}

	public class ReindexAttachmentApiTests : IClusterFixture<IntrusiveOperationCluster>
	{
		private readonly IOpenSearchClient _client;

		private static string RandomString { get; } = Guid.NewGuid().ToString("N").Substring(0, 8);
		private static string Index { get; } = "project" + RandomString;
		private static string PipelineId { get; } = "pipeline-" + RandomString;
		private static string Content => TestDocument.TestPdfDocument;

		private IngestedAttachment Document => new IngestedAttachment
		{
			Id = 1,
			Content = Content
		};

		public ReindexAttachmentApiTests(IntrusiveOperationCluster cluster)
		{
			_client = cluster.Client;

			_client.Ingest.PutPipeline(new PutPipelineRequest(PipelineId)
			{
				Description = "Attachment pipeline test",
				Processors = new List<IProcessor>
				{
					new AttachmentProcessor
					{
						Field = "content",
						TargetField = "attachment"
					}
				}
			});

			var createIndexResponse = _client.Indices.Create(Index, c => c
				.Map<IngestedAttachment>(mm => mm
					.Properties(p => p
						.Text(s => s
							.Name(f => f.Content)
						)
						.Object<Attachment>(o => o
							.Name(f => f.Attachment)
						)
					)
				)
			);

			createIndexResponse.ShouldBeValid();
			var ndexResponse = _client.Index(Document, i => i
				.Index(Index)
				.Refresh(Refresh.True)
				.Pipeline(PipelineId)
			);

			ndexResponse.ShouldBeValid();
		}

		[I]
		public void ReindexIntoAnotherIndexCorrectlySerializesAttachment()
		{
			var getResponse = _client.Get<IngestedAttachment>(1, g => g
				.Index(Index)
			);

			getResponse.ShouldBeValid();
			var ingestedAttachment = getResponse.Source;
			ingestedAttachment.Should().NotBeNull();

			var ndexResponse = _client.Index(ingestedAttachment, i => i
				.Index(Index + "2")
				.Refresh(Refresh.True)
			);

			ndexResponse.ShouldBeValid();

			getResponse = _client.Get<IngestedAttachment>(1, g => g
				.Index(Index + "2")
			);

			getResponse.ShouldBeValid();
			ingestedAttachment.Should().NotBeNull();
			ingestedAttachment.Attachment.Title.Should().Be("Attachment Test Document");
			ingestedAttachment.Attachment.Keywords.Should().Be("osc,test,document");
			ingestedAttachment.Attachment.Date.Should().Be(new DateTime(2016, 12, 08, 3, 5, 13, DateTimeKind.Utc));
			ingestedAttachment.Attachment.ContentType.Should().Be("application/pdf");
			ingestedAttachment.Attachment.Author.Should().Be("Russ Cam");
			ingestedAttachment.Attachment.Language.Should().Be("fr");
			ingestedAttachment.Attachment.ContentLength.Should().Be(96);
		}
	}
}