You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
97 lines
3.1 KiB
97 lines
3.1 KiB
--
|
|
-- Copyright 2013-2020 SipWise Team <development@sipwise.com>
|
|
--
|
|
-- This program is free software: you can redistribute it and/or modify
|
|
-- it under the terms of the GNU General Public License as published by
|
|
-- the Free Software Foundation, either version 3 of the License, or
|
|
-- (at your option) any later version.
|
|
--
|
|
-- This package is distributed in the hope that it will be useful,
|
|
-- but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
-- MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
-- GNU General Public License for more details.
|
|
--
|
|
-- You should have received a copy of the GNU General Public License
|
|
-- along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
-- .
|
|
-- On Debian systems, the complete text of the GNU General
|
|
-- Public License version 3 can be found in "/usr/share/common-licenses/GPL-3".
|
|
--
|
|
local utils = require 'ngcp.utils'
|
|
local NGCPPrefs = require 'ngcp.pref'
|
|
|
|
-- class NGCPContractPrefs
|
|
local NGCPContractPrefs = utils.inheritsFrom(NGCPPrefs)
|
|
|
|
NGCPContractPrefs.__class__ = 'NGCPContractPrefs'
|
|
NGCPContractPrefs.group = "contract_prefs"
|
|
NGCPContractPrefs.db_table = "contract_preferences"
|
|
NGCPContractPrefs.query = "SELECT * FROM %s WHERE uuid ='%s' AND location_id IS NULL"
|
|
NGCPContractPrefs.query_location_id = [[
|
|
SELECT location_id FROM provisioning.voip_contract_locations cl JOIN
|
|
provisioning.voip_contract_location_blocks cb ON cb.location_id = cl.id
|
|
WHERE cl.contract_id = %s AND
|
|
_%s_net_from <= UNHEX(HEX(INET_ATON('%s'))) AND
|
|
_%s_net_to >= UNHEX(HEX(INET_ATON('%s')))
|
|
ORDER BY cb.ip DESC, cb.mask DESC LIMIT 1
|
|
]]
|
|
-- luacheck: globals KSR
|
|
function NGCPContractPrefs:new(config)
|
|
local instance = NGCPContractPrefs:create()
|
|
-- creates xavp usr
|
|
instance:init(config)
|
|
return instance
|
|
end
|
|
|
|
function NGCPContractPrefs:caller_load(contract, ip)
|
|
if not contract then
|
|
return {}
|
|
end
|
|
return self:_load("caller", contract, ip)
|
|
end
|
|
|
|
function NGCPContractPrefs:callee_load(contract, ip)
|
|
if not contract then
|
|
return {}
|
|
end
|
|
return self:_load("callee", contract, ip)
|
|
end
|
|
|
|
function NGCPContractPrefs:_get_location_id(con, id, ip)
|
|
local ip_type = "ipv4"
|
|
if string.find(ip, ':') ~= nil then
|
|
ip_type = "ipv6"
|
|
end
|
|
local query = self.query_location_id:format(id, ip_type, ip, ip_type, ip)
|
|
local cur = assert(con:execute(query))
|
|
|
|
local row = cur:fetch({}, "a")
|
|
cur:close()
|
|
|
|
if row then
|
|
return row.location_id
|
|
end
|
|
end
|
|
|
|
function NGCPContractPrefs:_load(level, contract, ip)
|
|
local con = self.config:getDBConnection()
|
|
local location_id = nil
|
|
local query = self.query:format(self.db_table, contract)
|
|
if ip then
|
|
location_id = self:_get_location_id(con, contract, ip)
|
|
if location_id then
|
|
query = string.format("SELECT * FROM %s WHERE uuid ='%s' AND location_id = %d",
|
|
self.db_table, contract, location_id)
|
|
end
|
|
end
|
|
local cur = assert(con:execute(query))
|
|
local keys = self:_set_xavp(level, cur, query)
|
|
local xavp = self:xavp(level)
|
|
xavp("location_id", location_id)
|
|
|
|
return keys
|
|
end
|
|
|
|
-- class
|
|
return NGCPContractPrefs
|