medvedych/fotostore
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Merge pull request #6 from alpha6/check_scale_exists

Browse source 
Check scale exists
This commit is contained in:
Denis Fedoseev 2017-08-02 17:36:43 +03:00 committed by GitHub
commit 5012d8f123
2 changed files with 81 additions and 37 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
application.conf.example
View file

@ -1,4 +1,11 @@
{
db_file => 'sql/fotostore.db',
invite_code => 'very_secure_invite_code',
thumbnails_size => 200,
image_scales => {
640 => 1,
800 => 1,
1024 => 1,
2048 => 1,
},
}

101
fotostore.pl
View file

@ -17,7 +17,7 @@ use Digest::SHA;
use FotoStore::DB;
use Data::Dumper;
$Data::Dumper::Maxdepth = 3;
$Data::Dumper::Maxdepth = 2;
my $config = plugin 'Config' => { file => 'application.conf' };
@ -27,14 +27,20 @@ my $db = FotoStore::DB->new( $config->{'db_file'} );
my $IMAGE_BASE = 'images';
my $ORIG_DIR = 'orig';
my $thumbs_size = 200;
# set allowed thumbnails scale and image scales
my $thumbs_size = $config->{'thumbnails_size'};
my @scale_width = ( $thumbs_size, 640, 800, 1024 );
my $scales_map = $config->{'image_scales'};
$scales_map->{$thumbs_size} = 1;
#Sort and filter values for array of available scales
my @scale_width =
map { $scales_map->{$_} == 1 ? $_ : undef }
sort { $a <=> $b } keys(%$scales_map);
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 );
plugin 'authentication', {
@ -54,7 +60,6 @@ plugin 'authentication', {
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;
},
@ -86,37 +91,42 @@ get '/register' => ( authenticated => 0 ) => sub {
};
post '/register' => ( authenticated => 0 ) => sub {
my $self = shift;
my $username = $self->req->param('username');
my $password = $self->req->param('password');
my $self = shift;
my $username = $self->req->param('username');
my $password = $self->req->param('password');
my $fullname = $self->req->param('fullname');
my $invite = $self->req->param('invite');
my $invite = $self->req->param('invite');
if ( $invite eq $config->{'invite_code'} ) {
if ($invite eq $config->{'invite_code'}) {
#chek that username is not taken
my $user = $db->get_user($username);
if ($user->{'user_id'} > 0) {
$self->render(template => 'error', message => 'Username already taken!');
if ( $user->{'user_id'} > 0 ) {
$self->render(
template => 'error',
message => 'Username already taken!'
);
return 0;
}
if ($fullname eq '') {
if ( $fullname eq '' ) {
$fullname = $username;
}
my $digest = $sha->add($password);
$db->add_user($username, $digest->hexdigest(), $fullname);
$db->add_user( $username, $digest->hexdigest(), $fullname );
#Authenticate user after add
if ( $self->authenticate( $username, $password ) ) {
$self->redirect_to('/');
}
else {
$self->render( text => 'Login failed :(' );
$self->render( template => 'error', message => 'Login failed :(' );
}
} else {
$self->render(template => 'error', message => 'invalid invite code');
}
else {
$self->render( template => 'error', message => 'invalid invite code' );
}
};
@ -132,33 +142,51 @@ get '/get_images' => ( authenticated => 1 ) => sub {
my $self = shift;
my $current_user = $self->current_user;
my $user_id = $current_user->{'user_id'};
my $files_list = $db->get_files($current_user->{'user_id'}, 20);
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 );
my $thumbs_dir =
File::Spec->catfile( $IMAGE_DIR, $current_user->{'user_id'},
$thumbs_size );
my @images = map { $_->{'file_name'} } @$files_list;
my $images = [];
for my $img_item (@$files_list) {
my $file = $img_item->{'file_name'};
my $file = $img_item->{'file_name'};
my $img_hash = {};
$img_hash->{'filename'} = $img_item->{'original_filename'};
$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 );
$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 );
my @scaled = ();
for my $scale (@scale_width) {
push(@scaled, {'size' => $scale, 'url' => File::Spec->catfile( '/', $IMAGE_BASE, $current_user->{'user_id'}, $scale, $file )}) ;
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
)
}
);
}
}
$img_hash->{'scales'} = \@scaled;
push(@$images, $img_hash);
push( @$images, $img_hash );
}
# Render
return $self->render( json => $images );
};
@ -171,7 +199,7 @@ post '/upload' => ( authenticated => 1 ) => sub {
# Uploaded image(Mojo::Upload object)
my $image = $self->req->upload('image');
my $user = $self->current_user();
my $user = $self->current_user();
my $user_id = $user->{'user_id'};
$self->app->log->debug( "user:" . Dumper($user) );
@ -205,7 +233,8 @@ post '/upload' => ( authenticated => 1 ) => sub {
# Image file
my $filename = sprintf( '%s.%s', create_hash( $image->slurp() ), $ext );
my $image_file = File::Spec->catfile( get_path($user_id, $ORIG_DIR), $filename );
my $image_file =
File::Spec->catfile( get_path( $user_id, $ORIG_DIR ), $filename );
# Save to file
$image->move_to($image_file);
@ -226,15 +255,23 @@ post '/upload' => ( authenticated => 1 ) => sub {
$imager = $imager->rotate( degrees => 90 );
}
my $original_width = $imager->getwidth();
for my $scale (@scale_width) {
#Skip sizes which more than original image
if ( $scale >= $original_width ) {
next;
}
my $scaled = $imager->scale( xpixels => $scale );
$scaled->write(
file => File::Spec->catfile( get_path($user_id, $scale), $filename ) )
$scaled->write( file =>
File::Spec->catfile( get_path( $user_id, $scale ), $filename ) )
or die $scaled->errstr;
}
if ( !$db->add_file( $user->{'user_id'}, $filename, $image->filename) ) {
if ( !$db->add_file( $user->{'user_id'}, $filename, $image->filename ) ) {
#TODO: Send error msg
}
@ -265,9 +302,9 @@ sub create_hash {
}
sub get_path {
my ($user_id, $size) = @_;
my ( $user_id, $size ) = @_;
my $path = File::Spec->catfile( $IMAGE_DIR, $user_id, $size );
unless (-d $path) {
unless ( -d $path ) {
mkpath $path or die "Cannot create directory: $path";
}
return $path;