File: AssetCategory.pm

package info (click to toggle)
ocsinventory-server 2.5%2Bdfsg1-1
  • links: PTS, VCS
  • area: main
  • in suites: buster
  • size: 7,088 kB
  • sloc: php: 27,462; perl: 8,241; sh: 1,680; sql: 1,355; xml: 1,041; makefile: 34
file content (93 lines) | stat: -rw-r--r-- 2,936 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
###############################################################################
## Copyright 2005-2016 OCSInventory-NG/OCSInventory-Server contributors.
## See the Contributors file for more details about them.
##
## This file is part of OCSInventory-NG/OCSInventory-ocsreports.
##
## OCSInventory-NG/OCSInventory-Server is free software: you can redistribute
## it and/or modify it under the terms of the GNU General Public License as
## published by the Free Software Foundation, either version 2 of the License,
## or (at your option) any later version.
##
## OCSInventory-NG/OCSInventory-Server is distributed in the hope that it
## will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty
## of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
## GNU General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with OCSInventory-NG/OCSInventory-ocsreports. if not, write to the
## Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
## MA 02110-1301, USA.
################################################################################
package Apache::Ocsinventory::Interface::AssetCategory;

use Apache::Ocsinventory::Map;
use Apache::Ocsinventory::Interface::Database;
use Apache::Ocsinventory::Interface::Internals;

use strict;
use warnings;

use DBI qw(:sql_types);

require Exporter;

our @ISA = qw /Exporter/;

our @EXPORT = qw /
  set_asset_category
/;

sub set_asset_category{
    my @cats = get_asset_category();
    my $dbh = $Apache::Ocsinventory::CURRENT_CONTEXT{'DBI_HANDLE'};
    my $hardware = $Apache::Ocsinventory::CURRENT_CONTEXT{'XML_ENTRY'}->{CONTENT}->{HARDWARE};

    foreach my $cat (@cats) {
        my @args = split(/,/, $cat->{SQL_ARGS});
        my @part_query = split(/\?/, $cat->{SQL_QUERY});

        my $query;
        my $execution;

        for (my $i = 0; $i < scalar @part_query; $i++){
            $query .= @part_query[$i] . @args[$i];
        }

        $execution = $dbh->prepare($query);
        $execution->execute();

        while (my $row = $execution->fetchrow_hashref()){
            if ($row->{hardwareNAME} eq $hardware->{NAME}) {
                $Apache::Ocsinventory::CURRENT_CONTEXT{'XML_ENTRY'}->{CONTENT}->{HARDWARE}->{CATEGORY_ID} = $cat->{ID};
            }
        }
    }
    return 1;
}

sub get_asset_category{
    my $sth;
    my $dbh = $Apache::Ocsinventory::CURRENT_CONTEXT{'DBI_HANDLE'};
    my $sql;
    my @cats;

    $sql = "SELECT ID, SQL_QUERY, SQL_ARGS FROM assets_categories";

    my $result = $dbh->prepare($sql);
    $result->execute;

    while( my $row = $result->fetchrow_hashref() ){
        my $sql_query = $row->{SQL_QUERY};
        $sql_query =~ s/%s/?/g;

        push @cats, {
            'ID' => $row->{ID},
            'SQL_QUERY' => $sql_query,
            'SQL_ARGS' =>  $row->{SQL_ARGS}
        }
    }

    return @cats;
}
1;