#!/usr/bin/env perl
use strict;
use warnings;

use lib 'lib';
use Mojolicious::Lite;    # app, get, post is exported.

use File::Basename 'basename';
use File::Path 'mkpath';
use File::Spec 'catfile';
use Cwd;

use Imager;
use DBI;
use Digest::SHA;

use FotoStore::DB;

use Data::Dumper;
$Data::Dumper::Maxdepth = 3;

my $config = plugin 'Config' => { file => 'application.conf' };

my $db = FotoStore::DB->new( $config->{'db_file'} );

# Image base URL
my $IMAGE_BASE = 'images';
my $ORIG_DIR   = 'orig';

my $thumbs_size = 200;

my @scale_width = ( $thumbs_size, 640, 800, 1024 );

my $sha = Digest::SHA->new('sha256');

# Directory to save image files
# (app is Mojolicious object. static is MojoX::Dispatcher::Static object)
my $IMAGE_DIR = File::Spec->catfile( getcwd(), 'public', $IMAGE_BASE );

# Create directory if not exists
unless ( -d $IMAGE_DIR ) {
    mkpath $IMAGE_DIR or die "Cannot create directory: $IMAGE_DIR";
}

my $ORIG_PATH = File::Spec->catfile( $IMAGE_DIR, $ORIG_DIR );
unless ( -d $ORIG_PATH ) {
    mkpath $ORIG_PATH or die "Cannot create directory: $ORIG_PATH";
}

for my $dir (@scale_width) {
    my $scaled_dir_path = File::Spec->catfile( $IMAGE_DIR, $dir );
    unless ( -d $scaled_dir_path ) {
        mkpath $scaled_dir_path
          or die "Cannot create directory: $scaled_dir_path";
    }
}

plugin 'authentication', {
    autoload_user => 1,
    load_user     => sub {
        my $self = shift;
        my $uid  = shift;

        return $db->get_user($uid);
    },
    validate_user => sub {
        my $self      = shift;
        my $username  = shift || '';
        my $password  = shift || '';
        my $extradata = shift || {};

        my $digest = $sha->add($password);

        my $user_id = $db->check_user( $username, $digest->hexdigest() );
        $self->app->log->debug("user id: [$user_id]");

        return $user_id;
    },
};

post '/login' => sub {
    my $self = shift;
    my $u    = $self->req->param('username');
    my $p    = $self->req->param('password');

    if ( $self->authenticate( $u, $p ) ) {
        $self->redirect_to('/');
    }
    else {
        $self->render( text => 'Login failed :(' );
    }

};

get '/logout' => sub {
    my $self = shift;

    $self->logout();
    $self->render( text => 'bye' );
};

# Display top page
get '/' => sub {
    my $self = shift;

    my $thumbs_dir = File::Spec->catfile( $IMAGE_DIR, $thumbs_size );

    # Get file names(Only base name)
    my @images =
      map { basename($_) }
      glob("$thumbs_dir/*.jpg $thumbs_dir/*.gif $thumbs_dir/*.png");

    # Sort by new order
    @images = sort { $b cmp $a } @images;

    # Render
    return $self->render(
        images      => \@images,
        image_base  => $IMAGE_BASE,
        orig        => $ORIG_DIR,
        thumbs_size => $thumbs_size,
        scales      => \@scale_width
    );

} => 'index';

# Upload image file
post '/upload' => ( authenticated => 1 ) => sub {
    my $self = shift;

    # Uploaded image(Mojo::Upload object)
    my $image = $self->req->upload('image');

    my $user = $self->current_user();
    $self->app->log->debug( "user:" . Dumper($user) );

    # Not upload
    unless ($image) {
        return $self->render(
            template => 'error',
            message  => "Upload fail. File is not specified."
        );
    }

    # Upload max size
    #my $upload_max_size = 3 * 1024 * 1024;

    # Over max size
    #if ($image->size > $upload_max_size) {
    #    return $self->render(
    #        template => 'error',
    #        message  => "Upload fail. Image size is too large."
    #    );
    #}

    # Check file type
    my $image_type = $image->headers->content_type;
    my %valid_types = map { $_ => 1 } qw(image/gif image/jpeg image/png);

    # Content type is wrong
    unless ( $valid_types{$image_type} ) {
        return $self->render(
            template => 'error',
            message  => "Upload fail. Content type is wrong."
        );
    }

    # Extention
    my $exts = {
        'image/gif'  => 'gif',
        'image/jpeg' => 'jpg',
        'image/png'  => 'png'
    };
    my $ext = $exts->{$image_type};

    # Image file
    my $filename = sprintf( '%s.%s', create_hash( $image->slurp() ), $ext );
    my $image_file = File::Spec->catfile( $ORIG_PATH, $filename );

    # Save to file
    $image->move_to($image_file);

    my $imager = Imager->new();
    $imager->read( file => $image_file ) or die $imager->errstr;

    #http://sylvana.net/jpegcrop/exif_orientation.html
    #http://myjaphoo.de/docs/exifidentifiers.html
    my $rotation_angle = $imager->tags( name => "exif_orientation" ) || 1;
    $self->app->log->info(
        "Rotation angle [" . $rotation_angle . "] [" . $image->filename . "]" );

    if ( $rotation_angle == 3 ) {
        $imager = $imager->rotate( degrees => 180 );
    }
    elsif ( $rotation_angle == 6 ) {
        $imager = $imager->rotate( degrees => 90 );
    }

    for my $scale (@scale_width) {
        my $scaled = $imager->scale( xpixels => $scale );

        $scaled->write(
            file => File::Spec->catfile( $IMAGE_DIR, $scale, $filename ) )
          or die $scaled->errstr;
    }

    if ( !$db->add_file( $user->{'user_id'}, $filename ) ) {

        #TODO: Send error msg
    }

    $self->render(
        json => {
            files => [
                {
                    name         => $image->filename,
                    size         => $image->size,
                    url          => sprintf( '/images/orig/%s', $filename ),
                    thumbnailUrl => sprintf( '/images/200/%s', $filename ),
                }
            ]
        }
    );

    # Redirect to top page
    # $self->redirect_to('index');

} => 'upload';

sub create_hash {
    my $data_to_hash = shift;

    $sha->add($data_to_hash);
    return $sha->hexdigest();
}

app->start;