//------------------------------------------------------------------------------
// <copyright file="WebServiceVSPackage.cs" company="Company">
//     Copyright (c) Company.  All rights reserved.
// </copyright>
//------------------------------------------------------------------------------

using System;
using System.ComponentModel.Design;
using System.Diagnostics;
using System.Diagnostics.CodeAnalysis;
using System.Globalization;
using System.Runtime.InteropServices;
using Microsoft.VisualStudio;
using Microsoft.VisualStudio.OLE.Interop;
using Microsoft.VisualStudio.Shell;
using Microsoft.VisualStudio.Shell.Interop;
using Microsoft.Win32;
using EnvDTE;
using EnvDTE80;
using WebTest.WebService.Plugin.ResultTab;
using Microsoft.VisualStudio.Settings;
using Microsoft.VisualStudio.Shell.Settings;
using System.IO;

namespace WebTest.WebServive.Plugin
{
    /// <summary>
    /// This is the class that implements the package exposed by this assembly.
    /// </summary>
    /// <remarks>
    /// <para>
    /// The minimum requirement for a class to be considered a valid package for Visual Studio
    /// is to implement the IVsPackage interface and register itself with the shell.
    /// This package uses the helper classes defined inside the Managed Package Framework (MPF)
    /// to do it: it derives from the Package class that provides the implementation of the
    /// IVsPackage interface and uses the registration attributes defined in the framework to
    /// register itself and its components with the shell. These attributes tell the pkgdef creation
    /// utility what data to put into .pkgdef file.
    /// </para>
    /// <para>
    /// To get loaded into VS, the package must be referred by &lt;Asset Type="Microsoft.VisualStudio.VsPackage" ...&gt; in .vsixmanifest file.
    /// </para>
    /// </remarks>
    [PackageRegistration(UseManagedResourcesOnly = true)]
    [InstalledProductRegistration("#110", "#112", "1.0", IconResourceID = 400)] // Info on this package for Help/About
    [Guid(WebServiceVSPackage.PackageGuidString)]
    [SuppressMessage("StyleCop.CSharp.DocumentationRules", "SA1650:ElementDocumentationMustBeSpelledCorrectly", Justification = "pkgdef, VS and vsixmanifest are valid VS terms")]
    [ProvideAutoLoad(UIContextGuids80.SolutionExists)]
    [ProvideMenuResource("Menus.ctmenu", 1)]
    public sealed class WebServiceVSPackage : Package
    {
        /// <summary>
        /// WebServiceVSPackage GUID string.
        /// </summary>
        public const string PackageGuidString = "7e15e6a8-3708-4be9-92f4-92218254a23d";

        /// <summary>
        /// Initializes a new instance of the <see cref="WebServiceVSPackage"/> class.
        /// </summary>
        public WebServiceVSPackage()
        {
            // Inside this method you can place any initialization code that does not require
            // any Visual Studio service because at this point the package object is created but
            // not sited yet inside Visual Studio environment. The place to do all the other
            // initialization is the Initialize method.
        }

        #region Package Members

        /// <summary>
        /// Initialization of the package; this method is called right after the package is sited, so this is the place
        /// where you can put all the initialization code that rely on services provided by VisualStudio.
        /// </summary>
        protected override void Initialize()
        {
            // check setup
            checkSetup();

            base.Initialize();
            DTE2 dte = (DTE2)GetService(typeof(DTE));

            _wsp_connect = new WebTest.WebService.Plugin.ResultTab.Connect();
            _wsp_connect.OnConnection(dte, Extensibility.ext_ConnectMode.ext_cm_Startup, null, ref _arr);
            WebTest.WebServive.Plugin.InfoCommand.Initialize(this);

        }

        protected override void Dispose(bool disposing)
        {
            if(_wsp_connect != null)
            {
                _wsp_connect.OnDisconnection(Extensibility.ext_DisconnectMode.ext_dm_HostShutdown, ref _arr);
            }
            base.Dispose(disposing);
        }

         bool checkSetup()
        {
            RegistryKey rk = Registry.CurrentUser.OpenSubKey("SOFTWARE\\WebServicePlugin");
            bool installOK = false;
            if(rk == null)
            {
                // Registry key not found, i.e. we did not install the dll's before
                installOK = false;
            } else
            {
                // Registry key found, check if we are installed
                Object regInstalled = rk.GetValue("INSTALLED");
                if (regInstalled != null)
                {
                    Int32 val = (Int32)regInstalled;
                    if (val == 1) installOK = true;
                    else installOK = false;
                } else
                {
                    installOK = false;
                }
            }

            // Install if we need to install, and show info message when done
            if(installOK == false)
            {
                string messageLoc = new Uri(typeof(WebService.Plugin.MessageEditors.XmlMessageEditor).Assembly.CodeBase).LocalPath;
                string runtimeLoc = new Uri(typeof(WebService.Plugin.Runtime.WebServicePlugin).Assembly.CodeBase).LocalPath;

                // Check if the old runtime was found in PublicAssemblies (old version installed there)
                if (runtimeLoc.Contains("PublicAssemblies"))
                {
                    // CONFLICT
                    VsShellUtilities.ShowMessageBox(this, "Previous version of the runtime dll was found in the PublicAssemblies folder. This version will conflict with this new version of the plugin. Please remove it and restart Visual Studio.\n\n Path to the DLL: " + runtimeLoc,
                        "Conflicting previous version", OLEMSGICON.OLEMSGICON_CRITICAL, OLEMSGBUTTON.OLEMSGBUTTON_OK, OLEMSGDEFBUTTON.OLEMSGDEFBUTTON_FIRST);
                    return false;
                }
                else
                {
                    // OK to proceed
                    // Copy message editor plugins and recorder plugin to users webtest dir
                    ShellSettingsManager sm = new ShellSettingsManager(this);
                    string pluginPath = sm.GetApplicationDataFolder(ApplicationDataFolder.Documents) + @"\WebTestPlugins\";
                    string assPath = sm.GetApplicationDataFolder(ApplicationDataFolder.Documents) + @"\MyAssemblies\";
                    Directory.CreateDirectory(pluginPath);
                    Directory.CreateDirectory(assPath);
                    string messageLocNew = pluginPath + Path.GetFileName(messageLoc);
                    File.Copy(messageLoc, messageLocNew, true);
                    string runtimeLocNew = assPath + Path.GetFileName(runtimeLoc);
                    File.Copy(runtimeLoc, runtimeLocNew, true);

                    //Update the registry
                    RegistryKey wrk = Registry.CurrentUser.CreateSubKey(@"SOFTWARE\WebServicePlugin");
                    wrk.SetValue("INSTALLED", 1, RegistryValueKind.DWord);
                    RegistryKey wark = Registry.CurrentUser.CreateSubKey(@"SOFTWARE\Microsoft\.NETFramework\AssemblyFolders\MyAssemblies");
                    wark.SetValue(null, assPath);

                    //Tell the user
                    VsShellUtilities.ShowMessageBox(this, "Recorder plugin copied to: " + pluginPath + "\n\nUser assemblies copied to: " + assPath, "Plugin files installed", OLEMSGICON.OLEMSGICON_INFO, OLEMSGBUTTON.OLEMSGBUTTON_OK, OLEMSGDEFBUTTON.OLEMSGDEFBUTTON_FIRST);
                }
            }

            return installOK;
        }

        #endregion
        Connect _wsp_connect = null;
        Array _arr = new object[0];
    }
}