de.symeda.sormas.ui.caze.CaseGrid.java Source code

Introduction

Here is the source code for de.symeda.sormas.ui.caze.CaseGrid.java

Source

/*******************************************************************************
 * SORMAS - Surveillance Outbreak Response Management & Analysis System
 * Copyright  2016-2018 Helmholtz-Zentrum fr Infektionsforschung GmbH (HZI)
 *
 * 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 <https://www.gnu.org/licenses/>.
 *******************************************************************************/
package de.symeda.sormas.ui.caze;

import java.util.Date;
import java.util.stream.Collectors;

import com.vaadin.data.provider.DataProvider;
import com.vaadin.shared.data.sort.SortDirection;
import com.vaadin.ui.renderers.DateRenderer;

import de.symeda.sormas.api.DiseaseHelper;
import de.symeda.sormas.api.FacadeProvider;
import de.symeda.sormas.api.caze.CaseCriteria;
import de.symeda.sormas.api.caze.CaseIndexDto;
import de.symeda.sormas.api.caze.CaseOrigin;
import de.symeda.sormas.api.i18n.Captions;
import de.symeda.sormas.api.i18n.I18nProperties;
import de.symeda.sormas.api.user.UserRight;
import de.symeda.sormas.api.user.UserRole;
import de.symeda.sormas.api.utils.DateHelper;
import de.symeda.sormas.api.utils.SortProperty;
import de.symeda.sormas.ui.ControllerProvider;
import de.symeda.sormas.ui.UserProvider;
import de.symeda.sormas.ui.utils.FilteredGrid;
import de.symeda.sormas.ui.utils.UuidRenderer;

@SuppressWarnings("serial")
public class CaseGrid extends FilteredGrid<CaseIndexDto, CaseCriteria> {

    public static final String DISEASE_SHORT = Captions.columnDiseaseShort;

    @SuppressWarnings("unchecked")
    public CaseGrid() {
        super(CaseIndexDto.class);
        setSizeFull();

        if (UserProvider.getCurrent().hasUserRight(UserRight.PERFORM_BULK_OPERATIONS)) {
            setSelectionMode(SelectionMode.MULTI);
        } else {
            setSelectionMode(SelectionMode.NONE);
        }

        DataProvider<CaseIndexDto, CaseCriteria> dataProvider = DataProvider
                .fromFilteringCallbacks(query -> FacadeProvider.getCaseFacade()
                        .getIndexList(UserProvider.getCurrent().getUuid(), query.getFilter().orElse(null),
                                query.getOffset(), query.getLimit(),
                                query.getSortOrders().stream()
                                        .map(sortOrder -> new SortProperty(sortOrder.getSorted(),
                                                sortOrder.getDirection() == SortDirection.ASCENDING))
                                        .collect(Collectors.toList()))
                        .stream(), query -> {
                            return (int) FacadeProvider.getCaseFacade().count(UserProvider.getCurrent().getUuid(),
                                    query.getFilter().orElse(null));
                        });
        setDataProvider(dataProvider);

        Column<CaseIndexDto, String> diseaseShortColumn = addColumn(
                caze -> DiseaseHelper.toString(caze.getDisease(), caze.getDiseaseDetails()));
        diseaseShortColumn.setId(DISEASE_SHORT);
        diseaseShortColumn.setSortProperty(CaseIndexDto.DISEASE);

        setColumns(CaseIndexDto.UUID, CaseIndexDto.EPID_NUMBER, DISEASE_SHORT, CaseIndexDto.CASE_CLASSIFICATION,
                CaseIndexDto.OUTCOME, CaseIndexDto.INVESTIGATION_STATUS, CaseIndexDto.PERSON_FIRST_NAME,
                CaseIndexDto.PERSON_LAST_NAME, CaseIndexDto.DISTRICT_NAME, CaseIndexDto.HEALTH_FACILITY_NAME,
                CaseIndexDto.POINT_OF_ENTRY_NAME, CaseIndexDto.REPORT_DATE, CaseIndexDto.CREATION_DATE);

        ((Column<CaseIndexDto, String>) getColumn(CaseIndexDto.UUID)).setRenderer(new UuidRenderer());
        ((Column<CaseIndexDto, Date>) getColumn(CaseIndexDto.REPORT_DATE))
                .setRenderer(new DateRenderer(DateHelper.getLocalDateTimeFormat()));

        if (UserProvider.getCurrent().hasUserRight(UserRight.CASE_IMPORT)) {
            ((Column<CaseIndexDto, Date>) getColumn(CaseIndexDto.CREATION_DATE))
                    .setRenderer(new DateRenderer(DateHelper.getLocalDateTimeFormat()));
        } else {
            removeColumn(CaseIndexDto.CREATION_DATE);
        }

        for (Column<?, ?> column : getColumns()) {
            column.setCaption(I18nProperties.getPrefixCaption(CaseIndexDto.I18N_PREFIX, column.getId().toString(),
                    column.getCaption()));
        }

        addItemClickListener(e -> {
            if ((e.getColumn() != null && CaseIndexDto.UUID.equals(e.getColumn().getId()))
                    || e.getMouseEventDetails().isDoubleClick()) {
                ControllerProvider.getCaseController().navigateToCase(e.getItem().getUuid());
            }
        });
    }

    public void reload() {
        if (getSelectionModel().isUserSelectionAllowed()) {
            deselectAll();
        }

        if (getCriteria().getOutcome() == null) {
            this.getColumn(CaseIndexDto.OUTCOME).setHidden(false);
        } else if (this.getColumn(CaseIndexDto.OUTCOME) != null) {
            this.getColumn(CaseIndexDto.OUTCOME).setHidden(true);
        }

        if (UserRole.isPortHealthUser(UserProvider.getCurrent().getUserRoles())) {
            removeColumn(CaseIndexDto.HEALTH_FACILITY_NAME);
        } else {
            if (getCriteria().getCaseOrigin() == CaseOrigin.IN_COUNTRY) {
                removeColumn(CaseIndexDto.POINT_OF_ENTRY_NAME);
            } else if (getCriteria().getCaseOrigin() == CaseOrigin.POINT_OF_ENTRY) {
                removeColumn(CaseIndexDto.HEALTH_FACILITY_NAME);
            }
        }

        getDataProvider().refreshAll();
    }

}