Commit | Line | Data |
32b29b79 |
1 | package HTTP::Body; |
2 | |
3 | use strict; |
4 | |
58050177 |
5 | use Carp qw[ ]; |
6 | use List::Util qw[ first ]; |
32b29b79 |
7 | |
7e2df1d9 |
8 | our $TYPES = { |
4f5db602 |
9 | 'application/octet-stream' => 'HTTP::Body::OctetStream', |
10 | 'application/x-www-form-urlencoded' => 'HTTP::Body::UrlEncoded', |
11 | 'multipart/form-data' => 'HTTP::Body::MultiPart' |
32b29b79 |
12 | }; |
13 | |
14 | sub new { |
15 | my ( $class, $content_type, $content_length ) = @_; |
16 | |
17 | unless ( @_ == 3 ) { |
18 | Carp::croak( $class, '->new( $content_type, $content_length )' ); |
19 | } |
7e2df1d9 |
20 | |
21 | my $type = first { index( lc($content_type), $_ ) >= 0 } keys %{$TYPES}; |
22 | my $body = $TYPES->{ $type || 'application/octet-stream' }; |
23 | |
32b29b79 |
24 | eval "require $body"; |
7e2df1d9 |
25 | |
26 | if ($@) { |
32b29b79 |
27 | die $@; |
28 | } |
7e2df1d9 |
29 | |
32b29b79 |
30 | my $self = { |
31 | buffer => '', |
32 | content_length => $content_length, |
33 | content_type => $content_type, |
58050177 |
34 | length => 0, |
7e2df1d9 |
35 | param => {}, |
36 | state => 'buffering', |
37 | upload => {} |
32b29b79 |
38 | }; |
39 | |
40 | bless( $self, $body ); |
7e2df1d9 |
41 | |
32b29b79 |
42 | return $self->init; |
43 | } |
44 | |
45 | sub add { |
58050177 |
46 | my $self = shift; |
7e2df1d9 |
47 | |
58050177 |
48 | if ( defined $_[0] ) { |
49 | $self->{buffer} .= $_[0]; |
7e2df1d9 |
50 | $self->{length} += length( $_[0] ); |
58050177 |
51 | } |
52 | |
7e2df1d9 |
53 | unless ( $self->state eq 'done' ) { |
54 | $self->spin; |
55 | } |
56 | |
58050177 |
57 | return ( $self->length - $self->content_length ); |
32b29b79 |
58 | } |
59 | |
60 | sub body { |
61 | my $self = shift; |
62 | $self->{body} = shift if @_; |
63 | return $self->{body}; |
64 | } |
65 | |
58050177 |
66 | sub buffer { |
67 | return shift->{buffer}; |
68 | } |
69 | |
32b29b79 |
70 | sub content_length { |
71 | return shift->{content_length}; |
72 | } |
73 | |
74 | sub content_type { |
75 | return shift->{content_type}; |
76 | } |
77 | |
58050177 |
78 | sub init { |
79 | return $_[0]; |
80 | } |
81 | |
82 | sub length { |
83 | return shift->{length}; |
84 | } |
85 | |
86 | sub spin { |
87 | Carp::croak('Define abstract method spin() in implementation'); |
88 | } |
89 | |
7e2df1d9 |
90 | sub state { |
91 | my $self = shift; |
92 | $self->{state} = shift if @_; |
93 | return $self->{state}; |
94 | } |
95 | |
32b29b79 |
96 | sub param { |
97 | my $self = shift; |
98 | |
99 | if ( @_ == 2 ) { |
100 | |
101 | my ( $name, $value ) = @_; |
102 | |
103 | if ( exists $self->{param}->{$name} ) { |
104 | for ( $self->{param}->{$name} ) { |
105 | $_ = [$_] unless ref($_) eq "ARRAY"; |
106 | push( @$_, $value ); |
107 | } |
108 | } |
109 | else { |
110 | $self->{param}->{$name} = $value; |
111 | } |
112 | } |
113 | |
114 | return $self->{param}; |
115 | } |
116 | |
117 | sub upload { |
118 | my $self = shift; |
119 | |
120 | if ( @_ == 2 ) { |
121 | |
122 | my ( $name, $upload ) = @_; |
123 | |
124 | if ( exists $self->{upload}->{$name} ) { |
125 | for ( $self->{upload}->{$name} ) { |
126 | $_ = [$_] unless ref($_) eq "ARRAY"; |
127 | push( @$_, $upload ); |
128 | } |
129 | } |
130 | else { |
131 | $self->{upload}->{$name} = $upload; |
132 | } |
133 | } |
134 | |
135 | return $self->{upload}; |
136 | } |
137 | |
138 | 1; |