File Coverage

blib/lib/JavaScript/Writer/Function.pm

Criterion	Covered	Total	%
statement	16	18	88.8
branch			n/a
condition			n/a
subroutine	6	6	100.0
pod			n/a
total	22	24	91.6

package JavaScript::Writer::Function;

use strict;
use warnings;
use 5.008;
use base 'Class::Accessor::Fast';

__PACKAGE__->mk_accessors(qw[ name body ]);

use overload '""' => \&as_string;

our $VERSION = '0.0.2';

use JavaScript::Writer::Block;

sub new {
    my $class = shift;

    my $self = bless {}, $class;
    if (ref($_[0]) eq 'CODE') {
        $self->body($_[0]);
    }
    else {
        my %args = @_;
        while ( my ($k, $v) = each %args) {
            $self->$k($v);
        }
    }

    return $self;
}

sub arguments {
    my ($self, @args) = @_;
    if (@args) {
        $self->{arguments} = \@args;
        return $self;
    }
    return @{$self->{arguments}||[]}
}

sub as_string {
    my $self = shift;
    my $sub = $self->body;
    my $function_body = JavaScript::Writer::Block->new;
    $function_body->body($self->body);
    my $name = $self->name ? " $self->{name}" : "";
    my $args = join(",", $self->arguments);
    return "function${name}($args)${function_body}";
}

1;
__END__