tribufu/ServerManagers
1
0
Fork 0
mirror of https://github.com/tribufu/ServerManagers synced 2026-05-18 09:35:48 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

source code checkin

Browse source
This commit is contained in:
Brett Hewitson 2021-01-07 16:23:23 +10:00
parent 5f8fb2c825
commit 7e57b72e35
675 changed files with 168433 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

196
src/ServerManager.Common/Lib/SteamCmdUpdater.cs Normal file
View file

@ -0,0 +1,196 @@
using ServerManagerTool.Common.Utils;
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.IO;
using System.IO.Compression;
using System.Net;
using System.Threading;
using System.Threading.Tasks;
namespace ServerManagerTool.Common.Lib
{
public delegate void ProgressDelegate(int progress, string message, bool newLine = true);
public class SteamCmdUpdater
{
public const string OUTPUT_PREFIX = "[UPDATER]";
public struct Update
{
public Update(string statusKey, float completionPercent)
{
this.StatusKey = statusKey;
this.CompletionPercent = completionPercent;
this.Cancelled = false;
this.FailureText = null;
}
public Update SetFailed(string failureText)
{
this.FailureText = failureText;
return this;
}
public static Update AsCompleted(string statusKey)
{
return new Update { StatusKey = statusKey, CompletionPercent = 100, Cancelled = false };
}
public static Update AsCancelled(string statusKey)
{
return new Update { StatusKey = statusKey, CompletionPercent = 100, Cancelled = true };
}
public string StatusKey;
public float CompletionPercent;
public bool Cancelled;
public string FailureText;
}
enum Status
{
CleaningSteamCmd,
DownloadingSteamCmd,
UnzippingSteamCmd,
RunningSteamCmd,
InstallSteamCmdComplete,
Complete,
Cancelled
}
Dictionary<Status, Update> statuses = new Dictionary<Status, Update>()
{
{ Status.CleaningSteamCmd, new Update("AutoUpdater_Status_CleaningSteamCmd", 0) },
{ Status.DownloadingSteamCmd, new Update("AutoUpdater_Status_DownloadingSteamCmd", 10) },
{ Status.UnzippingSteamCmd, new Update("AutoUpdater_Status_UnzippingSteamCmd", 30) },
{ Status.RunningSteamCmd, new Update("AutoUpdater_Status_RunningSteamCmd", 50) },
{ Status.InstallSteamCmdComplete, new Update("AutoUpdater_Status_InstallSteamCmdComplete", 80) },
{ Status.Complete, Update.AsCompleted("AutoUpdater_Status_Complete") },
{ Status.Cancelled, Update.AsCancelled("AutoUpdater_Status_Cancelled") }
};
public static string GetSteamCmdFile(string dataPath) => IOUtils.NormalizePath(Path.Combine(dataPath, CommonConfig.Default.SteamCmdRelativePath, CommonConfig.Default.SteamCmdExeFile));
public async Task ReinstallSteamCmdAsync(string dataPath, IProgress<Update> reporter, CancellationToken cancellationToken)
{
if (string.IsNullOrWhiteSpace(dataPath))
return;
try
{
reporter?.Report(statuses[Status.CleaningSteamCmd]);
string steamCmdDirectory = Path.Combine(dataPath, CommonConfig.Default.SteamCmdRelativePath);
if (Directory.Exists(steamCmdDirectory))
{
Directory.Delete(steamCmdDirectory, true);
}
await Task.Delay(5000);
await InstallSteamCmdAsync(dataPath, reporter, cancellationToken);
reporter?.Report(statuses[Status.InstallSteamCmdComplete]);
reporter?.Report(statuses[Status.Complete]);
}
catch (TaskCanceledException)
{
reporter?.Report(statuses[Status.Cancelled]);
}
catch (Exception ex)
{
reporter?.Report(statuses[Status.Complete].SetFailed(ex.ToString()));
}
}
private async Task InstallSteamCmdAsync(string dataPath, IProgress<Update> reporter, CancellationToken cancellationToken)
{
if (string.IsNullOrWhiteSpace(dataPath))
return;
string steamCmdDirectory = Path.Combine(dataPath, CommonConfig.Default.SteamCmdRelativePath);
if (!Directory.Exists(steamCmdDirectory))
{
Directory.CreateDirectory(steamCmdDirectory);
}
reporter?.Report(statuses[Status.DownloadingSteamCmd]);
// Get SteamCmd.exe if necessary
string steamCmdPath = Path.Combine(steamCmdDirectory, CommonConfig.Default.SteamCmdExeFile);
if (!File.Exists(steamCmdPath))
{
// download the SteamCMD zip file
var steamZipPath = Path.Combine(steamCmdDirectory, CommonConfig.Default.SteamCmdZipFile);
using (var webClient = new WebClient())
{
using (var cancelRegistration = cancellationToken.Register(webClient.CancelAsync))
{
await webClient.DownloadFileTaskAsync(CommonConfig.Default.SteamCmdUrl, steamZipPath);
}
}
// Unzip the downloaded file
reporter?.Report(statuses[Status.UnzippingSteamCmd]);
ZipFile.ExtractToDirectory(steamZipPath, steamCmdDirectory);
File.Delete(steamZipPath);
// Run the SteamCmd updater
reporter?.Report(statuses[Status.RunningSteamCmd]);
var arguments = SteamUtils.BuildSteamCmdArguments(false, CommonConfig.Default.SteamCmdInstallArgs);
ProcessStartInfo startInfo = new ProcessStartInfo()
{
FileName = steamCmdPath,
Arguments = arguments,
UseShellExecute = false,
};
var process = Process.Start(startInfo);
process.EnableRaisingEvents = true;
var ts = new TaskCompletionSource<bool>();
using (var cancelRegistration = cancellationToken.Register(() =>
{
try
{
process.Kill();
}
finally
{
ts.TrySetCanceled();
}
}))
{
process.Exited += (s, e) =>
{
ts.TrySetResult(process.ExitCode == 0);
};
await ts.Task;
}
}
return;
}
public async void UpdateSteamCmdAsync(string dataPath, IProgress<Update> reporter, CancellationToken cancellationToken)
{
try
{
await InstallSteamCmdAsync(dataPath, reporter, cancellationToken);
reporter?.Report(statuses[Status.InstallSteamCmdComplete]);
reporter?.Report(statuses[Status.Complete]);
}
catch (TaskCanceledException)
{
reporter?.Report(statuses[Status.Cancelled]);
}
catch(Exception ex)
{
reporter?.Report(statuses[Status.Complete].SetFailed(ex.ToString()));
}
}
}
}