package view;

import model.UmlRefType;
import model.UmlMethod;
import model.UmlAttribute;
import model.UmlEntity;
import model.UmlEnum;
import model.UmlInterface;

import javax.swing.JPanel;
import javax.swing.JPopupMenu;

import controller.AttributeEditorController;
import controller.ClassEditorController;
import controller.MethodEditorController;

import javax.swing.BorderFactory;
import javax.swing.BoxLayout;
import javax.swing.JButton;
import javax.swing.ImageIcon;
import javax.swing.JLabel;
import javax.swing.JMenuItem;

import java.awt.BorderLayout;
import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.util.HashMap;
import java.util.Observable;
import java.util.Observer;

/**
 * This class is the visual representation of an UML Object (e.g. a class, an interface...).<br>
 * It is a JPanel with added functionality to automatically update it's content to match
 * a given UmlComponent.
 *
 * <b>ATTENTION:</b> while this object is a JPanel and have all JPanel methods, it is 
 * discouraged to use them.
 */
public class UMLObjectDisplay extends JPanel implements Observer {
	
	/**
	 * Generated serial ID
	 */
	private static final long serialVersionUID = 8971524820842081306L;

	private static final HashMap<Class<? extends UmlRefType>,String> stereotypeMap;
	
	static {
		stereotypeMap = new HashMap<>();
		//stereotypeMap.put(UmlInterface.class, "interface");
		stereotypeMap.put(UmlEnum.class, "enum");
		stereotypeMap.put(UmlInterface.class, "interface");
	}
	
	/** The inner label that display the name*/
	private JLabel classname;
	
	/** the container of the attribute displays */
	private JPanel attributeContainer;

	/** the container of the function/method displays */
	private JPanel functionContainer;
	
	/**
	 * Construct a new UMLObjectDisplay backed by an UmlComponent and 
	 * potentially display an archetype.
	 * @param umlobject
	 * 		The (non null) object to visually represent
	 */
	public UMLObjectDisplay(UmlRefType umlobject) {
		super();
		
		if (umlobject == null) throw new IllegalArgumentException("uml object can't be null");
		
		//creating object attributes
		umlobject.addObserver(this);
		
		//Creating Swing architecture
		this.buildInnerSwingArchitecture(umlobject);
		
		this.classname.addMouseListener(new ClassEditorController(umlobject));
		this.updateDisplay(umlobject);
	}

	/**
	 * Generate the inner SWING Architecture of this JPanel to properly display
	 * an UML Object.
	 */
	private void buildInnerSwingArchitecture(UmlRefType umlobject) {
		this.setLayout(new BorderLayout());
		this.setBorder(Uml7JFrame.objectBorder);
		this.setBackground(Uml7JFrame.objectBackgroundColor);
		
		this.classname = new JLabel();
		this.classname.setBackground(Uml7JFrame.objectBackgroundColor);
		
		
		JLabel addElement = new JLabel(
				new ImageIcon(
						this.getClass().getClassLoader().getResource("resource/plus.png")
						)
				);
		
		
		addElement.addMouseListener(new MouseAdapter() {

			public void mouseClicked(MouseEvent arg0) {
				// Show popup menu
				JPopupMenu popup = new JPopupMenu(); 
		        JMenuItem methodItem = new JMenuItem("Add method");
				methodItem.addActionListener(new ActionListener() {

					@Override
					public void actionPerformed(ActionEvent arg0) {
						new MethodEditorController(umlobject);
					}
					
				});
		        JMenuItem attributeItem = new JMenuItem("Add attribute");
		        attributeItem.addActionListener(new ActionListener() {

					@Override
					public void actionPerformed(ActionEvent arg0) {
						new AttributeEditorController(umlobject);
					}
		        	
		        });
				
				popup.add(methodItem);
				popup.add(attributeItem);
				popup.show(arg0.getComponent(), arg0.getX(), arg0.getY());
				
			}

		});
		
		classname.setHorizontalAlignment(JLabel.CENTER);
		
		//Creating title area
		JPanel titleArea = new JPanel(new BorderLayout());
		titleArea.setBorder(Uml7JFrame.objectInnerSeparation);
		titleArea.add(this.classname,BorderLayout.CENTER);
		titleArea.add(addElement, BorderLayout.EAST);
		titleArea.setBackground(Uml7JFrame.objectBackgroundColor);
		
		String stereotype = stereotypeMap.get(umlobject.getClass());
		if (stereotype != null) {
			titleArea.add(new JLabel("<<"+stereotype+">>"),BorderLayout.NORTH);
		}
		
		this.add(titleArea, BorderLayout.NORTH);
		
		//Content has 2 categories : attributes and methods/functions
		JPanel listsContainer = new JPanel();
		listsContainer.setLayout(new BoxLayout(listsContainer,BoxLayout.Y_AXIS));
		listsContainer.setBorder(Uml7JFrame.objectInnerSeparation);
		listsContainer.setBackground(Uml7JFrame.objectBackgroundColor);
		
		this.attributeContainer = new JPanel();
		attributeContainer.setLayout(new BoxLayout(attributeContainer,BoxLayout.Y_AXIS));
		attributeContainer.setBorder(Uml7JFrame.objectInnerSeparation);
		attributeContainer.setBackground(Uml7JFrame.objectBackgroundColor);
		
		this.functionContainer = new JPanel();
		functionContainer.setLayout(new BoxLayout(functionContainer,BoxLayout.Y_AXIS));
		functionContainer.setBorder(Uml7JFrame.objectInnerSeparation);
		functionContainer.setBackground(Uml7JFrame.objectBackgroundColor);
		
		listsContainer.add(attributeContainer);
		listsContainer.add(functionContainer);
		
		this.add(listsContainer, BorderLayout.CENTER);
	}

	@Override
	public void update(Observable o, Object arg) {
		if (o instanceof UmlRefType) {
			this.updateDisplay((UmlRefType) o);
		}
	}
	
	private void updateDisplay(UmlRefType uc) {
		this.classname.setText(uc.getName());
		this.attributeContainer.removeAll();
		for (UmlAttribute a : uc.getAttributesList()) {
			this.attributeContainer.add(new SubComponentDeleteWrapper(a,uc,false));
		}
		this.attributeContainer.repaint();
		this.attributeContainer.revalidate();
		
		this.functionContainer.removeAll();
		for (UmlMethod m : uc.getMethodsList()) {
			this.functionContainer.add(new SubComponentDeleteWrapper(m,uc,true));
		}
		this.functionContainer.repaint();
		this.functionContainer.revalidate();
		this.revalidate();
	}
	
	private class SubComponentDeleteWrapper extends JPanel{
		
		/**
		 * Generated serial ID
		 */
		private static final long serialVersionUID = 1776459837985581481L;
		
		private JButton deleteButton;
		
		SubComponentDeleteWrapper(UmlEntity umlent, UmlRefType umlrt, boolean isMethod) {
			super(new BorderLayout());
			this.setBackground(Uml7JFrame.objectBackgroundColor);
			if (isMethod) {
				this.add(new MethodDisplay((UmlMethod)umlent), BorderLayout.CENTER);
			} else {
				this.add(new AttributeDisplay((UmlAttribute)umlent), BorderLayout.CENTER);
			}
			deleteButton = new JButton(" - ");
			deleteButton.setBorder(Uml7JFrame.deleteButtonBorder);
			this.add(deleteButton, BorderLayout.EAST);
			deleteButton.addActionListener(e -> {
				if (isMethod) {
					umlrt.removeMethod((UmlMethod)umlent);
				} else {
					umlrt.removeAttribute((UmlAttribute)umlent);
				}
			});
			this.setBorder(BorderFactory.createEmptyBorder(2, 2, 2, 2));
		}
		
	}
}