SL/Helper/Csv/Dispatcher.pm

   1 package SL::Helper::Csv::Dispatcher;
   2
   3 use strict;
   4
   5 use Data::Dumper;
   6 use Carp;
   7 use Scalar::Util qw(weaken);
   8 use List::MoreUtils qw(all pairwise);
   9 use Rose::Object::MakeMethods::Generic scalar => [ qw(
  10   _specs _row_class _row_spec _errors
  11 ) ];
  12
  13 use SL::Helper::Csv::Error;
  14
  15 sub new {
  16   my ($class, $parent) = @_;
  17   my $self = bless { }, $class;
  18
  19   weaken($self->{_csv} = $parent);
  20   $self->_errors([]);
  21
  22   return $self;
  23 }
  24
  25 sub dispatch {
  26   my ($self, $line) = @_;
  27
  28   my $class = $self->_class_by_line($line);
  29   croak 'no class given' unless $class;
  30
  31   eval "require " . $class;
  32   my $obj = $class->new;
  33
  34   my $specs = $self->_specs_by_line($line);
  35   for my $spec (@{ $specs }) {
  36     $self->apply($obj, $spec, $line->{$spec->{key}});
  37   }
  38
  39   return $obj;
  40 }
  41
  42 sub _class_by_line {
  43   my ($self, $line) = @_;
  44
  45   # initialize lookup hash if not already done
  46   if ($self->_csv->is_multiplexed && ! defined $self->_row_class ) {
  47     $self->_row_class({ map { $_->{row_ident} => $_->{class} } @{ $self->_csv->profile } });
  48   }
  49
  50   if ($self->_csv->is_multiplexed) {
  51     return $self->_row_class->{$line->{datatype}};
  52   } else {
  53     return $self->_csv->profile->[0]->{class};
  54   }
  55 }
  56
  57 sub _specs_by_line {
  58   my ($self, $line) = @_;
  59
  60   # initialize lookup hash if not already done
  61   if ($self->_csv->is_multiplexed && ! defined $self->_row_spec ) {
  62     $self->_row_spec({ pairwise { no warnings 'once'; $a->{row_ident} => $b } @{ $self->_csv->profile }, @{ $self->_specs } });
  63   }
  64
  65   if ($self->_csv->is_multiplexed) {
  66     return $self->_row_spec->{$line->{datatype}};
  67   } else {
  68     return $self->_specs->[0];
  69   }
  70 }
  71
  72
  73 sub apply {
  74   my ($self, $obj, $spec, $value) = @_;
  75   return unless $value;
  76
  77   for my $step (@{ $spec->{steps} }) {
  78     my ($acc, $class, $index) = @$step;
  79     if ($class) {
  80
  81       # autovifify
  82       if (defined $index) {
  83         if (! $obj->$acc || !$obj->$acc->[$index]) {
  84           my @objects = $obj->$acc;
  85           $obj->$acc(@objects, map { $class->new } 0 .. $index - @objects);
  86         }
  87         $obj = $obj->$acc->[$index];
  88       } else {
  89         if (! $obj->$acc) {
  90           $obj->$acc($class->new);
  91         }
  92         $obj = $obj->$acc;
  93       }
  94
  95     } else {
  96       $obj->$acc($value);
  97     }
  98   }
  99 }
 100
 101 sub is_known {
 102   my ($self, $col) = @_;
 103   return grep { $col eq $_->{key} } $self->_specs;
 104 }
 105
 106 sub parse_profile {
 107   my ($self, %params) = @_;
 108
 109   my @specs;
 110
 111   my $csv_profile = $self->_csv->profile;
 112   my $h_aref = ($self->_csv->is_multiplexed)? $self->_csv->header : [ $self->_csv->header ];
 113   my $i = 0;
 114   foreach my $header (@{ $h_aref }) {
 115     my $spec = $self->_parse_profile(profile => $csv_profile->[$i]->{profile},
 116                                      class   => $csv_profile->[$i]->{class},
 117                                      header  => $header);
 118     push @specs, $spec;
 119     $i++;
 120   }
 121
 122   $self->_specs(\@specs);
 123
 124   $self->_csv->_push_error($self->errors);
 125
 126   return ! $self->errors;
 127 }
 128
 129 sub _parse_profile {
 130   my ($self, %params) = @_;
 131
 132   my $profile = $params{profile};
 133   my $class   = $params{class};
 134   my $header  = $params{header};
 135
 136   my @specs;
 137
 138   for my $col (@$header) {
 139     next unless $col;
 140     if ($self->_csv->strict_profile) {
 141       if (exists $profile->{$col}) {
 142         push @specs, $self->make_spec($col, $profile->{$col}, $class);
 143       } else {
 144         $self->unknown_column($col, undef);
 145       }
 146     } else {
 147       if (exists $profile->{$col}) {
 148         push @specs, $self->make_spec($col, $profile->{$col}, $class);
 149       } else {
 150         push @specs, $self->make_spec($col, $col, $class);
 151       }
 152     }
 153   }
 154
 155   return \@specs;
 156 }
 157
 158 sub make_spec {
 159   my ($self, $col, $path, $cur_class) = @_;
 160
 161   my $spec = { key => $col, steps => [] };
 162
 163   return unless $path;
 164
 165   return unless $cur_class;
 166
 167   for my $step_index ( split /\.(?!\d)/, $path ) {
 168     my ($step, $index) = split /\./, $step_index;
 169     if ($cur_class->can($step)) {
 170       if (my $rel = $cur_class->meta->relationship($step)) { #a
 171         if ($index && ! $rel->isa('Rose::DB::Object::Metadata::Relationship::OneToMany')) {
 172           $self->_push_error([
 173             $path,
 174             undef,
 175             "Profile path error. Indexed relationship is not OneToMany around here: '$step_index'",
 176             undef,
 177             0,
 178           ]);
 179           return;
 180         } else {
 181           my $next_class = $cur_class->meta->relationship($step)->class;
 182           push @{ $spec->{steps} }, [ $step, $next_class, $index ];
 183           $cur_class = $next_class;
 184           eval "require $cur_class; 1" or die "could not load class '$cur_class'";
 185         }
 186       } else { # simple dispatch
 187         push @{ $spec->{steps} }, [ $step ];
 188         last;
 189       }
 190     } else {
 191       $self->unknown_column($col, $path);
 192     }
 193   }
 194
 195   return $spec;
 196 }
 197
 198 sub unknown_column {
 199   my ($self, $col, $path) = @_;
 200   return if $self->_csv->ignore_unknown_columns;
 201
 202   $self->_push_error([
 203     $col,
 204     undef,
 205     "header field '$col' is not recognized",
 206     undef,
 207     0,
 208   ]);
 209 }
 210
 211 sub _csv {
 212   $_[0]->{_csv};
 213 }
 214
 215 sub errors {
 216   @{ $_[0]->_errors }
 217 }
 218
 219 sub _push_error {
 220   my ($self, @errors) = @_;
 221   my @new_errors = ($self->errors, map { SL::Helper::Csv::Error->new(@$_) } @errors);
 222   $self->_errors(\@new_errors);
 223 }
 224
 225 1;