pl.allegro.tech.hermes.management.api.ModeEndpoint Maven / Gradle / Ivy

Go to download
package pl.allegro.tech.hermes.management.api;

import static jakarta.ws.rs.core.MediaType.APPLICATION_JSON;
import static jakarta.ws.rs.core.MediaType.TEXT_PLAIN;
import static pl.allegro.tech.hermes.management.domain.mode.ModeService.ManagementMode.READ_ONLY_ADMIN;
import static pl.allegro.tech.hermes.management.domain.mode.ModeService.ManagementMode.READ_WRITE;

import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import jakarta.annotation.security.RolesAllowed;
import jakarta.ws.rs.GET;
import jakarta.ws.rs.HttpMethod;
import jakarta.ws.rs.POST;
import jakarta.ws.rs.Path;
import jakarta.ws.rs.Produces;
import jakarta.ws.rs.QueryParam;
import jakarta.ws.rs.core.Response;
import org.springframework.stereotype.Component;
import pl.allegro.tech.hermes.management.api.auth.Roles;
import pl.allegro.tech.hermes.management.domain.mode.ModeService;

@Component
@Path("/mode")
@Api(value = "/mode", description = "Operations on management mode")
public class ModeEndpoint {

  private final ModeService modeService;

  public ModeEndpoint(ModeService modeService) {
    this.modeService = modeService;
  }

  @GET
  @Produces(TEXT_PLAIN)
  @ApiOperation(value = "Get management mode", response = String.class, httpMethod = HttpMethod.GET)
  public String getMode() {
    return modeService.getMode().toString();
  }

  @POST
  @Produces(APPLICATION_JSON)
  @ApiOperation(
      value = "Set management mode",
      response = String.class,
      httpMethod = HttpMethod.POST)
  @RolesAllowed(Roles.ADMIN)
  public Response setMode(@QueryParam("mode") String mode) {
    if (mode == null) {
      return Response.status(Response.Status.BAD_REQUEST).build();
    }
    switch (mode) {
      case ModeService.READ_WRITE:
        modeService.setModeByAdmin(READ_WRITE);
        break;
      case ModeService.READ_ONLY:
      case ModeService.READ_ONLY_ADMIN:
        modeService.setModeByAdmin(READ_ONLY_ADMIN);
        break;
      default:
        return Response.status(Response.Status.BAD_REQUEST).build();
    }
    return Response.status(Response.Status.OK).build();
  }
}