From 50b503a3b52692bb9debdee42150cd2bffcf581f Mon Sep 17 00:00:00 2001
From: Jeremy Tellier - ROG <JeremyTellier@gmail.com>
Date: Thu, 11 Jun 2020 20:32:36 +0100
Subject: [PATCH] Configure location of versioning

---
 ZitiUpdateService/App.config               | 10 +++++++---
 ZitiUpdateService/UpdateService.cs         |  7 ++++++-
 ZitiUpdateService/ZitiUpdateService.csproj |  1 +
 3 files changed, 14 insertions(+), 4 deletions(-)

diff --git a/ZitiUpdateService/App.config b/ZitiUpdateService/App.config
index 193aecc6..18ca8d20 100644
--- a/ZitiUpdateService/App.config
+++ b/ZitiUpdateService/App.config
@@ -1,6 +1,10 @@
 <?xml version="1.0" encoding="utf-8" ?>
 <configuration>
-    <startup> 
-        <supportedRuntime version="v4.0" sku=".NETFramework,Version=v4.8" />
-    </startup>
+  <appSettings>
+    <add key="UpdateUrl" value="https://netfoundry.jfrog.io/netfoundry/ziti-maven-snapshot/ziti-tunnel-win/amd64/windows/ziti-tunnel-win/maven-metadata.xml"/>
+    <add key="Version" value="release"/>
+  </appSettings>
+  <startup> 
+    <supportedRuntime version="v4.0" sku=".NETFramework,Version=v4.8" />
+  </startup>
 </configuration>
\ No newline at end of file
diff --git a/ZitiUpdateService/UpdateService.cs b/ZitiUpdateService/UpdateService.cs
index fa622d09..73746d05 100644
--- a/ZitiUpdateService/UpdateService.cs
+++ b/ZitiUpdateService/UpdateService.cs
@@ -13,6 +13,8 @@ using System.Net;
 using System.Timers;
 using System.Management.Automation;
 using System.Xml;
+using System.Configuration;
+using System.Collections.Specialized;
 
 namespace ZitiUpdateService {
 	public partial class UpdateService : ServiceBase {
@@ -27,6 +29,7 @@ namespace ZitiUpdateService {
 		private string _rootDirectory = "";
 		private string _logDirectory = "";
 		private bool _isJustStopped = true;
+		private string _versionType = "release";
 
 		ServiceController controller;
 		public UpdateService() {
@@ -34,6 +37,8 @@ namespace ZitiUpdateService {
 		}
 
 		protected override void OnStart(string[] args) {
+			if (ConfigurationManager.AppSettings.Get("UpdateUrl")!=null) _versionUrl = ConfigurationManager.AppSettings.Get("UpdateUrl");
+			if (ConfigurationManager.AppSettings.Get("Version") !=null) _versionType = ConfigurationManager.AppSettings.Get("Version");
 			_rootDirectory = Path.Combine(Environment.GetFolderPath(Environment.SpecialFolder.CommonApplicationData), "NetFoundry");
 			if (!Directory.Exists(_rootDirectory)) Directory.CreateDirectory(_rootDirectory);
 			_logDirectory = Path.Combine(_rootDirectory, "Logs");
@@ -90,7 +95,7 @@ namespace ZitiUpdateService {
 			var result = readStream.ReadToEnd();
 			XmlDocument xmlDoc = new XmlDocument();
 			xmlDoc.LoadXml(result);
-			XmlNode node = xmlDoc.SelectSingleNode("metadata/versioning/release");
+			XmlNode node = xmlDoc.SelectSingleNode("metadata/versioning/"+ _versionType);
 			string version = node.InnerText;
 			Log("Version Checked: " + version+" on "+_version);
 			if (version != _version) {
diff --git a/ZitiUpdateService/ZitiUpdateService.csproj b/ZitiUpdateService/ZitiUpdateService.csproj
index 2bc88441..6b0fedfc 100644
--- a/ZitiUpdateService/ZitiUpdateService.csproj
+++ b/ZitiUpdateService/ZitiUpdateService.csproj
@@ -75,6 +75,7 @@
   </PropertyGroup>
   <ItemGroup>
     <Reference Include="System" />
+    <Reference Include="System.Configuration" />
     <Reference Include="System.Configuration.Install" />
     <Reference Include="System.Core" />
     <Reference Include="System.IO.Compression" />