docuservix/widgets/chat: добавление поддержки markdown

docusaurus.config.ts

@@ -35,7 +35,7 @@ const config: Config = {
     markdown: {
         mermaid: true,
     },
-    plugins: [docuservix()],
+    plugins: [docuservix],
     themes: ['@docusaurus/theme-mermaid'],
 
     // Future flags, see https://docusaurus.io/docs/api/docusaurus-config#future

plugins/docuservix/entities/markdown/MD.module.css

@@ -1,59 +0,0 @@
-.MD p:last-child {
-    margin-bottom: 0;
-}
-
-.MD p:first-child {
-    margin-top: 0;
-}
-
-.MD code {
-    padding: 0.15em 0.4em;
-    border-radius: 4px;
-    background: var(--ifm-color-emphasis-200);
-    font-size: 0.85em;
-}
-
-.MD pre {
-    margin: 0.5em 0;
-    padding: 0.75em;
-    overflow-x: auto;
-    border-radius: 6px;
-    background: var(--ifm-color-emphasis-100);
-}
-
-.MD pre code {
-    padding: 0;
-    background: none;
-}
-
-.MD ul,
-.MD ol {
-    padding-left: 1.5em;
-    margin: 0.5em 0;
-}
-
-.MD table {
-    width: 100%;
-    margin: 0.5em 0;
-    border-collapse: collapse;
-    font-size: 0.9em;
-}
-
-.MD th,
-.MD td {
-    padding: 0.4em 0.75em;
-    border: 1px solid var(--ifm-color-emphasis-300);
-    text-align: left;
-}
-
-.MD th {
-    background: var(--ifm-color-emphasis-100);
-    font-weight: var(--ifm-font-weight-semibold);
-}
-
-.MD blockquote {
-    margin: 0.5em 0;
-    padding: 0.25em 1em;
-    border-left: 3px solid var(--ifm-color-emphasis-300);
-    color: var(--ifm-color-emphasis-700);
-}

plugins/docuservix/entities/markdown/MD.tsx

@@ -1,17 +0,0 @@
-import React, { ReactNode } from 'react';
-import Markdown from 'react-markdown';
-import remarkGfm from 'remark-gfm';
-
-import styles from './MD.module.css';
-
-interface MDProps {
-    children: string;
-}
-
-export function MD({ children }: MDProps): ReactNode {
-    return (
-        <div className={styles.MD}>
-            <Markdown remarkPlugins={[remarkGfm]}>{children}</Markdown>
-        </div>
-    );
-}

plugins/docuservix/entities/markdown/index.ts

@@ -1 +0,0 @@
-export { MD } from './MD';

plugins/docuservix/hooks/useChat.ts

@@ -1,88 +0,0 @@
-import { useLocation } from '@docusaurus/router';
-import { useCallback, useEffect, useRef, useState } from 'react';
-
-import { useOptions } from '@docuservix/hooks/useOptions';
-import { IChat, IChatMessage, IChatSource } from '@docuservix/models/chat';
-
-interface UseChatResult {
-    dialog: IChat;
-    typing: boolean;
-    statusMessage?: string;
-    sendMessage: (text: string) => void;
-}
-
-function useQuery(): string {
-    const location = useLocation();
-    const params = new URLSearchParams(location.search);
-
-    return params.get('q') ?? '';
-}
-
-export function useChat(): UseChatResult {
-    const chatEndpoint = useOptions().api + '/v1/chat';
-    const urlQuery = useQuery();
-
-    const [messages, setMessages] = useState<IChatMessage[]>([]);
-    const [loading, setLoading] = useState(false);
-    const [error, setError] = useState<string | null>(null);
-
-    const initialSentRef = useRef(false);
-    const messagesEndRef = useRef(messages);
-
-    messagesEndRef.current = messages;
-
-    const sendMessage = useCallback(
-        async (text: string) => {
-            const content = text.trim();
-
-            if (!content) {
-                return;
-            }
-
-            const userMessage: IChatMessage = { role: 'user', content };
-            const newHistory = [...messagesEndRef.current, userMessage];
-
-            setMessages(newHistory);
-            setLoading(true);
-            setError(null);
-
-            try {
-                const res = await fetch(chatEndpoint, {
-                    method: 'POST',
-                    headers: { 'Content-Type': 'application/json' },
-                    body: JSON.stringify({ messages: newHistory }),
-                });
-
-                if (!res.ok) {
-                    throw new Error(`HTTP ${res.status}`);
-                }
-
-                const data: { answer: string; sources?: IChatSource[] } = await res.json();
-
-                setMessages((prev) => [
-                    ...prev,
-                    { role: 'assistant', content: data.answer, sources: data.sources },
-                ]);
-            } catch (err) {
-                setError(err instanceof Error ? err.message : 'Ошибка при обращении к серверу');
-            } finally {
-                setLoading(false);
-            }
-        },
-        [chatEndpoint],
-    );
-
-    useEffect(() => {
-        if (urlQuery && !initialSentRef.current) {
-            initialSentRef.current = true;
-            sendMessage(urlQuery);
-        }
-    }, [urlQuery, sendMessage]);
-
-    return {
-        dialog: { messages },
-        typing: loading,
-        statusMessage: error ?? undefined,
-        sendMessage,
-    };
-}

plugins/docuservix/hooks/useChatStatus.ts

@@ -1,119 +0,0 @@
-import { useCallback, useEffect, useRef, useState } from 'react';
-
-import { useOptions } from '@docuservix/hooks/useOptions';
-import { ChatStatus } from '@docuservix/models/chat';
-
-interface UseChatStatusResult {
-    status: ChatStatus;
-    apiUrl: string;
-    recheck: () => void;
-}
-
-const POLL_INTERVAL_MS = 30_000;
-const REQUEST_TIMEOUT_MS = 5_000;
-
-function sameStatus(a: ChatStatus, b: ChatStatus): boolean {
-    if (a.kind !== b.kind) return false;
-    if (a.kind === 'server' && b.kind === 'server') return a.label === b.label;
-    return true;
-}
-
-export function useChatStatus(): UseChatStatusResult {
-    const apiUrl = useOptions().api ?? '';
-    const statusApiUrl = apiUrl + '/v1/status';
-
-    const [status, setStatus] = useState<ChatStatus>({ kind: 'connecting' });
-
-    const abortRef = useRef<AbortController | null>(null);
-    const timerRef = useRef<ReturnType<typeof setTimeout> | null>(null);
-
-    const updateStatus = useCallback((next: ChatStatus): void => {
-        setStatus((prev) => (sameStatus(prev, next) ? prev : next));
-    }, []);
-
-    const check = useCallback(async (): Promise<void> => {
-        abortRef.current?.abort();
-
-        const controller = new AbortController();
-
-        abortRef.current = controller;
-
-        let timedOut = false;
-        const timeoutId = setTimeout(() => {
-            timedOut = true;
-            controller.abort();
-        }, REQUEST_TIMEOUT_MS);
-
-        try {
-            const res = await fetch(statusApiUrl, { signal: controller.signal });
-
-            if (!res.ok) {
-                updateStatus({ kind: 'offline' });
-                return;
-            }
-
-            const parsed = (await res.json()) as { status?: unknown } | null | undefined;
-            const raw = typeof parsed?.status === 'string' ? parsed.status : null;
-
-            if (raw === null || raw.length === 0) {
-                updateStatus({ kind: 'offline' });
-                return;
-            }
-
-            updateStatus({
-                kind: 'server',
-                label: raw.charAt(0).toUpperCase() + raw.slice(1),
-            });
-        } catch {
-            if (!controller.signal.aborted || timedOut) {
-                updateStatus({ kind: 'offline' });
-            }
-        } finally {
-            clearTimeout(timeoutId);
-        }
-    }, [statusApiUrl, updateStatus]);
-
-    const scheduleNext = useCallback((): void => {
-        if (timerRef.current !== null) {
-            clearTimeout(timerRef.current);
-        }
-
-        timerRef.current = setTimeout(async () => {
-            if (document.visibilityState === 'visible') {
-                await check();
-            }
-            scheduleNext();
-        }, POLL_INTERVAL_MS);
-    }, [check]);
-
-    const recheck = useCallback((): void => {
-        updateStatus({ kind: 'connecting' });
-        void check();
-        scheduleNext();
-    }, [check, scheduleNext, updateStatus]);
-
-    useEffect(() => {
-        void check();
-        scheduleNext();
-
-        const onVisibilityChange = (): void => {
-            if (document.visibilityState === 'visible') {
-                void check();
-                scheduleNext();
-            }
-        };
-
-        document.addEventListener('visibilitychange', onVisibilityChange);
-
-        return (): void => {
-            document.removeEventListener('visibilitychange', onVisibilityChange);
-            abortRef.current?.abort();
-
-            if (timerRef.current !== null) {
-                clearTimeout(timerRef.current);
-            }
-        };
-    }, [check, scheduleNext]);
-
-    return { status, apiUrl, recheck };
-}

plugins/docuservix/hooks/useOptions.ts

@@ -1,7 +0,0 @@
-import { usePluginData } from '@docusaurus/useGlobalData';
-
-import { DocuservixOptions } from '@docuservix/models/docuservix';
-
-export function useOptions(): DocuservixOptions {
-    return usePluginData('docuservix') as DocuservixOptions;
-}

plugins/docuservix/index.ts

@@ -2,12 +2,7 @@ import path from 'path';
 
 import type { LoadContext, Plugin } from '@docusaurus/types';
 
-import { DocuservixOptions } from '@docuservix/models/docuservix';
+function pluginDocuservix(_context: LoadContext): Plugin {
-
-export default function docuservix(options: Partial<DocuservixOptions> = {}) {
-    const api = process.env.DOCUSERVIX_API || options.api || '/api';
-
-    return function pluginDocuservix(_context: LoadContext): Plugin {
     return {
         name: 'docuservix',
 
@@ -22,11 +17,7 @@ export default function docuservix(options: Partial<DocuservixOptions> = {}) {
         },
 
         async contentLoaded({ actions }) {
-                const { addRoute, setGlobalData } = actions;
+            const { addRoute } = actions;
-
-                setGlobalData({
-                    api,
-                });
 
             addRoute({
                 path: '/chat',
@@ -35,5 +26,6 @@ export default function docuservix(options: Partial<DocuservixOptions> = {}) {
             });
         },
     };
-    };
 }
+
+export default pluginDocuservix;

plugins/docuservix/models/chat.ts

@@ -1,42 +1,8 @@
-export type ChatStatus =
-    | { kind: 'connecting' }
-    | { kind: 'offline' }
-    | { kind: 'server'; label: string };
-
 export interface IChat {
     messages: IChatMessage[];
 }
 
-export interface IChatSource {
-    file: string;
-    heading: string;
-    anchor: string;
-    score: number;
-}
-
 export interface IChatMessage {
     role: 'user' | 'assistant';
     content: string;
-    sources?: IChatSource[];
-}
-
-function stripNumericPrefixes(p: string): string {
-    return p
-        .split('/')
-        .map((seg) => seg.replace(/^\d+-/, ''))
-        .join('/');
-}
-
-export function sourceToUrl(file: string, anchor: string): string {
-    let p = file.replace(/^docs\//, '').replace(/\.md$/, '');
-
-    p = stripNumericPrefixes(p);
-
-    return `/docs/${p}${anchor ? `#${anchor}` : ''}`;
-}
-
-export function sourceToPath(file: string): string {
-    const p = file.replace(/^docs\//, '').replace(/\.md$/, '');
-
-    return stripNumericPrefixes(p);
 }

plugins/docuservix/models/docuservix.ts

@@ -1,3 +0,0 @@
-export interface DocuservixOptions {
-    api?: string;
-}

plugins/docuservix/pages/chat/ChatPage.tsx

@@ -1,20 +1,30 @@
 import Layout from '@theme/Layout';
 import { ReactNode } from 'react';
 
-import { useChat } from '@docuservix/hooks/useChat';
+import { IChat } from '@docuservix/models/chat';
 import { Chat } from '@docuservix/widgets/chat';
 
-export function ChatPage(): ReactNode {
+const dialog: IChat = {
-    const { dialog, typing, statusMessage, sendMessage } = useChat();
+    messages: [
+        {
+            role: 'user',
+            content: 'Can you show me some CSS animations? It can be simple tools like chatbots...',
+        },
+        {
+            role: 'assistant',
+            content: "Hello! I'm your **AI assistant**. How can I help you today?",
+        },
+    ],
+};
 
+export function ChatPage(): ReactNode {
     return (
         <Layout title="Чат">
             <main className="container margin-vert--lg">
                 <Chat
                     dialog={dialog}
-                    typing={typing}
+                    statusMessage="Unable to connect to the server"
-                    statusMessage={statusMessage}
+                    typing
-                    onSend={sendMessage}
                 />
             </main>
         </Layout>

plugins/docuservix/widgets/chat/Chat.tsx

@@ -29,7 +29,7 @@ export function Chat({ dialog, typing, statusMessage, onSend }: ChatProps): Reac
             />
             {statusMessage && <div className={b('statusMessage')}>{statusMessage}</div>}
             <Input
-                disabled={typing}
+                loading={typing}
                 onSend={onSend}
             />
         </div>

plugins/docuservix/widgets/chat/Header.module.css

@@ -24,3 +24,9 @@
     color: var(--ifm-font-color-base);
     font-size: 1.125rem;
 }
+
+.Header__info p {
+    margin: 0;
+    color: var(--ifm-color-emphasis-600);
+    font-size: 0.85rem;
+}

plugins/docuservix/widgets/chat/Header.tsx

@@ -3,7 +3,6 @@ import React, { ReactNode } from 'react';
 
 import styles from './Header.module.css';
 import { RobotIcon } from './icons';
-import { Status } from './Status';
 
 const b = block(styles, 'Header');
 
@@ -15,7 +14,7 @@ export function Header(): ReactNode {
             </div>
             <div className={b('info')}>
                 <h3>AI Assistant</h3>
-                <Status />
+                <p>Ready to help</p>
             </div>
         </div>
     );

plugins/docuservix/widgets/chat/Input.tsx

@@ -7,17 +7,17 @@ import styles from './Input.module.css';
 const b = block(styles, 'Input');
 
 interface InputProps {
-    disabled?: boolean;
+    loading?: boolean;
     onSend?: (text: string) => void;
 }
 
-export function Input({ disabled, onSend }: InputProps): ReactNode {
+export function Input({ loading, onSend }: InputProps): ReactNode {
     const [input, setInput] = useState('');
 
     const handleSend = () => {
         const text = input.trim();
 
-        if (!text || disabled) {
+        if (!text || loading) {
             return;
         }
 
@@ -41,12 +41,12 @@ export function Input({ disabled, onSend }: InputProps): ReactNode {
                 onKeyDown={handleKeyDown}
                 placeholder="Type your message here..."
                 rows={1}
-                disabled={disabled}
+                disabled={loading}
             />
             <button
                 className={b('send')}
                 onClick={handleSend}
-                disabled={disabled || !input.trim()}
+                disabled={loading || !input.trim()}
             >
                 <PaperPlaneIcon />
             </button>

plugins/docuservix/widgets/chat/Message.module.css

@@ -1,5 +1,5 @@
 .Message {
-    max-width: 90%;
+    max-width: 85%;
     animation: slideIn 0.3s ease-out;
 }
 
@@ -34,37 +34,94 @@
     border-bottom-right-radius: 0.25rem;
 }
 
-.Message__sources {
+/* Markdown typography */
-    margin-top: 8px;
+
-    padding: 8px 1rem 0;
+.Message__content p:last-child {
-    border-top: 1px solid var(--ifm-color-emphasis-200);
+    margin-bottom: 0;
 }
 
-.Message__sourcesLabel {
+.Message__content p:first-child {
-    margin-bottom: 4px;
+    margin-top: 0;
-    color: var(--ifm-color-emphasis-600);
+}
+
+.Message__content code {
+    padding: 0.15em 0.4em;
+    border-radius: 4px;
+    background: var(--ifm-color-emphasis-200);
+    font-size: 0.85em;
+}
+
+.Message__content pre {
+    margin: 0.5em 0;
+    padding: 0.75em;
+    overflow-x: auto;
+    border-radius: 6px;
+    background: var(--ifm-color-emphasis-100);
+}
+
+.Message__content pre code {
+    padding: 0;
+    background: none;
+}
+
+.Message__content ul,
+.Message__content ol {
+    padding-left: 1.5em;
+    margin: 0.5em 0;
+}
+
+.Message__content table {
+    width: 100%;
+    margin: 0.5em 0;
+    border-collapse: collapse;
+    font-size: 0.9em;
+}
+
+.Message__content th,
+.Message__content td {
+    padding: 0.4em 0.75em;
+    border: 1px solid var(--ifm-color-emphasis-300);
+    text-align: left;
+}
+
+.Message__content th {
+    background: var(--ifm-color-emphasis-100);
     font-weight: var(--ifm-font-weight-semibold);
-    font-size: 0.75rem;
-    letter-spacing: 0.04em;
-    text-transform: uppercase;
 }
 
-.Message__sourceLink {
+.Message__content blockquote {
-    display: block;
+    margin: 0.5em 0;
-    overflow: hidden;
+    padding: 0.25em 1em;
-    color: var(--ifm-color-primary);
+    border-left: 3px solid var(--ifm-color-emphasis-300);
-    font-size: 0.8rem;
+    color: var(--ifm-color-emphasis-700);
-    white-space: nowrap;
-    text-decoration: none;
-    text-overflow: ellipsis;
 }
 
-.Message__sourceLink:hover {
+/* User role overrides for light-on-dark text */
-    text-decoration: underline;
+
+.Message_role_user .Message__content code {
+    background: rgba(255, 255, 255, 0.2);
 }
 
-@media (max-width: 576px) {
+.Message_role_user .Message__content pre {
+    background: rgba(255, 255, 255, 0.15);
+}
+
+.Message_role_user .Message__content th,
+.Message_role_user .Message__content td {
+    border-color: rgba(255, 255, 255, 0.3);
+}
+
+.Message_role_user .Message__content th {
+    background: rgba(255, 255, 255, 0.15);
+}
+
+.Message_role_user .Message__content blockquote {
+    border-left-color: rgba(255, 255, 255, 0.4);
+    color: rgba(255, 255, 255, 0.85);
+}
+
+@media (min-width: 576px) {
     .Message {
-        max-width: 100%;
+        max-width: 75%;
     }
 }

plugins/docuservix/widgets/chat/Message.tsx

@@ -1,10 +1,7 @@
-import Link from '@docusaurus/Link';
 import block from 'bem-css-modules';
 import React, { ReactNode } from 'react';
-
+import Markdown from 'react-markdown';
-import { MD } from '@docuservix/entities/markdown';
+import remarkGfm from 'remark-gfm';
-
-import { IChatSource, sourceToPath, sourceToUrl } from '@docuservix/models/chat';
 
 import styles from './Message.module.css';
 
@@ -13,30 +10,14 @@ const b = block(styles, 'Message');
 interface MessageProps {
     role: 'user' | 'assistant';
     content: string;
-    sources?: IChatSource[];
 }
 
-export function Message({ role, content, sources }: MessageProps): ReactNode {
+export function Message({ role, content }: MessageProps): ReactNode {
     return (
         <div className={b({ role })}>
             <div className={b('content')}>
-                <MD>{content}</MD>
+                <Markdown remarkPlugins={[remarkGfm]}>{content}</Markdown>
             </div>
-
-            {sources && sources.length > 0 && (
-                <div className={b('sources')}>
-                    <div className={b('sourcesLabel')}>Источники:</div>
-                    {sources.map((src, j) => (
-                        <Link
-                            key={j}
-                            to={sourceToUrl(src.file, src.anchor)}
-                            className={b('sourceLink')}
-                        >
-                            {src.heading || sourceToPath(src.file)}
-                        </Link>
-                    ))}
-                </div>
-            )}
         </div>
     );
 }

plugins/docuservix/widgets/chat/Messages.tsx

@@ -21,7 +21,6 @@ export function Messages({ messages, typing }: MessagesProps): ReactNode {
                     key={i}
                     role={msg.role}
                     content={msg.content}
-                    sources={msg.sources}
                 />
             ))}
 

plugins/docuservix/widgets/chat/Status.module.css

@@ -1,53 +0,0 @@
-.Status {
-    display: inline-flex;
-    align-items: center;
-    gap: 0.4rem;
-    padding: 0;
-    background: none;
-    border: none;
-    cursor: pointer;
-    color: var(--ifm-color-emphasis-600);
-    font-size: 0.85rem;
-    line-height: 1;
-}
-
-.Status:hover .Status__label {
-    color: var(--ifm-color-emphasis-800);
-}
-
-.Status:focus-visible {
-    outline: 2px solid var(--ifm-color-primary);
-    outline-offset: 2px;
-    border-radius: 4px;
-}
-
-.Status__dot {
-    width: 8px;
-    height: 8px;
-    border-radius: 50%;
-    background: var(--ifm-color-emphasis-400);
-    flex-shrink: 0;
-}
-
-.Status_kind_server .Status__dot {
-    background: #22c55e;
-}
-
-.Status_kind_offline .Status__dot {
-    background: #ef4444;
-}
-
-.Status_kind_connecting .Status__dot {
-    background: #eab308;
-    animation: Status__pulse 1s ease-in-out infinite;
-}
-
-@keyframes Status__pulse {
-    0%,
-    100% {
-        opacity: 1;
-    }
-    50% {
-        opacity: 0.4;
-    }
-}

plugins/docuservix/widgets/chat/Status.tsx

@@ -1,38 +0,0 @@
-import block from 'bem-css-modules';
-import React, { ReactNode } from 'react';
-
-import { useChatStatus } from '@docuservix/hooks/useChatStatus';
-import { ChatStatus } from '@docuservix/models/chat';
-
-import styles from './Status.module.css';
-
-const b = block(styles, 'Status');
-
-function getLabel(status: ChatStatus): string {
-    switch (status.kind) {
-        case 'connecting':
-            return 'Подключение…';
-        case 'offline':
-            return 'Offline';
-        case 'server':
-            return status.label;
-    }
-}
-
-export function Status(): ReactNode {
-    const { status, apiUrl, recheck } = useChatStatus();
-    const label = getLabel(status);
-
-    return (
-        <button
-            type="button"
-            className={b({ kind: status.kind })}
-            onClick={recheck}
-            title={apiUrl}
-            aria-label={`Статус: ${label}. Нажмите для проверки`}
-        >
-            <span className={b('dot')} />
-            <span className={b('label')}>{label}</span>
-        </button>
-    );
-}