Source file content_length.ml

(* This file is part of Dream, released under the MIT license. See LICENSE.md
   for details, or visit https://github.com/aantron/dream.

   Copyright 2021 Anton Bachin *)



module Message = Dream_pure.Message



(* TODO Also mind Connection: close. *)
(* TODO Test in integration with HTTP/2. *)
(* TODO This could be renamed transfer_encoding at this point. *)
(* Add a Content-Length header to HTTP 1.x responses that have a fixed body but
   don't yet have the header. *)
let content_length next_handler request =
  if fst (Message.version request) <> 1 then
    next_handler request
  else
    let%lwt (response : Message.response) = next_handler request in
    if not (Message.has_header response "Transfer-Encoding") then
      if not (Message.has_header response "Content-Length") then
        Message.add_header response "Transfer-Encoding" "chunked";
    Lwt.return response