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

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

import com.fasterxml.jackson.databind.ObjectMapper;
import com.google.inject.Inject;
import com.hubspot.singularity.Singularity;
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;
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;

@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,
    @Singularity ObjectMapper objectMapper,
    NotificationsManager notificationsManager
  ) {
    super(httpClient, leaderLatch, objectMapper);
    this.notificationsManager = notificationsManager;
  }

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

  @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.addToBlocklist(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.removeFromBlocklist(getFormattedEmail(email));
        return null;
      }
    );
  }

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