/*
TShock, a server mod for Terraria
Copyright (C) 2011-2018 Pryaxis & TShock Contributors
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see .
*/
using System;
using System.Collections.Generic;
using System.Net;
using System.Threading;
using Newtonsoft.Json;
using Microsoft.Xna.Framework;
using System.Net.Http;
using System.Threading.Tasks;
using TShockAPI.Extensions;
namespace TShockAPI
{
///
/// Responsible for checking for and notifying users about new updates to TShock
///
public class UpdateManager
{
private const string UpdateUrl = "https://update.tshock.co/latest/";
private HttpClient _client = new HttpClient();
///
/// Check once every X minutes.
///
private int CheckXMinutes = 30;
///
/// Creates a new instance of and starts the update thread
///
public UpdateManager()
{
//5 second timeout
_client.Timeout = new TimeSpan(0, 0, 5);
Thread t = new Thread(async () =>
{
do
{
//Sleep for X minutes
await Task.Delay(1000 * 60 * CheckXMinutes);
//Then check again
await CheckForUpdatesAsync(null);
} while (true);
})
{
Name = "TShock Update Thread",
IsBackground = true
};
t.Start();
}
private async Task CheckForUpdatesAsync(object state)
{
try
{
CheckXMinutes = 30;
await UpdateCheckAsync(state);
}
catch (Exception ex)
{
// Skip this run and check more frequently...
string msg = ex.BuildExceptionString();
//Give the console a brief
Console.ForegroundColor = ConsoleColor.Yellow;
Console.WriteLine($"UpdateManager warning: {msg}");
Console.ForegroundColor = ConsoleColor.Gray;
//And log the full exception
TShock.Log.Warn($"UpdateManager warning: {ex.ToString()}");
TShock.Log.ConsoleError("Retrying in 5 minutes.");
CheckXMinutes = 5;
}
}
///
/// Checks for updates to the TShock server
///
///
///
public async Task UpdateCheckAsync(object o)
{
var updates = await ServerIsOutOfDateAsync();
if (updates != null)
{
NotifyAdministrators(updates);
}
}
///
/// Checks to see if the server is out of date.
///
///
private async Task> ServerIsOutOfDateAsync()
{
var resp = await _client.GetAsync(UpdateUrl);
if (resp.StatusCode != HttpStatusCode.OK)
{
string reason = resp.ReasonPhrase;
if (string.IsNullOrWhiteSpace(reason))
{
reason = "none";
}
throw new WebException("Update server did not respond with an OK. "
+ $"Server message: [error {resp.StatusCode}] {reason}");
}
string json = await resp.Content.ReadAsStringAsync();
var update = JsonConvert.DeserializeObject>(json);
var version = new Version(update["version"]);
if (TShock.VersionNum.CompareTo(version) < 0)
{
return update;
}
return null;
}
private void NotifyAdministrators(Dictionary update)
{
var changes = update["changes"].Split(new[] {'\n'}, StringSplitOptions.RemoveEmptyEntries);
NotifyAdministrator(TSPlayer.Server, changes);
foreach (TSPlayer player in TShock.Players)
{
if (player != null && player.Active && player.HasPermission(Permissions.maintenance))
{
NotifyAdministrator(player, changes);
}
}
}
private void NotifyAdministrator(TSPlayer player, string[] changes)
{
player.SendMessage("The server is out of date. Latest version: ", Color.Red);
for (int j = 0; j < changes.Length; j++)
{
player.SendMessage(changes[j], Color.Red);
}
}
}
}