Programowo dodaj funkcję do istniejącego pliku C# za pomocą EnvDTE

Question

Chcę programowo dodać funkcję (TestMethod) do istniejącego pliku C#. Po pewnym przeszukaniu go znalazłem bibliotekę EnvDTE i CodeModel.AddFunction -Method, ale nie mogę znaleźć dobrego przykładu tego, czego chcę.

Chciałbym dodać funkcję z kodem już w tej nowo utworzonej funkcji, a także z atrybutem. Coś takiego:

/// <summary> 
/// Documentation 
/// </summary> 
[TestMethod] 
public void TestMethod1() 
{ 
    string test = Helper.CodeExample(); 
} 

Czy ktoś może pokazać mi przykład, jak to zrobić?

EDYCJA: Chcę edytować plik C#, tak jak w przypadku edycji pliku tekstowego. Wiem, że możesz to zrobić za pomocą StreamWriter, ale czy jest lepszy sposób na zrobienie tego?

Answer 1

EnvDTE może być najlepszym rozwiązaniem. Możesz rozwinąć dodatek VisualStudio, a następnie zmodyfikować metodę Exec. W tej metodzie musisz uzyskać aktywny dokument i jego ProjectItem. Tutaj znajdziesz CodeModel, który zawiera dużo CodeElements. Wśród tych elementów musisz znaleźć CodeNamespace, a wewnątrz tego elementu CodeClass. Jest to obiekt, który odpowiada na AddFunction zwracający nową funkcję CodeFunction, do której możesz dodać swój atrybut i kod (jest to część, której nie lubię zbyt wiele, ponieważ musisz używać EditPoint).

Jest to bardzo prosta wersja Exec, które można wykorzystać jako punkt wyjścia:

public void Exec(string commandName, vsCommandExecOption executeOption, ref object varIn, ref object varOut, ref bool handled) 
    { 
     handled = false; 
     if(executeOption == vsCommandExecOption.vsCommandExecOptionDoDefault) 
     { 
      handled = true; 
      if (commandName == "TestAddMethod.Connect.TestAddMethod") 
      { 
       Document activeDoc = _applicationObject.ActiveDocument; 
       if (activeDoc == null) 
        return; 
       ProjectItem prjItem = activeDoc.ProjectItem; 
       if (prjItem == null) 
        return; 
       FileCodeModel fcm = prjItem.FileCodeModel; 
       if (fcm == null) 
        return; 

       CodeElements ces = fcm.CodeElements; 
       // look for the namespace in the active document 
       CodeNamespace cns = null; 
       foreach (CodeElement ce in ces) 
       { 
        if (ce.Kind == vsCMElement.vsCMElementNamespace) 
        { 
         cns = ce as CodeNamespace; 
         break; 
        } 
       } 
       if (cns == null) 
        return; 
       ces = cns.Members; 
       if (ces == null) 
        return; 
       // look for the first class 
       CodeClass cls = null; 
       foreach (CodeElement ce in ces) 
       { 
        if (ce.Kind == vsCMElement.vsCMElementClass) 
        { 
         cls = ce as CodeClass; 
         break; 
        } 
       } 
       if (cls == null) 
        return; 
       CodeFunction cf = cls.AddFunction("TestMethod1", vsCMFunction.vsCMFunctionFunction, vsCMTypeRef.vsCMTypeRefVoid, -1, vsCMAccess.vsCMAccessPrivate); 
       cf.AddAttribute("TestMethod", "true"); 
       TextPoint tp = cf.GetStartPoint(vsCMPart.vsCMPartBody); 
       EditPoint ep = tp.CreateEditPoint(); 
       ep.Indent(); 
       ep.Indent(); 
       ep.Indent(); 
       ep.Insert("string test = Helper.CodeExample();"); 
      } 
     } 
    }