Add more data to publish state (#52)

chkr1011 · Aug 3, 2022 · 0a49875 · 0a49875
1 parent c1a6c81
commit 0a49875
Show file tree

Hide file tree

Showing 12 changed files with 129 additions and 13 deletions.
diff --git a/Source/Controls/ErrorBox/ErrorBox.axaml.cs b/Source/Controls/ErrorBox/ErrorBox.axaml.cs
@@ -35,6 +35,6 @@ void OnButtonCloseClicked(object? sender, RoutedEventArgs e)
 
     void OnButtonCopyClicked(object? sender, RoutedEventArgs e)
     {
-        _ = Application.Current.Clipboard.SetTextAsync(Message);
+        _ = Application.Current?.Clipboard?.SetTextAsync(Message);
     }
 }
diff --git a/Source/Controls/UserProperties/UserPropertiesView.axaml b/Source/Controls/UserProperties/UserPropertiesView.axaml
@@ -18,7 +18,7 @@
                     Orientation="Horizontal">
             <Button Classes="tool_bar_button"
                     ToolTip.Tip="Add user property"
-                    Command="{Binding AddItem}">
+                    Command="{Binding AddEmptyItem}">
                 <PathIcon Data="{StaticResource add_square_regular}" />
             </Button>
             <Button Classes="tool_bar_button"

diff --git a/Source/Controls/UserProperties/UserPropertiesViewModel.cs b/Source/Controls/UserProperties/UserPropertiesViewModel.cs
@@ -19,9 +19,20 @@ public bool IsReadOnly
 
     public ObservableCollection<UserPropertyViewModel> Items { get; } = new();
 
-    public void AddItem()
+    public void AddEmptyItem()
     {
-        Items.Add(new UserPropertyViewModel(this));
+        AddItem(string.Empty, string.Empty);
+    }
+
+    public void AddItem(string name, string value)
+    {
+        var item = new UserPropertyViewModel(this)
+        {
+            Name = name,
+            Value = value
+        };
+
+        Items.Add(item);
     }
 
     public void ClearItems()

diff --git a/Source/Pages/Connection/ConnectionItemViewModel.cs b/Source/Pages/Connection/ConnectionItemViewModel.cs
@@ -13,7 +13,7 @@ public ConnectionItemViewModel(ConnectionPageViewModel ownerPage)
     {
         OwnerPage = ownerPage ?? throw new ArgumentNullException(nameof(ownerPage));
 
-        SessionOptions.UserProperties.AddItem();
+        SessionOptions.UserProperties.AddEmptyItem();
     }
 
     public string Name

diff --git a/Source/Pages/Publish/PayloadInputFormat.cs b/Source/Pages/Publish/PayloadInputFormat.cs
@@ -0,0 +1,10 @@
+namespace MQTTnetApp.Pages.Publish;
+
+public enum PayloadInputFormat
+{
+    PlainText,
+
+    Base64String,
+
+    FilePath
+}
diff --git a/Source/Pages/Publish/PayloadInputFormatSelectorViewModel.cs b/Source/Pages/Publish/PayloadInputFormatSelectorViewModel.cs
@@ -10,7 +10,12 @@ public sealed class PayloadInputFormatSelectorViewModel : BaseViewModel
 {
     bool _isBase64String;
     bool _isFilePath;
-    bool _isPlainText = true;
+    bool _isPlainText;
+
+    public PayloadInputFormatSelectorViewModel()
+    {
+        Value = PayloadInputFormat.PlainText;
+    }
 
     public bool IsBase64String
     {
@@ -30,6 +35,36 @@ public bool IsPlainText
         set => this.RaiseAndSetIfChanged(ref _isPlainText, value);
     }
 
+    public PayloadInputFormat Value
+    {
+        get
+        {
+            if (IsPlainText)
+            {
+                return PayloadInputFormat.PlainText;
+            }
+
+            if (IsBase64String)
+            {
+                return PayloadInputFormat.Base64String;
+            }
+
+            if (IsFilePath)
+            {
+                return PayloadInputFormat.FilePath;
+            }
+
+            throw new NotSupportedException();
+        }
+
+        set
+        {
+            IsPlainText = value == PayloadInputFormat.PlainText;
+            IsBase64String = value == PayloadInputFormat.Base64String;
+            IsFilePath = value == PayloadInputFormat.FilePath;
+        }
+    }
+
     public byte[] ConvertPayloadInput(string? payloadInput)
     {
         if (string.IsNullOrEmpty(payloadInput))

diff --git a/Source/Pages/Publish/PublishPageViewModel.cs b/Source/Pages/Publish/PublishPageViewModel.cs
@@ -36,7 +36,7 @@ public void AddItem()
 
         // Prepare the UI with at lest one user property.
         // It will not be send when the name is empty.
-        newItem.UserProperties.AddItem();
+        newItem.UserProperties.AddEmptyItem();
 
         Items.Collection.Add(newItem);
         Items.SelectedItem = newItem;

diff --git a/Source/Pages/Publish/State/PublishPageStateFactory.cs b/Source/Pages/Publish/State/PublishPageStateFactory.cs
@@ -1,4 +1,5 @@
 using System;
+using MQTTnetApp.Services.State.Model;
 
 namespace MQTTnetApp.Pages.Publish.State;
 
@@ -15,15 +16,32 @@ public static PublishPageState Create(PublishPageViewModel viewModel)
 
         foreach (var item in viewModel.Items.Collection)
         {
-            state.Publishes.Add(new PublishState
+            var publishState = new PublishState
             {
                 Name = item.Name,
                 Topic = item.Topic,
                 Retain = item.Retain,
                 ContentType = item.ContentType,
+                ResponseTopic = item.ResponseTopic,
+                SubscriptionIdentifier = item.SubscriptionIdentifier,
+                TopicAlias = item.TopicAlias,
+                MessageExpiryInterval = item.MessageExpiryInterval, 
                 PayloadFormatIndicator = item.PayloadFormatIndicator.Value,
-                QualityOfServiceLevel = item.QualityOfServiceLevel.Value
-            });
+                QualityOfServiceLevel = item.QualityOfServiceLevel.Value,
+                PayloadInputFormat = item.PayloadInputFormat.Value,
+                Payload = item.Payload
+            };
+
+            foreach (var userProperty in item.UserProperties.Items)
+            {
+                publishState.UserProperties.Add(new UserProperty
+                {
+                    Name = userProperty.Name,
+                    Value = userProperty.Value
+                });
+            }
+
+            state.Publishes.Add(publishState);
         }
 
         return state;

diff --git a/Source/Pages/Publish/State/PublishPageStateLoader.cs b/Source/Pages/Publish/State/PublishPageStateLoader.cs
@@ -29,15 +29,33 @@ static PublishItemViewModel CreateItem(PublishPageViewModel ownerPage, PublishSt
         var item = new PublishItemViewModel(ownerPage)
         {
             Name = publishState.Name ?? string.Empty,
-            Topic = publishState.Topic,
+            Topic = publishState.Topic ?? string.Empty,
             Retain = publishState.Retain,
-            ContentType = publishState.ContentType,
+            ContentType = publishState.ContentType ?? string.Empty,
+            ResponseTopic = publishState.ResponseTopic ?? string.Empty,
+            SubscriptionIdentifier = publishState.SubscriptionIdentifier,
+            TopicAlias = publishState.TopicAlias,
+            MessageExpiryInterval = publishState.MessageExpiryInterval,
+            Payload = publishState.Payload ?? string.Empty,
+            PayloadFormatIndicator =
+            {
+                Value = publishState.PayloadFormatIndicator
+            },
             QualityOfServiceLevel =
             {
                 Value = publishState.QualityOfServiceLevel
+            },
+            PayloadInputFormat =
+            {
+                Value = publishState.PayloadInputFormat
             }
         };
 
+        foreach (var userProperty in publishState.UserProperties)
+        {
+            item.UserProperties.AddItem(userProperty.Name ?? string.Empty, userProperty.Value ?? string.Empty);
+        }
+
         return item;
     }
 }
diff --git a/Source/Pages/Publish/State/PublishState.cs b/Source/Pages/Publish/State/PublishState.cs
@@ -1,18 +1,34 @@
+using System.Collections.Generic;
 using MQTTnet.Protocol;
+using MQTTnetApp.Services.State.Model;
 
 namespace MQTTnetApp.Pages.Publish.State;
 
 public sealed class PublishState
 {
     public string? ContentType { get; set; }
 
+    public uint MessageExpiryInterval { get; set; }
+
     public string? Name { get; set; }
 
+    public string? Payload { get; set; }
+
     public MqttPayloadFormatIndicator PayloadFormatIndicator { get; set; }
 
+    public PayloadInputFormat PayloadInputFormat { get; set; }
+
     public MqttQualityOfServiceLevel QualityOfServiceLevel { get; set; }
 
+    public string? ResponseTopic { get; set; }
+
     public bool Retain { get; set; }
 
+    public uint SubscriptionIdentifier { get; set; }
+
     public string? Topic { get; set; }
+
+    public ushort TopicAlias { get; set; }
+
+    public List<UserProperty> UserProperties { get; set; } = new();
 }
diff --git a/Source/Pages/Subscriptions/SubscriptionsPageViewModel.cs b/Source/Pages/Subscriptions/SubscriptionsPageViewModel.cs
@@ -34,7 +34,7 @@ public void AddItem()
             Name = "Untitled"
         };
 
-        newItem.UserProperties.AddItem();
+        newItem.UserProperties.AddEmptyItem();
 
         Items.Collection.Add(newItem);
     }

diff --git a/Source/Services/State/Model/UserProperty.cs b/Source/Services/State/Model/UserProperty.cs
@@ -0,0 +1,8 @@
+namespace MQTTnetApp.Services.State.Model;
+
+public sealed class UserProperty
+{
+    public string? Name { get; set; }
+
+    public string? Value { get; set; }
+}