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.
lua-ngcp-kamailio/ngcp/pp.lua

113 lines
3.5 KiB

--
-- Copyright 2013-2015 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'
local NGCPXAvp = require 'ngcp.xavp'
-- class NGCPPeerPrefs
local NGCPPeerPrefs = utils.inheritsFrom(NGCPPrefs)
NGCPPeerPrefs.__class__ = 'NGCPPeerPrefs'
NGCPPeerPrefs.group = "peer_prefs"
NGCPPeerPrefs.db_table = "peer_preferences"
NGCPPeerPrefs.query = "SELECT * FROM %s WHERE uuid = '%s'"
-- joins three tables:
-- *kamailio.peer_preferences
-- *provisioning.voip_preferences
-- *provisioning.voip_preference_groups
-- links peers attributes to the preferences list,
-- to the preferences group id
NGCPPeerPrefs.group_query = [[
SELECT kp.*
FROM kamailio.peer_preferences AS kp
JOIN provisioning.voip_preferences AS vp
ON vp.attribute = kp.attribute
JOIN provisioning.voip_preference_groups AS vpg
ON vpg.id = vp.voip_preference_groups_id
WHERE kp.uuid = '%s' AND vpg.id = %s
]]
NGCPPeerPrefs.select_id_query = "SELECT id FROM provisioning.voip_preference_groups WHERE name = '%s'"
-- luacheck: globals KSR
function NGCPPeerPrefs:new(config)
local instance = NGCPPeerPrefs:create()
-- creates xavp usr
instance:init(config)
return instance
end
function NGCPPeerPrefs:clean(vtype)
NGCPPrefs.clean(self, vtype)
if not vtype then
NGCPXAvp:new('callee', 'prefs'):clean()
NGCPXAvp:new('caller', 'prefs'):clean()
else
NGCPXAvp:new(vtype, 'prefs'):clean()
end
end
function NGCPPeerPrefs:get_pref_group_id(name)
local con = assert(self.config:getDBConnection())
local query = self.select_id_query:format(name)
local cur = assert(con:execute(query))
local row = cur:fetch({}, "a")
cur:close()
if row and row.id then
return tonumber(row.id)
end
KSR.err(string.format("[NGCP] preference group '%s' not found\n", name))
return nil
end
function NGCPPeerPrefs:load_group(level, uuid, group_id)
local con = assert(self.config:getDBConnection())
local query = self.group_query:format(uuid, group_id)
local cur = assert(con:execute(query))
return self:_set_xavp(level, cur, query)
end
function NGCPPeerPrefs:caller_load_group(uuid, name)
local group_id = self:get_pref_group_id(name)
if not group_id then
KSR.err("[NGCP] Cannot load group '%s', skipping\n", name)
return {}
end
if not uuid or uuid == '' then return {} end
return self:load_group("caller", uuid, group_id)
end
function NGCPPeerPrefs:callee_load_group(uuid, name)
local group_id = self:get_pref_group_id(name)
if not group_id then
KSR.err("[NGCP] Cannot load group '%s', skipping\n", name)
return {}
end
if not uuid or uuid == '' then return {} end
return self:load_group("callee", uuid, group_id)
end
-- class
return NGCPPeerPrefs