File Coverage

blib/lib/Gantry/Utils/Threeway.pm

Criterion	Covered	Total	%
statement	3	62	4.8
branch	0	20	0.0
condition	0	15	0.0
subroutine	1	3	33.3
pod	2	2	100.0
total	6	102	5.8

package Gantry::Utils::Threeway;
use strict;

############################################################
# Functions                                                #
############################################################
sub new {
    my ( $class, $opt ) = @_;

    my $self = { options => $opt };
    bless( $self, $class );

    my @errors;
    foreach( qw/join_table primary_table secondary_table self/ ) {
        push( @errors, "$_ is not set properly" ) if ! $opt->{$_};
    }

    if ( scalar( @errors ) ) {
        die join( "\n", @errors );
    }

    # populate self with data from site
    return( $self );

} # end new

#-------------------------------------------------
# process()
#-------------------------------------------------
sub process {
    my( $self ) = @_;

    my $gself = $self->{options}{self};

    my $template        = $self->{options}{template} || 'threeway.tt';
    my $join_table      = $self->{options}{join_table};
    my $form_action     = $self->{options}{action} || $gself->uri;
    my $type            = $self->{options}{type} || 'checkbox';
    my $primary_table   = $self->{options}{primary_table};
    my $secondary_table = $self->{options}{secondary_table};
    my $redirect_loc    = $self->{options}{redirect_loc} || $gself->location;
    my $order_by        = $self->{options}{order_by} || 'id';
    my $legend          = $self->{options}{legend} || "Add ${secondary_table}s";

    my $sch   = $gself->get_schema();
    my %param = $gself->get_param_hash;

    # pull from the member table
    my $selected = $sch->resultset( $join_table )->search(
        { $primary_table => $self->{options}{primary_id} }, { }
    );

    my %selected;
    while ( my $r = $selected->next ) {
        $selected{$r->$secondary_table . ''} = $r->id;
    }

    if ( $gself->is_post() ) {

        if ( $param{cancel} ) {
            $gself->relocate( $redirect_loc );
            return;
        }

        my $available = $sch->resultset( $secondary_table )->search(
            {}, { order_by => $order_by }
        );

        my %available;
        while ( my $r = $available->next ) {
            ++$available{$r->id . ''};
        }

        my $values_ref;
        if ( $type eq 'checkbox' ) {
            foreach my $k ( keys %param ) {
                next if $k !~ /^subscribe/;

                my @val = split( ':', $k );
                push( @{ $values_ref }, $val[1] );
            }
        }
        # multiselect
        else {

            if ( $gself->engine =~ /CGI/ ) {
                $values_ref = $gself->cgi->{subscribe};
            }
            else {
                $values_ref = ref( $gself->params->{'subscribe'} ) eq 'ARRAY'
                    ? $gself->params->{'subscribe'}
                    : [ $gself->params->{'subscribe'} ];
            }
        }

        foreach my $val ( @{ $values_ref } ) {

            $sch->resultset( $join_table )->update_or_create(
                {
                    $primary_table   => $self->{options}{primary_id},
                    $secondary_table => $val }
            );

            # delete from available hash
            delete( $selected{$val} );

        }

        my @remove;
        foreach my $k ( keys %selected ) {
            push( @remove, $selected{$k} );
        }

        if ( scalar( @remove ) > 0 ) {
            $sch->resultset( $join_table )->search(
                id => \@remove
            )->delete;
        }

        $gself->relocate(
            ref( $redirect_loc ) eq 'CODE'
                ? $redirect_loc->( $gself ) : $redirect_loc
        );

        return;

    }

    my $available = $sch->resultset( $secondary_table )->search( {}, {
        order_by => $order_by,
    } );

    $gself->stash->view->template( $template );
    $gself->stash->view->form( {
        legend => $legend,
        action => $form_action
    } );
    $gself->stash->view->data( {
        type      => $type,
        available => $available,
        selected => \%selected
    } );

} # END process

# EOF
1;

__END__