feat(room): redesign mention system with AST-based format

Backend:
- Add MENTION_TAG_RE matching new `<mention type="..." id="...">label</mention>` format
- Extend extract_mentions() and resolve_mentions() to parse new format (legacy backward-compatible)

Frontend:
- New src/lib/mention-ast.ts: AST types (TextNode, MentionNode, AiActionNode),
  parse() and serialize() functions for AST↔HTML conversion
- MentionPopover: load @repository: from project repos, @ai: from room_ai configs
  (not room members); output new HTML format with ID instead of label
- MessageMentions: use AST parse() for rendering (falls back to legacy parser)
- ChatInputArea: insertMention now produces `<mention type="user" id="...">label</mention>`
- RoomParticipantsPanel: onMention passes member UUID to insertMention
- RoomContext: add projectRepos and roomAiConfigs for mention data sources
This commit is contained in:
ZhenYi 2026-04-17 23:43:26 +08:00
parent 0ea6440ea3
commit 26682973e7
8 changed files with 461 additions and 81 deletions

View File

@ -27,6 +27,15 @@ const DEFAULT_MAX_CONCURRENT_WORKERS: usize = 1024;
static USER_MENTION_RE: LazyLock<regex_lite::Regex, fn() -> regex_lite::Regex> = static USER_MENTION_RE: LazyLock<regex_lite::Regex, fn() -> regex_lite::Regex> =
LazyLock::new(|| regex_lite::Regex::new(r"<user>\s*([^<]+?)\s*</user>").unwrap()); LazyLock::new(|| regex_lite::Regex::new(r"<user>\s*([^<]+?)\s*</user>").unwrap());
/// Matches <mention type="..." id="...">label</mention>
static MENTION_TAG_RE: LazyLock<regex_lite::Regex, fn() -> regex_lite::Regex> =
LazyLock::new(|| {
regex_lite::Regex::new(
r#"<mention\s+type="([^"]+)"\s+id="([^"]+)"[^>]*>\s*([^<]*?)\s*</mention>"#,
)
.unwrap()
});
#[derive(Clone)] #[derive(Clone)]
pub struct RoomService { pub struct RoomService {
pub db: AppDatabase, pub db: AppDatabase,
@ -533,8 +542,12 @@ impl RoomService {
Ok(()) Ok(())
} }
/// Extracts user UUIDs from both the legacy `<user>uuid</user>` format
/// and the new `<mention type="user" id="uuid">label</mention>` format.
pub fn extract_mentions(content: &str) -> Vec<Uuid> { pub fn extract_mentions(content: &str) -> Vec<Uuid> {
let mut mentioned = Vec::new(); let mut mentioned = Vec::new();
// Legacy <user>uuid</user> format
for cap in USER_MENTION_RE.captures_iter(content) { for cap in USER_MENTION_RE.captures_iter(content) {
if let Some(inner) = cap.get(1) { if let Some(inner) = cap.get(1) {
let token = inner.as_str().trim(); let token = inner.as_str().trim();
@ -546,9 +559,25 @@ impl RoomService {
} }
} }
// New <mention type="user" id="...">label</mention> format
for cap in MENTION_TAG_RE.captures_iter(content) {
if let (Some(type_m), Some(id_m)) = (cap.get(1), cap.get(2)) {
if type_m.as_str() == "user" {
if let Ok(uuid) = Uuid::parse_str(id_m.as_str().trim()) {
if !mentioned.contains(&uuid) {
mentioned.push(uuid);
}
}
}
}
}
mentioned mentioned
} }
/// Resolves user mentions from both the legacy `<user>...` format and the
/// new `<mention type="user" id="uuid">label</mention>` format.
/// Repository and AI mention types are accepted but produce no user UUIDs.
pub async fn resolve_mentions(&self, content: &str) -> Vec<Uuid> { pub async fn resolve_mentions(&self, content: &str) -> Vec<Uuid> {
use models::users::User; use models::users::User;
use sea_orm::EntityTrait; use sea_orm::EntityTrait;
@ -556,6 +585,7 @@ impl RoomService {
let mut resolved: Vec<Uuid> = Vec::new(); let mut resolved: Vec<Uuid> = Vec::new();
let mut seen_usernames: Vec<String> = Vec::new(); let mut seen_usernames: Vec<String> = Vec::new();
// Legacy <user>uuid</user> or <user>username</user> format
for cap in USER_MENTION_RE.captures_iter(content) { for cap in USER_MENTION_RE.captures_iter(content) {
if let Some(inner) = cap.get(1) { if let Some(inner) = cap.get(1) {
let token = inner.as_str().trim(); let token = inner.as_str().trim();
@ -587,6 +617,46 @@ impl RoomService {
} }
} }
// New <mention type="user" id="uuid">label</mention> format
for cap in MENTION_TAG_RE.captures_iter(content) {
if let (Some(type_m), Some(id_m)) = (cap.get(1), cap.get(2)) {
if type_m.as_str() == "user" {
let id = id_m.as_str().trim();
if let Ok(uuid) = Uuid::parse_str(id) {
if !resolved.contains(&uuid) {
resolved.push(uuid);
}
} else {
// Fall back to label-based username lookup
if let Some(label_m) = cap.get(3) {
let label = label_m.as_str().trim();
if !label.is_empty() {
let label_lower = label.to_lowercase();
if seen_usernames.contains(&label_lower) {
continue;
}
seen_usernames.push(label_lower.clone());
if let Some(user) = User::find()
.filter(models::users::user::Column::Username.eq(label_lower))
.one(&self.db)
.await
.ok()
.flatten()
{
if !resolved.contains(&user.uid) {
resolved.push(user.uid);
}
}
}
}
}
}
// `repository` and `ai` mention types are accepted but do not
// produce user notification UUIDs — no-op here.
}
}
resolved resolved
} }

View File

@ -1,15 +1,25 @@
import type { RoomMemberResponse } from '@/client'; import type { ProjectRepositoryItem, RoomMemberResponse } from '@/client';
import { buildMentionHtml, type MentionMentionType } from '@/lib/mention-ast';
import { cn } from '@/lib/utils'; import { cn } from '@/lib/utils';
import { Check, CornerDownLeft, Keyboard, SearchX } from 'lucide-react'; import { Bot, Check, CornerDownLeft, Keyboard, SearchX } from 'lucide-react';
import { useCallback, useEffect, useMemo, useRef, useState } from 'react'; import { useCallback, useEffect, useMemo, useRef, useState } from 'react';
import type { SetStateAction } from 'react'; import type { SetStateAction } from 'react';
import { ModelIcon } from './icon-match'; import { ModelIcon } from './icon-match';
/** Room AI config — the configured model for this room */
export interface RoomAiConfig {
model: string; // model UID / model ID
modelName?: string;
}
interface MentionSuggestion { interface MentionSuggestion {
type: 'category' | 'item'; type: 'category' | 'item';
category?: 'repository' | 'user' | 'ai'; category?: MentionMentionType;
label: string; label: string;
/** Raw value stored in suggestion (for display) */
value: string; value: string;
/** ID used when building HTML mention (absent for category headers) */
mentionId?: string;
avatar?: string | null; avatar?: string | null;
} }
@ -27,10 +37,12 @@ interface PopoverPosition {
} }
interface MentionPopoverProps { interface MentionPopoverProps {
/** Available members for user/ai mention suggestions */ /** Available members for @user: mention suggestions */
members: RoomMemberResponse[]; members: RoomMemberResponse[];
/** Repository names available for @repository: mention */ /** Available repositories for @repository: mention suggestions */
repos?: string[]; repos?: ProjectRepositoryItem[];
/** Room AI configs for @ai: mention suggestions */
aiConfigs?: RoomAiConfig[];
inputValue: string; inputValue: string;
cursorPosition: number; cursorPosition: number;
onSelect: (newValue: string, newCursorPosition: number) => void; onSelect: (newValue: string, newCursorPosition: number) => void;
@ -41,6 +53,7 @@ interface MentionPopoverProps {
export function MentionPopover({ export function MentionPopover({
members, members,
repos = [], repos = [],
aiConfigs = [],
inputValue, inputValue,
cursorPosition, cursorPosition,
onSelect, onSelect,
@ -109,12 +122,13 @@ export function MentionPopover({
if (category === 'repository') { if (category === 'repository') {
return repos return repos
.filter((repo) => !item || repo.toLowerCase().includes(item)) .filter((repo) => !item || repo.repo_name.toLowerCase().includes(item))
.map((repo) => ({ .map((repo) => ({
type: 'item' as const, type: 'item' as const,
category: 'repository' as const, category: 'repository' as const,
label: repo, label: repo.repo_name,
value: `@repository:${repo}`, value: `@repository:${repo.repo_name}`,
mentionId: repo.uid,
avatar: null, avatar: null,
})); }));
} }
@ -132,33 +146,34 @@ export function MentionPopover({
type: 'item' as const, type: 'item' as const,
category: 'user' as const, category: 'user' as const,
label: username, label: username,
value: `@user:${m.user}`, value: `@user:${username}`,
mentionId: m.user,
avatar: m.user_info?.avatar_url ?? null, avatar: m.user_info?.avatar_url ?? null,
}; };
}); });
} }
if (category === 'ai') { if (category === 'ai') {
return members return aiConfigs
.filter((m) => m.role === 'ai') .filter((cfg) => {
.filter((m) => { const name = cfg.modelName ?? cfg.model;
const username = m.user_info?.username ?? m.user; return !item || name.toLowerCase().includes(item);
return !item || username.toLowerCase().includes(item);
}) })
.map((m) => { .map((cfg) => {
const username = m.user_info?.username ?? m.user; const label = cfg.modelName ?? cfg.model;
return { return {
type: 'item' as const, type: 'item' as const,
category: 'ai' as const, category: 'ai' as const,
label: username, label,
value: `@ai:${m.user}`, value: `@ai:${label}`,
avatar: m.user_info?.avatar_url ?? null, mentionId: cfg.model,
avatar: null,
}; };
}); });
} }
return []; return [];
}, [mentionState, members, repos]); }, [mentionState, members, repos, aiConfigs]);
const visibleSuggestions = useMemo(() => suggestions.slice(0, 8), [suggestions]); const visibleSuggestions = useMemo(() => suggestions.slice(0, 8), [suggestions]);
const validSelectedIndex = selectedIndex >= visibleSuggestions.length ? 0 : selectedIndex; const validSelectedIndex = selectedIndex >= visibleSuggestions.length ? 0 : selectedIndex;
@ -173,16 +188,27 @@ export function MentionPopover({
const before = inputValue.slice(0, mentionState.startPos); const before = inputValue.slice(0, mentionState.startPos);
const after = inputValue.slice(cursorPosition); const after = inputValue.slice(cursorPosition);
const spacer = suggestion.type === 'item' ? ' ' : '';
const newValue = before + suggestion.value + spacer + after;
const newCursorPos = mentionState.startPos + suggestion.value.length + spacer.length;
onSelect(newValue, newCursorPos); let newValue: string;
let newCursorPos: number;
if (suggestion.type === 'item') { if (suggestion.type === 'category') {
closePopover(); // User selected a category header (e.g., @user:) — keep the partial
} else { newValue = before + suggestion.value + after;
newCursorPos = mentionState.startPos + suggestion.value.length;
onOpenChange(true); onOpenChange(true);
} else {
// Build new HTML mention string
const html = buildMentionHtml(
suggestion.category!,
suggestion.mentionId!,
suggestion.label,
);
const spacer = ' ';
newValue = before + html + spacer + after;
newCursorPos = mentionState.startPos + html.length + spacer.length;
onSelect(newValue, newCursorPos);
closePopover();
} }
}; };
}, [mentionState, inputValue, cursorPosition, onSelect, closePopover, onOpenChange]); }, [mentionState, inputValue, cursorPosition, onSelect, closePopover, onOpenChange]);
@ -351,7 +377,11 @@ export function MentionPopover({
{suggestion.type === 'category' ? ( {suggestion.type === 'category' ? (
'>' '>'
) : suggestion.category === 'ai' ? ( ) : suggestion.category === 'ai' ? (
<ModelIcon modelId={suggestion.value.replace(/^@ai:/, '')} className="h-4 w-4" /> suggestion.avatar ? (
<ModelIcon modelId={suggestion.mentionId} className="h-4 w-4" />
) : (
<Bot className="h-4 w-4" />
)
) : ( ) : (
suggestion.label[0]?.toUpperCase() suggestion.label[0]?.toUpperCase()
)} )}

View File

@ -1,10 +1,10 @@
import { memo, useMemo } from 'react'; import { memo, useMemo } from 'react';
import { cn } from '@/lib/utils'; import { cn } from '@/lib/utils';
import { parse, type Node, type MentionMentionType } from '@/lib/mention-ast';
type MentionType = 'repository' | 'user' | 'ai' | 'notify'; type MentionType = 'repository' | 'user' | 'ai' | 'notify';
interface MentionToken { interface LegacyMentionToken {
full: string; full: string;
type: MentionType; type: MentionType;
name: string; name: string;
@ -14,12 +14,17 @@ function isMentionNameChar(ch: string): boolean {
return /[A-Za-z0-9._:\/-]/.test(ch); return /[A-Za-z0-9._:\/-]/.test(ch);
} }
function extractMentionTokens(text: string): MentionToken[] { /**
const tokens: MentionToken[] = []; * Parses legacy mention formats for backward compatibility with old messages:
* - `<user>name</user>` (old backend format)
* - `@type:name` (old colon format)
* - `<type>name</type>` (old XML format)
*/
function extractLegacyMentionTokens(text: string): LegacyMentionToken[] {
const tokens: LegacyMentionToken[] = [];
let cursor = 0; let cursor = 0;
while (cursor < text.length) { while (cursor < text.length) {
// Find next potential mention: <user> or <user: or @user:
const angleOpen = text.indexOf('<', cursor); const angleOpen = text.indexOf('<', cursor);
const atOpen = text.indexOf('@', cursor); const atOpen = text.indexOf('@', cursor);
@ -36,13 +41,10 @@ function extractMentionTokens(text: string): MentionToken[] {
} }
const typeStart = next + 1; const typeStart = next + 1;
// Check for colon format: <type:name> or @type:name
const colon = text.indexOf(':', typeStart); const colon = text.indexOf(':', typeStart);
const validTypes: MentionType[] = ['repository', 'user', 'ai', 'notify']; const validTypes: MentionType[] = ['repository', 'user', 'ai', 'notify'];
if (colon >= 0 && colon - typeStart <= 20) { if (colon >= 0 && colon - typeStart <= 20) {
// Colon format: <type:name> or @type:name
const typeRaw = text.slice(typeStart, colon).toLowerCase(); const typeRaw = text.slice(typeStart, colon).toLowerCase();
if (!validTypes.includes(typeRaw as MentionType)) { if (!validTypes.includes(typeRaw as MentionType)) {
cursor = next + 1; cursor = next + 1;
@ -54,7 +56,6 @@ function extractMentionTokens(text: string): MentionToken[] {
end++; end++;
} }
// For angle style, require closing >
const closeBracket = style === 'angle' && text[end] === '>' ? end + 1 : end; const closeBracket = style === 'angle' && text[end] === '>' ? end + 1 : end;
const name = text.slice(colon + 1, style === 'angle' ? end : closeBracket); const name = text.slice(colon + 1, style === 'angle' ? end : closeBracket);
@ -65,10 +66,7 @@ function extractMentionTokens(text: string): MentionToken[] {
} }
} }
// Check for XML-like format: <type>name</type>
// Only for angle style
if (style === 'angle') { if (style === 'angle') {
// Find space or > to determine type end
let typeEnd = typeStart; let typeEnd = typeStart;
while (typeEnd < text.length && /[A-Za-z]/.test(text[typeEnd])) { while (typeEnd < text.length && /[A-Za-z]/.test(text[typeEnd])) {
typeEnd++; typeEnd++;
@ -76,11 +74,8 @@ function extractMentionTokens(text: string): MentionToken[] {
const typeCandidate = text.slice(typeStart, typeEnd); const typeCandidate = text.slice(typeStart, typeEnd);
if (validTypes.includes(typeCandidate as MentionType)) { if (validTypes.includes(typeCandidate as MentionType)) {
// Found opening tag like <user>
const closeTag = `</${typeCandidate}>`; const closeTag = `</${typeCandidate}>`;
const contentStart = typeEnd; const contentStart = typeEnd;
// Find the closing tag
const tagClose = text.indexOf(closeTag, cursor); const tagClose = text.indexOf(closeTag, cursor);
if (tagClose >= 0) { if (tagClose >= 0) {
const name = text.slice(contentStart, tagClose); const name = text.slice(contentStart, tagClose);
@ -99,7 +94,7 @@ function extractMentionTokens(text: string): MentionToken[] {
} }
function extractFirstMentionName(text: string, type: MentionType): string | null { function extractFirstMentionName(text: string, type: MentionType): string | null {
const token = extractMentionTokens(text).find((item) => item.type === type); const token = extractLegacyMentionTokens(text).find((item) => item.type === type);
return token?.name ?? null; return token?.name ?? null;
} }
@ -107,38 +102,73 @@ interface MessageContentWithMentionsProps {
content: string; content: string;
} }
const mentionStyles: Record<MentionType, string> = { const mentionStyles: Record<string, string> = {
user: 'inline-flex items-center rounded bg-blue-100/80 px-1.5 py-0.5 text-blue-700 dark:bg-blue-900/40 dark:text-blue-300 font-medium cursor-pointer hover:bg-blue-200 dark:hover:bg-blue-900/60 transition-colors text-sm leading-5', user: 'inline-flex items-center rounded bg-blue-100/80 px-1.5 py-0.5 text-blue-700 dark:bg-blue-900/40 dark:text-blue-300 font-medium cursor-pointer hover:bg-blue-200 dark:hover:bg-blue-900/60 transition-colors text-sm leading-5',
repository: 'inline-flex items-center rounded bg-purple-100/80 px-1.5 py-0.5 text-purple-700 dark:bg-purple-900/40 dark:text-purple-300 font-medium cursor-pointer hover:bg-purple-200 dark:hover:bg-purple-900/60 transition-colors text-sm leading-5', repository: 'inline-flex items-center rounded bg-purple-100/80 px-1.5 py-0.5 text-purple-700 dark:bg-purple-900/40 dark:text-purple-300 font-medium cursor-pointer hover:bg-purple-200 dark:hover:bg-purple-900/60 transition-colors text-sm leading-5',
ai: 'inline-flex items-center rounded bg-green-100/80 px-1.5 py-0.5 text-green-700 dark:bg-green-900/40 dark:text-green-300 font-medium cursor-pointer hover:bg-green-200 dark:hover:bg-green-900/60 transition-colors text-sm leading-5', ai: 'inline-flex items-center rounded bg-green-100/80 px-1.5 py-0.5 text-green-700 dark:bg-green-900/40 dark:text-green-300 font-medium cursor-pointer hover:bg-green-200 dark:hover:bg-green-900/60 transition-colors text-sm leading-5',
notify: 'inline-flex items-center rounded bg-yellow-100/80 px-1.5 py-0.5 text-yellow-700 dark:bg-yellow-900/40 dark:text-yellow-300 font-medium cursor-pointer hover:bg-yellow-200 dark:hover:bg-yellow-900/60 transition-colors text-sm leading-5', notify: 'inline-flex items-center rounded bg-yellow-100/80 px-1.5 py-0.5 text-yellow-700 dark:bg-yellow-900/40 dark:text-yellow-300 font-medium cursor-pointer hover:bg-yellow-200 dark:hover:bg-yellow-900/60 transition-colors text-sm leading-5',
}; };
/** Renders message content with @mention highlighting using styled spans */ function renderNode(node: Node, index: number): React.ReactNode {
if (node.type === 'text') {
return <span key={index}>{node.text}</span>;
}
if (node.type === 'mention') {
return (
<span key={index} className={mentionStyles[node.mentionType] ?? mentionStyles.user}>
@{node.label}
</span>
);
}
if (node.type === 'ai_action') {
return (
<span
key={index}
className="inline-flex items-center rounded bg-green-100/80 px-1.5 py-0.5 text-green-700 dark:bg-green-900/40 dark:text-green-300 font-mono text-xs"
>
/{node.action}
{node.args ? ` ${node.args}` : ''}
</span>
);
}
return null;
}
/** Renders message content with @mention highlighting using styled spans.
* Supports the new `<mention type="..." id="...">label</mention>` format
* and legacy formats for backward compatibility with old messages.
*/
export const MessageContentWithMentions = memo(function MessageContentWithMentions({ export const MessageContentWithMentions = memo(function MessageContentWithMentions({
content, content,
}: MessageContentWithMentionsProps) { }: MessageContentWithMentionsProps) {
const processed = useMemo(() => { const nodes = useMemo(() => {
const tokens = extractMentionTokens(content); // Try the new AST parser first (handles <mention> and <ai> tags)
if (tokens.length === 0) return [{ type: 'text' as const, content }]; const ast = parse(content);
if (ast.length > 0) return ast;
const parts: Array<{ type: 'text'; content: string } | { type: 'mention'; mention: MentionToken }> = []; // Fall back to legacy parser for old-format messages
const legacy = extractLegacyMentionTokens(content);
if (legacy.length === 0) return [{ type: 'text' as const, text: content }];
const parts: Node[] = [];
let cursor = 0; let cursor = 0;
for (const token of legacy) {
for (const token of tokens) {
const idx = content.indexOf(token.full, cursor); const idx = content.indexOf(token.full, cursor);
if (idx === -1) continue; if (idx === -1) continue;
if (idx > cursor) { if (idx > cursor) {
parts.push({ type: 'text', content: content.slice(cursor, idx) }); parts.push({ type: 'text', text: content.slice(cursor, idx) });
} }
parts.push({ type: 'mention', mention: token }); parts.push({
type: 'mention',
mentionType: token.type as MentionMentionType,
id: '',
label: token.name,
});
cursor = idx + token.full.length; cursor = idx + token.full.length;
} }
if (cursor < content.length) { if (cursor < content.length) {
parts.push({ type: 'text', content: content.slice(cursor) }); parts.push({ type: 'text', text: content.slice(cursor) });
} }
return parts; return parts;
}, [content]); }, [content]);
@ -151,18 +181,7 @@ export const MessageContentWithMentions = memo(function MessageContentWithMentio
'[&_pre]:rounded-md [&_pre]:bg-muted [&_pre]:p-3 [&_pre]:overflow-x-auto', '[&_pre]:rounded-md [&_pre]:bg-muted [&_pre]:p-3 [&_pre]:overflow-x-auto',
)} )}
> >
{processed.map((part, i) => {nodes.map((node, i) => renderNode(node, i))}
part.type === 'mention' ? (
<span
key={i}
className={mentionStyles[part.mention.type]}
>
@{part.mention.name}
</span>
) : (
<span key={i}>{part.content}</span>
),
)}
</div> </div>
); );
}); });

View File

@ -1,5 +1,6 @@
import type { RoomResponse, RoomMemberResponse, RoomMessageResponse, RoomThreadResponse } from '@/client'; import type { ProjectRepositoryItem, RoomResponse, RoomMemberResponse, RoomMessageResponse, RoomThreadResponse } from '@/client';
import { useRoom, type MessageWithMeta } from '@/contexts'; import { useRoom, type MessageWithMeta } from '@/contexts';
import { type RoomAiConfig } from '@/contexts/room-context';
import { useRoomDraft } from '@/hooks/useRoomDraft'; import { useRoomDraft } from '@/hooks/useRoomDraft';
import { Button } from '@/components/ui/button'; import { Button } from '@/components/ui/button';
import { Textarea } from '@/components/ui/textarea'; import { Textarea } from '@/components/ui/textarea';
@ -30,7 +31,7 @@ const MENTION_PATTERN = /@([^:@\s]*)(:([^\s]*))?$/;
const MENTION_POPOVER_KEYS = ['Enter', 'Tab', 'ArrowUp', 'ArrowDown']; const MENTION_POPOVER_KEYS = ['Enter', 'Tab', 'ArrowUp', 'ArrowDown'];
export interface ChatInputAreaHandle { export interface ChatInputAreaHandle {
insertMention: (name: string, type: 'user' | 'ai') => void; insertMention: (id: string, label: string, type: 'user' | 'ai') => void;
} }
interface ChatInputAreaProps { interface ChatInputAreaProps {
@ -38,6 +39,8 @@ interface ChatInputAreaProps {
onSend: (content: string) => void; onSend: (content: string) => void;
isSending: boolean; isSending: boolean;
members: RoomMemberResponse[]; members: RoomMemberResponse[];
repos?: ProjectRepositoryItem[];
aiConfigs?: RoomAiConfig[];
replyingTo?: { id: string; display_name?: string; content: string } | null; replyingTo?: { id: string; display_name?: string; content: string } | null;
onCancelReply?: () => void; onCancelReply?: () => void;
draft: string; draft: string;
@ -51,6 +54,8 @@ const ChatInputArea = memo(function ChatInputArea({
onSend, onSend,
isSending, isSending,
members, members,
repos,
aiConfigs,
replyingTo, replyingTo,
onCancelReply, onCancelReply,
draft, draft,
@ -75,11 +80,14 @@ const ChatInputArea = memo(function ChatInputArea({
}, [onDraftChange]); }, [onDraftChange]);
useImperativeHandle(ref, () => ({ useImperativeHandle(ref, () => ({
insertMention: (name: string) => { insertMention: (id: string, label: string) => {
if (!textareaRef.current) return; if (!textareaRef.current) return;
const value = textareaRef.current.value; const value = textareaRef.current.value;
const cursorPos = textareaRef.current.selectionStart; const cursorPos = textareaRef.current.selectionStart;
const mentionText = `<user>${name}</user> `; // Build new HTML mention: <mention type="user" id="uuid">label</mention>
const escapedLabel = label.replace(/</g, '&lt;').replace(/>/g, '&gt;');
const escapedId = id.replace(/"/g, '&quot;');
const mentionText = `<mention type="user" id="${escapedId}">${escapedLabel}</mention> `;
const before = value.substring(0, cursorPos); const before = value.substring(0, cursorPos);
const after = value.substring(cursorPos); const after = value.substring(cursorPos);
const newValue = before + mentionText + after; const newValue = before + mentionText + after;
@ -169,6 +177,8 @@ const ChatInputArea = memo(function ChatInputArea({
{showMentionPopover && ( {showMentionPopover && (
<MentionPopover <MentionPopover
members={members} members={members}
repos={repos}
aiConfigs={aiConfigs}
inputValue={draft} inputValue={draft}
cursorPosition={cursorPosition} cursorPosition={cursorPosition}
onSelect={handleMentionSelect} onSelect={handleMentionSelect}
@ -244,6 +254,8 @@ export function RoomChatPanel({ room, isAdmin, onClose, onDelete }: RoomChatPane
wsClient, wsClient,
threads, threads,
refreshThreads, refreshThreads,
projectRepos,
roomAiConfigs,
} = useRoom(); } = useRoom();
const messagesEndRef = useRef<HTMLDivElement>(null); const messagesEndRef = useRef<HTMLDivElement>(null);
@ -322,8 +334,8 @@ export function RoomChatPanel({ room, isAdmin, onClose, onDelete }: RoomChatPane
); );
// Stable: chatInputRef is stable, no deps that change on message updates // Stable: chatInputRef is stable, no deps that change on message updates
const handleMention = useCallback((name: string, type: 'user' | 'ai') => { const handleMention = useCallback((id: string, label: string) => {
chatInputRef.current?.insertMention(name, type); chatInputRef.current?.insertMention(id, label, 'user');
}, []); }, []);
const handleSelectSearchResult = useCallback((message: RoomMessageResponse) => { const handleSelectSearchResult = useCallback((message: RoomMessageResponse) => {
@ -530,6 +542,8 @@ export function RoomChatPanel({ room, isAdmin, onClose, onDelete }: RoomChatPane
onSend={handleSend} onSend={handleSend}
isSending={false} isSending={false}
members={members} members={members}
repos={projectRepos}
aiConfigs={roomAiConfigs}
replyingTo={replyingTo ? { id: replyingTo.id, display_name: replyingTo.display_name ?? undefined, content: replyingTo.content } : null} replyingTo={replyingTo ? { id: replyingTo.id, display_name: replyingTo.display_name ?? undefined, content: replyingTo.content } : null}
onCancelReply={() => setReplyingTo(null)} onCancelReply={() => setReplyingTo(null)}
draft={draft} draft={draft}

View File

@ -9,7 +9,7 @@ import type { ReactNode } from 'react';
interface RoomParticipantsPanelProps { interface RoomParticipantsPanelProps {
members: RoomMemberResponse[]; members: RoomMemberResponse[];
membersLoading: boolean; membersLoading: boolean;
onMention?: (name: string, type: 'user' | 'ai') => void; onMention?: (id: string, label: string, type: 'user' | 'ai') => void;
} }
export const RoomParticipantsPanel = memo(function RoomParticipantsPanel({ export const RoomParticipantsPanel = memo(function RoomParticipantsPanel({
@ -97,7 +97,7 @@ function ParticipantRow({
onMention, onMention,
}: { }: {
member: RoomMemberResponse; member: RoomMemberResponse;
onMention?: (name: string, type: 'user' | 'ai') => void; onMention?: (id: string, label: string, type: 'user' | 'ai') => void;
}) { }) {
const username = member.user_info?.username ?? member.user; const username = member.user_info?.username ?? member.user;
const avatarUrl = member.user_info?.avatar_url; const avatarUrl = member.user_info?.avatar_url;
@ -110,7 +110,7 @@ function ParticipantRow({
className={cn( className={cn(
'flex w-full items-center gap-2 rounded-md px-2 py-1.5 text-left transition-colors hover:bg-muted/60', 'flex w-full items-center gap-2 rounded-md px-2 py-1.5 text-left transition-colors hover:bg-muted/60',
)} )}
onClick={() => onMention?.(username, 'user')} onClick={() => onMention?.(member.user, username, 'user')}
disabled={!onMention} disabled={!onMention}
> >
<Avatar className="h-7 w-7"> <Avatar className="h-7 w-7">

View File

@ -1,5 +1,5 @@
import { memo, useState, useEffect, useCallback } from 'react'; import { memo, useState, useEffect, useCallback } from 'react';
import type { ModelResponse, RoomResponse } from '@/client'; import type { ModelResponse, RoomResponse, RoomAiResponse, RoomAiUpsertRequest } from '@/client';
import { aiList, aiUpsert, aiDelete, modelList } from '@/client'; import { aiList, aiUpsert, aiDelete, modelList } from '@/client';
import { Button } from '@/components/ui/button'; import { Button } from '@/components/ui/button';
import { Input } from '@/components/ui/input'; import { Input } from '@/components/ui/input';
@ -15,7 +15,6 @@ import {
} from '@/components/ui/dialog'; } from '@/components/ui/dialog';
import { Loader2, Plus, Trash2, Bot, ChevronDown, ChevronRight } from 'lucide-react'; import { Loader2, Plus, Trash2, Bot, ChevronDown, ChevronRight } from 'lucide-react';
import { toast } from 'sonner'; import { toast } from 'sonner';
import type { ModelResponse, RoomAiResponse, RoomAiUpsertRequest } from '@/client';
interface RoomSettingsPanelProps { interface RoomSettingsPanelProps {
room: RoomResponse; room: RoomResponse;

View File

@ -11,6 +11,7 @@ import {
import { useNavigate } from 'react-router-dom'; import { useNavigate } from 'react-router-dom';
import { toast } from 'sonner'; import { toast } from 'sonner';
import { import {
type ProjectRepositoryItem,
type RoomCategoryResponse, type RoomCategoryResponse,
type RoomMemberResponse, type RoomMemberResponse,
type RoomMessageResponse, type RoomMessageResponse,
@ -39,6 +40,11 @@ import {
export type { RoomWsStatus, RoomWsClient } from '@/lib/room-ws-client'; export type { RoomWsStatus, RoomWsClient } from '@/lib/room-ws-client';
export interface RoomAiConfig {
model: string;
modelName?: string;
}
export interface ReactionGroup { export interface ReactionGroup {
emoji: string; emoji: string;
count: number; count: number;
@ -146,6 +152,11 @@ interface RoomContextValue {
updateRoom: (roomId: string, name?: string, isPublic?: boolean, categoryId?: string) => Promise<void>; updateRoom: (roomId: string, name?: string, isPublic?: boolean, categoryId?: string) => Promise<void>;
deleteRoom: (roomId: string) => Promise<void>; deleteRoom: (roomId: string) => Promise<void>;
streamingMessages: Map<string, string>; streamingMessages: Map<string, string>;
/** Project repositories for @repository: mention suggestions */
projectRepos: ProjectRepositoryItem[];
/** Room AI configs for @ai: mention suggestions */
roomAiConfigs: RoomAiConfig[];
} }
const RoomContext = createContext<RoomContextValue | null>(null); const RoomContext = createContext<RoomContextValue | null>(null);
@ -415,6 +426,13 @@ export function RoomProvider({
const [streamingContent, setStreamingContent] = useState<Map<string, string>>(new Map()); const [streamingContent, setStreamingContent] = useState<Map<string, string>>(new Map());
// Project repos for @repository: mention suggestions
const [projectRepos, setProjectRepos] = useState<ProjectRepositoryItem[]>([]);
// Room AI configs for @ai: mention suggestions
const [roomAiConfigs, setRoomAiConfigs] = useState<RoomAiConfig[]>([]);
// Available models (for looking up AI model names)
const [availableModels, setAvailableModels] = useState<{ id: string; name: string }[]>([]);
useEffect(() => { useEffect(() => {
const baseUrl = import.meta.env.VITE_API_BASE_URL ?? window.location.origin; const baseUrl = import.meta.env.VITE_API_BASE_URL ?? window.location.origin;
const client = createRoomWsClient( const client = createRoomWsClient(
@ -1095,6 +1113,67 @@ export function RoomProvider({
[activeRoomId], [activeRoomId],
); );
// Fetch project repos for @repository: mention suggestions
const fetchProjectRepos = useCallback(async () => {
if (!projectName) {
setProjectRepos([]);
return;
}
try {
const baseUrl = import.meta.env.VITE_API_BASE_URL ?? window.location.origin;
const resp = await fetch(`${baseUrl}/api/projects/${encodeURIComponent(projectName)}/repos`);
if (!resp.ok) throw new Error(`HTTP ${resp.status}`);
const json: { data?: { items?: ProjectRepositoryItem[] } } = await resp.json();
setProjectRepos(json.data?.items ?? []);
} catch {
setProjectRepos([]);
}
}, [projectName]);
// Fetch room AI configs for @ai: mention suggestions
const fetchRoomAiConfigs = useCallback(async () => {
const client = wsClientRef.current;
if (!activeRoomId || !client) {
setRoomAiConfigs([]);
return;
}
try {
const configs = await client.aiList(activeRoomId);
// Look up model names from the available models list
setRoomAiConfigs(
configs.map((cfg) => ({
model: cfg.model,
modelName: availableModels.find((m) => m.id === cfg.model)?.name,
})),
);
} catch {
setRoomAiConfigs([]);
}
}, [activeRoomId, availableModels]);
// Fetch available models (for AI model name lookup)
const fetchAvailableModels = useCallback(async () => {
try {
const resp = await (await import('@/client')).modelList({});
const inner = (resp.data as { data?: { data?: { id: string; name: string }[] } } | undefined);
setAvailableModels(inner?.data?.data ?? []);
} catch {
// Non-fatal
}
}, []);
useEffect(() => {
fetchProjectRepos();
}, [fetchProjectRepos]);
useEffect(() => {
fetchRoomAiConfigs();
}, [fetchRoomAiConfigs]);
useEffect(() => {
fetchAvailableModels();
}, [fetchAvailableModels]);
const createRoom = useCallback( const createRoom = useCallback(
async (name: string, isPublic: boolean, categoryId?: string) => { async (name: string, isPublic: boolean, categoryId?: string) => {
const client = wsClientRef.current; const client = wsClientRef.current;
@ -1222,6 +1301,8 @@ export function RoomProvider({
updateRoom, updateRoom,
deleteRoom, deleteRoom,
streamingMessages: streamingContent, streamingMessages: streamingContent,
projectRepos,
roomAiConfigs,
}), }),
[ [
wsStatus, wsStatus,
@ -1270,6 +1351,8 @@ export function RoomProvider({
updateRoom, updateRoom,
deleteRoom, deleteRoom,
streamingContent, streamingContent,
projectRepos,
roomAiConfigs,
], ],
); );

165
src/lib/mention-ast.ts Normal file
View File

@ -0,0 +1,165 @@
/**
* Mention AST system see architecture spec.
*
* Node types:
* text plain text
* mention @user:, @repository:, @ai: mentions
* ai_action <ai action="..."> structured AI commands (future)
*
* HTML serialization:
* <mention type="user" id="uuid-or-username">label</mention>
* <mention type="repository" id="repo-uid">repo_name</mention>
* <mention type="ai" id="model-uid">model_name</mention>
* <ai action="..."></ai>
*
* Key principle: always use ID for logic, label only for display.
*/
// ─── AST Node Types ──────────────────────────────────────────────────────────
export type MentionMentionType = 'user' | 'repository' | 'ai';
export type TextNode = {
type: 'text';
text: string;
};
export type MentionNode = {
type: 'mention';
mentionType: MentionMentionType;
id: string;
label: string;
};
export type AiActionNode = {
type: 'ai_action';
action: string;
args?: string;
};
export type Node = TextNode | MentionNode | AiActionNode;
export type Document = Node[];
// ─── Serialization (AST → HTML) ───────────────────────────────────────────────
/**
* Serialize a single AST node to HTML string.
*/
function serializeNode(node: Node): string {
if (node.type === 'text') return node.text;
if (node.type === 'ai_action') return `<ai action="${node.action}">${node.args ?? ''}</ai>`;
// mention node
const escapedId = node.id.replace(/"/g, '&quot;');
const escapedLabel = node.label.replace(/</g, '&lt;').replace(/>/g, '&gt;');
return `<mention type="${node.mentionType}" id="${escapedId}">${escapedLabel}</mention>`;
}
/**
* Serialize a document (list of AST nodes) to HTML string.
*/
export function serialize(doc: Document): string {
return doc.map(serializeNode).join('');
}
// ─── Parsing (HTML → AST) ────────────────────────────────────────────────────
// Regex to match <mention type="..." id="...">label</mention>
// Works whether attributes are on one line or spread across lines.
const MENTION_RE =
/<mention\s+type="([^"]+)"\s+id="([^"]+)"[^>]*>\s*([^<]*?)\s*<\/mention>/gi;
// Regex to match <ai action="...">args</ai>
const AI_ACTION_RE = /<ai\s+action="([^"]+)"[^>]*>\s*([^<]*?)\s*<\/ai>/gi;
/**
* Parse an HTML string into an AST document.
* Falls back to a single text node if no structured tags are found.
*/
export function parse(html: string): Document {
if (!html) return [];
const nodes: Document = [];
let lastIndex = 0;
// We interleave all three patterns to find the earliest match.
const matchers: Array<{
re: RegExp;
type: 'mention' | 'ai_action';
}> = [
{ re: MENTION_RE, type: 'mention' },
{ re: AI_ACTION_RE, type: 'ai_action' },
];
// Reset regex lastIndex
for (const m of matchers) m.re.lastIndex = 0;
while (true) {
let earliest: { match: RegExpExecArray; type: 'mention' | 'ai_action' } | null = null;
for (const m of matchers) {
m.re.lastIndex = lastIndex;
const match = m.re.exec(html);
if (match) {
if (!earliest || match.index < earliest.match.index) {
earliest = { match, type: m.type };
}
}
}
if (!earliest) break;
const { match, type } = earliest;
// Text before this match
if (match.index > lastIndex) {
const text = html.slice(lastIndex, match.index);
if (text) nodes.push({ type: 'text', text });
}
if (type === 'mention') {
const mentionType = match[1] as MentionMentionType;
const id = match[2];
const label = match[3] ?? '';
if (
mentionType === 'user' ||
mentionType === 'repository' ||
mentionType === 'ai'
) {
nodes.push({ type: 'mention', mentionType, id, label });
} else {
// Unknown mention type — treat as text
nodes.push({ type: 'text', text: match[0] });
}
} else if (type === 'ai_action') {
const action = match[1];
const args = match[2] ?? '';
nodes.push({ type: 'ai_action', action, args });
}
lastIndex = match.index + match[0].length;
}
// Trailing text
if (lastIndex < html.length) {
const text = html.slice(lastIndex);
if (text) nodes.push({ type: 'text', text });
}
return nodes;
}
// ─── Suggestion value builders ───────────────────────────────────────────────
/**
* Build the HTML mention string from a suggestion selection.
*/
export function buildMentionHtml(
mentionType: MentionMentionType,
id: string,
label: string,
): string {
const escapedId = id.replace(/"/g, '&quot;');
const escapedLabel = label.replace(/</g, '&lt;').replace(/>/g, '&gt;');
return `<mention type="${mentionType}" id="${escapedId}">${escapedLabel}</mention>`;
}