class WebSocketServletImpl extends JettyWebSocketServlet {

	private static final long serialVersionUID = -8292051630200394714L;

	/** The websocket creator */
	private final JettyWebSocketCreator creator;

	/** The underlying Jetty context */
	private ServletContextHandler context;

	/** Used to get attributes from the Jetty context */
	private ServletContext proxiedContext;

	/**
	 * @param creator the non-null creator
	 */
	WebSocketServletImpl(JettyWebSocketCreator creator) {
		if (creator == null)
			throw new IllegalArgumentException("creator is null");

		this.creator = creator;
	}

	@Override
	public void init() throws ServletException {
		final ServletContext osgiServletContext = super.getServletContext();
		context = ServletContextHandler.getServletContextHandler(osgiServletContext, "WebSockets");

		final JettyWebSocketServerContainer serverContainer = JettyWebSocketServerContainer
				.getContainer(osgiServletContext);
		if (serverContainer == null) {
			final Context jettyServletContext = context.getServletContext();
			WebSocketServerComponents.ensureWebSocketComponents(context.getServer(), jettyServletContext);
			WebSocketUpgradeFilter.ensureFilter(jettyServletContext);
			WebSocketMappings.ensureMappings(jettyServletContext);
			JettyServerFrameHandlerFactory.getFactory(jettyServletContext);
			JettyWebSocketServerContainer.ensureContainer(jettyServletContext);
		}

		super.init();
	}

	@Override
	public synchronized ServletContext getServletContext() {
		if (proxiedContext == null) {
			proxiedContext = (ServletContext) Proxy.newProxyInstance(WebSocketServletImpl.class.getClassLoader(),
					new Class[] { ServletContext.class }, (proxy, method, methodArgs) -> {
						final ServletContext osgiServletContext = super.getServletContext();
						if (!"getAttribute".equals(method.getName())) {
							return method.invoke(osgiServletContext, methodArgs);
						}

						final String name = (String) methodArgs[0];
						Object value = osgiServletContext.getAttribute(name);
						if (value == null && context != null) {
							final Context jettyServletContext = context.getServletContext();
							value = jettyServletContext.getAttribute(name);
						}
						return value;
					});
		}

		return proxiedContext;
	}

	@Override
	public void configure(JettyWebSocketServletFactory factory) {
		factory.setMaxBinaryMessageSize(1024 * 1024);
		factory.setMaxTextMessageSize(128 * 1024);
		// factory.setAsyncWriteTimeout(5000);
		factory.setCreator(creator);
		getServletContext().setAttribute(DecoratedObjectFactory.ATTR, new DecoratedObjectFactory());
	}

	@Override
	protected void service(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
		final HttpServletRequest proxiedReq = (HttpServletRequest) Proxy.newProxyInstance(
				WebSocketServletImpl.class.getClassLoader(), new Class[] { HttpServletRequest.class },
				(proxy, method, methodArgs) -> {
					if (!"getServletContext".equals(method.getName())) {
						return method.invoke(req, methodArgs);
					}

					if (context != null)
						return context.getServletContext();

					return req.getServletContext();
				});

		super.service(proxiedReq, resp);
	}
}