package org.activityinfo.server.database.hibernate.entity;

/*
 * #%L
 * ActivityInfo Server
 * %%
 * Copyright (C) 2009 - 2013 UNICEF
 * %%
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as
 * published by the Free Software Foundation, either version 3 of the 
 * License, or (at your option) any later version.
 * 
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public 
 * License along with this program.  If not, see
 * <http://www.gnu.org/licenses/gpl-3.0.html>.
 * #L%
 */

import com.bedatadriven.rebar.time.calendar.LocalDate;

import javax.persistence.*;
import java.io.Serializable;
import java.util.Date;

@Entity
public class LockedPeriod implements Serializable, ReallyDeleteable {

    private Date fromDate;
    private Date toDate;
    private String name;
    private int id;
    private UserDatabase userDatabase;
    private Project project;
    private Activity activity;
    private boolean enabled;

    public LockedPeriod() {
    }

    public LockedPeriod(LockedPeriod lockedPeriod) {
        this.fromDate = lockedPeriod.fromDate;
        this.toDate = lockedPeriod.toDate;
        this.name = lockedPeriod.name;
        this.userDatabase = lockedPeriod.userDatabase;
        this.project = lockedPeriod.project;
        this.activity = lockedPeriod.activity;
        this.enabled = lockedPeriod.enabled;
    }

    @Column(nullable = false) @Temporal(TemporalType.DATE)
    public Date getFromDate() {
        return fromDate;
    }

    public void setFromDate(Date fromDate) {
        this.fromDate = fromDate;
    }

    public void setFromDate(LocalDate fromDate) {
        this.fromDate = fromDate.atMidnightInMyTimezone();
    }

    @Column(nullable = false) @Temporal(TemporalType.DATE)
    public Date getToDate() {
        return toDate;
    }

    public void setToDate(Date toDate) {
        this.toDate = toDate;
    }

    public void setToDate(LocalDate toDate) {
        this.toDate = toDate.atMidnightInMyTimezone();
    }

    @Column(nullable = false)
    public String getName() {
        return name;
    }

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

    @Id @GeneratedValue(strategy = GenerationType.AUTO)
    @Column(name = "LockedPeriodId", unique = true, nullable = false)
    public int getId() {
        return id;
    }

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

    @ManyToOne(fetch = FetchType.EAGER) @JoinColumn(name = "UserDatabaseId", nullable = true)
    public UserDatabase getUserDatabase() {
        return userDatabase;
    }

    public void setUserDatabase(UserDatabase userDatabase) {
        this.userDatabase = userDatabase;
    }

    public void setProject(Project project) {
        this.project = project;
    }

    @ManyToOne(fetch = FetchType.EAGER) @JoinColumn(name = "ProjectId", nullable = true)
    public Project getProject() {
        return project;
    }

    public void setActivity(Activity activity) {
        this.activity = activity;
    }

    @ManyToOne(fetch = FetchType.EAGER) @JoinColumn(name = "ActivityId", nullable = true)
    public Activity getActivity() {
        return activity;
    }

    public void setEnabled(boolean isEnabled) {
        this.enabled = isEnabled;
    }

    public boolean isEnabled() {
        return enabled;
    }

    @Transient
    public UserDatabase getParentDatabase() {
        if (userDatabase != null) {
            return userDatabase;
        } else if (activity != null) {
            return activity.getDatabase();
        } else if (project != null) {
            return project.getUserDatabase();
        }

        return null;
    }

    @Override
    public void deleteReferences() {
        if (activity != null) {
            activity.getLockedPeriods().remove(this);
        }
        if (userDatabase != null) {
            userDatabase.getLockedPeriods().remove(this);
        }
        if (project != null) {
            project.getLockedPeriods().remove(this);
        }
        this.activity = null;
        this.userDatabase = null;
        this.project = null;
    }

}