Unix.wget: added --user-agent option

pveber · Jul 25, 2019 · 52afaec · 52afaec
1 parent 6c9397e
commit 52afaec
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/lib/unix/bistro_unix.ml b/lib/unix/bistro_unix.ml
@@ -5,9 +5,10 @@ open Printf
 let ident x = x
 
 module Cmd = struct
-  let wget ?no_check_certificate ?user ?password ?dest url =
+  let wget ?no_check_certificate ?user_agent ?user ?password ?dest url =
     cmd "wget" [
       option (flag string "--no-check-certificate") no_check_certificate ;
+      option (opt "--user-agent" string) user_agent ;
       option (opt "--user" string) user ;
       option (opt "--password" string) password ;
       option (opt "-O" ident) dest ;
@@ -19,14 +20,14 @@ module Cmd = struct
     seq ~sep:"" [ string "<(gunzip -c " ; dep x ; string ";)" ]
 end
 
-let wget_dyn ?descr_url ?no_check_certificate ?user ?password url =
+let wget_dyn ?descr_url ?no_check_certificate ?user_agent ?user ?password url =
   let info = match descr_url with None -> "" | Some i -> sprintf "(%s)" i in
   Workflow.shell ~descr:("unix.wget" ^ info) [
-    Cmd.wget ?no_check_certificate ?user ?password ~dest url
+    Cmd.wget ?no_check_certificate ?user_agent ?user ?password ~dest url
   ]
 
-let wget ?descr_url ?no_check_certificate ?user ?password url =
-  wget_dyn ?descr_url ?no_check_certificate ?user ?password (Workflow.string url)
+let wget ?descr_url ?no_check_certificate ?user_agent ?user ?password url =
+  wget_dyn ?descr_url ?no_check_certificate ?user_agent ?user ?password (Workflow.string url)
 
 let unzip zip =
   Workflow.shell ~descr:"unix.unzip" [

diff --git a/lib/unix/bistro_unix.mli b/lib/unix/bistro_unix.mli
@@ -3,6 +3,7 @@ open Bistro
 module Cmd : sig
   val wget :
     ?no_check_certificate:bool ->
+    ?user_agent:string ->
     ?user:string ->
     ?password:string ->
     ?dest:Shell_dsl.template ->
@@ -14,13 +15,15 @@ end
 val wget :
   ?descr_url:string ->
   ?no_check_certificate:bool ->
+  ?user_agent:string ->
   ?user:string ->
   ?password:string ->
   string -> #file pworkflow
 
 val wget_dyn :
   ?descr_url:string ->
   ?no_check_certificate:bool ->
+  ?user_agent:string ->
   ?user:string ->
   ?password:string ->
   string workflow -> #file pworkflow