Anri/pusk
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

Download now works

Browse source
This commit is contained in:
Mylloon 2023-05-12 18:40:47 +02:00
parent 3d392c1063
commit 837b885d27
Signed by: Anri
GPG key ID: A82D63DFF8D1317F
1 changed files with 20 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

25
bin/main.ml
View file

@ -4,15 +4,30 @@ open Cohttp_lwt_unix
let fmt = Printf.sprintf let fmt = Printf.sprintf
let download uri dest = let rec download uri dest =
let* response, body = Client.get uri in let* response, body = Client.get uri in
let status = Response.status response in let status = Response.status response in
if Cohttp.Code.code_of_status status = 302 let code = Cohttp.Code.code_of_status status in
if Cohttp.Code.is_redirection code
then ( then (
print_endline (fmt "\nDownloading %s ..." (Uri.to_string uri)); let headers = Response.headers response in
match Cohttp.Header.get headers "location" with
| Some url ->
let uri = Uri.of_string url in
let redirect_url = Uri.resolve "" uri uri in
download redirect_url dest
| None -> Lwt.fail_with "Redirect location not found")
else if Cohttp.Code.is_success code
then (
print_endline "Downloading...";
let stream = Body.to_stream body in let stream = Body.to_stream body in
Lwt_io.with_file ~mode:Lwt_io.output dest (fun chan -> let res =
Lwt_stream.iter_s (Lwt_io.write chan) stream)) Lwt_io.with_file ~mode:Lwt_io.output dest (fun chan ->
Lwt_stream.iter_s (Lwt_io.write chan) stream)
in
let* () = res in
print_endline "Download done!";
Lwt.return_unit)
else else
Lwt.fail_with Lwt.fail_with
("Failed to download file. HTTP status: " ^ Cohttp.Code.string_of_status status) ("Failed to download file. HTTP status: " ^ Cohttp.Code.string_of_status status)