fotostore/fotostore.pl

279 lines
7.3 KiB
Perl
Raw Normal View History

#!/usr/bin/perl
use strict;
use warnings;
use Mojolicious::Lite; # app, get, post is exported.
use File::Basename 'basename';
use File::Path 'mkpath';
use File::Spec 'catfile';
use Cwd;
use Getopt::Long;
use Imager;
my $predefined_user = 'alpha6';
my $predefined_password = "";
GetOptions('password=s', \$predefined_password);
die "No user password defined!" unless($predefined_password);
# Image base URL
my $IMAGE_BASE = 'images';
my $ORIG_DIR = 'orig';
my $thumbs_size = 200;
my @scale_width = ($thumbs_size, 640, 800, 1024);
# 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 {
'username' => $predefined_user,
'password' => $predefined_password,
'name' => 'User Name'
} if ($uid eq 'userid' || $uid eq 'useridwithextradata');
return undef;
},
validate_user => sub {
my $self = shift;
my $username = shift || '';
my $password = shift || '';
my $extradata = shift || {};
# return 'useridwithextradata' if($username eq 'alpha6' && $password eq 'qwerty' && ( $extradata->{'ohnoes'} || '' ) eq 'itsameme');
return 'userid' if($username eq $predefined_user && $password eq $predefined_password);
return undef;
},
};
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');
# 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 = create_filename($ext);
my $image_file = File::Spec->catfile($ORIG_PATH, $filename);
# If file is exists, Retry creating filename
while(-f $image_file){
$filename = create_filename();
$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;
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;
}
# Redirect to top page
$self->redirect_to('index');
} => 'upload';
sub create_filename {
my $ext = shift || 'jpg';
# Date and time
my ($sec, $min, $hour, $mday, $month, $year) = localtime;
$month = $month + 1;
$year = $year + 1900;
# Random number(0 ~ 999999)
my $rand_num = int(rand 1000000);
# Create file name form datatime and random number
# (like image-20091014051023-78973)
my $name = sprintf('image-%04s%02s%02s%02s%02s%02s-%06s.%s',
$year, $month, $mday, $hour, $min, $sec, $rand_num, $ext);
return $name;
}
app->start;
__DATA__
@@ error.html.ep
<html>
<head>
<meta http-equiv="Content-Type" content="text/html;charset=UTF-8" >
<title>Error</title>
</head>
<body>
<%= $message %>
</body>
</html>
@@ no_logged.html.ep
<html>
<head>
<meta http-equiv="Content-Type" content="text/html;charset=UTF-8" >
<title>Rough, Slow, Stupid, Contrary Photohosting</title>
</head>
<body>
<h1>Rough, Slow, Stupid, Contrary Photohosting</h1>
<form method="post" action="<%= url_for('login') %>" >
<div>
<input type="text" name="username" >
<input type="password" name="password">
<input type="submit" value="Login">
</div>
</form>
</body>
</html>
@@ index.html.ep
<html>
<head>
<meta http-equiv="Content-Type" content="text/html;charset=UTF-8" >
<title>Rough, Slow, Stupid, Contrary Photohosting</title>
</head>
<body>
<h1>Rough, Slow, Stupid, Contrary Photohosting</h1>
<% if (is_user_authenticated()) { %>
<div><a href="/logout">Logout</a></div>
<hr>
<form method="post" action="<%= url_for('upload') %>" enctype ="multipart/form-data">
<div>
File name
<input type="file" name="image" >
<input type="submit" value="Upload" >
</div>
</form>
<div>
<% foreach my $image (@$images) { %>
<div>
<hr>
<div>
<a href='<%= "/$image_base/$orig/$image" %>'>Image original</a>
<% for my $scale (@$scales) { %>
<a href='<%= "/$image_base/$scale/$image" %>'><%= $scale %></a>
<% } %>
</div>
<div>
<img src="<%= "/$image_base/$thumbs_size/$image" %>">
</div>
<div>
<% } %>
</div>
<% } else { %>
<form method="post" action="<%= url_for('login') %>" >
<div>
<input type="text" name="username" >
<input type="password" name="password">
<input type="submit" value="Login">
</div>
</form>
<% } %>
</body>
</html>