X-Git-Url: http://matita.cs.unibo.it/gitweb/?a=blobdiff_plain;f=helm%2Focaml%2Fgetter%2Fhttp_getter_common.ml;h=c85f680c28fa49fe7eb87aeb78eeacc90a5576bd;hb=52fdcda3e0083391fa04a064f3e07279d975d5ba;hp=15810b93c2b4b8fb535c187eda61ae6eed1785fb;hpb=f693e2de79c1a95422a5efd22acbcea5003eeaba;p=helm.git

diff --git a/helm/ocaml/getter/http_getter_common.ml b/helm/ocaml/getter/http_getter_common.ml
index 15810b93c..c85f680c2 100644
--- a/helm/ocaml/getter/http_getter_common.ml
+++ b/helm/ocaml/getter/http_getter_common.ml
@@ -112,9 +112,9 @@ let patch_xml ?via_http ?xmlbases () =
           (patch_doctype ?via_http () (patch_entity ?via_http () line))
 
 let return_file
-  ~fname ?contype ?contenc
-  ?(patch_fun = fun x -> x) ?(gunzip = false) ?(via_http = true) outchan
-  =
+  ~fname ?contype ?contenc ?patch_fun ?(gunzip = false) ?(via_http = true)
+  ~enc outchan
+=
   let headers =
     match (contype, contenc) with
     | (Some t, Some e) -> ["Content-Encoding", e; "Content-Type", t]
@@ -127,32 +127,40 @@ let return_file
     Http_daemon.send_headers headers outchan;
     Http_daemon.send_CRLF outchan
   end;
-  if gunzip then begin  (* gunzip needed, uncompress file, apply patch_fun to
-                        it, compress the result and sent it to client *)
-    let (tmp1, tmp2) =
-      (Http_getter_misc.tempfile (), Http_getter_misc.tempfile ())
-    in
-    try
-      Http_getter_misc.gunzip ~keep:true ~output:tmp1 fname;(* gunzip to tmp1 *)
-      let new_file = open_out tmp2 in
-      Http_getter_misc.iter_file  (* tmp2 = patch(tmp1) *)
-        (fun line ->
-          output_string new_file (patch_fun line ^ "\n");
-          flush outchan)
-        tmp1;
-      close_out new_file;
-      Http_getter_misc.gzip ~output:tmp1 tmp2;  (* tmp1 = gzip(tmp2); rm tmp2 *)
-      Http_getter_misc.iter_file  (* send tmp1 to client as is*)
-        (fun line -> output_string outchan (line ^ "\n"); flush outchan)
-        tmp1;
-      Sys.remove tmp1       (* rm tmp1 *)
-    with e ->
-      Sys.remove tmp1;
-      raise e
-  end else begin (* no need to gunzip, apply patch_fun directly to file *)
-    Http_getter_misc.iter_file
-      (fun line -> output_string outchan (patch_fun line ^ "\n"); flush outchan)
-      fname;
-  end
+  match gunzip, patch_fun with
+  | true, Some patch_fun ->
+      Http_getter_logger.log ~level:2
+        "Patch required, uncompress/compress cycle needed :-(";
+      (* gunzip needed, uncompress file, apply patch_fun to it, compress the
+       * result and sent it to client *)
+      let (tmp1, tmp2) =
+        (Http_getter_misc.tempfile (), Http_getter_misc.tempfile ())
+      in
+      (try
+        Http_getter_misc.gunzip ~keep:true ~output:tmp1 fname; (* gunzip tmp1 *)
+        let new_file = open_out tmp2 in
+        Http_getter_misc.iter_file  (* tmp2 = patch(tmp1) *)
+          (fun line ->
+            output_string new_file (patch_fun line ^ "\n");
+            flush outchan)
+          tmp1;
+        close_out new_file;
+        Http_getter_misc.gzip ~output:tmp1 tmp2;(* tmp1 = gzip(tmp2); rm tmp2 *)
+        Http_getter_misc.iter_file  (* send tmp1 to client as is*)
+          (fun line -> output_string outchan (line ^ "\n"); flush outchan)
+          tmp1;
+        Sys.remove tmp1       (* rm tmp1 *)
+      with e ->
+        Sys.remove tmp1;
+        raise e)
+  | false, Some patch_fun ->
+      (match enc with
+      | `Normal ->
+          Http_getter_misc.iter_file
+            (fun line -> output_string outchan (patch_fun (line ^ "\n")))
+            fname
+      | `Gzipped -> assert false)
+        (* dangerous case, if this happens it needs to be investigated *)
+  | _, None -> Http_getter_misc.iter_file_data (output_string outchan) fname
 ;;