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