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插件来有针对性地批量修改代码的方法，在此基础上，我们可以做很多有趣的事情，但是目的是一个，就是尽可能偷懒，即加快我们开发又不出错。