package code.ast;

import java.util.List;
import java.util.Map;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;

import models.algebra.Type;

public class MethodDeclaration extends BodyDeclaration implements IAnnotatable {
	private String name = null;
	private boolean isConstructor = false;
	private Type returnType = null;
	private List<VariableDeclaration> parameters = null;
	private Block body = null;
	private Map<String, Annotation> annotations = new HashMap<>();
	private Throws thrws = null;
	
	public MethodDeclaration(String methodName) {
		this(methodName, false);
	}
	
	public MethodDeclaration(String methodName, Type returnType) {
		this(methodName, false);
		this.returnType = returnType;
	}
	
	public MethodDeclaration(String methodName, boolean isConstructor) {
		this.name = methodName;
		this.isConstructor = isConstructor;
	}
	
	public MethodDeclaration(String methodName, boolean isConstructor, Type returnType, List<VariableDeclaration> parameters) {
		this(methodName, isConstructor, returnType, parameters, null);
	}
	
	public MethodDeclaration(String methodName, boolean isConstructor, Type returnType, List<VariableDeclaration> parameters, Block body) {
		this(methodName, isConstructor);
		this.returnType = returnType;
		this.parameters = parameters;
		this.body = body;
	}

	public String getName() {
		return name;
	}

	public void setName(String name) {
		this.name = name;
	}

	public boolean isConstructor() {
		return isConstructor;
	}

	public void setConstructor(boolean isConstructor) {
		this.isConstructor = isConstructor;
	}

	public Type getReturnType() {
		return returnType;
	}

	public void setReturnType(Type returnType) {
		this.returnType = returnType;
	}

	public List<VariableDeclaration> getParameters() {
		return parameters;
	}

	public void setParameters(List<VariableDeclaration> parameters) {
		this.parameters = parameters;
	}
	
	public void addParameter(VariableDeclaration parameter) {
		if (parameters == null) {
			parameters = new ArrayList<>();
		}
		parameters.add(parameter);
	}

	public Block getBody() {
		return body;
	}

	public void setBody(Block body) {
		this.body = body;
	}
	
	public void addStatement(String statement) {
		if (body == null) {
			body = new Block();
		}
		body.addStatement(statement);
	}
	
	public void addFirstStatement(String statement) {
		if (body == null) {
			body = new Block();
		}
		body.addFirstStatement(statement);
	}
	
	public void addThrow(String exception) {
		if (thrws == null) {
			thrws = new Throws();
		}
		thrws.addException(exception);
	}
	
	public Throws getThrows() {
		return thrws;
	}	

	@Override
	public Annotation getAnnotation(String name) {
		return annotations.get(name);
	}

	@Override
	public Collection<Annotation> getAnnotations() {
		return annotations.values();
	}

	@Override
	public void addAnnotation(Annotation annotation) {
		annotations.put(annotation.getElementName(), annotation);
	}
	
	public String toString() {
		String code = "";
		for (Annotation annotation: getAnnotations()) {
			code += annotation.toString() + "\n";
		}
		code += "public ";
		if (returnType == null) {
			if(!isConstructor) code += "void ";
		}else {
			code += returnType.getInterfaceTypeName() + " ";
		}
		code += (name + "(");
		if (parameters != null) {
			for (int i = 0; i < parameters.size(); i++) {
				code += parameters.get(i).toString();
				if (i < parameters.size() - 1) code += ", ";
			}
		}
		code += ") ";

		if (body != null) {
			code += body.toString();
		} else {
			code += ";";
		}
		return code;
	}
}