keywhiz.service.resources.automation.AutomationSecretGeneratorsResource Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of keywhiz-server Show documentation
There is a newer version: 0.10.1
/*
 * Copyright (C) 2015 Square, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * 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 License for the specific language governing permissions and
 * limitations under the License.
 */

package keywhiz.service.resources.automation;

import com.fasterxml.jackson.databind.ObjectMapper;
import io.dropwizard.auth.Auth;
import java.io.IOException;
import java.util.List;
import java.util.Map;
import javax.inject.Inject;
import javax.ws.rs.BadRequestException;
import javax.ws.rs.Consumes;
import javax.ws.rs.NotFoundException;
import javax.ws.rs.POST;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.Produces;
import keywhiz.api.model.AutomationClient;
import keywhiz.api.model.SanitizedSecret;
import keywhiz.generators.SecretGenerator;
import keywhiz.service.exceptions.UnprocessableEntityException;

import static javax.ws.rs.core.MediaType.APPLICATION_JSON;

/**
 * @parentEndpointName secrets-generator-automation
 *
 * @resourceDescription Create secrets with a generator
 */
@Path("/automation/secrets/generators")
@Produces(APPLICATION_JSON)
public class AutomationSecretGeneratorsResource {
  private final ObjectMapper mapper;
  private final Map generatorMap;

  @Inject
  public AutomationSecretGeneratorsResource(ObjectMapper mapper, Map generatorMap) {
    this.mapper = mapper;
    this.generatorMap = generatorMap;
  }

  /**
   * Generate Secrets from a single generator request
   *
   * @excludeParams client
   * @param generatorName the name of the generator to use for creating secrets
   *
   * @description Creates Secrets from a single generator request, using the specified generator
   * @responseMessage 200 Successfully generated secrets according to the request
   * @responseMessage 400 Malformed or incorrect request parameters provided
   * @responseMessage 422 Request was formed correctly but was semantically incorrect
   */
  @Path("{generatorName}")
  @POST
  @Consumes(APPLICATION_JSON)
  public List generate(@Auth AutomationClient client,
      @PathParam("generatorName") String generatorName, String requestBody) {
    SecretGenerator generator = getGeneratorOrThrow(generatorName);

    Object requestParams;
    try {
      requestParams = mapper.readValue(requestBody, generator.getRequestType());
    } catch (IOException e) {
      throw new UnprocessableEntityException(e.getMessage());
    }

    return SanitizedSecret.fromSecrets(generator.generate(client.getName(), requestParams));
  }

  /**
   * Generate Secrets from a batch of generator requests
   *
   * @excludeParams client
   * @param generatorName the name of the generator to use for creating secrets
   *
   * @description Creates Secrets from a batch of generator requests, using the specified generator
   * @responseMessage 200 Successfully generated secrets according to the request
   * @responseMessage 400 Malformed or incorrect request parameters provided
   * @responseMessage 422 Request was formed correctly but was semantically incorrect, batch may have been empty
   */
  @Path("{generatorName}/batch")
  @POST
  @Consumes(APPLICATION_JSON)
  public List batchGenerate(@Auth AutomationClient client,
      @PathParam("generatorName") String generatorName, String requestBody) {
    SecretGenerator generator = getGeneratorOrThrow(generatorName);

    List