sentilo/sentilo

/*
 * Sentilo
 *
 * Original version 1.4 Copyright (C) 2013 Institut Municipal d’Informàtica, Ajuntament de
 * Barcelona. Modified by Opentrends adding support for multitenant deployments and SaaS.
 * Modifications on version 1.5 Copyright (C) 2015 Opentrends Solucions i Sistemes, S.L.
 *
 *
 * This program is licensed and may be used, modified and redistributed under the terms of the
 * European Public License (EUPL), either version 1.1 or (at your option) any later version as soon
 * as they are approved by the European Commission.
 *
 * Alternatively, you may redistribute and/or modify this program 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.
 *
 * Unless required by applicable law or agreed to in writing, software distributed under the License
 * is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express
 * or implied.
 *
 * See the licenses for the specific language governing permissions, limitations and more details.
 *
 * You should have received a copy of the EUPL1.1 and the LGPLv3 licenses along with this program;
 * if not, you may find them at:
 *
 * https://joinup.ec.europa.eu/software/page/eupl/licence-eupl http://www.gnu.org/licenses/ and
 * https://www.gnu.org/licenses/lgpl.txt
 */
package org.sentilo.web.catalog.controller.admin;

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

import javax.servlet.http.HttpServletRequest;

import org.sentilo.common.enums.AlertTriggerType;
import org.sentilo.common.enums.AlertType;
import org.sentilo.web.catalog.controller.CrudController;
import org.sentilo.web.catalog.domain.Alert;
import org.sentilo.web.catalog.domain.Application;
import org.sentilo.web.catalog.domain.Provider;
import org.sentilo.web.catalog.dto.OptionDTO;
import org.sentilo.web.catalog.service.AlertService;
import org.sentilo.web.catalog.service.ApplicationService;
import org.sentilo.web.catalog.service.CrudService;
import org.sentilo.web.catalog.service.ProviderService;
import org.sentilo.web.catalog.service.SectorService;
import org.sentilo.web.catalog.utils.CatalogUtils;
import org.sentilo.web.catalog.utils.Constants;
import org.sentilo.web.catalog.utils.ExcelGeneratorUtils;
import org.sentilo.web.catalog.utils.FormatUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.i18n.LocaleContextHolder;
import org.springframework.stereotype.Controller;
import org.springframework.ui.Model;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.ModelAttribute;
import org.springframework.web.bind.annotation.RequestMapping;

@Controller
@RequestMapping("/admin/alert")
public class AlertController extends CrudController<Alert> {

  @Autowired
  private AlertService alertService;

  @Autowired
  private ApplicationService applicationService;

  @Autowired
  private ProviderService providerService;

  @Autowired
  private SectorService sectorService;

  @ModelAttribute(Constants.MODEL_ACTIVE_MENU)
  public String getActiveMenu() {
    return Constants.MENU_ALERT;
  }

  @Override
  protected void initViewNames() {
    getViewNames().put(LIST_ACTION, Constants.VIEW_ALERT_LIST);
    getViewNames().put(DETAIL_ACTION, Constants.VIEW_ALERT_DETAIL);
    getViewNames().put(NEW_ACTION, Constants.VIEW_NEW_ALERT);
  }

  @Override
  protected List<String> toRow(final Alert alert) {
    final List<String> row = new ArrayList<String>();
    row.add(alert.getId()); // checkbox
    row.add(alert.getId());
    row.add(FormatUtils.label(messageSource.getMessage("alert.type." + alert.getType().toString(), null, LocaleContextHolder.getLocale())));
    row.add(FormatUtils.formatAlertTriggerColumn(alert));
    row.add(String.valueOf(alert.isActive()));
    row.add(getLocalDateFormat().printAsLocalTime(alert.getCreatedAt(), Constants.DATETIME_FORMAT));
    return row;
  }

  @Override
  protected List<String> toExcelRow(final Alert alert) {
    return ExcelGeneratorUtils.getAlertExcelRowsData(alert, messageSource, getLocalDateFormat());
  }

  private void addAlertMasterDataTo(final Model model) {
    addProviderListTo(model);
    addApplicationListTo(model);
    addAlertTypesTo(model);
    addAlertTriggersTo(model);
  }

  @SuppressWarnings("unchecked")
  private void addProviderListTo(final Model model) {
    final List<Provider> providers = (List<Provider>) getResourcesUserSectorsGranted(providerService);
    final List<OptionDTO> providersDTO = CatalogUtils.toOptionList(providers);
    model.addAttribute(Constants.MODEL_PROVIDERS, providersDTO);
  }

  @SuppressWarnings("unchecked")
  private void addApplicationListTo(final Model model) {
    final List<Application> applications = (List<Application>) getResourcesUserSectorsGranted(applicationService);
    final List<OptionDTO> applicationDto = CatalogUtils.toOptionList(applications);
    model.addAttribute(Constants.MODEL_APPLICATIONS, applicationDto);
  }

  private void addAlertTypesTo(final Model model) {
    model.addAttribute(Constants.MODEL_ALERT_TYPES, CatalogUtils.toOptionList(AlertType.class, "alert.type", messageSource));
  }

  private void addAlertTriggersTo(final Model model) {
    model.addAttribute(Constants.MODEL_ALERT_TRIGGERS, CatalogUtils.toOptionList(AlertTriggerType.class, "alert.trigger", messageSource));
  }

  @Override
  protected CrudService<Alert> getService() {
    return alertService;
  }

  @Override
  protected Alert buildNewEntity(final String id) {
    return new Alert(id);
  }

  @Override
  protected String getEntityModelKey() {
    return Constants.MODEL_ALERT;
  }

  /*
   * (non-Javadoc)
   *
   * @see org.sentilo.web.catalog.controller.CrudController#doBeforeNewResource(javax.servlet.http.
   * HttpServletRequest, org.springframework.ui.Model)
   */
  @Override
  protected void doBeforeNewResource(final HttpServletRequest request, final Model model) {
    super.doBeforeNewResource(request, model);
    addAlertMasterDataTo(model);
  }

  /*
   * (non-Javadoc)
   *
   * @see org.sentilo.web.catalog.controller.CrudController#doBeforeEditResource(java.lang.String,
   * org.springframework.ui.Model)
   */
  @Override
  protected void doBeforeEditResource(final String id, final Model model) {
    super.doBeforeEditResource(id, model);
    addAlertMasterDataTo(model);
  }

  /*
   * (non-Javadoc)
   *
   * @see org.sentilo.web.catalog.controller.CrudController#doBeforeCreateResource(org.sentilo.web.
   * catalog .domain.CatalogDocument, org.springframework.ui.Model)
   */
  @Override
  protected void doBeforeCreateResource(final Alert alert, final Model model) {
    setToNullAlertFields(alert);
  }

  /*
   * (non-Javadoc)
   *
   * @see org.sentilo.web.catalog.controller.CrudController#doBeforeUpdateResource(org.sentilo.web.
   * catalog .domain.CatalogDocument, org.springframework.ui.Model)
   */
  @Override
  protected void doBeforeUpdateResource(final Alert alert, final Model model) {
    setToNullAlertFields(alert);
  }

  private void setToNullAlertFields(final Alert alert) {
    // If the alert is internal we put to null all specific attributes of an external alert.
    // Same with an external alert
    if (AlertType.INTERNAL.equals(alert.getType())) {
      alert.setApplicationId(null);
    } else {
      alert.setComponentId(null);
      alert.setSensorId(null);
      alert.setTrigger(null);
      alert.setExpression(null);

      if (StringUtils.hasText(alert.getProviderId())) {
        alert.setApplicationId(null);
      } else if (StringUtils.hasText(alert.getApplicationId())) {
        alert.setProviderId(null);
      }
    }
  }

  @Override
  protected void doAfterViewResource(final Model model) {
    super.doAfterViewResource(model);
    addSectors(model, sectorService);
  }

}