// ==========================================================================
//  Squidex Headless CMS
// ==========================================================================
//  Copyright (c) Squidex UG (haftungsbeschraenkt)
//  All rights reserved. Licensed under the MIT license.
// ==========================================================================

using Newtonsoft.Json;
using Newtonsoft.Json.Linq;
using Squidex.ClientLibrary;
using Squidex.ClientLibrary.Utils;
using TestSuite.Model;

#pragma warning disable SA1300 // Element should begin with upper-case letter

namespace TestSuite.ApiTests
{
    public class ContentQueryTests : IClassFixture<ContentQueryFixture>
    {
        public ContentQueryFixture _ { get; }

        public ContentQueryTests(ContentQueryFixture fixture)
        {
            _ = fixture;
        }

        [Fact]
        public async Task Should_query_newly_created_schema()
        {
            for (var i = 0; i < 20; i++)
            {
                var schemaName = $"schema-{Guid.NewGuid()}";

                await TestEntity.CreateSchemaAsync(_.Schemas, _.AppName, schemaName);

                var contentClient = _.ClientManager.CreateContentsClient<TestEntity, TestEntityData>(schemaName);
                var contentItems = await contentClient.GetAsync();

                Assert.Equal(0, contentItems.Total);
            }
        }

        [Fact]
        public async Task Should_query_by_ids()
        {
            var q = new ContentQuery { OrderBy = "data/number/iv asc" };

            var items = await _.Contents.GetAsync(q);

            var itemsById = await _.Contents.GetAsync(new HashSet<string>(items.Items.Take(3).Select(x => x.Id)));

            Assert.Equal(3, itemsById.Items.Count);
            Assert.Equal(3, itemsById.Total);

            foreach (var item in itemsById.Items)
            {
                Assert.Equal(_.AppName, item.AppName);
                Assert.Equal(_.SchemaName, item.SchemaName);
            }
        }

        [Fact]
        public async Task Should_return_all_with_odata()
        {
            var q = new ContentQuery { OrderBy = "data/number/iv asc" };

            var items = await _.Contents.GetAsync(q);

            AssertItems(items, 10, new[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 });
        }

        [Fact]
        public async Task Should_return_all_with_json()
        {
            var q = new ContentQuery
            {
                JsonQuery = new
                {
                    sort = new[]
                    {
                        new
                        {
                            path = "data.number.iv"
                        }
                    }
                }
            };

            var items = await _.Contents.GetAsync(q);

            AssertItems(items, 10, new[] { 1, 2, 3, 4, 5, 6, 7, 8, 9, 10 });
        }

        [Fact]
        public async Task Should_return_items_by_skip_with_odata()
        {
            var q = new ContentQuery { OrderBy = "data/number/iv asc", Skip = 5 };

            var items = await _.Contents.GetAsync(q);

            AssertItems(items, 10, new[] { 6, 7, 8, 9, 10 });
        }

        [Fact]
        public async Task Should_return_items_by_skip_with_json()
        {
            var q = new ContentQuery
            {
                JsonQuery = new
                {
                    sort = new[]
                    {
                        new
                        {
                            path = "data.number.iv"
                        }
                    },
                    skip = 5
                }
            };

            var items = await _.Contents.GetAsync(q);

            AssertItems(items, 10, new[] { 6, 7, 8, 9, 10 });
        }

        [Fact]
        public async Task Should_return_items_by_skip_and_top_with_odata()
        {
            var q = new ContentQuery { Skip = 2, Top = 5, OrderBy = "data/number/iv asc" };

            var items = await _.Contents.GetAsync(q);

            AssertItems(items, 10, new[] { 3, 4, 5, 6, 7 });
        }

        [Fact]
        public async Task Should_return_items_by_skip_and_top_with_json()
        {
            var q = new ContentQuery
            {
                JsonQuery = new
                {
                    skip = 2,
                    sort = new[]
                    {
                        new
                        {
                            path = "data.number.iv"
                        }
                    },
                    top = 5
                }
            };

            var items = await _.Contents.GetAsync(q);

            AssertItems(items, 10, new[] { 3, 4, 5, 6, 7 });
        }

        [Fact]
        public async Task Should_return_items_by_filter_with_odata()
        {
            var q = new ContentQuery { Filter = "data/number/iv gt 3 and data/number/iv lt 7", OrderBy = "data/number/iv asc" };

            var items = await _.Contents.GetAsync(q);

            AssertItems(items, 3, new[] { 4, 5, 6 });
        }

        [Fact]
        public async Task Should_return_items_by_filter_with_json()
        {
            var q = new ContentQuery
            {
                JsonQuery = new
                {
                    sort = new[]
                    {
                        new
                        {
                            path = "data.number.iv"
                        }
                    },
                    filter = new
                    {
                        and = new[]
                        {
                            new
                            {
                                path = "data.number.iv",
                                op = "gt",
                                value = 3
                            },
                            new
                            {
                                path = "data.number.iv",
                                op = "lt",
                                value = 7
                            }
                        }
                    }
                }
            };

            var items = await _.Contents.GetAsync(q);

            AssertItems(items, 3, new[] { 4, 5, 6 });
        }

        [Fact]
        public async Task Should_return_items_by_json_filter_with_json()
        {
            var q = new ContentQuery
            {
                JsonQuery = new
                {
                    sort = new[]
                    {
                        new
                        {
                            path = "data.json.iv.nested1.nested2"
                        }
                    },
                    filter = new
                    {
                        and = new[]
                        {
                            new
                            {
                                path = "data.json.iv.nested1.nested2",
                                op = "gt",
                                value = 3
                            },
                            new
                            {
                                path = "data.json.iv.nested1.nested2",
                                op = "lt",
                                value = 7
                            }
                        }
                    }
                }
            };

            var items = await _.Contents.GetAsync(q);

            AssertItems(items, 3, new[] { 4, 5, 6 });
        }

        [Fact]
        public async Task Should_return_items_by_full_text_with_odata()
        {
            var q = new ContentQuery { Search = "2" };

            var items = await _.Contents.WaitForContentAsync(q, x => true, TimeSpan.FromSeconds(30));

            AssertItems(items, 1, new[] { 2 });
        }

        [Fact]
        public async Task Should_return_items_by_full_text_with_json()
        {
            var q = new ContentQuery
            {
                JsonQuery = new
                {
                    fullText = "2"
                }
            };

            var items = await _.Contents.WaitForContentAsync(q, x => true, TimeSpan.FromSeconds(30));

            AssertItems(items, 1, new[] { 2 });
        }

        [Fact]
        public async Task Should_return_items_by_near_location_with_odata()
        {
            var q = new ContentQuery { Filter = "geo.distance(data/geo/iv, geography'POINT(3 3)') lt 1000" };

            var items = await _.Contents.WaitForContentAsync(q, x => true, TimeSpan.FromSeconds(30));

            AssertItems(items, 1, new[] { 3 });
        }

        [Fact]
        public async Task Should_return_items_by_near_location_with_json()
        {
            var q = new ContentQuery
            {
                JsonQuery = new
                {
                    filter = new
                    {
                        path = "data.geo.iv",
                        op = "lt",
                        value = new
                        {
                            longitude = 3,
                            latitude = 3,
                            distance = 1000
                        }
                    }
                }
            };

            var items = await _.Contents.WaitForContentAsync(q, x => true, TimeSpan.FromSeconds(30));

            AssertItems(items, 1, new[] { 3 });
        }

        [Fact]
        public async Task Should_return_items_by_near_geoson_location_with_odata()
        {
            var q = new ContentQuery { Filter = "geo.distance(data/geo/iv, geography'POINT(4 4)') lt 1000" };

            var items = await _.Contents.WaitForContentAsync(q, x => true, TimeSpan.FromSeconds(30));

            AssertItems(items, 1, new[] { 4 });
        }

        [Fact]
        public async Task Should_query_json_with_dot()
        {
            TestEntity content = null;
            try
            {
                // STEP 1: Create a content item with a text that caused a bug before.
                content = await _.Contents.CreateAsync(new TestEntityData
                {
                    Json = new JObject
                    {
                        ["search.field.with.dot"] = 42
                    }
                }, ContentCreateOptions.AsPublish);


                // STEP 2: Get the item and ensure that the text is the same.
                var q = new ContentQuery
                {
                    JsonQuery = new
                    {
                        filter = new
                        {
                            and = new[]
                            {
                                new
                                {
                                    path = "data.json.iv.search\\.field\\.with\\.dot",
                                    op = "eq",
                                    value = 42
                                }
                            }
                        }
                    }
                };

                var queried = await _.Contents.GetAsync(q);

                Assert.Equal(42, (int)queried.Items[0].Data.Json["search.field.with.dot"]);
            }
            finally
            {
                if (content != null)
                {
                    await _.Contents.DeleteAsync(content.Id);
                }
            }
        }

        [Fact]
        public async Task Should_return_items_by_near_geoson_location_with_json()
        {
            var q = new ContentQuery
            {
                JsonQuery = new
                {
                    filter = new
                    {
                        path = "data.geo.iv",
                        op = "lt",
                        value = new
                        {
                            longitude = 4,
                            latitude = 4,
                            distance = 1000
                        }
                    }
                }
            };

            var items = await _.Contents.WaitForContentAsync(q, x => true, TimeSpan.FromSeconds(30));

            AssertItems(items, 1, new[] { 4 });
        }

        [Fact]
        public async Task Should_create_and_query_with_inline_graphql()
        {
            var query = new
            {
                query = @"
                    mutation {
                        createMyReadsContent(data: {
                            number: {
                                iv: 555
                            }
                        }) {
                            id,
                            data {
                                number {
                                    iv
                                }
                            }
                        }
                    }"
            };

            var result = await _.Contents.GraphQlAsync<JObject>(query);

            var value = result["createMyReadsContent"]["data"]["number"]["iv"].Value<int>();

            Assert.Equal(555, value);
        }

        [Fact]
        public async Task Should_return_items_by_ids()
        {
            var q0 = new ContentQuery { Filter = "data/number/iv gt 3 and data/number/iv lt 7", OrderBy = "data/number/iv asc" };

            var items_0 = await _.Contents.GetAsync(q0);

            var q1 = new ContentQuery
            {
                JsonQuery = new
                {
                    sort = new[]
                    {
                        new
                        {
                            path = "data.number.iv"
                        }
                    },
                    filter = new
                    {
                        or = items_0.Items.Select(x => new
                        {
                            path = "id",
                            op = "eq",
                            value = x.Id
                        }).ToArray()
                    }
                }
            };

            var items_1 = await _.Contents.GetAsync(q1);

            AssertItems(items_0, 3, new[] { 4, 5, 6 });
            AssertItems(items_1, 3, new[] { 4, 5, 6 });
        }

        [Fact]
        public async Task Should_create_and_query_with_variable_graphql()
        {
            var query = new
            {
                query = @"
                    mutation Mutation($data: MyReadsDataInputDto!) {
                        createMyReadsContent(data: $data) {
                            id,
                            data {
                                number {
                                    iv
                                }
                            }
                        }
                    }",
                variables = new
                {
                    data = new
                    {
                        number = new
                        {
                            iv = 998
                        }
                    }
                }
            };

            var result = await _.Contents.GraphQlAsync<JObject>(query);

            var value = result["createMyReadsContent"]["data"]["number"]["iv"].Value<int>();

            Assert.Equal(998, value);
        }

        [Fact]
        public async Task Should_batch_query_items_with_graphql()
        {
            var query1 = new
            {
                query = @"
                    query ContentsQuery($filter: String!) {
                        queryMyReadsContents(filter: $filter, orderby: ""data/number/iv asc"") {
                            id,
                            data {
                                number {
                                    iv
                                }
                            }
                        }
                    }",
                variables = new
                {
                    filter = @"data/number/iv gt 3 and data/number/iv lt 7"
                }
            };

            var query2 = new
            {
                query = @"
                    query ContentsQuery($filter: String!) {
                        queryMyReadsContents(filter: $filter, orderby: ""data/number/iv asc"") {
                            id,
                            data {
                                number {
                                    iv
                                }
                            }
                        }
                    }",
                variables = new
                {
                    filter = @"data/number/iv gt 4 and data/number/iv lt 7"
                }
            };

            var results = await _.Contents.GraphQlAsync<QueryResult>(new[] { query1, query2 });

            var items1 = results.ElementAt(0).Data.Items;
            var items2 = results.ElementAt(1).Data.Items;

            Assert.Equal(items1.Select(x => x.Data.Number).ToArray(), new[] { 4, 5, 6 });
            Assert.Equal(items2.Select(x => x.Data.Number).ToArray(), new[] { 5, 6 });
        }

        [Fact]
        public async Task Should_query_items_with_graphql()
        {
            var query = new
            {
                query = @"
                    query ContentsQuery($filter: String!) {
                        queryMyReadsContents(filter: $filter, orderby: ""data/number/iv asc"") {
                            id,
                            data {
                                number {
                                    iv
                                }
                            }
                        }
                    }",
                variables = new
                {
                    filter = @"data/number/iv gt 3 and data/number/iv lt 7"
                }
            };

            var result = await _.Contents.GraphQlAsync<QueryResult>(query);

            var items = result.Items;

            Assert.Equal(items.Select(x => x.Data.Number).ToArray(), new[] { 4, 5, 6 });
        }

        [Fact]
        public async Task Should_query_items_with_graphql_get()
        {
            var query = new
            {
                query = @"
                    query ContentsQuery($filter: String!) {
                        queryMyReadsContents(filter: $filter, orderby: ""data/number/iv asc"") {
                            id,
                            data {
                                number {
                                    iv
                                }
                            }
                        }
                    }",
                variables = new
                {
                    filter = @"data/number/iv gt 3 and data/number/iv lt 7"
                }
            };

            var result = await _.Contents.GraphQlGetAsync<QueryResult>(query);

            var items = result.Items;

            Assert.Equal(items.Select(x => x.Data.Number).ToArray(), new[] { 4, 5, 6 });
        }

        [Fact]
        public async Task Should_query_items_with_graphql_with_dynamic()
        {
            var query = new
            {
                query = @"
                {
                    queryMyReadsContents(filter: ""data/number/iv gt 3 and data/number/iv lt 7"", orderby: ""data/number/iv asc"") {
                      id,
                      data {
                        number {
                          iv
                        }
                      }
                    }
                }"
            };

            var result = await _.Contents.GraphQlAsync<JObject>(query);

            var items = result["queryMyReadsContents"];

            Assert.Equal(items.Select(x => x["data"]["number"]["iv"].Value<int>()).ToArray(), new[] { 4, 5, 6 });
        }

        [Fact]
        public async Task Should_query_items_complex_search()
        {
            var query = new
            {
                query = @"
                    query ContentsQuery($search: String!) {
                        queryMyReadsContents(search: $search) {
                            id,
                            data {
                                number {
                                    iv
                                }
                            }
                        }
                    }",
                variables = new
                {
                    search = @"The answer is 42"
                }
            };

            await _.Contents.GraphQlAsync<QueryResult>(query);
        }

        [Fact]
        public async Task Should_return_correct_content_type_for_graphql()
        {
            var query = new
            {
                query = @"
                    query ContentsQuery($filter: String!) {
                        queryMyReadsContents(filter: $filter, orderby: ""data/number/iv asc"") {
                            id,
                            data {
                                number {
                                    iv
                                }
                            }
                        }
                    }",
                variables = new
                {
                    filter = @"data/number/iv gt 3 and data/number/iv lt 7"
                }
            };

            using (var client = _.ClientManager.CreateHttpClient())
            {
                var response = await client.PostAsync(_.ClientManager.GenerateUrl($"api/content/{_.AppName}/graphql/batch"), query.ToContent());

                Assert.Equal("application/json", response.Content.Headers.ContentType.MediaType);

                var result = await response.Content.ReadAsJsonAsync<GraphQlResponse<QueryResult>>();

                Assert.Equal(result.Data.Items.Select(x => x.Data.Number).ToArray(), new[] { 4, 5, 6 });
            }
        }

        private sealed class QueryResult
        {
            [JsonProperty("queryMyReadsContents")]
            public QueryItem[] Items { get; set; }
        }

        private sealed class QueryItem
        {
            public Guid Id { get; set; }

            public QueryItemData Data { get; set; }
        }

        private sealed class QueryItemData
        {
            [JsonConverter(typeof(InvariantConverter))]
            public int Number { get; set; }
        }

        private static void AssertItems(ContentsResult<TestEntity, TestEntityData> entities, int total, int[] expected)
        {
            Assert.Equal(total, entities.Total);
            Assert.Equal(expected, entities.Items.Select(x => x.Data.Number).ToArray());
        }
    }
}