lib/MojoFacets/Data.pm

   1 package MojoFacets::Data;
   2
   3 use strict;
   4 use warnings;
   5
   6 use base 'Mojolicious::Controller';
   7
   8 use Data::Dump qw(dump);
   9 use File::Slurp;
  10 use JSON;
  11 use Encode;
  12 use locale;
  13 use File::Find;
  14
  15 sub index {
  16         my $self = shift;
  17
  18         my $path = $self->app->home->rel_dir('data');
  19         die "no data dir $path" unless -d $path;
  20
  21         my @files;
  22         find( sub {
  23                 my $file = $File::Find::name;
  24                 if ( -f $file && $file =~ m/\.(js(on)?|txt)$/ ) {
  25                         $file =~ s/$path\/*//;
  26                         push @files, $file;
  27                 } else {
  28                         warn "IGNORE: $file\n";
  29                 }
  30         }, $path);
  31
  32         $self->render( files => [ @files ] );
  33 }
  34
  35 our $data;
  36 our $stats;
  37
  38 sub load {
  39         my $self = shift;
  40
  41         my $path = $self->app->home->rel_file( 'data/' . $self->param('path') );
  42         die "$path $!" unless -r $path;
  43
  44         $self->session('path' => $self->param('path'));
  45
  46         # we could use Mojo::JSON here, but it's too slow
  47 #       $data = from_json read_file $path;
  48         $data = read_file $path;
  49         warn "# data snippet: ", substr($data,0,200);
  50         if ( $path =~ m/\.js/ ) {
  51                 Encode::_utf8_on($data);
  52                 $data = from_json $data;
  53                 $self->session( 'header' => 0 ); # generate later
  54         } elsif ( $path =~ m/\.txt/ ) {
  55                 my @lines = split(/\r?\n/, $data);
  56                 $data = { items => [] };
  57
  58                 my $headers = shift @lines;
  59                 my $multiline = $headers =~ s/\^//g;
  60                 my @header = split(/\|/, $headers );
  61                 warn "# header ", dump( @header );
  62                 $self->session( 'header' => [ @header ] );
  63                 $self->session( 'columns' => [ @header ] );
  64                 while ( my $line = shift @lines ) {
  65                         chomp $line;
  66                         my @v = split(/\|/, $line);
  67                         while ( @lines && $#v < $#header ) {
  68                                 $line = $lines[0];
  69                                 $line =~ s/\^//g;
  70                                 chomp $line;
  71                                 my @more_v = split(/\|/, $line);
  72                                 if ( $#v + $#more_v > $#header ) {
  73                                         warn "short line: ",dump( @v );
  74                                         last;
  75                                 }
  76                                 shift @lines;
  77                                 $v[ $#v ] .= shift @more_v;
  78                                 push @v, @more_v if @more_v;
  79
  80                                 if ( $#v > $#header ) {
  81                                         die "# splice $#header ", dump( @v );
  82                                         @v = splice @v, 0, $#header;
  83                                 }
  84                         }
  85                         my $item;
  86                         $item->{ $header[$_] || "f_$_" } = [ $v[$_] ] foreach ( 0 .. $#v );
  87                         push @{ $data->{items} }, $item;
  88                 }
  89         } else {
  90                 warn "file format unknown $path";
  91         }
  92
  93         $stats = {};
  94
  95         foreach my $e ( @{ $data->{items} } ) {
  96                 foreach my $n ( keys %$e ) {
  97                         $stats->{$n}->{count}++;
  98                         if ( ref $e->{$n} eq 'ARRAY' ) {
  99
 100                                 $stats->{$n}->{array} += $#{ $e->{$n} } + 1;
 101
 102                                 foreach my $x ( @{$e->{$n}} ) {
 103                                         $stats->{$n}->{numeric}++
 104                                                 if $x =~ m/^[-+]?([0-9]*\.[0-9]+|[0-9]+)$/;
 105                                 }
 106
 107                         } else {
 108                                 $stats->{$n}->{numeric}++
 109                                         if $e->{$n} =~ m/^[-+]?([0-9]*\.[0-9]+|[0-9]+)$/;
 110                         }
 111                 }
 112         }
 113
 114         foreach my $n ( keys %$stats ) {
 115                 next unless defined $stats->{$n}->{array};
 116                 delete $stats->{$n}->{array}
 117                         if $stats->{$n}->{array} == $stats->{$n}->{count};
 118         }
 119
 120         $self->session( 'header' => [
 121                 sort { $stats->{$b}->{count} <=> $stats->{$a}->{count} }
 122                 grep { defined $stats->{$_}->{count} } keys %$stats
 123         ] ) unless $self->session( 'header' );
 124
 125         warn dump($stats);
 126
 127         $self->session( 'filters' => {} );
 128
 129         $self->redirect_to( '/data/columns' );
 130 }
 131
 132
 133 sub columns {
 134     my $self = shift;
 135
 136         $self->redirect_to( '/data/index' ) unless $self->session('header');
 137
 138         my @columns;
 139         @columns = grep { defined $stats->{$_}->{count} } @{ $self->session('columns') } if $self->session('columns');
 140
 141         foreach my $c ( @{ $self->session( 'header' ) } ) {
 142                 push @columns, $c unless grep { /^\Q$c\E$/ } @columns;
 143         }
 144
 145     $self->render(
 146                 message => 'Select columns to display',
 147                 stats => $stats,
 148                 columns => \@columns,
 149                 checked => $self->_checked( $self->_perm_array('columns') ),
 150         );
 151 }
 152
 153 sub _perm_array {
 154     my ($self,$name) = @_;
 155
 156         my @array = $self->param($name);
 157
 158         if ( @array ) {
 159                 $self->session($name => [ @array ]);
 160         } elsif ( my $session = $self->session($name) ) {
 161                 if ( ref $session eq 'ARRAY' ) {
 162                         @array = @$session;
 163                 } else {
 164                         die "$name not array ",dump($session);
 165                 }
 166         }
 167         warn "# $name ",dump @array;
 168         return @array;
 169 }
 170
 171 sub _perm_scalar {
 172     my ($self,$name,$default) = @_;
 173
 174         my $scalar = $self->param($name);
 175
 176         if ( defined $scalar ) {
 177                 $self->session($name => $scalar);
 178         } else {
 179                 $scalar = $self->session($name);
 180         }
 181
 182         if ( ! defined $scalar ) {
 183                 $scalar = $default;
 184                 die "no default for $name" unless defined $scalar;
 185                 $self->session($name => $scalar);
 186         }
 187
 188         warn "# $name ",dump $scalar;
 189         return $scalar;
 190 }
 191
 192 sub filter {
 193         my $self = shift;
 194
 195         my $name = $self->param('filter_name') || die "name?";
 196         my @vals = $self->param('filter_vals');
 197
 198         warn "# filter $name vals ",dump(@vals);
 199
 200         my $filters = $self->session('filters');
 201         if ( @vals ) {
 202                 $filters->{$name} = [ @vals ];
 203         } else {
 204                 delete $filters->{$name};
 205         }
 206         $self->session( 'filters' => $filters );
 207
 208         warn "# filters ",dump($self->session('filters'));
 209
 210         $self->session( 'offset' => 0 );
 211
 212         $self->redirect_to('/data/items');
 213 }
 214
 215 sub _filter_item {
 216         my ( $self, $filters, $i ) = @_;
 217         my $pass = 1;
 218         foreach my $n ( keys %$filters ) {
 219                 my @filter_values = @{ $filters->{$n} };
 220                 my $include_missing = grep { /^_missing/ } @filter_values;
 221                 if ( ! exists $i->{$n} ) {
 222                         if ( $include_missing ) {
 223                                 $pass = 1;
 224                                 next;
 225                         } else {
 226                                 $pass = 0;
 227                                 last;
 228                         }
 229                 }
 230                 # and match any of values in element
 231                 my $have_values = 0;
 232                 foreach my $v ( @{ $i->{$n} } ) { # FIXME not array?
 233                         $have_values ||= 1 if grep { m/^\Q$v\E$/ } @filter_values;
 234                 }
 235                 if ( ! $have_values ) {
 236                         $pass = 0;
 237                         last;
 238                 }
 239         }
 240         return $pass;
 241 }
 242
 243 sub _data_items {
 244         my $self = shift;
 245         my $filters = $self->session('filters');
 246         grep {
 247                 $filters ? $self->_filter_item( $filters, $_ ) : 1;
 248         } @{ $data->{items} };
 249 }
 250
 251 sub items {
 252     my $self = shift;
 253
 254         $self->redirect_to('/data/index') unless $data->{items};
 255
 256         my @columns = $self->_perm_array('columns');
 257         $self->redirect_to('/data/columns') unless @columns;
 258         my $order   = $self->_perm_scalar('order', $columns[0]);
 259         my $sort    = $self->_perm_scalar('sort', 'a');
 260         my $offset  = $self->_perm_scalar('offset', 0);
 261         my $limit   = $self->_perm_scalar('limit', 20);
 262         $self->_perm_scalar('show', 'table');
 263
 264         # fix offset when changing limit
 265         $offset = int( $offset / $limit ) * $limit;
 266
 267         # FIXME - multi-level sort
 268         my $numeric = $self->_is_numeric($order);
 269         my $missing = $numeric ? 0 : '';
 270         no warnings qw(numeric);
 271         my @sorted = sort {
 272                 my $v1 = exists $a->{$order} ? join('', @{$a->{$order}}) : $missing;
 273                 my $v2 = exists $b->{$order} ? join('', @{$b->{$order}}) : $missing;
 274                 ($v1,$v2) = ($v2,$v1) if $sort eq 'd';
 275                 $numeric ? $v1 <=> $v2 : $v1 cmp $v2 ;
 276         } $self->_data_items;
 277
 278 #       warn "# sorted ", dump @sorted;
 279
 280         my $rows = $#sorted + 1;
 281
 282         $self->render(
 283                 order => $order,
 284                 offset => $offset,
 285                 limit => $limit,
 286                 sorted => [ splice @sorted, $offset, $limit ],
 287                 columns => [ @columns ],
 288                 rows => $rows,
 289                 numeric => { map { $_, $self->_is_numeric($_) } @columns },
 290         );
 291
 292 }
 293
 294
 295 sub order {
 296         my $self = shift;
 297         $self->session('order', $self->param('order'));
 298         $self->session('sort', $self->param('sort'));
 299         $self->redirect_to('/data/items');
 300 }
 301
 302 sub _is_numeric {
 303         my ( $self, $name ) = @_;
 304
 305         # sort facet numerically if more >50% elements are numeric
 306         defined $stats->{$name}->{numeric} &&
 307                 $stats->{$name}->{numeric} > $stats->{$name}->{count} / 2;
 308 }
 309
 310 sub facet {
 311         my $self = shift;
 312
 313         if ( my $remove = $self->param('remove') ) {
 314                 my $f = $self->session('filters');
 315                 delete $f->{$remove};
 316                 $self->session( 'filters' => $f );
 317                 $self->redirect_to( '/data/items' );
 318         }
 319
 320         my $facet;
 321         my $name = $self->param('name') || die "no name";
 322
 323         foreach my $i ( $self->_data_items ) {
 324                 if ( ! exists $i->{$name} ) {
 325                         $facet->{ _missing }++;
 326                 } elsif ( ref $i->{$name} eq 'ARRAY' ) {
 327                         $facet->{$_}++ foreach @{ $i->{$name} };
 328                 } else {
 329                         $facet->{ $i->{$name} }++;
 330                 }
 331         }
 332
 333 #       warn "# facet $name ",dump $facet;
 334
 335         my $checked;
 336         if ( my $f = $self->session('filters') ) {
 337                 if ( defined $f->{$name} ) {
 338                         $checked = $self->_checked( @{ $f->{$name} } );
 339                 }
 340         }
 341
 342         my $sort = $self->param('sort') || 'c';
 343
 344         # sort facet numerically if more >50% elements are numeric
 345         my $numeric = $self->_is_numeric($name);
 346
 347         my @facet_names = sort {
 348                 if ( $sort =~ m/a/i ) {
 349                         $numeric ? $a <=> $b : lc $a cmp lc $b;
 350                 } elsif ( $sort =~ m/d/i ) {
 351                         $numeric ? $b <=> $a : lc $b cmp lc $a;
 352                 } elsif ( $sort =~ m/c/i ) {
 353                         $facet->{$b} <=> $facet->{$a};
 354                 } else {
 355                         warn "unknown sort: $sort";
 356                         $a cmp $b;
 357                 }
 358         } keys %$facet;
 359
 360         $self->render( name => $name, facet => $facet, checked => $checked,
 361                 facet_names => \@facet_names, sort => $sort, numeric => $numeric,
 362         );
 363 }
 364
 365 sub _checked {
 366         my $self = shift;
 367         my $checked;
 368         $checked->{$_}++ foreach @_;
 369         warn "# _checked ",dump($checked);
 370         return $checked;
 371 }
 372
 373 1;