From be0e1841b89d11ecbea93d227c1ec21be6dd6169 Mon Sep 17 00:00:00 2001
From: LAGonauta <lagonauta@gmail.com>
Date: Sat, 27 Feb 2021 12:52:03 -0300
Subject: [PATCH] Anitial builder

---
 plugins/windows-notification/CMakeLists.txt   |   1 +
 .../api/src/converter.cpp                     |   2 +-
 .../src/toast_notification_builder.vala       | 166 ++++++++++++++++++
 .../src/windows_notifications_plugin.vala     | 100 ++---------
 4 files changed, 182 insertions(+), 87 deletions(-)
 create mode 100644 plugins/windows-notification/src/toast_notification_builder.vala

diff --git a/plugins/windows-notification/CMakeLists.txt b/plugins/windows-notification/CMakeLists.txt
index 9cef5965..915fd121 100644
--- a/plugins/windows-notification/CMakeLists.txt
+++ b/plugins/windows-notification/CMakeLists.txt
@@ -12,6 +12,7 @@ vala_precompile(WINDOWS_NOTIFICATION_VALA_C
 SOURCES
     src/windows_notifications_plugin.vala
     src/windows_notifications_register_plugin.vala
+    src/toast_notification_builder.vala
     # src/win_notification_provider.vala
 CUSTOM_VAPIS
     ${CMAKE_BINARY_DIR}/exports/xmpp-vala.vapi
diff --git a/plugins/windows-notification/api/src/converter.cpp b/plugins/windows-notification/api/src/converter.cpp
index b6ab3c7b..67a52071 100644
--- a/plugins/windows-notification/api/src/converter.cpp
+++ b/plugins/windows-notification/api/src/converter.cpp
@@ -10,7 +10,7 @@ char* wsview_to_char(const std::wstring_view wstr)
         return nullptr;
     }
     int final_size = WideCharToMultiByte(CP_UTF8, 0, wstr.data(), (int)wstr.size(), nullptr, 0, nullptr, nullptr);
-    char* strTo = new char[final_size];
+    char* strTo = new char[final_size + 1]{ 0 };
     WideCharToMultiByte(CP_UTF8, 0, wstr.data(), (int)wstr.size(), strTo, final_size, nullptr, nullptr);
     return strTo;
 }
diff --git a/plugins/windows-notification/src/toast_notification_builder.vala b/plugins/windows-notification/src/toast_notification_builder.vala
new file mode 100644
index 00000000..6abf3f65
--- /dev/null
+++ b/plugins/windows-notification/src/toast_notification_builder.vala
@@ -0,0 +1,166 @@
+using Dino.Entities;
+using Dino.Plugins.WindowsNotification.Vapi;
+using winrt.Windows.UI.Notifications;
+using Dino.Plugins.WindowsNotification.Vapi.Win32Api;
+using Xmpp;
+
+namespace Dino.Plugins.WindowsNotification {
+    private delegate void NodeFunction(StanzaNode node);
+    private class Button {
+        public string title;
+        public string arguments;
+    }
+
+    public class ToastNotificationBuilder  {
+        private static bool _supportsModernFeatures = SupportsModernNotifications();
+        private Gee.List<Button> _buttons = new Gee.ArrayList<Button>();
+        private string _header = null;
+        private string _body = null;
+        private string _imagePath = null;
+
+        public ToastNotificationBuilder() {
+        }
+
+        public ToastNotificationBuilder AddButton(string title, string arguments) {
+            _buttons.add(new Button() { title = title, arguments = arguments });
+            return this;
+        }
+
+        public ToastNotificationBuilder SetHeader(string header) {
+            _header = header;
+            return this;
+        }
+
+        public ToastNotificationBuilder SetBody(string body) {
+            _body = body;
+            return this;
+        }
+
+        public ToastNotificationBuilder SetImage(string image_path) {
+            _imagePath = image_path;
+            return this;
+        }
+
+        // legacy templates: https://docs.microsoft.com/en-us/previous-versions/windows/apps/hh761494(v=win.10)
+        public ToastNotification Build() {
+            ToastTemplateType templateType = _header != null ? ToastTemplateType.ToastText02 : ToastTemplateType.ToastText01;
+            if (_imagePath != null) {
+                if (templateType == ToastTemplateType.ToastText02) {
+                    templateType = ToastTemplateType.ToastImageAndText02;
+                } else {
+                    templateType = ToastTemplateType.ToastImageAndText01;
+                }
+            }
+
+            var template = BuildStanzaFromXml(ToastNotificationManager.GetTemplateContent(templateType));
+            { // add header and body
+                var attributes = new Gee.ArrayList<StanzaAttribute>();
+                attributes.add(new StanzaAttribute.build("", "id", "1"));
+                attributes.add(new StanzaAttribute.build("", "id", "2"));
+    
+                var nodes = FindRecursive(template, "text", attributes);
+                foreach (var node in nodes) {
+                    var attr = node.get_attribute_raw("id", "");
+                    if (attr != null) {
+                        if (attr.val == "1") {
+                            if (templateType == ToastTemplateType.ToastText02 || templateType ==  ToastTemplateType.ToastImageAndText02) {
+                                node.put_node(new StanzaNode.text(_header));
+                            } else {
+                                node.put_node(new StanzaNode.text(_body));
+                            }
+                        } else if (attr.val == "2") {
+                            node.put_node(new StanzaNode.text(_body));
+                        }
+                    }
+                }
+            }
+
+            { // add image
+                var nodes = FindRecursive(template, "image", null);
+                foreach (var node in nodes) {
+                    var attr = node.get_attribute_raw("src", "");
+                    if (attr != null) {
+                        attr.val = _imagePath;
+                    }
+                }
+            }
+
+            if (_supportsModernFeatures && _buttons.size > 0) { // allow buttons
+                var actions = new StanzaNode.build("actions", "", null, null);
+                foreach (var button in _buttons) {
+                    var attributes = new Gee.ArrayList<StanzaAttribute>();
+                    attributes.add(new StanzaAttribute.build("", "content", button.title));
+                    attributes.add(new StanzaAttribute.build("", "arguments", button.arguments));
+
+                    var action = new StanzaNode.build("action", "", null, attributes);
+                    actions.get_all_subnodes().add(action);
+                }
+                template.get_all_subnodes().add(actions);
+            }
+
+            var xml = ToXml(template);
+            return new ToastNotification(xml);
+        }
+
+        private StanzaNode BuildStanzaFromXml(string xml) {
+            var reader = new Xmpp.StanzaReader.for_string(xml);
+
+            StanzaNode root_node = null;
+            var loop = new MainLoop();
+            reader.read_node
+                .begin((obj, res) => {
+                root_node = reader.read_node.end(res);
+                loop.quit();
+            });
+            loop.run();
+
+            ExecuteOnAllSubNodes(root_node, (node) => {
+                node.ns_uri = "";
+                foreach (var attr in node.attributes){
+                    attr.ns_uri = "";
+                }
+            });
+
+            return root_node;
+        }
+
+        private Gee.ArrayList<StanzaNode> FindRecursive(StanzaNode node, string tag_name, Gee.List<StanzaAttribute>? attributes) {
+            var ret = new Gee.ArrayList<StanzaNode>();
+            FindRecursiveInternal(node, tag_name, attributes, ret);
+            return ret;
+        }
+
+        private void FindRecursiveInternal(StanzaNode root_node, string tag_name, Gee.List<StanzaAttribute>? attributes, Gee.List<StanzaNode> list) {
+            if (root_node.name == tag_name) {
+                if (attributes != null) {
+                    foreach (var attr in attributes) {
+                        var node_attr = root_node.get_attribute_raw(attr.name, attr.ns_uri);
+                        if (node_attr != null && node_attr.equals(attr)) {
+                            list.add(root_node);
+                            break;
+                        }
+                    }
+                }
+                else {
+                    list.add(root_node);
+                }
+            }
+            foreach (var node in root_node.get_all_subnodes()) {
+                FindRecursiveInternal(node, tag_name, attributes, list);
+            }
+        }
+
+        private string ToXml(StanzaNode node) {
+            var namespace_state = new NamespaceState();
+            namespace_state.set_current("");
+            return node.to_xml(namespace_state);
+        }
+
+        private void ExecuteOnAllSubNodes(StanzaNode root_node, NodeFunction func) {
+            func(root_node);
+            foreach (var node in root_node.get_all_subnodes()) {
+                ExecuteOnAllSubNodes(node, func);
+            }
+        }
+    }
+}
diff --git a/plugins/windows-notification/src/windows_notifications_plugin.vala b/plugins/windows-notification/src/windows_notifications_plugin.vala
index f6d536dc..c5a49f61 100644
--- a/plugins/windows-notification/src/windows_notifications_plugin.vala
+++ b/plugins/windows-notification/src/windows_notifications_plugin.vala
@@ -8,70 +8,7 @@ public class Plugin : RootInterface, Object {
 
     private static string AUMID = "org.dino.Dino";
     private ToastNotifier notifier;
-    private ToastNotification notification; // Notifications remove their actions when they get out of scope
-
-    private delegate void NodeFunction(StanzaNode node);
-
-    private StanzaNode Build(string xml) {
-        var reader = new Xmpp.StanzaReader.for_string(xml);
-
-        StanzaNode root_node = null;
-        var loop = new MainLoop();
-        reader.read_node
-            .begin((obj, res) => {
-              root_node = reader.read_node.end(res);
-              loop.quit();
-        });
-        loop.run();
-
-        ExecuteOnAllSubNodes(root_node, (node) => {
-            node.ns_uri = "";
-            foreach (var attr in node.attributes){
-                attr.ns_uri = "";
-            }
-        });
-
-        return root_node;
-    }
-
-    private Gee.ArrayList<StanzaNode> FindRecursive(StanzaNode node, string tag_name, Gee.List<StanzaAttribute>? attributes) {
-        var ret = new Gee.ArrayList<StanzaNode>();
-        FindRecursiveInternal(node, tag_name, attributes, ret);
-        return ret;
-    }
-
-    private void FindRecursiveInternal(StanzaNode root_node, string tag_name, Gee.List<StanzaAttribute>? attributes, Gee.List<StanzaNode> list) {
-        if (root_node.name == tag_name) {
-            if (attributes != null) {
-                foreach (var attr in attributes) {
-                    var node_attr = root_node.get_attribute_raw(attr.name, attr.ns_uri);
-                    if (node_attr != null && node_attr.equals(attr)) {
-                        list.add(root_node);
-                        break;
-                    }
-                }
-            }
-            else {
-                list.add(root_node);
-            }
-        }
-        foreach (var node in root_node.get_all_subnodes()) {
-            FindRecursiveInternal(node, tag_name, attributes, list);
-        }
-    }
-
-    private string ToXml(StanzaNode node) {
-        var namespace_state = new NamespaceState();
-        namespace_state.set_current("");
-        return node.to_xml(namespace_state);
-    }
-
-    private void ExecuteOnAllSubNodes(StanzaNode root_node, NodeFunction func) {
-        func(root_node);
-        foreach (var node in root_node.get_all_subnodes()) {
-            ExecuteOnAllSubNodes(node, func);
-        }
-    }
+    private ToastNotification notification; // Notifications remove their actions when they go out of scope
 
     public void registered(Dino.Application app) {
         if (!winrt.InitApartment())
@@ -90,31 +27,22 @@ public class Plugin : RootInterface, Object {
         }
 
         {
-            var give_me_template = ToastTemplateType.ToastText02;
-            var template = ToastNotificationManager.GetTemplateContent(give_me_template);
-            var node = Build(template);
-            {
-                var attributes = new Gee.ArrayList<StanzaAttribute>();
-                attributes.add(new StanzaAttribute.build("", "id", "1"));
-                attributes.add(new StanzaAttribute.build("", "id", "2"));
-
-                var nodes = FindRecursive(node, "text", attributes);
-                foreach (var node_ in nodes) {
-                    var attr = node_.get_attribute_raw("id", "");
-                    if (attr != null) {
-                        if (attr.val == "1") {
-                            node_.put_node(new StanzaNode.text("Header!"));
-                        } else if (attr.val == "2") {
-                            node_.put_node(new StanzaNode.text("Text!"));
-                        }
-                    }
-                }
-            }
+            //var notificationBuilder = new ToastNotificationBuilder.from_string(template);
+            this.notification = new ToastNotificationBuilder()
+                .SetHeader("Hello")
+                .SetBody("World")
+                .SetImage("C:\\Users\\lfsaf\\Pictures\\14236067.png")
+                .AddButton("Clique aqui", "argumento")
+                .Build();
             
             this.notifier = new ToastNotifier(AUMID);
-            this.notification = new ToastNotification(ToXml(node));
-            var token = notification.Activated((c, d) => {
+            var token = this.notification.Activated((c, d) => {
               stdout.printf("\nYay! Activated!\n");
+              var tr = false;
+              if (c != null && c == "argumento") {
+                    tr = true;
+              }
+              
               stdout.flush();
             });