package jp.ac.u_ryukyu.ie.cr.jungle;


import jp.ac.u_ryukyu.ie.cr.jungle.data.list.List;
import jp.ac.u_ryukyu.ie.cr.jungle.persistent.ChangeList;
import jp.ac.u_ryukyu.ie.cr.jungle.persistent.Journal;
import jp.ac.u_ryukyu.ie.cr.jungle.persistent.NullJournal;
import jp.ac.u_ryukyu.ie.cr.jungle.query.traverser.DifferentialInterfaceTraverser;
import jp.ac.u_ryukyu.ie.cr.jungle.query.traverser.InterfaceTraverser;
import jp.ac.u_ryukyu.ie.cr.jungle.query.traverser.RedBlackTreeInterfaceTraverser;
import jp.ac.u_ryukyu.ie.cr.jungle.store.TreeContext;
import jp.ac.u_ryukyu.ie.cr.jungle.store.logger.DefaultTreeOperationLog;
import jp.ac.u_ryukyu.ie.cr.jungle.store.logger.TreeOperationLog;
import jp.ac.u_ryukyu.ie.cr.jungle.store.nodepath.DefaultNodePath;
import jp.ac.u_ryukyu.ie.cr.jungle.store.operations.TreeOperation;
import jp.ac.u_ryukyu.ie.cr.jungle.transaction.context.DefaultTreeContext;
import jp.ac.u_ryukyu.ie.cr.jungle.transaction.context.DifferenceTreeContext;
import jp.ac.u_ryukyu.ie.cr.jungle.transaction.editor.jungleTreeEditor.JungleTreeEditor;
import jp.ac.u_ryukyu.ie.cr.jungle.transaction.editor.treeEditor.DefaultTreeEditor;
import jp.ac.u_ryukyu.ie.cr.jungle.transaction.editor.treeEditor.DifferenceTreeEditor;
import jp.ac.u_ryukyu.ie.cr.jungle.transaction.editor.treeEditor.RedBlackTreeEditor;
import jp.ac.u_ryukyu.ie.cr.jungle.transaction.editor.treeEditor.TreeEditor;
import jp.ac.u_ryukyu.ie.cr.jungle.transaction.node.Default.DefaultTreeNode;
import jp.ac.u_ryukyu.ie.cr.jungle.transaction.node.Differencial.DifferencialTreeNode;
import jp.ac.u_ryukyu.ie.cr.jungle.transaction.node.TreeNode;
import jp.ac.u_ryukyu.ie.cr.jungle.transaction.node.redBlackTree.EmptyTreeNode;
import jp.ac.u_ryukyu.ie.cr.jungle.traverser.DefaultTraverser;
import jp.ac.u_ryukyu.ie.cr.jungle.query.traverser.DefaultInterfaceTraverser;
import jp.ac.u_ryukyu.ie.cr.jungle.traverser.RedBlackTreeTraverser;
import jp.ac.u_ryukyu.ie.cr.jungle.traverser.Traverser;
import jp.ac.u_ryukyu.ie.cr.jungle.tree.DefaultJungleTree;
import jp.ac.u_ryukyu.ie.cr.jungle.tree.DifferenceListJungleTree;
import jp.ac.u_ryukyu.ie.cr.jungle.tree.JungleTree;
import jp.ac.u_ryukyu.ie.cr.jungle.tree.RedBlackJungleTree;
import jp.ac.u_ryukyu.ie.cr.jungle.util.Either;
import jp.ac.u_ryukyu.ie.cr.jungle.util.Error.Error;

import java.util.Enumeration;
import java.util.Iterator;
import java.util.concurrent.ConcurrentHashMap;

public class DefaultJungle implements Jungle {
    private Journal journal;
    private ConcurrentHashMap<String, JungleTree> trees;
    private String uuid;
    private TreeEditor treeEditor;
    private TreeEditor differenceEditor;
    private TreeEditor redBlackTreeEditor;

    public static void main(String args[]) {
        DefaultJungle j = new DefaultJungle(null, "hoge");
        JungleTree t = j.createNewTree("fuga");

        JungleTreeEditor e1 = t.getJungleTreeEditor();

        DefaultNodePath root = new DefaultNodePath();

        Either<Error, JungleTreeEditor> either = e1.addNewChildAt(root, 0);
        e1 = either.b();
        either = e1.addNewChildAt(root.add(0), 0);
        e1 = either.b();
        e1.success();
    }

    public DefaultJungle(Journal journal, String uuid) {
        this.journal = new NullJournal();
        this.trees = new ConcurrentHashMap<String, JungleTree>();
        this.uuid = uuid;
        Traverser defaultTraverser = new DefaultTraverser();
        Traverser redBlackTreeTraverser = new RedBlackTreeTraverser();
        this.treeEditor = new DefaultTreeEditor(defaultTraverser);
        this.differenceEditor = new DifferenceTreeEditor(defaultTraverser);
        this.redBlackTreeEditor = new RedBlackTreeEditor(redBlackTreeTraverser);
    }

    @Override
    public JungleTree getTreeByName(String name) {
        return trees.get(name);
    }

    @Override
    public JungleTree createNewTree(final String name) {
        return createNewTree(name, new DefaultTreeNode());
    }

    @Override
    public JungleTree createNewTree(final String name, TreeNode rootNode) {

        ChangeList list = new ChangeList() {
            @Override
            public Iterator<TreeOperation> iterator() {
                List<TreeOperation> nil = new List<>();
                return nil.iterator();
            }

            @Override
            public String uuid() {
                return uuid;
            }

            @Override
            public String getTreeName() {
                return name;
            }

            @Override
            public TreeOperationLog getLog() {
                return new DefaultTreeOperationLog();
            }

        };
        InterfaceTraverser traverser = new DefaultInterfaceTraverser(rootNode, true);
        TreeContext tc = new DefaultTreeContext(rootNode, null, list, uuid, name, 0, traverser);
        JungleTree newTree = new DefaultJungleTree(tc, uuid, journal.getWriter(), treeEditor);
        if (trees.putIfAbsent(name, newTree) != null) {
            return null;
        }
        return newTree;
    }

    @Override
    public JungleTree createNewDifferenceTree(final String name) {
        TreeNode rootNode = new DifferencialTreeNode();
        return createNewDifferenceTree(name, rootNode);
    }

    @Override
    public JungleTree createNewDifferenceTree(final String name, TreeNode rootNode) {
        ChangeList list = new ChangeList() {
            @Override
            public Iterator<TreeOperation> iterator() {
                List<TreeOperation> nil = new List<>();
                return nil.iterator();
            }

            @Override
            public String uuid() {
                return uuid;
            }

            @Override
            public String getTreeName() {
                return name;
            }

            @Override
            public TreeOperationLog getLog() {
                return new DefaultTreeOperationLog();
            }

        };
        InterfaceTraverser traverser = new DifferentialInterfaceTraverser(rootNode,rootNode, true);
        TreeContext tc = new DifferenceTreeContext(rootNode, rootNode, null, list, uuid, name, 0, traverser);
        JungleTree newTree = new DifferenceListJungleTree(tc, uuid, journal.getWriter(), differenceEditor);
        if (trees.putIfAbsent(name, newTree) != null) {
            return null;
        }
        return newTree;
    }

    @Override
    public JungleTree createNewRedBlackTree(String treeName, String balanceKey) {
        ChangeList list = new ChangeList() {
            @Override
            public Iterator<TreeOperation> iterator() {
                List<TreeOperation> nil = new List<>();
                return nil.iterator();
            }

            @Override
            public String uuid() {
                return uuid;
            }

            @Override
            public String getTreeName() {
                return treeName;
            }

            @Override
            public TreeOperationLog getLog() {
                return new DefaultTreeOperationLog();
            }

        };
        TreeNode rootNode = new EmptyTreeNode();
        InterfaceTraverser traverser = new RedBlackTreeInterfaceTraverser(rootNode);
        TreeContext tc = new DefaultTreeContext(rootNode, null, list, uuid, treeName, 0, traverser);
        JungleTree newTree = new RedBlackJungleTree(tc, uuid, journal.getWriter(), redBlackTreeEditor, balanceKey);
        if (trees.putIfAbsent(treeName, newTree) != null) {
            return null;
        }
        return newTree;
    }

        @Override public Iterator<String> getTreeNames () {
            Enumeration<String> treeNames = trees.keys();
            return new Iterator<String>() {

                @Override
                public boolean hasNext() {
                    return treeNames.hasMoreElements();
                }

                @Override
                public String next() {
                    return treeNames.nextElement();
                }
            };
        }
    }