103 lines
2.3 KiB
Perl
103 lines
2.3 KiB
Perl
package App::Netdisco::Util::API;
|
|
|
|
use strict;
|
|
use warnings;
|
|
use Dancer ':syntax';
|
|
use Dancer::Plugin::DBIC 'schema';
|
|
|
|
use List::MoreUtils 'singleton';
|
|
|
|
use base 'Exporter';
|
|
our @EXPORT = ();
|
|
our @EXPORT_OK = qw/
|
|
resultsource_to_openapi_params
|
|
parse_search_params
|
|
format_data
|
|
format_error
|
|
/;
|
|
our %EXPORT_TAGS = (all => \@EXPORT_OK);
|
|
|
|
sub resultsource_to_openapi_params {
|
|
my $sourcename = shift or return ();
|
|
my @params = ();
|
|
|
|
my $rs = schema('netdisco')->source($sourcename) or return ();
|
|
my $columns = $rs->columns_info;
|
|
|
|
foreach my $col ($rs->primary_columns,
|
|
(singleton ($rs->primary_columns, keys %{ $columns }))) {
|
|
my $data = $columns->{$col};
|
|
next if $data->{extra}->{hide_from_api};
|
|
push @params, (
|
|
$col => {
|
|
description => $data->{extra}->{descr},
|
|
type => ($data->{data_type} =~ m/int/ ? 'integer'
|
|
: $data->{data_type} eq 'boolean' ? 'boolean' : 'string'),
|
|
}
|
|
);
|
|
}
|
|
|
|
return @params;
|
|
}
|
|
|
|
sub parse_search_params {
|
|
my $params = shift;
|
|
my $search = {};
|
|
my $partial = $params->{partial} || false;
|
|
|
|
foreach my $param (keys %{$params}) {
|
|
if ($param ne 'return_url' and $param ne 'partial') {
|
|
if ($partial eq 'true') {
|
|
$search->{"text(".$param.")"} = { -ilike => '%'.$params->{$param}.'%'};
|
|
}
|
|
else {
|
|
$search->{$param} = $params->{$param};
|
|
}
|
|
}
|
|
}
|
|
|
|
return $search;
|
|
}
|
|
|
|
sub format_data {
|
|
my $items = shift;
|
|
my $results = {};
|
|
|
|
if (ref($items) =~ m/ResultSet/) {
|
|
my @hashes;
|
|
|
|
foreach my $item ($items->all) {
|
|
my $c = {};
|
|
my $columns = $item->{_column_data};
|
|
|
|
foreach my $col (keys %{$columns}) {
|
|
$c->{$col} = $columns->{$col};
|
|
}
|
|
|
|
push @hashes, $c;
|
|
}
|
|
|
|
$results->{data} = \@hashes;
|
|
}
|
|
elsif (ref($items) =~ m/Result/) {
|
|
$results->{data} = $items->{_column_data};
|
|
}
|
|
else {
|
|
$results->{data} = $items;
|
|
}
|
|
|
|
header('Content-Type' => 'application/json');
|
|
return to_json $results;
|
|
};
|
|
|
|
sub format_error {
|
|
my $status = shift;
|
|
my $message = shift;
|
|
|
|
status $status;
|
|
header('Content-Type' => 'application/json');
|
|
return to_json { error => $message };
|
|
}
|
|
|
|
true;
|