// EXECUTE METHOD IN LIBS WITHOUT LOCK FILE
 private static object Call(string assemblyPath, string typeName, string method, Type[] paramTypes, bool createInstance, params object[] methodParams) {
     AppDomain domain = AppDomain.CreateDomain(UPDATER_DOMAIN);
     AssemblyCaller assemblyCaller = new AssemblyCaller();
     assemblyCaller = (AssemblyCaller)domain.CreateInstanceAndUnwrap(Assembly.GetExecutingAssembly().FullName, typeof(AssemblyCaller).FullName);
     object result = assemblyCaller.PrivateCall(assemblyPath, typeName, method, paramTypes, createInstance, methodParams);
     AppDomain.Unload(domain);
     return result;
 }
예제 #2
0
        // EXECUTE METHOD IN LIBS WITHOUT LOCK FILE
        private static object Call(string assemblyPath, string typeName, string method, Type[] paramTypes, bool createInstance, params object[] methodParams)
        {
            AppDomain      domain         = AppDomain.CreateDomain(UPDATER_DOMAIN);
            AssemblyCaller assemblyCaller = (AssemblyCaller)domain.CreateInstanceAndUnwrap(Assembly.GetExecutingAssembly().FullName, typeof(AssemblyCaller).FullName);
            object         result         = assemblyCaller.PrivateCall(assemblyPath, typeName, method, paramTypes, createInstance, methodParams);

            AppDomain.Unload(domain);
            return(result);
        }
        private void MergeXmlRepository(string sourceFile, string destFile)
        {
            var repo = String.Empty;

            switch (Path.GetFileName(sourceFile))
            {
            case "RocksmithToolkitLib.Config.xml":
                repo = "ConfigRepository";
                break;

            case "RocksmithToolkitLib.SongAppId.xml":
                repo = "SongAppIdRepository";
                break;

            case "RocksmithToolkitLib.TuningDefinition.xml":
                repo = "TuningDefinitionRepository";
                break;
            }

            if (!String.IsNullOrEmpty(repo))
            {
                AssemblyCaller.CallStatic(Path.Combine(workDir, APP_RSLIB), String.Format("RocksmithToolkitLib.XmlRepository.{0}", repo), "Merge", sourceFile, destFile);
            }
        }
        /// <summary>
        /// AutoUpdater with command line args for localToolkitDir and tempToolkitDir
        /// </summary>
        public AutoUpdaterForm(string[] args)
        {
            InitializeComponent();

            // force showing autoupdater and progress bar on startup
            this.Show();
            this.Location = new Point(100, 100);
            this.BringToFront();
            ShowCurrentOperation("Starting the AutoUpdater Engine ...");
            pbUpdate.Style = ProgressBarStyle.Marquee;
            pbUpdate.Refresh();
            this.Refresh();

            // catch if there are no cmd line arguments
            if (args.GetLength(0) == 0)
            {
                args = new string[1] {
                    "?"
                }
            }
            ;
            if (args[0].Equals("?"))
            {
                StringBuilder sb = new StringBuilder();
                sb.AppendLine("Song Creator Toolkit for Rocksmith");
                sb.AppendLine("- commonly known as, 'the toolkit'");
                sb.AppendLine("");
                sb.AppendLine("You are probably seeing this informative message");
                sb.AppendLine("because you started the RocksmithToolkitUpdater.exe");
                sb.AppendLine("application by double clicking on it directly.");
                sb.AppendLine("");
                sb.AppendLine("- Normal Usage:");
                sb.AppendLine("  AutoUpdater is normally run programmatically by the");
                sb.AppendLine("  toolkit to automatically update the toolkit files.");
                sb.AppendLine("");
                sb.AppendLine("- Alternate Usage:");
                sb.AppendLine("  AutoUpdater can be run by double clicking on the");
                sb.AppendLine("  application.  This will force the toolkit to be");
                sb.AppendLine("  updated to the latest available online revision.");
                sb.AppendLine("");
                //sb.AppendLine("- WARNING:");
                //sb.AppendLine("  All user customized toolkit settings are overwritten");
                //sb.AppendLine("  if AutoUpdater is run in the Alternate Usage mode.");
                //sb.AppendLine("  A warning will popup during update to remind you.");
                //sb.AppendLine("");
                sb.AppendLine("Continue running AutoUpdater in Alternate Usage mode?   ");

                if (DialogResult.Yes != MessageBox.Show(sb.ToString(), "AutoUpdater", MessageBoxButtons.YesNo, MessageBoxIcon.Information))
                {
                    Environment.Exit(0);
                }

                // confirm toolkit process is not running before continuing
                Thread.Sleep(500);
                Process[] processesByName = Process.GetProcessesByName("RocksmithToolkitGUI");
                if (processesByName.Length != 0)
                {
                    MessageBox.Show("<ERROR> Detected that RocksmithToolkitGUI is running ..." + Environment.NewLine +
                                    "The toolkit must be closed before running the AutoUpdater.  ", "RocksmithToolkit AutoUpdater", MessageBoxButtons.OK, MessageBoxIcon.Error);

                    Environment.Exit(2);
                }
            }

            // turn on/off debugging MessageBox displays
            isDebugMe      = false;
            isInDesignMode = Helpers.IsInDesignMode;
            appExecPath    = Application.ExecutablePath;
            appExecDir     = Path.GetDirectoryName(appExecPath);
            appExecFile    = Path.GetFileName(appExecPath);
            // if (isInDesignMode) debugMe = true; // overrides initiation

            // running RocksmithToolkitUpdating.exe programatically (Primary Usage Mode)
            if (args.Length == 2 && appExecFile.Equals(APP_UPDATING, StringComparison.InvariantCultureIgnoreCase) || appExecFile.Equals(APP_RSGUI, StringComparison.InvariantCultureIgnoreCase))
            {
                if (isDebugMe)
                {
                    MessageBox.Show("Starting Auto Update ... Primary Usage", "DPDM");
                }

                localToolkitDir = args[0];
                tempToolkitDir  = args[1];
            }
            // running RocksmithToolkitUpdater.exe (Alternate Usage Mode) or developer running project in VS IDE Debug mode
            else if (appExecFile.Equals(APP_UPDATER, StringComparison.InvariantCultureIgnoreCase))
            {
                // the user double clicked on RocksmithToolkitUpdater.exe (w/o cmd line args)
                localToolkitDir = appExecDir;
                tempToolkitDir  = Path.Combine(Path.GetTempPath(), "RocksmithToolkit");

                if (Directory.Exists(tempToolkitDir))
                {
                    Directory.Delete(tempToolkitDir, true);
                }

                Directory.CreateDirectory(tempToolkitDir);

                // copy required files for debugging the AutoUpdater as a standalone project in VS IDE Debug mode
                if (isInDesignMode)
                {
                    if (isDebugMe)
                    {
                        MessageBox.Show("Starting Alternate Usage In Design Mode ...", "DPDM");
                    }

                    try
                    {
                        var rootBinDebugDir = Path.Combine(Directory.GetParent(Directory.GetCurrentDirectory()).Parent.Parent.FullName, "RocksmithTookitGUI\\bin\\Debug");
                        File.Copy(Path.Combine(rootBinDebugDir, APP_CSZIPLIB), Path.Combine(appExecDir, APP_CSZIPLIB), true);
                        File.Copy(Path.Combine(rootBinDebugDir, APP_RSLIB), Path.Combine(appExecDir, APP_RSLIB), true);
                        File.Copy(Path.Combine(rootBinDebugDir, APP_RSGUI), Path.Combine(appExecDir, APP_RSGUI), true);
                        File.Copy(Path.Combine(rootBinDebugDir, REPO_CONFIG), Path.Combine(appExecDir, REPO_CONFIG), true);
                        File.Copy(Path.Combine(rootBinDebugDir, REPO_SONGAPPID), Path.Combine(appExecDir, REPO_SONGAPPID), true);
                        File.Copy(Path.Combine(rootBinDebugDir, REPO_TUNINGDEF), Path.Combine(appExecDir, REPO_TUNINGDEF), true);
                    }
                    catch (Exception ex)
                    {
                        MessageBox.Show("<ERROR> Can not find required file(s) to run AutoUpdater project in VS IDE Debug mode.  " + Environment.NewLine + "Make sure the RocksmithToolkitGUI project has been run in VS IDE Debug mode first. " + Environment.NewLine + Environment.NewLine + ex.Message, "DPDM");

                        Environment.Exit(1);
                    }
                }
                else
                {
                    // use some whacky, hacky, trickery
                    // make a copy of AutoUpdater to prevent locking the process during update
                    var updaterAppPath  = Path.Combine(localToolkitDir, APP_UPDATER);
                    var updatingAppPath = Path.Combine(tempToolkitDir, APP_UPDATING);
                    File.Copy(updaterAppPath, updatingAppPath, true);
                    var cmdArgs = String.Format("\"{0}\" \"{1}\"", localToolkitDir, tempToolkitDir);

                    if (isDebugMe)
                    {
                        MessageBox.Show("Starting Auto Update ... Alternate Usage Release Mode" + Environment.NewLine + "cmdArgs = " + cmdArgs, "DPDM");
                    }

                    var startInfo = new ProcessStartInfo
                    {
                        FileName        = updatingAppPath,
                        Arguments       = cmdArgs,
                        UseShellExecute = false,
                        CreateNoWindow  = true, // hide command window
                    };

                    using (var updater = new Process())
                    {
                        updater.StartInfo = startInfo;
                        updater.Start();
                    }

                    Thread.Sleep(500);
                    // Kill current process (RocksmithToolkitUpdater.exe) now
                    // that new process (RocksmithToolkitUpdating.exe) is started
                    Environment.Exit(0);
                }
            }
            else
            {
                MessageBox.Show("<ERROR> Unexpected updater usage ..." + Environment.NewLine + "appExecFile = " + appExecFile, "RocksmithToolkit AutoUpdater", MessageBoxButtons.OK, MessageBoxIcon.Error);

                Environment.Exit(1);
            }

            localToolkitDirRoot = Path.GetDirectoryName(localToolkitDir);
            // toolkit archive directory/file structure effects newLocalToolkitDir path
            newLocalToolkitDir = Path.Combine(localToolkitDirRoot, "RocksmithToolkit");

            if (isDebugMe)
            {
                MessageBox.Show("IsDesignMode = " + isInDesignMode.ToString() + Environment.NewLine + "Currently running: " + Application.ExecutablePath + Environment.NewLine + "localToolkitDir = " + localToolkitDir + Environment.NewLine + "newlocalToolkitDir = " + newLocalToolkitDir + Environment.NewLine + "localToolkitDirRoot = " + localToolkitDirRoot + Environment.NewLine + "tempToolkitDir = " + tempToolkitDir + Environment.NewLine + "args[0] = " + args[0] + Environment.NewLine + "args[1] = " + args[1], "DPDM");
            }

            // switch progress bar style
            pbUpdate.Style = ProgressBarStyle.Continuous;
            pbUpdate.Value = 0;

            try
            {
                // backup the local process
                BackupProcessDir(localToolkitDir, tempToolkitDir);
            }
            catch (Exception ex)
            {
                MessageBox.Show("<ERROR> Process backup failure ..." + Environment.NewLine + "Please manually download and install the latest toolkit revision." + Environment.NewLine + ex.Message, "RocksmithToolkit AutoUpdater", MessageBoxButtons.OK, MessageBoxIcon.Error);

                Environment.Exit(1);
            }

            try
            {
                // www.rscustom.net/builds/latest.zip or *.tar.gz
                // www.rscustom.net/builds/latest_test.zip or *.tar.gz
                // get latest zip url
                latestZipUrl = (string)AssemblyCaller.Call(Path.Combine(appExecDir, APP_RSLIB), "RocksmithToolkitLib.ToolkitVersionOnline", "GetFileUrl", null, true);

                if (String.IsNullOrEmpty(latestZipUrl))
                {
                    throw new Exception("latestZipUrl is null/empty");
                }
            }
            catch (Exception ex)
            {
                MessageBox.Show("<ERROR> latestZipUrl AssemblyCaller failure ..." + Environment.NewLine + ex.InnerException.Message, "RocksmithToolkit AutoUpdater", MessageBoxButtons.OK, MessageBoxIcon.Error);

                Environment.Exit(1);
            }

            var latestZipUri = new Uri(latestZipUrl);

            latestZipPath = Path.Combine(tempToolkitDir, Path.GetFileName(latestZipUri.LocalPath));
            DownloadFile(latestZipUri, latestZipPath);

            if (isDebugMe)
            {
                MessageBox.Show("Check backup files: " + tempToolkitDir + Environment.NewLine + "latestZipUri dlStatus: " + dlStatus.ToString() + Environment.NewLine + "latestZipPath: " + latestZipPath, "DPDM");
                //ExtractFile(latestZipPath, localToolkitDirRoot);
                //MessageBox.Show("Check ExtractFile destination: " + localToolkitDirRoot, "DPDM");
                //MergeXmlRepository(tempToolkitDir, newLocalToolkitDir);
                //MessageBox.Show("Check MergeXmlRepository destination: " + newLocalToolkitDir, "DPDM");
            }

            if (dlStatus == DownloadStatus.SUCCESS && File.Exists(latestZipPath))
            {
                // bulldoze the local process directory
                if (args.Length == 2 && appExecFile.Equals(APP_UPDATING, StringComparison.InvariantCultureIgnoreCase))
                {
                    var lockedLocalFiles = DeleteDirectory(localToolkitDir);
                    if (lockedLocalFiles.Any())
                    {
                        ShowCurrentOperation("<WARNING> localToolkitDir cleanup failed ...");
                        if (!ShowLockedFilesAndContinue(lockedLocalFiles))
                        {
                            Environment.Exit(1);
                        }
                    }
                }

                try
                {
                    // extract latest toolkit revision to the localToolkitDirRoot
                    // revised archive directory structure to be more like an installer
                    ExtractFile(latestZipPath, localToolkitDirRoot);

                    if (isDebugMe)
                    {
                        MessageBox.Show("Check unzipped files in: " + localToolkitDirRoot, "DPDM");
                    }
                }
                catch (Exception ex)
                {
                    if (DialogResult.No == MessageBox.Show("<ERROR> Could not unzip file: " + Path.GetFileName(latestZipPath) + Environment.NewLine +
                                                           "The AutoUpdater can not continue." + Environment.NewLine +
                                                           "Do you want to roll back the installation process?" + Environment.NewLine +
                                                           ex.Message, "RocksmithToolkit AutoUpdater", MessageBoxButtons.YesNo, MessageBoxIcon.Error))
                    {
                        Environment.Exit(1);
                    }

                    // rollback the process to its original state
                    DeleteDirectory(newLocalToolkitDir);
                    if (!Directory.Exists(localToolkitDir))
                    {
                        Directory.CreateDirectory(localToolkitDir);
                    }

                    RollBack(tempToolkitDir, localToolkitDir);
                    RestartToolkitGUI(localToolkitDir);
                }

                try
                {
                    // merge xml repo files
                    MergeXmlRepository(tempToolkitDir, newLocalToolkitDir);
                }
                catch (Exception ex)
                {
                    MessageBox.Show("<ERROR> Could not merge repositories ... " + Environment.NewLine + ex.Message, "RocksmithToolkit AutoUpdater", MessageBoxButtons.OK, MessageBoxIcon.Error);
                }

                // merge cgm inlay files
                var cgmFiles = Directory.EnumerateFiles(Path.Combine(tempToolkitDir, "cgm"), "*", SearchOption.AllDirectories);
                foreach (var cgmFile in cgmFiles)
                {
                    try
                    {
                        File.Copy(cgmFile, cgmFile.Replace(tempToolkitDir, newLocalToolkitDir));
                    }
                    catch { /* Do nothing */ }
                }

                // TODO: merge custom/user dds xml/cfg files

                // cleanup tempToolkitDir
                var lockedTempFiles = DeleteDirectory(tempToolkitDir);
                if (lockedTempFiles.Any())
                {
                    ShowCurrentOperation("<WARNING> tempToolkitDir cleanup failed ...");
                    if (!ShowLockedFilesAndContinue(lockedTempFiles))
                    {
                        Environment.Exit(1);
                    }
                }
            }

            if (File.Exists(latestZipPath))
            {
                File.Delete(latestZipPath);
            }

            if (localToolkitDir != newLocalToolkitDir)
            {
                // find open localToolkitDir and attempt to close it so it can be deleted
                var shellWindows = new SHDocVw.ShellWindows();
                foreach (SHDocVw.InternetExplorer shellWindow in shellWindows)
                {
                    var processType      = Path.GetFileNameWithoutExtension(shellWindow.FullName).ToLower();
                    var cleanLocationUrl = shellWindow.LocationURL.ToLower().Replace(@"/", @"\");
                    if (processType.Equals("explorer") && cleanLocationUrl.Contains(localToolkitDir.ToLower()))
                    {
                        shellWindow.Quit();
                        Thread.Sleep(100);
                        shellWindow.Quit();
                        Thread.Sleep(100);

                        MessageBox.Show("The old local toolkit directory should now be closed: " + Environment.NewLine +
                                        localToolkitDir + Environment.NewLine + Environment.NewLine +
                                        "Manually close the directory if it is still open" + Environment.NewLine +
                                        "before pressing 'OK' to continue.  The old toolkit" + Environment.NewLine +
                                        "may be deleted after the update finishes ...", "Close Toolkit Directory ...", MessageBoxButtons.OK, MessageBoxIcon.Hand);
                        break;
                    }
                }

                // attempt to delete the closed directory
                DeleteDirectory(localToolkitDir);
            }

            if (isDebugMe)
            {
                MessageBox.Show("Before Restart: " + Environment.NewLine +
                                "Check localToolkitDir deleted: " + localToolkitDir + Environment.NewLine +
                                "Check tempToolkitDir deleted: " + tempToolkitDir + Environment.NewLine +
                                "Check latestZipPath deleted: " + latestZipPath, "DPDM");
            }

            ShowCurrentOperation("Please wait ... Restarting ToolkitGUI ...");
            Thread.Sleep(1000); // settle down before restart
            RestartToolkitGUI(newLocalToolkitDir);
        }
        private void MergeXmlRepository(string srcDir, string destDir)
        {
            ShowCurrentOperation("Preparing to merge repositories ...");

            // this check must happen after download has been unzipped to the newLocalToolkitPath
            try
            {
                bool?replaceRepo = null;
                // get old replace repo from old ConfigRepo
                replaceRepo = (bool)AssemblyCaller.CallStatic(Path.Combine(appExecDir, APP_RSLIB), "RocksmithToolkitLib.XmlRepository.ConfigRepository", "GetBoolean", "general_replacerepo");
                // get old config version from old ConfigRepo
                var xmlConfigVersion = (string)AssemblyCaller.CallStatic(Path.Combine(appExecDir, APP_RSLIB), "RocksmithToolkitLib.XmlRepository.ConfigRepository", "GetString", "general_configversion");
                // get new config version from new GeneralConfig
                var generalConfigVersion = (string)AssemblyCaller.CallStatic(Path.Combine(appExecDir, APP_RSGUI), "RocksmithToolkitGUI.Config.GeneralConfig", "GetConfigVersion", null);

                if (String.IsNullOrEmpty(xmlConfigVersion))
                {
                    replaceRepo = true;
                }
                else if (String.IsNullOrEmpty(generalConfigVersion))
                {
                    replaceRepo = true;
                }
                else if (xmlConfigVersion != generalConfigVersion) // compare versions
                {
                    replaceRepo = true;
                }
                else if (replaceRepo == null)
                {
                    replaceRepo = true;
                }

                if (replaceRepo == true)
                {
                    if (isDebugMe)
                    {
                        MessageBox.Show("XmlRepositories were not merged ..." + Environment.NewLine +
                                        "replaceRepo = " + replaceRepo.ToString() + Environment.NewLine +
                                        "xmlConfigVersion = " + xmlConfigVersion + Environment.NewLine +
                                        "generalConfigVersion = " + generalConfigVersion, "DEBUG ME");
                    }

                    return;
                }
            }
            catch (Exception ex)
            {
                MessageBox.Show("<WARNING> XmlRepositories could not be merged ..." + Environment.NewLine +
                                ex.Message, "RocksmithToolkit AutoUpdater", MessageBoxButtons.OK, MessageBoxIcon.Error);

                return;
            }

            ShowCurrentOperation("Merging repositories ...");
            var repositories = new Dictionary <string, string>()
            {
                { "ConfigRepository", REPO_CONFIG },
                { "SongAppIdRepository", REPO_SONGAPPID },
                { "TuningDefinitionRepository", REPO_TUNINGDEF }
            };

            foreach (KeyValuePair <string, string> repo in repositories)
            {
                var srcPath  = Path.Combine(srcDir, repo.Value);
                var destPath = Path.Combine(destDir, repo.Value);

                // merge if srcPath and destPath exist
                if (File.Exists(srcPath) && File.Exists(destPath))
                {
                    try
                    {
                        AssemblyCaller.CallStatic(Path.Combine(appExecDir, APP_RSLIB), String.Format("RocksmithToolkitLib.XmlRepository.{0}", repo.Key), "Merge", srcPath, destPath);
                    }
                    catch (Exception ex)
                    {
                        MessageBox.Show("<ERROR> MergeXmlRepository file, AssemblyCaller failure ..." + Environment.NewLine +
                                        "srcPath = " + srcPath + Environment.NewLine +
                                        "destPath = " + destPath + Environment.NewLine +
                                        ex.InnerException.Message, "RocksmithToolkit AutoUpdater", MessageBoxButtons.OK, MessageBoxIcon.Error);
                    }
                }
            }
        }
 private void ExtractFile(string srcPath, string destDir)
 {
     // only works for zip files
     ShowCurrentOperation("Extracting: " + srcPath);
     AssemblyCaller.Call(Path.Combine(tempToolkitDir, APP_CSZIPLIB), "ICSharpCode.SharpZipLib.Zip.FastZip", "ExtractZip", new Type[] { typeof(string), typeof(string), typeof(string) }, new object[] { srcPath, destDir, null });
 }
 public void ExtractFile(string output)   //works only for ZIP files.
 {
     AssemblyCaller.Call(Path.Combine(workDir, APP_CSZIP), "ICSharpCode.SharpZipLib.Zip.FastZip", "ExtractZip", new Type[] { typeof(string), typeof(string), typeof(string) }, new object[] { localFile, output, null });
 }