From 5161b1c8fdc8ff69005f864a89127fc18db6d4ed Mon Sep 17 00:00:00 2001
From: Cash Costello <cash.costello@gmail.com>
Date: Sun, 24 Oct 2010 01:53:21 +0000
Subject: started cleaning up the upload action

---
 lib/upload.php | 38 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)
 create mode 100644 lib/upload.php

(limited to 'lib/upload.php')

diff --git a/lib/upload.php b/lib/upload.php
new file mode 100644
index 000000000..a7dd7611d
--- /dev/null
+++ b/lib/upload.php
@@ -0,0 +1,38 @@
+<?php
+/**
+ * Helper library for working with uploads
+ */
+
+function tp_upload_check_format($mime) {
+	$accepted_formats = array(
+		'image/jpeg',
+		'image/png',
+		'image/gif',
+		'image/pjpeg',
+		'image/x-png',
+	);
+
+	if (!in_array($mime, $accepted_formats)) {
+		return false;
+	}
+	return true;
+}
+
+function tp_upload_memory_check($image_lib, $num_pixels) {
+	if ($image_lib !== 'GD') {
+		return true;
+	}
+
+	$mem_avail = ini_get('memory_limit');
+	$mem_avail = rtrim($mem_avail, 'M');
+	$mem_avail = $mem_avail * 1024 * 1024;
+	$mem_used = memory_get_usage();
+	$mem_required = ceil(5.35 * $num_pixels);
+
+	$mem_avail = $mem_avail - $mem_used - 2097152; // 2 MB buffer
+	if ($mem_required > $mem_avail) {
+		return false;
+	}
+
+	return true;
+}
\ No newline at end of file
-- 
cgit v1.2.3