mirror of
https://github.com/langgenius/dify.git
synced 2026-02-12 04:01:29 -05:00
772dbe620d
Simpler approach: disable the view picker toggle when preview is running, preventing users from switching views during active runs. This replaces the previous mounted ref guard approach (commitsa0188bd9b5,b7f1eb9b7b,8332f0de2b) which added complexity to handle post-unmount operations. Disabling the toggle is more direct and follows KISS principle. Changes: - Add disabled prop to ViewPicker based on isResponding state - Revert mounted ref guards in use-chat-flow-control.ts - Revert isMountedRef parameter in use-nodes/edges-interactions-without-sync.ts - Revert defensive type check in markdown-utils.ts (no longer needed)
96 lines
3.5 KiB
TypeScript
96 lines
3.5 KiB
TypeScript
/**
|
|
* @fileoverview Utility functions for preprocessing Markdown content.
|
|
* These functions were extracted from the main markdown renderer for better separation of concerns.
|
|
* Includes preprocessing for LaTeX and custom "think" tags.
|
|
*/
|
|
import { flow } from 'es-toolkit/compat'
|
|
import { ALLOW_UNSAFE_DATA_SCHEME } from '@/config'
|
|
|
|
export const preprocessLaTeX = (content: string) => {
|
|
if (typeof content !== 'string')
|
|
return content
|
|
|
|
const codeBlockRegex = /```[\s\S]*?```/g
|
|
const codeBlocks = content.match(codeBlockRegex) || []
|
|
const escapeReplacement = (str: string) => str.replace(/\$/g, '_TMP_REPLACE_DOLLAR_')
|
|
let processedContent = content.replace(codeBlockRegex, 'CODE_BLOCK_PLACEHOLDER')
|
|
|
|
processedContent = flow([
|
|
(str: string) => str.replace(/\\\[(.*?)\\\]/g, (_, equation) => `$$${equation}$$`),
|
|
(str: string) => str.replace(/\\\[([\s\S]*?)\\\]/g, (_, equation) => `$$${equation}$$`),
|
|
(str: string) => str.replace(/\\\((.*?)\\\)/g, (_, equation) => `$$${equation}$$`),
|
|
(str: string) => str.replace(/(^|[^\\])\$(.+?)\$/g, (_, prefix, equation) => `${prefix}$${equation}$`),
|
|
])(processedContent)
|
|
|
|
codeBlocks.forEach((block) => {
|
|
processedContent = processedContent.replace('CODE_BLOCK_PLACEHOLDER', escapeReplacement(block))
|
|
})
|
|
|
|
processedContent = processedContent.replace(/_TMP_REPLACE_DOLLAR_/g, '$')
|
|
|
|
return processedContent
|
|
}
|
|
|
|
export const preprocessThinkTag = (content: string) => {
|
|
const thinkOpenTagRegex = /(<think>\s*)+/g
|
|
const thinkCloseTagRegex = /(\s*<\/think>)+/g
|
|
return flow([
|
|
(str: string) => str.replace(thinkOpenTagRegex, '<details data-think=true>\n'),
|
|
(str: string) => str.replace(thinkCloseTagRegex, '\n[ENDTHINKFLAG]</details>'),
|
|
(str: string) => str.replace(/(<\/details>)(?![^\S\r\n]*[\r\n])(?![^\S\r\n]*$)/g, '$1\n'),
|
|
])(content)
|
|
}
|
|
|
|
/**
|
|
* Transforms a URI for use in react-markdown, ensuring security and compatibility.
|
|
* This function is designed to work with react-markdown v9+ which has stricter
|
|
* default URL handling.
|
|
*
|
|
* Behavior:
|
|
* 1. Always allows the custom 'abbr:' protocol.
|
|
* 2. Always allows page-local fragments (e.g., "#some-id").
|
|
* 3. Always allows protocol-relative URLs (e.g., "//example.com/path").
|
|
* 4. Always allows purely relative paths (e.g., "path/to/file", "/abs/path").
|
|
* 5. Allows absolute URLs if their scheme is in a permitted list (case-insensitive):
|
|
* 'http:', 'https:', 'mailto:', 'xmpp:', 'irc:', 'ircs:'.
|
|
* 6. Intelligently distinguishes colons used for schemes from colons within
|
|
* paths, query parameters, or fragments of relative-like URLs.
|
|
* 7. Returns the original URI if allowed, otherwise returns `undefined` to
|
|
* signal that the URI should be removed/disallowed by react-markdown.
|
|
*/
|
|
export const customUrlTransform = (uri: string): string | undefined => {
|
|
const PERMITTED_SCHEME_REGEX = /^(https?|ircs?|mailto|xmpp|abbr):$/i
|
|
|
|
if (uri.startsWith('#'))
|
|
return uri
|
|
|
|
if (uri.startsWith('//'))
|
|
return uri
|
|
|
|
const colonIndex = uri.indexOf(':')
|
|
|
|
if (colonIndex === -1)
|
|
return uri
|
|
|
|
const slashIndex = uri.indexOf('/')
|
|
const questionMarkIndex = uri.indexOf('?')
|
|
const hashIndex = uri.indexOf('#')
|
|
|
|
if (
|
|
(slashIndex !== -1 && colonIndex > slashIndex)
|
|
|| (questionMarkIndex !== -1 && colonIndex > questionMarkIndex)
|
|
|| (hashIndex !== -1 && colonIndex > hashIndex)
|
|
) {
|
|
return uri
|
|
}
|
|
|
|
const scheme = uri.substring(0, colonIndex + 1).toLowerCase()
|
|
if (PERMITTED_SCHEME_REGEX.test(scheme))
|
|
return uri
|
|
|
|
if (ALLOW_UNSAFE_DATA_SCHEME && scheme === 'data:')
|
|
return uri
|
|
|
|
return undefined
|
|
}
|