abonbot/modules/cdata.pm

use Modern::Perl;
use utf8;

use telnet;

use Mojo::Base -strict, -async_await;
use Data::Dumper;

our $log;

sub cdata_extract_onu_info {
    my @lines = @_;

#    say Dumper @lines;

    my @onu_info = @lines[4..@lines-5];

    return map([ split(' ', $_) ], @onu_info );
};

async sub cdata_login
{
    my $t        = shift;
    my $login    = shift;
    my $password = shift;

    await $t->connect;

    await $t->reply(qr/User name:/, $login);
    await $t->reply(qr/User password:/, $password);

    $t->print("enable");
    await $t->waitfor("#");
    $t->prompt(qr/#\s?$/);

    await $t->cmd("config");
    await $t->cmd("vty output show-all");

};

async sub cdata_get_onu_info
{
    my $t    = shift;
    my $tree = shift;

    await $t->cmd("interface epon 0/0");
    my @lines = await $t->cmd("show ont info $tree all");

    my $onu_count_line = @lines[@lines-3];

    $onu_count_line =~ m|Total: (\d+), online (\d+)|;
    my $total_onu_cnt   = $1;
    my $online_onu_cnt  = $2;
    my $offline_onu_cnt = $total_onu_cnt - $online_onu_cnt;

    if ($lines[2] =~ /Error/) {
        $total_onu_cnt   = 0;
        $online_onu_cnt  = 0;
        $offline_onu_cnt = 0;
    }

    await $t->cmd("exit");

    my @onu_info = cdata_extract_onu_info(@lines);

    my @online_onu = grep { @$_[5] =~ "online" }  @onu_info;
    @online_onu = map [@$_[2,3]], @online_onu;

    my @offline_onu = grep { @$_[5] =~ "offline|powerdown" }  @onu_info;
    @offline_onu = map [@$_[2,3]], @offline_onu;

    my %res = (
        'total_cnt'    => $total_onu_cnt,
        'active_cnt'   => $online_onu_cnt,
        'inactive_cnt' => $offline_onu_cnt,
        'active'       => [ @online_onu ],
        'inactive'     => [ @offline_onu ],
    );

    return %res;

};

async sub cdata_fun_purge_tree
{
    my $t        = shift;
    my $tree     = shift;
    my $onu_info = shift;
    my %onu_info = %{$onu_info};


    #say Dumper \%onu_info;

    await $t->cmd("interface epon 0/0");

    foreach my $i ( @{$onu_info{'inactive'}} ) {
        #say Dumper @$i[0];
        my $onu_num = @$i[0];
#        say Dumper $onu_num;
        my @tmp = await $t->cmd("ont delete $tree $onu_num");
        #say Dumper @tmp;
    }

    await $t->cmd("exit");

    my @deleted_onu = map { sprintf("%2d %s", $_->[0], $_->[1]) } @{$onu_info{inactive}};

    return @deleted_onu;
};

async sub cdata_inspect
{
    my $ip = shift;
    my $login = shift;
    my $password = shift;
    my $tree = shift;

    my $t = new telnet($ip);
#    $t->debug(1);

    await cdata_login($t, $login, $password);

    my %onu_info = await cdata_get_onu_info($t, $tree);

#    say Dumper \%onu_info;
    
    $t->close;

    my $res = "Всего: $onu_info{'total_cnt'}
Количество активных ONU: $onu_info{'active_cnt'}
Количество неактивных ONU: $onu_info{'inactive_cnt'}
\n";

    $res = $res . "Неактивные ONU:\n" . join("\n", map( join(" ", @$_), @{$onu_info{'inactive'}} ) ) . "\n\n";

    $res = $res . "Активные ONU:\n" . join("\n", map( join(" ", @$_), @{$onu_info{'active'}} ) );

    return $res;
};

async sub cdata_onu
{
    my $ip = shift;
    my $login = shift;
    my $password = shift;
    my $tree = shift;

    my $t = new telnet($ip);
#    $t->debug(1);

    await cdata_login($t, $login, $password);

    my %onu_info = await cdata_get_onu_info($t, $tree);

    $t->close;

    my $res = "Всего: $onu_info{'total_cnt'}
Количество активных ONU: $onu_info{'active_cnt'}
Количество неактивных ONU: $onu_info{'inactive_cnt'}
\n";

    return $res;
};

async sub cdata_purge
{
    my $ip = shift;
    my $login = shift;
    my $password = shift;
    my $tree = shift;
    my $save = shift;

    my $t = new telnet($ip);
#    $t->debug(1);

    await cdata_login($t, $login, $password);

    my %onu_info = await cdata_get_onu_info($t, $tree);

    if ($onu_info{'inactive_cnt'} == 0) {
        $t->close;
        return "Нечего чистить.";
    }

    my @deleted_onu = await cdata_fun_purge_tree($t, $tree, \%onu_info);

    %onu_info = await cdata_get_onu_info($t, $tree);

    if ($save) {
        await $t->cmd("save");
    }

    $t->close;

    my $res = "Осталось ONU: $onu_info{'total_cnt'}\n";

    if ( $onu_info{'active_cnt'} != $onu_info{'total_cnt'} ) {
        $res = $res . "Общее количество ONU и количество активных ONU не совпадает.
Лучше обратиться к Вашему системному администратору\n";
    }

    $res = $res . "Удалённые ONU: " . scalar @deleted_onu . "\n";

    $log->info("IP $ip TREE $tree\n" . join("\n", @deleted_onu));


    return $res;
};

async sub cdata_purgeall
{
    my $ip = shift;
    my $login = shift;
    my $password = shift;
    my $tree_count = shift;
    my $save = shift;

    my @res;

    my $t = new telnet($ip);
#    $t->debug(1);

    await cdata_login($t, $login, $password);

    for (my $tree=1; $tree <= $tree_count; $tree++) {
        my $res = "Дерево $tree\n";

        my %onu_info = await cdata_get_onu_info($t, $tree);

#        say Dumper %onu_info;

        if ($onu_info{'inactive_cnt'} == 0) {
            $res = $res . "Не имеет неактивных ONU\n";
            push @res, $res;
            next;
        }

        my @deleted_onu = await cdata_fun_purge_tree($t, $tree, \%onu_info);

#        say Dumper @deleted_onu;

        %onu_info = await cdata_get_onu_info($t, $tree);

        $res = $res . "Осталось ONU: $onu_info{'total_cnt'}\n";
        $res = $res . "Удалённые ONU: " . scalar @deleted_onu . "\n";

        $log->info("IP $ip TREE $tree\n" . join("\n", @deleted_onu));

        push @res, $res;

    }

    if ($save) {
        await $t->cmd("save");
    }

    $t->close;

    return \@res;
};

1;