SuspendFileChanges.cs :  » Development » StyleCop » Microsoft » VisualStudio » Package » C# / CSharp Open Source

/***************************************************************************

Copyright (c) Microsoft Corporation. All rights reserved.
This code is licensed under the Visual Studio SDK license terms.
THIS CODE IS PROVIDED *AS IS* WITHOUT WARRANTY OF
ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING ANY
IMPLIED WARRANTIES OF FITNESS FOR A PARTICULAR
PURPOSE, MERCHANTABILITY, OR NON-INFRINGEMENT.

***************************************************************************/

using System;
using Microsoft.VisualStudio.TextManager.Interop;
using Microsoft.VisualStudio.OLE.Interop;
using Microsoft.VisualStudio.Shell.Interop;
using Microsoft.VisualStudio.Designer.Interfaces;
using Microsoft.VisualStudio.Shell;
using System.Runtime.InteropServices;
using System.Windows.Forms;
using System.Diagnostics;
using Microsoft.Win32;
using System.Globalization;
using System.IO;
using System.Collections;
using System.Xml;
using System.Text;
using System.Net;
using IOleServiceProviderMicrosoft.VisualStudio.OLE.Interop.IServiceProvider;
using IServiceProviderSystem.IServiceProvider;
using ShellConstantsMicrosoft.VisualStudio.Shell.Interop.Constants;
using OleConstantsMicrosoft.VisualStudio.OLE.Interop.Constants;

namespace Microsoft.VisualStudio.Package{
  /// <summary>
  /// helper to make the editor ignore external changes
  /// </summary>
  internal class SuspendFileChanges
  {
    private string documentFileName;

    private bool isSuspending;

    private IServiceProvider site;

    private IVsDocDataFileChangeControl fileChangeControl;

    public SuspendFileChanges(IServiceProvider site, string document)
    {
      this.site = site;
      this.documentFileName = document;
    }
  
    
    public void Suspend()
    {
      if (this.isSuspending)
        return;

      IntPtr docData = IntPtr.Zero;
      try
      {
        IVsRunningDocumentTable rdt = this.site.GetService(typeof(SVsRunningDocumentTable)) as IVsRunningDocumentTable;
        
        IVsHierarchy hierarchy;
        uint itemId;
        uint docCookie;
        IVsFileChangeEx fileChange;


        if (rdt == null) return;

        ErrorHandler.ThrowOnFailure(rdt.FindAndLockDocument((uint)_VSRDTFLAGS.RDT_NoLock, this.documentFileName, out hierarchy, out itemId, out docData, out docCookie));

        if ((docCookie == (uint)ShellConstants.VSDOCCOOKIE_NIL) || docData == IntPtr.Zero)
          return;

        fileChange = this.site.GetService(typeof(SVsFileChangeEx)) as IVsFileChangeEx;

        if (fileChange != null)
        {
          this.isSuspending = true;
          ErrorHandler.ThrowOnFailure(fileChange.IgnoreFile(0, this.documentFileName, (int)1));
          if (docData != IntPtr.Zero)
          {
            IVsPersistDocData persistDocData = null;

            // if interface is not supported, return null
            object unknown = Marshal.GetObjectForIUnknown(docData);
            if (unknown is IVsPersistDocData)
            {
              persistDocData = (IVsPersistDocData)unknown;
              if (persistDocData is IVsDocDataFileChangeControl)
              {
                this.fileChangeControl = (IVsDocDataFileChangeControl)persistDocData;
                if (this.fileChangeControl != null)
                {
                  ErrorHandler.ThrowOnFailure(this.fileChangeControl.IgnoreFileChanges(1));
                }
              }
            }
          }
        }
      }
      catch (InvalidCastException e)
      {
        Trace.WriteLine("Exception" + e.Message);
      }  
      finally
      {
        if (docData != IntPtr.Zero)
        {
          Marshal.Release(docData);
        }
      }
      return;
    }

    public void Resume()
    {
      if (!this.isSuspending)
        return;
      IVsFileChangeEx fileChange;
      fileChange = this.site.GetService(typeof(SVsFileChangeEx)) as IVsFileChangeEx;
      if (fileChange != null)
      {
        this.isSuspending = false;
        ErrorHandler.ThrowOnFailure(fileChange.IgnoreFile(0, this.documentFileName, 0));
        if (this.fileChangeControl != null)
        {
          ErrorHandler.ThrowOnFailure(this.fileChangeControl.IgnoreFileChanges(0));
        }
      }
    }
  }
}