public void Download(WebClient client, string updatePath) { var cancelArgs = new CancelEventArgs(false); DownloadProgressEventArgs downloadArgs = null; int fileDownloading = 0; void DownloadComplete(object sender, AsyncCompletedEventArgs args) { lock (args.UserState) { Monitor.Pulse(args.UserState); } } void DownloadProgressChanged(object sender, DownloadProgressChangedEventArgs args) { downloadArgs = new DownloadProgressEventArgs(args, fileDownloading, FilesToDownload); if (OnDownloadProgress(downloadArgs)) { ((WebClient)sender).CancelAsync(); } } switch (Type) { case ModDownloadType.Archive: { var uri = new Uri(Url); if (!uri.Host.EndsWith("github.com")) { HttpWebRequest req = (HttpWebRequest)WebRequest.Create(uri); req.Method = "HEAD"; HttpWebResponse res = (HttpWebResponse)req.GetResponse(); uri = res.ResponseUri; res.Close(); } string filePath = Path.Combine(updatePath, uri.Segments.Last()); var info = new FileInfo(filePath); if (info.Exists && info.Length == Size) { if (OnDownloadCompleted(cancelArgs)) { return; } } else { if (OnDownloadStarted(cancelArgs)) { return; } client.DownloadFileCompleted += DownloadComplete; client.DownloadProgressChanged += DownloadProgressChanged; ++fileDownloading; var sync = new object(); lock (sync) { client.DownloadFileAsync(uri, filePath, sync); Monitor.Wait(sync); } client.DownloadProgressChanged -= DownloadProgressChanged; client.DownloadFileCompleted -= DownloadComplete; if (cancelArgs.Cancel || downloadArgs?.Cancel == true) { return; } if (OnDownloadCompleted(cancelArgs)) { return; } } string dataDir = Path.Combine(updatePath, Path.GetFileNameWithoutExtension(filePath)); if (!Directory.Exists(dataDir)) { Directory.CreateDirectory(dataDir); } if (OnExtracting(cancelArgs)) { return; } Process.Start(new ProcessStartInfo("7za.exe", $"x -aoa -o\"{dataDir}\" \"{filePath}\"") { UseShellExecute = false, CreateNoWindow = true }).WaitForExit(); string workDir = Path.GetDirectoryName(ModInfo.GetModFiles(new DirectoryInfo(dataDir)).FirstOrDefault()); if (string.IsNullOrEmpty(workDir)) { throw new DirectoryNotFoundException("Unable to locate mod.ini in " + dataDir); } string newManPath = Path.Combine(workDir, "mod.manifest"); string oldManPath = Path.Combine(Folder, "mod.manifest"); if (OnParsingManifest(cancelArgs)) { return; } if (!File.Exists(newManPath) || !File.Exists(oldManPath)) { CopyDirectory(new DirectoryInfo(workDir), Directory.CreateDirectory(Folder)); Directory.Delete(dataDir, true); if (File.Exists(filePath)) { File.Delete(filePath); } return; } if (OnParsingManifest(cancelArgs)) { return; } List <ModManifest> newManifest = ModManifest.FromFile(newManPath); if (OnApplyingManifest(cancelArgs)) { return; } List <ModManifest> oldManifest = ModManifest.FromFile(oldManPath); List <string> oldFiles = oldManifest.Except(newManifest) .Select(x => Path.Combine(Folder, x.FilePath)) .ToList(); foreach (string file in oldFiles) { if (File.Exists(file)) { File.Delete(file); } } RemoveEmptyDirectories(oldManifest, newManifest); foreach (ModManifest file in newManifest) { string dir = Path.GetDirectoryName(file.FilePath); if (!string.IsNullOrEmpty(dir)) { string newDir = Path.Combine(Folder, dir); if (!Directory.Exists(newDir)) { Directory.CreateDirectory(newDir); } } string dest = Path.Combine(Folder, file.FilePath); if (File.Exists(dest)) { File.Delete(dest); } File.Move(Path.Combine(workDir, file.FilePath), dest); } File.Copy(newManPath, oldManPath, true); Directory.Delete(dataDir, true); File.WriteAllText(Path.Combine(Folder, "mod.version"), Updated); if (File.Exists(filePath)) { File.Delete(filePath); } break; } case ModDownloadType.Modular: { // First let's download all the new stuff. List <ModManifestDiff> newEntries = ChangedFiles .Where(x => x.State == ModManifestState.Added || x.State == ModManifestState.Changed) .ToList(); var uri = new Uri(Url); string tempDir = Path.Combine(updatePath, uri.Segments.Last()); if (!Directory.Exists(tempDir)) { Directory.CreateDirectory(tempDir); } var sync = new object(); foreach (ModManifestDiff i in newEntries) { string filePath = Path.Combine(tempDir, i.Current.FilePath); string dir = Path.GetDirectoryName(filePath); if (!string.IsNullOrEmpty(dir) && !Directory.Exists(dir)) { Directory.CreateDirectory(dir); } if (OnDownloadStarted(cancelArgs)) { return; } var info = new FileInfo(filePath); ++fileDownloading; if (!info.Exists || info.Length != i.Current.FileSize || !i.Current.Checksum.Equals(ModManifestGenerator.GetFileHash(filePath), StringComparison.InvariantCultureIgnoreCase)) { client.DownloadFileCompleted += DownloadComplete; client.DownloadProgressChanged += DownloadProgressChanged; lock (sync) { client.DownloadFileAsync(new Uri(uri, i.Current.FilePath), filePath, sync); Monitor.Wait(sync); } client.DownloadProgressChanged -= DownloadProgressChanged; client.DownloadFileCompleted -= DownloadComplete; info.Refresh(); if (info.Length != i.Current.FileSize) { throw new Exception(string.Format("Size of downloaded file \"{0}\" ({1}) differs from manifest ({2}).", i.Current.FilePath, SizeSuffix.GetSizeSuffix(info.Length), SizeSuffix.GetSizeSuffix(i.Current.FileSize))); } var hash = ModManifestGenerator.GetFileHash(filePath); if (!i.Current.Checksum.Equals(hash, StringComparison.InvariantCultureIgnoreCase)) { throw new Exception(string.Format("Checksum of downloaded file \"{0}\" ({1}) differs from manifest ({2}).", i.Current.FilePath, hash, i.Current.Checksum)); } } if (cancelArgs.Cancel || downloadArgs?.Cancel == true) { return; } if (OnDownloadCompleted(cancelArgs)) { return; } } client.DownloadFileCompleted += DownloadComplete; lock (sync) { client.DownloadFileAsync(new Uri(uri, "mod.manifest"), Path.Combine(tempDir, "mod.manifest"), sync); Monitor.Wait(sync); } client.DownloadFileCompleted -= DownloadComplete; // Now handle all file operations except where removals are concerned. List <ModManifestDiff> movedEntries = ChangedFiles.Except(newEntries) .Where(x => x.State == ModManifestState.Moved) .ToList(); if (OnApplyingManifest(cancelArgs)) { return; } // Handle existing entries marked as moved. foreach (ModManifestDiff i in movedEntries) { ModManifest old = i.Last; // This would be considered an error... if (old == null) { continue; } string oldPath = Path.Combine(Folder, old.FilePath); string newPath = Path.Combine(tempDir, i.Current.FilePath); string dir = Path.GetDirectoryName(newPath); if (!string.IsNullOrEmpty(dir) && !Directory.Exists(dir)) { Directory.CreateDirectory(dir); } File.Copy(oldPath, newPath, true); } // Now move the stuff from the temporary folder over to the working directory. foreach (ModManifestDiff i in newEntries.Concat(movedEntries)) { string tempPath = Path.Combine(tempDir, i.Current.FilePath); string workPath = Path.Combine(Folder, i.Current.FilePath); string dir = Path.GetDirectoryName(workPath); if (!string.IsNullOrEmpty(dir) && !Directory.Exists(dir)) { Directory.CreateDirectory(dir); } File.Copy(tempPath, workPath, true); } // Once that has succeeded we can safely delete files that have been marked for removal. List <ModManifestDiff> removedEntries = ChangedFiles .Where(x => x.State == ModManifestState.Removed) .ToList(); foreach (string path in removedEntries.Select(i => Path.Combine(Folder, i.Current.FilePath)).Where(File.Exists)) { File.Delete(path); } // Same for files that have been moved. foreach (string path in movedEntries .Where(x => newEntries.All(y => y.Current.FilePath != x.Last.FilePath)) .Select(i => Path.Combine(Folder, i.Last.FilePath)).Where(File.Exists)) { File.Delete(path); } string oldManPath = Path.Combine(Folder, "mod.manifest"); string newManPath = Path.Combine(tempDir, "mod.manifest"); if (File.Exists(oldManPath)) { List <ModManifest> oldManifest = ModManifest.FromFile(oldManPath); List <ModManifest> newManifest = ModManifest.FromFile(newManPath); // Remove directories that are now empty. RemoveEmptyDirectories(oldManifest, newManifest); } // And last but not least, copy over the new manifest. File.Copy(newManPath, oldManPath, true); break; } default: throw new ArgumentOutOfRangeException(); } }
public ModDownload CheckModularVersion(ModInfo mod, string folder, List <ModManifestEntry> localManifest, UpdaterWebClient client, List <string> errors) { if (!mod.UpdateUrl.StartsWith("http://", StringComparison.InvariantCulture) && !mod.UpdateUrl.StartsWith("https://", StringComparison.InvariantCulture)) { mod.UpdateUrl = "http://" + mod.UpdateUrl; } if (!mod.UpdateUrl.EndsWith("/", StringComparison.InvariantCulture)) { mod.UpdateUrl += "/"; } var url = new Uri(mod.UpdateUrl); url = new Uri(url, "mod.ini"); ModInfo remoteInfo; try { Dictionary <string, Dictionary <string, string> > dict = IniFile.IniFile.Load(client.OpenRead(url)); remoteInfo = IniSerializer.Deserialize <ModInfo>(dict); } catch (Exception ex) { errors.Add($"[{mod.Name}] Error pulling mod.ini from \"{mod.UpdateUrl}\": {ex.Message}"); return(null); } if (!ForceUpdate && remoteInfo.Version == mod.Version) { return(null); } string manString; try { manString = client.DownloadString(new Uri(new Uri(mod.UpdateUrl), "mod.manifest")); } catch (Exception ex) { errors.Add($"[{mod.Name}] Error pulling mod.manifest from \"{mod.UpdateUrl}\": {ex.Message}"); return(null); } List <ModManifestEntry> remoteManifest; try { remoteManifest = ModManifest.FromString(manString); } catch (Exception ex) { errors.Add($"[{mod.Name}] Error parsing remote manifest from \"{mod.UpdateUrl}\": {ex.Message}"); return(null); } List <ModManifestDiff> diff = ModManifestGenerator.Diff(remoteManifest, localManifest); if (diff.Count < 1 || diff.All(x => x.State == ModManifestState.Unchanged)) { return(null); } string changes; if (!string.IsNullOrEmpty(mod.ChangelogUrl)) { try { changes = client.DownloadString(new Uri(mod.ChangelogUrl)); } catch (Exception ex) { changes = ex.Message; } } else { try { changes = client.DownloadString(new Uri(new Uri(mod.UpdateUrl), "changelog.txt")); } catch { // ignored changes = string.Empty; } } if (!string.IsNullOrEmpty(changes)) { changes = Regex.Replace(changes, "(?<!\r)\n", "\r\n"); } return(new ModDownload(mod, Path.Combine("mods", folder), mod.UpdateUrl, changes, diff)); }