Kyoo/Kyoo.CommonAPI/JsonSerializer.cs

using System;
using System.Buffers;
using System.Collections.Generic;
using System.Reflection;
using Kyoo.Models;
using Kyoo.Models.Attributes;
using Microsoft.AspNetCore.Mvc;
using Microsoft.AspNetCore.Mvc.Filters;
using Microsoft.AspNetCore.Mvc.Formatters;
using Microsoft.Extensions.DependencyInjection;
using Newtonsoft.Json;
using Newtonsoft.Json.Serialization;

namespace Kyoo.Controllers
{
	public class JsonPropertyIgnorer : CamelCasePropertyNamesContractResolver
	{
		protected override JsonProperty CreateProperty(MemberInfo member, MemberSerialization memberSerialization)
		{
			JsonProperty property = base.CreateProperty(member, memberSerialization);
			
			property.ShouldSerialize = i => member.GetCustomAttribute<JsonReadOnly>(true) == null;
			property.ShouldDeserialize = i => member.GetCustomAttribute<JsonIgnore>(true) == null;
			return property;
		}
	}
	
	public class JsonPropertySelector : JsonPropertyIgnorer
	{
		private readonly Dictionary<Type, HashSet<string>> _ignored;
		private readonly Dictionary<Type, HashSet<string>> _forceSerialize;

		public JsonPropertySelector()
		{
			_ignored = new Dictionary<Type, HashSet<string>>();	
			_forceSerialize = new Dictionary<Type, HashSet<string>>();
		}
		
		public JsonPropertySelector(Dictionary<Type, HashSet<string>> ignored, 
			Dictionary<Type, HashSet<string>> forceSerialize = null)
		{
			_ignored = ignored ?? new Dictionary<Type, HashSet<string>>();
			_forceSerialize = forceSerialize ?? new Dictionary<Type, HashSet<string>>();
		}

		private bool IsIgnored(Type type, string propertyName)
		{
			while (type != null)
			{
				if (_ignored.ContainsKey(type) && _ignored[type].Contains(propertyName))
					return true;
				type = type.BaseType;
			}

			return false;
		}

		private bool IsSerializationForced(Type type, string propertyName)
		{
			while (type != null)
			{
				if (_forceSerialize.ContainsKey(type) && _forceSerialize[type].Contains(propertyName))
					return true;
				type = type.BaseType;
			}

			return false;
		}

		protected override JsonProperty CreateProperty(MemberInfo member, MemberSerialization memberSerialization)
		{
			JsonProperty property = base.CreateProperty(member, memberSerialization);

			if (IsSerializationForced(property.DeclaringType, property.PropertyName))
				property.ShouldSerialize = i => true;
			else if (IsIgnored(property.DeclaringType, property.PropertyName))
				property.ShouldSerialize = i => false;
			return property;
		}
	}
	
	public class JsonDetailed : ActionFilterAttribute
	{
		public override void OnActionExecuted(ActionExecutedContext context)
		{
			if (context.Result is ObjectResult result)
			{
				result.Formatters.Add(new NewtonsoftJsonOutputFormatter(
					new JsonSerializerSettings
					{
						ContractResolver = new JsonPropertySelector(null, new Dictionary<Type, HashSet<string>>
						{
							{typeof(Show), new HashSet<string> {"genres", "studio"}},
							{typeof(Episode), new HashSet<string> {"tracks"}},
							{typeof(PeopleRole), new HashSet<string> {"show"}}
						})
					},
				context.HttpContext.RequestServices.GetRequiredService<ArrayPool<char>>(), 
					new MvcOptions()));
			}
		}
	}
}