feat: add image support in chat

src/components/chat-view/chat-input/ChatUserInput.tsx

@@ -2,15 +2,22 @@ import { useQuery } from '@tanstack/react-query'
 import { $nodesOfType, LexicalEditor, SerializedEditorState } from 'lexical'
 import {
   forwardRef,
+  useCallback,
   useEffect,
   useImperativeHandle,
+  useMemo,
   useRef,
   useState,
 } from 'react'
 
 import { useApp } from '../../../contexts/app-context'
 import { useDarkModeContext } from '../../../contexts/dark-mode-context'
-import { Mentionable, SerializedMentionable } from '../../../types/mentionable'
+import {
+  Mentionable,
+  MentionableImage,
+  SerializedMentionable,
+} from '../../../types/mentionable'
+import { fileToMentionableImage } from '../../../utils/image'
 import {
   deserializeMentionable,
   getMentionableKey,
@@ -19,6 +26,7 @@ import {
 import { openMarkdownFile, readTFileContent } from '../../../utils/obsidian'
 import { MemoizedSyntaxHighlighterWrapper } from '../SyntaxHighlighterWrapper'
 
+import { ImageUploadButton } from './ImageUploadButton'
 import LexicalContentEditable from './LexicalContentEditable'
 import MentionableBadge from './MentionableBadge'
 import { ModelSelect } from './ModelSelect'
@@ -57,7 +65,6 @@ const ChatUserInput = forwardRef<ChatUserInputRef, ChatUserInputProps>(
     ref,
   ) => {
     const app = useApp()
-    const { isDarkMode } = useDarkModeContext()
 
     const editorRef = useRef<LexicalEditor | null>(null)
     const contentEditableRef = useRef<HTMLDivElement>(null)
@@ -139,6 +146,29 @@ const ChatUserInput = forwardRef<ChatUserInputRef, ChatUserInputProps>(
       }
     }
 
+    const handleCreateImageMentionables = useCallback(
+      (mentionableImages: MentionableImage[]) => {
+        const newMentionableImages = mentionableImages.filter(
+          (m) =>
+            !mentionables.some(
+              (mentionable) =>
+                getMentionableKey(serializeMentionable(mentionable)) ===
+                getMentionableKey(serializeMentionable(m)),
+            ),
+        )
+        if (newMentionableImages.length === 0) return
+        setMentionables([...mentionables, ...newMentionableImages])
+        setDisplayedMentionableKey(
+          getMentionableKey(
+            serializeMentionable(
+              newMentionableImages[newMentionableImages.length - 1],
+            ),
+          ),
+        )
+      },
+      [mentionables, setMentionables],
+    )
+
     const handleMentionableDelete = (mentionable: Mentionable) => {
       const mentionableKey = getMentionableKey(
         serializeMentionable(mentionable),
@@ -158,47 +188,12 @@ const ChatUserInput = forwardRef<ChatUserInputRef, ChatUserInputProps>(
       })
     }
 
-    const { data: fileContent } = useQuery({
-      queryKey: [
-        'file',
-        displayedMentionableKey,
-        mentionables.map((m) => getMentionableKey(serializeMentionable(m))), // should be updated when mentionables change (especially on delete)
-      ],
-      queryFn: async () => {
-        if (!displayedMentionableKey) return null
-
-        const displayedMentionable = mentionables.find(
-          (m) =>
-            getMentionableKey(serializeMentionable(m)) ===
-            displayedMentionableKey,
-        )
-
-        if (!displayedMentionable) return null
-
-        if (
-          displayedMentionable.type === 'file' ||
-          displayedMentionable.type === 'current-file'
-        ) {
-          if (!displayedMentionable.file) return null
-          return await readTFileContent(displayedMentionable.file, app.vault)
-        } else if (displayedMentionable.type === 'block') {
-          const fileContent = await readTFileContent(
-            displayedMentionable.file,
-            app.vault,
-          )
-
-          return fileContent
-            .split('\n')
-            .slice(
-              displayedMentionable.startLine - 1,
-              displayedMentionable.endLine,
-            )
-            .join('\n')
-        }
-
-        return null
-      },
-    })
+    const handleUploadImages = async (images: File[]) => {
+      const mentionableImages = await Promise.all(
+        images.map((image) => fileToMentionableImage(image)),
+      )
+      handleCreateImageMentionables(mentionableImages)
+    }
 
     const handleSubmit = (options: { useVaultSearch?: boolean } = {}) => {
       const content = editorRef.current?.getEditorState()?.toJSON()
@@ -235,23 +230,19 @@ const ChatUserInput = forwardRef<ChatUserInputRef, ChatUserInputProps>(
                     setDisplayedMentionableKey(mentionableKey)
                   }
                 }}
+                isFocused={
+                  getMentionableKey(serializeMentionable(m)) ===
+                  displayedMentionableKey
+                }
               />
             ))}
           </div>
         )}
 
-        {fileContent && (
-          <div className="smtcmp-chat-user-input-file-content-preview">
-            <MemoizedSyntaxHighlighterWrapper
-              isDarkMode={isDarkMode}
-              language="markdown"
-              hasFilename={false}
-              wrapLines={false}
-            >
-              {fileContent}
-            </MemoizedSyntaxHighlighterWrapper>
-          </div>
-        )}
+        <MentionableContentPreview
+          displayedMentionableKey={displayedMentionableKey}
+          mentionables={mentionables}
+        />
 
         <LexicalContentEditable
           initialEditorState={(editor) => {
@@ -267,6 +258,7 @@ const ChatUserInput = forwardRef<ChatUserInputRef, ChatUserInputProps>(
           onEnter={() => handleSubmit({ useVaultSearch: false })}
           onFocus={onFocus}
           onMentionNodeMutation={handleMentionNodeMutation}
+          onCreateImageMentionables={handleCreateImageMentionables}
           autoFocus={autoFocus}
           plugins={{
             onEnter: {
@@ -281,8 +273,11 @@ const ChatUserInput = forwardRef<ChatUserInputRef, ChatUserInputProps>(
         />
 
         <div className="smtcmp-chat-user-input-controls">
-          <ModelSelect />
-          <div className="smtcmp-chat-user-input-controls-buttons">
+          <div className="smtcmp-chat-user-input-controls__model-select-container">
+            <ModelSelect />
+          </div>
+          <div className="smtcmp-chat-user-input-controls__buttons">
+            <ImageUploadButton onUpload={handleUploadImages} />
             <SubmitButton onClick={() => handleSubmit()} />
             <VaultChatButton
               onClick={() => {
@@ -296,6 +291,84 @@ const ChatUserInput = forwardRef<ChatUserInputRef, ChatUserInputProps>(
   },
 )
 
+function MentionableContentPreview({
+  displayedMentionableKey,
+  mentionables,
+}: {
+  displayedMentionableKey: string | null
+  mentionables: Mentionable[]
+}) {
+  const app = useApp()
+  const { isDarkMode } = useDarkModeContext()
+
+  const displayedMentionable: Mentionable | null = useMemo(() => {
+    return (
+      mentionables.find(
+        (m) =>
+          getMentionableKey(serializeMentionable(m)) ===
+          displayedMentionableKey,
+      ) ?? null
+    )
+  }, [displayedMentionableKey, mentionables])
+
+  const { data: displayFileContent } = useQuery({
+    enabled:
+      !!displayedMentionable &&
+      ['file', 'current-file', 'block'].includes(displayedMentionable.type),
+    queryKey: [
+      'file',
+      displayedMentionableKey,
+      mentionables.map((m) => getMentionableKey(serializeMentionable(m))), // should be updated when mentionables change (especially on delete)
+    ],
+    queryFn: async () => {
+      if (!displayedMentionable) return null
+      if (
+        displayedMentionable.type === 'file' ||
+        displayedMentionable.type === 'current-file'
+      ) {
+        if (!displayedMentionable.file) return null
+        return await readTFileContent(displayedMentionable.file, app.vault)
+      } else if (displayedMentionable.type === 'block') {
+        const fileContent = await readTFileContent(
+          displayedMentionable.file,
+          app.vault,
+        )
+
+        return fileContent
+          .split('\n')
+          .slice(
+            displayedMentionable.startLine - 1,
+            displayedMentionable.endLine,
+          )
+          .join('\n')
+      }
+
+      return null
+    },
+  })
+
+  const displayImage: MentionableImage | null = useMemo(() => {
+    return displayedMentionable?.type === 'image' ? displayedMentionable : null
+  }, [displayedMentionable])
+
+  return displayFileContent ? (
+    <div className="smtcmp-chat-user-input-file-content-preview">
+      <MemoizedSyntaxHighlighterWrapper
+        isDarkMode={isDarkMode}
+        language="markdown"
+        hasFilename={false}
+        wrapLines={false}
+      >
+        {displayFileContent}
+      </MemoizedSyntaxHighlighterWrapper>
+    </div>
+  ) : displayImage ? (
+    <div className="smtcmp-chat-user-input-file-content-preview">
+      <img src={displayImage.data} alt={displayImage.name} />
+    </div>
+  ) : null
+}
+
 ChatUserInput.displayName = 'ChatUserInput'
 
 export default ChatUserInput

src/components/chat-view/chat-input/ImageUploadButton.tsx

@@ -0,0 +1,30 @@
+import { ImageIcon } from 'lucide-react'
+
+export function ImageUploadButton({
+  onUpload,
+}: {
+  onUpload: (files: File[]) => void
+}) {
+  const handleFileChange = (event: React.ChangeEvent<HTMLInputElement>) => {
+    const files = Array.from(event.target.files ?? [])
+    if (files.length > 0) {
+      onUpload(files)
+    }
+  }
+
+  return (
+    <label className="smtcmp-chat-user-input-submit-button">
+      <input
+        type="file"
+        accept="image/*"
+        multiple
+        onChange={handleFileChange}
+        style={{ display: 'none' }}
+      />
+      <div className="smtcmp-chat-user-input-submit-button-icons">
+        <ImageIcon size={12} />
+      </div>
+      <div>Image</div>
+    </label>
+  )
+}

src/components/chat-view/chat-input/LexicalContentEditable.tsx

@@ -13,8 +13,11 @@ import { LexicalEditor, SerializedEditorState } from 'lexical'
 import { RefObject, useCallback, useEffect } from 'react'
 
 import { useApp } from '../../../contexts/app-context'
+import { MentionableImage } from '../../../types/mentionable'
 import { fuzzySearch } from '../../../utils/fuzzy-search'
 
+import DragDropPaste from './plugins/image/DragDropPastePlugin'
+import ImagePastePlugin from './plugins/image/ImagePastePlugin'
 import AutoLinkMentionPlugin from './plugins/mention/AutoLinkMentionPlugin'
 import { MentionNode } from './plugins/mention/MentionNode'
 import MentionPlugin from './plugins/mention/MentionPlugin'
@@ -33,6 +36,7 @@ export type LexicalContentEditableProps = {
   onEnter?: (evt: KeyboardEvent) => void
   onFocus?: () => void
   onMentionNodeMutation?: (mutations: NodeMutations<MentionNode>) => void
+  onCreateImageMentionables?: (mentionables: MentionableImage[]) => void
   initialEditorState?: InitialEditorStateType
   autoFocus?: boolean
   plugins?: {
@@ -52,6 +56,7 @@ export default function LexicalContentEditable({
   onEnter,
   onFocus,
   onMentionNodeMutation,
+  onCreateImageMentionables,
   initialEditorState,
   autoFocus = false,
   plugins,
@@ -134,6 +139,8 @@ export default function LexicalContentEditable({
       <EditorRefPlugin editorRef={editorRef} />
       <NoFormatPlugin />
       <AutoLinkMentionPlugin />
+      <ImagePastePlugin onCreateImageMentionables={onCreateImageMentionables} />
+      <DragDropPaste onCreateImageMentionables={onCreateImageMentionables} />
       <TemplatePlugin />
       {plugins?.templatePopover && (
         <CreateTemplatePopoverPlugin