Sponsoring The Perl Toolchain Summit 2025: Help make this important event another success Learn more

use Mojo::Base -strict;
use Carp 'croak';
use Exporter 'import';
use Mojo::Util 'deprecated';
use Scalar::Util 'blessed';
# DEPRECATED in Tiger Face!
use overload '""' => sub {
deprecated 'Stringification support in Mojo::Collection is DEPRECATED'
. ' in favor of Mojo::Collection::join';
shift->join("\n");
};
use overload bool => sub {1}, fallback => 1;
our @EXPORT_OK = ('c');
# DEPRECATED in Tiger Face!
sub AUTOLOAD {
my $self = shift;
my ($package, $method) = our $AUTOLOAD =~ /^(.+)::(.+)$/;
deprecated "Mojo::Collection::AUTOLOAD ($method) is DEPRECATED"
. ' in favor of Mojo::Collection::map';
croak "Undefined subroutine &${package}::$method called"
unless blessed $self && $self->isa(__PACKAGE__);
return $self->map($method, @_);
}
# DEPRECATED in Tiger Face!
sub DESTROY { }
sub c { __PACKAGE__->new(@_) }
sub compact {
$_[0]->new(grep { defined && (ref || length) } @{$_[0]});
}
sub each {
my ($self, $cb) = @_;
return @$self unless $cb;
my $i = 1;
$_->$cb($i++) for @$self;
return $self;
}
sub first {
my ($self, $cb) = @_;
return $self->[0] unless $cb;
return List::Util::first { $cb->($_) } @$self if ref $cb eq 'CODE';
return List::Util::first { $_ =~ $cb } @$self;
}
sub flatten { $_[0]->new(_flatten(@{$_[0]})) }
sub grep {
my ($self, $cb) = @_;
return $self->new(grep { $cb->($_) } @$self) if ref $cb eq 'CODE';
return $self->new(grep { $_ =~ $cb } @$self);
}
sub join {
Mojo::ByteStream->new(join $_[1] // '', map {"$_"} @{$_[0]});
}
sub last { shift->[-1] }
sub map {
my ($self, $cb) = (shift, shift);
return $self->new(map { $_->$cb(@_) } @$self);
}
sub new {
my $class = shift;
return bless [@_], ref $class || $class;
}
# DEPRECATED in Tiger Face!
sub pluck {
deprecated
'Mojo::Collection::pluck is DEPRECATED in favor of Mojo::Collection::map';
my ($self, $key) = (shift, shift);
return $self->new(map { ref eq 'HASH' ? $_->{$key} : $_->$key(@_) } @$self);
}
sub reduce {
my $self = shift;
@_ = (@_, @$self);
goto &List::Util::reduce;
}
sub reverse { $_[0]->new(reverse @{$_[0]}) }
sub shuffle { $_[0]->new(List::Util::shuffle @{$_[0]}) }
sub size { scalar @{$_[0]} }
sub slice {
my $self = shift;
return $self->new(@$self[@_]);
}
sub sort {
my ($self, $cb) = @_;
return $self->new(sort @$self) unless $cb;
my $caller = caller;
no strict 'refs';
my @sorted = sort {
local (*{"${caller}::a"}, *{"${caller}::b"}) = (\$a, \$b);
$a->$cb($b);
} @$self;
return $self->new(@sorted);
}
sub tap { shift->Mojo::Base::tap(@_) }
sub to_array { [@{shift()}] }
sub uniq {
my %seen;
return $_[0]->new(grep { !$seen{$_}++ } @{$_[0]});
}
sub _flatten {
map { _ref($_) ? _flatten(@$_) : $_ } @_;
}
sub _ref { ref $_[0] eq 'ARRAY' || blessed $_[0] && $_[0]->isa(__PACKAGE__) }
1;
=encoding utf8
=head1 NAME
Mojo::Collection - Collection
=head1 SYNOPSIS
use Mojo::Collection;
# Manipulate collection
my $collection = Mojo::Collection->new(qw(just works));
unshift @$collection, 'it';
say $collection->join("\n");
# Chain methods
$collection->map(sub { ucfirst })->shuffle->each(sub {
my ($word, $num) = @_;
say "$num: $word";
});
# Use the alternative constructor
use Mojo::Collection 'c';
c(qw(a b c))->join('/')->url_escape->say;
=head1 DESCRIPTION
L<Mojo::Collection> is an array-based container for collections.
# Access array directly to manipulate collection
my $collection = Mojo::Collection->new(1 .. 25);
$collection->[23] += 100;
say for @$collection;
=head1 FUNCTIONS
L<Mojo::Collection> implements the following functions, which can be imported
individually.
=head2 c
my $collection = c(1, 2, 3);
Construct a new array-based L<Mojo::Collection> object.
=head1 METHODS
L<Mojo::Collection> implements the following methods.
=head2 compact
my $new = $collection->compact;
Create a new collection with all elements that are defined and not an empty
string.
# "0, 1, 2, 3"
Mojo::Collection->new(0, 1, undef, 2, '', 3)->compact->join(', ');
=head2 each
my @elements = $collection->each;
$collection = $collection->each(sub {...});
Evaluate callback for each element in collection or return all elements as a
list if none has been provided. The element will be the first argument passed
to the callback and is also available as C<$_>.
# Make a numbered list
$collection->each(sub {
my ($e, $num) = @_;
say "$num: $e";
});
=head2 first
my $first = $collection->first;
my $first = $collection->first(qr/foo/);
my $first = $collection->first(sub {...});
Evaluate regular expression or callback for each element in collection and
return the first one that matched the regular expression, or for which the
callback returned true. The element will be the first argument passed to the
callback and is also available as C<$_>.
# Find first value that contains the word "mojo"
my $interesting = $collection->first(qr/mojo/i);
# Find first value that is greater than 5
my $greater = $collection->first(sub { $_ > 5 });
=head2 flatten
my $new = $collection->flatten;
Flatten nested collections/arrays recursively and create a new collection with
all elements.
# "1, 2, 3, 4, 5, 6, 7"
Mojo::Collection->new(1, [2, [3, 4], 5, [6]], 7)->flatten->join(', ');
=head2 grep
my $new = $collection->grep(qr/foo/);
my $new = $collection->grep(sub {...});
Evaluate regular expression or callback for each element in collection and
create a new collection with all elements that matched the regular expression,
or for which the callback returned true. The element will be the first
argument passed to the callback and is also available as C<$_>.
# Find all values that contain the word "mojo"
my $interesting = $collection->grep(qr/mojo/i);
# Find all values that are greater than 5
my $greater = $collection->grep(sub { $_ > 5 });
=head2 join
my $stream = $collection->join;
my $stream = $collection->join("\n");
Turn collection into L<Mojo::ByteStream>.
# Join all values with commas
$collection->join(', ')->say;
=head2 last
my $last = $collection->last;
Return the last element in collection.
=head2 map
my $new = $collection->map(sub {...});
my $new = $collection->map($method);
my $new = $collection->map($method, @args);
Evaluate callback for, or call method on, each element in collection and
create a new collection from the results. The element will be the first
argument passed to the callback and is also available as C<$_>.
# Longer version
my $new = $collection->map(sub { $_->$method(@args) });
# Append the word "mojo" to all values
my $mojoified = $collection->map(sub { $_ . 'mojo' });
=head2 new
my $collection = Mojo::Collection->new(1, 2, 3);
Construct a new array-based L<Mojo::Collection> object.
=head2 reduce
my $result = $collection->reduce(sub {...});
my $result = $collection->reduce(sub {...}, $initial);
Reduce elements in collection with callback, the first element will be used as
initial value if none has been provided.
# Calculate the sum of all values
my $sum = $collection->reduce(sub { $a + $b });
# Count how often each value occurs in collection
my $hash = $collection->reduce(sub { $a->{$b}++; $a }, {});
=head2 reverse
my $new = $collection->reverse;
Create a new collection with all elements in reverse order.
=head2 slice
my $new = $collection->slice(4 .. 7);
Create a new collection with all selected elements.
# "B C E"
Mojo::Collection->new('A', 'B', 'C', 'D', 'E')->slice(1, 2, 4)->join(' ');
=head2 shuffle
my $new = $collection->shuffle;
Create a new collection with all elements in random order.
=head2 size
my $size = $collection->size;
Number of elements in collection.
=head2 sort
my $new = $collection->sort;
my $new = $collection->sort(sub {...});
Sort elements based on return value of callback and create a new collection
from the results.
# Sort values case insensitive
my $insensitive = $collection->sort(sub { uc($a) cmp uc($b) });
=head2 tap
$collection = $collection->tap(sub {...});
Alias for L<Mojo::Base/"tap">.
=head2 to_array
my $array = $collection->to_array;
Turn collection into array reference.
=head2 uniq
my $new = $collection->uniq;
Create a new collection without duplicate elements.
# "foo bar baz"
Mojo::Collection->new('foo', 'bar', 'bar', 'baz')->uniq->join(' ');
=head1 SEE ALSO
L<Mojolicious>, L<Mojolicious::Guides>, L<http://mojolicio.us>.
=cut