add filter for m2ts

YouROK · Sep 18, 2018 · abcea96 · abcea96
1 parent c9bb93c
commit abcea96
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/app/src/main/java/ru/yourok/torrserve/server/torrent/Torrent.kt b/app/src/main/java/ru/yourok/torrserve/server/torrent/Torrent.kt
@@ -3,6 +3,7 @@ package ru.yourok.torrserve.server.torrent
 import ru.yourok.torrserve.server.api.Api
 import ru.yourok.torrserve.server.api.JSObject
 import ru.yourok.torrserve.utils.Mime
+import java.io.File
 import kotlin.concurrent.thread
 
 object Torrent {
@@ -27,8 +28,17 @@ object Torrent {
         val files = getFiles(torr)
         val retList = mutableListOf<JSObject>()
         files.forEach {
-            if (Mime.getMimeType(it.getString("Name", "")) != "*/*")
+            val name = it.get("Name", "")
+            if (Mime.getMimeType(name) != "*/*") {
+                val size = it.getLong("Size", 0L)
+                if (File(name).extension.toLowerCase() == "m2ts") {
+                    if (size > 1073741824L)
+                        retList.add(it)
+                } else
+                    retList.add(it)
+            } else if (name.toLowerCase().contains("bdmv/index.bdmv")) {
                 retList.add(it)
+            }
         }
         return retList
     }