--  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.

-- This example depends on "libmaxminddb".
-- It illustrates how to connect to MaxMind DB and uses it to look up country of an IP address.
-- It can be used in plugin.config with the lua plugin.

-- Setup Instructions
-- 1. install libmaxminddb 1.6.0 (https://github.com/maxmind/libmaxminddb)
--   a. wget https://github.com/maxmind/libmaxminddb/releases/download/1.6.0/libmaxminddb-1.6.0.tar.gz
--   b. tar zxvf libmaxminddb-1.6.0.tar.gz
--   c. cd libmaxminddb-1.6.0
--   d. ./configure; make; make install
-- 2) Get GeoLite2 country database from https://dev.maxmind.com/geoip/geolite2-free-geolocation-data and put it in /usr/share/GeoIP/GeoLite2-Country.mmdb
-- 3. install luajit-geoip v2.1.0 (https://github.com/leafo/luajit-geoip)
--   a. wget https://github.com/leafo/luajit-geoip/archive/refs/tags/v2.1.0.tar.gz
--   b. tar zxvf v2.1.0.tar.gz
--   c. mkdir -p /usr/local/share/lua/5.1/geoip
--   d. cp luajit-geoip-2.1.0/geoip.lua /usr/local/share/lua/5.1/geoip.lua
--   e. cp luajit-geoip-2.1.0/geoip/*.lua /usr/local/share/lua/5.1/geoip/

ts.add_package_path('/usr/local/share/lua/5.1/?.lua')

local geoip = require 'geoip.mmdb'

function do_global_send_response()
  local mmdb = geoip.load_database("/usr/share/GeoIP/GeoLite2-Country.mmdb")

  local result = mmdb:lookup("8.8.8.8")

  ts.client_response.header['X-Maxmind-Info'] = result.country.iso_code
end