File Coverage

blib/lib/Math/CheckDigits.pm

Criterion	Covered	Total	%
statement	14	67	20.9
branch	0	32	0.0
condition	0	6	0.0
subroutine	5	12	41.6
pod	0	6	0.0
total	19	123	15.4

package Math::CheckDigits;

use 5.006;
use strict;
use warnings;
use integer;
use utf8;
our $VERSION = '0.01_01';
$VERSION = eval $VERSION;

my %DEFAULT = (
    TRANS_TABLE => {},
    OPTIONS     => {
        start_at_right  => 1, # multipule
        DSR             => 1, # use DSR or DR
        runes           => 0, # use runes
    },
);

sub new {
    my $cls = shift;
    my $self = \%DEFAULT;
    if ( @_ == 2 ){
        ( $self->{modulus}, $self->{weight} ) = @_
    }
    else{
        $self = { %$self, ref $_[0] ? %{$_[0]} : @_ };
    }
    die 'not enough arguments!'
        if !$self->{modulus} || !$self->{weight};

    bless $self, $cls;
}

sub checkdigit {
    my $self = shift;
    my @digits = split //, shift;

    @digits = reverse @digits if $self->options('start_at_right');

    # only support non runes format at present.
    my $check_sum = $self->_calc_check_sum( $self->{weight}, @digits );
    my $check_digit = $check_sum % $self->{modulus};

    # DSR or DR ?
    $check_digit = $self->{modulus} - $check_digit if $self->options('DSR');

    # see trans table if exists. ( eg. 16 => 'g' )
    my %trans_table = $self->trans_table;
    $check_digit =
        defined $trans_table{$check_digit} ?
        $trans_table{$check_digit} : $check_digit;
    $check_digit = 0 if length( $check_digit ) >= 2;

    return $check_digit;
}

sub is_valid {
    my ( $self, $digits ) = @_;
    ( $digits, my $check_num )
        = $digits =~ /^(.*)(.)$/;
    return $self->checkdigit( $digits ) == $check_num;
}

sub complete {
    my ( $self, $digits ) = @_;
    return $digits . $self->checkdigit( $digits );
}

sub trans_table {
    my $self = shift;
    if ( @_ ){
        $self->{TRANS_TABLE} = ref $_[0] ? shift : { @_ };
        return $self;
    }
    return %{$self->{TRANS_TABLE}};
}

sub options {
    my $self = shift;

    return %{$self->{OPTIONS}} if @_ == 0;
    return $self->{OPTIONS}{$_[0]} if (@_ == 1) && (!ref $_[0]);

    $self->{OPTIONS}
        = { %{$self->{OPTIONS}}, ref $_[0] ? %{$_[0]} : @_ };

    return $self;
}

sub _calc_check_sum {
    my $self = shift;
    my ( $weight, @digits ) = @_;

    my %trans_table = reverse $self->trans_table;
    for ( keys %trans_table ){
        delete $trans_table{$_} if /\d/;
    }

    my ( $i, $check_sum ) = ( 0, 0 );
    for my $digit ( @digits ){
        my $num = defined $trans_table{$digit} ? $trans_table{$digit} : $digit;
        die "'$num' does not map to number. Use trans_table method." if $num =~ /\D/;

        $num = $weight->[ $i % @$weight ] * $num;
        if ( !$self->options('runes') ){
            $check_sum += $num;
        }
        else{
            my @nums = split //, $num;
            $check_sum += $_ for @nums;
        }
        $i++;
    }

    return $check_sum;
}

1;
__END__