Search Fix

Swati4star · Jul 28, 2018 · 93b391d · 93b391d
1 parent 759b22b
commit 93b391d
Showing 1 changed file with 20 additions and 1 deletion.
diff --git a/app/src/main/java/swati4star/createpdf/util/DirectoryUtils.java b/app/src/main/java/swati4star/createpdf/util/DirectoryUtils.java
@@ -28,7 +28,7 @@ public DirectoryUtils(Context context) {
     public ArrayList<File> searchPDF(String query) {
         ArrayList<File> searchResult = new ArrayList<>();
         final File[] files = getOrCreatePdfDirectory().listFiles();
-        ArrayList<File> pdfs = getPdfsFromPdfFolder(files);
+        ArrayList<File> pdfs = searchPdfsFromPdfFolder(files);
         for (File pdf : pdfs) {
             String path = pdf.getPath();
             String[] fileName = path.split("/");
@@ -82,6 +82,25 @@ public ArrayList<File> getPdfsFromPdfFolder(File[] files) {
         return pdfFiles;
     }
 
+    private ArrayList<File> searchPdfsFromPdfFolder(File[] files) {
+        final ArrayList<File> pdfFiles = new ArrayList<>();
+        for (File file : files) {
+            if (!file.isDirectory() && file.getName().endsWith(mContext.getString(R.string.pdf_ext))) {
+                pdfFiles.add(file);
+                Log.v("adding", file.getName());
+            }
+            if (file.isDirectory()) {
+                for (File dirFiles : file.listFiles()) {
+                    if (!dirFiles.isDirectory() && dirFiles.getName().endsWith(mContext.getString(R.string.pdf_ext))) {
+                        pdfFiles.add(dirFiles);
+                        Log.v("adding", dirFiles.getName());
+                    }
+                }
+            }
+        }
+        return pdfFiles;
+    }
+
     /**
      * create PDF directory if directory does not exists
      */