Working Copy

src/main/java/HamletParser.java

@@ -3,6 +3,9 @@ import java.io.IOException;
 import java.util.Scanner;
 import java.lang.*;
 import java.util.*;
+import java.util.regex.Pattern;
+import java.util.regex.Matcher;
+
 /**
  * Created by thook on 10/7/15.
  */
@@ -10,92 +13,125 @@ public class HamletParser {
 
     private String hamletData;
     private Formatter formatter;
+    private String hamletFile = "hamlet.txt";
 
     public HamletParser(){
-        this.hamletData = loadFile();
+        // this.hamletData = loadFile(fileName);
+    }
+
+    public String getHamletFile() {
+        return hamletFile;
     }
 
-    private String loadFile(){
+    protected String loadFile(String fileName){
         ClassLoader classLoader = getClass().getClassLoader();
-        File file = new File(classLoader.getResource("hamlet.txt").getFile());
+        File file = new File(classLoader.getResource(fileName).getFile());
         StringBuilder result = new StringBuilder("");
-        StringBuilder lineResult;
-        String[] wordArr;
 
         try(Scanner scanner = new Scanner(file)){
             while(scanner.hasNextLine()){
-                String line = scanner.nextLine();
-                wordArr = line.split(" ");
-                lineResult = new StringBuilder("");
-                for (String s: wordArr) {
-                    if (findHamlet(s)) {
-                        lineResult.append(changeHamletToLeon()).append(" ");
-                    } else if (findHoratio(s)) {
-                        lineResult.append(changeChangeHoratioToTariq()).append(" ");
-                    } else {
-                        lineResult.append(s).append(" ");
-                    }
-                }
-                result.append(lineResult.toString().trim());
+                String line = replaceHamletAndHoratio(scanner.nextLine());
+                result.append(line).append("\n");
             }
-
             scanner.close();
         }catch(IOException e){
             e.printStackTrace();
         }
-        openOutFile();
-        addText(result.toString());
-        closeOutFile();
         return result.toString();
     }
+//    protected String loadFile(){
+//        ClassLoader classLoader = getClass().getClassLoader();
+//        File file = new File(classLoader.getResource("hamlet.txt").getFile());
+//        StringBuilder result = new StringBuilder("");
+//        StringBuilder lineResult;
+//        String[] wordArr;
+//
+//        try(Scanner scanner = new Scanner(file)){
+//            while(scanner.hasNextLine()){
+//                String line = scanner.nextLine();
+//                wordArr = line.split(" ");
+//                lineResult = new StringBuilder("");
+//                for (String s: wordArr) {
+//                    if (findHamlet(s)) {
+//                        lineResult.append(changeHamletToLeon()).append(" ");
+//                    } else if (findHoratio(s)) {
+//                        lineResult.append(changeChangeHoratioToTariq()).append(" ");
+//                    } else {
+//                        lineResult.append(s).append(" ");
+//                    }
+//                }
+//                result.append(lineResult.toString().trim() + "\n");
+//            }
+//
+//            scanner.close();
+//        }catch(IOException e){
+//            e.printStackTrace();
+//        }
+//        openOutFile();
+//        addText(result.toString());
+//        closeOutFile();
+//        return result.toString();
+//    }
 
     public String getHamletData(){
         return hamletData;
     }
 
-    public boolean findHamlet(String s){
-        if (s.toLowerCase().matches("hamlet")){
-            return true;
-        } else {
-            return false;
-        }
-    }
-
-    public boolean findHoratio(String s){
-        if (s.toLowerCase().matches("horatio")){
-            return true;
-        } else {
-            return false;
-        }
-    }
-
-    public String changeHamletToLeon(){
-        return "Leon";
-    }
-
-    public String changeChangeHoratioToTariq(){
-        return "Tariq";
-    }
+//    public boolean findHamlet(String s){
+//        if (s.toLowerCase().matches("hamlet")){
+//            return true;
+//        } else {
+//            return false;
+//        }
+//    }
+//
+//    public boolean findHoratio(String s){
+//        if (s.toLowerCase().matches("horatio")){
+//            return true;
+//        } else {
+//            return false;
+//        }
+//    }
+//
+//    public String changeHamletToLeon(){
+//        return "Leon";
+//    }
+//
+//    public String changeChangeHoratioToTariq(){
+//        return "Tariq";
+//    }
 
     /**
      * Methods to print out to a new file
      */
-    public void openOutFile(){
+    public void openOutFile(String fileName){
         try {
-            formatter = new Formatter("output.txt");
+            formatter = new Formatter(fileName);
         }
         catch (Exception e){
             System.out.println("Error writing to file.");
         }
     }
-
-    public String addText(String s){
-        formatter.format("%s", s);
-        return loadFile();
+    public String writeToFile(String output){
+        formatter.format("%s", output);
+        return output;
     }
-
     public void closeOutFile(){
         formatter.close();
     }
 
+    public String replaceHamletAndHoratio(String input) {
+//        String noHamlet = input.replaceAll("(?i)hamlet", "Leon");
+//        String noHoratioOrHamlet = noHamlet.replaceAll("(?i)horatio", "Tarik");
+//        return noHoratioOrHamlet;
+        String returnString = input;
+        String[] replaceArr = {"Hamlet", "Horatio", "hamlet", "horatio", "HAMLET", "HORATIO"};
+        String[] changeToArr = {"Leon", "Tarik", "leon", "tarik", "LEON", "TARIK"};
+
+        for (int i = 0; i < replaceArr.length; i++){
+            returnString = returnString.replaceAll(replaceArr[i], changeToArr[i]);
+        }
+        return returnString;
+    }
+
 }

src/main/java/MainApp.java

@@ -0,0 +1,10 @@
+public class MainApp {
+    public static void main(String[] args) {
+        String noHamletOrHoratio;
+        HamletParser hamletParser = new HamletParser();
+        noHamletOrHoratio = hamletParser.loadFile(hamletParser.getHamletFile());
+        hamletParser.openOutFile("output.txt");
+        hamletParser.writeToFile(noHamletOrHoratio);
+        hamletParser.closeOutFile();
+    }
+}

src/main/resources/hamletTest.txt

@@ -0,0 +1,2 @@
+The Tragedy of Hamlet, Prince of Denmark
+HORATIO

src/test/java/HamletParserTest.java

@@ -15,68 +15,92 @@ public class HamletParserTest {
     }
 
     @Test
-    public void testChangeHamletToLeon() {
+    public void loadFileTest() {
         //GIVEN
 
         //WHEN
-        String expected = "Leon";
+        String expected = "The Tragedy of Leon, Prince of Denmark\nTARIK\n";
         //THEN
-        String actual = hamletParser.changeHamletToLeon();
+        String actual = hamletParser.loadFile("hamletTest.txt");
         Assert.assertEquals(expected, actual);
-    }
-
-    @Test
-    public void testChangeHoratioToTariq() {
-        //GIVEN
-
-        //WHEN
-        String expected = "Tariq";
-        //THEN
-        String actual = hamletParser.changeChangeHoratioToTariq();
-        Assert.assertEquals(expected, actual);
-    }
 
-    @Test
-    public void testFindHoratio() {
-        //GIVEN
-        String input = "horatio";
-        //WHEN
-        boolean expected = true;
-        //THEN
-        boolean actual = hamletParser.findHoratio(input);
-        Assert.assertEquals(expected, actual);
+        // the file to output, then test output for 'has hamlet or horatio
     }
 
-    @Test
-    public void testFindHamlet() {
-        //GIVEN
-        String input = "HAMLET";
-        //WHEN
-        boolean expected = true;
-        //THEN
-        boolean actual = hamletParser.findHamlet(input);
-        Assert.assertEquals(expected, actual);
-    }
-
-    @Test
-    public void testFindHoratio2() {
-        //GIVEN
-        String input = "nope";
-        //WHEN
-        boolean expected = false;
-        //THEN
-        boolean actual = hamletParser.findHoratio(input);
-        Assert.assertEquals(expected, actual);
-    }
+//    @Test
+//    public void testChangeHamletToLeon() {
+//        //GIVEN
+//
+//        //WHEN
+//        String expected = "Leon";
+//        //THEN
+//        String actual = hamletParser.changeHamletToLeon();
+//        Assert.assertEquals(expected, actual);
+//    }
+//
+//    @Test
+//    public void testChangeHoratioToTariq() {
+//        //GIVEN
+//
+//        //WHEN
+//        String expected = "Tariq";
+//        //THEN
+//        String actual = hamletParser.changeChangeHoratioToTariq();
+//        Assert.assertEquals(expected, actual);
+//    }
+//
+//    @Test
+//    public void testFindHoratio() {
+//        //GIVEN
+//        String input = "horatio";
+//        //WHEN
+//        boolean expected = true;
+//        //THEN
+//        boolean actual = hamletParser.findHoratio(input);
+//        Assert.assertEquals(expected, actual);
+//    }
+//
+//    @Test
+//    public void testFindHamlet() {
+//        //GIVEN
+//        String input = "HAMLET";
+//        //WHEN
+//        boolean expected = true;
+//        //THEN
+//        boolean actual = hamletParser.findHamlet(input);
+//        Assert.assertEquals(expected, actual);
+//    }
+//
+//    @Test
+//    public void testFindHoratio2() {
+//        //GIVEN
+//        String input = "nope";
+//        //WHEN
+//        boolean expected = false;
+//        //THEN
+//        boolean actual = hamletParser.findHoratio(input);
+//        Assert.assertEquals(expected, actual);
+//    }
+//
+//    @Test
+//    public void testFindHamlet2() {
+//        //GIVEN
+//        String input = "hhamlet";
+//        //WHEN
+//        boolean expected = false;
+//        //THEN
+//        boolean actual = hamletParser.findHamlet(input);
+//        Assert.assertEquals(expected, actual);
+//    }
 
     @Test
-    public void testFindHamlet2() {
+    public void hamletReplaceTest() {
         //GIVEN
-        String input = "hhamlet";
+        String input = "Hamlet Horatio hamlet horatio HAMLET HORATIO";
         //WHEN
-        boolean expected = false;
+        String expected = "Leon Tarik leon tarik LEON TARIK";
         //THEN
-        boolean actual = hamletParser.findHamlet(input);
+        String actual = hamletParser.replaceHamletAndHoratio(input);
         Assert.assertEquals(expected, actual);
     }
 }