dify/web/hooks/use-async-window-open.ts

import { useCallback } from 'react'

type GetUrl = () => Promise<string | null | undefined>

type AsyncWindowOpenOptions = {
  immediateUrl?: string | null
  target?: string
  features?: string
  onError?: (error: Error) => void
}

export const useAsyncWindowOpen = () =>
  useCallback(async (getUrl: GetUrl, options?: AsyncWindowOpenOptions) => {
    const { immediateUrl, target = '_blank', features, onError } = options ?? {}

    const secureImmediateFeatures = features
      ? `${features},noopener,noreferrer`
      : 'noopener,noreferrer'

    if (immediateUrl) {
      const newWindow = window.open(immediateUrl, target, secureImmediateFeatures)
      if (!newWindow) {
        onError?.(new Error('Failed to open new window'))
        return
      }
      try {
        newWindow.opener = null
      } catch {
        /* noop */
      }
      return
    }

    const newWindow = window.open('about:blank', target, features)
    if (!newWindow) {
      onError?.(new Error('Failed to open new window'))
      return
    }

    try {
      newWindow.opener = null
    } catch {
      /* noop */
    }

    try {
      const url = await getUrl()
      if (url) {
        newWindow.location.href = url
        return
      }
      newWindow.close()
      onError?.(new Error('No url resolved for new window'))
    } catch (error) {
      newWindow.close()
      onError?.(error instanceof Error ? error : new Error(String(error)))
    }
  }, [])