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