Visual Studio 核心代码模型（Addin+CodeModel）使用心得 .

我们知道Visual Studio 2010的UML设计功能已经可以做到正向工程了，但是这些生成的代码都是空架子，还得经过手工处理填充上业务逻辑代码，如果系统是基于某个具体技术框架进行开发的话，还必须加上这些技术框架的特性代码。比如：

/// <summary>

/// 布尔值选项

/// </summary>

[Serializable]

public enum BooleanOption

{

/// <summary>

/// 否

/// </summary>

[EnumCaption("否")]

N,

/// <summary>

/// 是

/// </summary>

[EnumCaption("是")]

Y

}

代码里面的[Serializable]、[EnumCaption("否")]都是需要附加到代码里的。

为了实现快速开发，减少开发工作量，这些工作是否可以让机器代劳？利用“Visual Studio 自动化对象模型”技术似乎是个可行方案。

http://msdn.microsoft.com/zh-cn/library/za2b25t3(v=VS.80).aspx

只要我们在VS的IDE里利用Addin插件技术，参与到它的代码编辑功能中，通过分析类的定义，将技术框架的特征码嵌入到代码文本中。

基于上述的想法，我们开始探索之旅：

1，在代码编辑窗体的弹出菜单里嵌入一个菜单项，借此触发实现我们的需求：

基于网上有很多编写Connect类的介绍，本文不再多说，代码节选如下：

/// <summary>实现 IDTExtensibility2 接口的 OnConnection 方法。接收正在加载外接程序的通知。</summary>

/// <param term='application'>宿主应用程序的根对象。</param>

/// <param term='connectMode'>描述外接程序的加载方式。</param>

/// <param term='addInInst'>表示此外接程序的对象。</param>

/// <seealso class='IDTExtensibility2' />

public void OnConnection(object Application, ext_ConnectMode ConnectMode, object AddInInst, ref Array custom)

{

_applicationObject = (DTE2)Application;

_addInInstance = (AddIn)AddInInst;

//如果是安装状态或是插件刚被启动的状态，则创建菜单

if (ConnectMode == Extensibility.ext_ConnectMode.ext_cm_UISetup

|| ConnectMode == Extensibility.ext_ConnectMode.ext_cm_Startup)

{

//取代码编辑窗口右键弹出菜单的工具条

CommandBar codeWindowBar = ((CommandBars)_applicationObject.CommandBars)["Code Window"];

//将菜单项加入工具条

object[] contextGUIDS = new object[] { };

Command commandObj = _applicationObject.Commands.AddNamedCommand(_addInInstance,

_buildClassMenuName, "构建业务类/规则类", "业务类由指定的数据库表/视图映射",

true, 137, ref contextGUIDS,

(int)vsCommandStatus.vsCommandStatusSupported + (int)vsCommandStatus.vsCommandStatusEnabled

);

commandObj.AddControl(codeWindowBar);

}

}

2，通过当前活动的编辑窗体ActiveDocument找到它的工程项ProjectItem，通过工程项得到我们需要的FileCodeModel：

FileCodeModel2 codeModel = (FileCodeModel2)_applicationObject.ActiveDocument.ProjectItem.FileCodeModel;

我们知道FileCodeModel是操作代码模型的入口：

http://msdn.microsoft.com/zh-cn/library/envdte.filecodemodel.aspx

3，通过FileCodeModel，我们可以对当前代码编辑框内的代码进行改写了，但是首先要找到改写的点，由于FileCodeModel是一个树，我们要遍历它，为了方便遍历，写了这样的公共函数：

internal static class CodeModelHelper

{

public static List<CodeElement2> FindAllCodeElement(CodeElements elements, vsCMElement kind)

{

List<CodeElement2> result = new List<CodeElement2>();

FindAllCodeElement(elements, kind, ref result);

return result;

}

private static void FindAllCodeElement(CodeElements elements, vsCMElement kind, ref List<CodeElement2> elementList)

{

if (elements != null)

foreach (CodeElement2 item in elements)

{

if (item.Kind == kind)

elementList.Add(item);

FindAllCodeElement(item.Children, kind, ref elementList);

}

}

}

我们要找到这个代码文本中所有的枚举定义，应该这么做：

CodeModelHelper.FindAllCodeElement(codeModel.CodeElements, vsCMElement.vsCMElementEnum)

返回的是符合vsCMElement.vsCMElementEnum条件的CodeElement的集合，我们遍历这个集合，可以将把[Serializable]添加到枚举上：

public void Exec(string CmdName, vsCommandExecOption ExecuteOption, ref object VariantIn, ref object VariantOut, ref bool Handled)

{

if (ExecuteOption == vsCommandExecOption.vsCommandExecOptionDoDefault)

{

if (CmdName == _buildClassCmdName)

{

FileCodeModel2 codeModel = (FileCodeModel2)_applicationObject.ActiveDocument.ProjectItem.FileCodeModel;

ExecuteBuildEnumClass(CodeModelHelper.FindAllCodeElement(codeModel.CodeElements, vsCMElement.vsCMElementEnum));

Handled = true;

return;

}

}

Handled = false;

}

private static void ExecuteBuildEnumClass(List<CodeElement2> elementList)

{

foreach (CodeEnum item in elementList)

{

item.AddAttribute(typeof(SerializableAttribute).Name, null);

}

}

结论：

上述简单地演示了一个通过Addin插件来有针对性地批量修改代码的方法，在此基础上，我们可以做很多有趣的事情，但是目的是一个，就是尽可能偷懒，即加快我们开发又不出错。