namespace STDotCom
{
    using System;
    using System.Web.Mvc;

    /// <summary>
    ///  Used to make razor aware of where views are located on a module by module basis.
    /// </summary>
    public class WidgetFolderViewEngine : VirtualPathProviderViewEngine
    {
        /// <summary>
        ///  Initializes a new instance of the <see cref="WidgetFolderViewEngine"/> class.
        /// </summary>
        public WidgetFolderViewEngine()
        {
            string[] widgetFolderFormats = new[]
            {
                "~/Widgets/{1}/Views/{0}.cshtml"
            };

            this.FileExtensions = new string[] { "cshtml", "vbhtml", "aspx", "ascx" };
            this.AreaViewLocationFormats = widgetFolderFormats;
            this.AreaMasterLocationFormats = widgetFolderFormats;
            this.AreaPartialViewLocationFormats = widgetFolderFormats;
            this.ViewLocationFormats = widgetFolderFormats;
            this.MasterLocationFormats = widgetFolderFormats;
            this.PartialViewLocationFormats = widgetFolderFormats;
        }

        public override ViewEngineResult FindPartialView(ControllerContext controllerContext, string partialViewName, bool useCache)
        {
            return base.FindPartialView(controllerContext, partialViewName, useCache);
        }

        public override ViewEngineResult FindView(ControllerContext controllerContext, string viewName, string masterName, bool useCache)
        {
            return base.FindView(controllerContext, viewName, masterName, useCache);
        }

        protected override bool FileExists(ControllerContext controllerContext, string virtualPath)
        {
            return base.FileExists(controllerContext, virtualPath);
        }

        protected override IView CreatePartialView(ControllerContext controllerContext, string partialPath)
        {
            if (partialPath.EndsWith(".cshtml") || partialPath.EndsWith(".vbhtml"))
            {
                return new RazorView(controllerContext, partialPath, null, false, null);
            }
            else
            {
                return new WebFormView(controllerContext, partialPath);
            }
        }

        protected override IView CreateView(ControllerContext controllerContext, string viewPath, string masterPath)
        {
            if (viewPath.EndsWith(".cshtml") || viewPath.EndsWith(".vbhtml"))
            {
                return new RazorView(controllerContext, viewPath, masterPath, false, null);
            }
            else
            {
                return new WebFormView(controllerContext, viewPath, masterPath);
            }
        }
    }
}