com.hubspot.singularity.resources.NotificationsResource Maven / Gradle / Ivy

Go to download
package com.hubspot.singularity.resources;

import java.util.List;

import javax.servlet.http.HttpServletRequest;
import javax.ws.rs.GET;
import javax.ws.rs.POST;
import javax.ws.rs.Path;
import javax.ws.rs.Produces;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.MediaType;

import org.apache.curator.framework.recipes.leader.LeaderLatch;

import com.fasterxml.jackson.databind.ObjectMapper;
import com.google.inject.Inject;
import com.hubspot.singularity.SingularityUser;
import com.hubspot.singularity.config.ApiPaths;
import com.hubspot.singularity.data.NotificationsManager;
import com.ning.http.client.AsyncHttpClient;

import io.dropwizard.auth.Auth;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.media.Schema;
import io.swagger.v3.oas.annotations.parameters.RequestBody;
import io.swagger.v3.oas.annotations.tags.Tag;
import io.swagger.v3.oas.annotations.tags.Tags;

@Path(ApiPaths.NOTIFICATIONS_RESOURCE_PATH)
@Produces({ MediaType.APPLICATION_JSON })
@Schema(title = "Manage email notifications")
@Tags({@Tag(name = "Notifications")})
public class NotificationsResource extends AbstractLeaderAwareResource {
  private final NotificationsManager notificationsManager;

  @Inject
  public NotificationsResource(AsyncHttpClient httpClient, LeaderLatch leaderLatch, ObjectMapper objectMapper, NotificationsManager notificationsManager) {
    super(httpClient, leaderLatch, objectMapper);
    this.notificationsManager = notificationsManager;
  }

  @GET
  @Path("/blacklist")
  @Operation(summary = "Retrieve the list of blacklisted emails")
  public List getBlacklist(@Parameter(hidden = true) @Auth SingularityUser user) {
    return notificationsManager.getBlacklist();
  }

  @POST
  @Path("/unsubscribe")
  @Operation(summary = "Unsubscribe from Singularity emails.")
  public void unsubscribe(
      @Parameter(hidden = true) @Auth SingularityUser user,
      @RequestBody(required = true, description = "The email address to unsubscribe") String email,
      @Context HttpServletRequest requestContext) {
    maybeProxyToLeader(requestContext, Void.class, email, () -> {
      notificationsManager.addToBlacklist(getFormattedEmail(email));
      return null; });
  }

  @POST
  @Path("/subscribe")
  @Operation(summary = "Delete an unsubscription for an email address")
  public void subscribe(
      @Parameter(hidden = true) @Auth SingularityUser user,
      @RequestBody(required = true, description = "The email address to re-subscribe") String email,
      @Context HttpServletRequest requestContext) {
    maybeProxyToLeader(requestContext, Void.class, email, () -> {
      notificationsManager.removeFromBlacklist(getFormattedEmail(email));
      return null; });
  }

  private String getFormattedEmail(String email) {
    return email.replaceAll("^\"|\"$", "");
  }
}