/// <summary>
/// Adds all services required for the <see cref="RenderRazor"/> module.
/// </summary>
/// <param name="serviceCollection">The service collection to register services in.</param>
/// <param name="fileSystem">The file system or <c>null</c> to skip.</param>
/// <param name="classCatalog">An existing class catalog or <c>null</c> to scan assemblies during registration.</param>
/// <returns>The service collection.</returns>
public static IServiceCollection AddRazor(this IServiceCollection serviceCollection, IReadOnlyFileSystem fileSystem, ClassCatalog classCatalog = null)
{
// Register the file system if we're not expecting one from an engine
if (fileSystem != null)
{
serviceCollection.TryAddSingleton(fileSystem);
}
// Register some of our own types if not already registered
serviceCollection.TryAddSingleton <Microsoft.Extensions.FileProviders.IFileProvider, FileSystemFileProvider>();
serviceCollection.TryAddSingleton <DiagnosticSource, SilentDiagnosticSource>();
serviceCollection.TryAddSingleton(new DiagnosticListener("Razor"));
serviceCollection.TryAddSingleton <IWebHostEnvironment, HostEnvironment>();
serviceCollection.TryAddSingleton <ObjectPoolProvider, DefaultObjectPoolProvider>();
serviceCollection.TryAddSingleton <StatiqRazorProjectFileSystem>();
serviceCollection.TryAddSingleton <RazorProjectFileSystem, StatiqRazorProjectFileSystem>();
// Register the view location expander if not already registered
serviceCollection.Configure <RazorViewEngineOptions>(x =>
{
if (!x.ViewLocationExpanders.OfType <ViewLocationExpander>().Any())
{
x.ViewLocationExpanders.Add(new ViewLocationExpander());
}
});
// Add the default services _after_ adding our own
// (most default registration use .TryAdd...() so they skip already registered types)
IMvcCoreBuilder builder = serviceCollection
.AddMvcCore()
.AddRazorViewEngine()
.AddRazorRuntimeCompilation();
// Add all loaded assemblies
CompilationReferencesProvider referencesProvider = new CompilationReferencesProvider();
referencesProvider.Assemblies.AddRange((classCatalog ?? new ClassCatalog()).GetAssemblies());
// And a couple needed assemblies that might not be loaded in the AppDomain yet
referencesProvider.Assemblies.Add(typeof(IHtmlContent).Assembly);
referencesProvider.Assemblies.Add(Assembly.Load(new AssemblyName("Microsoft.CSharp")));
// Add the reference provider as an ApplicationPart
builder.ConfigureApplicationPartManager(x => x.ApplicationParts.Add(referencesProvider));
return(serviceCollection);
}
public RazorCompiler(CompilationParameters parameters, IExecutionContext context)
{
// Create the service collection that MVC needs and add default MVC services
ServiceCollection serviceCollection = new ServiceCollection();
DiagnosticListener listener = new DiagnosticListener("Razor");
// Register some of our own types
serviceCollection
.AddSingleton(parameters.FileSystem)
.AddSingleton <Microsoft.Extensions.FileProviders.IFileProvider, FileSystemFileProvider>()
.AddSingleton(context.GetRequiredService <ILoggerFactory>())
.AddSingleton <DiagnosticSource, SilentDiagnosticSource>()
.AddSingleton <DiagnosticListener>(listener)
.AddSingleton <IWebHostEnvironment, HostEnvironment>()
.AddSingleton <ObjectPoolProvider, DefaultObjectPoolProvider>()
// .AddSingleton<IViewCompilerProvider, StatiqRazorViewCompilerProvider>()
.AddSingleton <StatiqRazorProjectFileSystem>()
.AddSingleton <RazorProjectFileSystem, StatiqRazorProjectFileSystem>();
// Register the view location expander
serviceCollection.Configure <RazorViewEngineOptions>(x => x.ViewLocationExpanders.Add(new ViewLocationExpander()));
// Add the default services _after_ adding our own
// (most default registration use .TryAdd...() so they skip already registered types)
IMvcCoreBuilder builder = serviceCollection
.AddMvcCore()
.AddRazorViewEngine()
.AddRazorRuntimeCompilation();
// Add all loaded assemblies
CompilationReferencesProvider referencesProvider = new CompilationReferencesProvider();
referencesProvider.Assemblies.AddRange(AppDomain.CurrentDomain.GetAssemblies());
// And a couple needed assemblies that might not be loaded in the AppDomain yet
referencesProvider.Assemblies.Add(typeof(IHtmlContent).Assembly);
referencesProvider.Assemblies.Add(Assembly.Load(new AssemblyName("Microsoft.CSharp")));
// Add the reference provider as an ApplicationPart
builder.ConfigureApplicationPartManager(x => x.ApplicationParts.Add(referencesProvider));
// Build the service provider and local scope
IServiceProvider serviceProvider = serviceCollection.BuildServiceProvider();
_serviceScopeFactory = serviceProvider.GetRequiredService <IServiceScopeFactory>();
// Calculate the base page type
Type basePageType = parameters.BasePageType ?? typeof(StatiqRazorPage <>);
string baseClassName = basePageType.FullName;
int tickIndex = baseClassName.IndexOf('`');
if (tickIndex > 0)
{
baseClassName = baseClassName.Substring(0, tickIndex);
}
string baseType = basePageType.IsGenericTypeDefinition ? $"{baseClassName}<TModel>" : baseClassName;
// We need to register a new document classifier phase because builder.SetBaseType() (which uses builder.ConfigureClass())
// use the DefaultRazorDocumentClassifierPhase which stops applying document classifier passes after DocumentIntermediateNode.DocumentKind is set
// (which gets set by the Razor document classifier passes registered in RazorExtensions.Register())
// Also need to add it just after the DocumentClassifierPhase, otherwise it'll miss the C# lowering phase
RazorProjectEngine razorProjectEngine = serviceProvider.GetRequiredService <RazorProjectEngine>();
List <IRazorEnginePhase> phases = razorProjectEngine.Engine.Phases.ToList();
phases.Insert(
phases.IndexOf(phases.OfType <IRazorDocumentClassifierPhase>().Last()) + 1,
new StatiqDocumentPhase(baseType, parameters.Namespaces)
{
Engine = razorProjectEngine.Engine
});
FieldInfo phasesField = razorProjectEngine.Engine.GetType().GetField("<Phases>k__BackingField", BindingFlags.Instance | BindingFlags.NonPublic);
phasesField.SetValue(razorProjectEngine.Engine, phases.ToArray());
}
