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 $i = 0;
 113   foreach my $header (@{ $self->_csv->header }) {
 114     my $spec = $self->_parse_profile(profile => $csv_profile->[$i]->{profile},
 115                                      class   => $csv_profile->[$i]->{class},
 116                                      header  => $header);
 117     push @specs, $spec;
 118     $i++;
 119   }
 120
 121   $self->_specs(\@specs);
 122
 123   $self->_csv->_push_error($self->errors);
 124
 125   return ! $self->errors;
 126 }
 127
 128 sub _parse_profile {
 129   my ($self, %params) = @_;
 130
 131   my $profile = $params{profile};
 132   my $class   = $params{class};
 133   my $header  = $params{header};
 134
 135   my @specs;
 136
 137   for my $col (@$header) {
 138     next unless $col;
 139     if ($self->_csv->strict_profile) {
 140       if (exists $profile->{$col}) {
 141         push @specs, $self->make_spec($col, $profile->{$col}, $class);
 142       } else {
 143         $self->unknown_column($col, undef);
 144       }
 145     } else {
 146       if (exists $profile->{$col}) {
 147         push @specs, $self->make_spec($col, $profile->{$col}, $class);
 148       } else {
 149         push @specs, $self->make_spec($col, $col, $class);
 150       }
 151     }
 152   }
 153
 154   return \@specs;
 155 }
 156
 157 sub make_spec {
 158   my ($self, $col, $path, $cur_class) = @_;
 159
 160   my $spec = { key => $col, steps => [] };
 161
 162   return unless $path;
 163
 164   return unless $cur_class;
 165
 166   for my $step_index ( split /\.(?!\d)/, $path ) {
 167     my ($step, $index) = split /\./, $step_index;
 168     if ($cur_class->can($step)) {
 169       if (my $rel = $cur_class->meta->relationship($step)) { #a
 170         if ($index && ! $rel->isa('Rose::DB::Object::Metadata::Relationship::OneToMany')) {
 171           $self->_push_error([
 172             $path,
 173             undef,
 174             "Profile path error. Indexed relationship is not OneToMany around here: '$step_index'",
 175             undef,
 176             0,
 177           ]);
 178           return;
 179         } else {
 180           my $next_class = $cur_class->meta->relationship($step)->class;
 181           push @{ $spec->{steps} }, [ $step, $next_class, $index ];
 182           $cur_class = $next_class;
 183           eval "require $cur_class; 1" or die "could not load class '$cur_class'";
 184         }
 185       } else { # simple dispatch
 186         push @{ $spec->{steps} }, [ $step ];
 187         last;
 188       }
 189     } else {
 190       $self->unknown_column($col, $path);
 191     }
 192   }
 193
 194   return $spec;
 195 }
 196
 197 sub unknown_column {
 198   my ($self, $col, $path) = @_;
 199   return if $self->_csv->ignore_unknown_columns;
 200
 201   $self->_push_error([
 202     $col,
 203     undef,
 204     "header field '$col' is not recognized",
 205     undef,
 206     0,
 207   ]);
 208 }
 209
 210 sub _csv {
 211   $_[0]->{_csv};
 212 }
 213
 214 sub errors {
 215   @{ $_[0]->_errors }
 216 }
 217
 218 sub _push_error {
 219   my ($self, @errors) = @_;
 220   my @new_errors = ($self->errors, map { SL::Helper::Csv::Error->new(@$_) } @errors);
 221   $self->_errors(\@new_errors);
 222 }
 223
 224 1;