|
| 1 | +import React, { useRef, useEffect, useState, useCallback } from "react"; |
| 2 | +import { AbstractInputSuggest, App } from "obsidian"; |
| 3 | +import { usePlugin } from "./PluginContext"; |
| 4 | + |
| 5 | +class GenericSuggestInput<T> extends AbstractInputSuggest<T> { |
| 6 | + private getSuggestionsFn: (query: string) => T[]; |
| 7 | + private onSelectCallback: (item: T) => void; |
| 8 | + private getDisplayTextFn: (item: T) => string; |
| 9 | + private renderItemFn?: (item: T, el: HTMLElement) => void; |
| 10 | + |
| 11 | + constructor( |
| 12 | + app: App, |
| 13 | + private textInputEl: HTMLInputElement, |
| 14 | + config: { |
| 15 | + getSuggestions: (query: string) => T[]; |
| 16 | + onSelect: (item: T) => void; |
| 17 | + getDisplayText: (item: T) => string; |
| 18 | + renderItem?: (item: T, el: HTMLElement) => void; |
| 19 | + }, |
| 20 | + ) { |
| 21 | + super(app, textInputEl); |
| 22 | + this.getSuggestionsFn = config.getSuggestions; |
| 23 | + this.onSelectCallback = config.onSelect; |
| 24 | + this.getDisplayTextFn = config.getDisplayText; |
| 25 | + this.renderItemFn = config.renderItem; |
| 26 | + } |
| 27 | + |
| 28 | + getSuggestions(inputStr: string): T[] { |
| 29 | + return this.getSuggestionsFn(inputStr); |
| 30 | + } |
| 31 | + |
| 32 | + renderSuggestion(item: T, el: HTMLElement): void { |
| 33 | + if (this.renderItemFn) { |
| 34 | + this.renderItemFn(item, el); |
| 35 | + } else { |
| 36 | + el.createDiv({ |
| 37 | + text: this.getDisplayTextFn(item), |
| 38 | + cls: "suggestion-item", |
| 39 | + }); |
| 40 | + } |
| 41 | + } |
| 42 | + |
| 43 | + selectSuggestion(item: T, evt: MouseEvent | KeyboardEvent): void { |
| 44 | + this.textInputEl.value = this.getDisplayTextFn(item); |
| 45 | + this.onSelectCallback(item); |
| 46 | + this.close(); |
| 47 | + } |
| 48 | +} |
| 49 | + |
| 50 | +type SuggestInputProps<T> = { |
| 51 | + value: string; |
| 52 | + onChange: (value: string) => void; |
| 53 | + getSuggestions: (query: string) => T[]; |
| 54 | + getDisplayText: (item: T) => string; |
| 55 | + onSelect?: (item: T) => void; |
| 56 | + renderItem?: (item: T, el: HTMLElement) => void; |
| 57 | + placeholder?: string; |
| 58 | + className?: string; |
| 59 | + disabled?: boolean; |
| 60 | +}; |
| 61 | + |
| 62 | +const SuggestInput = <T,>({ |
| 63 | + value, |
| 64 | + onChange, |
| 65 | + getSuggestions, |
| 66 | + getDisplayText, |
| 67 | + onSelect, |
| 68 | + renderItem, |
| 69 | + placeholder = "Enter value", |
| 70 | + className = "", |
| 71 | + disabled = false, |
| 72 | +}: SuggestInputProps<T>) => { |
| 73 | + const plugin = usePlugin(); |
| 74 | + const inputRef = useRef<HTMLInputElement>(null); |
| 75 | + const [suggest, setSuggest] = useState<GenericSuggestInput<T> | null>(null); |
| 76 | + |
| 77 | + const handleSelect = useCallback( |
| 78 | + (item: T) => { |
| 79 | + const displayText = getDisplayText(item); |
| 80 | + onChange(displayText); |
| 81 | + onSelect?.(item); |
| 82 | + }, |
| 83 | + [getDisplayText, onChange, onSelect], |
| 84 | + ); |
| 85 | + |
| 86 | + useEffect(() => { |
| 87 | + if (inputRef.current && !suggest && !disabled) { |
| 88 | + const genericSuggest = new GenericSuggestInput<T>( |
| 89 | + plugin.app, |
| 90 | + inputRef.current, |
| 91 | + { |
| 92 | + getSuggestions, |
| 93 | + onSelect: handleSelect, |
| 94 | + getDisplayText, |
| 95 | + renderItem, |
| 96 | + }, |
| 97 | + ); |
| 98 | + setSuggest(genericSuggest); |
| 99 | + |
| 100 | + return () => { |
| 101 | + genericSuggest.close(); |
| 102 | + setSuggest(null); |
| 103 | + }; |
| 104 | + } |
| 105 | + }, [ |
| 106 | + plugin.app, |
| 107 | + getSuggestions, |
| 108 | + getDisplayText, |
| 109 | + renderItem, |
| 110 | + disabled, |
| 111 | + handleSelect, |
| 112 | + ]); |
| 113 | + |
| 114 | + useEffect(() => { |
| 115 | + if (inputRef.current && inputRef.current.value !== value) { |
| 116 | + inputRef.current.value = value; |
| 117 | + } |
| 118 | + }, [value]); |
| 119 | + |
| 120 | + const handleChange = (e: React.ChangeEvent<HTMLInputElement>) => { |
| 121 | + onChange(e.target.value); |
| 122 | + }; |
| 123 | + |
| 124 | + return ( |
| 125 | + <input |
| 126 | + ref={inputRef} |
| 127 | + type="text" |
| 128 | + value={value} |
| 129 | + onChange={handleChange} |
| 130 | + placeholder={placeholder} |
| 131 | + className={className} |
| 132 | + disabled={disabled} |
| 133 | + autoComplete="off" |
| 134 | + /> |
| 135 | + ); |
| 136 | +}; |
| 137 | + |
| 138 | +export default SuggestInput; |
0 commit comments