From 9a715a4a09d2470f8c0fc0ab72ae7473a61294b3 Mon Sep 17 00:00:00 2001
From: dizda13 <dino_dizda@live.com>
Date: Sun, 16 Apr 2017 17:22:23 +0200
Subject: [PATCH] event gallery upload/download

---
 .../storage/FileSystemStorageService.java     |  8 ++
 .../ba/steleks/storage/StorageException.java  |  0
 .../storage/StorageFileNotFoundException.java |  0
 .../ba/steleks/storage/StorageProperties.java |  0
 .../ba/steleks/storage/StorageService.java    |  0
 .../java/ba/steleks/EventsApplication.java    |  4 +
 .../controller/EventGalleryController.java    | 81 +++++++++++++++++++
 7 files changed, 93 insertions(+)
 rename {users => common}/src/main/java/ba/steleks/storage/FileSystemStorageService.java (88%)
 rename {users => common}/src/main/java/ba/steleks/storage/StorageException.java (100%)
 rename {users => common}/src/main/java/ba/steleks/storage/StorageFileNotFoundException.java (100%)
 rename {users => common}/src/main/java/ba/steleks/storage/StorageProperties.java (100%)
 rename {users => common}/src/main/java/ba/steleks/storage/StorageService.java (100%)
 create mode 100644 events/src/main/java/ba/steleks/controller/EventGalleryController.java

diff --git a/users/src/main/java/ba/steleks/storage/FileSystemStorageService.java b/common/src/main/java/ba/steleks/storage/FileSystemStorageService.java
similarity index 88%
rename from users/src/main/java/ba/steleks/storage/FileSystemStorageService.java
rename to common/src/main/java/ba/steleks/storage/FileSystemStorageService.java
index 4906eac..9e91224 100644
--- a/users/src/main/java/ba/steleks/storage/FileSystemStorageService.java
+++ b/common/src/main/java/ba/steleks/storage/FileSystemStorageService.java
@@ -12,6 +12,7 @@ import java.net.MalformedURLException;
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.nio.file.Paths;
+import java.nio.file.attribute.FileOwnerAttributeView;
 import java.util.stream.Stream;
 
 @Service
@@ -30,6 +31,13 @@ public class FileSystemStorageService implements StorageService {
             if (file.isEmpty()) {
                 throw new StorageException("Failed to store empty file " + file.getOriginalFilename());
             }
+
+            String tempDest="";
+            String[] locations=this.rootLocation.resolve(dest).toString().split("/");
+            for(int i=0; i<locations.length-1;i++)
+                    tempDest=tempDest+"/"+locations[i];
+            Files.createDirectory(Paths.get(tempDest));
+
             Files.copy(file.getInputStream(),
                     this.rootLocation.resolve(dest));
         } catch (IOException e) {
diff --git a/users/src/main/java/ba/steleks/storage/StorageException.java b/common/src/main/java/ba/steleks/storage/StorageException.java
similarity index 100%
rename from users/src/main/java/ba/steleks/storage/StorageException.java
rename to common/src/main/java/ba/steleks/storage/StorageException.java
diff --git a/users/src/main/java/ba/steleks/storage/StorageFileNotFoundException.java b/common/src/main/java/ba/steleks/storage/StorageFileNotFoundException.java
similarity index 100%
rename from users/src/main/java/ba/steleks/storage/StorageFileNotFoundException.java
rename to common/src/main/java/ba/steleks/storage/StorageFileNotFoundException.java
diff --git a/users/src/main/java/ba/steleks/storage/StorageProperties.java b/common/src/main/java/ba/steleks/storage/StorageProperties.java
similarity index 100%
rename from users/src/main/java/ba/steleks/storage/StorageProperties.java
rename to common/src/main/java/ba/steleks/storage/StorageProperties.java
diff --git a/users/src/main/java/ba/steleks/storage/StorageService.java b/common/src/main/java/ba/steleks/storage/StorageService.java
similarity index 100%
rename from users/src/main/java/ba/steleks/storage/StorageService.java
rename to common/src/main/java/ba/steleks/storage/StorageService.java
diff --git a/events/src/main/java/ba/steleks/EventsApplication.java b/events/src/main/java/ba/steleks/EventsApplication.java
index a047ca1..7ca5dec 100644
--- a/events/src/main/java/ba/steleks/EventsApplication.java
+++ b/events/src/main/java/ba/steleks/EventsApplication.java
@@ -1,10 +1,13 @@
 package ba.steleks;
 
+
 import ba.steleks.service.ServiceConstants;
+import ba.steleks.storage.StorageProperties;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.beans.factory.annotation.Value;
 import org.springframework.boot.SpringApplication;
 import org.springframework.boot.autoconfigure.SpringBootApplication;
+import org.springframework.boot.context.properties.EnableConfigurationProperties;
 import org.springframework.cloud.client.ServiceInstance;
 import org.springframework.cloud.client.discovery.DiscoveryClient;
 import org.springframework.cloud.client.discovery.EnableDiscoveryClient;
@@ -18,6 +21,7 @@ import java.util.List;
 
 @EnableDiscoveryClient
 @SpringBootApplication
+@EnableConfigurationProperties(StorageProperties.class)
 public class EventsApplication {
 
 	public static void main(String[] args) {
diff --git a/events/src/main/java/ba/steleks/controller/EventGalleryController.java b/events/src/main/java/ba/steleks/controller/EventGalleryController.java
new file mode 100644
index 0000000..06e8945
--- /dev/null
+++ b/events/src/main/java/ba/steleks/controller/EventGalleryController.java
@@ -0,0 +1,81 @@
+package ba.steleks.controller;
+
+import ba.steleks.error.exception.ExternalServiceException;
+import ba.steleks.model.Media;
+import ba.steleks.repository.EventsJpaRepository;
+import ba.steleks.repository.MediaJpaRepository;
+import ba.steleks.storage.StorageFileNotFoundException;
+import ba.steleks.storage.StorageService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.cloud.client.ServiceInstance;
+import org.springframework.cloud.client.discovery.DiscoveryClient;
+import org.springframework.core.io.Resource;
+import org.springframework.http.HttpHeaders;
+import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.*;
+import org.springframework.web.multipart.MultipartFile;
+import org.springframework.web.servlet.mvc.support.RedirectAttributes;
+
+import java.util.Date;
+import java.util.List;
+
+/**
+ * Created by admin on 16/04/2017.
+ */
+@RestController
+public class EventGalleryController {
+
+    private final StorageService storageService;
+    private final MediaJpaRepository repository;
+    private final DiscoveryClient discoveryClient;
+
+    @Autowired
+    public EventGalleryController(StorageService storageService, MediaJpaRepository repository, DiscoveryClient discoveryClient) {
+        this.storageService = storageService;
+        this.repository = repository;
+        this.discoveryClient=discoveryClient;
+    }
+
+    @GetMapping("/eventPictures/{filename:.+}")
+    @ResponseBody
+    public ResponseEntity<Resource> serveFile(@PathVariable String filename) {
+
+        Resource file = storageService.loadAsResource("mediaEvent/"+filename);
+        return ResponseEntity
+                .ok()
+                .header(HttpHeaders.CONTENT_DISPOSITION, "attachment; filename=\""+file.getFilename()+"\"")
+                .body(file);
+    }
+
+    @PostMapping("/media/{mediaId}/picture")
+    public String handleFileUpload(@PathVariable Long mediaId, @RequestParam("file") MultipartFile file,
+                                   RedirectAttributes redirectAttributes) throws ExternalServiceException {
+
+        List<ServiceInstance> usersInstances = discoveryClient.getInstances("events");
+        if(usersInstances == null || usersInstances.size() == 0) {
+            System.err.print("Users service not found!");
+            throw new ExternalServiceException();
+        }
+
+        ServiceInstance usersService = usersInstances.get(0);
+        String mediaServiceBase = usersService.getUri().toString();
+
+        String[] names = file.getOriginalFilename().split("\\.");
+        String dest = String.valueOf("mediaEvent/" + mediaId + "_" + new Date().getTime()) + "." + names[names.length - 1];
+        storageService.store(file, dest);
+        redirectAttributes.addFlashAttribute("message",
+                "You successfully uploaded " + file.getOriginalFilename() + "!");
+
+        Media media = repository.findOne(mediaId);
+        media.setContentUrl(mediaServiceBase+"/profilePictures/" + dest);
+
+        repository.save(media);
+
+        return "redirect:/";
+    }
+
+    @ExceptionHandler(StorageFileNotFoundException.class)
+    public ResponseEntity handleStorageFileNotFound(StorageFileNotFoundException exc) {
+        return ResponseEntity.notFound().build();
+    }
+}