fotostore/fotostore.pl

314 lines
7.9 KiB
Perl
Raw Normal View History

2017-07-19 07:19:46 +03:00
#!/usr/bin/env perl
use strict;
use warnings;
2017-07-26 08:16:18 +03:00
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;
2017-07-22 11:08:15 +03:00
use DBI;
use Digest::SHA;
2017-07-26 08:16:18 +03:00
use FotoStore::DB;
2017-07-19 07:49:54 +03:00
2017-07-26 08:16:18 +03:00
use Data::Dumper;
$Data::Dumper::Maxdepth = 2;
2017-07-26 08:16:18 +03:00
my $config = plugin 'Config' => { file => 'application.conf' };
2017-07-22 11:08:15 +03:00
2017-07-26 08:16:18 +03:00
my $db = FotoStore::DB->new( $config->{'db_file'} );
# Image base URL
my $IMAGE_BASE = 'images';
2017-07-26 08:16:18 +03:00
my $ORIG_DIR = 'orig';
# set allowed thumbnails scale and image scales
my $thumbs_size = $config->{'thumbnails_size'};
my $scales_map = $config->{'image_scales'};
$scales_map->{$thumbs_size} = 1;
#Sort and filter values for array of available scales
2017-08-02 17:30:44 +03:00
my @scale_width =
map { $scales_map->{$_} == 1 ? $_ : undef }
sort { $a <=> $b } keys(%$scales_map);
2017-07-26 08:16:18 +03:00
my $sha = Digest::SHA->new('sha256');
# Directory to save image files
2017-07-26 08:16:18 +03:00
my $IMAGE_DIR = File::Spec->catfile( getcwd(), 'public', $IMAGE_BASE );
plugin 'authentication', {
autoload_user => 1,
2017-07-26 08:16:18 +03:00
load_user => sub {
my $self = shift;
my $uid = shift;
2017-07-26 08:16:18 +03:00
return $db->get_user($uid);
},
validate_user => sub {
2017-07-26 08:16:18 +03:00
my $self = shift;
my $username = shift || '';
my $password = shift || '';
my $extradata = shift || {};
2017-07-26 08:16:18 +03:00
my $digest = $sha->add($password);
my $user_id = $db->check_user( $username, $digest->hexdigest() );
return $user_id;
},
};
post '/login' => sub {
my $self = shift;
my $u = $self->req->param('username');
my $p = $self->req->param('password');
2017-07-26 08:16:18 +03:00
if ( $self->authenticate( $u, $p ) ) {
$self->redirect_to('/');
}
2017-07-26 08:16:18 +03:00
else {
$self->render( text => 'Login failed :(' );
}
};
get '/logout' => sub {
my $self = shift;
2017-07-26 08:16:18 +03:00
$self->logout();
2017-08-01 15:55:21 +03:00
$self->render( message => 'bye' );
};
2017-07-31 10:14:59 +03:00
get '/register' => ( authenticated => 0 ) => sub {
};
post '/register' => ( authenticated => 0 ) => sub {
2017-08-02 17:30:44 +03:00
my $self = shift;
my $username = $self->req->param('username');
my $password = $self->req->param('password');
2017-07-31 10:14:59 +03:00
my $fullname = $self->req->param('fullname');
2017-08-02 17:30:44 +03:00
my $invite = $self->req->param('invite');
if ( $invite eq $config->{'invite_code'} ) {
2017-07-31 10:14:59 +03:00
#chek that username is not taken
my $user = $db->get_user($username);
2017-08-02 17:30:44 +03:00
if ( $user->{'user_id'} > 0 ) {
$self->render(
template => 'error',
message => 'Username already taken!'
);
return 0;
2017-07-31 10:14:59 +03:00
}
2017-08-02 17:30:44 +03:00
if ( $fullname eq '' ) {
2017-07-31 10:14:59 +03:00
$fullname = $username;
}
my $digest = $sha->add($password);
2017-08-02 17:30:44 +03:00
$db->add_user( $username, $digest->hexdigest(), $fullname );
2017-07-31 10:14:59 +03:00
#Authenticate user after add
if ( $self->authenticate( $username, $password ) ) {
$self->redirect_to('/');
}
else {
$self->render( template => 'error', message => 'Login failed :(' );
2017-07-31 10:14:59 +03:00
}
}
2017-08-02 17:30:44 +03:00
else {
$self->render( template => 'error', message => 'invalid invite code' );
}
};
2017-07-31 10:14:59 +03:00
# Display top page
get '/' => sub {
my $self = shift;
2017-07-26 08:16:18 +03:00
2017-07-26 09:50:45 +03:00
my $current_user = $self->current_user;
2017-07-26 08:16:18 +03:00
} => 'index';
2017-08-01 15:06:08 +03:00
get '/get_images' => ( authenticated => 1 ) => sub {
my $self = shift;
my $current_user = $self->current_user;
2017-08-02 17:30:44 +03:00
my $user_id = $current_user->{'user_id'};
my $files_list = $db->get_files( $current_user->{'user_id'}, 20 );
my $thumbs_dir =
File::Spec->catfile( $IMAGE_DIR, $current_user->{'user_id'},
$thumbs_size );
2017-08-01 15:06:08 +03:00
my @images = map { $_->{'file_name'} } @$files_list;
my $images = [];
for my $img_item (@$files_list) {
2017-08-02 17:30:44 +03:00
my $file = $img_item->{'file_name'};
2017-08-01 15:06:08 +03:00
my $img_hash = {};
2017-08-02 07:55:54 +03:00
$img_hash->{'filename'} = $img_item->{'original_filename'};
2017-08-02 17:30:44 +03:00
$img_hash->{'original_url'} =
File::Spec->catfile( '/', $IMAGE_BASE, $current_user->{'user_id'},
$ORIG_DIR, $file );
$img_hash->{'thumbnail_url'} =
File::Spec->catfile( '/', $IMAGE_BASE, $current_user->{'user_id'},
$thumbs_size, $file );
2017-08-01 15:06:08 +03:00
2017-08-01 15:33:29 +03:00
my @scaled = ();
2017-08-01 15:06:08 +03:00
for my $scale (@scale_width) {
2017-08-02 17:30:44 +03:00
if ( -r File::Spec->catfile( get_path( $user_id, $scale ), $file ) )
{
push(
@scaled,
{
'size' => $scale,
'url' => File::Spec->catfile(
'/', $IMAGE_BASE, $user_id, $scale, $file
)
}
);
}
2017-08-02 17:30:44 +03:00
2017-08-01 15:06:08 +03:00
}
2017-08-01 15:33:29 +03:00
$img_hash->{'scales'} = \@scaled;
2017-08-02 17:30:44 +03:00
push( @$images, $img_hash );
}
2017-08-01 15:06:08 +03:00
# Render
return $self->render( json => $images );
};
# Upload image file
2017-07-31 20:21:03 +03:00
# There is no restriction for file size in app because restriction is present in nginx configuration
2017-07-26 08:16:18 +03:00
post '/upload' => ( authenticated => 1 ) => sub {
my $self = shift;
# Uploaded image(Mojo::Upload object)
my $image = $self->req->upload('image');
2017-07-26 08:16:18 +03:00
2017-08-02 17:30:44 +03:00
my $user = $self->current_user();
2017-07-26 09:50:45 +03:00
my $user_id = $user->{'user_id'};
2017-07-26 08:16:18 +03:00
$self->app->log->debug( "user:" . Dumper($user) );
# Not upload
unless ($image) {
return $self->render(
2017-07-26 08:16:18 +03:00
template => 'error',
message => "Upload fail. File is not specified."
);
}
2017-07-26 08:16:18 +03:00
# Check file type
my $image_type = $image->headers->content_type;
2017-07-26 08:16:18 +03:00
my %valid_types = map { $_ => 1 } qw(image/gif image/jpeg image/png);
# Content type is wrong
2017-07-26 08:16:18 +03:00
unless ( $valid_types{$image_type} ) {
return $self->render(
template => 'error',
message => "Upload fail. Content type is wrong."
);
}
2017-07-26 08:16:18 +03:00
# Extention
2017-07-26 08:16:18 +03:00
my $exts = {
'image/gif' => 'gif',
'image/jpeg' => 'jpg',
'image/png' => 'png'
};
my $ext = $exts->{$image_type};
2017-07-26 08:16:18 +03:00
# Image file
2017-07-26 08:16:18 +03:00
my $filename = sprintf( '%s.%s', create_hash( $image->slurp() ), $ext );
2017-08-02 17:30:44 +03:00
my $image_file =
File::Spec->catfile( get_path( $user_id, $ORIG_DIR ), $filename );
2017-07-26 08:16:18 +03:00
# Save to file
$image->move_to($image_file);
2017-07-26 08:16:18 +03:00
my $imager = Imager->new();
2017-07-26 08:16:18 +03:00
$imager->read( file => $image_file ) or die $imager->errstr;
2017-07-19 10:01:12 +03:00
#http://sylvana.net/jpegcrop/exif_orientation.html
2017-07-19 10:01:12 +03:00
#http://myjaphoo.de/docs/exifidentifiers.html
2017-07-26 08:16:18 +03:00
my $rotation_angle = $imager->tags( name => "exif_orientation" ) || 1;
$self->app->log->info(
"Rotation angle [" . $rotation_angle . "] [" . $image->filename . "]" );
2017-07-26 08:16:18 +03:00
if ( $rotation_angle == 3 ) {
$imager = $imager->rotate( degrees => 180 );
}
elsif ( $rotation_angle == 6 ) {
$imager = $imager->rotate( degrees => 90 );
}
my $original_width = $imager->getwidth();
2017-07-19 10:21:20 +03:00
for my $scale (@scale_width) {
2017-08-02 17:30:44 +03:00
#Skip sizes which more than original image
2017-08-02 17:30:44 +03:00
if ( $scale >= $original_width ) {
next;
}
2017-07-26 08:16:18 +03:00
my $scaled = $imager->scale( xpixels => $scale );
2017-08-02 17:30:44 +03:00
$scaled->write( file =>
File::Spec->catfile( get_path( $user_id, $scale ), $filename ) )
2017-07-26 08:16:18 +03:00
or die $scaled->errstr;
}
2017-08-02 17:30:44 +03:00
if ( !$db->add_file( $user->{'user_id'}, $filename, $image->filename ) ) {
2017-07-26 08:16:18 +03:00
#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');
2017-07-26 08:16:18 +03:00
} => 'upload';
2017-07-26 08:16:18 +03:00
sub create_hash {
my $data_to_hash = shift;
$sha->add($data_to_hash);
return $sha->hexdigest();
}
2017-07-26 09:50:45 +03:00
sub get_path {
2017-08-02 17:30:44 +03:00
my ( $user_id, $size ) = @_;
2017-07-26 09:50:45 +03:00
my $path = File::Spec->catfile( $IMAGE_DIR, $user_id, $size );
2017-08-02 17:30:44 +03:00
unless ( -d $path ) {
2017-07-26 09:50:45 +03:00
mkpath $path or die "Cannot create directory: $path";
}
return $path;
}
2017-07-26 08:16:18 +03:00
app->start;