From 9e004f5b3c5c7945a09b2c70c628d7767d2bb79c Mon Sep 17 00:00:00 2001
From: Shaun Walker <shaun.walker@siliqon.com>
Date: Sun, 27 Jun 2021 08:48:18 -0400
Subject: [PATCH] add ability to register for updates

---
 Oqtane.Client/Installer/Installer.razor       | 11 ++++-
 .../Modules/Admin/SystemInfo/Index.razor      | 22 ++++++++++
 .../Resources/Installer/Installer.resx        |  7 ++-
 .../Modules/Admin/SystemInfo/Index.resx       | 14 ++++--
 Oqtane.Client/Services/InstallationService.cs |  5 +++
 .../Interfaces/IInstallationService.cs        |  1 +
 .../Controllers/InstallationController.cs     | 43 ++++++++++++++++---
 .../Controllers/PackageController.cs          | 15 +------
 Oqtane.Server/Controllers/SystemController.cs | 13 +-----
 Oqtane.Server/Infrastructure/ConfigManager.cs | 27 ++++++++++++
 .../Interfaces/IConfigManager.cs              |  5 +++
 .../EntityBuilders/BaseEntityBuilder.cs       | 11 +++++
 Oqtane.Shared/Shared/InstallConfig.cs         |  1 +
 13 files changed, 136 insertions(+), 39 deletions(-)

diff --git a/Oqtane.Client/Installer/Installer.razor b/Oqtane.Client/Installer/Installer.razor
index eb0c96b1..a3812b22 100644
--- a/Oqtane.Client/Installer/Installer.razor
+++ b/Oqtane.Client/Installer/Installer.razor
@@ -101,6 +101,11 @@
         </div>
         <div class="app-progress-indicator" style="@_loadingDisplay"></div>
     </div>
+    <div class="row">
+        <div class="mx-auto text-center">
+            <input type="checkbox" @bind="@_register" /> @Localizer["Register"]
+        </div>
+    </div>
 </div>
 
 @code {
@@ -114,6 +119,7 @@
     private string _hostPassword = string.Empty;
     private string _confirmPassword = string.Empty;
     private string _hostEmail = string.Empty;
+    private bool _register = true;
     private string _message = string.Empty;
     private string _loadingDisplay = "display: none;";
 
@@ -169,7 +175,7 @@
             connectionString = databaseConfigControl.GetConnectionString();
         }
 
-        if (connectionString != "" && _hostUsername != "" && _hostPassword.Length >= 6 && _hostPassword == _confirmPassword && _hostEmail != "")
+        if (connectionString != "" && _hostUsername != "" && _hostPassword.Length >= 6 && _hostPassword == _confirmPassword && _hostEmail != "" && _hostEmail.Contains("@"))
         {
             _loadingDisplay = "";
             StateHasChanged();
@@ -188,7 +194,8 @@
                 HostName = UserNames.Host,
                 TenantName = TenantNames.Master,
                 IsNewTenant = true,
-                SiteName = Constants.DefaultSite
+                SiteName = Constants.DefaultSite,
+                Register = _register
             };
 
             var installation = await InstallationService.Install(config);
diff --git a/Oqtane.Client/Modules/Admin/SystemInfo/Index.razor b/Oqtane.Client/Modules/Admin/SystemInfo/Index.razor
index 173d9e2b..333f47d7 100644
--- a/Oqtane.Client/Modules/Admin/SystemInfo/Index.razor
+++ b/Oqtane.Client/Modules/Admin/SystemInfo/Index.razor
@@ -56,6 +56,12 @@
                 </td>
             </tr>
         </table>
+        <div class="row">
+            <div class="mx-auto text-center">
+                <input type="checkbox" @onchange="(e => RegisterChecked(e))" /> @Localizer["Register"]
+            </div>
+        </div>
+        <br /><br />
         <a class="btn btn-primary" href="swagger/index.html" target="_new">@Localizer["Access.ApiFramework"]</a>&nbsp;
         <ActionDialog Header="Restart Application" Message="Are You Sure You Wish To Restart The Application?" Action="Restart Application" Security="SecurityAccessLevel.Host" Class="btn btn-danger" OnClick="@(async () => await RestartApplication())" ResourceKey="RestartApplication" />
     </TabPanel>
@@ -183,4 +189,20 @@
             await logger.LogError(ex, "Error Restarting Application");
         }
     }
+
+    private async Task RegisterChecked(ChangeEventArgs e)
+    {
+        try
+        {
+            if ((bool)e.Value)
+            {
+                await InstallationService.RegisterAsync(PageState.User.Email);
+                AddModuleMessage(Localizer["Success.Register"], MessageType.Success);
+            }
+        }
+        catch (Exception ex)
+        {
+            await logger.LogError(ex, "Error On Register");
+        }
+    }
 }
\ No newline at end of file
diff --git a/Oqtane.Client/Resources/Installer/Installer.resx b/Oqtane.Client/Resources/Installer/Installer.resx
index f0bb6212..16a4ef42 100644
--- a/Oqtane.Client/Resources/Installer/Installer.resx
+++ b/Oqtane.Client/Resources/Installer/Installer.resx
@@ -1,4 +1,4 @@
-<?xml version="1.0" encoding="utf-8"?>
+<?xml version="1.0" encoding="utf-8"?>
 <root>
   <!-- 
     Microsoft ResX Schema 
@@ -148,6 +148,9 @@
     <value>Error loading Database Configuration Control</value>
   </data>
   <data name="Message.Require.DbInfo" xml:space="preserve">
-    <value>Please Enter All Fields And Ensure Passwords Match And Are Greater Than 5 Characters In Length</value>
+    <value>Please Enter All Required Fields. Ensure Passwords Match And Are Greater Than 5 Characters In Length. Ensure Email Address Provided Is Valid.</value>
+  </data>
+  <data name="Register" xml:space="preserve">
+    <value>Please Register Me For Major Product Updates And Security Bulletins</value>
   </data>
 </root>
\ No newline at end of file
diff --git a/Oqtane.Client/Resources/Modules/Admin/SystemInfo/Index.resx b/Oqtane.Client/Resources/Modules/Admin/SystemInfo/Index.resx
index a94fa5c7..05ff2846 100644
--- a/Oqtane.Client/Resources/Modules/Admin/SystemInfo/Index.resx
+++ b/Oqtane.Client/Resources/Modules/Admin/SystemInfo/Index.resx
@@ -204,16 +204,16 @@
   <data name="LoggingLevel.HelpText" xml:space="preserve">
     <value>The Minimum Logging Level For The Event Log. This Option Can Be Used To Control The Volume Of Items Stored In Your Event Log.</value>
   </data>
-    <data name="Trace" xml:space="preserve">
+  <data name="Trace" xml:space="preserve">
     <value>Trace</value>
   </data>
-    <data name="Debug" xml:space="preserve">
+  <data name="Debug" xml:space="preserve">
     <value>Debug</value>
   </data>
-    <data name="Information" xml:space="preserve">
+  <data name="Information" xml:space="preserve">
     <value>Information</value>
   </data>
-    <data name="Warning" xml:space="preserve">
+  <data name="Warning" xml:space="preserve">
     <value>Warning</value>
   </data>
   <data name="Error" xml:space="preserve">
@@ -228,4 +228,10 @@
   <data name="Options" xml:space="preserve">
     <value>Options</value>
   </data>
+  <data name="Register" xml:space="preserve">
+    <value>Please Register Me For Major Product Updates And Security Bulletins</value>
+  </data>
+  <data name="Success.Register" xml:space="preserve">
+    <value>You Have Been Successfully Registered For Updates</value>
+  </data>
 </root>
\ No newline at end of file
diff --git a/Oqtane.Client/Services/InstallationService.cs b/Oqtane.Client/Services/InstallationService.cs
index 70bfc7f6..3fd73dbd 100644
--- a/Oqtane.Client/Services/InstallationService.cs
+++ b/Oqtane.Client/Services/InstallationService.cs
@@ -45,5 +45,10 @@ namespace Oqtane.Services
         {
             await PostAsync($"{ApiUrl}/restart");
         }
+
+        public async Task RegisterAsync(string email)
+        {
+            await PostAsync($"{ApiUrl}/register?email={WebUtility.UrlEncode(email)}");
+        }
     }
 }
diff --git a/Oqtane.Client/Services/Interfaces/IInstallationService.cs b/Oqtane.Client/Services/Interfaces/IInstallationService.cs
index da9222e9..e9eb961b 100644
--- a/Oqtane.Client/Services/Interfaces/IInstallationService.cs
+++ b/Oqtane.Client/Services/Interfaces/IInstallationService.cs
@@ -10,5 +10,6 @@ namespace Oqtane.Services
         Task<Installation> Install(InstallConfig config);
         Task<Installation> Upgrade();
         Task RestartAsync();
+        Task RegisterAsync(string email);
     }
 }
diff --git a/Oqtane.Server/Controllers/InstallationController.cs b/Oqtane.Server/Controllers/InstallationController.cs
index dc51076f..09032b46 100644
--- a/Oqtane.Server/Controllers/InstallationController.cs
+++ b/Oqtane.Server/Controllers/InstallationController.cs
@@ -5,7 +5,6 @@ using System.Linq;
 using System.IO.Compression;
 using Microsoft.AspNetCore.Authorization;
 using Microsoft.AspNetCore.Mvc;
-using Microsoft.Extensions.Configuration;
 using Oqtane.Infrastructure;
 using Oqtane.Models;
 using Oqtane.Modules;
@@ -16,13 +15,15 @@ using System.Net;
 using Oqtane.Repository;
 using Microsoft.AspNetCore.Http;
 using System.Diagnostics;
+using System.Net.Http;
+using System.Net.Http.Headers;
 
 namespace Oqtane.Controllers
 {
     [Route(ControllerRoutes.ApiRoute)]
     public class InstallationController : Controller
     {
-        private readonly IConfigurationRoot _config;
+        private readonly IConfigManager _configManager;
         private readonly IInstallationManager _installationManager;
         private readonly IDatabaseManager _databaseManager;
         private readonly ILocalizationManager _localizationManager;
@@ -30,9 +31,9 @@ namespace Oqtane.Controllers
         private readonly IHttpContextAccessor _accessor;
         private readonly IAliasRepository _aliases;
 
-        public InstallationController(IConfigurationRoot config, IInstallationManager installationManager, IDatabaseManager databaseManager, ILocalizationManager localizationManager, IMemoryCache cache, IHttpContextAccessor accessor, IAliasRepository aliases)
+        public InstallationController(IConfigManager configManager, IInstallationManager installationManager, IDatabaseManager databaseManager, ILocalizationManager localizationManager, IMemoryCache cache, IHttpContextAccessor accessor, IAliasRepository aliases)
         {
-            _config = config;
+            _configManager = configManager;
             _installationManager = installationManager;
             _databaseManager = databaseManager;
             _localizationManager = localizationManager;
@@ -47,9 +48,14 @@ namespace Oqtane.Controllers
         {
             var installation = new Installation {Success = false, Message = ""};
 
-            if (ModelState.IsValid && (User.IsInRole(RoleNames.Host) || string.IsNullOrEmpty(_config.GetConnectionString(SettingKeys.ConnectionStringKey))))
+            if (ModelState.IsValid && (User.IsInRole(RoleNames.Host) || string.IsNullOrEmpty(_configManager.GetSetting("ConnectionStrings:" + SettingKeys.ConnectionStringKey, ""))))
             {
                 installation = _databaseManager.Install(config);
+
+                if (config.Register)
+                {
+                    RegisterContact(config.HostEmail);
+                }
             }
             else
             {
@@ -93,7 +99,7 @@ namespace Oqtane.Controllers
         [HttpGet("load")]
         public IActionResult Load()
         {
-            if (_config.GetSection("Runtime").Value == "WebAssembly")
+            if (_configManager.GetSection("Runtime").Value == "WebAssembly")
             {
                 return File(GetAssemblies(), System.Net.Mime.MediaTypeNames.Application.Octet, "oqtane.dll");
             }
@@ -199,5 +205,30 @@ namespace Oqtane.Controllers
                 }
             });
         }
+
+        private void RegisterContact(string email)
+        {
+            try
+            {
+                using (var client = new HttpClient())
+                {
+                    client.DefaultRequestHeaders.Add("Referer", HttpContext.Request.Scheme + "://" + HttpContext.Request.Host.Value);
+                    client.DefaultRequestHeaders.UserAgent.Add(new ProductInfoHeaderValue(Constants.PackageId, Constants.Version));
+                    client.GetAsync(Constants.PackageRegistryUrl + $"/api/registry/contact/?id={_configManager.GetInstallationId()}&email={WebUtility.UrlEncode(email)}");
+                }
+            }
+            catch
+            {
+                // error calling registry service
+            }
+        }
+
+        // GET api/<controller>/register?email=x
+        [HttpPost("register")]
+        [Authorize(Roles = RoleNames.Host)]
+        public void Register(string email)
+        {
+            RegisterContact(email);
+        }
     }
 }
diff --git a/Oqtane.Server/Controllers/PackageController.cs b/Oqtane.Server/Controllers/PackageController.cs
index 8cc1c458..c6ae0585 100644
--- a/Oqtane.Server/Controllers/PackageController.cs
+++ b/Oqtane.Server/Controllers/PackageController.cs
@@ -42,7 +42,7 @@ namespace Oqtane.Controllers
             {
                 client.DefaultRequestHeaders.Add("Referer", HttpContext.Request.Scheme + "://" + HttpContext.Request.Host.Value);
                 client.DefaultRequestHeaders.UserAgent.Add(new ProductInfoHeaderValue(Constants.PackageId, Constants.Version));
-                packages = await GetJson<List<Package>>(client, Constants.PackageRegistryUrl + $"/api/registry/packages/?id={GetInstallationId()}&type={type.ToLower()}&version={Constants.Version}&search={search}");
+                packages = await GetJson<List<Package>>(client, Constants.PackageRegistryUrl + $"/api/registry/packages/?id={_configManager.GetInstallationId()}&type={type.ToLower()}&version={Constants.Version}&search={search}");
             }
             return packages;
         }
@@ -57,7 +57,7 @@ namespace Oqtane.Controllers
             {
                 client.DefaultRequestHeaders.Add("Referer", HttpContext.Request.Scheme + "://" + HttpContext.Request.Host.Value);
                 client.DefaultRequestHeaders.UserAgent.Add(new ProductInfoHeaderValue(Constants.PackageId, Constants.Version));
-                package = await GetJson<Package>(client, Constants.PackageRegistryUrl + $"/api/registry/package/?id={GetInstallationId()}&package={packageid}&version={version}");
+                package = await GetJson<Package>(client, Constants.PackageRegistryUrl + $"/api/registry/package/?id={_configManager.GetInstallationId()}&package={packageid}&version={version}");
             }
 
             if (package != null)
@@ -86,17 +86,6 @@ namespace Oqtane.Controllers
             }
         }
 
-        private string GetInstallationId()
-        {
-            var installationid = _configManager.GetSetting("InstallationId", "");
-            if (installationid == "")
-            {
-                installationid = Guid.NewGuid().ToString();
-                _configManager.AddOrUpdateSetting("InstallationId", installationid, true);
-            }
-            return installationid;
-        }
-
         private async Task<T> GetJson<T>(HttpClient httpClient, string url)
         {
             Uri uri = new Uri(url);
diff --git a/Oqtane.Server/Controllers/SystemController.cs b/Oqtane.Server/Controllers/SystemController.cs
index ab0ddc50..90ca3e95 100644
--- a/Oqtane.Server/Controllers/SystemController.cs
+++ b/Oqtane.Server/Controllers/SystemController.cs
@@ -32,7 +32,7 @@ namespace Oqtane.Controllers
             systeminfo.Add("machinename", Environment.MachineName);
             systeminfo.Add("serverpath", _environment.ContentRootPath);
             systeminfo.Add("servertime", DateTime.Now.ToString());
-            systeminfo.Add("installationid", GetInstallationId());
+            systeminfo.Add("installationid", _configManager.GetInstallationId());
 
             systeminfo.Add("runtime", _configManager.GetSetting("Runtime", "Server"));
             systeminfo.Add("rendermode", _configManager.GetSetting("RenderMode", "ServerPrerendered"));
@@ -65,16 +65,5 @@ namespace Oqtane.Controllers
                 }
             }
         }
-
-        private string GetInstallationId()
-        {
-            var installationid = _configManager.GetSetting("InstallationId", "");
-            if (installationid == "")
-            {
-                installationid = Guid.NewGuid().ToString();
-                _configManager.AddOrUpdateSetting("InstallationId", installationid, true);
-            }
-            return installationid;
-        }
     }
 }
diff --git a/Oqtane.Server/Infrastructure/ConfigManager.cs b/Oqtane.Server/Infrastructure/ConfigManager.cs
index cd38a9d1..6b1d2349 100644
--- a/Oqtane.Server/Infrastructure/ConfigManager.cs
+++ b/Oqtane.Server/Infrastructure/ConfigManager.cs
@@ -4,6 +4,7 @@ using System.IO;
 using Microsoft.Extensions.Configuration;
 using Newtonsoft.Json;
 using Newtonsoft.Json.Linq;
+using Oqtane.Shared;
 
 namespace Oqtane.Infrastructure
 {
@@ -115,5 +116,31 @@ namespace Oqtane.Infrastructure
         {
             _config.Reload();
         }
+
+        public string GetConnectionString()
+        {
+            return _config.GetConnectionString(SettingKeys.ConnectionStringKey);
+        }
+
+        public string GetConnectionString(string name)
+        {
+            return _config.GetConnectionString(name);
+        }
+
+        public bool IsInstalled()
+        {
+            return !string.IsNullOrEmpty(GetConnectionString());
+        }
+
+        public string GetInstallationId()
+        {
+            var installationid = GetSetting("InstallationId", "");
+            if (installationid == "")
+            {
+                installationid = Guid.NewGuid().ToString();
+                AddOrUpdateSetting("InstallationId", installationid, true);
+            }
+            return installationid;
+        }
     }
 }
diff --git a/Oqtane.Server/Infrastructure/Interfaces/IConfigManager.cs b/Oqtane.Server/Infrastructure/Interfaces/IConfigManager.cs
index 400c5e6a..5bff86b9 100644
--- a/Oqtane.Server/Infrastructure/Interfaces/IConfigManager.cs
+++ b/Oqtane.Server/Infrastructure/Interfaces/IConfigManager.cs
@@ -12,5 +12,10 @@ namespace Oqtane.Infrastructure
         void RemoveSetting(string key, bool reload);
         void RemoveSetting(string file, string key, bool reload);
         void Reload();
+
+        public string GetConnectionString();
+        public string GetConnectionString(string name);
+        public bool IsInstalled();
+        public string GetInstallationId();
     }
 }
diff --git a/Oqtane.Server/Migrations/EntityBuilders/BaseEntityBuilder.cs b/Oqtane.Server/Migrations/EntityBuilders/BaseEntityBuilder.cs
index f46db6b4..e60aaade 100644
--- a/Oqtane.Server/Migrations/EntityBuilders/BaseEntityBuilder.cs
+++ b/Oqtane.Server/Migrations/EntityBuilders/BaseEntityBuilder.cs
@@ -143,6 +143,17 @@ namespace Oqtane.Migrations.EntityBuilders
                 unique: isUnique);
         }
 
+        public virtual void AddForeignKey(string foreignKeyName, string columnName, string principalTable, string principalColumn, ReferentialAction onDelete)
+        {
+            _migrationBuilder.AddForeignKey(
+                name: RewriteName(foreignKeyName),
+                table: RewriteName(EntityTableName),
+                column: RewriteName(columnName),
+                principalTable: RewriteName(principalTable),
+                principalColumn: RewriteName(principalColumn),
+                onDelete: onDelete );
+        }
+
         /// <summary>
         /// Creates a Migration to add an Index to the Entity (table)
         /// </summary>
diff --git a/Oqtane.Shared/Shared/InstallConfig.cs b/Oqtane.Shared/Shared/InstallConfig.cs
index 358e6465..9a160c3d 100644
--- a/Oqtane.Shared/Shared/InstallConfig.cs
+++ b/Oqtane.Shared/Shared/InstallConfig.cs
@@ -16,5 +16,6 @@ namespace Oqtane.Shared
         public string DefaultTheme { get; set; }
         public string DefaultContainer { get; set; }
         public string DefaultAdminContainer { get; set; }
+        public bool Register { get; set; }
     }
 }