using System; using System.Collections.Generic; using FluentValidation; using Nancy; using NzbDrone.Api.Extensions; using System.Linq; namespace NzbDrone.Api.REST { public abstract class RestModule : NancyModule where TResource : RestResource, new() { protected ResourceValidator PostValidator { get; private set; } protected ResourceValidator PutValidator { get; private set; } protected ResourceValidator SharedValidator { get; private set; } private const string ROOT_ROUTE = "/"; private const string ID_ROUTE = "/{id}"; protected RestModule() : this(new TResource().ResourceName) { } protected RestModule(string modulePath) : base(modulePath) { PostValidator = new ResourceValidator(); PutValidator = new ResourceValidator(); SharedValidator = new ResourceValidator(); Get[ROOT_ROUTE] = options => { EnsureImplementation(GetResourceAll); var resource = GetResourceAll(); return resource.AsResponse(); }; Get[ID_ROUTE] = options => { EnsureImplementation(GetResourceById); var resource = GetResourceById((int)options.Id); return resource.AsResponse(); }; Post[ROOT_ROUTE] = options => { EnsureImplementation(CreateResource); var resource = CreateResource(ReadFromRequest()); return resource.AsResponse(HttpStatusCode.Created); }; Put[ROOT_ROUTE] = options => { EnsureImplementation(UpdateResource); var resource = UpdateResource(ReadFromRequest()); return resource.AsResponse(HttpStatusCode.Accepted); }; Delete[ID_ROUTE] = options => { EnsureImplementation(DeleteResource); DeleteResource((int)options.Id); return new Response { StatusCode = HttpStatusCode.OK }; }; } protected Action DeleteResource { get; set; } protected Func GetResourceById { get; set; } protected Func> GetResourceAll { get; set; } protected Func CreateResource { get; set; } protected Func UpdateResource { get; set; } private void EnsureImplementation(Delegate implementation) { if (implementation == null) { throw new NotImplementedException(); } } private TResource ReadFromRequest() { var resource = Request.Body.FromJson(); var errors = SharedValidator.Validate(resource).Errors.ToList(); if (Request.Method.Equals("POST", StringComparison.InvariantCultureIgnoreCase)) { errors.AddRange(PostValidator.Validate(resource).Errors); } else if (Request.Method.Equals("PUT", StringComparison.InvariantCultureIgnoreCase)) { errors.AddRange(PutValidator.Validate(resource).Errors); } if (errors.Any()) { throw new ValidationException(errors); } return resource; } } }