org.squashtest.tm.web.internal.controller.milestone.MilestoneController.java Source code

Introduction

Here is the source code for org.squashtest.tm.web.internal.controller.milestone.MilestoneController.java

Source

/**
 *     This file is part of the Squashtest platform.
 *     Copyright (C) 2010 - 2016 Henix, henix.fr
 *
 *     See the NOTICE file distributed with this work for additional
 *     information regarding copyright ownership.
 *
 *     This is free software: you can redistribute it and/or modify
 *     it under the terms of the GNU Lesser General Public License as published by
 *     the Free Software Foundation, either version 3 of the License, or
 *     (at your option) any later version.
 *
 *     this software 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 Lesser General Public License for more details.
 *
 *     You should have received a copy of the GNU Lesser General Public License
 *     along with this software.  If not, see <http://www.gnu.org/licenses/>.
 */
package org.squashtest.tm.web.internal.controller.milestone;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.Comparator;
import java.util.HashSet;
import java.util.List;
import java.util.Locale;
import java.util.Set;

import javax.inject.Inject;
import javax.inject.Provider;

import org.apache.commons.collections.CollectionUtils;
import org.springframework.context.i18n.LocaleContextHolder;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestMethod;
import org.springframework.web.bind.annotation.ResponseBody;
import org.squashtest.tm.core.foundation.collection.PagedCollectionHolder;
import org.squashtest.tm.core.foundation.collection.SinglePageCollectionHolder;
import org.squashtest.tm.domain.milestone.Milestone;
import org.squashtest.tm.domain.project.Project;
import org.squashtest.tm.domain.projectfilter.ProjectFilter;
import org.squashtest.tm.service.milestone.MilestoneFinderService;
import org.squashtest.tm.service.project.ProjectFilterModificationService;
import org.squashtest.tm.web.internal.i18n.InternationalizationHelper;
import org.squashtest.tm.web.internal.model.datatable.DataTableModel;

@Controller
@RequestMapping("/milestones")
public class MilestoneController {

    @Inject
    private Provider<MilestoneStatusComboDataBuilder> statusComboDataBuilderProvider;

    @Inject
    private MilestoneFinderService milestoneFinder;

    @Inject
    private InternationalizationHelper i18nHelper;

    @Inject
    private ProjectFilterModificationService projectFilterService;

    @RequestMapping(value = "/status-combo-data", method = RequestMethod.GET)
    @ResponseBody
    public String buildStatusComboData(Locale locale) {
        return statusComboDataBuilderProvider.get().useLocale(locale).buildMarshalled();
    }

    @RequestMapping(params = "selectable", method = RequestMethod.GET)
    @ResponseBody
    public DataTableModel<Milestone> findUserSelectableMilestones() {

        List<Milestone> milestones = milestoneFinder.findAllVisibleToCurrentUser();

        //checking global project filter and filter milestone who aren't binded to at least one project in filter
        ProjectFilter projectFilter = projectFilterService.findProjectFilterByUserLogin();
        if (projectFilter.isEnabled()) {
            Collection<Milestone> milestonesCollection = CollectionUtils.retainAll(milestones,
                    getMilestoneFromProjectFilter(projectFilter));
            milestones = new ArrayList<>(0);
            milestones.addAll(milestonesCollection);
        }

        // they must be initially sorted by date descending
        Collections.sort(milestones, new Comparator<Milestone>() {
            @Override
            public int compare(Milestone o1, Milestone o2) {
                return o2.getEndDate().before(o1.getEndDate()) ? -1 : 1;
            }
        });

        // now make the model
        PagedCollectionHolder<List<Milestone>> holderCollection = new SinglePageCollectionHolder<>(milestones);

        Locale locale = LocaleContextHolder.getLocale();
        return new MilestoneTableModelHelper(i18nHelper, locale).buildDataModel(holderCollection, "0");

    }

    //--------------------------------- PRIVATE STUFF ---------------------------------//

    private Set<Milestone> getMilestoneFromProjectFilter(ProjectFilter projectFilter) {
        HashSet<Milestone> milestoneFiltered = new HashSet<>();

        List<Project> projects = projectFilter.getProjects();
        for (Project project : projects) {
            milestoneFiltered.addAll(project.getMilestones());
        }
        return milestoneFiltered;
    }

}