lib/MojoFacets/Import/CSV.pm

   1 package MojoFacets::Import::CSV;
   2
   3 use warnings;
   4 use strict;
   5
   6 use base 'Mojo::Base';
   7
   8 use File::Slurp;
   9 use Data::Dump qw(dump);
  10 use Encode;
  11
  12 __PACKAGE__->attr('path');
  13 __PACKAGE__->attr('full_path'); # FIXME remove full_path
  14
  15 my $null = ''; # FIXME undef?
  16
  17 sub _split_line {
  18         my ( $delimiter, $line ) = @_;
  19         my @v;
  20         while ( $line ) {
  21                 if ( $line =~ s/^"([^"]+)"\Q$delimiter\E?// ) {
  22                         push @v, $1;
  23                 } elsif ( $line =~ s/^([^\Q$delimiter\E]+)\Q$delimiter\E?// ) {
  24                         push @v, $1;
  25                 } elsif ( $line =~ s/^\Q$delimiter\E// ) {
  26                         push @v, $null;
  27                 } else {
  28                         die "can't parse [$line]\n";
  29                 }
  30         }
  31
  32         return @v;
  33 }
  34
  35 sub data {
  36         my $self = shift;
  37
  38         my $path = $self->full_path || $self->path;
  39
  40         my $data = read_file $path, { binmode => ':raw' }; # FIXME configurable!
  41         my $encoding = 'utf-8';
  42         if ( $path =~ m/\.(\w+).csv/i ) {
  43                 $encoding = $1;
  44         }
  45         warn "decoding ", length($data), " bytes using $encoding\n";
  46         $data = decode($encoding, $data);
  47
  48         my @lines = split(/\r?\n/, $data);
  49         $data = { items => [] };
  50
  51         my $delimiter = ',';
  52
  53         if ( $lines[0] !~ /;/ && $lines[1] =~ /;/ ) {
  54                 shift @lines; # FIXME ship non-header line
  55                 $delimiter = ';';
  56         }
  57
  58         warn "$path ", $#lines + 1, " lines encoding: $encoding delimiter:",dump($delimiter);
  59
  60         my $header_line = shift @lines;
  61
  62         my @header = _split_line( $delimiter, $header_line );
  63         warn "# header ",dump( @header );
  64
  65         while ( my $line = shift @lines ) {
  66                 chomp $line;
  67                 my @v = _split_line($delimiter, $line);
  68                 my $item;
  69                 foreach my $i ( 0 .. $#v ) {
  70                         $item->{ $header[$i] || "f_$i" } = [ $v[$i] ];
  71                 }
  72                 push @{ $data->{items} }, $item;
  73         }
  74
  75         $data->{header} = [ @header ];
  76
  77         return $data;
  78
  79 }
  80
  81 1