#---------------------------------------------------------------
# ____                   _ _         ____  _
#|  _ \ ___  _   _  __ _| | |_ _   _/ ___|| |__   __ _ _ __ ___
#| |_) / _ \| | | |/ _` | | __| | | \___ \| '_ \ / _` | '__/ _ \
#|  _ < (_) | |_| | (_| | | |_| |_| |___) | | | | (_| | | |  __/
#|_| \_\___/ \__, |\__,_|_|\__|\__, |____/|_| |_|\__,_|_|  \___|
#            |___/             |___/
#
# Copyright (C) 2009 RoyaltyShare, Inc.      All Rights Reserved
#---------------------------------------------------------------
package Job::HostClassSet;

use strict;
use Data::Dumper;
use lib '/app/tools/common/lib';
use Common::FormHash;

use lib '/app/tools/job/lib';
use Job::DB::Item::JobHostClass;
use Job::HostClass;

use base 'Common::FormHash';

sub _tagName {
    return 'JobHostClass';
}

sub _getItemCollection {
    my ( $self, %args ) = @_;

    my $collection;
    if ( $self->{_hostname} ) {
        $collection = Job::DB::Item::JobHostClass->GetAllForHostname( $self->{_hostname} );
    } else {
        $collection = Job::DB::Item::JobHostClass->GetAll();
    }

    return $collection;
}

sub _keyMethodName {
    return 'JobHostClassID';
}

sub _getNewObject {
    my ($self) = @_;

    if ( $self->{_hostname} ) {
        return Job::HostClass->new( hash => { hostname => $self->{_hostname} } );
    }

    return Job::HostClass->new();
}

sub _getObjectFromDBItem {
    my ( $self, $dbItem ) = @_;
    return Job::HostClass->new( dbItem => $dbItem );
}

sub _init {
    my ( $self, %args ) = @_;

    if ( $args{hostname} ) {
        $self->{_hostname} = $args{hostname};
    }

    return $self->SUPER::_init(%args);
}

sub getHash {
    my ($self) = @_;
    return $self->{JobHostClass};
}

# We want to sort by hostname and class
#
sub getKeys {
    my ($self) = @_;

    my $tag  = $self->_tagName();
    my $hash = $self->getHash();

    my @keys =
      sort { $hash->{$a}->Hostname() cmp $hash->{$b}->Hostname() || $hash->{$a}->Class() cmp $hash->{$b}->Class() } keys %{ $self->{$tag} };

    return \@keys;
}

sub save {
    my ($self) = @_;

    my $keys = $self->getKeys();

    foreach my $id (@$keys) {
        my $rule = $self->{JobHostClass}{$id};
        $rule->save();
    }
}

1;