add default content type notion

[jSite2.git] / src / net / pterodactylus / jsite / project / Project.java

diff --git a/src/net/pterodactylus/jsite/project/Project.java b/src/net/pterodactylus/jsite/project/Project.java
index 5c58763..36a27f9 100644 (file)
--- a/src/net/pterodactylus/jsite/project/Project.java
+++ b/src/net/pterodactylus/jsite/project/Project.java
@@ -25,6 +25,7 @@ import java.util.ArrayList;
 import java.util.List;
 
 import net.pterodactylus.util.beans.AbstractBean;
+import net.pterodactylus.util.io.MimeTypes;
 
 /**
  * Container for project information. A Project is capable of notifying
@@ -299,6 +300,12 @@ public class Project extends AbstractBean {
                        entry.setPath(file.getPath());
                        entry.setDefaultInsert(!file.isHidden());
                        entry.setInsert(!file.isHidden());
+                       String extension = fileName.substring(fileName.lastIndexOf('.') + 1);
+                       List<String> mimeTypes = MimeTypes.getMimeTypes(extension);
+                       if (!mimeTypes.isEmpty()) {
+                               entry.setDefaultContentType(mimeTypes.get(0));
+                               entry.setContentType(mimeTypes.get(0));
+                       }
                        entries.add(entry);
                }
        }