Ticket #675: StyledEditFileChecker.2.diff

src/apps/stylededit/StyledEditApp.cpp

@@ -16 +16 @@
 #include <CharacterSetRoster.h>
 #include <Screen.h>
 #include <stdio.h>
+#include <NodeInfo.h>
 
 #include "Constants.h"
 #include "StyledEditApp.h"
@@ -165 +166 @@
     fWindowCount++;
 }
 
+bool StyledEditApp::CheckMimeType(entry_ref* ref)
+{
+    BEntry entry(ref, true); // traverse an eventual link
+    if (entry.InitCheck() != B_OK)
+        return false;
 
+    if (entry.IsDirectory())
+        return false;
+
+    BNode fileNode(&entry);
+    if (fileNode.InitCheck() != B_OK)
+        return false;
+    
+    BNodeInfo nodeInfo(&fileNode);
+    char fileTypeStr[256];
+    nodeInfo.GetType(fileTypeStr);
+    
+    BMimeType fileType(fileTypeStr);
+
+    // Check for malformed or inexistant mime type
+    if (!fileType.IsValid()){
+        char fileName[B_FILE_NAME_LENGTH];
+        entry.GetName(fileName);            
+        BString alertText;          
+        alertText << "Sorry! Can't open \"" << fileName << "\"!\nStyledEdit can't determine the type of the file!";
+        BAlert* alert = new BAlert("StyledEdit Load Failed", alertText.String(), "Ok", 0, 0, B_WIDTH_AS_USUAL, B_EVEN_SPACING, B_STOP_ALERT);
+        alert->SetShortcut(0, B_ESCAPE);
+        alert->Go();
+        return false;
+    }
+
+    BMimeType desiredSuperType("text");
+        
+    if (desiredSuperType.Contains(&fileType)) {
+        return true;
+    } else {            
+        char fileName[B_FILE_NAME_LENGTH];
+        entry.GetName(fileName);        
+        BString alertText;          
+        alertText << "Sorry! Can't open \"" << fileName << "\"!\nStyledEdit can't load files of type \"" << fileTypeStr << "\"!";
+        BAlert* alert = new BAlert("StyledEdit Load Failed", alertText.String(), "Ok", 0, 0, B_WIDTH_AS_USUAL, B_EVEN_SPACING, B_STOP_ALERT);
+        alert->SetShortcut(0, B_ESCAPE);
+        alert->Go();            
+        return false;   
+    }
+}
+
 void
 StyledEditApp::OpenDocument(entry_ref* ref)
 {
-    cascade();
-    new StyledEditWindow(gWindowRect, ref, fOpenAsEncoding);
-    fWindowCount++;
+    if (CheckMimeType(ref)){
+        cascade();
+        new StyledEditWindow(gWindowRect, ref, fOpenAsEncoding);
+        fWindowCount++;
+    }
 }
 
 

src/apps/stylededit/StyledEditApp.h

@@ -31 +31 @@
                 void    OpenDocument();
                 void    OpenDocument(entry_ref * ref);
                 void    CloseDocument();
+                bool    CheckMimeType(entry_ref* ref);
 
     private:
         BFilePanel*     fOpenPanel;