open FSharp.CosmosDb
open FSharp.Control
open Azure.Cosmos.Serialization
open System.Text
open Microsoft.FSharpLu.Json

module Serializer =
    type CompactCosmosSerializer() =
        inherit CosmosSerializer()

        let defaultEncoding = new UTF8Encoding(false, true)
        let jsonSerializerSettings = new JsonSerializerSettings()
        do jsonSerializerSettings.Converters <- [| CompactUnionJsonConverter(true) |]

        let serializer =
            JsonSerializer.Create(jsonSerializerSettings)

        override u.FromStream<'T>(stream: Stream): 'T =
            let returnCastedStream () = (box stream) :?> 'T

            let readStreamAndConvert () =
                use streamReader = new StreamReader(stream)
                use jsonTextReader = new JsonTextReader(streamReader)
                serializer.Deserialize<'T>(jsonTextReader)

            try
                if typeof<Stream>.IsAssignableFrom(typeof<'T>)
                then returnCastedStream ()
                else readStreamAndConvert ()
            finally
                stream.Dispose()

        override u.ToStream<'T>(input: 'T): Stream =
            let streamPayload = new MemoryStream()

            use streamWriter =
                new StreamWriter(streamPayload, encoding = defaultEncoding, bufferSize = 1024, leaveOpen = true)

            use writer = new JsonTextWriter(streamWriter)
            writer.Formatting <- Newtonsoft.Json.Formatting.None
            serializer.Serialize(writer, input)
            writer.Flush()
            streamWriter.Flush()

            streamPayload.Position <- 0L
            streamPayload :> Stream

module Domain = 
    type FamilyMember = Mum | Dad | Child
    type Person = { id: Guid; FirstName : string; LastName : string; FamilyType: FamilyMember }

module Programme = 
    [<EntryPoint>]
    let main argv =
        async {

            let person = { id = Guid.NewGuid(); FirstName = "Minecraft"; LastName = "Steve"; FamiltyType = Dad }

            let connString = "..."

            let op: Azure.Cosmos.CosmosClientOptions = Azure.Cosmos.CosmosClientOptions()
            op.Serializer <- Serializer.CompactCosmosSerializer()

            let insertPerson =
                Cosmos.fromConnectionStringWithOptions connString op
                |> Cosmos.database "Sample"
                |> Cosmos.container "People"
                |> Cosmos.insert<Person> person
                |> Cosmos.execAsync

            let person = insertPerson 
            do! person 
                |> AsyncSeq.iter (fun u -> printfn "%A" u)

        return 0 // return an integer exit code
    }
    |> Async.RunSynchronously