feat: add image-select modal (#49)

2025-04-24 23:18:15 +08:00
parent 1b43bac6a0
commit 6db44f5ea0
3 changed files with 212 additions and 14 deletions
--- a/src/components/modals/ImageSelectorModal.tsx
+++ b/src/components/modals/ImageSelectorModal.tsx
@@ -0,0 +1,114 @@
+import { App, Modal, TFile } from 'obsidian'
+import React, { useEffect, useState } from 'react'
+import { createRoot } from 'react-dom/client'
+
+type ImageSelectorProps = {
+  onClose: () => void
+  onSelectImages: (files: File[]) => void
+  onSelectVaultImages: (files: TFile[]) => void
+  app: App
+}
+
+const ImageSelector: React.FC<ImageSelectorProps> = ({
+  onClose,
+  onSelectImages,
+  onSelectVaultImages,
+  app,
+}) => {
+  const [searchTerm, setSearchTerm] = useState('')
+  const [vaultImages, setVaultImages] = useState<TFile[]>([])
+  
+  useEffect(() => {
+    const images = app.vault.getFiles()
+      .filter((file) => file.extension.match(/png|jpg|jpeg|gif|svg/i))
+      .filter((file) => 
+        searchTerm ? file.path.toLowerCase().includes(searchTerm.toLowerCase()) : true
+      )
+    setVaultImages(images)
+  }, [searchTerm, app.vault])
+
+  const handleFileUpload = (event: React.ChangeEvent<HTMLInputElement>) => {
+    const files = Array.from(event.target.files ?? [])
+    if (files.length > 0) {
+      onSelectImages(files)
+      onClose()
+    }
+  }
+
+  return (
+    <div className="infio-image-selector">
+      <div className="infio-image-selector-header">
+        <input
+          type="text"
+          placeholder="Search images in vault..."
+          value={searchTerm}
+          onChange={(e) => setSearchTerm(e.target.value)}
+          className="infio-image-search"
+        />
+        <label className="infio-upload-button">
+          Upload New Image
+          <input
+            type="file"
+            accept="image/*"
+            multiple
+            onChange={handleFileUpload}
+            style={{ display: 'none' }}
+          />
+        </label>
+      </div>
+      
+      <div className="infio-image-grid">
+        {vaultImages.map((file) => (
+          <div 
+            key={file.path}
+            className="infio-image-item"
+            onClick={() => {
+              onSelectVaultImages([file])
+              onClose()
+            }}
+          >
+            <img 
+              src={app.vault.adapter.getResourcePath(file.path)} 
+              alt={file.name} 
+            />
+            <div className="infio-image-name">{file.name}</div>
+          </div>
+        ))}
+      </div>
+    </div>
+  )
+}
+
+export class ImageSelectorModal extends Modal {
+  private readonly onSelectImages: (files: File[]) => void
+  private readonly onSelectVaultImages: (files: TFile[]) => void
+
+  constructor(
+    app: App,
+    onSelectImages: (files: File[]) => void,
+    onSelectVaultImages: (files: TFile[]) => void,
+  ) {
+    super(app)
+    this.onSelectImages = onSelectImages
+    this.onSelectVaultImages = onSelectVaultImages
+  }
+
+  onOpen(): void {
+    const { contentEl } = this
+    const root = createRoot(contentEl)
+    
+    root.render(
+      <ImageSelector
+        onClose={() => this.close()}
+        onSelectImages={this.onSelectImages}
+        onSelectVaultImages={this.onSelectVaultImages}
+        app={this.app}
+      />
+    )
+  }
+
+  onClose(): void {
+    const { contentEl } = this
+    contentEl.empty()
+  }
+}