prs/prs.desktop/Utils/SelectDatabase.xaml.cs

using System;
using System.Collections.Generic;
using System.Drawing;
using System.IO;
using System.Linq;
using System.Threading;
using System.Threading.Tasks;
using System.Windows;
using System.Windows.Controls;
using System.Windows.Media;
using Comal.Classes;
using InABox.IPC;
using InABox.Clients;
using InABox.Configuration;
using InABox.Core;
using InABox.Database;
using InABox.IPC;
using InABox.Rpc;
using InABox.Wpf;
using InABox.WPF;
using PRSServer;
using Image = System.Windows.Controls.Image;
using Size = System.Drawing.Size;

namespace PRSDesktop
{
    /// <summary>
    ///     Interaction logic for SelectDatabase.xaml
    /// </summary>
    public partial class SelectDatabase : ThemableWindow
    {

        private CancellationTokenSource cancel = new CancellationTokenSource();
        
        public SelectDatabase(Dictionary<string, DatabaseSettings> databases)
        {
            InitializeComponent();

            var active = databases.Where(x => x.Value.IsActive).ToArray().Length;

            var cols = (int)Math.Ceiling(Math.Sqrt(active));
            for (var iCol = 0; iCol < cols; iCol++)
                Grid.ColumnDefinitions.Add(new ColumnDefinition { Width = new GridLength(1, GridUnitType.Auto) });
            var rows = (int)Math.Ceiling(active / (double)cols);
            for (var iRow = 0; iRow < cols; iRow++)
                Grid.RowDefinitions.Add(new RowDefinition { Height = new GridLength(1, GridUnitType.Auto) });

            var i = 0;
            foreach (var key in databases.Keys)
                if (databases[key].IsActive)
                {
                    var db = databases[key];
                    
                    var button = new Button
                    {
                        Background = new SolidColorBrush(Colors.Transparent),
                        BorderThickness = new Thickness(0),
                        Padding = new Thickness(0)
                    };

                    var color = Colors.Red;
                    // try
                    // {
                    //     color = (System.Windows.Media.Color)System.Windows.Media.ColorConverter.ConvertFromString(db.ColorScheme);
                    // }
                    // catch (Exception e)
                    // {
                    //     color = Colors.Red;
                    // }
                    
                    var border = new Border
                    {
                        CornerRadius = new CornerRadius(5),
                        BorderBrush = new SolidColorBrush(color),
                        BorderThickness = new Thickness(0.75),
                        Background = db.DatabaseType == DatabaseType.Standalone ? 
                            new SolidColorBrush(Colors.WhiteSmoke) : 
                            new SolidColorBrush(Colors.White),
                        Width = 200,
                        Height = 200
                    };

                    var panel = new DockPanel { VerticalAlignment = VerticalAlignment.Stretch };
                    border.Child = panel;

                    StackPanel stack = new StackPanel() { Orientation = Orientation.Horizontal, HorizontalAlignment = HorizontalAlignment.Center };
                    stack.SetValue(DockPanel.DockProperty, Dock.Top);
                    Label dbtype = new Label()
                    {
                        Content = db.DatabaseType.ToString(), 
                        Margin = new Thickness(5,2,0,0), 
                        HorizontalContentAlignment = HorizontalAlignment.Left,
                        FontSize = 8,
                        Foreground = new SolidColorBrush(Colors.DimGray)
                    };
                    stack.Children.Add(dbtype);                    
                    Label dbver = new Label()
                    {
                        Margin = new Thickness(0,2,5,0), 
                        HorizontalContentAlignment = HorizontalAlignment.Right,
                        FontSize = 8,
                        Foreground = new SolidColorBrush(Colors.DimGray)
                    };
                    stack.Children.Add(dbver);
                    panel.Children.Add(stack);
                    
                    var label = new Label
                    {
                        Content = key,
                        HorizontalContentAlignment = HorizontalAlignment.Center,
                        VerticalContentAlignment = VerticalAlignment.Center,
                        Width = 150,
                        Height = 35
                    };
                    label.SetValue(DockPanel.DockProperty, Dock.Bottom);
                    panel.Children.Add(label);
                    
                    var image = new Image();
                    image.Height = 150;
                    image.Width = 150;
                    image.SetValue(DockPanel.DockProperty, Dock.Top);
                    panel.Children.Add(image);

                    button.Content = border;
                    button.Margin = new Thickness(5, 5, 0, 0);
                    button.Click += Button_Click;
                    button.Height = 200;
                    button.Width = 200;
                    button.Tag = key;
                    var col = i % cols;
                    var row = i / cols;
                    button.SetValue(Grid.ColumnProperty, col);
                    button.SetValue(Grid.RowProperty, row);
                    Grid.Children.Add(button);

                    Task.Run(
                        () =>
                        {
                            while (!cancel.IsCancellationRequested)
                            {
                                DatabaseInfo? info = null;

                                if(db.DatabaseType == DatabaseType.Standalone)
                                {
                                    info = new DatabaseInfo(db.ColorScheme, db.Logo, CoreUtils.GetVersion(), true, DbFactory.RestPort, DbFactory.RPCPort, DbFactory.ID);
                                }
                                if (db.DatabaseType == DatabaseType.Local)
                                {
                                    info = new IPCClient<User>(DatabaseServerProperties.GetPipeName(db.LocalServerName, false)).Info();
                                    //info = new RpcClientPipeTransport(DatabaseServerProperties.GetPipeName(db.LocalServerName, true)).Info();
                                }
                                else if (db.DatabaseType == DatabaseType.Networked)
                                {
                                    if (db.Protocol == SerializerProtocol.RPC)
                                        info = new RpcClientSocketTransport(db.URLs).Info();
                                    else
                                    {
                                        RestClient<User>.Ping(db.URLs, out DatabaseInfo i);
                                        info = i;
                                    }
                                }

                                UpdateInfo(key, db, info, border, image, dbver);
                                Task.Delay(1000).Wait();
                            }
                        }
                        ,cancel.Token);

                    i++;
                }
        }

        private bool LogoEqual(byte[]? first, byte[]? second)
        {
            if ((first == null) && (second == null))
                return true;
            if ((first == null) || (second == null))
                return false;
            return first.SequenceEqual(second);
        }

        private void UpdateInfo(String key, DatabaseSettings db, DatabaseInfo? info, Border border, Image image, Label version)
        {

            
            Dispatcher.Invoke(() =>
            {
                if (info == null)
                {
                    border.BorderBrush = new SolidColorBrush(Colors.Red);
                    border.Background = db.DatabaseType == DatabaseType.Standalone
                        ? new SolidColorBrush(Colors.WhiteSmoke)
                        : new SolidColorBrush(Colors.White);
                    image.Source = null;
                    version.Content = "";
                    return;
                }

                if (info?.ColorScheme is not null)
                {
                    try
                    {
                        var color = (System.Windows.Media.Color)System.Windows.Media.ColorConverter.ConvertFromString(info.ColorScheme);
                        border.BorderBrush = new SolidColorBrush(color);
                        border.Background = new SolidColorBrush(color.AdjustLightness(90));
                    }
                    catch
                    {
                    }
                }

                Bitmap? bitmap = null; //PRSDesktop.Resources.splash_small;
                try
                {
                    if (info?.Logo != null && info.Logo.Any())
                        using (var ms = new MemoryStream(info.Logo))
                            bitmap = new Bitmap(ms);
                }
                catch
                {
                }

                if (bitmap != null)
                {
                    var size = new Size(bitmap.Width, bitmap.Height).Adjust(150, 150);
                    image.Source = bitmap.AsBitmapImage(size.Height, size.Width);
                }

                try
                {
                    version.Content = String.Equals(info?.Version,"???") ? "" : info?.Version;
                }
                catch
                {
                }

                if (!string.Equals(db.ColorScheme, info?.ColorScheme) || !LogoEqual(db.Logo, info?.Logo))
                {
                    db.ColorScheme = info?.ColorScheme ?? DatabaseSettings.DefaultColorScheme;
                    db.Logo = info?.Logo;
                    new LocalConfiguration<DatabaseSettings>(key).Save(db);
                }
            });
        }

        public string Database { get; private set; }

        private void Button_Click(object sender, RoutedEventArgs e)
        {
            cancel.Cancel();
            Database = (sender as Button).Tag as string;
            DialogResult = true;
        }

    }
}