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                 my $v;
  22                 $line =~ s/""/_qq_/gc;
  23                 if ( $line =~ s/^"\s*([^"]+)\s*"\Q$delimiter\E?// ) {
  24                         $v = $1;
  25                 } elsif ( $line =~ s/^\s*([^\Q$delimiter\E]+)\s*\Q$delimiter\E?// ) {
  26                         $v = $1;
  27                 } elsif ( $line =~ s/^\s*\Q$delimiter\E// ) {
  28                         $v = $null;
  29                 } else {
  30                         die "can't parse [$line]\n";
  31                 }
  32
  33                 $v =~ s/^\s*(.+?)\s*$/$1/;
  34                 $v =~ s/_qq_/"/g;
  35                 push @v, $v;
  36         }
  37
  38         return @v;
  39 }
  40
  41 sub data {
  42         my $self = shift;
  43
  44         my $path = $self->full_path || $self->path;
  45
  46         my $data = read_file $path, { binmode => ':raw' }; # FIXME configurable!
  47         my $encoding = 'utf-8';
  48         if ( $path =~ m/\.(\w+).csv/i ) {
  49                 $encoding = $1;
  50         }
  51         warn "decoding ", length($data), " bytes using $encoding\n";
  52         $data = decode($encoding, $data);
  53
  54         my @lines = split(/\r?\n/, $data);
  55         $data = { items => [] };
  56
  57         my $delimiter = ',';
  58
  59         if ( $lines[0] !~ /;/ && $lines[1] =~ /;/ ) {
  60                 shift @lines; # FIXME ship non-header line
  61                 $delimiter = ';';
  62         }
  63
  64         warn "$path ", $#lines + 1, " lines encoding: $encoding delimiter:",dump($delimiter);
  65
  66         my $header_line = shift @lines;
  67
  68         my @header = _split_line( $delimiter, $header_line );
  69         warn "# header ",dump( @header );
  70
  71         while ( my $line = shift @lines ) {
  72                 chomp $line;
  73                 my @v = _split_line($delimiter, $line);
  74                 my $item;
  75                 foreach my $i ( 0 .. $#v ) {
  76                         $item->{ $header[$i] || "f_$i" } = [ $v[$i] ];
  77                 }
  78                 push @{ $data->{items} }, $item;
  79         }
  80
  81         $data->{header} = [ @header ];
  82
  83         return $data;
  84
  85 }
  86
  87 1