Storage.java

package duke;

import duke.exception.BadDateArgumentException;
import duke.exception.EmptyArgumentException;
import duke.task.Deadline;
import duke.task.Event;
import duke.task.Task;
import duke.task.ToDos;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class Storage {
    public static final String FILE_DIR = "data";
    public static final String FILE_NAME = "duke.txt";
    static int badLines = 0;//Last call bad lines

    public static TaskList LoadTaskList() throws IOException {
        badLines = 0;
        List<Task> store = new ArrayList<>();
        File file = getOrCreateFile();
        Scanner s = new Scanner(file);
        generateLines: //TODO: Whats the code style for this.
        while(s.hasNextLine()){
            Task t;
            String line = s.nextLine();
            String pattern = "([TED]),([01]),(\\d*),(.*)";
            Pattern r = Pattern.compile(pattern);
            Matcher m = r.matcher(line);
            if(!m.find()){
                badLines++;
                break;
            }
            String type = m.group(1);
            boolean isDone = m.group(2).equals("1");
            int taskLength = Integer.parseInt(m.group(3));
            String task = m.group(4).substring(0,taskLength);
            String leftover = m.group(4).substring(taskLength);
            try {
                if (type.equals("E") || type.equals("D") ) {
                    line = leftover.substring(1);
                    pattern = "(\\d*),(.*)";
                    r = Pattern.compile(pattern);
                    m = r.matcher(line);
                    if (!m.find()) {
                        badLines++;
                        break;
                    }
                    int timeLength = Integer.parseInt(m.group(1));
                    String timeData = m.group(2).substring(0, timeLength);

                    switch (type) {
                    case "E":
                        t = new Event(task, timeData);
                        break;
                    case "D":
                        t = new Deadline(task, timeData);
                        break;
                    default:
                        badLines++;
                        break generateLines;
                    }
                } else {
                    t = new ToDos(task);
                }
            }catch(EmptyArgumentException e){
                badLines++;
                break;
            }catch(BadDateArgumentException e){
                badLines++;
                break;
            }
            if (isDone){
                t.setDone();
            }
            store.add(t);
        }
        s.close();
        return new TaskList(store);
    }

    public static void saveTaskList(TaskList data) throws IOException {
        List<Task> store = data.getRawData();
        StringBuilder saveText = new StringBuilder();
        for (Task t: store){
            saveText.append(t.toFileString());
            saveText.append('\n');
        }
        File f = getOrCreateFile();
        FileWriter writer = new FileWriter(f);
        writer.write(saveText.toString());
        writer.close();
    }

    private static File getOrCreateFile() throws IOException {
        Files.createDirectories(Paths.get(FILE_DIR));
        File file = new File(FILE_DIR, FILE_NAME); // create a File for the given file path
        file.createNewFile();
        return file;
    }
}