From 85306e4e7b6fe796d113d1e78e07f135f7a6dd39 Mon Sep 17 00:00:00 2001
From: Rudi Grinberg <rudi.grinberg@gmail.com>
Date: Fri, 6 Feb 2015 13:48:22 -0500
Subject: [PATCH] Add location accessor for headers

---
 lib/header.ml  | 5 +++++
 lib/header.mli | 1 +
 2 files changed, 6 insertions(+)

diff --git a/lib/header.ml b/lib/header.ml
index 6f84439d79..9722af2e65 100644
--- a/lib/header.ml
+++ b/lib/header.ml
@@ -193,6 +193,11 @@ let get_authorization headers =
 let is_form headers =
   get_media_type headers = (Some "application/x-www-form-urlencoded")
 
+let get_location headers =
+  match get headers "location" with
+  | None -> None
+  | Some u -> Some (Uri.of_string u)
+
 let prepend_user_agent headers user_agent =
   let k = "user-agent" in
   match get headers k with
diff --git a/lib/header.mli b/lib/header.mli
index 02f2a19eec..1ca096ce5f 100644
--- a/lib/header.mli
+++ b/lib/header.mli
@@ -75,6 +75,7 @@ val add_authorization : t -> Auth.credential -> t
 val get_authorization : t -> Auth.credential option
 val add_authorization_req : t -> Auth.challenge -> t
 val is_form : t -> bool
+val get_location : t -> Uri.t option
 
 val user_agent : string
 val prepend_user_agent : t -> string -> t