using System;

using System.Collections;

using System.Collections.Generic;

using System.Linq;

using System.Text;

using System.Runtime.Serialization.Formatters;

using System.IO;

using System.Data;

using System.ComponentModel.DataAnnotations;

using System.Globalization;

using System.Reflection;

using System.Threading;

​

using Newtonsoft.Json;

using Newtonsoft.Json.Linq;

using Newtonsoft.Json.Converters;

using Newtonsoft.Json.Serialization;    

​

using NUnit.Framework; // Throws NUnit.Framework.AssertionException

​

    // https://stackoverflow.com/questions/29651433/json-net-limit-maxdepth-when-serializing

​

    public struct ObjectId

    {

        static int lastId = 1;

        int id;

​

        public static ObjectId Create()

        {

            return new ObjectId(Interlocked.Increment(ref lastId));

        }

​

        public ObjectId(int id)

        {

            this.id = id - 1;

        }

​

        [JsonProperty]

        public int Id { get { return id + 1; } private set { id = value - 1; } }

​

        [JsonIgnore]

        public ObjectId Next { get { return new ObjectId(Id + 1); } }

    }

​

    public class MaxDepthJsonTextWriter : JsonTextWriter

    {

        public int? MaxDepth { get; set; }

        public int MaxObservedDepth { get; private set; }

​

        public MaxDepthJsonTextWriter(TextWriter writer, JsonSerializerSettings settings)

            : base(writer)

        {

            this.MaxDepth = (settings == null ? null : settings.MaxDepth);

            this.MaxObservedDepth = 0;

        }

​

        public MaxDepthJsonTextWriter(TextWriter writer, int? maxDepth)

            : base(writer)

        {

            this.MaxDepth = maxDepth;

        }

​

        public override void WriteStartArray()

        {

            base.WriteStartArray();

            CheckDepth();

        }

​

        public override void WriteStartConstructor(string name)

        {

            base.WriteStartConstructor(name);

            CheckDepth();

        }

​

        public override void WriteStartObject()

        {

            base.WriteStartObject();

            CheckDepth();

        }

​

        private void CheckDepth()

        {

            MaxObservedDepth = Math.Max(MaxObservedDepth, Top);

            if (Top > MaxDepth)

                throw new JsonSerializationException(string.Format("Depth {0} Exceeds MaxDepth {1} at path \"{2}\"", Top, MaxDepth, Path));

        }

    }

​

    public class SimpleClass

    {

        public string Value { get; set; }

    }

​

    public class TestClass

    {

        public ObjectId ObjectId;

​

        public static void Test()

        {

            var myClass = new TestClass { ObjectId = ObjectId.Create() };

            var myList = Enumerable.Range(0, 10).Select(i => new TestClass { ObjectId = ObjectId.Create() }).ToList();

​

            Assert.DoesNotThrow(() => Test(new [] { 1, 2, 3  }, 1));

            Assert.DoesNotThrow(() => Test(ObjectId.Create(), 1));

            Assert.DoesNotThrow(() => Test(new SimpleClass { Value = "some value" }, 1));

            Assert.DoesNotThrow(() => Test(new SimpleClass { Value = "some value" }, 2));

            Assert.DoesNotThrow(() => Test(new SimpleClass { Value = "some value" }, 3));

            Assert.DoesNotThrow(() => Test(myClass, 3));

            Assert.DoesNotThrow(() => Test(myClass, 2));

            Assert.Throws<JsonSerializationException>(() => Test(myClass, 1));

            Assert.DoesNotThrow(() => Test(myList, 3));

            Assert.Throws<JsonSerializationException>(() => Test(myList, 2));

        }

        static void Test<T>(T root, int maxDepth)

        {

            var settings = new JsonSerializerSettings { MaxDepth = maxDepth };

            string json;

            try

            {

                Console.WriteLine("Testing serialization of {0} with MaxDepth {1}:", root, maxDepth);

                using (var writer = new StringWriter())

                {

                    using (var jsonWriter = new MaxDepthJsonTextWriter(writer, settings))

                    {

                        JsonSerializer.Create(settings).Serialize(jsonWriter, root);

                        // Log the MaxObservedDepth here, if you want to.

                    }

                    json = writer.ToString();

                }

                Console.Write("  passed, result: ");

                Console.WriteLine(json);

            }

            catch (Exception ex)

            {

                Console.WriteLine("  failed with exception {0}", ex.Message);

                //Console.WriteLine(ex);

                throw;

            }

        }

    }

​

    public class Program

    {

        public static void Main()

        {

            Console.WriteLine("Environment version: " + Environment.Version);

            Console.WriteLine("Json.NET version: " + typeof(JsonSerializer).Assembly.FullName);

            Console.WriteLine();

​

            try

            {

                TestClass.Test();

            }

            catch (Exception ex)

            {

                Console.WriteLine("Failed with unhandled exception: ");

                Console.WriteLine(ex);

                throw;

            }

        }

    }

​