using System;
using System.Collections.Generic;
using System.Linq;
using System.Web;
using System.Web.Mvc;
using System.IO;

namespace CustomViewEngines.Infrastructure
{
    public class CustomRazorViewEngine : RazorViewEngine
    {
        public CustomRazorViewEngine() : base()
        {
            AreaViewLocationFormats = new string[] { "~/Areas/{2}/Views/{1}/%1/{0}.cshtml", 
                                                     "~/Areas/{2}/Views/{1}/%1/{0}.vbhtml", 
                                                     "~/Areas/{2}/Views/Shared/%1/{0}.cshtml", 
                                                     "~/Areas/{2}/Views/Shared/%1/{0}.vbhtml" };
            AreaMasterLocationFormats = new string[] { "~/Areas/{2}/Views/{1}/%1/{0}.cshtml", 
                                                       "~/Areas/{2}/Views/{1}/%1/{0}.vbhtml", 
                                                       "~/Areas/{2}/Views/Shared/%1/{0}.cshtml", 
                                                       "~/Areas/{2}/Views/Shared/%1/{0}.vbhtml" };
            AreaPartialViewLocationFormats = new string[] { "~/Areas/{2}/Views/{1}/%1/{0}.cshtml", 
                                                            "~/Areas/{2}/Views/{1}/%1/{0}.vbhtml", 
                                                            "~/Areas/{2}/Views/Shared/%1/{0}.cshtml", 
                                                            "~/Areas/{2}/Views/Shared/%1/{0}.vbhtml" };
            ViewLocationFormats = new string[] { "~/Views/{1}/%1/{0}.cshtml", 
                                                 "~/Views/{1}/%1/{0}.vbhtml", 
                                                 "~/Views/Shared/%1/{0}.cshtml", 
                                                 "~/Views/Shared/%1/{0}.vbhtml" };
            MasterLocationFormats = new string[] { "~/Views/{1}/%1/{0}.cshtml", 
                                                   "~/Views/{1}/%1/{0}.vbhtml", 
                                                   "~/Views/Shared/%1/{0}.cshtml", 
                                                   "~/Views/Shared/%1/{0}.vbhtml" };
            PartialViewLocationFormats = new string[] { "~/Views/{1}/%1/{0}.cshtml", 
                                                        "~/Views/{1}/%1/{0}.vbhtml", 
                                                        "~/Views/Shared/%1/{0}.cshtml", 
                                                        "~/Views/Shared/%1/{0}.vbhtml" };
            FileExtensions = new string[] { "cshtml", "vbhtml" };
        }

        protected override IView CreateView(ControllerContext controllerContext, string viewPath, string masterPath)
        {
            return base.CreateView(controllerContext, viewPath.Replace("%1","Pages"), masterPath.Replace("%1","Pages"));
        }

        protected override IView CreatePartialView(ControllerContext controllerContext, string partialPath)
        {
            return base.CreatePartialView(controllerContext, partialPath.Replace("%1","PartialPages"));
        }

        protected override bool FileExists(ControllerContext controllerContext, string virtualPath)
        {
            return (base.FileExists(controllerContext, virtualPath.Replace("%1", "Pages")) ||
                    base.FileExists(controllerContext, virtualPath.Replace("%1", "PartialPages")));
        }
    }
}