#!/opt/bin/perl

use Gtk2 -init;
use Gtk2::Gdk::Keysyms;

use Gtk2::CV::ImageWindow;
use Gtk2::CV::Schnauzer;

$VERSION = 0.1;

my $viewer;

package cluster;

use Glib::Object::Subclass Gtk2::Window;

use Gtk2::SimpleList;

sub move_pics {
   my ($self, $spec, $specdir) = @_;
   return unless defined $ARGV[1];

   # dirname: 2nd program argument + category + name from analyze (in the list)
   my $dirname = $ARGV[1] . '/'.$self->{dirlbl}->get_text;

   # only add the special when ctrl is pressed
   if ((not defined $specdir) and $spec) {
      my @s = $self->{list}->get_selected_indices;
      unless (defined $s[0]) {
         print "nothing selected!\n";
         return;
      }
      my $d = $self->{list}->{data}->[$s[0]]->[1];
      unless (defined $d) {
         print "no such name on idx $s[0]\n";
      }
      mkdir $dirname;
      $dirname .= '/'.$d;
   } elsif (defined $specdir) {
      $dirname .= '/'.$specdir;
   }
   print "dirname: '$dirname'\n";
   mkdir $dirname;
   $self->{schnauzer}->movesel ($dirname);
}

sub INIT_INSTANCE {
   my ($self) = @_;
   
   $self->set_default_size (900, 700);

   my $box = new Gtk2::HBox;
   $self->add ($box);

   $box->add (my $sw = new Gtk2::ScrolledWindow);
   $sw->add (
      $self->{list} = new Gtk2::SimpleList
         "#"    => "int",
         "Name" => "text",
   );

   $box->add ($self->{schnauzer} = new Gtk2::CV::Schnauzer);
   #$self->{schnauzer}->{draw}->signal_connect (key_press_event => sub {
   #   my $key = $_[1]->keyval;
   #   my $state = $_[1]->state;
   #
   #  my $ctrl = grep $_ eq "control-mask", @{$_[1]->state};
   #   if ($ctrl && $key == $Gtk2::Gdk::Keysyms{c}) {
   #      $self->move_pics (0, $self->{direntry}->get_text())
   #         if ($self->{direntry}->get_text() ne "")
   #   } else {
   #      return 0;
   #   }
   #   return 1;
   #});

   $self->{schnauzer}->signal_connect (activate => sub { $viewer->load_image ($_[1]) });

   $self->{list}->get_column(0)->set_sort_column_id(0);
   $self->{list}->get_column(1)->set_sort_column_id(1);

   $self->{list}->signal_connect (key_press_event => sub {
      my $key = $_[1]->keyval;
      my $state = $_[1]->state;

      my $ctrl = grep $_ eq "control-mask", @{$_[1]->state};

      if ($key == $Gtk2::Gdk::Keysyms{t}) {
         $self->cycle_move_dir_idx;
      } elsif ($key == $Gtk2::Gdk::Keysyms{m}) {
         $self->move_pics ($ctrl);
      } elsif ($key == $Gtk2::Gdk::Keysyms{c}) {
         $self->move_pics (0, $self->{direntry}->get_text())
            if ($self->{direntry}->get_text() ne "")
      } elsif ($key == $Gtk2::Gdk::Keysyms{Up}) {
         return 0;
      } elsif ($key == $Gtk2::Gdk::Keysyms{Down}) {
         return 0;
      } else {
         $self->{schnauzer}->handle_key ($key, $state);
      }
      return 1;

   });

   $self->{list}->signal_connect (cursor_changed => sub {
      my $row = scalar +($_[0]->get_selection->get_selected_rows)[0]->get_indices;

      my $k = $_[0]{data}[$row][1];
      $k = $self->{cluster}{$k};

      $self->{schnauzer}->set_paths (
         [map "$self->{path}/$_", @$k],
      );

      1;

   });
}

sub analyse {
   my ($self, $path) = @_;

   opendir my $dir, $path
      or die "$path: $!";

   $self->{path} = $path;

   my @files = map {
                     my $path = $_ = Glib::filename_to_unicode $_;
                     s/[\-_ ]+/ /g;
                     s/\.[^\.]+$//g;
                     s/(?<=[0-9 ])(?:fc|bc|front|back|cover|lq|hq)?[a-z]{0,3}$//;
                     s/(?<=[0-9])\[\d+\]$//;
                     s/ +$//;
                     [$_, $path]
                   }
                   grep !/\.(sfv|crc|par|par2)$/i, readdir $dir;

   my %cluster;
   $self->{cluster} = \%cluster;

   my @regexps = (
      qr<^(.+?)\ *[0-9 ]+$>,
      qr<^(.+?)\ *\([0-9 ]+\)$>,
      qr<^(.+?)\d+\ +[0-9 ]+$>,
      qr<^(.+?)\d+\ +\([0-9 ]+\)$>,
      qr<^[0-9 ]+\ *(.+)$>,
   );

   for my $re (@regexps) {
      for (@files) {
         #print "$_->[0]\n" if $_->[1] =~ /^kisa.tan/;
         #if ($_->[0] =~ $re && $_->[0] =~ $info->[1]) {
         if ($_->[0] =~ $re) {
            $cluster{$1}{$_->[1]} = 1;
            # print "PUSH $1 : $_->[1] $_->[0] $re $info->[1] \n";
         }
      }
   }
   #push @{$cluster{"THE REST"}->[1]}, $_ for keys %chk_files;

   my %file_sorted;

   my @rest;
   for my $k (keys %cluster) {
      my @v = keys %{ $cluster{$k }};
      $cluster{$k} = \@v;
      if (@v > 2) {
         push @{$self->{list}{data}}, [scalar @v, $k];
         $file_sorted{$_} = 1 for @v;
      }
   }

   my @remaining = grep !$file_sorted{$_},
                        map $_->[1],
                            @files;
   $cluster{"REMAINING FILES"} = \@remaining;
   push @{$self->{list}{data}}, [(scalar @remaining), "REMAINING FILES"];
}

package main;

sub new_schnauzer {
   my $w = new Gtk2::Window;
   $w->add (my $s = new Gtk2::CV::Schnauzer);

   $s->signal_connect (activate => sub { $viewer->load_image ($_[1]) });
   $s->signal_connect (key_press_event => \&std_keys);

   $s->set_dir (".");
   $w->show_all;

   $w;
}

sub std_keys {
   my $key = $_[1]->keyval;

   my $ctrl = grep $_ eq "control-mask", @{$_[1]->state};

   if ($key == $Gtk2::Gdk::Keysyms{q}) {
      main_quit Gtk2;
   } elsif ($ctrl && $key == $Gtk2::Gdk::Keysyms{v}) {
      new_schnauzer;
   } else {
      return 0;
   }

   1;
}

$viewer = new Gtk2::CV::ImageWindow;
$viewer->signal_connect (key_press_event => \&std_keys);
$viewer->signal_connect (delete_event => sub { main_quit Gtk2 });
$viewer->show_all;

my $cluster = new cluster;

$cluster->analyse ($ARGV[0]);
$cluster->show_all;

#$viewer->set (path => "/fs/samsung/store-anime-done/yuumi_kazuaki_-_love_to_hajieki_to_sayonara_to_-_150.jpg");
#$viewer->set (path => "/root/pix/kvvz.jpg");

main Gtk2;