File Coverage

blib/lib/Quota/OO.pm

Criterion	Covered	Total	%
statement	18	100	18.0
branch	0	44	0.0
condition	0	3	0.0
subroutine	6	33	18.1
pod	15	27	55.5
total	39	207	18.8

package Quota::OO;

use strict;
use warnings;

use version;our $VERSION = qv('0.0.1');

use Class::Std;
use Class::Std::Utils;

use Quota;

{
    my %dev :ATTR('set' => 'dev', 'get' => 'dev', 'init_arg' => 'dev');
    my %uid :ATTR('set' => 'uid', 'get' => 'uid', 'init_arg' => 'uid');
    my %gid :ATTR('set' => 'gid', 'get' => 'gid', 'init_arg' => 'gid');

    my %rpc_host :ATTR('set' => 'rpc_host', 'get' => 'rpc_host', 'init_arg' => 'rpc_host');
    my %rpc_path :ATTR('set' => 'rpc_path', 'get' => 'rpc_path', 'init_arg' => 'rpc_path');
    my %rpc_port :ATTR('set' => 'rpc_port', 'get' => 'rpc_port', 'init_arg' => 'rpc_port');
    my %rpc_uid  :ATTR('set' => 'rpc_uid',  'get' =>  'rpc_uid', 'init_arg' => 'rpc_uid');
    my %rpc_gid  :ATTR('set' => 'rpc_gid',  'get' =>  'rpc_gid', 'init_arg' => 'rpc_gid');
    my %rpc_use_tcp :ATTR('set' => 'rpc_use_tcp', 'get' => 'rpc_use_tcp', 'init_arg' => 'rpc_use_tcp');
    my %rpc_timeout :ATTR('set' => 'rpc_timeout', 'get' => 'rpc_timeout', 'init_arg' => 'rpc_timeout');

    sub BUILD {
        my ($self, $obj_id, $arg_ref) = @_;

        $dev{$obj_id} = $self->set_dev( $arg_ref->{'dev'} );
        $uid{$obj_id} = $self->set_uid( $arg_ref->{'uid'} );
        $gid{$obj_id} = $self->set_gid( $arg_ref->{'gid'} );
    }

    sub set_dev {
        my ($self, $dev)	= @_;
        $dev{ ident $self }   = $self->getqcarg($dev);
    }

    sub set_uid {
        my ($self, $uid) = @_;
        $uid{ ident $self } = $uid =~ m{^\d+$} ? $uid : $>;
    }

    sub set_gid {
        my ($self, $gid) = @_;
        $gid{ ident $self }
            = $gid =~ m{^\d+$} ? $gid : ( getpwuid($uid{ ident $self }) )[3];
    }

#### do rcp* ##
#### Quota::rpcquery ##
	sub rpcquery {
    	my ($self, $host, $path, $uid) = @_;

        Quota::rpcquery(
	        defined $host ? $host : $rpc_host{ ident $self },
	        defined $path ? $path : $rpc_path{ ident $self },
	        defined $uid  ? $uid  : $rpc_uid{ ident $self },
	    );
	}

#### Quota::rpcpeer ##
	sub rpcpeer {
    	my ($self, $port, $use_tcp, $timeout) = @_;

        Quota::rpcpeer(
	        defined $port    ? $port    : $rpc_port{ ident $self },
	        defined $use_tcp ? $use_tcp : $rpc_use_tcp{ ident $self },
	        defined $timeout ? $timeout : $rpc_timeout{ ident $self },
	    );
	}

#### Quota::rpcauth ##
	sub rpcauth {
    	my ($self, $uid, $gid, $ host) = @_;

        Quota::rpcauth(
	        defined $uid  ? $uid  : $rpc_uid{ ident $self },
	        defined $gid  ? $gid  : $rpc_gid{ ident $self },
	        defined $host ? $host : $rpc_host{ ident $self },
	    );
	}

	sub get_rpc_query {
    	my ($self, $args_hr) = @_;

    	$self->rpcpeer(
            $args_hr->{'rpc_port'},
            $args_hr->{'rpc_use_tcp'},
            $args_hr->{'rpc_timeout'},
        );

    	$self->rpcauth(
            $args_hr->{'rpc_uid'},
            $args_hr->{'rpc_gid'},
            $args_hr->{'rpc_host'},
        );

        my @rpc_results = $self->rcpquery(
            				  $args_hr->{'rpc_host'},
            		 		  $args_hr->{'rpc_path'},
            		          $args_hr->{'rpc_uid'},
                          );
    	return wantarray ? @rpc_results : \@rpc_results;

    }

	sub get_rpc_query_hash {
    	my ($self, $args_hr) = @_;

    	my $query_ar = $self->get_rpc_query( $args_hr );

    	my $query_hr = {
        	'block_current'   => $query_ar->[0],
        	'block_soft'      => $query_ar->[1],
        	'block_hard'      => $query_ar->[2],
        	'block_timelimit' => $query_ar->[3],
        	'inode_current'   => $query_ar->[4],
        	'inode_soft'      => $query_ar->[5],
        	'inode_hard'      => $query_ar->[6],
        	'inode_timelimit' => $query_ar->[7],
    	};

    	return wantarray ? %{ $query_hr } : $query_hr;
	}

#### Quota::query ##
    sub query {
       my($self, $dev, $id, $id_is_group) = @_;

       my $original_dev = $dev{ ident $self };
       $dev{ ident $self } = $self->getqcarg( $dev );

       my @query = $id_is_group ? $self->get_gid_query($id)
                                : $self->get_uid_query($id);

       $dev{ ident $self } = $original_dev;
       return @query;
    }

    sub get_query {
        my($self, $id, $id_is_group) = @_;

        $id_is_group = 0 if !defined $id_is_group;

        if(!defined $id) {
        	$id = $id_is_group  ? $gid{ ident $self }
                                : $uid{ ident $self };
        }

        return wantarray ?  $self->query($dev{ ident $self }, $id, $id_is_group)
                         : [$self->query($dev{ ident $self }, $id, $id_is_group)];
    }

    sub get_uid_query {
        my($self, $uid) = @_;
        return $self->get_query($uid, 0);
    }

    sub get_gid_query {
        my($self, $gid) = @_;
        return $self->get_query($gid, 1);
    }

    sub get_query_hash {
        my($self, $id, $id_is_group) = @_;

        my $query_ar = $self->get_query($id, $id_is_group);

        my $query_hr = {
            'block_current'   => $query_ar->[0],
            'block_soft'      => $query_ar->[1],
            'block_hard'      => $query_ar->[2],
            'block_timelimit' => $query_ar->[3],
            'inode_current'   => $query_ar->[4],
            'inode_soft'      => $query_ar->[5],
            'inode_hard'      => $query_ar->[6],
            'inode_timelimit' => $query_ar->[7],
        };

        return wantarray ? %{ $query_hr } : $query_hr;
    }

    sub get_uid_query_hash {
        my($self, $uid) = @_;
        return $self->get_query_hash($uid, 0);
    }

    sub get_gid_query_hash {
        my($self, $uid) = @_;
        return $self->get_query_hash($uid, 1);
    }

#### Quota::setqlim ##
    sub setqlim {
	    shift;
        Quota::setqlim(@_);
    }

    sub set_quota_limit {
	    my($self, $args_hr) = @_;

        my $dev = !defined $args_hr->{'dev'} ? $dev{ ident $self }
		                                     : $self->getqcarg( $args_hr->{'dev'} );
		$args_hr->{'id'}  = $uid{ ident $self } if !defined $args_hr->{'uid'};
		$args_hr->{'block_soft'} = int( $args_hr->{'block_soft'} );
		$args_hr->{'block_hard'} = int( $args_hr->{'block_hard'} );
	    $args_hr->{'inode_soft'} = int( $args_hr->{'inode_soft'} );
		$args_hr->{'inode_hard'} = int( $args_hr->{'inaode_hard'} );
		$args_hr->{'time_limit'} = 0 if !defined $args_hr->{'time_limit'};
		$args_hr->{'id_is_gid'}  = 0 if !defined $args_hr->{'id_is_gid'};

        $self->setqlim(
			$dev,
			$args_hr->{'id'},
			$args_hr->{'block_soft'},
			$args_hr->{'block_hard'},
			$args_hr->{'inode_soft'},
			$args_hr->{'inode_hard'},
			$args_hr->{'time_limit'},
			$args_hr->{'id_is_gid'},
	    );
    }

    sub set_uid_quota_limit {
        my($self, $args_hr) = @_;
        $args_hr->{'id_is_gid'} = 0;
        $self->set_quota_limit($args_hr);
    }

    sub set_gid_quota_limit {
        my($self, $args_hr) = @_;
        $args_hr->{'id_is_gid'} = 1;
        $self->set_quota_limit($args_hr);
    }

#### Quota::sync ##
    sub sync {
        my($self, $dev) = @_;
        Quots::sync( $self->getqcarg( $dev ) || $dev{ ident $self } );
    }

    sub sync_all {
        Quota::sync();
    }

#### Quota::getqcarg ##
    sub getqcarg {
	    shift;
        Quota::getqcarg(@_);
    }

#### Quota::getdev ##
    sub getdev {
        shift;
        Quota::getdev(@_);
    }

#### Quota::strerr ##
    sub strerr {
        Quota::strerr();
    }

    sub get_errstr {
        Quota::strerr();
    }

#### *mntent ##
#### Quota::setmntent ##
    sub setmntent {
	    Quota::setmntent();
    }

#### Quota::getmntent ##
    sub getmntent {
        Quota::getmntent();
    }

#### Quota::endmntent ##
    sub endmntent {
	    Quota::endmntent();
    }

    sub get_mntent_hash {
	     my($self) = @_;
	     my $hr = {};
	     $self->setmntent();
	     while( my @ent = $self->getmntent() ){
		     $hr->{ $ent[0] } = {
			     'path' => $ent[1],
			     'type' => $ent[2],
			     'opts' => $ent[3],
	   	     };
	     }
	     $self->endmntent();

	     return wantarray ? %{ $hr } : $hr;
    }
}

1;

__END__