X-Git-Url: https://git.sesse.net/?a=blobdiff_plain;f=perl%2FSesse%2Fpr0n%2FCommon.pm;h=a9ee2119db8751d3c97770f5cb6f7aa68f9864e0;hb=9b45ad1ecd9f1eace384f825a7dc6b2b7920af65;hp=6317ca4c09f66010a222518cd0a61751426cbe65;hpb=b14e19cdd4aa848f623687e5b0ac446f39c02b17;p=pr0n

diff --git a/perl/Sesse/pr0n/Common.pm b/perl/Sesse/pr0n/Common.pm
index 6317ca4..a9ee211 100644
--- a/perl/Sesse/pr0n/Common.pm
+++ b/perl/Sesse/pr0n/Common.pm
@@ -211,7 +211,7 @@ sub get_base {
 	my $r = shift;
 	return $r->dir_config('ImageBase');
 }
-
+				
 sub get_disk_location {
 	my ($r, $id) = @_;
         my $dir = POSIX::floor($id / 256);
@@ -231,6 +231,23 @@ sub get_cache_location {
 	}
 }
 
+sub ensure_disk_location_exists {
+	my ($r, $id) = @_;
+        my $dir = POSIX::floor($id / 256);
+
+	my $img_dir = get_base($r) . "/images/$dir/";
+	if (! -d $img_dir) {
+		$r->log->info("Need to create new image directory $img_dir");
+		mkdir($img_dir) or die "Couldn't create new image directory $img_dir";
+	}
+
+	my $cache_dir = get_base($r) . "/cache/$dir/";
+	if (! -d $cache_dir) {
+		$r->log->info("Need to create new cache directory $cache_dir");
+		mkdir($cache_dir) or die "Couldn't create new image directory $cache_dir";
+	}
+}
+
 sub get_mipmap_location {
 	my ($r, $id, $width, $height) = @_;
         my $dir = POSIX::floor($id / 256);