SierraSystems/ntt-data-notification-service

package com.nttdata.nttdatanotificationservice.controller;

import com.nttdata.nttdatanotificationservice.configuration.NotificationBody;
import com.nttdata.nttdatanotificationservice.configuration.NotificationServiceProperties;
import com.nttdata.nttdatanotificationservice.service.WebHookService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.context.properties.EnableConfigurationProperties;
import org.springframework.http.HttpStatus;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.*;

@RestController
@EnableConfigurationProperties(NotificationServiceProperties.class)
public class NotificationController {

  @Autowired
  WebHookService webHookService;

  @Autowired
  NotificationServiceProperties notificationServiceProperties;

  Logger logger = LoggerFactory.getLogger(NotificationController.class);

  @PostMapping(value = "generic/{token}", produces = MediaType.APPLICATION_JSON_VALUE)
  public ResponseEntity<String> alert(@PathVariable("token") String token,
                                      @RequestBody NotificationBody notificationBody) {
    if (!notificationServiceProperties.getTokens().contains(token)) {
      logger.error("Token failed to validate");
      return new ResponseEntity<>("Token validation failed", HttpStatus.UNAUTHORIZED);
    }

    return webHookService.postMessage(notificationBody.getNotification(), notificationBody.getWebHookParams());
  }
}