File Coverage

blib/lib/Net/Stomp/Frame.pm

Criterion	Covered	Total	%
statement	48	55	87.2
branch	15	22	68.1
condition	4	5	80.0
subroutine	9	9	100.0
pod	3	3	100.0
total	79	94	84.0

package Net::Stomp::Frame;
use strict;
use warnings;

our $VERSION='0.61';

use base 'Class::Accessor::Fast';
__PACKAGE__->mk_accessors(qw(command headers body));

BEGIN {
    for my $header (
        qw(destination exchange content-type content-length message-id reply-to))
    {
        my $method = $header;
        $method =~ s/-/_/g;
        no strict 'refs';
        *$method = sub {
            my $self = shift;
            $self->headers->{$header} = shift if @_;
            $self->headers->{$header};
        };
    }
}

sub new {
    my $class = shift;
    my $self  = $class->SUPER::new(@_);
    $self->headers({}) unless defined $self->headers;
    return $self;
}

sub as_string {
    my $self    = shift;
    my $command = $self->command;
    my $headers = $self->headers;
    my $body    = $self->body;
    my $frame   = $command . "\n";

    # insert a content-length header
    my $bytes_message = 0;
    if ( $headers->{bytes_message} ) {
        $bytes_message = 1;
        delete $headers->{bytes_message};
        $headers->{"content-length"} = length( $self->body );
    }

    while ( my ( $key, $value ) = each %{ $headers || {} } ) {
        $frame .= $key . ':' . (defined $value ? $value : '') . "\n";
    }
    $frame .= "\n";
    $frame .= $body || '';
    $frame .= "\0";
}

sub parse {
    my ($class,$string) = @_;

    $string =~ s{
      \A\s*
      ([A-Z]+)\n #command
      ((?:[^\n]+\n)*)\n # header block
    }{}smx;
    my ($command,$headers_block) = ($1,$2);

    return unless $command;

    my ($headers,$body);
    if ($headers_block) {
        foreach my $line (split(/\n/, $headers_block)) {
            my ($key, $value) = split(/\s*:\s*/, $line, 2);
            $headers->{$key} = $value;
        }
    }

    if ($headers && $headers->{'content-length'}) {
        if (length($string) >= $headers->{'content-length'}) {
            $body = substr($string,
                           0,
                           $headers->{'content-length'},
                           '' );
        }
        else { return } # not enough body
    } elsif ($string =~ s/\A(.*?)\0//s) {
        # No content-length header.
        $body = $1 if length($1);
    }
    else { return } # no body

    return $class->new({
        command => $command,
        headers => $headers,
        body => $body,
    });
}

1;

__END__