Source

ocaml-bitbucket / xcurl.ml

Full commit
let ok200 = function
  | 200, v -> `Ok v
  | n, mes -> `Error (`Http (n, mes))

let get_string f = 
  let h = new Curl.handle in
  f h;
  let buf = Buffer.create 100 in
  h#set_writefunction (fun s -> Buffer.add_string buf s; String.length s);
  h#perform;
  h#cleanup; (* Need to flush out cookies *)
  ok200 (h#get_httpcode, Buffer.contents buf)

let download dst f =
  let h = new Curl.handle in
  f h;
  let tmp = dst ^ ".tmp" in
  let oc = open_out_bin tmp in
  h#set_writefunction (fun s -> 
    output_string oc s; String.length s);
  h#perform;
  h#cleanup; (* Need to flush out cookies *)
  close_out oc;
  (* CR jfuruse: what to do files when code <> 200? *)
  ok200 (h#get_httpcode, Unix.rename tmp dst)