package net.sourceforge.tuned.ui;


import java.awt.Color;
import java.awt.Component;
import java.awt.Dimension;
import java.awt.Font;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.awt.Insets;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;
import java.awt.geom.GeneralPath;
import java.awt.geom.Path2D;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;

import javax.swing.Icon;
import javax.swing.JButton;
import javax.swing.JMenuItem;
import javax.swing.JPopupMenu;
import javax.swing.SwingConstants;


public class SelectButton<T> extends JButton implements ActionListener {
	
	private boolean hover = false;
	
	private Color beginColor = new Color(0xF0EEE4);
	private Color endColor = new Color(0xE0DED4);
	
	private Color beginColorHover = beginColor;
	private Color endColorHover = new Color(0xD8D7CD);
	
	private Entry<T> selectedEntry = null;
	
	private List<Entry<T>> entries = new ArrayList<Entry<T>>();
	
	public static final String SELECTED_VALUE_PROPERTY = "SELECTED_VALUE_PROPERTY";
	
	
	public SelectButton(Collection<Entry<T>> entries) {
		if (entries.isEmpty())
			throw new IllegalArgumentException("Entries must not be empty");
		
		this.entries.addAll(entries);
		
		setContentAreaFilled(false);
		setFocusable(false);
		
		addActionListener(this);
		
		setHorizontalAlignment(SwingConstants.CENTER);
		setVerticalAlignment(SwingConstants.CENTER);
		
		addMouseListener(new MouseInputListener());
		
		setPreferredSize(new Dimension(32, 22));
		
		// select first entry
		setSelectedIndex(0);
	}
	

	public void setSelectedValue(T value) {
		Entry<T> entry = find(value);
		
		if (entry == null)
			return;
		
		selectedEntry = entry;
		setIcon(new SelectIcon(selectedEntry.getIcon()));
		
		firePropertyChange(SELECTED_VALUE_PROPERTY, null, selectedEntry);
	}
	

	public T getSelectedEntry() {
		return selectedEntry.getValue();
	}
	

	public void setSelectedIndex(int i) {
		setSelectedValue(entries.get(i).getValue());
	}
	

	public int getSelectedIndex() {
		return entries.indexOf(selectedEntry);
	}
	

	private Entry<T> find(T value) {
		for (Entry<T> entry : entries) {
			if (entry.value == value)
				return entry;
		}
		
		return null;
	}
	

	public void spinValue(int spin) {
		spin = spin % entries.size();
		
		int next = getSelectedIndex() + spin;
		
		if (next < 0)
			next += entries.size();
		else if (next >= entries.size())
			next -= entries.size();
		
		setSelectedIndex(next);
	}
	

	public void actionPerformed(ActionEvent e) {
		JPopupMenu popup = new JPopupMenu();
		
		for (Entry<T> entry : entries) {
			popup.add(new SelectMenuItem(entry));
		}
		
		popup.show(this, 0, getHeight() - 1);
	}
	

	@Override
	protected void paintComponent(Graphics g) {
		Graphics2D g2d = (Graphics2D) g;
		
		if (hover)
			g2d.setPaint(GradientStyle.TOP_TO_BOTTOM.getGradientPaint(getBounds(), beginColorHover, endColorHover));
		else
			g2d.setPaint(GradientStyle.TOP_TO_BOTTOM.getGradientPaint(getBounds(), beginColor, endColor));
		
		g2d.fill(getBounds());
		
		super.paintComponent(g);
	}
	
	
	private class SelectMenuItem extends JMenuItem implements ActionListener {
		
		private T value;
		
		
		public SelectMenuItem(Entry<T> entry) {
			super(entry.toString(), entry.getIcon());
			this.value = entry.getValue();
			
			this.setMargin(new Insets(3, 0, 3, 0));
			this.addActionListener(this);
			
			if (this.value == getSelectedEntry())
				this.setFont(this.getFont().deriveFont(Font.BOLD));
		}
		

		@Override
		public void actionPerformed(ActionEvent e) {
			setSelectedValue(value);
		}
		
	}
	

	private static class SelectIcon implements Icon {
		
		private Icon icon;
		private GeneralPath arrow;
		
		
		public SelectIcon(Icon icon) {
			this.icon = icon;
			
			arrow = new GeneralPath(Path2D.WIND_EVEN_ODD, 3);
			int x = 25;
			int y = 10;
			
			arrow.moveTo(x - 2, y);
			arrow.lineTo(x, y + 3);
			arrow.lineTo(x + 3, y);
			arrow.lineTo(x - 2, y);
		}
		

		public synchronized void paintIcon(Component c, Graphics g, int x, int y) {
			Graphics2D g2d = (Graphics2D) g;
			
			icon.paintIcon(c, g2d, 4, 3);
			
			g2d.setPaint(Color.BLACK);
			g2d.fill(arrow);
		}
		

		public int getIconWidth() {
			return 30;
		}
		

		public int getIconHeight() {
			return 20;
		}
	}
	

	private class MouseInputListener extends MouseAdapter {
		
		@Override
		public void mouseEntered(MouseEvent e) {
			hover = true;
			repaint();
		}
		

		@Override
		public void mouseExited(MouseEvent e) {
			hover = false;
			repaint();
		}
		
	}
	

	public static class Entry<T> {
		
		private T value;
		private Icon icon;
		
		
		public Entry(T value, Icon icon) {
			this.value = value;
			this.icon = icon;
		}
		

		public T getValue() {
			return value;
		}
		

		public Icon getIcon() {
			return icon;
		}
		

		@Override
		public String toString() {
			return value.toString();
		}
	}
	
}