How to manage undo/redo stack in VSPackage?

Question

I have created VSPackage that provides certain functionality in Visual Studio code window context menu. This action consists of several code edits plus some stuff around.

The problem is, each of these code edits get added to the undo stack separately. What I want is to handle this action as one atomic unit, i.e. pressing CTRL+Z rollbacks all code edits and everything else (plus of course, places the unit on top of redo stack).

Documentation regarding this topic is extremely poor, the only thing I found is something about IOleParentUndo unit - but I wasn't succesfull in implementing it.

I use

IVsTextLines.GetUndoManager()

to get the undo manager - that appears to be a good start.

Answer 1

I encapsulated below code to undo couple of actions.

public class VSUndo : IDisposable
{
    public static UndoContext undoContext;

    public static VSUndo StartUndo()
    {
        undoContext = ((DTE2)Package.GetGlobalService(typeof(DTE))).UndoContext; 
        undoContext.Open(Guid.NewGuid().ToString());
        // return new instance for calling dispose to close current undocontext
        return new VSUndo(); 
    }

    public void Dispose()
    {
        undoContext.Close();
    }

}

then, you can just use:

using (VSUndo.StartUndo())
{
   // couple of actions that may need to undo together
}