package com.sleekbyte.tailor.output;

import com.sleekbyte.tailor.common.Location;
import com.sleekbyte.tailor.common.Messages;

import java.io.File;
import java.io.IOException;
import java.util.*;

/**
 * Generates and outputs formatted analysis messages for Xcode
 */
public class Printer implements AutoCloseable {

    private File inputFile;
    private HashMap<String, ViolationMessage> msgBuffer = new HashMap<String, ViolationMessage>();

    public Printer(File inputFile) {
        this.inputFile = inputFile;
    }

    /**
     * Prints warning message
     *
     * @param warningMsg warning message to print
     * @param location   location object containing line and column number for printing
     */
    public void warn(String warningMsg, Location location) {
        print(Messages.WARNING, warningMsg, location);
    }

    /**
     * Prints error message
     *
     * @param errorMsg error message to print
     * @param location location object containing line and column number for printing
     */
    public void error(String errorMsg, Location location) {
        print(Messages.ERROR, errorMsg, location);
    }

    private void print(String classification, String msg, Location location) {
        ViolationMessage violationMessage = null;
        try {
            violationMessage = new ViolationMessage(this.inputFile.getCanonicalPath(),
                                location.line,
                                location.column,
                                classification,
                                msg);

        } catch (IOException e) {
            System.err.println("Error in getting canonical path of input file: " + e.getMessage());
        }
        this.msgBuffer.put(violationMessage.toString(), violationMessage);
    }

    // Visible for testing only
    public static String genOutputStringForTest(String filePath, int line, String classification,
                                                String msg) {
        return filePath + ":" + line + ": " + classification + ": " + msg;
    }

    // Visible for testing only
    public static String genOutputStringForTest(String filePath, int line, int column, String classification,
                                                String msg) {
        return filePath + ":" + line + ":" + column + ": " + classification + ": " + msg;
    }

    @Override
    public void close() {
        flushBuffer();
    }

    private void flushBuffer() {

        ArrayList<ViolationMessage> outputList = new ArrayList<>();
        for(Map.Entry<String, ViolationMessage> map : msgBuffer.entrySet()){
            outputList.add(map.getValue());
        }
        Collections.sort(outputList);
        outputList.forEach(System.out::println);
        this.msgBuffer.clear();
    }
}

@@ -1263,11 +1263,46 @@
  = n
-ull
+ew ViolationMessage(%22%22, -1, 0, %22%22, %22%22)
 ;%0A  

Fix potential null reference bug