Razor Engine，实现代码生成器的又一件利器

Razor Engine，之前仅仅是ASP.NET MVC的一种View引擎，目前已经完全成为一种可以独立使用的模版引擎，并且已经成为了CodePlex上一个开源的项目（http://razorengine.codeplex.com/）。对于使用过ASP.NET MVC Razor视图引擎的朋友们一定已经领略过它的灵活性和易用性，在这篇文章中我们将利用它来实现一个代码生成器使我们可以以Razor的语法来定义代码模版。[源代码从这里下载]

在《一个简易版的T4代码生成"框架"》这篇文章中，我创建了一个能够生成多个文件的代码生成器，其核心是两个基本的类型：Template类生成代码文件的内容，Generator执行Template。现在我将通过自定义一个基于Razor Engine的Template，让它读取以Razor语法编写的模版文件并生成相应的代码。

不过在具体介绍实现原理之前，还是先来看看它能够达到的效果。如右图所示，我们还是通过一个.tt文件（Generator.tt）来生成相应的代码文件，具体生成的代码文件就是依赖于该.tt文件的三个.cs 文件：Foo.cs、Bar.cs和Baz.cs。简单起见，我们仅仅在这三个文件中生成了一个与文件同名的空Class，它们的定义如下。

public class Foo
{
}

public class Bar
{
}

public class Baz
{
}

如下所示的是Generator.tt的内容，它直接执行一个类型为DemoGenerator的生成器。

<#@ template hostspecific="true" language="C#" #>
<#@ assembly name="$(TargetDir)Artech.CodeGeneration.dll" #>
<#@ import namespace="Artech.CodeGeneration" #>
<#@ output extension=".empty" #>
<#this.RunCodeGenerator(this.Host, new DemoGenerator());#>

作为代码生成器的DemoGenerator定义如下，它重写了CreateTemplates方法并返回一个包含三个RazorTemplate<TModel>的字典，该字典的Key表示生成的文件名。

public class DemoGenerator: Generator
{
    protected override IDictionary<string, Template> CreateTemplates()
    {
        Dictionary<string, Template> templates = new Dictionary<string, Template>();
        templates.Add("Foo.cs", new RazorTemplate<string>("template.razor", "Foo"));
        templates.Add("Bar.cs", new RazorTemplate<string>("template.razor", "Bar"));
        templates.Add("Baz.cs", new RazorTemplate<string>("template.razor", "Baz"));
        return templates;
    }
}

构建RazorTemplate<TModel>需要传入两个参数，第一个参数是用于定义Razor模版的文件路径（可以是绝对路径，也可以是相对于.tt文件的相对路径），另一个是作为Model的对象（相当于ASP.NET强类型View的Model）。通过上面的代码可以看出，三个RazorTemplate<TModel>均使用同一个Model类型为字符串的模版文件template.razor，该模版内容定义如下。可以看出作为Model的字符串在模版中作为了生成类型的名称。

public class @Model
{
}

我假设你已经了解了《一个简易版的T4代码生成"框架"》这篇文章的内容，所以我们只需要简单介绍RazorTemplate<TModel>的定义就可以了。首选RazorTemplate<TModel>是一个具有如下定义的RazorTemplate类型的子类。RazorTemplate的只读属性TemplateFileName 表示Razor模版的路径，用于读取模版文本内容的虚方法GetTemplateText直接读取指定模版文件的内容。在重写的TransformText方法中，我们调用Razor的静态方法Parse对模版内容进行解析并生成代码文本。

public class RazorTemplate : Template
{
    public string TemplateFileName { get; private set; }
    public RazorTemplate(string templateFileName)
    {
        this.TemplateFileName = templateFileName;
    }
    protected virtual string GetTemplateText()
    {
        string path = this.TemplateFileName;
        if (!Path.IsPathRooted(path))
        {
            path = Path.Combine(Path.GetDirectoryName(TransformContext.Current.Host.TemplateFile), path);
        }
        return File.ReadAllText(path);
    }

    public override string TransformText()
    {
        return Razor.Parse(this.GetTemplateText());
    }
}

继承自RazorTemplate的泛型RazorTemplate<TModel>定义如下。它具有一个表示Model的只读属性，该属性在构造函数中被初始化。在重写的TransformText方法中，我们调用Razor泛型的静态方法Parse<TModel>解析模版文本并传入Model对象。

public class RazorTemplate<TModel> : RazorTemplate
{
    public TModel Model { get; private set; }
    public RazorTemplate(string templateFileName, TModel model)
        : base(templateFileName)
    {
        this.Model = model;
    }

    public override string TransformText()
    {
        return Razor.Parse<TModel>(this.GetTemplateText(), this.Model);
    }
}

 

相关阅读：
与VS集成的若干种代码生成解决方案[博文汇总(共8篇)]
通过CodeDOM定义生成代码的结构
通过Visual Studio的Custom Tool定义代码生成器
不同于CodeDOM的代码生成机制——T4
通过T4模板实现单文件的代码生成
通过T4模板实现多文件的代码生成
解决T4模板的程序集引用的五种方案
编写T4模板进行代码生成无法避免的两个话题："Assembly Locking"&"Debug"
通过自定义BuildProvider为ASP.NET提供代码生成