From a9ba35c0d2c2e30e9b88aa60e0463c44ae8c7d82 Mon Sep 17 00:00:00 2001
From: Alejandro Alonso <alejandroalonsofernandez@gmail.com>
Date: Tue, 5 Dec 2023 13:53:27 +0100
Subject: [PATCH] :bug: Fix ungroup a component copy

---
 .../src/app/main/data/workspace/groups.cljs   | 52 ++-----------------
 .../app/main/ui/workspace/context_menu.cljs   |  5 +-
 2 files changed, 6 insertions(+), 51 deletions(-)

diff --git a/frontend/src/app/main/data/workspace/groups.cljs b/frontend/src/app/main/data/workspace/groups.cljs
index 9faf7515e..76e8e0df1 100644
--- a/frontend/src/app/main/data/workspace/groups.cljs
+++ b/frontend/src/app/main/data/workspace/groups.cljs
@@ -13,14 +13,10 @@
    [app.common.geom.shapes :as gsh]
    [app.common.types.component :as ctk]
    [app.common.types.container :as ctn]
-   [app.common.types.pages-list :as ctpl]
    [app.common.types.shape :as cts]
-   [app.common.types.shape-tree :as ctst]
    [app.common.types.shape.layout :as ctl]
-   [app.common.uuid :as uuid]
    [app.main.data.workspace.changes :as dch]
    [app.main.data.workspace.selection :as dws]
-   [app.main.data.workspace.shapes :as dwsh]
    [app.main.data.workspace.state-helpers :as wsh]
    [app.main.data.workspace.undo :as dwu]
    [beicon.core :as rx]
@@ -178,45 +174,6 @@
         (pcb/remove-objects [(:id frame)]))))
 
 
-(defn- clone-component-shapes-changes
-  [changes shape objects]
-  (let [shape-parent-id (:parent-id shape)
-        new-shape-id (uuid/next)
-        [_ new-shapes _]
-        (ctst/clone-object shape
-                      shape-parent-id
-                      objects
-                      (fn [object _]
-                        (cond-> object
-                          (= new-shape-id (:parent-id object))
-                          (assoc :parent-id shape-parent-id)))
-                      (fn [object _] object)
-                      new-shape-id
-                      false)
-
-        new-shapes (->> new-shapes
-                        (filter #(not= (:id %) new-shape-id)))]
-    (reduce
-     (fn [changes shape]
-       (pcb/add-object changes shape))
-     changes
-     new-shapes)))
-
-(defn remove-component-changes
-  [it page-id shape objects file-data file]
-  (let [page (ctpl/get-page file-data page-id)
-        components-v2 (dm/get-in file-data [:options :components-v2])
-        ;; In order to ungroup a component, we first make a clone of its shapes,
-        ;; and then we delete it
-        changes (-> (pcb/empty-changes it page-id)
-                    (pcb/with-objects objects)
-                    (pcb/with-library-data file-data)
-                    (pcb/with-page page)
-                    (clone-component-shapes-changes shape objects)
-                    (dwsh/delete-shapes-changes file page objects [(:id shape)] it components-v2))]
-    ;; TODO: Should we call detach-comment-thread ?
-    changes))
-
 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
 ;; GROUPS
 ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;
@@ -245,17 +202,12 @@
     (watch [it state _]
       (let [page-id   (:current-page-id state)
             objects   (wsh/lookup-page-objects state page-id)
-            file-data (get state :workspace-data)
-            file      (wsh/get-local-file state)
 
             prepare
             (fn [shape-id]
               (let [shape (get objects shape-id)
                     changes
                     (cond
-                      (ctk/main-instance? shape)
-                      (remove-component-changes it page-id shape objects file-data file)
-
                       (or (cfh/group-shape? shape) (cfh/bool-shape? shape))
                       (remove-group-changes it page-id shape objects)
 
@@ -267,7 +219,9 @@
                   (pcb/update-shapes [(:parent-id shape)] ctl/assign-cells))))
 
             selected (->> (wsh/lookup-selected state)
-                          (remove #(ctn/has-any-copy-parent? objects (get objects %))))
+                          (remove #(ctn/has-any-copy-parent? objects (get objects %)))
+                          ;; components can't be ungrouped
+                          (remove #(ctk/instance-head? (get objects %))))
             changes-list (sequence
                           (keep prepare)
                           selected)
diff --git a/frontend/src/app/main/ui/workspace/context_menu.cljs b/frontend/src/app/main/ui/workspace/context_menu.cljs
index d8ac33625..b028ad001 100644
--- a/frontend/src/app/main/ui/workspace/context_menu.cljs
+++ b/frontend/src/app/main/ui/workspace/context_menu.cljs
@@ -253,8 +253,9 @@
         single?   (= (count shapes) 1)
         do-create-artboard-from-selection #(st/emit! (dwsh/create-artboard-from-selection))
 
-        has-frame? (->> shapes (d/seek cfh/frame-shape?))
-        has-group? (->> shapes (d/seek cfh/group-shape?))
+        ;; components can't be ungrouped
+        has-frame? (->> shapes (d/seek #(and (cfh/frame-shape? %) (not (ctk/instance-head? %)))))
+        has-group? (->> shapes (d/seek #(and (cfh/group-shape? %) (not (ctk/instance-head? %)))))
         has-bool? (->> shapes (d/seek cfh/bool-shape?))
         has-mask? (->> shapes (d/seek :masked-group))