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