package eu.eudat.entities;


import org.hibernate.annotations.GenericGenerator;
import org.hibernate.annotations.Type;

import javax.persistence.*;
import java.util.*;

@Entity
@Table(name = "\"Dataset\"")
@NamedEntityGraphs({
        @NamedEntityGraph(
                name = "datasetListingModel",
                attributeNodes = {@NamedAttributeNode("services"), @NamedAttributeNode("dataRepositories"), @NamedAttributeNode("externalDatasets"), @NamedAttributeNode("registries"),
                        @NamedAttributeNode(value = "dmp", subgraph = "dmp"), @NamedAttributeNode("profile"), @NamedAttributeNode("creator")},
                subgraphs = @NamedSubgraph(name = "dmp", attributeNodes = {@NamedAttributeNode("creator"), @NamedAttributeNode("users")})),
        @NamedEntityGraph(
                name = "datasetWizardModel",
                attributeNodes = {@NamedAttributeNode("services"), @NamedAttributeNode("dataRepositories"), @NamedAttributeNode("externalDatasets"), @NamedAttributeNode("registries"),
                        @NamedAttributeNode("dmp"), @NamedAttributeNode("profile"), @NamedAttributeNode("creator")})
})
public class Dataset implements DataEntity<Dataset,UUID> {

    public static Set<String> getHints() {
        return hints;
    }

    private static final Set<String> hints = new HashSet<>(Arrays.asList("datasetListingModel"));

    public enum Status {
        SAVED((short) 0), FINALISED((short) 1), DELETED((short) 99);

        private short value;

        private Status(short value) {
            this.value = value;
        }

        public short getValue() {
            return value;
        }

        public static Dataset.Status fromInteger(int value) {
            switch (value) {
                case 0:
                    return SAVED;
                case 1:
                    return FINALISED;
                default:
                    throw new RuntimeException("Unsupported Project Status");
            }
        }
    }

    @Id
    @GeneratedValue
    @GenericGenerator(name = "uuid2", strategy = "uuid2")
    @Column(name = "\"ID\"", updatable = false, nullable = false, columnDefinition = "BINARY(16)")
    private UUID id;

    @Column(name = "\"Label\"")
    private String label;


    @ManyToOne(fetch = FetchType.LAZY)
//	@Cascade(value=org.hibernate.annotations.CascadeType.ALL)
    @JoinColumn(name = "\"DMP\"", nullable = true)
    private DMP dmp;


    @Column(name = "\"Uri\"")
    private String uri;

    @Type(type = "eu.eudat.typedefinition.XMLType")
    @Column(name = "\"Properties\"", columnDefinition = "xml", nullable = true)
    private String properties;


    @ManyToOne(fetch = FetchType.LAZY)
    //@Cascade(value=org.hibernate.annotations.CascadeType.ALL)
    @JoinColumn(name = "\"Profile\"", nullable = true)
    private DatasetProfile profile;

    @Type(type = "eu.eudat.typedefinition.XMLType")
    @Column(name = "\"Reference\"", columnDefinition = "xml", nullable = true)
    private String reference;

    @OneToMany(fetch = FetchType.LAZY)
    @JoinTable(name = "\"DatasetDataRepository\"",
            joinColumns = {@JoinColumn(name = "\"Dataset\"", referencedColumnName = "\"ID\"")},
            inverseJoinColumns = {@JoinColumn(name = "\"DataRepository\"", referencedColumnName = "\"ID\"")}
    )
    private Set<DataRepository> dataRepositories;


    @OneToMany(fetch = FetchType.LAZY)
    @JoinTable(name = "\"DatasetRegistry\"",
            joinColumns = {@JoinColumn(name = "\"Dataset\"", referencedColumnName = "\"ID\"")},
            inverseJoinColumns = {@JoinColumn(name = "\"Registry\"", referencedColumnName = "\"ID\"")}
    )
    private Set<Registry> registries;


    @OneToMany(fetch = FetchType.LAZY)
    @JoinTable(name = "\"DatasetService\"",
            joinColumns = {@JoinColumn(name = "\"Dataset\"", referencedColumnName = "\"ID\"")},
            inverseJoinColumns = {@JoinColumn(name = "\"Service\"", referencedColumnName = "\"ID\"")}
    )
    private Set<Service> services;

    @OneToMany(fetch = FetchType.LAZY)
    @JoinTable(name = "\"DatasetExternalDataset\"",
            joinColumns = {@JoinColumn(name = "\"Dataset\"", referencedColumnName = "\"ID\"")},
            inverseJoinColumns = {@JoinColumn(name = "\"ExternalDataset\"", referencedColumnName = "\"Id\"")}
    )
    private Set<ExternalDataset> externalDatasets;


    @Column(name = "\"Status\"", nullable = false)
    private Short status;

    @Column(name = "\"IsPublic\"", nullable = false)
    private boolean isPublic;

    @Column(name = "\"Created\"")
    private Date created = null;

    @Column(name = "\"Modified\"")
    private Date modified = new Date();

    @ManyToOne(fetch = FetchType.LAZY)
    @JoinColumn(name = "\"Creator\"", nullable = true)
    private UserInfo creator;

    @Column(name = "\"Description\"")
    private String description;


    public String getDescription() {
        return description;
    }

    public void setDescription(String description) {
        this.description = description;
    }


    public UserInfo getCreator() {
        return creator;
    }

    public void setCreator(UserInfo creator) {
        this.creator = creator;
    }

    public Short getStatus() {
        return status;
    }


    public void setStatus(Short status) {
        this.status = status;
    }


    public Date getCreated() {
        return created;
    }


    public void setCreated(Date created) {
        this.created = created;
    }


    public Date getModified() {
        return modified;
    }


    public void setModified(Date modified) {
        this.modified = modified;
    }


    public Set<Registry> getRegistries() {
        return registries;
    }


    public void setRegistries(Set<Registry> registries) {
        this.registries = registries;
    }


    public Set<Service> getServices() {
        return services;
    }


    public void setServices(Set<Service> services) {
        this.services = services;
    }


    public UUID getId() {
        return id;
    }


    public void setId(UUID id) {
        this.id = id;
    }


    public String getLabel() {
        return label;
    }


    public void setLabel(String label) {
        this.label = label;
    }


    public DMP getDmp() {
        return dmp;
    }


    public void setDmp(DMP dmp) {
        this.dmp = dmp;
    }


    public String getUri() {
        return uri;
    }


    public void setUri(String uri) {
        this.uri = uri;
    }


    public String getProperties() {
        return properties;
    }


    public void setProperties(String properties) {
        this.properties = properties;
    }


    public DatasetProfile getProfile() {
        return profile;
    }


    public void setProfile(DatasetProfile profile) {
        this.profile = profile;
    }


    public Set<DataRepository> getDataRepositories() {
        return dataRepositories;
    }


    public void setDataRepositories(Set<DataRepository> dataRepositories) {
        this.dataRepositories = dataRepositories;
    }


    public String getReference() {
        return reference;
    }


    public void setReference(String reference) {
        this.reference = reference;
    }

    public Set<ExternalDataset> getExternalDatasets() {
        return externalDatasets;
    }

    public void setExternalDatasets(Set<ExternalDataset> externalDatasets) {
        this.externalDatasets = externalDatasets;
    }

    public boolean isPublic() {
        return isPublic;
    }

    public void setPublic(boolean aPublic) {
        isPublic = aPublic;
    }

    @Override
    public void update(Dataset entity) {
        this.setRegistries(entity.getRegistries());
        this.setDataRepositories(entity.getDataRepositories());
        this.setDescription(entity.getDescription());
        this.setLabel(entity.getLabel());
        this.setProperties(entity.getProperties());
        this.setExternalDatasets(entity.getExternalDatasets());
        this.setStatus(entity.getStatus());
        this.setProfile(entity.getProfile());
        if (entity.getCreator() != null) this.creator = entity.getCreator();
    }

    @Override
    public UUID getKeys() {
        return this.id;
    }
}