Implement update check on startup

JustArchiNET · Oct 29, 2015 · a0ba148 · a0ba148
1 parent abe3bea
commit a0ba148
Show file tree

Hide file tree

Showing 21 changed files with 53,431 additions and 9 deletions.
diff --git a/ArchiSteamFarm/ArchiSteamFarm.csproj b/ArchiSteamFarm/ArchiSteamFarm.csproj
@@ -56,6 +56,10 @@
       <HintPath>..\packages\HtmlAgilityPack.1.4.9\lib\Net45\HtmlAgilityPack.dll</HintPath>
       <Private>True</Private>
     </Reference>
+    <Reference Include="Newtonsoft.Json, Version=8.0.0.0, Culture=neutral, PublicKeyToken=30ad4fe6b2a6aeed, processorArchitecture=MSIL">
+      <HintPath>..\packages\Newtonsoft.Json.8.0.1-beta1\lib\net45\Newtonsoft.Json.dll</HintPath>
+      <Private>True</Private>
+    </Reference>
     <Reference Include="protobuf-net, Version=2.0.0.668, Culture=neutral, PublicKeyToken=257b51d87d2e4d67, processorArchitecture=MSIL">
       <HintPath>..\packages\protobuf-net.2.0.0.668\lib\net40\protobuf-net.dll</HintPath>
       <Private>True</Private>

diff --git a/ArchiSteamFarm/Bot.cs b/ArchiSteamFarm/Bot.cs
@@ -55,7 +55,7 @@ internal class Bot {
 		internal Trading Trading { get; private set; }
 
 		// Config variables
-		private bool Enabled { get { return bool.Parse(Config["Enabled"]); } }
+		internal bool Enabled { get { return bool.Parse(Config["Enabled"]); } }
 		private string SteamLogin { get { return Config["SteamLogin"]; } }
 		private string SteamPassword { get { return Config["SteamPassword"]; } }
 		private string SteamNickname { get { return Config["SteamNickname"]; } }
@@ -76,7 +76,6 @@ internal Bot(string botName) {
 			ReadConfig();
 
 			if (!Enabled) {
-				Logging.LogGenericInfo(BotName, "Not starting this instance, because it's disabled in config file");
 				return;
 			}
 

diff --git a/ArchiSteamFarm/Logging.cs b/ArchiSteamFarm/Logging.cs
@@ -50,6 +50,10 @@ internal static void LogGenericInfo(string botName, string message, [CallerMembe
 			Log("[*] INFO: " + previousMethodName + "() <" + botName + "> " + message);
 		}
 
+		internal static void LogGenericNotice(string botName, string message, [CallerMemberName] string previousMethodName = "") {
+			Log("[*] NOTICE: " + previousMethodName + "() <" + botName + "> " + message);
+		}
+
 		[Conditional("DEBUG")]
 		internal static void LogGenericDebug(string botName, string message, [CallerMemberName] string previousMethodName = "") {
 			Log("[#] DEBUG: " + previousMethodName + "() <" + botName + "> " + message);

diff --git a/ArchiSteamFarm/Program.cs b/ArchiSteamFarm/Program.cs
@@ -22,10 +22,13 @@ limitations under the License.
 
 */
 
+using Newtonsoft.Json.Linq;
 using System;
 using System.Collections.Generic;
 using System.IO;
+using System.Reflection;
 using System.Threading;
+using System.Threading.Tasks;
 
 namespace ArchiSteamFarm {
 	internal static class Program {
@@ -39,8 +42,33 @@ internal enum EUserInputType {
 
 		internal const ulong ArchiSCFarmGroup = 103582791440160998;
 		internal const string ConfigDirectoryPath = "config";
+		private const string LatestGithubReleaseURL = "https://api.github.com/repos/JustArchi/ArchiSteamFarm/releases/latest";
+
 		private static readonly HashSet<Bot> Bots = new HashSet<Bot>();
 		internal static readonly object ConsoleLock = new object();
+		internal static string Version { get { return Assembly.GetExecutingAssembly().GetName().Version.ToString(); } }
+
+		private static async Task CheckForUpdate() {
+			JObject response = await Utilities.UrlToJObject(LatestGithubReleaseURL).ConfigureAwait(false);
+			if (response == null) {
+				return;
+			}
+
+			string remoteVersion = response["tag_name"].ToString();
+			if (string.IsNullOrEmpty(remoteVersion)) {
+				return;
+			}
+
+			string localVersion = Version;
+
+			if (localVersion.CompareTo(remoteVersion) < 0) {
+				Logging.LogGenericNotice("", "New version is available!");
+				Logging.LogGenericNotice("", "Local version: " + localVersion);
+				Logging.LogGenericNotice("", "Remote version: " + remoteVersion);
+				Logging.LogGenericNotice("", "Consider updating yourself!");
+				Thread.Sleep(5000);
+			}
+		}
 
 		internal static void Exit(int exitCode = 0) {
 			ShutdownAllBots();
@@ -83,6 +111,10 @@ private static void ShutdownAllBots() {
 		}
 
 		private static void Main(string[] args) {
+			Logging.LogGenericInfo("Main", "Archi's Steam Farm, version " + Version);
+
+			Task.Run(async () => await CheckForUpdate().ConfigureAwait(false)).Wait();
+
 			// Config directory may not be in the same directory as the .exe, check maximum of 3 levels lower
 			for (var i = 0; i < 4 && !Directory.Exists(ConfigDirectoryPath); i++) {
 				Directory.SetCurrentDirectory("..");
@@ -97,7 +129,11 @@ private static void Main(string[] args) {
 			lock (Bots) {
 				foreach (var configFile in Directory.EnumerateFiles(ConfigDirectoryPath, "*.xml")) {
 					string botName = Path.GetFileNameWithoutExtension(configFile);
-					Bots.Add(new Bot(botName));
+					Bot bot = new Bot(botName);
+					Bots.Add(bot);
+					if (!bot.Enabled) {
+						Logging.LogGenericInfo(botName, "Not starting this instance because it's disabled in config file");
+					}
 				}
 			}
 

diff --git a/ArchiSteamFarm/Properties/AssemblyInfo.cs b/ArchiSteamFarm/Properties/AssemblyInfo.cs
@@ -32,5 +32,5 @@
 // You can specify all the values or you can default the Build and Revision Numbers 
 // by using the '*' as shown below:
 // [assembly: AssemblyVersion("1.0.*")]
-[assembly: AssemblyVersion("1.0.0.0")]
-[assembly: AssemblyFileVersion("1.0.0.0")]
+[assembly: AssemblyVersion("0.4.0.0")]
+[assembly: AssemblyFileVersion("0.4.0.0")]
diff --git a/ArchiSteamFarm/Utilities.cs b/ArchiSteamFarm/Utilities.cs
@@ -23,6 +23,7 @@ limitations under the License.
 */
 
 using HtmlAgilityPack;
+using Newtonsoft.Json.Linq;
 using System;
 using System.Collections.Generic;
 using System.Globalization;
@@ -62,6 +63,7 @@ internal static async Task<HttpResponseMessage> UrlToHttpResponse(string website
 				using (HttpClientHandler clientHandler = new HttpClientHandler { UseCookies = false }) {
 					using (HttpClient client = new HttpClient(clientHandler)) {
 						client.Timeout = TimeSpan.FromSeconds(10);
+						client.DefaultRequestHeaders.UserAgent.ParseAdd("ArchiSteamFarm/1.0");
 						HttpRequestMessage requestMessage = new HttpRequestMessage(HttpMethod.Get, websiteAddress);
 						if (cookieVariables != null) {
 							StringBuilder cookie = new StringBuilder();
@@ -122,7 +124,8 @@ internal static async Task<bool> UrlPostRequest(string request, Dictionary<strin
 			try {
 				using (HttpClientHandler clientHandler = new HttpClientHandler { UseCookies = false }) {
 					using (HttpClient client = new HttpClient(clientHandler)) {
-						client.Timeout = TimeSpan.FromSeconds(15);
+						client.Timeout = TimeSpan.FromSeconds(10);
+						client.DefaultRequestHeaders.UserAgent.ParseAdd("ArchiSteamFarm/1.0");
 						HttpRequestMessage requestMessage = new HttpRequestMessage(HttpMethod.Post, request);
 						requestMessage.Content = new FormUrlEncodedContent(postData);
 						if (cookieVariables != null && cookieVariables.Count > 0) {
@@ -159,6 +162,7 @@ internal static async Task<HttpResponseMessage> UrlPostRequestWithResponse(strin
 				using (HttpClientHandler clientHandler = new HttpClientHandler { UseCookies = false }) {
 					using (HttpClient client = new HttpClient(clientHandler)) {
 						client.Timeout = TimeSpan.FromSeconds(10);
+						client.DefaultRequestHeaders.UserAgent.ParseAdd("ArchiSteamFarm/1.0");
 						HttpRequestMessage requestMessage = new HttpRequestMessage(HttpMethod.Post, request);
 						requestMessage.Content = new FormUrlEncodedContent(postData);
 						if (cookieVariables != null && cookieVariables.Count > 0) {
@@ -183,5 +187,29 @@ internal static async Task<HttpResponseMessage> UrlPostRequestWithResponse(strin
 
 			return result;
 		}
+
+		internal static async Task<JObject> UrlToJObject(string request) {
+			if (string.IsNullOrEmpty(request)) {
+				return null;
+			}
+
+			HttpResponseMessage httpResponseMessage = await UrlToHttpResponse(request, null).ConfigureAwait(false);
+			if (httpResponseMessage == null) {
+				return null;
+			}
+
+			string source = await httpResponseMessage.Content.ReadAsStringAsync().ConfigureAwait(false);
+			if (string.IsNullOrEmpty(source)) {
+				return null;
+			}
+
+			JObject result = null;
+			try {
+				result = JObject.Parse(source);
+			} catch {
+			}
+
+			return result;
+		}
 	}
 }
diff --git a/ArchiSteamFarm/packages.config b/ArchiSteamFarm/packages.config
@@ -1,6 +1,7 @@
 <?xml version="1.0" encoding="utf-8"?>
 <packages>
-	<package id="HtmlAgilityPack" version="1.4.9" targetFramework="net45" />
-	<package id="protobuf-net" version="2.0.0.668" targetFramework="net45" />
-	<package id="SteamKit2" version="1.6.5" targetFramework="net45" />
+  <package id="HtmlAgilityPack" version="1.4.9" targetFramework="net45" />
+  <package id="Newtonsoft.Json" version="8.0.1-beta1" targetFramework="net45" />
+  <package id="protobuf-net" version="2.0.0.668" targetFramework="net45" />
+  <package id="SteamKit2" version="1.6.5" targetFramework="net45" />
 </packages>
diff --git a/packages/Newtonsoft.Json.8.0.1-beta1/Newtonsoft.Json.8.0.1-beta1.nupkg b/packages/Newtonsoft.Json.8.0.1-beta1/Newtonsoft.Json.8.0.1-beta1.nupkg
diff --git a/packages/Newtonsoft.Json.8.0.1-beta1/lib/net20/Newtonsoft.Json.dll b/packages/Newtonsoft.Json.8.0.1-beta1/lib/net20/Newtonsoft.Json.dll