package org.jabref.gui.mergeentries.multiwaymerge;

import java.util.Arrays;
import java.util.List;
import java.util.stream.Collectors;

import javafx.scene.text.Text;

import com.github.difflib.DiffUtils;
import com.github.difflib.patch.AbstractDelta;

public class DiffHighlighting {

    private DiffHighlighting() {
    }

    public static List<Text> generateDiffHighlighting(String baseString, String modifiedString, String separator) {
        List<String> baseStringSplit = Arrays.asList(baseString.split(separator));
        List<String> modifiedStringSplit = Arrays.asList(modifiedString.split(separator));
        List<AbstractDelta<String>> deltaList = DiffUtils.diff(baseStringSplit, modifiedStringSplit).getDeltas();
        List<Text> result = baseStringSplit.stream().map(text -> forUnchanged(text + separator)).collect(Collectors.toList());
        for (AbstractDelta<String> delta : deltaList.reversed()) {
            int startPos = delta.getSource().getPosition();
            List<String> lines = delta.getSource().getLines();
            int offset = 0;
            switch (delta.getType()) {
                case CHANGE:
                    for (String line : lines) {
                        result.set(startPos + offset, forRemoved(line + separator));
                        offset++;
                    }
                    result.set(startPos + offset - 1, forRemoved(baseStringSplit.get((startPos + offset) - 1) + separator));
                    result.add(startPos + offset, forAdded(String.join(separator, delta.getTarget().getLines())));
                    break;
                case DELETE:
                    for (String line : lines) {
                        result.set(startPos + offset, forRemoved(line + separator));
                        offset++;
                    }
                    break;
                case INSERT:
                    result.add(delta.getSource().getPosition(), forAdded(String.join(separator, delta.getTarget().getLines())));
                    break;
                default:
                    break;
            }
        }
        return result;
    }

    public static Text forChanged(String text) {
        Text node = new Text(text);
        node.getStyleClass().add("text-changed");
        return node;
    }

    public static Text forUnchanged(String text) {
        Text node = new Text(text);
        node.getStyleClass().add("text-unchanged");
        return node;
    }

    public static Text forAdded(String text) {
        Text node = new Text(text);
        node.getStyleClass().add("text-added");
        return node;
    }

    public static Text forRemoved(String text) {
        Text node = new Text(text);
        node.getStyleClass().add("text-removed");
        return node;
    }
}
