浏览代码

fix: export workflow image

tags/2.0.0-beta.1
zxhlyh 1 个月前
父节点
当前提交
1bf04b7b74

+ 1
- 1
web/app/components/workflow-app/hooks/use-workflow-init.ts 查看文件

@@ -30,7 +30,7 @@ export const useWorkflowInit = () => {
const [data, setData] = useState<FetchWorkflowDraftResponse>()
const [isLoading, setIsLoading] = useState(true)
useEffect(() => {
workflowStore.setState({ appId: appDetail.id })
workflowStore.setState({ appId: appDetail.id, appName: appDetail.name })
}, [appDetail.id, workflowStore])

const handleUpdateWorkflowConfig = useCallback((config: Record<string, any>) => {

+ 2
- 0
web/app/components/workflow-app/store/workflow/workflow-slice.ts 查看文件

@@ -2,6 +2,7 @@ import type { StateCreator } from 'zustand'

export type WorkflowSliceShape = {
appId: string
appName: string
notInitialWorkflow: boolean
setNotInitialWorkflow: (notInitialWorkflow: boolean) => void
nodesDefaultConfigs: Record<string, any>
@@ -11,6 +12,7 @@ export type WorkflowSliceShape = {
export type CreateWorkflowSlice = StateCreator<WorkflowSliceShape>
export const createWorkflowSlice: StateCreator<WorkflowSliceShape> = set => ({
appId: '',
appName: '',
notInitialWorkflow: false,
setNotInitialWorkflow: notInitialWorkflow => set(() => ({ notInitialWorkflow })),
nodesDefaultConfigs: {},

+ 5
- 6
web/app/components/workflow/operator/export-image.tsx 查看文件

@@ -10,7 +10,6 @@ import { useNodesReadOnly } from '../hooks'
import TipPopup from './tip-popup'
import { RiExportLine } from '@remixicon/react'
import cn from '@/utils/classnames'
import { useStore as useAppStore } from '@/app/components/app/store'
import {
PortalToFollowElem,
PortalToFollowElemContent,
@@ -25,14 +24,14 @@ const ExportImage: FC = () => {
const { getNodesReadOnly } = useNodesReadOnly()
const reactFlow = useReactFlow()

const appDetail = useAppStore(s => s.appDetail)
const [open, setOpen] = useState(false)
const [previewUrl, setPreviewUrl] = useState('')
const [previewTitle, setPreviewTitle] = useState('')
const knowledgeName = useStore(s => s.knowledgeName)
const appName = useStore(s => s.appName)

const handleExportImage = useCallback(async (type: 'png' | 'jpeg' | 'svg', currentWorkflow = false) => {
if (!appDetail && !knowledgeName)
if (!appName && !knowledgeName)
return

if (getNodesReadOnly())
@@ -43,6 +42,7 @@ const ExportImage: FC = () => {
if (!flowElement) return

try {
let filename = appName || knowledgeName
const filter = (node: HTMLElement) => {
if (node instanceof HTMLImageElement)
return node.complete && node.naturalHeight !== 0
@@ -51,7 +51,6 @@ const ExportImage: FC = () => {
}

let dataUrl
let filename = `${appDetail.name}`

if (currentWorkflow) {
// Get all nodes and their bounds
@@ -158,7 +157,7 @@ const ExportImage: FC = () => {
// For current view, just download
const link = document.createElement('a')
link.href = dataUrl
link.download = `${appDetail ? filename : knowledgeName}.${type}`
link.download = `${filename}.${type}`
document.body.appendChild(link)
link.click()
document.body.removeChild(link)
@@ -167,7 +166,7 @@ const ExportImage: FC = () => {
catch (error) {
console.error('Export image failed:', error)
}
}, [getNodesReadOnly, appDetail, reactFlow, knowledgeName])
}, [getNodesReadOnly, appName, reactFlow, knowledgeName])

const handleTrigger = useCallback(() => {
if (getNodesReadOnly())

正在加载...
取消
保存