-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathProgram.cs
More file actions
130 lines (107 loc) · 4.28 KB
/
Program.cs
File metadata and controls
130 lines (107 loc) · 4.28 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
using System;
using System.Diagnostics;
using System.IO;
using System.IO.Compression;
using System.Net.Http;
using System.Net.Http.Headers;
using System.Threading;
using System.Threading.Tasks;
class Program
{
static async Task Main(string[] args)
{
if (args.Length < 4) return;
string mainApp = args[0];
string downloadUrl = args[1];
string destinationDir = args[2];
string excludedFolders = args[3];
try
{
Console.WriteLine("Waiting for the main application to complete...");
string mainAppName = Path.GetFileNameWithoutExtension(mainApp);
Process[] processes = Process.GetProcessesByName(mainAppName);
foreach (var process in processes) process.WaitForExit(10000);
Thread.Sleep(1000);
Console.WriteLine("Downloading update...");
string tempZipPath = Path.Combine(Path.GetTempPath(), "update.zip");
using (var client = new HttpClient())
{
client.Timeout = Timeout.InfiniteTimeSpan;
client.DefaultRequestHeaders.ConnectionClose = false;
client.DefaultRequestHeaders.UserAgent.ParseAdd(
"Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36");
client.DefaultRequestHeaders.CacheControl = new CacheControlHeaderValue
{
NoCache = true,
NoStore = true
};
var response = await client.GetAsync(downloadUrl, HttpCompletionOption.ResponseHeadersRead);
response.EnsureSuccessStatusCode();
using (var contentStream = await response.Content.ReadAsStreamAsync())
using (var fileStream = new FileStream
(
tempZipPath,
FileMode.Create,
FileAccess.Write,
FileShare.None,
81920,
true
))
{
await contentStream.CopyToAsync(fileStream, 81920);
}
}
Console.WriteLine("Unpacking update...");
var excluded = excludedFolders.Split(';');
string tempExtractDir = Path.Combine(Path.GetTempPath(), "update_extract");
if (Directory.Exists(tempExtractDir)) Directory.Delete(tempExtractDir, true);
ZipFile.ExtractToDirectory(tempZipPath, tempExtractDir);
CopyDirectory(tempExtractDir, destinationDir, excluded);
Console.WriteLine("Launching main application...");
Process.Start(mainApp);
File.Delete(tempZipPath);
Directory.Delete(tempExtractDir, true);
Console.WriteLine("Update completed!");
}
catch (Exception ex)
{
Console.WriteLine($"Error: {ex.Message}");
Console.ReadKey();
}
}
static void CopyDirectory(string sourceDir, string destinationDir, string[] excludedFolders)
{
try
{
if (!Directory.Exists(destinationDir)) Directory.CreateDirectory(destinationDir);
foreach (string file in Directory.GetFiles(sourceDir))
{
string destinationFile = Path.Combine(destinationDir, Path.GetFileName(file));
File.Copy(file, destinationFile, true);
}
foreach (string directory in Directory.GetDirectories(sourceDir))
{
string directoryName = Path.GetFileName(directory);
bool shouldExclude = false;
foreach (string excluded in excludedFolders)
{
if (directoryName.Equals(excluded, StringComparison.OrdinalIgnoreCase))
{
shouldExclude = true;
break;
}
}
if (!shouldExclude)
{
string destinationSubDir = Path.Combine(destinationDir, directoryName);
CopyDirectory(directory, destinationSubDir, excludedFolders);
}
}
}
catch (Exception ex)
{
Console.WriteLine($"Error: {ex.Message}");
Console.ReadKey();
}
}
}