package nodebox.client;

import nodebox.node.Node;

import javax.imageio.ImageIO;
import javax.swing.*;
import java.awt.*;
import java.awt.event.MouseEvent;
import java.awt.event.MouseListener;
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;

public class AddressBar extends JPanel implements MouseListener {

    public static Image addressGradient;
    public static Image addressArrow;

    static {
        try {
            addressGradient = ImageIO.read(new File("res/address-gradient.png"));
            addressArrow = ImageIO.read(new File("res/address-arrow.png"));
        } catch (IOException e) {
            throw new RuntimeException(e);
        }
    }

    //private ArrayList<Node> parts = new List<Node>[]{"root", "poster", "background"};
    private int[] positions;
    private int armed = -1;
    private OnPartClickListener onPartClickListener;
    private Node activeNetwork;
    private JProgressBar progressBar;

    public AddressBar() {
        addMouseListener(this);
        setMinimumSize(new Dimension(0, 25));
        setMaximumSize(new Dimension(Integer.MAX_VALUE, 25));
        setPreferredSize(new Dimension(Integer.MAX_VALUE, 25));
        setLayout(null);
        progressBar = new JProgressBar();
        progressBar.putClientProperty("JProgressBar.style", "circular");
        progressBar.setIndeterminate(true);
        progressBar.setBorderPainted(false);
        progressBar.setVisible(false);
        add(progressBar);
    }

    public Node getActiveNetwork() {
        return activeNetwork;
    }

    public void setActiveNetwork(Node activeNetwork) {
        this.activeNetwork = activeNetwork;
        repaint();
    }

    /**
     * Returns the part-click callback registered for this address bar.
     *
     * @return The callback, or null if one is not registered.
     */
    public OnPartClickListener getOnPartClickListener() {
        return onPartClickListener;
    }

    /**
     * Register a callback to be invoked when a part was clicked in the address bar.
     *
     * @param l the callback that will run.
     */
    public void setOnPartClickListener(OnPartClickListener l) {
        onPartClickListener = l;
    }

    public boolean getProgressVisible() {
        return progressBar.isVisible();
    }

    public void setProgressVisible(boolean visible) {
        progressBar.setVisible(visible);
    }

    private java.util.List<Node> getNetworkParts() {
        ArrayList<Node> parts = new ArrayList<Node>();
        if (activeNetwork == null) return parts;
        Node currentNode = activeNetwork;
        parts.add(0, currentNode);
        while (currentNode.getParent() != null) {
            parts.add(0, currentNode.getParent());
            currentNode = currentNode.getParent();
        }
        return parts;
    }

    @Override
    protected void paintComponent(Graphics g) {
        java.util.List<Node> nodes = getNetworkParts();
        positions = new int[nodes.size()];
        Graphics2D g2 = (Graphics2D) g;

        g2.setFont(Theme.SMALL_BOLD_FONT);

        g2.drawImage(addressGradient, 0, 0, getWidth(), 25, null);

        int x = 10;

        for (int i = 0; i < nodes.size(); i++) {
            Node part = nodes.get(i);
            if (i == armed) {
                g2.setColor(Theme.TEXT_ARMED_COLOR);
            } else {
                g2.setColor(Theme.TEXT_NORMAL_COLOR);
            }
            SwingUtils.drawShadowText(g2, part.getName(), x, 16);

            int width = (int) g2.getFontMetrics().stringWidth(part.getName());
            x += width + 5;
            positions[i] = x + 10;
            g2.drawImage(addressArrow, x, 1, null);
            x += 15;
        }
    }

    public void mouseClicked(MouseEvent e) {
    }

    public void mousePressed(MouseEvent e) {
        int mx = e.getX();
        armed = partIndex(mx);
        repaint();
    }

    public void mouseReleased(MouseEvent e) {
        armed = -1;
        int mx = e.getX();
        int partIndex = partIndex(mx);
        if (partIndex == -1) return;
        java.util.List<Node> nodes = getNetworkParts();
        Node selectedNode = nodes.get(partIndex);
        if (selectedNode != null && onPartClickListener != null)
            onPartClickListener.onPartClicked(selectedNode);
        repaint();
    }

    public void mouseEntered(MouseEvent e) {
    }

    public void mouseExited(MouseEvent e) {
        armed = -1;
        repaint();
    }


    private int partIndex(int x) {
        if (positions == null) return -1;
        for (int i = 0; i < positions.length; i++) {
            if (x < positions[i])
                return i;
        }
        return -1;
    }

    @Override
    public void doLayout() {
        final int width = getWidth();
        progressBar.setBounds(width - 23, 3, 20, 20);
    }

    /**
     * Callback listener to be invoked when an address part has been clicked.
     */
    public static interface OnPartClickListener {

        /**
         * Called when a part has been clicked.
         *
         * @param n the part that was clicked.
         */
        public void onPartClicked(Node n);

    }

}