From fe48161dbc7511220b0b91b06c26d2beba7a6586 Mon Sep 17 00:00:00 2001
From: Ingela Anderton Andin <ingela@erlang.org>
Date: Wed, 30 Sep 2015 10:24:35 +0200
Subject: inets: Add new customize function response_default_headers

This enables the user to provide default HTTP header values for headers
that should always be sent. Note that these values
may override built in defaults.
---
 lib/inets/src/http_server/httpd_custom.erl   | 23 +++++++++++++++++++++--
 lib/inets/src/http_server/httpd_response.erl | 21 ++++++++++++++-------
 2 files changed, 35 insertions(+), 9 deletions(-)

(limited to 'lib/inets/src')

diff --git a/lib/inets/src/http_server/httpd_custom.erl b/lib/inets/src/http_server/httpd_custom.erl
index b62fc25fac..2b9701ef75 100644
--- a/lib/inets/src/http_server/httpd_custom.erl
+++ b/lib/inets/src/http_server/httpd_custom.erl
@@ -20,8 +20,8 @@
 %%
 -module(httpd_custom).
 
--export([response_header/1, request_header/1]).
--export([customize_headers/3]).
+-export([response_header/1, request_header/1, response_default_headers/0]).
+-export([customize_headers/3, response_default_headers/1]).
 
 -include("../inets_app/inets_internal.hrl").
 
@@ -35,7 +35,12 @@ response_header(Header) ->
     {true, httpify(Header)}.
 request_header(Header) -> 
     {true, Header}.
+response_default_headers() ->
+    [].
 
+%%--------------------------------------------------------------------
+%% Internal API  -----------------------------------
+%%--------------------------------------------------------------------
 customize_headers(?MODULE, Function, Arg) ->
     ?MODULE:Function(Arg);
 customize_headers(Module, Function, Arg) ->
@@ -49,6 +54,20 @@ customize_headers(Module, Function, Arg) ->
 	    ?MODULE:Function(Arg)
     end.
 
+response_default_headers(?MODULE) ->
+    response_default_headers();
+response_default_headers(Module) ->
+    try Module:response_default_headers() of
+	Defaults ->
+	    [{http_util:to_lower(Key), Value} ||  {Key, Value} <- Defaults, 
+						  is_list(Key), is_list(Value)]
+    catch
+	_:_ ->
+	    ?MODULE:response_default_headers()
+    end.
+%%--------------------------------------------------------------------
+%% Internal functions -----------------------------------
+%%--------------------------------------------------------------------
 httpify({Key0, Value}) ->
     %% make sure first letter is capital (defacto standard)
     Words1 = string:tokens(Key0, "-"),
diff --git a/lib/inets/src/http_server/httpd_response.erl b/lib/inets/src/http_server/httpd_response.erl
index 7e73da7060..71243f525a 100644
--- a/lib/inets/src/http_server/httpd_response.erl
+++ b/lib/inets/src/http_server/httpd_response.erl
@@ -287,14 +287,21 @@ create_header(ConfigDb, KeyValueTupleHeaders) ->
     Date        = httpd_util:rfc1123_date(), 
     ContentType = "text/html", 
     Server      = server(ConfigDb),
-    Headers0  = add_default_headers([{"date",         Date},
-				     {"content-type", ContentType}
-				     | if Server=="" -> [];
-					  true -> [{"server",       Server}]
-				       end
-				    ], 
-				    KeyValueTupleHeaders),
     CustomizeCB = httpd_util:lookup(ConfigDb, customize, httpd_custom),
+
+    CustomDefaults = httpd_custom:response_default_headers(CustomizeCB),
+    SystemDefaultes = ([{"date", Date},
+			{"content-type", ContentType}
+			| if Server=="" -> [];
+			     true -> [{"server", Server}]
+			  end
+		       ]),
+
+    %% System defaults not present in custom defaults will be added
+    %% to defaults    
+    Defaults = add_default_headers(SystemDefaultes, CustomDefaults),
+    
+    Headers0 = add_default_headers(Defaults, KeyValueTupleHeaders),
     lists:filtermap(fun(H) ->
 			    httpd_custom:customize_headers(CustomizeCB, response_header, H)
 		    end,
-- 
cgit v1.2.3