fix: attachment open handler

This commit is contained in:
kangfenmao 2024-09-20 11:38:30 +08:00
parent 4dadf98909
commit 83925832be
2 changed files with 7 additions and 4 deletions

View File

@ -1,5 +1,5 @@
import FileManager from '@renderer/services/file'
import { FileTypes, Message } from '@renderer/types'
import { getFileDirectory } from '@renderer/utils'
import { Image as AntdImage, Upload } from 'antd'
import { FC } from 'react'
import styled from 'styled-components'
@ -9,6 +9,10 @@ interface Props {
}
const MessageAttachments: FC<Props> = ({ message }) => {
if (!message.files) {
return null
}
if (message?.files && message.files[0]?.type === FileTypes.IMAGE) {
return (
<Container>
@ -22,10 +26,9 @@ const MessageAttachments: FC<Props> = ({ message }) => {
<Upload
listType="picture"
disabled
onPreview={(item) => item.url && window.open(getFileDirectory(item.url))}
fileList={message.files?.map((file) => ({
uid: file.id,
url: 'file://' + file.path,
url: 'file://' + FileManager.getSafePath(file),
status: 'done',
name: file.origin_name
}))}

View File

@ -55,7 +55,7 @@ class FileManager {
}
static isDangerFile(file: FileType) {
return ['.sh', '.bat', '.cmd', '.ps1'].includes(file.ext)
return ['.sh', '.bat', '.cmd', '.ps1', '.vbs', 'reg'].includes(file.ext)
}
static getSafePath(file: FileType) {