/*
 * Copyright (c) 2012. The Genome Analysis Centre, Norwich, UK
 * MISO project contacts: Robert Davey, Mario Caccamo @ TGAC
 * *********************************************************************
 *
 * This file is part of MISO.
 *
 * MISO 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.
 *
 * MISO 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 MISO.  If not, see <http://www.gnu.org/licenses/>.
 *
 * *********************************************************************
 */

package uk.ac.bbsrc.tgac.miso.webapp.controller;

import com.eaglegenomics.simlims.core.manager.SecurityManager;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.integration.Message;
import org.springframework.integration.support.MessageBuilder;
import org.springframework.stereotype.Controller;
import org.springframework.util.LinkedMultiValueMap;
import org.springframework.util.MultiValueMap;
import org.springframework.web.bind.annotation.*;
import uk.ac.bbsrc.tgac.miso.core.data.type.PlatformType;
import uk.ac.bbsrc.tgac.miso.core.manager.RequestManager;
import uk.ac.bbsrc.tgac.miso.core.service.integration.NotificationGatewayService;
import uk.ac.bbsrc.tgac.miso.core.service.integration.strategy.NotificationGateway;
import uk.ac.bbsrc.tgac.miso.core.util.LimsUtils;

import javax.servlet.http.HttpServletRequest;
import java.io.*;
import java.util.*;

/**
 * A message consumer that will listen for messages and pass them on to respective handlers. This is a REST-aware controller, which means
 * message producers can POST messages to this controller.
 * <p/>
 * Info
 *
 * @author Rob Davey
 * @since 0.0.2
 */
@Controller
@RequestMapping(value="/consumer")
public class MessageConsumerController {
  protected static final Logger log = LoggerFactory.getLogger(MessageConsumerController.class);

  @Autowired
  private NotificationGatewayService notificationGatewayService;

  @Autowired
  private SecurityManager securityManager;

  @Autowired
  private RequestManager requestManager;

  public void setNotificationGatewayService(NotificationGatewayService notificationGatewayService) {
    this.notificationGatewayService = notificationGatewayService;
  }

  public void setRequestManager(RequestManager requestManager) {
    this.requestManager = requestManager;
  }

  public void setSecurityManager(SecurityManager securityManager) {
    this.securityManager = securityManager;
  }

  @RequestMapping(value="/illumina/run/status", method = RequestMethod.POST)
  public String consumeGatewayIlluminaStatus(HttpServletRequest request) throws IOException {
    for (NotificationGateway s : notificationGatewayService.getGatewaysFor(PlatformType.ILLUMINA)) {
      log.debug("Using " + s.toString());
      s.consume(buildMessage(exposeRequest(request)));
    }
    return "redirect:/miso";
  }

  @RequestMapping(value="/ls454/run/status", method = RequestMethod.POST)
  public String consumeGateway454Status(HttpServletRequest request) throws IOException {
    for (NotificationGateway s : notificationGatewayService.getGatewaysFor(PlatformType.LS454)) {
      log.debug("Using " + s.toString());
      s.consume(buildMessage(exposeRequest(request)));
    }
    return "redirect:/miso";
  }

  @RequestMapping(value="/solid/run/status", method = RequestMethod.POST)
  public String consumeGatewaySolidStatus(HttpServletRequest request) throws IOException {
    for (NotificationGateway s : notificationGatewayService.getGatewaysFor(PlatformType.SOLID)) {
      log.debug("Using " + s.toString());
      s.consume(buildMessage(exposeRequest(request)));
    }
    return "redirect:/miso";
  }

  @RequestMapping(value="/pacbio/run/status", method = RequestMethod.POST)
  public String consumeGatewayPacbioStatus(HttpServletRequest request) throws IOException {
    for (NotificationGateway s : notificationGatewayService.getGatewaysFor(PlatformType.PACBIO)) {
      log.debug("Using " + s.toString());
      s.consume(buildMessage(exposeRequest(request)));
    }
    return "redirect:/miso";
  }

  @RequestMapping(value="/iontorrent/run/status", method = RequestMethod.POST)
  public String consumeGatewayIonTorrentStatus(HttpServletRequest request) throws IOException {
    for (NotificationGateway s : notificationGatewayService.getGatewaysFor(PlatformType.IONTORRENT)) {
      log.debug("Using " + s.toString());
      s.consume(buildMessage(exposeRequest(request)));
    }
    return "redirect:/miso";
  }

  private MultiValueMap<String, String> exposeRequest(HttpServletRequest request) {
    log.debug("Request size: "+request.getContentLength());
    Map<String, Object> map = request.getParameterMap();
    log.debug("RAW MAP: " + map.toString());
    MultiValueMap<String, String> message = new LinkedMultiValueMap<String, String>();
    for (String s : map.keySet()) {
      log.debug("EXPOSING PARAM '" + s + "': " + LimsUtils.join(request.getParameterValues(s), ","));
      message.put(s, Arrays.asList(request.getParameterValues(s)));
    }

    return message;
  }

  private <T> Message<T> buildMessage(T payload) {
    return MessageBuilder.withPayload(payload).setHeader("handler", requestManager).build();
  }
}