using System;
using System.Collections.Generic;
using System.Threading.Tasks;
using Microsoft.Azure.Cosmos;
using Microsoft.Azure.Cosmos.Fluent;
using Microsoft.Azure.Cosmos.Scripts;
using Newtonsoft.Json;

namespace Cosmos409
{
    public class Program
    {
        public const string LocalConnectionString =
            "AccountEndpoint=https://localhost:8081/;AccountKey=C2y6yDjf5/R+ob0N8A7Cgv30VRDJIWEHLM+4QDU5DE2nQ9nDuVTqobD4b8mGGyPMbIZnqyMsEcaGQy67XIw/Jw==;";

        public const string DatabaseName = "test";
        public const string ContainerName = "jobs";
        public const string SetJobNumberTrigger = "SetJobNumber";

        public static async Task Main(string[] args)
        {
            var cosmosClient = new CosmosClientBuilder(LocalConnectionString).Build();

            try
            {
                await cosmosClient.GetDatabase(DatabaseName).DeleteAsync();
            } catch (CosmosException)
            {
                // data doesn't exist
            }

            var databaseResponse = await cosmosClient.CreateDatabaseAsync(DatabaseName, 400);
            var containerResponse = await databaseResponse.Database.DefineContainer(ContainerName, "/investigationKey")
                .WithUniqueKey().Path("/investigationKey").Attach()
                .WithUniqueKey().Path("/jobNumber").Attach()
                .CreateIfNotExistsAsync();

            await containerResponse.Container.Scripts.CreateTriggerAsync(new TriggerProperties
            {
                Id = SetJobNumberTrigger,
                TriggerType = TriggerType.Pre,
                TriggerOperation = TriggerOperation.Create,
                Body = @"function setJobNumber() {
                    var context = getContext();
                    var request = context.getRequest();      
                    var containerManager = context.getCollection();
                    var containerLink = containerManager.getSelfLink()

                    var documentToCreate = request.getBody();

                    var jobNumberQuery = ""SELECT VALUE MAX(r.jobNumber) from root r WHERE r.investigationKey = '"" + documentToCreate['investigationKey'] + ""'"";
                    containerManager.queryDocuments(containerLink,
                        jobNumberQuery,
                        function(err, countValue) {
                            if (err) throw new Error(err.message);
                            documentToCreate['jobNumber'] = (countValue.length > 0 ? countValue[0] : 0) + 1;
                        });

                    // update the document that will be created
                    request.setBody(documentToCreate);
                    }",
            });

            var job = new Job {Id = Guid.NewGuid(), InvestigationKey = "test"};

            await containerResponse.Container.CreateItemAsync(job, null,
                new ItemRequestOptions {PreTriggers = new List<string> {SetJobNumberTrigger}});

            job.Id = Guid.NewGuid();

            try
            {
                // this will currently fail due to the ill defined unique key, however, the 409 response doesn't indicate any reason at all
                await containerResponse.Container.CreateItemAsync(job, null,
                    new ItemRequestOptions {PreTriggers = new List<string> {SetJobNumberTrigger}});
            }
            catch (CosmosException e)
            {
                Console.WriteLine(e.Message);
            }

            Console.WriteLine("<3");
            Console.ReadLine();
        }
    }

    public class Job
    {
        [JsonProperty("id")]
        public Guid Id { get; set; }
        [JsonProperty("investigationKey")]
        public string InvestigationKey { get; set; }
        [JsonProperty("jobNumber")]
        public int JobNumber { get; set; }
    }
}