web/src/hooks/useDownloadRawFile.ts - main - githubmirror/gitness

web/src/hooks/useDownloadRawFile.ts
70 строк2 KB
Johannes Batzill
[OPEN-SOURCE] Prepare Repository (#605)
24 сен 2023, 10:33
24 сен 2023, 10:33a83cd21
/*
 * Copyright 2023 Harness, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import { useCallback, useEffect, useState } from 'react'
import { useToaster } from '@harnessio/uicore'
import { useGet } from 'restful-react'
import { getErrorMessage } from 'utils/Utils'
import type { GitInfoProps } from 'utils/GitUtils'

interface UseDownloadRawFileParams extends Pick<GitInfoProps, 'repoMetadata' | 'gitRef' | 'resourcePath'> {
  filename: string
}

export function useDownloadRawFile() {
  const { error, response, refetch } = useGet({
    path: '',
    lazy: true
  })
  const [name, setName] = useState('')
  const { showError } = useToaster()
  const callback = useCallback(async () => {
    if (response) {
      const imageBlog = await response.blob()
      const imageURL = URL.createObjectURL(imageBlog)

      const anchor = document.createElement('a')
      anchor.href = imageURL
      anchor.download = name

      document.body.appendChild(anchor)
      anchor.click()
      // Cleaning up requires a timeout to work under Firefox
      setTimeout(() => {
        document.body.removeChild(anchor)
        URL.revokeObjectURL(imageURL)
      }, 100)
      return { status: true }
    }
  }, [name, response])

  useEffect(() => {
    if (error) {
      showError(getErrorMessage(error))
    } else if (response) {
      callback()
    }
  }, [error, showError, response, callback])

  return useCallback(
    ({ repoMetadata, resourcePath, gitRef, filename = 'download' }: UseDownloadRawFileParams) => {
      const rawURL = `/api/v1/repos/${repoMetadata?.path}/+/raw/${resourcePath}`
      setName(filename)
      refetch({ path: rawURL, queryParams: { git_ref: gitRef } })
    },
    [refetch]
  )
}