Aggiunta di funzioni al file C# esistente con EnvDTE

Question

Desidero aggiungere una funzione (un TestMethod) a un file C# esistente. Dopo aver cercato su Google ho trovato la libreria EnvDTE e CodeModel.AddFunction -Method, ma non riesco a trovare un buon esempio di ciò che voglio.

Vorrei aggiungere una funzione con codice già in quella funzione appena creata e anche con un attributo. Qualcosa del genere:

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

Qualcuno può mostrarmi un esempio su come farlo?

EDIT: Voglio modificare un file C#, come se si modificasse un file di testo. So che potresti farlo con uno StreamWriter, ma c'è forse un modo migliore per farlo?

Answer 1

EnvDTE può essere la strada da percorrere. È possibile sviluppare un componente aggiuntivo VisualStudio e quindi modificare il metodo Exec. In questo metodo devi ottenere il documento attivo e il suo oggetto ProjectItem. Questo è dove si trova il CodeModel che contiene molti CodeElements. Tra questi elementi devi trovare lo spazio CodeNames e all'interno di questo elemento CodeClass. Questo è l'oggetto che risponde ad AddFunction restituendo il nuovo CodeFunction al quale è possibile aggiungere il proprio attributo e il codice (questa è una parte che non mi piace troppo, dal momento che è necessario utilizzare EditPoint).

Questa è una versione molto semplice di Exec che è possibile utilizzare come punto di partenza:

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();"); 
      } 
     } 
    }