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

View File

@ -55,7 +55,7 @@ class FileManager {
} }
static isDangerFile(file: FileType) { 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) { static getSafePath(file: FileType) {