--  Licensed to the Apache Software Foundation (ASF) under one
--  or more contributor license agreements.  See the NOTICE file
--  distributed with this work for additional information
--  regarding copyright ownership.  The ASF licenses this file
--  to you under the Apache License, Version 2.0 (the
--  "License"); you may not use this file except in compliance
--  with the License.  You may obtain a copy of the License at
--
--  http://www.apache.org/licenses/LICENSE-2.0
--
--  Unless required by applicable law or agreed to in writing, software
--  distributed under the License is distributed on an "AS IS" BASIS,
--  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
--  See the License for the specific language governing permissions and
--  limitations under the License.

local function host_or_nil(v)
    if v == nil then return "<nil>" end
    return v
end

function send_response()
    local hook_val = ts.remap.get_from_url_host()
    ts.client_response.header['Remap-From-Host'] = ts.ctx['remap_from_host']
    ts.client_response.header['Hook-From-Host']  = host_or_nil(hook_val)
end

function do_remap()
    ts.ctx['remap_from_host'] = host_or_nil(ts.remap.get_from_url_host())
    ts.hook(TS_LUA_HOOK_SEND_RESPONSE_HDR, send_response)
end
