mirror of
https://github.com/langgenius/dify.git
synced 2026-03-22 14:57:58 +08:00
71 lines
2.2 KiB
TypeScript
71 lines
2.2 KiB
TypeScript
import { FileAppearanceTypeEnum } from '@/app/components/base/file-uploader/types'
|
|
|
|
const MARKDOWN_EXTENSIONS = ['md', 'markdown', 'mdx']
|
|
const CODE_EXTENSIONS = ['json', 'yaml', 'yml', 'toml', 'js', 'jsx', 'ts', 'tsx', 'py', 'schema']
|
|
const TEXT_EXTENSIONS = ['txt', 'log', 'ini', 'env']
|
|
const IMAGE_EXTENSIONS = ['png', 'jpg', 'jpeg', 'gif', 'webp', 'svg', 'bmp', 'ico']
|
|
const VIDEO_EXTENSIONS = ['mp4', 'mov', 'webm', 'mpeg', 'mpg', 'm4v', 'avi']
|
|
const OFFICE_EXTENSIONS = ['pdf', 'doc', 'docx', 'xls', 'xlsx', 'ppt', 'pptx']
|
|
|
|
export const getFileExtension = (name?: string, extension?: string) => {
|
|
if (extension)
|
|
return extension.toLowerCase()
|
|
if (!name)
|
|
return ''
|
|
return name.split('.').pop()?.toLowerCase() ?? ''
|
|
}
|
|
|
|
export const getFileIconType = (name: string) => {
|
|
const extension = name.split('.').pop()?.toLowerCase() ?? ''
|
|
|
|
if (MARKDOWN_EXTENSIONS.includes(extension))
|
|
return FileAppearanceTypeEnum.markdown
|
|
|
|
if (CODE_EXTENSIONS.includes(extension))
|
|
return FileAppearanceTypeEnum.code
|
|
|
|
return FileAppearanceTypeEnum.document
|
|
}
|
|
|
|
export const isMarkdownFile = (extension: string) => MARKDOWN_EXTENSIONS.includes(extension)
|
|
export const isCodeOrTextFile = (extension: string) => CODE_EXTENSIONS.includes(extension) || TEXT_EXTENSIONS.includes(extension)
|
|
export const isImageFile = (extension: string) => IMAGE_EXTENSIONS.includes(extension)
|
|
export const isVideoFile = (extension: string) => VIDEO_EXTENSIONS.includes(extension)
|
|
export const isOfficeFile = (extension: string) => OFFICE_EXTENSIONS.includes(extension)
|
|
|
|
/**
|
|
* Get Monaco editor language from file name extension
|
|
*/
|
|
export const getFileLanguage = (name: string): string => {
|
|
const extension = name.split('.').pop()?.toLowerCase() ?? ''
|
|
|
|
const languageMap: Record<string, string> = {
|
|
// Markdown
|
|
md: 'markdown',
|
|
markdown: 'markdown',
|
|
mdx: 'markdown',
|
|
// JSON
|
|
json: 'json',
|
|
jsonl: 'json',
|
|
// YAML
|
|
yaml: 'yaml',
|
|
yml: 'yaml',
|
|
// JavaScript/TypeScript
|
|
js: 'javascript',
|
|
jsx: 'javascript',
|
|
ts: 'typescript',
|
|
tsx: 'typescript',
|
|
// Python
|
|
py: 'python',
|
|
// Others
|
|
html: 'html',
|
|
css: 'css',
|
|
xml: 'xml',
|
|
sql: 'sql',
|
|
sh: 'shell',
|
|
bash: 'shell',
|
|
}
|
|
|
|
return languageMap[extension] ?? 'plaintext'
|
|
}
|