From da3f2f820cb34029ddb3d1e86f2ccd700a7321b2 Mon Sep 17 00:00:00 2001
From: Florian Schroedl <flo.schroedl@gmail.com>
Date: Tue, 23 Jul 2024 15:28:13 +0200
Subject: [PATCH] Add generic context menu actions

---
 .../ui/workspace/tokens/context_menu.cljs     | 32 +++++++++++--------
 1 file changed, 19 insertions(+), 13 deletions(-)

diff --git a/frontend/src/app/main/ui/workspace/tokens/context_menu.cljs b/frontend/src/app/main/ui/workspace/tokens/context_menu.cljs
index 32a500c9c..9a0affd8c 100644
--- a/frontend/src/app/main/ui/workspace/tokens/context_menu.cljs
+++ b/frontend/src/app/main/ui/workspace/tokens/context_menu.cljs
@@ -361,10 +361,28 @@
             [:separator]
             gap-items)))
 
+(defn generic-attribute-actions [attributes title {:keys [token selected-shapes]}]
+  (let [{:keys [on-update-shape] :as p} (get wtc/token-types (:type token))
+        {:keys [selected-pred shape-ids]} (attribute-actions token selected-shapes attributes)]
+    (map (fn [attribute]
+           (let [selected? (selected-pred attribute)
+                 props {:attributes #{attribute}
+                        :token token
+                        :shape-ids shape-ids}]
+
+             {:title title
+              :selected? selected?
+              :action #(if selected?
+                         (st/emit! (wtc/unapply-token props))
+                         (st/emit! (wtc/apply-token (assoc props :on-update-shape on-update-shape))))}))
+         attributes)))
+
 (def shape-attribute-actions-map
   {:border-radius border-radius-attribute-actions
    :spacing spacing-attribute-actions
-   :sizing nil})
+   :rotation (partial generic-attribute-actions #{:rotation} "Rotation")
+   :opacity (partial generic-attribute-actions #{:opacity} "Opacity")
+   :stroke-width (partial generic-attribute-actions #{:stroke-width} "Stroke Width")})
 
 (defn shape-attribute-actions [{:keys [token] :as context-data}]
   (when-let [with-actions (get shape-attribute-actions-map (:type token))]
@@ -381,8 +399,6 @@
                                            :action #(update-fn context-data attributes)
                                            :selected? selected?)))))]
     (case (:type token)
-      :border-radius (border-radius-attribute-actions context-data)
-      :spacing (spacing-attribute-actions context-data)
       :sizing       (attributes->actions
                      apply-sizing-token
                      [{:title "All" :attributes #{:width :height :layout-item-min-w :layout-item-max-w :layout-item-min-h :layout-item-max-h}}
@@ -403,17 +419,7 @@
                        {:title "y" :attributes #{:y}}])
                       ;;TODO: Background blur {:title "Background blur" :attributes #{:width}}])
 
-      :opacity      (attributes->actions
-                     apply-rotation-opacity-stroke-token
-                     [{:title "opacity" :attributes #{:opacity}}])
 
-      :rotation     (attributes->actions
-                     apply-rotation-opacity-stroke-token
-                     [{:title "rotation" :attributes #{:rotation}}])
-
-      :stroke-width (attributes->actions
-                     apply-rotation-opacity-stroke-token
-                     [{:title "stroke width" :attributes #{:stroke-width}}])
 
       [])))