import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Vector;

public class ChainReaction implements ChainReactionInterface {
    private final Map<Integer, Node> nodes = new HashMap<>();
    private volatile int workingThreads = 0;
    private DamagePropagationInterface damagePropagationInterface;

    @Override
    public synchronized void addObject(int id) {
        nodes.put(id, new Node(id));
    }

    @Override
    public synchronized void neighbour(int idA, int idB) {
        Node nodeA = nodes.get(idA);
        Node nodeB = nodes.get(idB);

        nodeA.addNeighbor(nodeB);
        nodeB.addNeighbor(nodeA);
    }

    @Override
    public boolean resultReady() {
        return workingThreads == 0;
    }

    @Override
    public boolean isDestroyed(int id) {
        return nodes.get(id).isDestroyed();
    }

    @Override
    public synchronized void setDamagePropagator(DamagePropagationInterface dpi) {
        this.damagePropagationInterface = dpi;
    }

    @Override
    public void destroyed(int id) {
        Node node = nodes.get(id);
        synchronized (node) {
            if (node.isLive()) {
                node.destroy();
            }
        }
        getNeighboursToProcess(node).stream()
                .map(DestroyingRunnable::new)
                .forEach(this::startTrackedThread);
    }

    private List<Node> getNeighboursToProcess(Node node) {
        List<Node> neighboursToDestroy = new LinkedList<>();
        for (Node neighbour : node.getNeighbours()) {
            synchronized (neighbour) {
                if (neighbour.isLive() && damagePropagationInterface.destroyedBy(node.id, neighbour.id)) {
                    neighbour.markToDestroy();
                    neighboursToDestroy.add(neighbour);
                }
            }
        }
        return neighboursToDestroy;
    }

    private void startTrackedThread(DestroyingRunnable runnable) {
        new Thread(new TrackingRunnable(runnable)).start();
    }

    private class DestroyingRunnable implements Runnable {
        private final Node node;

        private DestroyingRunnable(Node node) {
            this.node = node;
        }

        @Override
        public void run() {
            node.destroy();

            List<Node> neighboursToProcess = getNeighboursToProcess(node);
            if (!neighboursToProcess.isEmpty()) {
                neighboursToProcess.subList(1, neighboursToProcess.size()).stream()
                        .map(DestroyingRunnable::new)
                        .forEach(ChainReaction.this::startTrackedThread);

                new DestroyingRunnable(neighboursToProcess.get(0)).run();
            }
        }
    }

    private class TrackingRunnable implements Runnable {
        private final Runnable runnable;

        private TrackingRunnable(Runnable runnable) {
            this.runnable = runnable;
        }

        @Override
        public void run() {
            try {
                synchronized (ChainReaction.this) {
                    workingThreads++;
                }
                runnable.run();
            } finally {
                synchronized (ChainReaction.this) {
                    workingThreads--;
                }
            }
        }
    }

    private static class Node {
        private final int id;
        private final List<Node> neighbours = new Vector<>();
        private volatile NodeState state = NodeState.LIVE;

        private Node(int id) {
            this.id = id;
        }

        private List<Node> getNeighbours() {
            return neighbours;
        }

        private void addNeighbor(Node neighbor) {
            neighbours.add(neighbor);
        }

        private boolean isLive() {
            return state == NodeState.LIVE;
        }

        private boolean isDestroyed() {
            return state == NodeState.DESTROYED;
        }

        private synchronized void markToDestroy() {
            state = NodeState.MARKED_TO_DESTROY;
        }

        private synchronized void destroy() {
            state = NodeState.DESTROYED;
        }

        private enum NodeState {
            LIVE, MARKED_TO_DESTROY, DESTROYED
        }
    }
}