Allows the parsing of whole folders

Author: Heleno Campos

README.md

@@ -18,6 +18,7 @@ full: outputs the bibtex references for a pdf file
 short: outputs only titles/year/publication venue of the references for a pdf file. They are separeted by semicollons, to make it easier to create csv files.
 
 
+
 How to use:
 java -jar refExtractor.jar full pdffile.pdf
 
@@ -26,6 +27,8 @@ or
 java -jar refExtractor.jar short pdffile.pdf
 
 
+If you pass a folder instead of a file as argument, it will attempt to parse all files in that folder and will output in the selected mode.
+
 
 Alternatively, if you dont want to output to be printed in the terminal, you can redirect the output using "> output.txt"
 

nbactions.xml

@@ -10,7 +10,7 @@
                 <goal>org.codehaus.mojo:exec-maven-plugin:1.5.0:exec</goal>
             </goals>
             <properties>
-                <exec.args>-classpath %classpath info.heleno.references_extractor.App short C:\Users\Heleno.DESKTOP-89HH2F6\Downloads\840.pdf</exec.args>
+                <exec.args>-classpath %classpath info.heleno.references_extractor.App short C:\Users\Heleno.DESKTOP-89HH2F6\Downloads\papers</exec.args>
                 <exec.executable>java</exec.executable>
             </properties>
         </action>
@@ -24,7 +24,7 @@
                 <goal>org.codehaus.mojo:exec-maven-plugin:1.5.0:exec</goal>
             </goals>
             <properties>
-                <exec.args>-agentlib:jdwp=transport=dt_socket,server=n,address=${jpda.address} -classpath %classpath info.heleno.references_extractor.App short C:\Users\Heleno.DESKTOP-89HH2F6\Downloads\840.pdf</exec.args>
+                <exec.args>-agentlib:jdwp=transport=dt_socket,server=n,address=${jpda.address} -classpath %classpath info.heleno.references_extractor.App short C:\Users\Heleno.DESKTOP-89HH2F6\Downloads\papers</exec.args>
                 <exec.executable>java</exec.executable>
                 <jpda.listen>true</jpda.listen>
             </properties>
@@ -39,7 +39,7 @@
                 <goal>org.codehaus.mojo:exec-maven-plugin:1.5.0:exec</goal>
             </goals>
             <properties>
-                <exec.args>-classpath %classpath info.heleno.references_extractor.App short C:\Users\Heleno.DESKTOP-89HH2F6\Downloads\840.pdf</exec.args>
+                <exec.args>-classpath %classpath info.heleno.references_extractor.App short C:\Users\Heleno.DESKTOP-89HH2F6\Downloads\papers</exec.args>
                 <exec.executable>java</exec.executable>
             </properties>
         </action>

src/main/java/info/heleno/references_extractor/App.java

@@ -5,8 +5,8 @@
  */
 package info.heleno.references_extractor;
 
+import java.io.File;
 import java.io.FileInputStream;
-import java.io.FileNotFoundException;
 import java.io.IOException;
 import java.util.List;
 import java.util.logging.Level;
@@ -30,34 +30,53 @@ public static String getField(List<String> values) {
         return field;
     }
 
-    public static void main(String[] args) {
+    public static void printFullReferences(List<BibEntry> references) {
+        for (BibEntry entry : references) {
+            System.out.println(entry.toBibTeX());
+        }
+    }
+
+    public static void printShortReferences(List<BibEntry> references, String source) {
+        System.out.println("Source;Title;Year;Publication");
+        source = source.replaceFirst("[.][^.]+$", "");  //removes file name extension
+        for (BibEntry entry : references) {
+            String title = getField(entry.getAllFieldValues(BibEntryFieldType.TITLE));
+            String year = entry.getFirstFieldValue(BibEntryFieldType.YEAR);
+            String publication = getField(entry.getAllFieldValues(BibEntryFieldType.JOURNAL));
+            System.out.println("\"" + source + "\";"+"\"" + title + "\"" + ";" + "\"" + year + "\"" + ";" + "\"" + publication + "\"");
+        }
+    }
+
+    public static void extractFile(File file, String mode) {
         try {
-            if (args.length >= 2) {
-
-                ContentExtractor extractor = new ContentExtractor();
-                FileInputStream inputStream = new FileInputStream(args[1]);
-                extractor.setPDF(inputStream);
-                List<BibEntry> references = extractor.getReferences();
-                if (args[0].equals("short")) {
-                    for (BibEntry entry : references) {
-                        String title = getField(entry.getAllFieldValues(BibEntryFieldType.TITLE));
-                        String year = entry.getFirstFieldValue(BibEntryFieldType.YEAR);
-                        String publication = getField(entry.getAllFieldValues(BibEntryFieldType.JOURNAL));
-                        System.out.println("\""+title +"\""+ ";" +"\""+ year +"\""+ ";" +"\""+ publication+"\"");
-                    }
-                } else if (args[0].equals("full")) {
-                    for (BibEntry entry : references) {
-                        System.out.println(entry.toBibTeX());
+            ContentExtractor extractor = new ContentExtractor();
+            FileInputStream inputStream = new FileInputStream(file);
+            extractor.setPDF(inputStream);
+            List<BibEntry> references = extractor.getReferences();
+            if (mode.equals("short")) {
+                printShortReferences(references, file.getName());
+            } else if (mode.equals("full")) {
+                printFullReferences(references);
+            }
+
+        } catch (AnalysisException | IOException ex) {
+            Logger.getLogger(App.class.getName()).log(Level.SEVERE, null, ex);
+        }
+    }
+
+    public static void main(String[] args) {
+        if (args.length >= 2) {
+            File file = new File(args[1]);
+            if (file.exists()) {
+                if (file.isDirectory()) {
+                    for(File subfile: file.listFiles()){
+                        extractFile(subfile, args[0]);
                     }
+                } else {
+                    extractFile(new File(args[1]), args[0]);
                 }
             }
 
-        } catch (AnalysisException ex) {
-            Logger.getLogger(App.class.getName()).log(Level.SEVERE, null, ex);
-        } catch (FileNotFoundException ex) {
-            Logger.getLogger(App.class.getName()).log(Level.SEVERE, null, ex);
-        } catch (IOException ex) {
-            Logger.getLogger(App.class.getName()).log(Level.SEVERE, null, ex);
         }
     }
 }