wip: club

src/main/java/fr/titionfire/ffsaf/utils/Utils.java

@@ -1,8 +1,19 @@
 package fr.titionfire.ffsaf.utils;
 
+import io.smallrye.mutiny.Uni;
+import jakarta.ws.rs.core.HttpHeaders;
+import jakarta.ws.rs.core.MediaType;
+import jakarta.ws.rs.core.Response;
 import jodd.net.MimeTypes;
+import net.sf.jmimemagic.Magic;
+import net.sf.jmimemagic.MagicException;
+import net.sf.jmimemagic.MagicMatchNotFoundException;
+import net.sf.jmimemagic.MagicParseException;
+import org.jboss.logging.Logger;
 
 import java.io.*;
+import java.net.URI;
+import java.net.URISyntaxException;
 import java.net.URLConnection;
 import java.nio.file.Files;
 import java.util.Calendar;
@@ -11,6 +22,7 @@ import java.util.concurrent.CompletableFuture;
 import java.util.concurrent.Future;
 
 public class Utils {
+    private static final org.jboss.logging.Logger LOGGER = Logger.getLogger(Utils.class);
 
     public static int getSaison() {
         return getSaison(new Date());
@@ -30,7 +42,12 @@ public class Utils {
     public static Future<String> replacePhoto(long id, byte[] input, String media, String dir) {
         return CompletableFuture.supplyAsync(() -> {
             try (InputStream is = new BufferedInputStream(new ByteArrayInputStream(input))) {
-                String mimeType = URLConnection.guessContentTypeFromStream(is);
+                String mimeType;
+                try {
+                    mimeType = Magic.getMagicMatch(input, false).getMimeType();
+                } catch (MagicParseException | MagicMatchNotFoundException | MagicException e) {
+                    mimeType = URLConnection.guessContentTypeFromStream(is);
+                }
                 String[] detectedExtensions = MimeTypes.findExtensionsByMimeTypes(mimeType, false);
                 if (detectedExtensions.length == 0)
                     throw new IOException("Fail to detect file extension for MIME type " + mimeType);
@@ -57,4 +74,39 @@ public class Utils {
             }
         });
     }
+
+    public static Uni<Response> getMediaFile(long id, String media, String dirname,
+                                              Uni<?> uniBase) throws URISyntaxException {
+        Future<Pair<File, byte[]>> future = CompletableFuture.supplyAsync(() -> {
+            FilenameFilter filter = (directory, filename) -> filename.startsWith(String.valueOf(id));
+            File[] files = new File(media, dirname).listFiles(filter);
+            if (files != null && files.length > 0) {
+                File file = files[0];
+                try {
+                    byte[] data = Files.readAllBytes(file.toPath());
+                    return new Pair<>(file, data);
+                } catch (IOException ignored) {
+                }
+            }
+            return null;
+        });
+
+        URI uri = new URI("https://mdbcdn.b-cdn.net/img/Photos/new-templates/bootstrap-chat/ava2.webp");
+
+        return uniBase.chain(__ -> Uni.createFrom().future(future)
+                .map(filePair -> {
+                    if (filePair == null)
+                        return Response.temporaryRedirect(uri).build();
+
+                    String mimeType = URLConnection.guessContentTypeFromName(filePair.getKey().getName());
+
+                    Response.ResponseBuilder resp = Response.ok(filePair.getValue());
+                    resp.type(MediaType.APPLICATION_OCTET_STREAM);
+                    resp.header(HttpHeaders.CONTENT_LENGTH, filePair.getValue().length);
+                    resp.header(HttpHeaders.CONTENT_TYPE, mimeType);
+                    resp.header(HttpHeaders.CONTENT_DISPOSITION, "inline; ");
+
+                    return resp.build();
+                }));
+    }
 }