From bb87401d1064e7c792467b49cb49e809edd2087f Mon Sep 17 00:00:00 2001
From: Dan Brown <ssddanbrown@googlemail.com>
Date: Sun, 17 Jan 2016 15:19:26 +0000
Subject: [PATCH] Improved image upload error handling and fixed bad
 user-avatar rewriting

---
 app/Exceptions/ImageUploadException.php  | 6 ++++++
 app/Http/Controllers/ImageController.php | 9 ++++++++-
 resources/assets/js/directives.js        | 1 +
 3 files changed, 15 insertions(+), 1 deletion(-)
 create mode 100644 app/Exceptions/ImageUploadException.php

diff --git a/app/Exceptions/ImageUploadException.php b/app/Exceptions/ImageUploadException.php
new file mode 100644
index 000000000..205bdd4ff
--- /dev/null
+++ b/app/Exceptions/ImageUploadException.php
@@ -0,0 +1,6 @@
+<?php namespace BookStack\Exceptions;
+
+
+use Exception;
+
+class ImageUploadException extends Exception {}
\ No newline at end of file
diff --git a/app/Http/Controllers/ImageController.php b/app/Http/Controllers/ImageController.php
index 146dd0c05..3fff28d3b 100644
--- a/app/Http/Controllers/ImageController.php
+++ b/app/Http/Controllers/ImageController.php
@@ -2,6 +2,7 @@
 
 namespace BookStack\Http\Controllers;
 
+use BookStack\Exceptions\ImageUploadException;
 use BookStack\Repos\ImageRepo;
 use Illuminate\Filesystem\Filesystem as File;
 use Illuminate\Http\Request;
@@ -69,7 +70,13 @@ class ImageController extends Controller
         ]);
 
         $imageUpload = $request->file('file');
-        $image = $this->imageRepo->saveNew($imageUpload, $type);
+
+        try {
+            $image = $this->imageRepo->saveNew($imageUpload, $type);
+        } catch (ImageUploadException $e) {
+            return response($e->getMessage(), 500);
+        }
+
         return response()->json($image);
     }
 
diff --git a/resources/assets/js/directives.js b/resources/assets/js/directives.js
index 758f16cd2..6ccfcf855 100644
--- a/resources/assets/js/directives.js
+++ b/resources/assets/js/directives.js
@@ -56,6 +56,7 @@ module.exports = function (ngApp) {
                 var usingIds = typeof scope.currentId !== 'undefined' || scope.currentId === 'false';
                 scope.image = scope.currentImage;
                 scope.value = scope.currentImage || '';
+                if (usingIds) scope.value = scope.currentId;
 
                 function setImage(imageModel, imageUrl) {
                     scope.image = imageUrl;