Added the ability to include or exclude specific sites

MANIFEST

@@ -9,5 +9,6 @@ MYMETA.yml
 README.md
 t/00-load.t
 t/01-basic.t
+t/02-include-exclude.t
 t/pod.t
 t/pod_coverage.t

lib/Amazon/Sites.pm

@@ -29,19 +29,38 @@ use Feature::Compat::Class;
 use feature 'signatures';
 no warnings 'experimental::signatures';
 
-our $VERSION = '0.0.5';
+our $VERSION = '0.1.0';
 
 class Amazon::Sites {
   use Amazon::Site;
 
-  field %sites = _init_sites();
+  field $include :param = [];
+  field $exclude :param = [];
+  field %sites = _init_sites($include, $exclude);
+
+  ADJUST {
+    if (@$include and @$exclude) {
+      die "You can't specify both include and exclude";
+    }
+  }
 
 =head1 METHODS
 
 =head2 new
 
 Creates a new Amazon::Sites object.
 
+    my $sites = Amazon::Sites->new;
+
+You can also specify a list of sites to include or exclude:
+
+    # Only include the US site
+    my $sites = Amazon::Sites->new(include => [ 'US' ]);
+    # Exclude the US site
+    my $sites = Amazon::Sites->new(exclude => [ 'US' ]);
+
+At most one of `include` or `exclude` can be specified.
+
 =head2 sites_hash
 
 Returns a hash where the keys are the two-letter country codes and the values are
@@ -73,17 +92,25 @@ Returns a list of L<Amazon::Site> objects, sorted by the sort order.
     return \@sites;
   }
 
-  sub _init_sites {
+  sub _init_sites($include, $exclude) {
     my %sites;
     my @cols = qw[code country tldn currency sort];
 
+    my $where = tell DATA;
+
     while (<DATA>) {
       chomp;
       my %site;
       @site{@cols} = split;
+
+      next if @$include and not grep { $site{code} eq $_ } @$include;
+      next if @$exclude and grep { $site{code} eq $_ } @$exclude;
+
       $sites{$site{code}} = Amazon::Site->new(%site);
     }
 
+    seek DATA, $where, 0;
+
     return %sites;
   }
 

t/02-include-exclude.t

@@ -0,0 +1,31 @@
+use strict;
+use warnings;
+
+use Test::More;
+use Test::Exception;
+
+use Amazon::Sites;
+
+throws_ok { Amazon::Sites->new( include => [ 'UK' ], exclude => [ 'US' ]) }
+  qr[You can't specify both include and exclude],
+  'Can\'t specify both include and exclude';
+
+my $sites = Amazon::Sites->new(exclude => [ 'US' ]);
+my $az_us = $sites->site('US');
+ok(! $az_us, 'US is excluded');
+my $az_uk = $sites->site('UK');
+ok($az_uk, 'UK is included');
+is_deeply([ $sites->codes ],
+  [ qw(AE AU BE BR CA CN DE EG ES FR IN IT JP MX NL PL SA SE SG TR UK) ],
+  'Correct codes are included');
+
+$sites = Amazon::Sites->new(include => [ 'UK' ]);
+$az_us = $sites->site('US');
+ok(! $az_us, 'US is excluded');
+$az_uk = $sites->site('UK');
+ok($az_uk, 'UK is included');
+is_deeply([ $sites->codes ],
+  [ qw(UK) ],
+  'Correct code is included');
+
+done_testing;