Skip to content

Commit

Permalink
🐛 Fix Download issue
Browse files Browse the repository at this point in the history
  • Loading branch information
@BerkeAras
BerkeAras committed Feb 16, 2022
1 parent 1cd06a7 commit 369c9ec
Show file tree
Hide file tree
Showing 3 changed files with 88 additions and 82 deletions.
162 changes: 86 additions & 76 deletions src/components/KnowledgeBase/KnowledgeBaseFileReader/index.jsx
View file
Original file line number Diff line number Diff line change
Expand Up @@ -14,92 +14,103 @@ const imageExtensions = ['jpg', 'jpeg', 'png', 'gif', 'tiff']

function KnowledgeBaseFileReader(props) {
const { fileId, folderId, historyId } = useParams()
const [isLoading, setIsLoading] = useState(false)
const [file, setFile] = useState(null)
const [fileContent, setFileContent] = useState('')
const [imageUrl, setImageUrl] = useState(null)
const [blobDownload, setBlobDownload] = useState('')
const [modifiedFileTextAreaContent, setModifiedFileTextAreaContent] = useState('')
const [errorNoIndex, setErrorNoIndex] = useState(false)
const [error, setError] = useState(false)
const [isRendered, setIsRendered] = useState(false)

useEffect(() => {
let tokenHeaders = new Headers()
tokenHeaders.append('Authorization', 'Bearer ' + localStorage.getItem('token'))
if (isRendered) {
let tokenHeaders = new Headers()
tokenHeaders.append('Authorization', 'Bearer ' + localStorage.getItem('token'))

let requestOptions = {
method: 'GET',
headers: tokenHeaders,
redirect: 'follow',
}
let requestOptions = {
method: 'GET',
headers: tokenHeaders,
redirect: 'follow',
}

let file_id = fileId
if (historyId != undefined && historyId != null) {
file_id = 'history_' + historyId
}
let file_id = fileId
if (historyId != undefined && historyId != null) {
file_id = 'history_' + historyId
}

fetch(
process.env.REACT_APP_API_URL + `/api/knowledgebase/getFile${file_id == null ? (props.editorMode ? `?file_id=` + file_id : `?folder_id=` + folderId) : `?file_id=` + file_id}`,
requestOptions
)
.then((response) => {
if (response.ok) {
return response.json()
} else {
setError(true)
return
}
})
.then((result) => {
if (result.error) {
if (result.error == 'Index not found') {
setErrorNoIndex(true)
fetch(
process.env.REACT_APP_API_URL + `/api/knowledgebase/getFile${file_id == null ? (props.editorMode ? `?file_id=` + file_id : `?folder_id=` + folderId) : `?file_id=` + file_id}`,
requestOptions
)
.then((response) => {
if (response.ok) {
return response.json()
} else {
setError(true)
return
}
} else {
setFile(result)
props.onFileExtensionChange(result.knowledge_base_file_extension)

if (!result.file_readable) {
fetch(
process.env.REACT_APP_API_URL +
`/api/knowledgebase/readFile${file_id == null ? (props.editorMode ? `?file_id=` + file_id : `?folder_id=` + folderId) : `?file_id=` + file_id}`,
requestOptions
)
.then((response) => {
if (!response.ok) {
setError(true)
return
}
response.blob()
})
.then((blobResult) => {
let url = window.URL.createObjectURL(blobResult)
let a = document.createElement('a')
a.href = url
a.download = result.knowledge_base_file_path
setBlobDownload(url)
document.body.appendChild(a)
a.click()
a.remove()
})
})
.then((result) => {
if (result.error) {
if (result.error == 'Index not found') {
setErrorNoIndex(true)
}
} else {
fetch(
process.env.REACT_APP_API_URL +
`/api/knowledgebase/readFile${file_id == null ? (props.editorMode ? `?file_id=` + file_id : `?folder_id=` + folderId) : `?file_id=` + file_id}`,
requestOptions
)
.then((response) => response.text())
.then((previewResult) => {
props.setModifiedFileContentProp(previewResult)
setFileContent(previewResult)
setFile(result)
props.onFileExtensionChange(result.knowledge_base_file_extension)

if (!result.file_readable) {
// Create a download token first
if (!isLoading) {
setIsLoading(true)
fetch(process.env.REACT_APP_API_URL + `/api/knowledgebase/createDownloadToken?file_id=` + file_id, requestOptions)
.then((response) => {
if (response.ok) {
return response.json()
} else {
setError(true)
return
}
})
.then((result) => {
let token = result.token

if (imageExtensions.includes(result.knowledge_base_file_extension)) {
setImageUrl(previewResult)
}
})
// Download the file
let a = document.createElement('a')
a.href = process.env.REACT_APP_API_URL + `/api/knowledgebase/readFile?file_id=${file_id}&token=${token}`
a.download = result.knowledge_base_file_name + '.' + result.knowledge_base_file_extension
a.click()
setTimeout(() => {
setIsLoading(false)
}, 1000) // Wait a second to prevent duplicate downloads
})
}
} else {
fetch(
process.env.REACT_APP_API_URL +
`/api/knowledgebase/readFile${file_id == null ? (props.editorMode ? `?file_id=` + file_id : `?folder_id=` + folderId) : `?file_id=` + file_id}`,
requestOptions
)
.then((response) => response.text())
.then((previewResult) => {
props.setModifiedFileContentProp(previewResult)
setFileContent(previewResult)

if (imageExtensions.includes(result.knowledge_base_file_extension)) {
setImageUrl(previewResult)
}
})
}
}
}
})
}, [historyId, fileId, folderId])
})
}

if (!isRendered) {
setIsRendered(true)
}
}, [historyId, fileId, folderId, isRendered])

return (
<div className="KnowledgeBaseFileReader">
Expand All @@ -118,14 +129,13 @@ function KnowledgeBaseFileReader(props) {
) : file != null ? (
<>
{!file.file_readable ? (
<center>
<center style={{ margin: '50px 0' }}>
<Zap size={35} strokeWidth={2} />
<h1>The file &quot;{file.knowledge_base_file_name}&quot; is being downloaded...</h1>
<h1>
The file &quot;{file.knowledge_base_file_name}.{file.knowledge_base_file_extension}&quot; is being downloaded...
</h1>
<span>
Download did not start?{' '}
<a href={blobDownload} rel="noreferrer" target="_blank" download={file.knowledge_base_file_path}>
Try again.
</a>
The Download did not start? <a href="?">Try again.</a>
</span>
</center>
) : (
Expand Down
4 changes: 1 addition & 3 deletions src/components/KnowledgeBase/KnowledgeBaseModifyFolderModal/index.jsx
View file
Original file line number Diff line number Diff line change
Expand Up @@ -286,9 +286,7 @@ export default function KnowledgeBaseModifyFolderModal(props) {
}}
>
<List.Content>
<div className="item-avatar">
{!user.user ? <Users size={18} /> : <User size={18} />}
</div>
<div className="item-avatar">{!user.user ? <Users size={18} /> : <User size={18} />}</div>
{!user.user ? 'Global' : user.user.name}
<ChevronRight />
</List.Content>
Expand Down
4 changes: 1 addition & 3 deletions src/components/KnowledgeBase/KnowledgeBaseNewFolderModal/index.jsx
View file
Original file line number Diff line number Diff line change
Expand Up @@ -271,9 +271,7 @@ export default function KnowledgeBaseNewFolderModal(props) {
}}
>
<List.Content>
<div className="item-avatar">
{!user.user ? <Users size={18} /> : <User size={18} />}
</div>
<div className="item-avatar">{!user.user ? <Users size={18} /> : <User size={18} />}</div>
{!user.user ? 'Global' : user.user.name}
<ChevronRight />
</List.Content>
Expand Down

0 comments on commit 369c9ec

Please sign in to comment.