329 lines
14 KiB
TypeScript
329 lines
14 KiB
TypeScript
import type { ClipboardEvent, DragEvent, KeyboardEvent, ReactElement } from 'react'
|
||
import type { Ask } from './qa'
|
||
import type { PendingImage } from './qaImages'
|
||
import type { RewriteRun } from './rewriteRun'
|
||
import type { ProfileChoice, QaEntryView } from './types'
|
||
import { ChevronDown, ChevronRight, CircleStop, LoaderCircle, SendHorizontal, Sparkles, Trash2, TriangleAlert, X } from 'lucide-react'
|
||
import { useEffect, useRef, useState } from 'react'
|
||
import { createPortal } from 'react-dom'
|
||
import { SelectMenu } from '@/components/ui/select-menu'
|
||
import { renderMarkdown, renderMarkdownOnce } from './markdown'
|
||
import { answerPreview, relativeTime } from './qa'
|
||
import { MAX_IMAGES, pickImages, readImage, transferFiles } from './qaImages'
|
||
|
||
/** 问题带的图片：一排小缩略图，点一张看大图；给了 onRemove 的（贴了还没发的）右上角有「去掉」。没有图片时什么都不画。 */
|
||
function Thumbs({ urls, pending, onOpen, onRemove }: { urls: string[], pending?: boolean, onOpen: (url: string) => void, onRemove?: (index: number) => void }): ReactElement | null {
|
||
if (urls.length === 0)
|
||
return null
|
||
return (
|
||
<div className="cr-qa-thumbs" data-pending={pending ? '' : undefined}>
|
||
{urls.map((url, index) => (
|
||
// 同一张图可能贴两次、地址一样；缩略图没有自己的状态，按位置认就行
|
||
<span key={index} className="cr-qa-thumb">
|
||
<button type="button" className="cr-qa-thumb-open" data-tooltip="看大图" aria-label={`看第 ${index + 1} 张图片的大图`} onClick={() => onOpen(url)}>
|
||
<img src={url} alt="" />
|
||
</button>
|
||
{onRemove && (
|
||
<button type="button" className="cr-qa-thumb-remove" data-tooltip="去掉这张图片" aria-label={`去掉第 ${index + 1} 张图片`} onClick={() => onRemove(index)}>
|
||
<X size={10} strokeWidth={2.4} aria-hidden />
|
||
</button>
|
||
)}
|
||
</span>
|
||
))}
|
||
</div>
|
||
)
|
||
}
|
||
|
||
/** 看大图：盖住整个预览，点哪儿或按 Esc 关上；打开时焦点在「关闭」上，关上后回到点开它的地方。 */
|
||
function ImagePreview({ url, onClose }: { url: string, onClose: () => void }): ReactElement {
|
||
const close = useRef<HTMLButtonElement>(null)
|
||
useEffect(() => {
|
||
const opener = document.activeElement
|
||
close.current?.focus()
|
||
return () => {
|
||
if (opener instanceof HTMLElement)
|
||
opener.focus()
|
||
}
|
||
}, [])
|
||
const onKeyDown = (event: KeyboardEvent<HTMLDivElement>): void => {
|
||
if (event.key === 'Escape') {
|
||
// 只关大图：不收起问答区，也不清名字高亮（那边认 defaultPrevented）
|
||
event.preventDefault()
|
||
event.stopPropagation()
|
||
onClose()
|
||
}
|
||
}
|
||
return createPortal(
|
||
<div className="cr-qa-lightbox" role="dialog" aria-modal="true" aria-label="图片大图" onClick={onClose} onKeyDown={onKeyDown}>
|
||
<img src={url} alt="" />
|
||
<button ref={close} type="button" className="cr-qa-lightbox-close" data-tooltip="关闭（Esc）" aria-label="关闭大图">
|
||
<X size={16} strokeWidth={1.8} aria-hidden />
|
||
</button>
|
||
</div>,
|
||
document.body,
|
||
)
|
||
}
|
||
|
||
/** 问过的一次，灰字：问题、时间、「已理解，删除」，问题带的图片，下面回答的前两行；点问题或预览展开整段回答（照样灰字，按 Markdown 排） */
|
||
function HistoryItem({ entry, now, onDelete, onOpenImage }: { entry: QaEntryView, now: number, onDelete: () => void, onOpenImage: (url: string) => void }): ReactElement {
|
||
const [expanded, setExpanded] = useState(false)
|
||
const toggle = (): void => setExpanded(value => !value)
|
||
return (
|
||
<div className="cr-qa-entry" data-history="">
|
||
<div className="cr-qa-head">
|
||
<Sparkles size={12} strokeWidth={1.8} className="cr-qa-icon" aria-hidden />
|
||
<button type="button" className="cr-qa-question" aria-expanded={expanded} data-tooltip={expanded ? '收起回答' : '展开整段回答'} onClick={toggle}>
|
||
{entry.question}
|
||
</button>
|
||
<span className="cr-qa-when">
|
||
<time dateTime={entry.createdAt} data-tooltip={new Date(entry.createdAt).toLocaleString()}>
|
||
{relativeTime(entry.createdAt, now)}
|
||
</time>
|
||
{entry.profile && <span data-tooltip="用这个 Claude Code 配置（profile）答的">{` · ${entry.profile}`}</span>}
|
||
</span>
|
||
<button type="button" className="cr-tool cr-qa-delete" data-tooltip="已理解，删除" aria-label="已理解，删除" onClick={onDelete}>
|
||
<Trash2 size={13} strokeWidth={1.8} aria-hidden />
|
||
</button>
|
||
</div>
|
||
<Thumbs urls={entry.images ?? []} onOpen={onOpenImage} />
|
||
{expanded
|
||
? <div className="cr-markdown cr-qa-answer" dangerouslySetInnerHTML={{ __html: renderMarkdown(entry.answer) }} />
|
||
: (
|
||
<button type="button" className="cr-qa-preview" onClick={toggle}>
|
||
{answerPreview(entry.answer)}
|
||
</button>
|
||
)}
|
||
</div>
|
||
)
|
||
}
|
||
|
||
/** 当前这一轮问的一次，照常颜色：问题，回答逐字长出来（按 Markdown 排），在做的事、出错、停止写在下面 */
|
||
function CurrentItem({ ask, onOpenImage, onStop }: { ask: Ask, onOpenImage: (url: string) => void, onStop?: () => void }): ReactElement {
|
||
const running = ask.status === 'running'
|
||
const label = ask.activity ?? (ask.text === '' ? '正在想…' : '正在回答…')
|
||
return (
|
||
<div className="cr-qa-entry" data-current="">
|
||
<div className="cr-qa-head">
|
||
<Sparkles size={12} strokeWidth={1.8} className="cr-qa-icon" aria-hidden />
|
||
<span className="cr-qa-question">{ask.question}</span>
|
||
</div>
|
||
<Thumbs urls={ask.images ?? []} onOpen={onOpenImage} />
|
||
{ask.text !== '' && (
|
||
<div
|
||
className="cr-markdown cr-qa-answer"
|
||
data-streaming={running ? '' : undefined}
|
||
dangerouslySetInnerHTML={{ __html: running ? renderMarkdownOnce(ask.text) : renderMarkdown(ask.text) }}
|
||
/>
|
||
)}
|
||
{running && (
|
||
<div className="cr-qa-status" role="status">
|
||
<LoaderCircle size={12} strokeWidth={1.8} className="cr-qa-spin" aria-hidden />
|
||
{onStop ? <span className="min-w-0 flex-1">{label}</span> : label}
|
||
{onStop && (
|
||
<button type="button" className="cr-tool" data-tooltip="停止" aria-label="停止" onClick={onStop}>
|
||
<CircleStop size={14} strokeWidth={1.8} aria-hidden />
|
||
</button>
|
||
)}
|
||
</div>
|
||
)}
|
||
{ask.status === 'failed' && (
|
||
<div className="cr-qa-error" role="alert">
|
||
<TriangleAlert size={13} strokeWidth={1.8} className="mt-[3px] shrink-0" aria-hidden />
|
||
<span>{ask.error}</span>
|
||
</div>
|
||
)}
|
||
{ask.status === 'stopped' && <div className="cr-qa-status">已停止，这次没存。</div>}
|
||
</div>
|
||
)
|
||
}
|
||
|
||
/** 底下的输入框：Enter 发送、Shift+Enter 换行、Esc 收起；正在答时发送换成停止。能贴图、拖图，贴的图在输入框上面一排，随下一个问题发 */
|
||
function AskInput({ running, first, profiles, onSelectProfile, onAsk, onStop, onClose, onOpenImage }: {
|
||
running: Ask | undefined
|
||
/** 这块还没问过：提示「问这块代码…」，问过了是「继续问…」 */
|
||
first: boolean
|
||
profiles: ProfileChoice
|
||
onSelectProfile: (id: string) => void
|
||
onAsk: (question: string, images: PendingImage[]) => void
|
||
onStop: (askId: string) => void
|
||
onClose: () => void
|
||
onOpenImage: (url: string) => void
|
||
}): ReactElement {
|
||
const [value, setValue] = useState('')
|
||
const [images, setImages] = useState<PendingImage[]>([])
|
||
/** 贴的、拖的有没加上的：给人看的原因，下次贴图或发送时清掉 */
|
||
const [notice, setNotice] = useState<string>()
|
||
const textarea = useRef<HTMLTextAreaElement>(null)
|
||
const attach = (files: File[]): void => {
|
||
if (files.length === 0)
|
||
return
|
||
const picked = pickImages(files, images.length)
|
||
setNotice(picked.notice)
|
||
Promise.all(picked.accepted.map(readImage)).then(
|
||
read => setImages(current => [...current, ...read].slice(0, MAX_IMAGES)),
|
||
() => setNotice('有图片读不出来，没加上。'),
|
||
)
|
||
}
|
||
const send = (): void => {
|
||
const question = value.trim()
|
||
if (question === '' || running)
|
||
return
|
||
onAsk(question, images)
|
||
setValue('')
|
||
setImages([])
|
||
setNotice(undefined)
|
||
}
|
||
const onKeyDown = (event: KeyboardEvent<HTMLTextAreaElement>): void => {
|
||
// 输入法选字时的回车不算发送
|
||
if (event.key === 'Enter' && !event.shiftKey && !event.nativeEvent.isComposing) {
|
||
event.preventDefault()
|
||
send()
|
||
}
|
||
else if (event.key === 'Escape') {
|
||
event.preventDefault()
|
||
onClose()
|
||
}
|
||
}
|
||
// 剪贴板里的图片加到上面的缩略图里；文字照常粘进输入框，不拦
|
||
const onPaste = (event: ClipboardEvent<HTMLTextAreaElement>): void => attach(transferFiles(event.clipboardData.items))
|
||
const onDragOver = (event: DragEvent<HTMLDivElement>): void => {
|
||
if (event.dataTransfer.types.includes('Files')) {
|
||
event.preventDefault()
|
||
event.dataTransfer.dropEffect = 'copy'
|
||
}
|
||
}
|
||
const onDrop = (event: DragEvent<HTMLDivElement>): void => {
|
||
const files = transferFiles(event.dataTransfer.items)
|
||
if (files.length > 0) {
|
||
event.preventDefault()
|
||
attach(files)
|
||
}
|
||
}
|
||
const remove = (index: number): void => {
|
||
setImages(current => current.filter((_, i) => i !== index))
|
||
textarea.current?.focus()
|
||
}
|
||
return (
|
||
<>
|
||
<Thumbs urls={images.map(image => image.url)} pending onOpen={onOpenImage} onRemove={remove} />
|
||
{notice && <div className="cr-qa-notice" role="status">{notice}</div>}
|
||
<div className="cr-qa-input" onDragOver={onDragOver} onDrop={onDrop}>
|
||
<textarea
|
||
ref={textarea}
|
||
rows={1}
|
||
autoFocus
|
||
value={value}
|
||
placeholder={`${first ? '问这块代码…' : '继续问…'} Enter 发送，Esc 收起`}
|
||
aria-label="问 AI 的问题"
|
||
onChange={event => setValue(event.target.value)}
|
||
onKeyDown={onKeyDown}
|
||
onPaste={onPaste}
|
||
/>
|
||
<SelectMenu
|
||
variant="ghost"
|
||
value={profiles.selected}
|
||
options={profiles.options.map(option => ({ value: option.id, label: option.label }))}
|
||
onChange={onSelectProfile}
|
||
placeholder="默认"
|
||
tooltip="用哪个 Claude Code 配置（profile）回答，下一个问题起用它"
|
||
className="cr-qa-profile"
|
||
/>
|
||
{running
|
||
? (
|
||
<button type="button" className="cr-tool" data-tooltip="停止" aria-label="停止" onClick={() => onStop(running.askId)}>
|
||
<CircleStop size={14} strokeWidth={1.8} aria-hidden />
|
||
</button>
|
||
)
|
||
: (
|
||
<button type="button" className="cr-tool" data-tooltip="发送（Enter）" aria-label="发送" disabled={value.trim() === ''} onClick={send}>
|
||
<SendHorizontal size={14} strokeWidth={1.8} aria-hidden />
|
||
</button>
|
||
)}
|
||
</div>
|
||
</>
|
||
)
|
||
}
|
||
|
||
export interface AskThreadProps {
|
||
history: QaEntryView[]
|
||
asks: Ask[]
|
||
/** 点了「问 AI」：展开全部历史、当前这一轮和输入框 */
|
||
open: boolean
|
||
/** 输入框旁边的 profile 下拉 */
|
||
profiles: ProfileChoice
|
||
now?: number
|
||
onSelectProfile: (id: string) => void
|
||
/** images 是贴进输入框的图片，随问题一起发 */
|
||
onAsk: (question: string, images: PendingImage[]) => void
|
||
onStop: (askId: string) => void
|
||
onDelete: (entryId: string) => void
|
||
onClose: () => void
|
||
}
|
||
|
||
/**
|
||
* 块下面的问答区（左边一条竖线）：展开时是灰字的历史、当前这一轮、输入框；
|
||
* 没展开但问过的块是一行「问过 N 次」、下面默认摊开灰字历史，点这一行收起 / 再摊开，读代码时一眼看到这里问过什么。
|
||
*/
|
||
export function AskThread({ history, asks, open, profiles, now = Date.now(), onSelectProfile, onAsk, onStop, onDelete, onClose }: AskThreadProps): ReactElement | null {
|
||
const [showHistory, setShowHistory] = useState(true)
|
||
/** 正在看大图的那张 */
|
||
const [preview, setPreview] = useState<string>()
|
||
const lightbox = preview !== undefined && <ImagePreview url={preview} onClose={() => setPreview(undefined)} />
|
||
if (!open) {
|
||
if (history.length === 0)
|
||
return null
|
||
const Chevron = showHistory ? ChevronDown : ChevronRight
|
||
return (
|
||
<div className="cr-qa" data-collapsed="">
|
||
<button type="button" className="cr-qa-summary" aria-expanded={showHistory} onClick={() => setShowHistory(value => !value)}>
|
||
<Sparkles size={12} strokeWidth={1.8} className="cr-qa-icon" aria-hidden />
|
||
{`问过 ${history.length} 次`}
|
||
<Chevron size={12} strokeWidth={1.8} aria-hidden />
|
||
</button>
|
||
{showHistory && history.map(entry => <HistoryItem key={entry.id} entry={entry} now={now} onDelete={() => onDelete(entry.id)} onOpenImage={setPreview} />)}
|
||
{lightbox}
|
||
</div>
|
||
)
|
||
}
|
||
return (
|
||
<section className="cr-qa" aria-label="问 AI">
|
||
{history.map(entry => <HistoryItem key={entry.id} entry={entry} now={now} onDelete={() => onDelete(entry.id)} onOpenImage={setPreview} />)}
|
||
{history.length > 0 && asks.length > 0 && <hr className="cr-qa-divider" />}
|
||
{asks.map(ask => <CurrentItem key={ask.askId} ask={ask} onOpenImage={setPreview} />)}
|
||
<AskInput
|
||
running={asks.find(ask => ask.status === 'running')}
|
||
first={history.length === 0 && asks.length === 0}
|
||
profiles={profiles}
|
||
onSelectProfile={onSelectProfile}
|
||
onAsk={onAsk}
|
||
onStop={onStop}
|
||
onClose={onClose}
|
||
onOpenImage={setPreview}
|
||
/>
|
||
{lightbox}
|
||
</section>
|
||
)
|
||
}
|
||
|
||
/**
|
||
* 注释块下面的重写过程，样子和展开的「问 AI」一样：左边竖线、问题、流出来的回答、正在读哪个文件。
|
||
*
|
||
* 问题是发给模型的那句固定要求。没有输入框，不记历史。写进文件后由调用方拿掉；失败或停下就留在这里。
|
||
*/
|
||
export function RewriteThread({ run, onStop }: { run: RewriteRun, onStop: () => void }): ReactElement {
|
||
const ask: Ask = {
|
||
askId: 'rewrite',
|
||
anchor: '',
|
||
question: run.question,
|
||
text: run.text,
|
||
...(run.activity !== undefined && { activity: run.activity }),
|
||
status: run.status,
|
||
...(run.error !== undefined && { error: run.error }),
|
||
}
|
||
return (
|
||
<section className="cr-qa" aria-label="按注释规范重写">
|
||
<CurrentItem ask={ask} onOpenImage={() => {}} onStop={run.status === 'running' ? onStop : undefined} />
|
||
</section>
|
||
)
|
||
}
|