package nl.andrewl.railsignalapi.live.websocket;

import lombok.RequiredArgsConstructor;
import org.springframework.context.annotation.Configuration;
import org.springframework.core.env.Environment;
import org.springframework.web.socket.config.annotation.EnableWebSocket;
import org.springframework.web.socket.config.annotation.WebSocketConfigurer;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistration;
import org.springframework.web.socket.config.annotation.WebSocketHandlerRegistry;

import java.util.Set;

/**
 * Configuration for Rail Signal's websockets. This includes both app and
 * component connections.
 */
@Configuration
@EnableWebSocket
@RequiredArgsConstructor
public class WebsocketConfig implements WebSocketConfigurer {
	private final ComponentWebsocketHandler componentHandler;
	private final ComponentWebsocketHandshakeInterceptor componentInterceptor;
	private final AppWebsocketHandler appHandler;
	private final AppWebsocketHandshakeInterceptor appInterceptor;
	private final Environment env;

	@Override
	public void registerWebSocketHandlers(WebSocketHandlerRegistry registry) {
		registry.addHandler(componentHandler, "/api/ws/component")
				.setAllowedOrigins("*")
				.addInterceptors(componentInterceptor);
		WebSocketHandlerRegistration appHandlerReg = registry.addHandler(appHandler, "/api/ws/app/*")
				.addInterceptors(appInterceptor);
		// If we're in a development profile, allow any origin to access the app websocket.
		// This is so that we can use a standalone JS dev server.
		if (Set.of(env.getActiveProfiles()).contains("development")) {
			appHandlerReg.setAllowedOrigins("*");
		}
	}
}