FormSummaryViewModel.java

package edu.rit.coopeval.viewmodel.form;

import edu.rit.coopeval.model.DepartmentTermForm;
import edu.rit.coopeval.model.Form;
import edu.rit.coopeval.form.FormType;

import java.util.ArrayList;
import java.util.List;

/**
 * Created by mhickson on 4/7/15.
 */
public class FormSummaryViewModel {
    private Long formId;
    private String name;
    private List<String> departmentCodes;
    private String isStudentForm;

    public FormSummaryViewModel(){}

    public FormSummaryViewModel(Form formEntity, FormType formType){
        this.formId = formEntity.getFormId();
        this.name = formEntity.getName();
        this.isStudentForm = formEntity.getIsStudentForm();
        this.departmentCodes = new ArrayList<String>();
        switch (formType) {
            case STUDENT:
                if(formEntity.getDepartmentTermFormsByFormByStudentFormId() != null){
                    for(DepartmentTermForm departmentTermForm: formEntity.getDepartmentTermFormsByFormByStudentFormId()){
                        departmentCodes.add(departmentTermForm.getDepartment().getDepartmentCode());
                    }
                }
                break;
            case EMPLOYER:
                if(formEntity.getDepartmentTermFormsByFormByEmployerFormId() != null) {
                    for (DepartmentTermForm departmentTermForm : formEntity.getDepartmentTermFormsByFormByEmployerFormId()) {
                        departmentCodes.add(departmentTermForm.getDepartment().getDepartmentCode());
                    }
                }
                break;
        }
    }

    public Long getFormId() {
        return formId;
    }

    public void setFormId(Long formId) {
        this.formId = formId;
    }

    public List<String> getDepartmentCodes() {
        return departmentCodes;
    }

    public void setDepartmentCodes(List<String> departmentCodes) {
        this.departmentCodes = departmentCodes;
    }

    public String getName() {
        return name;
    }

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

    public String getIsStudentForm() {
        return isStudentForm;
    }

    public void setIsStudentForm(String isStudentForm) {
        this.isStudentForm = isStudentForm;
    }
}