vite-plugin-ai-annotator 1.2.1 → 1.2.2

package/dist/annotator/selection.d.ts

@@ -20,6 +20,7 @@ export interface ElementSelectionManager {
     findSelectedChildren(element: Element): Element[];
     buildHierarchicalStructure(componentFinder?: ComponentFinder, imagePaths?: Map<Element, string>): ElementData[];
     setOnEditClick(callback: (element: Element) => void): void;
+    getBadgeForElement(element: Element): HTMLElement | null;
 }
 export declare function createElementSelectionManager(): ElementSelectionManager;
 export {};

package/dist/annotator-toolbar.js

@@ -4816,9 +4816,6 @@
   }
 
   // node_modules/@floating-ui/utils/dist/floating-ui.utils.mjs
-  var sides = ["top", "right", "bottom", "left"];
-  var alignments = ["start", "end"];
-  var placements = /* @__PURE__ */ sides.reduce((acc, side) => acc.concat(side, side + "-" + alignments[0], side + "-" + alignments[1]), []);
   var min = Math.min;
   var max = Math.max;
   var round = Math.round;
@@ -5144,102 +5141,6 @@
       right: (elementClientRect.right - clippingClientRect.right + paddingObject.right) / offsetScale.x
     };
   }
-  function getPlacementList(alignment, autoAlignment, allowedPlacements) {
-    const allowedPlacementsSortedByAlignment = alignment ? [...allowedPlacements.filter((placement) => getAlignment(placement) === alignment), ...allowedPlacements.filter((placement) => getAlignment(placement) !== alignment)] : allowedPlacements.filter((placement) => getSide(placement) === placement);
-    return allowedPlacementsSortedByAlignment.filter((placement) => {
-      if (alignment) {
-        return getAlignment(placement) === alignment || (autoAlignment ? getOppositeAlignmentPlacement(placement) !== placement : false);
-      }
-      return true;
-    });
-  }
-  var autoPlacement = function(options) {
-    if (options === void 0) {
-      options = {};
-    }
-    return {
-      name: "autoPlacement",
-      options,
-      async fn(state) {
-        var _middlewareData$autoP, _middlewareData$autoP2, _placementsThatFitOnE;
-        const {
-          rects,
-          middlewareData,
-          placement,
-          platform: platform2,
-          elements
-        } = state;
-        const {
-          crossAxis = false,
-          alignment,
-          allowedPlacements = placements,
-          autoAlignment = true,
-          ...detectOverflowOptions
-        } = evaluate(options, state);
-        const placements$1 = alignment !== void 0 || allowedPlacements === placements ? getPlacementList(alignment || null, autoAlignment, allowedPlacements) : allowedPlacements;
-        const overflow = await detectOverflow(state, detectOverflowOptions);
-        const currentIndex = ((_middlewareData$autoP = middlewareData.autoPlacement) == null ? void 0 : _middlewareData$autoP.index) || 0;
-        const currentPlacement = placements$1[currentIndex];
-        if (currentPlacement == null) {
-          return {};
-        }
-        const alignmentSides = getAlignmentSides(currentPlacement, rects, await (platform2.isRTL == null ? void 0 : platform2.isRTL(elements.floating)));
-        if (placement !== currentPlacement) {
-          return {
-            reset: {
-              placement: placements$1[0]
-            }
-          };
-        }
-        const currentOverflows = [overflow[getSide(currentPlacement)], overflow[alignmentSides[0]], overflow[alignmentSides[1]]];
-        const allOverflows = [...((_middlewareData$autoP2 = middlewareData.autoPlacement) == null ? void 0 : _middlewareData$autoP2.overflows) || [], {
-          placement: currentPlacement,
-          overflows: currentOverflows
-        }];
-        const nextPlacement = placements$1[currentIndex + 1];
-        if (nextPlacement) {
-          return {
-            data: {
-              index: currentIndex + 1,
-              overflows: allOverflows
-            },
-            reset: {
-              placement: nextPlacement
-            }
-          };
-        }
-        const placementsSortedByMostSpace = allOverflows.map((d3) => {
-          const alignment2 = getAlignment(d3.placement);
-          return [d3.placement, alignment2 && crossAxis ? (
-            // Check along the mainAxis and main crossAxis side.
-            d3.overflows.slice(0, 2).reduce((acc, v2) => acc + v2, 0)
-          ) : (
-            // Check only the mainAxis.
-            d3.overflows[0]
-          ), d3.overflows];
-        }).sort((a3, b3) => a3[1] - b3[1]);
-        const placementsThatFitOnEachSide = placementsSortedByMostSpace.filter((d3) => d3[2].slice(
-          0,
-          // Aligned placements should not check their opposite crossAxis
-          // side.
-          getAlignment(d3[0]) ? 2 : 3
-        ).every((v2) => v2 <= 0));
-        const resetPlacement = ((_placementsThatFitOnE = placementsThatFitOnEachSide[0]) == null ? void 0 : _placementsThatFitOnE[0]) || placementsSortedByMostSpace[0][0];
-        if (resetPlacement !== placement) {
-          return {
-            data: {
-              index: currentIndex + 1,
-              overflows: allOverflows
-            },
-            reset: {
-              placement: resetPlacement
-            }
-          };
-        }
-        return {};
-      }
-    };
-  };
   var flip = function(options) {
     if (options === void 0) {
       options = {};
@@ -6232,7 +6133,6 @@
     };
   }
   var offset2 = offset;
-  var autoPlacement2 = autoPlacement;
   var shift2 = shift;
   var flip2 = flip;
   var computePosition2 = (reference, floating, options) => {
@@ -6731,6 +6631,10 @@
       setOnEditClick(callback) {
         onEditClickCallback = callback;
       },
+      getBadgeForElement(element) {
+        const group = selectionGroups.get(element);
+        return group?.badge || null;
+      },
       buildHierarchicalStructure(componentFinder, imagePaths) {
         const rootElements = [];
         selectedElements.forEach((_2, element) => {
@@ -7982,25 +7886,17 @@
         if (!popoverEl || !element) return;
         if (textareaEl) {
           textareaEl.focus();
-          this.autoResizeTextarea(textareaEl, 37);
-        }
-        const virtualRef = {
-          getBoundingClientRect: () => {
-            const rect = element.getBoundingClientRect();
-            const x2 = Math.max(rect.left, 16);
-            const y3 = Math.max(rect.top, 16);
-            return new DOMRect(x2, y3, 0, 0);
-          }
-        };
-        this.popoverCleanup = autoUpdate(element, popoverEl, () => {
-          computePosition2(virtualRef, popoverEl, {
+          this.autoResizeTextarea(textareaEl);
+        }
+        const badge = this.selectionManager?.getBadgeForElement(element);
+        const referenceEl = badge || element;
+        this.popoverCleanup = autoUpdate(referenceEl, popoverEl, () => {
+          computePosition2(referenceEl, popoverEl, {
             strategy: "fixed",
+            placement: "bottom-start",
             middleware: [
-              offset2(8),
-              autoPlacement2({
-                alignment: "start",
-                allowedPlacements: ["bottom-start", "bottom-end", "top-start", "top-end"]
-              }),
+              offset2(6),
+              flip2({ fallbackPlacements: ["bottom-end", "top-start", "top-end"] }),
               shift2({ padding: 8 })
             ]
           }).then(({ x: x2, y: y3 }) => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "vite-plugin-ai-annotator",
-  "version": "1.2.1",
+  "version": "1.2.2",
   "description": "AI-powered element annotator for Vite - Pick elements and get instant AI code modifications",
   "type": "module",
   "main": "dist/vite-plugin.js",