package App::Netdisco::Web::Plugin::API::Device;

use Dancer ':syntax';

use Dancer::Plugin::Ajax;
use Dancer::Plugin::Auth::Extensible;
use Dancer::Plugin::DBIC;

use App::Netdisco::Web::Plugin;

sub api_array_json {
    my $items = shift;
    my @results;
    foreach my $item (@{$items}) {
        my $c = {};
        my $columns = $item->{_column_data};
        foreach my $col (keys %{$columns}) {
            $c->{$col} = $columns->{$col};
        }
        push @results, $c;
    }
    return (\@results);
};

ajax '/api/device/all' => require_login sub {
    my @devices=schema('netdisco')->resultset('Device')->all;
    return api_array_json(\@devices);
};

ajax '/api/device/search' => sub {
    my $para = params;
    my $search = {};
    foreach my $param (keys %{$para}) {
        if ($param ne 'return_url') {
            $search->{$param} = $para->{$param};
        }
    }
    my @devices;
    try {
       @devices=schema('netdisco')->resultset('Device')->search($search);
    };
    return api_array_json(\@devices);
};

ajax '/api/device/:device' => require_login sub {
    my $dev = params->{device};
    print "$dev\n";
    my $device = schema('netdisco')->resultset('Device')
      ->search_for_device($dev) or send_error('Bad Device', 404);
    return $device->{_column_data};
};

ajax '/api/device/:device/modules' => require_login sub {
    my $dev = params->{device};
    my $device = schema('netdisco')->resultset('Device')
      ->search_for_device($dev) or send_error('Bad Device', 404);
    my @modules = $device->modules;
    return api_array_json(\@modules);

};

ajax '/api/device/:device/vlans' => require_login sub {
    my $dev = params->{device};
    my $device = schema('netdisco')->resultset('Device')
      ->search_for_device($dev) or send_error('Bad Device', 404);
    my @vlans = $device->vlans;
    return api_array_json(\@vlans);
};

ajax '/api/device/:device/ports' => require_login sub {
    my $dev = params->{device};
    my $device = schema('netdisco')->resultset('Device')
      ->search_for_device($dev);
    my @ports = $device->ports->all;
    my @results;
    foreach my $item (@ports) {
        my $c = {};
        my $columns = $item->{_column_data};
        foreach my $col (keys %{$columns}) {
            $c->{$col} = $columns->{$col};
        }
        my @vlans = $item->vlans->all ;
        my @pvlans;
        my @vl = $item->port_vlans;
        foreach my $item (@vlans) {
            push @pvlans, $item->{_column_data}->{vlan};
        }
        $c->{vlans}=\@pvlans;
        push @results, $c;
    }
    return \@results;
};

ajax qr{/api/device/(?<ip>.*)/port/(?<port>.*)/nodes$} => require_login sub {
    my $param = captures;
    my @ports = schema('netdisco')->resultset('Device')
      ->search_for_device($$param{ip})->ports->search({port => $$param{port}});
    my @nodes = $ports[0]->nodes;
    return api_array_json(\@nodes);
};

ajax qr{/api/device/(?<ip>.*)/port/(?<port>.*)/neighbor$} => require_login sub {
    my $param = captures;
    my @ports = schema('netdisco')->resultset('Device')
      ->search_for_device($$param{ip})->ports->search({port => $$param{port}});
    my @neighbors = $ports[0]->neighbor;
    return api_array_json(\@neighbors);
};

ajax qr{/api/device/(?<ip>.*)/port/(?<port>.*)/power$} => require_login sub {
    my $param = captures;
    my @ports = schema('netdisco')->resultset('Device')
      ->search_for_device($$param{ip})->ports->search({port => $$param{port}});
    my @neighbors = $ports[0]->power;
    return api_array_json(\@neighbors);
};

ajax qr{/api/device/(?<ip>.*)/port/(?<port>.*)} => require_login sub {
    my $param = captures;
    my $port;
    try {
        my @ports = schema('netdisco')->resultset('Device')
          ->search_for_device($$param{ip})->ports->search({port => $$param{port}});
        my @vlans = $ports[0]->vlans->all ;
        my @pvlans;
        foreach my $item (@vlans) {
            push @pvlans, $item->{_column_data}->{vlan};
        }
        $port = @{api_array_json(\@ports)}[0];
        $port->{vlans} = \@pvlans;
    };

    return $port if defined $port;
    status 404;
    return { message => "Port not found" };
};

true;