package org.iri_research.renkan.models;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.data.mongodb.core.mapping.DBRef;

import org.springframework.data.mongodb.core.mapping.Document;

import org.springframework.data.mongodb.core.mapping.Field;

import com.fasterxml.jackson.annotation.JsonIgnore;

import com.fasterxml.jackson.annotation.JsonProperty;

@Document(collection="edges")

public class Edge extends AbstractRenkanModel<String> {

	@DBRef

	private Node from;

	@DBRef

	private Node to;

	@Field("project_id")

	@JsonProperty("project_id")

	private String projectId;

	@Field("created_by")

	@JsonProperty("created_by")

	private String createdBy;

	@SuppressWarnings("unused")

	private Edge() {		

	}

	public Edge(Edge edge, Node from, Node to, String projectId) {

	}	

	public Edge(Edge edge) {

		this(edge, edge.from, edge.to, edge.projectId);

	}	

	@Autowired(required=true)

	public Edge(String id, String title, String description, String uri, String color, Node from, Node to, String createdBy, String projectId) {

		super(id,title, description, uri, color);

		this.from = from;

		this.to = to;

		this.createdBy = createdBy;

		this.projectId = projectId;

	}

	@JsonProperty("project_id")

	public String getProjectId() {

		return projectId;

	}

	public String getCreatedBy() {

		return createdBy;

	}

	@JsonIgnore

	public Node getFromNode() {

		return this.from;

	}

	public String getFrom() {

		if(this.from != null) {

			return this.from.id;

		}

		else {

			return null;

		}

	}

	@JsonIgnore

	public Node getToNode() {

		return this.to;

	}

	public String getTo() {

		if(this.to != null) {

			return this.to.id;

		}

		else {

			return null;

		}

	}

}