MT#7571 bin/create_rewrite_rules.pl: use REST API

Change-Id: Icbefc9291c8c5719485b34c68fb5256517795de2
changes/74/3974/3
Victor Seva 9 years ago
parent 1c77b3df68
commit 9a7814994b

@ -151,7 +151,7 @@ function delete_voip
if [ -f "${SCEN_CHECK_DIR}/rewrite.yml" ]; then
echo "$(date) - Deleting rewrite rules"
"${BIN_DIR}/create_rewrite_rules.pl" -d "${SCEN_CHECK_DIR}/rewrite.yml"
"${BIN_DIR}/create_rewrite_rules.pl" -delete "${SCEN_CHECK_DIR}/rewrite.yml"
fi
if [ -f "${SCEN_CHECK_DIR}/hosts" ]; then

@ -22,66 +22,56 @@ use strict;
use warnings;
use English;
use Getopt::Std;
use Getopt::Long;
use Cwd 'abs_path';
use Config::Tiny;
use Sipwise::API qw(all);
use YAML;
use Getopt::Long;
use Sipwise::Provisioning::Voip;
use Sipwise::Provisioning::Billing;
use Sipwise::Provisioning::Config;
our %CONFIG = ( admin => 'cmd' );
my $config = Sipwise::Provisioning::Config->new()->get_config();
unless ($CONFIG{password} = $config->{acl}->{$CONFIG{admin}}->{password}) {
die "Error: No provisioning password found for user $CONFIG{admin}\n";
my $config = Config::Tiny->read('/etc/default/ngcp-api');
my $opts;
if ($config) {
$opts = {};
$opts->{host} = $config->{_}->{NGCP_API_IP};
$opts->{port} = $config->{_}->{NGCP_API_PORT};
$opts->{sslverify} = $config->{_}->{NGCP_API_SSLVERIFY};
}
my $api = Sipwise::API->new($opts);
$opts = $api->opts;
my $del;
sub usage;
sub call_prov;
sub usage {
return "Usage:\n$PROGRAM_NAME rewrite.yml\n".
"Options:\n".
" -delete\n".
" -d debug\n".
" -h this help\n";
}
my $help = 0;
my $del = 0;
GetOptions ("h|help" => \$help,
"d|delete" => \$del)
or die("Error in command line arguments\n".usage());
"d|debug" => \$opts->{verbose},
"delete" => \$del)
or die("Error in command line arguments\n".usage());
die(usage()) unless (!$help);
die("Wrong number of arguments\n".usage()) unless ($#ARGV == 0);
our $bprov = Sipwise::Provisioning::Billing->new();
our $vprov = Sipwise::Provisioning::Voip->new();
my $filename = abs_path($ARGV[0]);
my $r = YAML::LoadFile($filename);
if ($del)
{
do_delete($r);
}
else
{
do_create($r);
}
sub do_delete
{
my ($data) = @_;
for my $rule_set_name (keys $data)
for my $rule_set_name (keys %{$data})
{
my $rule_set;
my $result = call_prov( $vprov, 'get_rewrite_rule_sets');
foreach (@{$result})
{
$rule_set->{$_->{name}} = $_->{id};
}
if ($rule_set)
{
for my $rule_set_name (keys $rule_set)
{
call_prov($vprov, 'delete_rewrite_rule_set', {id => $rule_set->{$rule_set_name}});
my $param = { reseller_id => 1, name => $rule_set_name };
my $rws_id = $api->check_rewriteruleset_exists($param);
if(defined $rws_id) {
if($api->delete_rewriteruleset($rws_id)) {
print "rewriteruleset [$rule_set_name] deleted [$rws_id]\n";
}
else {
die "rewriteruleset [$rule_set_name] can't be removed [$rws_id]\n";
}
} else {
print "rewriteruleset [$rule_set_name] not there\n";
}
}
exit;
@ -90,58 +80,45 @@ sub do_delete
sub do_create
{
my ($data) = @_;
for my $rule_set_name (keys $data)
for my $rule_set_name (keys %{$data})
{
my $param = { name => $rule_set_name };
call_prov( $vprov, 'create_rewrite_rule_set', $param );
}
my $rule_set;
my $result = call_prov( $vprov, 'get_rewrite_rule_sets');
foreach (@{$result})
{
$rule_set->{$_->{name}} = $_->{id};
}
# rules
for my $rule_set_name (keys $data)
{
my $rs = $data->{$rule_set_name};
foreach (@{$rs})
{
my $param = { set_id => $rule_set->{$rule_set_name}, data => $_ };
call_prov($vprov, 'create_rewrite_rule', $param);
my $param = { reseller_id => 1, name => $rule_set_name };
my $rws_id = $api->check_rewriteruleset_exists($param);
if(defined $rws_id) {
print "rewriteruleset [$rule_set_name] already there [$rws_id]\n";
} else {
$rws_id = $api->create_rewriteruleset($param);
print "rewriteruleset [$rule_set_name] created [$rws_id]\n";
}
foreach my $param (@{$data->{$rule_set_name}}) {
$param->{set_id} = $rws_id;
my $rw_id = $api->create_rewriterule($param);
if(defined $rw_id) {
print "rewriterule [".
$param->{field}."/".$param->{direction}.
"] created [$rw_id]\n";
}
else {
die "Can't create rewriterule"
}
}
}
exit;
}
sub call_prov {
# scalar, scalar, hash-ref
my ($prov, $function, $parameter) = @_;
my $result;
eval {
$result = $prov->handle_request( $function,
{
authentication => {
type => 'system',
username => $CONFIG{admin},
password => $CONFIG{password},
},
parameters => $parameter,
});
};
sub main {
my $r = shift;
if($EVAL_ERROR) {
if(ref $EVAL_ERROR eq 'SOAP::Fault') {
die "Voip\::$function failed: ". $EVAL_ERROR->faultstring;
} else {
die "Voip\::$function failed: $EVAL_ERROR";
}
if ($del)
{
print "delete rewriterules\n" unless $opts->{debug};
return do_delete($r);
}
else
{
print "create rewriterules\n" unless $opts->{debug};
return do_create($r);
}
return $result;
}
sub usage {
return "Usage:\n$PROGRAM_NAME scenario.yml\n";
}
main(YAML::LoadFile(abs_path($ARGV[0])));

@ -97,9 +97,10 @@ sub do_query {
my $ua = shift;
my $url = shift;
my $data = shift;
my $req_type = shift || 'GET';
my $URL = URI->new($url);
$URL->query_form($data);
my $req = HTTP::Request->new('GET', $URL);
my $req = HTTP::Request->new($req_type, $URL);
my $res = $ua->request($req);
if(!$res->is_success) {
@ -153,6 +154,16 @@ sub _create {
return;
}
sub _delete {
my $self = shift;
my $urldata = shift;
my $urlbase = 'https://'.$self->{opts}->{host}.':'.$self->{opts}->{port};
my $ua = $self->create_ua();
my $res = $self->do_query($ua, $urlbase.$urldata, undef, 'DELETE');
return $res->is_success;
}
sub _get_content {
my $self = shift;
my $data = shift;
@ -195,7 +206,9 @@ sub _exists {
my $collection = $self->_get_content($data, $urldata);
if (defined $collection && $collection->{total_count} == 1) {
return $collection->{_embedded}->{$collection_id}->{id};
my $links = $collection->{_embedded}->{$collection_id}->{_links};
my $href = $links->{$collection_id}->{href};
return _get_id($urldata, $href);
}
return;
}
@ -363,7 +376,7 @@ sub check_rewriteruleset_exists {
my $self = shift;
my $data = shift;
my $urldata = '/api/rewriterulesets/';
my $collection_id = 'ngcp:rewriterules';
my $collection_id = 'ngcp:rewriterulesets';
return $self->_exists($data, $urldata, $collection_id);
}
@ -375,3 +388,11 @@ sub create_rewriteruleset {
return $self->_create($data, $urldata);
}
sub delete_rewriteruleset {
my $self = shift;
my $id = shift;
my $urldata = "/api/rewriterulesets/${id}";
return $self->_delete($urldata);
}

Loading…
Cancel
Save