Remotely/Server/Pages/ServerConfig.razor.cs

using Microsoft.AspNetCore.Components;
using Microsoft.AspNetCore.Components.Forms;
using Microsoft.AspNetCore.Components.Web;
using Microsoft.AspNetCore.Hosting;
using Microsoft.AspNetCore.SignalR;
using Microsoft.Extensions.Configuration;
using Microsoft.Extensions.Hosting;
using Remotely.Server.Components;
using Remotely.Server.Hubs;
using Remotely.Server.Services;
using Remotely.Shared.Enums;
using Remotely.Shared.Models;
using System;
using System.Collections.Generic;
using System.ComponentModel.DataAnnotations;
using System.Linq;
using System.Text.Json;
using System.Text.Json.Serialization;
using System.Threading;
using System.Threading.Tasks;

namespace Remotely.Server.Pages
{
    public class AppSettingsModel
    {
        [Display(Name = "Allow API Login")]
        public bool AllowApiLogin { get; set; }

        [Display(Name = "Banned Devices")]
        public List<string> BannedDevices { get; set; } = new();

        [Display(Name = "Data Retention (days)")]
        public double DataRetentionInDays { get; set; }

        [Display(Name = "Database Provider")]
        [JsonConverter(typeof(JsonStringEnumConverter))]
        public DbProvider DBProvider { get; set; }


        [Display(Name = "Enable Windows Event Log")]
        public bool EnableWindowsEventLog { get; set; }

        [Display(Name = "Enforce Attended Access")]
        public bool EnforceAttendedAccess { get; set; }

        [Display(Name = "Ice Servers")]
        public IceServerModel[] IceServers { get; set; }

        [Display(Name = "Known Proxies")]
        public List<string> KnownProxies { get; set; } = new();

        [Display(Name = "Max Concurrent Updates")]
        public int MaxConcurrentUpdates { get; set; }

        [Display(Name = "Max Organizations")]
        public int MaxOrganizationCount { get; set; }
        [Display(Name = "Message of the Day")]
        public string MessageOfTheDay { get; set; }

        [Display(Name = "Redirect To HTTPS")]
        public bool RedirectToHttps { get; set; }

        [Display(Name = "Remote Control Notify User")]
        public bool RemoteControlNotifyUser { get; set; }

        [Display(Name = "Remote Control Requires Authentication")]
        public bool RemoteControlRequiresAuthentication { get; set; }

        [Display(Name = "Remote Control Session Limit")]
        public double RemoteControlSessionLimit { get; set; }

        [Display(Name = "Require 2FA")]
        public bool Require2FA { get; set; }

        [Display(Name = "SMTP Display Name")]
        public string SmtpDisplayName { get; set; }

        [Display(Name = "SMTP Email")]
        [EmailAddress]
        public string SmtpEmail { get; set; }

        [Display(Name = "SMTP Host")]
        public string SmtpHost { get; set; }
        [Display(Name = "SMTP Local Domain")]
        public string SmtpLocalDomain { get; set; }

        [Display(Name = "SMTP Check Certificate Revocation")]
        public bool SmtpCheckCertificateRevocation { get; set; }

        [Display(Name = "SMTP Password")]
        public string SmtpPassword { get; set; }

        [Display(Name = "SMTP Port")]
        public int SmtpPort { get; set; }

        [Display(Name = "SMTP Username")]
        public string SmtpUserName { get; set; }

        [Display(Name = "Theme")]
        [JsonConverter(typeof(JsonStringEnumConverter))]
        public Theme Theme { get; set; }

        [Display(Name = "Trusted CORS Origins")]
        public List<string> TrustedCorsOrigins { get; set; } = new();

        [Display(Name = "Use HSTS")]
        public bool UseHsts { get; set; }

        [Display(Name = "Use WebRTC")]
        public bool UseWebRtc { get; set; }
    }

    public class ConnectionStringsModel
    {
        [Display(Name = "PostgreSQL")]
        public string PostgreSQL { get; set; }

        [Display(Name = "SQLite")]
        public string SQLite { get; set; }

        [Display(Name = "SQL Server")]
        public string SQLServer { get; set; }
    }

    public partial class ServerConfig : AuthComponentBase
    {
        private string _alertMessage;
        private string _bannedDeviceSelected;
        private string _bannedDeviceToAdd;

        private string _knownProxySelected;
        private string _knownProxyToAdd;

        private bool _showMyOrgAdminsOnly = true;
        private bool _showAdminsOnly;

        private string _trustedCorsOriginSelected;
        private string _trustedCorsOriginToAdd;

        private readonly List<RemotelyUser> _userList = new();


        [Inject]
        private IHubContext<AgentHub> AgentHubContext { get; set; }

        [Inject]
        private IConfiguration Configuration { get; set; }

        private ConnectionStringsModel ConnectionStrings { get; } = new();

        [Inject]
        private IDataService DataService { get; set; }

        [Inject]
        private IEmailSenderEx EmailSender { get; set; }

        [Inject]
        private IWebHostEnvironment HostEnv { get; set; }

        private AppSettingsModel Input { get; } = new();

        [Inject]
        private IModalService ModalService { get; set; }

        [Inject]
        private ICircuitManager CircuitManager { get; set; }

        private IEnumerable<string> OutdatedDevices => GetOutdatedDevices();

        [Inject]
        private IToastService ToastService { get; set; }
        private int TotalDevices => DataService.GetTotalDevices();
        private IEnumerable<RemotelyUser> UserList
        {
            get
            {
                return _userList.Where(x =>
                    (!_showAdminsOnly || x.IsServerAdmin) &&
                    (!_showMyOrgAdminsOnly || x.OrganizationID == User.OrganizationID));
            }
        }

        protected override async Task OnInitializedAsync()
        {
            await base.OnInitializedAsync();

            if (!User.IsServerAdmin)
            {
                return;
            }

            Configuration.Bind("ApplicationOptions", Input);
            Configuration.Bind("ConnectionStrings", ConnectionStrings);
            _userList.AddRange(DataService.GetAllUsersForServer().OrderBy(x => x.UserName));
        }

        private void AddBannedDevice()
        {
            if (string.IsNullOrWhiteSpace(_bannedDeviceToAdd))
            {
                return;
            }

            Input.BannedDevices.Add(_bannedDeviceToAdd);
            _bannedDeviceToAdd = string.Empty;
        }

        private void AddKnownProxy()
        {
            if (string.IsNullOrWhiteSpace(_knownProxyToAdd))
            {
                return;
            }

            Input.KnownProxies.Add(_knownProxyToAdd);
            _knownProxyToAdd = string.Empty;
        }

        private void AddTrustedCorsOrigin()
        {
            if (string.IsNullOrWhiteSpace(_trustedCorsOriginToAdd))
            {
                return;
            }

            Input.TrustedCorsOrigins.Add(_trustedCorsOriginToAdd);
            _trustedCorsOriginToAdd = null;
        }

        private IEnumerable<string> GetOutdatedDevices()
        {
            var highestVersion = AgentHub.ServiceConnections.Values.Max(x =>
                Version.TryParse(x.AgentVersion, out var result) ? result : default);

            return AgentHub.ServiceConnections.Values
                .Where(x => Version.TryParse(x.AgentVersion, out var result) && result != highestVersion)
                .Select(x => x.ID);
        }

        private void HandleBannedDeviceKeyDown(KeyboardEventArgs args)
        {
            if (args.Key.Equals("Enter", StringComparison.OrdinalIgnoreCase))
            {
                AddBannedDevice();
            }
        }

        private void HandleKnownProxyKeyDown(KeyboardEventArgs args)
        {
            if (args.Key.Equals("Enter", StringComparison.OrdinalIgnoreCase))
            {
                AddKnownProxy();
            }
        }

        private void HandleTrustedCorsKeyDown(KeyboardEventArgs args)
        {
            if (args.Key.Equals("Enter", StringComparison.OrdinalIgnoreCase))
            {
                AddTrustedCorsOrigin();
            }
        }
        private async Task HandleValidSubmit(EditContext context)
        {
            await Save();
        }

        private void RemoveBannedDevice()
        {
            if (string.IsNullOrWhiteSpace(_bannedDeviceSelected))
            {
                return;
            }

            Input.BannedDevices.Remove(_bannedDeviceSelected);
            _bannedDeviceSelected = null;
        }

        private void RemoveKnownProxy()
        {
            if (string.IsNullOrWhiteSpace(_knownProxySelected))
            {
                return;
            }
            Input.KnownProxies.Remove(_knownProxySelected);
            _knownProxySelected = null;
        }

        private void RemoveTrustedCorsOrigin()
        {
            if (string.IsNullOrWhiteSpace(_trustedCorsOriginSelected))
            {
                return;
            }

            Input.TrustedCorsOrigins.Remove(_trustedCorsOriginSelected);
            _trustedCorsOriginSelected = null;
        }

        private async Task Save()
        {
            var resetEvent = new ManualResetEventSlim();

            Configuration.GetReloadToken().RegisterChangeCallback((e) =>
            {
                resetEvent.Set();
            }, null);

            await SaveInputToAppSettings();

            resetEvent.Wait(5_000);

            ToastService.ShowToast("Configuration saved.");
            _alertMessage = "Configuration saved.";
        }

        private async Task SaveAndTestSmtpSettings()
        {
            await SaveInputToAppSettings();

            var success = await EmailSender.SendEmailAsync(User.Email, "Remotely Test Email", "Congratulations! Your SMTP settings are working!", User.OrganizationID);
            if (success)
            {
                ToastService.ShowToast($"Test email sent to {User.Email}.  Check your inbox (or spam folder).");
                _alertMessage = $"Test email sent to {User.Email}.  Check your inbox (or spam folder).";
            }
            else
            {
                ToastService.ShowToast("Error sending email.  Check the server logs for details.", classString: "bg-danger");
                _alertMessage = "Error sending email.  Check the server logs for details.";
            }
        }

        private async Task SaveInputToAppSettings()
        {
            string savePath;
            var prodSettings = HostEnv.ContentRootFileProvider.GetFileInfo("appsettings.Production.json");
            var stagingSettings = HostEnv.ContentRootFileProvider.GetFileInfo("appsettings.Staging.json");
            var settings = HostEnv.ContentRootFileProvider.GetFileInfo("appsettings.json");

            if (HostEnv.IsProduction() && prodSettings.Exists)
            {
                savePath = prodSettings.PhysicalPath;
            }
            else if (HostEnv.IsStaging() && stagingSettings.Exists)
            {
                savePath = stagingSettings.PhysicalPath;
            }
            else if (settings.Exists)
            {
                savePath = settings.PhysicalPath;
            }
            else
            {
                return;
            }

            var settingsJson = JsonSerializer.Deserialize<IDictionary<string, object>>(await System.IO.File.ReadAllTextAsync(savePath));
            Input.IceServers = Configuration.GetSection("ApplicationOptions:IceServers").Get<IceServerModel[]>();
            settingsJson["ApplicationOptions"] = Input;
            settingsJson["ConnectionStrings"] = ConnectionStrings;

            await System.IO.File.WriteAllTextAsync(savePath, JsonSerializer.Serialize(settingsJson, new JsonSerializerOptions() { WriteIndented = true }));
        }
        private void SetIsServerAdmin(ChangeEventArgs ev, RemotelyUser user)
        {
            var isAdmin = (bool)ev.Value;
            DataService.SetIsServerAdmin(user.Id, isAdmin, User.Id);
            ToastService.ShowToast("Server admins updated.");
        }

        private void ShowOutdatedDevices()
        {
            if (OutdatedDevices.Any())
            {
                var outdatedDeviceNames = DataService
                    .GetDevices(OutdatedDevices)
                    .Select(x => x.DeviceName);

                ModalService.ShowModal("Outdated Devices",
                    (new[] { "Outdated Devices:" }).Concat(outdatedDeviceNames).ToArray());
            }
            else
            {
                ModalService.ShowModal("Outdated Devices", new[] { "There are no outdated devices currently online." });
            }
        }

        private async Task UpdateAllDevices()
        {
            if (!User.IsServerAdmin)
            {
                return;
            }

            var outdatedDevices = OutdatedDevices;

            if (!outdatedDevices.Any())
            {
                ToastService.ShowToast("No agents need updating.");
                return;
            }

            var agentConnections = AgentHub.ServiceConnections.Where(x => outdatedDevices.Contains(x.Value.ID));

            await AgentHubContext.Clients.Clients(agentConnections.Select(x => x.Key)).SendAsync("ReinstallAgent");
            ToastService.ShowToast("Update command sent.");
        }
    }
}