io.yawp.servlet.FixturesServlet Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of yawp-core Show documentation

YAWP! core

There is a newer version: 2.08alpha

package io.yawp.servlet;

import io.yawp.commons.http.ErrorResponse;
import io.yawp.driver.api.Driver;
import io.yawp.repository.Repository;

import java.io.IOException;

import javax.servlet.ServletConfig;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

public class FixturesServlet extends EndpointServlet {

	private static final long serialVersionUID = -7833278558858095857L;

	@Override
	public void init(ServletConfig config) throws ServletException {
		super.init(config);
		setWithHooks(false);
	}

	@Override
	protected void service(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
		if (!enableFixtures(req)) {
			response(resp, new ErrorResponse(403));
			return;
		}
		super.service(req, resp);
	}

	private boolean enableFixtures(HttpServletRequest req) {
		Repository r = getRepository(makeParams(req));
		Driver driver = r.getDriver();

		if (!driver.environment().isProduction()) {
			return true;
		}

		return driver.environment().isAdmin();
	}

}