#!/usr/bin/perl
#  Copyright (C) 2002  Stanislav Sinyagin
#
#  This program 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.
#
#  This program 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 this program; if not, write to the Free Software
#  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.

# $Id$
# Stanislav Sinyagin <ssinyagin@yahoo.com>

BEGIN { require '/usr/share/torrus/conf_defaults/torrus-config.pl'; }

use strict;
use Getopt::Long;

use Torrus::Log;
use Torrus::ConfigTree;
use Torrus::SiteConfig;

exit(1) if not Torrus::SiteConfig::verify();

my $tree;
my $initPath = '/';
my $listWhat = 'rrdfiles';
my $selectType = 'all';

my $help_needed;

my %listingsSupported =
    (
     'rrdfiles' => {
         'collector'  => 1,
         'readonly'   => 1,
         'all'        => 1,
     },
     
     'snmphosts' => {
         'collector'  => 1,
     },
     );

my $ok = GetOptions ('tree=s'   => \$tree,
                     'path=s'   => \$initPath,
                     'what=s'   => \$listWhat,
                     'type=s'   => \$selectType,
                     'help'     => \$help_needed);

if( not $ok or not $tree or $help_needed or
    not $listingsSupported{$listWhat}{$selectType} or scalar(@ARGV) > 0 )
{
    print STDERR "Usage: $0 --tree=NAME [options...]\n",
    "Options:\n",
    "  --tree=NAME     tree name\n",
    "  --path=/PATH    [".$initPath."] subtree name\n",
    "  --what=WHAT     [".$listWhat."] what to list\n",
    "    Supported listings:\n",
    "      rrdfiles      List RRD file paths\n",
    "      snmphosts     List SNMP hosts\n",
    "  --type=TYPE     [".$selectType."] selection type\n",
    "    Supported types:\n",
    "      collector     Collector leaves\n",
    "      readonly      Read-only leaves\n",
    "      all           All of above\n",
    "  --help          this help message\n";
    exit 1;
}


if( not Torrus::SiteConfig::treeExists( $tree ) )
{
    Error('Tree ' . $tree . ' does not exist');
    exit 1;
}

&Torrus::DB::setSafeSignalHandlers();

my $config_tree = new Torrus::ConfigTree( -TreeName => $tree, -Wait => 1 );
if( not defined( $config_tree ) )
{
    exit 1;
}

my $initToken = $config_tree->token( $initPath );
if( not defined( $initToken ) )
{
    Error('No such subtree: ' . $initPath);
    exit 1;
}

my $listing = {};

my $listParams = {};
if( $selectType eq 'all' )
{
    foreach my $type ( keys %{$listingsSupported{$listWhat}} )
    {
        if( $type ne 'all' )
        {
            $listParams->{$type} = 1;
        }
    }
}
else
{
    $listParams->{$selectType} = 1;
}

pickup_data( $config_tree, $initToken, $listing, $listWhat, $listParams );

foreach my $item ( sort keys %{$listing} )
{
    print $item, "\n";
}

exit 0;

sub pickup_data
{
    my $config_tree = shift;
    my $token = shift;
    my $listing = shift;
    my $listWhat = shift;
    my $listParams = shift;

    foreach my $ctoken ( $config_tree->getChildren( $token ) )
    {
        &Torrus::DB::checkInterrupted();
        
        if( $config_tree->isSubtree( $ctoken ) )
        {
            pickup_data( $config_tree, $ctoken,
                         $listing, $listWhat, $listParams );
        }
        elsif( $config_tree->isLeaf( $ctoken ) )
        {
            if( $listWhat eq 'rrdfiles' and
                (
                 (
                  $listParams->{'collector'} and
                  $config_tree->getNodeParam( $ctoken, 'ds-type' ) eq
                  'collector' and
                  $config_tree->getNodeParam( $ctoken, 'storage-type' ) eq
                  'rrd'
                  ) or
                 (
                  $listParams->{'readonly'} and
                  $config_tree->getNodeParam( $ctoken, 'ds-type' ) eq
                  'rrd-file' and
                  $config_tree->getNodeParam( $ctoken, 'leaf-type' ) eq
                  'rrd-def'
                  )
                 )
                )
            {
                my $datafile =
                    $config_tree->getNodeParam( $ctoken, 'data-file' );
                my $datadir =
                    $config_tree->getNodeParam( $ctoken, 'data-dir' );
                $listing->{$datadir . '/' . $datafile} = 1;
            }
            elsif( $listWhat eq 'snmphosts' and
                   $listParams->{'collector'} and
                   $config_tree->getNodeParam( $ctoken, 'ds-type' ) eq
                   'collector' )
            {
                my $host =
                    $config_tree->getNodeParam( $ctoken, 'snmp-host' );
                my $oid =
                    $config_tree->getNodeParam( $ctoken, 'snmp-object' );
                
                if( defined( $host ) and length( $host ) > 0 and
                    defined( $oid ) and length( $oid ) > 0 )
                {
                    $listing->{$host} = 1;
                }                
            }
        }
    }
}



# Local Variables:
# mode: perl
# indent-tabs-mode: nil
# perl-indent-level: 4
# End:
