@docyrus/ui-pro-ai-assistant 0.0.1 → 0.0.2

package/src/internal/plate-editor/plate-ui/floating-discussion.tsx

@@ -344,13 +344,13 @@ const resolveOverlappingTop = (
 const useCommentingNode = () => useEditorSelector((editor) => {
   if (!editor?.api || !editor.selection || editor.api.isExpanded()) return;
 
-  return editor.api.node<TCommentText>({
-    match: n => TextApi.isText(n) && n[KEYS.comment] && n[getDraftCommentKey()]
-  })?.[0];
+  return (editor.api.node({
+    match: (n: TCommentText) => TextApi.isText(n) && n[KEYS.comment] && n[getDraftCommentKey()]
+  }) as NodeEntry<TCommentText> | undefined)?.[0];
 }, []);
 
 function FloatingDiscussionContent() {
-  const editorContainerRef = useEditorContainerRef();
+  const editorContainerRef = useEditorContainerRef() as RefObject<HTMLDivElement | null>;
   const editor = useEditorRef();
   const commentApi = editor.getApi(commentPlugin);
   const suggestionApi = editor.getApi(suggestionPlugin);
@@ -382,22 +382,22 @@ function FloatingDiscussionContent() {
 
     const allSuggestionNodes = suggestionApi.suggestion
       .nodes({ at: [] })
-      .filter(([node]) => !node[getTransientSuggestionKey()]);
+      .filter(([node]: NodeEntry<TElement | TSuggestionText>) => !node[getTransientSuggestionKey()]);
 
-    const suggestionIds = new Set(
+    const suggestionIds = new Set<string>(
       allSuggestionNodes
-        .flatMap(([node]) => {
+        .flatMap(([node]: NodeEntry<TElement | TSuggestionText>) => {
           if (TextApi.isText(node)) {
             const dataList = suggestionApi.suggestion.dataList(node);
             const includeUpdate = dataList.some(
-              data => data.type === 'update'
+              (data: { type: string }) => data.type === 'update'
             );
 
             if (!includeUpdate) return suggestionApi.suggestion.nodeId(node);
 
             return dataList
-              .filter(data => data.type === 'update')
-              .map(d => d.id);
+              .filter((data: { type: string }) => data.type === 'update')
+              .map((d: { id: string }) => d.id);
           }
           if (ElementApi.isElement(node)) {
             return suggestionApi.suggestion.nodeId(node);
@@ -405,20 +405,20 @@ function FloatingDiscussionContent() {
 
           return null;
         })
-        .filter(Boolean)
+        .filter((id: string | null): id is string => typeof id === 'string' && id.length > 0)
     );
 
     suggestionIds.forEach((id) => {
       if (!id) return;
 
       const entries = [
-        ...editor.api.nodes<TElement | TSuggestionText>({
+        ...editor.api.nodes({
           at: [],
           mode: 'all',
-          match: n => (n[KEYS.suggestion] && n[getSuggestionKey(id)])
+          match: (n: TElement | TSuggestionText) => (n[KEYS.suggestion] && n[getSuggestionKey(id)])
             || suggestionApi.suggestion.nodeId(n as TElement) === id
         })
-      ];
+      ] as NodeEntry<TElement | TSuggestionText>[];
 
       suggestionEntriesMap.current[id] = entries;
     });
@@ -454,7 +454,7 @@ function FloatingDiscussionContent() {
 
       const topDistance = getCommentTop(editor, {
         node: commentLeaf,
-        relativeElement: editorContainerRef.current!
+        relativeElement: editorContainerRef.current as HTMLDivElement
       });
 
       topRef.current[discussion.id] = topDistance;
@@ -465,7 +465,7 @@ function FloatingDiscussionContent() {
 
       const topDistance = getCommentTop(editor, {
         node: entries[0][0],
-        relativeElement: editorContainerRef.current!
+        relativeElement: editorContainerRef.current as HTMLDivElement
       });
 
       topRef.current[id] = topDistance;
@@ -487,7 +487,7 @@ function FloatingDiscussionContent() {
 
     const topDistance = getCommentTop(editor, {
       node: commentingNode,
-      relativeElement: editorContainerRef.current!
+      relativeElement: editorContainerRef.current as HTMLDivElement
     });
 
     topRef.current[getDraftCommentKey()] = topDistance;
@@ -561,7 +561,7 @@ function FloatingDiscussionContent() {
       activeId,
       getCommentTop(editor, {
         node: activeNode?.[0] || activeSuggestionNode![0],
-        relativeElement: editorContainerRef.current!
+        relativeElement: editorContainerRef.current as HTMLDivElement
       })
     );
 
@@ -696,7 +696,7 @@ function FloatingCommentsContent({
     editor.setOption(suggestionPlugin, 'activeId', null);
     const leaf = editor.api.node({
       at: [],
-      match: n => TextApi.isText(n)
+      match: (n: TElement | TSuggestionText) => TextApi.isText(n)
         && n[KEYS.comment]
         && editor.getApi(commentPlugin).comment.nodeId(n) === id
     });
@@ -944,7 +944,7 @@ const FloatingSuggestionContent = ({
 
     const leaf = editor.api.node({
       at: [],
-      match: n => n[KEYS.suggestion]
+      match: (n: TElement | TSuggestionText) => n[KEYS.suggestion]
         && editor.getApi(suggestionPlugin).suggestion.nodeId(n as any) === id
     });
 

package/src/internal/plate-editor/plate-ui/floating-toolbar.tsx

@@ -2,8 +2,6 @@
 
 import { type ComponentProps } from 'react';
 
-import * as React from 'react';
-
 import { AIChatPlugin } from '@platejs/ai/react';
 import {
   type FloatingToolbarState,

package/src/internal/plate-editor/plate-ui/font-color-toolbar-button.tsx

@@ -1,7 +1,5 @@
 'use client';
 
-import * as React from 'react';
-
 import { useCallback, useState, type ComponentProps } from 'react';
 
 import { KEYS } from 'platejs';

package/src/internal/plate-editor/plate-ui/font-size-toolbar-button.tsx

@@ -2,8 +2,6 @@
 
 import { useState } from 'react';
 
-import * as React from 'react';
-
 import { type TElement, KEYS } from 'platejs';
 
 import { toUnitLess } from '@platejs/basic-styles';
@@ -60,7 +58,7 @@ export function FontSizeToolbarButton() {
       return toUnitLess(fontSize as string);
     }
 
-    const [block] = editor.api.block<TElement>() || [];
+    const [block] = (editor.api.block() as [TElement, unknown] | undefined) || [];
 
     if (!block?.type) return DEFAULT_FONT_SIZE;
 

package/src/internal/plate-editor/plate-ui/history-toolbar-button.tsx

@@ -2,8 +2,6 @@
 
 import { type ComponentProps } from 'react';
 
-import * as React from 'react';
-
 import { Redo2Icon, Undo2Icon } from 'lucide-react';
 import { useEditorRef, useEditorSelector } from 'platejs/react';
 

package/src/internal/plate-editor/plate-ui/import-toolbar-button.tsx

@@ -2,8 +2,6 @@
 
 import { useState } from 'react';
 
-import * as React from 'react';
-
 import { importDocx } from '@platejs/docx-io';
 import { MarkdownPlugin } from '@platejs/markdown';
 import { ArrowUpToLineIcon } from 'lucide-react';

package/src/internal/plate-editor/plate-ui/indent-toolbar-button.tsx

@@ -2,8 +2,6 @@
 
 import { type ComponentProps } from 'react';
 
-import * as React from 'react';
-
 import { useIndentButton, useOutdentButton } from '@platejs/indent/react';
 import { IndentIcon, OutdentIcon } from 'lucide-react';
 

package/src/internal/plate-editor/plate-ui/input.tsx

@@ -2,8 +2,6 @@
 
 import { type ComponentProps } from 'react';
 
-import * as React from 'react';
-
 import { cva, type VariantProps } from 'class-variance-authority';
 
 import { cn } from '@docyrus/ui-pro-shared/lib/utils';

package/src/internal/plate-editor/plate-ui/insert-toolbar-button.tsx

@@ -2,8 +2,6 @@
 
 import { useState, type ReactNode } from 'react';
 
-import * as React from 'react';
-
 import {
   CalendarIcon,
   ChevronRightIcon,

package/src/internal/plate-editor/plate-ui/line-height-toolbar-button.tsx

@@ -2,8 +2,6 @@
 
 import { useState } from 'react';
 
-import * as React from 'react';
-
 import { LineHeightPlugin } from '@platejs/basic-styles/react';
 import { CheckIcon, WrapText } from 'lucide-react';
 import { useEditorRef, useSelectionFragmentProp } from 'platejs/react';

package/src/internal/plate-editor/plate-ui/link-toolbar-button.tsx

@@ -2,8 +2,6 @@
 
 import { type ComponentProps } from 'react';
 
-import * as React from 'react';
-
 import {
   useLinkToolbarButton,
   useLinkToolbarButtonState

package/src/internal/plate-editor/plate-ui/link-toolbar.tsx

@@ -2,8 +2,6 @@
 
 import { useMemo } from 'react';
 
-import * as React from 'react';
-
 import {
   type UseVirtualFloatingOptions,
   flip,

package/src/internal/plate-editor/plate-ui/list-toolbar-button.tsx

@@ -2,8 +2,6 @@
 
 import { useState, type ComponentProps } from 'react';
 
-import * as React from 'react';
-
 import { ListStyleType, someList, toggleList } from '@platejs/list';
 import {
   useIndentTodoToolBarButton,

package/src/internal/plate-editor/plate-ui/mark-toolbar-button.tsx

@@ -2,8 +2,6 @@
 
 import { type ComponentProps } from 'react';
 
-import * as React from 'react';
-
 import {
   useEditorRef,
   useMarkToolbarButton,

package/src/internal/plate-editor/plate-ui/media-toolbar-button.tsx

@@ -2,8 +2,6 @@
 
 import { useCallback, useState, type ReactNode } from 'react';
 
-import * as React from 'react';
-
 import { PlaceholderPlugin } from '@platejs/media/react';
 import {
   AudioLinesIcon,

package/src/internal/plate-editor/plate-ui/media-toolbar.tsx

@@ -2,8 +2,6 @@
 
 import { type ComponentProps, type Dispatch, type SetStateAction } from 'react';
 
-import * as React from 'react';
-
 import { type TMediaElement, type TTextAlignProps, KEYS } from 'platejs';
 
 import { showCaption } from '@platejs/caption/react';

package/src/internal/plate-editor/plate-ui/menu.tsx

@@ -1,7 +1,7 @@
 'use client';
 
 import {
-  createContext, use, useEffect, useRef, useState, type ComponentProps, type MouseEvent, type ReactNode
+  createContext, use, useEffect, useRef, useState, type ComponentProps, type ReactNode
 } from 'react';
 
 import * as Ariakit from '@ariakit/react';
@@ -147,7 +147,7 @@ const menuVariants = cva(
 export type MenuContentProps = Ariakit.MenuProps
   & VariantProps<typeof menuVariants> & {
     animateZoom?: boolean;
-    onClickOutside?: (event: MouseEvent) => void;
+    onClickOutside?: (event: globalThis.MouseEvent) => void;
   };
 
 export function MenuContent({
@@ -352,7 +352,7 @@ export function MenuItem({
     return <Ariakit.MenuItem {...baseProps} />;
   }
 
-  const hideOnClick = (event: MouseEvent<HTMLElement, MouseEvent>) => {
+  const hideOnClick: Ariakit.ComboboxItemProps['hideOnClick'] = (event) => {
     const expandable = event.currentTarget.hasAttribute('aria-expanded');
 
     if (expandable) return false;
@@ -485,11 +485,11 @@ export function filterMenuGroups(
   }, []);
 }
 
-export function useComboboxValueState() {
+export function useComboboxValueState(): readonly [string, (value: string) => void] {
   const store = Ariakit.useComboboxContext();
   const searchValue = store?.useState('value') ?? '';
 
-  return [searchValue, store!.setValue] as const;
+  return [searchValue, store!.setValue as (value: string) => void] as const;
 }
 
 export function useMenuSide() {
@@ -499,7 +499,11 @@ export function useMenuSide() {
   return currentPlacement;
 }
 
-export function useContextMenu(anchorRect: { x: number; y: number }) {
+export function useContextMenu(anchorRect: { x: number; y: number }): {
+  store: any;
+  getAnchorRect: () => { x: number; y: number };
+  show: () => void;
+} {
   const menu = Ariakit.useMenuStore();
 
   useEffect(() => {
@@ -509,7 +513,7 @@ export function useContextMenu(anchorRect: { x: number; y: number }) {
   }, [anchorRect, menu]);
 
   return {
-    store: menu,
+    store: menu as any,
     getAnchorRect: () => anchorRect,
     show: () => {
       menu.show();
@@ -518,11 +522,14 @@ export function useContextMenu(anchorRect: { x: number; y: number }) {
   };
 }
 
-export function useMenuStore() {
+export function useMenuStore(): {
+  store: any;
+  show: (anchorElement: HTMLElement) => void;
+} {
   const menu = Ariakit.useMenuStore();
 
   return {
-    store: menu,
+    store: menu as any,
     show: (anchorElement: HTMLElement) => {
       menu.setAnchorElement(anchorElement);
       menu.show();

package/src/internal/plate-editor/plate-ui/mode-toolbar-button.tsx

@@ -2,8 +2,6 @@
 
 import { useState, type ReactNode } from 'react';
 
-import * as React from 'react';
-
 import { SuggestionPlugin } from '@platejs/suggestion/react';
 import {
   CheckIcon, EyeIcon, PencilLineIcon, PenIcon

package/src/internal/plate-editor/plate-ui/popover.tsx

@@ -2,8 +2,6 @@
 
 import { type ComponentProps } from 'react';
 
-import * as React from 'react';
-
 import { cva, type VariantProps } from 'class-variance-authority';
 import { Popover as PopoverPrimitive } from 'radix-ui';
 

package/src/internal/plate-editor/plate-ui/remote-cursor-overlay.tsx

@@ -2,8 +2,6 @@
 
 import { useMemo } from 'react';
 
-import * as React from 'react';
-
 import {
   type CursorData,
   type CursorOverlayState,

package/src/internal/plate-editor/plate-ui/resize-handle.tsx

@@ -2,8 +2,6 @@
 
 import { type ComponentProps } from 'react';
 
-import * as React from 'react';
-
 import {
   Resizable as ResizablePrimitive,
   type ResizeHandle as ResizeHandlePrimitive,

package/src/internal/plate-editor/plate-ui/separator.tsx

@@ -2,8 +2,6 @@
 
 import { type ComponentProps } from 'react';
 
-import * as React from 'react';
-
 import { cva, type VariantProps } from 'class-variance-authority';
 import { Separator as SeparatorPrimitive } from 'radix-ui';
 

package/src/internal/plate-editor/plate-ui/slash-node.tsx

@@ -2,8 +2,6 @@
 
 import { type ReactNode } from 'react';
 
-import * as React from 'react';
-
 import { AIChatPlugin } from '@platejs/ai/react';
 import { BlockSelectionPlugin } from '@platejs/selection/react';
 import {

package/src/internal/plate-editor/plate-ui/suggestion-node-static.tsx

@@ -13,7 +13,7 @@ export function SuggestionLeafStatic(props: SlateLeafProps<TSuggestionText>) {
   const dataList = editor
     .getApi(BaseSuggestionPlugin)
     .suggestion.dataList(leaf);
-  const hasRemove = dataList.some(data => data.type === 'remove');
+  const hasRemove = dataList.some((data: { type: string }) => data.type === 'remove');
   const diffOperation = { type: hasRemove ? 'delete' : 'insert' } as const;
 
   const Component = ({ delete: 'del', insert: 'ins', update: 'span' } as const)[

package/src/internal/plate-editor/plate-ui/suggestion-node.tsx

@@ -2,8 +2,6 @@
 
 import { useRef, type ReactNode } from 'react';
 
-import * as React from 'react';
-
 import { type TSuggestionData, type TSuggestionText } from 'platejs';
 
 import { cva } from 'class-variance-authority';

package/src/internal/plate-editor/plate-ui/table-node-static.tsx

@@ -1,5 +1,4 @@
 import { type CSSProperties } from 'react';
-import * as React from 'react';
 
 import { type TTableCellElement, type TTableElement } from 'platejs';
 

package/src/internal/plate-editor/plate-ui/table-node.tsx

@@ -2,8 +2,6 @@
 
 import { type CSSProperties, type ComponentProps } from 'react';
 
-import * as React from 'react';
-
 import { ResizeHandle } from '@platejs/resizable';
 import { useBlockSelected } from '@platejs/selection/react';
 import {

package/src/internal/plate-editor/plate-ui/table-toolbar-button.tsx

@@ -2,8 +2,6 @@
 
 import { useState } from 'react';
 
-import * as React from 'react';
-
 import { TablePlugin, useTableMergeState } from '@platejs/table/react';
 import {
   ArrowDown,

package/src/internal/plate-editor/plate-ui/textarea.tsx

@@ -1,5 +1,4 @@
 import { type ComponentProps } from 'react';
-import * as React from 'react';
 
 import { TextareaAutosize as ReactTextareaAutosize } from '@platejs/caption/react';
 import { cva, type VariantProps } from 'class-variance-authority';

package/src/internal/plate-editor/plate-ui/toc-node-static.tsx

@@ -70,8 +70,8 @@ const getHeadingList = (editor?: SlateEditor) => {
 
   const values = editor.api.nodes({
     at: [],
-    match: n => isHeading(n)
-  });
+    match: (n: TElement) => isHeading(n)
+  }) as Array<[TElement, any]>;
 
   if (!values) return [];
 

package/src/internal/plate-editor/plate-ui/toc-sidebar.tsx

@@ -1,7 +1,5 @@
 'use client';
 
-import * as React from 'react';
-
 import { useTocSideBar, useTocSideBarState } from '@platejs/toc/react';
 import { cva } from 'class-variance-authority';
 

package/src/internal/plate-editor/plate-ui/toggle-toolbar-button.tsx

@@ -2,8 +2,6 @@
 
 import { type ComponentProps } from 'react';
 
-import * as React from 'react';
-
 import {
   useToggleToolbarButton,
   useToggleToolbarButtonState

package/src/internal/plate-editor/plate-ui/toolbar.tsx

@@ -4,8 +4,6 @@ import {
   useEffect, useState, type ComponentProps, type ComponentPropsWithoutRef, type ElementType, type ReactNode
 } from 'react';
 
-import * as React from 'react';
-
 import { type VariantProps, cva } from 'class-variance-authority';
 import { ChevronDown } from 'lucide-react';
 import { Toolbar as ToolbarPrimitive, Tooltip as TooltipPrimitive } from 'radix-ui';

package/src/internal/plate-editor/plate-ui/tooltip.tsx

@@ -1,6 +1,5 @@
 'use client';
 
-import * as React from 'react';
 import {
   useSyncExternalStore, type ComponentProps, type ComponentPropsWithoutRef, type ElementType, type ReactNode
 } from 'react';

package/src/internal/plate-editor/plate-ui/turn-into-toolbar-button.tsx

@@ -2,8 +2,6 @@
 
 import { useMemo } from 'react';
 
-import * as React from 'react';
-
 import {
   ChevronDownIcon,
   Code2Icon,

package/src/lib/assistant-config.tsx

@@ -6,6 +6,8 @@ import {
 
 import { Toaster } from 'sonner';
 
+import { TooltipProvider } from '@docyrus/ui-pro-shared/components/tooltip';
+
 import { AssistantI18nProvider } from '../i18n';
 
 // User interface for assistant
@@ -54,8 +56,10 @@ export function AssistantProvider({ config, children }: AssistantProviderProps)
 
   return (
     <AssistantConfigContext value={value}>
-      <AssistantI18nProvider>{children}</AssistantI18nProvider>
-      <Toaster />
+      <TooltipProvider>
+        <AssistantI18nProvider>{children}</AssistantI18nProvider>
+        <Toaster />
+      </TooltipProvider>
     </AssistantConfigContext>
   );
 }