ngcp-panel/lib/NGCP/Panel/Utils/RedisLocationResultSet.pm

package NGCP::Panel::Utils::RedisLocationResultSet;

use Moo;

use TryCatch;
use NGCP::Panel::Utils::Generic qw(:all);
use NGCP::Panel::Utils::RedisLocationResultSource;

use Data::Dumper;
use Data::Page;
use Time::HiRes qw(time);
use POSIX qw(strftime);

has usrloc_path => (
    is => 'ro',
    default => '1:location',
);

has _c => (
    is => 'ro',
    isa => sub { die "$_[0] must be NGCP::Panel" unless $_[0] && ref $_[0] eq 'NGCP::Panel' },
);

has _redis => (
    is => 'ro',
    isa => sub { die "$_[0] must be Redis" unless $_[0] && ref $_[0] eq 'Redis' },
);

has _rows => (
    is => 'rw',
    isa => sub { die "$_[0] must be ARRAY" unless $_[0] && ref $_[0] eq 'ARRAY' },
    default => sub {[]}
);

has _query_done => (
    is => 'rw',
    isa => sub { die "$_[0] must be int" unless defined $_[0] && is_int($_[0]) },
    default => 0,
);

has result_source => (
    is => 'ro',
    default => sub { NGCP::Panel::Utils::RedisLocationResultSource->new },
);

has result_class => (
    is => 'ro',
    default => 'dummy',
);

has current_source_alias => (
    is => 'ro',
    default => 'me',
);

sub count {
    my ($self) = @_;

    my $count = @{ $self->_rows };
    return $count;
}

sub first {
    my ($self) = @_;
    return $self->_rows->[0];
}

sub all {
    my ($self) = @_;
    return @{ $self->_rows };
}

sub find {
    my ($self, $filter) = @_;
    my $id;

    $filter = $self->_unalias($filter);
    if (ref $filter eq "") {
        $id = $filter;
        $filter = { id => $id };
    } elsif(ref $filter eq "HASH" && exists $filter->{id}) {
        $id = $filter->{id};
    } else {
        $self->_c->log->error("id filter is mandatory for redis find()");
        return;
    }

    my %entry = $self->_redis->hgetall($self->usrloc_path.":entry::$id");
    $entry{id} = $entry{ruid};
    return unless $entry{id};
    # deflate expires column
    if ($entry{expires}) {
        $entry{expires} = strftime("%Y-%m-%d %H:%M:%S", localtime($entry{expires}));
    } else {
        $entry{expires} = "1970-01-01 00:00:00";
    }
    my $subscribers_reseller = $self->_c->model('DB')->resultset('provisioning_voip_subscribers')->search(
        {
            '-or' => [
                { 'me.username' => $entry{username} },
                { 'voip_dbaliases.username' => $entry{username}, 'voip_dbaliases.is_devid' => 1 },
            ],
        },
        {
            join => [ { 'contract' => { 'contact' => 'reseller' } }, 'voip_dbaliases' ],
            '+select' => ['reseller.id'],
            '+as' => ['reseller_id']
        }
    )->first;
    return unless $subscribers_reseller;
    if (exists $filter->{reseller_id} && $filter->{reseller_id} != $subscribers_reseller->get_column('reseller_id')) {
        return;
    }
    return NGCP::Panel::Utils::RedisLocationResultSource->new(_data => \%entry);
}

sub search {
    my ($self, $filter, $opt) = @_;
    $filter //= {};

    $filter = $self->_unalias($filter // {});
    my $new_rs = $self->meta->clone_object($self);
    unless ($new_rs->_query_done) {
        if ($filter->{id}) {
            push @{ $new_rs->_rows }, $new_rs->find($filter);
        } elsif ($filter->{username} && ref $filter->{username} eq 'ARRAY') {
            foreach my $username (@{$filter->{username}}) {
                if ($filter->{domain}) {
                    push @{ $new_rs->_rows },
                        @{ $new_rs->_rows_from_mapkey($self->usrloc_path.":usrdom::" .
                            lc($username) . ":" . lc($filter->{domain}), $filter) };
                } else {
                    push @{ $new_rs->_rows },
                        @{ $new_rs->_rows_from_mapkey($self->usrloc_path.":usrdom::" .
                            lc($username), $filter) };
                }
            }
        } elsif ($filter->{username}) {
            if ($filter->{domain}) {
                push @{ $new_rs->_rows },
                    @{ $new_rs->_rows_from_mapkey($self->usrloc_path.":usrdom::" .
                        lc($filter->{username}) . ":" . lc($filter->{domain}), $filter) };
            } else {
                push @{ $new_rs->_rows },
                    @{ $new_rs->_rows_from_mapkey($self->usrloc_path.":usrdom::" .
                        lc($filter->{username}), $filter) };
            }
        } else {
            $new_rs->_scan($filter, $opt);
        }
        $new_rs->_query_done(1);
    }
    #domain and username already handled; if not deleted, would break filtering because of username duality (SCALAR and ARRAY)
    delete $filter->{username};
    delete $filter->{domain};
    $new_rs->_filter($filter);
    if ($opt->{order_by}) {
        my $sort_field;
        my $sort_order = '';
        if (!ref $opt->{order_by}) {
            $sort_field = $opt->{order_by};
        } elsif (ref $opt->{order_by} eq "HASH") {
            if ($opt->{order_by}->{'-desc'}) {
                $sort_field = $opt->{order_by}->{'-desc'};
                $sort_order = 'desc';
            } elsif ($opt->{order_by}->{'-asc'}) {
                $sort_field = $opt->{order_by}->{'-asc'};
            }
        }
        my $source_alias = $self->current_source_alias();
        $sort_field =~ s/^$source_alias\.//;
        if ($sort_order eq 'desc') {
            $new_rs->_rows([sort { $b->$sort_field cmp $a->$sort_field } @{ $new_rs->_rows }]);
        } else {
            $new_rs->_rows([sort { $a->$sort_field cmp $b->$sort_field } @{ $new_rs->_rows }]);
        }
    }

    if ($opt->{page} && $opt->{rows}) {
        my $pager = Data::Page->new();
        $pager->total_entries(scalar @{ $new_rs->_rows });
        $pager->entries_per_page($opt->{rows});
        $pager->current_page($opt->{page});
        $new_rs->_rows([$pager->splice(\@{ $new_rs->_rows })]);
    }

    return $new_rs;
}

sub _rows_from_mapkey {
    my ($self, $mapkey, $filter) = @_;
    my @rows = ();
    my $keys = $self->_redis->smembers($mapkey);
    foreach my $key (@{ $keys }) {
        my $res = $self->_row_from_key($key, $filter);
        push @rows, $res if $res;
    }
    return \@rows;
}

sub _row_from_key {
    my ($self, $key, $filter) = @_;

    my %entry = $self->_redis->hgetall($key);
    $entry{id} = $entry{ruid};
    next unless $entry{id};
    # deflate expires column
    if ($entry{expires}) {
        $entry{expires} = strftime("%Y-%m-%d %H:%M:%S", localtime($entry{expires}));
    } else {
        $entry{expires} = "1970-01-01 00:00:00";
    }
    my $subscribers_reseller = $self->_c->model('DB')->resultset('provisioning_voip_subscribers')->search(
        {
            '-or' => [
                { 'me.username' => $entry{username} },
                { 'voip_dbaliases.username' => $entry{username}, 'voip_dbaliases.is_devid' => 1 },
            ],
        },
        {
            join => [ { 'contract' => { 'contact' => 'reseller' } }, 'voip_dbaliases' ],
            '+select' => ['reseller.id'],
            '+as' => ['reseller_id']
        }
    )->first;
    next unless $subscribers_reseller;
    if (exists $filter->{reseller_id} && $filter->{reseller_id} != $subscribers_reseller->get_column('reseller_id')) {
        return;
    }
    my $res = NGCP::Panel::Utils::RedisLocationResultSource->new(_data => \%entry);
    return $res;
}

sub _filter {
    my ($self, $filter) = @_;
    my @newrows = ();
    my $i = 0;
    foreach my $row (@{ $self->_rows }) {
        my $match = 0;
        my $filter_applied = 0;
        foreach my $colname (keys %{ $filter }) {
            my $condition = $filter->{$colname};
            my $searchname = $colname;
            my $source_alias = $self->current_source_alias();
            $colname =~ s/^$source_alias\.//;
            next if ($colname =~ /\./); # we don't support joined table columns
            $filter_applied = 1;
            if (defined $condition && ref $condition eq "") {
                if ($row->$colname && lc($row->$colname) ne lc($condition)) {
                    $match = 0;
                    last;
                } else {
                    $match = 1;
                }
            } elsif (ref $condition eq "HASH" && exists $condition->{like}) {
                my $fil = $condition->{like};
                $fil =~ s/^\%//;
                $fil =~ s/\%$//;
                if ($row->$colname !~ /$fil/i) {
                    $match = 0;
                    last;
                } else {
                    $match = 1;
                }
            } else { # condition is undef
                if ($row->$colname) {
                    $match = 0;
                    last;
                } else {
                    $match = 1;
                }
            }
        }
        next if ($filter_applied && !$match);
        push @newrows, $row;
    }
    $self->_rows(\@newrows);
}

sub _scan {
    my ($self, $filter, $opt) = @_;
    $filter //= {};
    my $domain = ref $filter->{domain} eq "HASH"
                    ? ''
                    : ($filter->{domain} // "");
    my $match = ($filter->{username} // "") . ":" . ($domain);
    if ($match eq ":") {
        $match = "*";
    } elsif ($match =~ /:$/) {
        if (exists $filter->{domain}) {
            $match = substr($match, 0, -1);
        } else {
            $match .= '*';
        }
    }

    $self->_rows([]);
    if ($match ne '*') {
        my $cursor = 0;
        do {
            my $res = $self->_redis->scan($cursor, MATCH => $self->usrloc_path.":usrdom::$match", COUNT => 1000);
            $cursor = shift @{ $res };
            my $mapkeys = shift @{ $res };
            foreach my $mapkey (@{ $mapkeys }) {
                push @{ $self->_rows }, @{ $self->_rows_from_mapkey($mapkey, $filter) };
            }
        } while ($cursor);
    }
    else {
        my $cursor = -1;
        my $fetched_keys_count = 0;
        my $stored_keys_count = 0;
        my $res;
        my $page = $opt->{page} // 1;
        my $rows = $opt->{rows} // 10;
        while ($cursor) {
            $cursor == -1 and $cursor = 0; # init cursor first iteration
            $res = $self->_redis->scan($cursor, MATCH => $self->usrloc_path.":entry::*", COUNT => 1000);
            $cursor = shift @{ $res };

            last unless $res && ref $res eq 'ARRAY' && $#$res >= 0;

            my $keys = shift @{$res};
            my $keys_count = $#$keys+1;

            my $offset = $fetched_keys_count - ($page-1)*$rows;
            $fetched_keys_count += $keys_count;

            foreach my $key (@{$keys}) {
                if ($offset < 0) {
                    $offset++;
                    next;
                }
                if (my $v = $self->_row_from_key($key)) {
                    push @{$self->_rows}, $v;
                    $stored_keys_count++;
                    last if $stored_keys_count >= $rows;
                }
            }
            last if $stored_keys_count >= $rows;
        }
    }

    return 1;
}

sub _unalias {
    my ($self,$filter) = @_;
    if ('HASH' eq ref $filter) {
        my %unaliased_filter = ();
        my $source_alias = $self->current_source_alias();
        foreach my $key (keys %$filter) {
            my $k = $key;
            $k =~ s/^$source_alias\.//;
            $unaliased_filter{$k} = $filter->{$key};
        }
        $filter = \%unaliased_filter;
    }
    return $filter;
}

1;