fixed parsing

AlbertZyc · Dec 25, 2018 · 47f1938 · 47f1938
1 parent 213a4e8
commit 47f1938
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/main/kotlin/com/mistamek/drawablepreview/factories/Constants.kt b/src/main/kotlin/com/mistamek/drawablepreview/factories/Constants.kt
@@ -5,5 +5,5 @@ import com.intellij.util.ui.UIUtil
 object Constants {
     val ICON_SIZE = if (UIUtil.isRetina()) 36 else 16
     const val XML_TYPE = ".xml"
-    const val DRAWABLES_FOLDER_TYPE = "drawable"
+    val SUPPORTED_FOLDERS = arrayOf("drawable", "mipmap")
 }
diff --git a/src/main/kotlin/com/mistamek/drawablepreview/factories/XmlImageFactory.kt b/src/main/kotlin/com/mistamek/drawablepreview/factories/XmlImageFactory.kt
@@ -42,7 +42,8 @@ object XmlImageFactory {
     fun getDrawable(path: String): Drawable? = parseDocument(path)?.let { DrawableInflater.getDrawable(it.documentElement) }
 
     private fun parseDocument(path: String): Document? {
-        if (!(path.endsWith(Constants.XML_TYPE) && path.contains(Constants.DRAWABLES_FOLDER_TYPE))) {
+        val supportedFolder = Constants.SUPPORTED_FOLDERS.fold(false) { acc, next -> acc || path.contains(next) }
+        if (!(path.endsWith(Constants.XML_TYPE) && supportedFolder)) {
             return null
         }