Hesham/viddl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: Set content length header for comletion and time estiamte on

Browse Source 
browsers
This commit is contained in:
HeshamTB 2023-09-15 18:48:45 +03:00
parent aef6deb8d6
commit 9532b70458
1 changed files with 9 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
main.go
View File

@ -294,20 +294,22 @@ func main() {
if dataRequest.StatusCode != 200 { if dataRequest.StatusCode != 200 {
log.Println("Failed to get content for URL", userURL) log.Println("Failed to get content for URL", userURL)
return
} }
contentLength := dataRequest.Header.Get("Content-Length")
w.Header().Set(
"Content-Disposition",
fmt.Sprintf("attachment;filename=%s", filename),
)
w.WriteHeader(206)
if dataRequest.ContentLength == 0 { if dataRequest.ContentLength == 0 {
log.Println("Empty body from content url") log.Println("Empty body from content url")
w.WriteHeader(500) w.WriteHeader(500)
return return
} }
w.Header().Set(
"Content-Disposition",
fmt.Sprintf("attachment;filename=%s;", filename),
)
w.Header().Set("Content-Length", contentLength)
w.WriteHeader(206)
n, err := io.Copy(w, dataRequest.Body) n, err := io.Copy(w, dataRequest.Body)
if err != nil { if err != nil {
log.Println(err.Error()) log.Println(err.Error())