upstream_selector.lua

local ipairs = ipairs
local setmetatable = setmetatable

local Upstream = require('apicast.upstream')

local _M = {}

local mt = { __index = _M }

function _M.new()
  local self = setmetatable({}, mt)
  return self
end

--- Selects an upstream based on a list of rules.
-- @tparam rules table Table with instances of Rule
-- @tparam context table Context used to evaluate the conditions
-- @treturn Upstream Returns an instance of Upstream initialized with the url
--   of the first rule whose condition evaluates to true. If there are no rules
--   or none of them evaluate to true, this method returns nil.
function _M.select(_, rules, context)
  if not rules then return nil end

  local service = context.service or ngx.ctx.service or {}

  for _, rule in ipairs(rules) do
    local cond_is_true = rule.condition:evaluate(context)

    if cond_is_true then
      ngx.log(ngx.DEBUG, 'Rule with URL ', rule.url, ' matched')

      local upstream = Upstream.new(rule.url)

      local host_header = rule.host_header or service and service.hostname_rewrite
      if host_header and host_header ~= '' then
        upstream:use_host_header(host_header)
      end

      if rule.owner_id then
        upstream:set_owner_id(rule.owner_id)
      end

      if rule.replace_path then
        upstream:append_path(rule.replace_path:render(context))
        -- Set uri as nil if not will be appended to the upstream
        ngx.req.set_uri("/")
      end

      return upstream
    end
  end

  return nil
end

return _M