com.capitalone.dashboard.rest.DeployController Maven / Gradle / Ivy

Go to download
package com.capitalone.dashboard.rest;

import com.capitalone.dashboard.misc.HygieiaException;
import com.capitalone.dashboard.model.DataResponse;
import com.capitalone.dashboard.model.deploy.Environment;
import com.capitalone.dashboard.request.DeployDataCreateRequest;
import com.capitalone.dashboard.service.DeployService;
import org.bson.types.ObjectId;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.RequestBody;
import org.springframework.web.bind.annotation.RequestHeader;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import org.w3c.dom.Document;
import org.xml.sax.InputSource;
import org.xml.sax.SAXException;

import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.validation.Valid;
import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import java.io.IOException;
import java.util.List;

import static org.springframework.http.MediaType.APPLICATION_JSON_VALUE;
import static org.springframework.http.MediaType.TEXT_XML_VALUE;
import static org.springframework.web.bind.annotation.RequestMethod.GET;
import static org.springframework.web.bind.annotation.RequestMethod.POST;

@RestController
public class DeployController {


    private final DeployService deployService;

    private static DocumentBuilder documentBuilder;

    static {
        try {
            documentBuilder = getDocumentBuilder();
        }
        catch (ParserConfigurationException e) {}
    }

    @Autowired
    public DeployController(DeployService deployService) {
        this.deployService = deployService;
    }

    @RequestMapping(value = "/deploy/status/{componentId}", method = GET, produces = APPLICATION_JSON_VALUE)
    public DataResponse> deployStatus(@PathVariable ObjectId componentId) {
        return deployService.getDeployStatus(componentId);
    }

    @RequestMapping(value = "/deploy/status/application/{applicationName}", method = GET, produces = APPLICATION_JSON_VALUE)
    public DataResponse> deployStatus(@PathVariable String applicationName) {
        return deployService.getDeployStatus(applicationName);
    }

    @RequestMapping(value = "/deploy", method = POST,
            consumes = APPLICATION_JSON_VALUE, produces = APPLICATION_JSON_VALUE)
    public ResponseEntity createDeploy(@Valid @RequestBody DeployDataCreateRequest request) throws HygieiaException {
        String response = deployService.create(request);
        return ResponseEntity
                .status(HttpStatus.CREATED)
                .body(response);
    }

    @RequestMapping(value = "/v2/deploy", method = POST,
            consumes = APPLICATION_JSON_VALUE, produces = APPLICATION_JSON_VALUE)
    public ResponseEntity createDeployV2(@Valid @RequestBody DeployDataCreateRequest request) throws HygieiaException {
        String response = deployService.createV2(request);
        return ResponseEntity
                .status(HttpStatus.CREATED)
                .body(response);
    }

    @RequestMapping(value = "/v3/deploy", method = POST,
            consumes = APPLICATION_JSON_VALUE, produces = APPLICATION_JSON_VALUE)
    public ResponseEntity createDeployV3(@Valid @RequestBody DeployDataCreateRequest request) throws HygieiaException {
        String response = deployService.createV3(request);
        return ResponseEntity
                .status(HttpStatus.CREATED)
                .body(response);
    }

    @RequestMapping(value = "/deploy/rundeck", method = POST,
            consumes = TEXT_XML_VALUE, produces = APPLICATION_JSON_VALUE)
    public ResponseEntity createRundeckBuild(HttpServletRequest request,
            @RequestHeader("X-Rundeck-Notification-Execution-ID") String executionId,
            @RequestHeader("X-Rundeck-Notification-Trigger") String status) throws HygieiaException{
        String response = null;
        try {
            response = deployService.createRundeckBuild(getDocument(request.getInputStream()), request.getParameterMap(), executionId, status);
        } catch (IOException e) {
            throw new HygieiaException(e);
        }

        return ResponseEntity
                .status(HttpStatus.CREATED)
                .body(response);
    }

    @RequestMapping(value = "/v2/deploy/rundeck", method = POST,
            consumes = TEXT_XML_VALUE, produces = APPLICATION_JSON_VALUE)
    public ResponseEntity createRundeckBuildV2(HttpServletRequest request,
                                                     @RequestHeader("X-Rundeck-Notification-Execution-ID") String executionId,
                                                     @RequestHeader("X-Rundeck-Notification-Trigger") String status) throws HygieiaException{
        String response = null;
        try {
             response = deployService.createRundeckBuildV2(getDocument(request.getInputStream()), request.getParameterMap(), executionId, status);
        } catch (IOException e) {
            throw new HygieiaException(e);
        }

        return ResponseEntity
                .status(HttpStatus.CREATED)
                .body(response);
    }


    private Document getDocument (ServletInputStream inputStream) throws HygieiaException{
        Document doc = null;
        try {
            documentBuilder = (documentBuilder == null) ? getDocumentBuilder() : documentBuilder;
            doc = documentBuilder.parse(new InputSource(inputStream));
        } catch (ParserConfigurationException | SAXException | IOException e) {
            throw new HygieiaException(e);
        }
        return doc;
    }

    private static DocumentBuilder getDocumentBuilder() throws ParserConfigurationException{
        DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
        factory.setFeature("http://apache.org/xml/features/disallow-doctype-decl", true);
        return factory.newDocumentBuilder();
    }

}