@vllnt/ui 0.2.1 → 0.3.0-canary.a42d8f4

package/dist/components/code-block/code-block.js

@@ -1,13 +1,12 @@
 "use client";
 import { jsx, jsxs } from "react/jsx-runtime";
-import { useEffect, useRef, useState } from "react";
+import {
+  useEffect,
+  useRef,
+  useState
+} from "react";
 import { Check, Copy } from "lucide-react";
 import { useTheme } from "next-themes";
-import { Prism as SyntaxHighlighter } from "react-syntax-highlighter";
-import {
-  oneDark,
-  oneLight
-} from "react-syntax-highlighter/dist/esm/styles/prism";
 import { cn } from "../../lib/utils";
 import { Button } from "../button/button";
 function extractTextFromChildren(children) {
@@ -37,12 +36,31 @@ function CodeBlock({
   showLanguage = false
 }) {
   const [copied, setCopied] = useState(false);
+  const [highlighter, setHighlighter] = useState(
+    null
+  );
   const { systemTheme, theme } = useTheme();
   const resolvedTheme = theme === "system" ? systemTheme : theme;
   const isDark = resolvedTheme !== "light";
-  const codeStyle = isDark ? oneDark : oneLight;
   const code = extractTextFromChildren(children);
   const scrollRef = useRef(null);
+  useEffect(() => {
+    let active = true;
+    void Promise.all([
+      import("react-syntax-highlighter"),
+      import("react-syntax-highlighter/dist/esm/styles/prism")
+    ]).then(([module_, styles]) => {
+      if (!active) return;
+      setHighlighter({
+        oneDark: styles.oneDark,
+        oneLight: styles.oneLight,
+        SyntaxHighlighter: module_.Prism
+      });
+    });
+    return () => {
+      active = false;
+    };
+  }, []);
   useEffect(() => {
     const element = scrollRef.current;
     if (!element) return;
@@ -66,6 +84,8 @@ function CodeBlock({
       setCopied(false);
     }, 2e3);
   };
+  const SyntaxHighlighter = highlighter?.SyntaxHighlighter;
+  const codeStyle = isDark ? highlighter?.oneDark : highlighter?.oneLight;
   return /* @__PURE__ */ jsx(
     "div",
     {
@@ -79,7 +99,7 @@ function CodeBlock({
           className: "relative overflow-x-auto overflow-y-hidden touch-pan-y",
           ref: scrollRef,
           children: [
-            /* @__PURE__ */ jsx(
+            SyntaxHighlighter ? /* @__PURE__ */ jsx(
               SyntaxHighlighter,
               {
                 codeTagProps: {
@@ -90,7 +110,7 @@ function CodeBlock({
                   }
                 },
                 customStyle: {
-                  background: "hsl(var(--background))",
+                  background: "oklch(var(--background))",
                   fontSize: "0.875rem",
                   margin: 0,
                   minWidth: "fit-content",
@@ -101,17 +121,18 @@ function CodeBlock({
                 style: codeStyle,
                 children: code
               }
-            ),
+            ) : /* @__PURE__ */ jsx("pre", { className: "m-0 min-w-fit overflow-y-hidden p-4 font-mono text-sm", children: /* @__PURE__ */ jsx("code", { className: "block bg-transparent", children: code }) }),
             /* @__PURE__ */ jsxs("div", { className: "absolute right-2 top-2 flex items-center gap-2", children: [
               showLanguage ? /* @__PURE__ */ jsx("span", { className: "text-xs font-mono text-muted-foreground uppercase tracking-wider", children: language }) : null,
               /* @__PURE__ */ jsx(
                 Button,
                 {
-                  className: "h-8 w-8",
+                  "aria-label": copied ? "Copied" : "Copy code",
+                  className: "size-8",
                   onClick: handleCopy,
                   size: "icon",
                   variant: "ghost",
-                  children: copied ? /* @__PURE__ */ jsx(Check, { className: "h-3 w-3" }) : /* @__PURE__ */ jsx(Copy, { className: "h-3 w-3" })
+                  children: copied ? /* @__PURE__ */ jsx(Check, { className: "size-3" }) : /* @__PURE__ */ jsx(Copy, { className: "size-3" })
                 }
               )
             ] })

package/dist/components/code-playground/code-playground.js

@@ -3,7 +3,7 @@ import { jsx, jsxs } from "react/jsx-runtime";
 import { useState } from "react";
 import { Check, Code, Copy, FileCode } from "lucide-react";
 import { cn } from "../../lib/utils";
-import { Button } from "../button";
+import { Button } from "../button/button";
 function CodeLine({ highlightLines, line, lineNumber }) {
   const isHighlighted = highlightLines.includes(lineNumber);
   return /* @__PURE__ */ jsxs("div", { className: cn("flex", isHighlighted && "bg-primary/10 -mx-4 px-4"), children: [
@@ -11,11 +11,13 @@ function CodeLine({ highlightLines, line, lineNumber }) {
     /* @__PURE__ */ jsx("span", { children: line })
   ] });
 }
+const EMPTY_HIGHLIGHT_LINES = [];
 function CodePlayground({
+  as: Heading = "h4",
   children,
   description,
   filename,
-  highlightLines = [],
+  highlightLines = EMPTY_HIGHLIGHT_LINES,
   language = "typescript",
   showLineNumbers = false,
   title
@@ -33,25 +35,26 @@ function CodePlayground({
   return /* @__PURE__ */ jsxs("div", { className: "my-6 rounded-lg border bg-card overflow-hidden", children: [
     /* @__PURE__ */ jsxs("div", { className: "flex items-center justify-between px-4 py-3 border-b bg-muted/30", children: [
       /* @__PURE__ */ jsxs("div", { className: "flex items-center gap-3", children: [
-        /* @__PURE__ */ jsx("div", { className: "flex h-8 w-8 items-center justify-center rounded bg-primary/10", children: /* @__PURE__ */ jsx(Code, { className: "h-4 w-4 text-primary" }) }),
+        /* @__PURE__ */ jsx("div", { className: "flex size-8 items-center justify-center rounded bg-primary/10", children: /* @__PURE__ */ jsx(Code, { className: "size-4 text-primary" }) }),
         /* @__PURE__ */ jsxs("div", { children: [
-          /* @__PURE__ */ jsx("h4", { className: "font-semibold text-sm", children: title }),
+          /* @__PURE__ */ jsx(Heading, { className: "font-semibold text-sm", children: title }),
           description ? /* @__PURE__ */ jsx("p", { className: "text-xs text-muted-foreground", children: description }) : null
         ] })
       ] }),
       /* @__PURE__ */ jsxs("div", { className: "flex items-center gap-2", children: [
         filename ? /* @__PURE__ */ jsxs("div", { className: "flex items-center gap-1 text-xs text-muted-foreground", children: [
-          /* @__PURE__ */ jsx(FileCode, { className: "h-3 w-3" }),
+          /* @__PURE__ */ jsx(FileCode, { className: "size-3" }),
           /* @__PURE__ */ jsx("span", { className: "font-mono", children: filename })
         ] }) : null,
         /* @__PURE__ */ jsx(
           Button,
           {
-            className: "h-8 w-8",
+            "aria-label": copied ? "Copied" : "Copy code",
+            className: "size-8",
             onClick: handleCopy,
             size: "icon",
             variant: "ghost",
-            children: copied ? /* @__PURE__ */ jsx(Check, { className: "h-3 w-3" }) : /* @__PURE__ */ jsx(Copy, { className: "h-3 w-3" })
+            children: copied ? /* @__PURE__ */ jsx(Check, { className: "size-3" }) : /* @__PURE__ */ jsx(Copy, { className: "size-3" })
           }
         )
       ] })
@@ -63,7 +66,7 @@ function CodePlayground({
         line,
         lineNumber: index + 1
       },
-      index
+      `${line}-${index + 1}`
     )) }) : /* @__PURE__ */ jsx("code", { children: code }) }) }),
     /* @__PURE__ */ jsx("div", { className: "px-4 py-2 border-t bg-muted/30", children: /* @__PURE__ */ jsx("span", { className: "text-xs font-mono text-muted-foreground uppercase tracking-wider", children: language }) })
   ] });
@@ -74,7 +77,7 @@ function FileTree({
 }) {
   return /* @__PURE__ */ jsxs("div", { className: "my-6 rounded-lg border bg-card overflow-hidden", children: [
     /* @__PURE__ */ jsx("div", { className: "px-4 py-2 border-b bg-muted/30", children: /* @__PURE__ */ jsxs("h4", { className: "font-semibold text-sm flex items-center gap-2", children: [
-      /* @__PURE__ */ jsx(FileCode, { className: "h-4 w-4" }),
+      /* @__PURE__ */ jsx(FileCode, { className: "size-4" }),
       title
     ] }) }),
     /* @__PURE__ */ jsx("div", { className: "p-4 font-mono text-sm [&>ul]:m-0 [&>ul]:list-none [&_ul]:ml-4 [&_ul]:list-none [&_li]:text-muted-foreground", children })

package/dist/components/combobox/combobox.js

@@ -3,7 +3,7 @@ import { jsx, jsxs } from "react/jsx-runtime";
 import * as React from "react";
 import { Check, ChevronsUpDown } from "lucide-react";
 import { cn } from "../../lib/utils";
-import { Button } from "../button";
+import { Button } from "../button/button";
 import {
   Command,
   CommandEmpty,
@@ -15,11 +15,6 @@ import {
 import { Popover, PopoverContent, PopoverTrigger } from "../popover";
 function useComboboxValue(value, onValueChange) {
   const [internalValue, setInternalValue] = React.useState(value ?? "");
-  React.useEffect(() => {
-    if (value !== void 0) {
-      setInternalValue(value);
-    }
-  }, [value]);
   const resolvedValue = value ?? internalValue;
   const setResolvedValue = (nextValue) => {
     if (value === void 0) {
@@ -48,7 +43,7 @@ function ComboboxOptionItem({
           Check,
           {
             className: cn(
-              "mr-2 h-4 w-4",
+              "mr-2 size-4",
               selectedValue === option.value ? "opacity-100" : "opacity-0"
             )
           }
@@ -58,6 +53,38 @@ function ComboboxOptionItem({
     }
   );
 }
+function ComboboxListPanel({
+  className,
+  commandClassName,
+  emptyText,
+  listboxId,
+  onSelect,
+  options,
+  resolvedValue,
+  searchPlaceholder
+}) {
+  return /* @__PURE__ */ jsx(
+    PopoverContent,
+    {
+      className: cn("w-[var(--radix-popover-trigger-width)] p-0", className),
+      children: /* @__PURE__ */ jsxs(Command, { className: commandClassName, children: [
+        /* @__PURE__ */ jsx(CommandInput, { placeholder: searchPlaceholder }),
+        /* @__PURE__ */ jsxs(CommandList, { id: listboxId, children: [
+          /* @__PURE__ */ jsx(CommandEmpty, { children: emptyText }),
+          /* @__PURE__ */ jsx(CommandGroup, { children: options.map((option) => /* @__PURE__ */ jsx(
+            ComboboxOptionItem,
+            {
+              onSelect,
+              option,
+              selectedValue: resolvedValue
+            },
+            option.value
+          )) })
+        ] })
+      ] })
+    }
+  );
+}
 const Combobox = React.forwardRef(
   ({
     className,
@@ -71,6 +98,7 @@ const Combobox = React.forwardRef(
     value
   }, reference) => {
     const [open, setOpen] = React.useState(false);
+    const listboxId = React.useId();
     const { resolvedValue, setResolvedValue } = useComboboxValue(
       value,
       onValueChange
@@ -86,39 +114,30 @@ const Combobox = React.forwardRef(
       /* @__PURE__ */ jsx(PopoverTrigger, { asChild: true, children: /* @__PURE__ */ jsxs(
         Button,
         {
+          "aria-controls": listboxId,
           "aria-expanded": open,
+          "aria-haspopup": "listbox",
           className: cn("w-full justify-between", triggerClassName),
           ref: reference,
           role: "combobox",
           variant: "outline",
           children: [
             /* @__PURE__ */ jsx("span", { className: "truncate", children: selectedOption ? selectedOption.label : placeholder }),
-            /* @__PURE__ */ jsx(ChevronsUpDown, { className: "ml-2 h-4 w-4 shrink-0 opacity-50" })
+            /* @__PURE__ */ jsx(ChevronsUpDown, { className: "ml-2 size-4 shrink-0 opacity-50" })
           ]
         }
       ) }),
       /* @__PURE__ */ jsx(
-        PopoverContent,
+        ComboboxListPanel,
         {
-          className: cn(
-            "w-[var(--radix-popover-trigger-width)] p-0",
-            className
-          ),
-          children: /* @__PURE__ */ jsxs(Command, { className: commandClassName, children: [
-            /* @__PURE__ */ jsx(CommandInput, { placeholder: searchPlaceholder }),
-            /* @__PURE__ */ jsxs(CommandList, { children: [
-              /* @__PURE__ */ jsx(CommandEmpty, { children: emptyText }),
-              /* @__PURE__ */ jsx(CommandGroup, { children: options.map((option) => /* @__PURE__ */ jsx(
-                ComboboxOptionItem,
-                {
-                  onSelect: handleSelect,
-                  option,
-                  selectedValue: resolvedValue
-                },
-                option.value
-              )) })
-            ] })
-          ] })
+          className,
+          commandClassName,
+          emptyText,
+          listboxId,
+          onSelect: handleSelect,
+          options,
+          resolvedValue,
+          searchPlaceholder
         }
       )
     ] });

package/dist/components/command/command.js

@@ -26,7 +26,7 @@ const CommandDialog = ({ children, ...props }) => {
 const CommandInput = React.forwardRef(({ className, ...props }, reference) => (
   // eslint-disable-next-line react/no-unknown-property
   /* @__PURE__ */ jsxs("div", { className: "flex items-center border-b px-3", "cmdk-input-wrapper": "", children: [
-    /* @__PURE__ */ jsx(Search, { className: "mr-2 h-4 w-4 shrink-0 opacity-50" }),
+    /* @__PURE__ */ jsx(Search, { className: "mr-2 size-4 shrink-0 opacity-50" }),
     /* @__PURE__ */ jsx(
       CommandPrimitive.Input,
       {

package/dist/components/comment-pin/comment-pin.js

@@ -0,0 +1,104 @@
+"use client";
+import { Fragment, jsx, jsxs } from "react/jsx-runtime";
+import {
+  forwardRef
+} from "react";
+import { cn } from "../../lib/utils";
+const DEFAULT_LABELS = {
+  region: "Comment",
+  unreadSuffix: "unread"
+};
+const STATE_FILL = {
+  open: "bg-foreground text-background",
+  resolved: "bg-muted text-muted-foreground"
+};
+const PinBody = (props) => {
+  const showBadge = typeof props.unread === "number" && props.unread > 0;
+  const useAccent = props.accent && props.state === "open";
+  return /* @__PURE__ */ jsxs(Fragment, { children: [
+    /* @__PURE__ */ jsx(
+      "span",
+      {
+        "aria-hidden": "true",
+        className: cn(
+          "flex size-7 items-center justify-center rounded-full border border-background text-[11px] font-semibold shadow-sm",
+          STATE_FILL[props.state]
+        ),
+        "data-comment-pin-body": true,
+        style: useAccent ? { backgroundColor: props.accent, color: "white" } : void 0,
+        children: props.authorInitial ?? "\u2022"
+      }
+    ),
+    showBadge ? /* @__PURE__ */ jsx(
+      "span",
+      {
+        "aria-hidden": "true",
+        className: "absolute -right-1 -top-1 inline-flex min-h-[14px] min-w-[14px] items-center justify-center rounded-full bg-red-500 px-1 text-[9px] font-medium text-white",
+        "data-comment-pin-unread": true,
+        children: props.unread
+      }
+    ) : null
+  ] });
+};
+const CommentPin = forwardRef(
+  (props, ref) => {
+    const {
+      authorInitial,
+      className,
+      color,
+      labels,
+      onActivate,
+      state = "open",
+      unread,
+      x,
+      y,
+      ...rest
+    } = props;
+    const resolvedLabels = { ...DEFAULT_LABELS, ...labels };
+    const showBadge = typeof unread === "number" && unread > 0;
+    const ariaLabel = showBadge ? `${resolvedLabels.region}, ${unread} ${resolvedLabels.unreadSuffix}` : resolvedLabels.region;
+    const handleActivateComment = () => {
+      onActivate?.();
+    };
+    const body = /* @__PURE__ */ jsx(
+      PinBody,
+      {
+        accent: color,
+        authorInitial,
+        state,
+        unread
+      }
+    );
+    return /* @__PURE__ */ jsx(
+      "div",
+      {
+        "aria-label": ariaLabel,
+        className: cn(
+          "absolute z-30 inline-flex -translate-x-1/2 -translate-y-1/2",
+          className
+        ),
+        "data-comment-pin": true,
+        "data-comment-pin-state": state,
+        ref,
+        role: "img",
+        style: { left: x, top: y },
+        ...rest,
+        children: onActivate ? /* @__PURE__ */ jsx(
+          "button",
+          {
+            "aria-label": ariaLabel,
+            className: "relative inline-flex rounded-full focus-visible:outline-none focus-visible:ring-2 focus-visible:ring-ring",
+            "data-comment-pin-trigger": true,
+            onClick: handleActivateComment,
+            type: "button",
+            children: body
+          }
+        ) : /* @__PURE__ */ jsx("span", { className: "relative inline-flex", children: body })
+      }
+    );
+  }
+);
+CommentPin.displayName = "CommentPin";
+export {
+  CommentPin
+};

package/dist/components/comment-pin/index.js

@@ -0,0 +1,6 @@
+import {
+  CommentPin
+} from "./comment-pin";
+export {
+  CommentPin
+};

package/dist/components/comparison/comparison.js

@@ -24,6 +24,7 @@ const variantConfig = {
 };
 function Comparison({
   after,
+  as: Heading = "h4",
   before,
   title,
   ...rest
@@ -40,7 +41,7 @@ function Comparison({
   const BeforeIcon = beforeConfig.icon;
   const AfterIcon = afterConfig.icon;
   return /* @__PURE__ */ jsxs("div", { className: "my-6", children: [
-    title ? /* @__PURE__ */ jsx("h4", { className: "font-semibold mb-3", children: title }) : null,
+    title ? /* @__PURE__ */ jsx(Heading, { className: "font-semibold mb-3", children: title }) : null,
     /* @__PURE__ */ jsxs("div", { className: "grid grid-cols-1 md:grid-cols-2 gap-4", children: [
       /* @__PURE__ */ jsxs("div", { className: cn("rounded-lg border", beforeConfig.className), children: [
         /* @__PURE__ */ jsx(
@@ -53,18 +54,18 @@ function Comparison({
             children: before.title
           }
         ),
-        /* @__PURE__ */ jsx("ul", { className: "p-4 space-y-2", children: before.items.map((item, index) => /* @__PURE__ */ jsxs("li", { className: "flex items-start gap-2 text-sm", children: [
+        /* @__PURE__ */ jsx("ul", { className: "p-4 space-y-2", children: before.items.map((item) => /* @__PURE__ */ jsxs("li", { className: "flex items-start gap-2 text-sm", children: [
           /* @__PURE__ */ jsx(
             BeforeIcon,
             {
               className: cn(
-                "h-4 w-4 mt-0.5 flex-shrink-0",
+                "size-4 mt-0.5 flex-shrink-0",
                 beforeConfig.iconClass
               )
             }
           ),
           /* @__PURE__ */ jsx("span", { children: item })
-        ] }, index)) })
+        ] }, item)) })
       ] }),
       /* @__PURE__ */ jsxs("div", { className: cn("rounded-lg border", afterConfig.className), children: [
         /* @__PURE__ */ jsx(
@@ -77,18 +78,18 @@ function Comparison({
             children: after.title
           }
         ),
-        /* @__PURE__ */ jsx("ul", { className: "p-4 space-y-2", children: after.items.map((item, index) => /* @__PURE__ */ jsxs("li", { className: "flex items-start gap-2 text-sm", children: [
+        /* @__PURE__ */ jsx("ul", { className: "p-4 space-y-2", children: after.items.map((item) => /* @__PURE__ */ jsxs("li", { className: "flex items-start gap-2 text-sm", children: [
           /* @__PURE__ */ jsx(
             AfterIcon,
             {
               className: cn(
-                "h-4 w-4 mt-0.5 flex-shrink-0",
+                "size-4 mt-0.5 flex-shrink-0",
                 afterConfig.iconClass
               )
             }
           ),
           /* @__PURE__ */ jsx("span", { children: item })
-        ] }, index)) })
+        ] }, item)) })
       ] })
     ] })
   ] });
@@ -99,15 +100,15 @@ function BeforeAfter({ after, before, title }) {
     /* @__PURE__ */ jsxs("div", { className: "grid grid-cols-1 md:grid-cols-[1fr_auto_1fr] gap-4 items-start", children: [
       /* @__PURE__ */ jsxs("div", { className: "rounded-lg border border-red-500/30 bg-red-500/5 overflow-hidden", children: [
         /* @__PURE__ */ jsxs("div", { className: "px-4 py-2 bg-red-500/10 text-red-700 dark:text-red-300 font-medium text-sm flex items-center gap-2", children: [
-          /* @__PURE__ */ jsx(X, { className: "h-4 w-4" }),
+          /* @__PURE__ */ jsx(X, { className: "size-4" }),
           "Before"
         ] }),
         /* @__PURE__ */ jsx("div", { className: "p-4 text-sm [&>pre]:my-0", children: before })
       ] }),
-      /* @__PURE__ */ jsx("div", { className: "hidden md:flex items-center justify-center h-full", children: /* @__PURE__ */ jsx(ArrowRight, { className: "h-6 w-6 text-muted-foreground" }) }),
+      /* @__PURE__ */ jsx("div", { className: "hidden md:flex items-center justify-center h-full", children: /* @__PURE__ */ jsx(ArrowRight, { className: "size-6 text-muted-foreground" }) }),
       /* @__PURE__ */ jsxs("div", { className: "rounded-lg border border-green-500/30 bg-green-500/5 overflow-hidden", children: [
         /* @__PURE__ */ jsxs("div", { className: "px-4 py-2 bg-green-500/10 text-green-700 dark:text-green-300 font-medium text-sm flex items-center gap-2", children: [
-          /* @__PURE__ */ jsx(Check, { className: "h-4 w-4" }),
+          /* @__PURE__ */ jsx(Check, { className: "size-4" }),
           "After"
         ] }),
         /* @__PURE__ */ jsx("div", { className: "p-4 text-sm [&>pre]:my-0", children: after })

package/dist/components/completion-dialog/completion-dialog.js

@@ -1,9 +1,10 @@
 "use client";
 import { jsx, jsxs } from "react/jsx-runtime";
-import { memo, useCallback, useEffect, useRef } from "react";
+import { memo, useEffect, useRef } from "react";
 import { cn } from "../../lib/utils";
-import { Button } from "../button";
+import { Button } from "../button/button";
 function DialogContent({
+  as: Heading = "h2",
   cancelLabel = "Skip",
   cancelShortcut = "S",
   className,
@@ -39,7 +40,7 @@ function DialogContent({
             children: closeIcon ?? /* @__PURE__ */ jsx(
               "svg",
               {
-                className: "h-4 w-4",
+                className: "size-4",
                 fill: "none",
                 stroke: "currentColor",
                 viewBox: "0 0 24 24",
@@ -57,7 +58,7 @@ function DialogContent({
           }
         ),
         /* @__PURE__ */ jsxs("div", { className: "mb-4", children: [
-          /* @__PURE__ */ jsx("h2", { className: "text-lg font-semibold", id: "completion-dialog-title", children: title }),
+          /* @__PURE__ */ jsx(Heading, { className: "text-lg font-semibold", id: "completion-dialog-title", children: title }),
           description ? /* @__PURE__ */ jsx("div", { className: "text-sm text-muted-foreground mt-1.5", children: description }) : null
         ] }),
         /* @__PURE__ */ jsxs("div", { className: "flex flex-row gap-2", children: [
@@ -83,6 +84,7 @@ function DialogContent({
   );
 }
 function CompletionDialogImpl({
+  as,
   cancelLabel,
   cancelShortcut = "S",
   className,
@@ -96,8 +98,11 @@ function CompletionDialogImpl({
   onConfirm,
   title
 }) {
-  const handleKeyDown = useCallback(
-    (event) => {
+  const keyDownHandlerRef = useRef(() => {
+    return;
+  });
+  useEffect(() => {
+    keyDownHandlerRef.current = (event) => {
       if (!isOpen) return;
       if (event.key === "Escape") {
         event.preventDefault();
@@ -116,16 +121,18 @@ function CompletionDialogImpl({
         event.stopPropagation();
         onCancel();
       }
-    },
-    [isOpen, onClose, onConfirm, onCancel, confirmShortcut, cancelShortcut]
-  );
+    };
+  }, [cancelShortcut, confirmShortcut, isOpen, onCancel, onClose, onConfirm]);
   useEffect(() => {
     if (!isOpen) return;
-    document.addEventListener("keydown", handleKeyDown, true);
+    const onDocumentKeyDown = (event) => {
+      keyDownHandlerRef.current(event);
+    };
+    document.addEventListener("keydown", onDocumentKeyDown, true);
     return () => {
-      document.removeEventListener("keydown", handleKeyDown, true);
+      document.removeEventListener("keydown", onDocumentKeyDown, true);
     };
-  }, [isOpen, handleKeyDown]);
+  }, [isOpen]);
   if (!isOpen) return null;
   return /* @__PURE__ */ jsxs(
     "div",
@@ -149,6 +156,7 @@ function CompletionDialogImpl({
         /* @__PURE__ */ jsx(
           DialogContent,
           {
+            as,
             cancelLabel,
             cancelShortcut,
             className,

package/dist/components/connector-edge/connector-edge.js

@@ -0,0 +1,66 @@
+import { jsx, jsxs } from "react/jsx-runtime";
+import { forwardRef } from "react";
+import { cn } from "../../lib/utils";
+import { EdgeLabel } from "../edge-label/edge-label";
+const strokeClasses = {
+  active: "stroke-sky-500",
+  blocked: "stroke-amber-500",
+  idle: "stroke-muted-foreground/60"
+};
+const ConnectorEdge = forwardRef(
+  ({ className, end, label, start, state = "idle", ...props }, ref) => {
+    const width = Math.max(Math.abs(end.x - start.x), 32);
+    const height = Math.max(Math.abs(end.y - start.y), 32);
+    const midX = width / 2;
+    const startX = start.x <= end.x ? 4 : width - 4;
+    const endX = start.x <= end.x ? width - 4 : 4;
+    const startY = start.y <= end.y ? 4 : height - 4;
+    const endY = start.y <= end.y ? height - 4 : 4;
+    const path = `M ${startX} ${startY} C ${midX} ${startY}, ${midX} ${endY}, ${endX} ${endY}`;
+    const style = {
+      height,
+      width
+    };
+    return /* @__PURE__ */ jsxs(
+      "div",
+      {
+        className: cn("relative inline-flex", className),
+        ref,
+        style,
+        ...props,
+        children: [
+          /* @__PURE__ */ jsx(
+            "svg",
+            {
+              className: "overflow-visible",
+              height,
+              viewBox: `0 0 ${width} ${height}`,
+              width,
+              children: /* @__PURE__ */ jsx(
+                "path",
+                {
+                  className: cn("fill-none stroke-[2.5]", strokeClasses[state]),
+                  d: path,
+                  strokeDasharray: state === "blocked" ? "4 4" : void 0,
+                  strokeLinecap: "round"
+                }
+              )
+            }
+          ),
+          label ? /* @__PURE__ */ jsx(
+            EdgeLabel,
+            {
+              className: "absolute left-1/2 top-1/2 -translate-x-1/2 -translate-y-1/2",
+              emphasis: state === "active" ? "active" : "subtle",
+              children: label
+            }
+          ) : null
+        ]
+      }
+    );
+  }
+);
+ConnectorEdge.displayName = "ConnectorEdge";
+export {
+  ConnectorEdge
+};

package/dist/components/connector-edge/index.js

@@ -0,0 +1,6 @@
+import {
+  ConnectorEdge
+} from "./connector-edge";
+export {
+  ConnectorEdge
+};