openvidu-2/openvidu-server/src/main/java/io/openvidu/server/OpenViduServer.java

/*
 * (C) Copyright 2014 Kurento (http://kurento.org/)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *   http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package io.openvidu.server;

import static org.kurento.commons.PropertiesManager.getPropertyJson;

import java.util.List;

import org.kurento.jsonrpc.JsonUtils;
import org.kurento.jsonrpc.internal.server.config.JsonRpcConfiguration;
import org.kurento.jsonrpc.server.JsonRpcConfigurer;
import org.kurento.jsonrpc.server.JsonRpcHandlerRegistry;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;
import org.springframework.boot.autoconfigure.condition.ConditionalOnMissingBean;
import org.springframework.context.ConfigurableApplicationContext;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Import;
import org.springframework.web.socket.server.standard.ServletServerContainerFactoryBean;

import com.google.gson.Gson;
import com.google.gson.JsonArray;

import io.openvidu.server.core.NotificationRoomManager;
import io.openvidu.server.core.RoomManager;
import io.openvidu.server.core.api.KurentoClientProvider;
import io.openvidu.server.core.api.NotificationRoomHandler;
import io.openvidu.server.core.internal.DefaultNotificationRoomHandler;
import io.openvidu.server.kms.FixedOneKmsManager;
import io.openvidu.server.rest.NgrokController;
import io.openvidu.server.rpc.JsonRpcNotificationService;
import io.openvidu.server.rpc.JsonRpcUserControl;
import io.openvidu.server.security.OpenviduConfiguration;

/**
 * Room server application.
 *
 * @author Ivan Gracia (izanmail@gmail.com)
 * @author Micael Gallego (micael.gallego@gmail.com)
 * @author Radu Tom Vlad (rvlad@naevatec.com)
 * @since 1.0.0
 */
@Import({ JsonRpcConfiguration.class, InfoSocketConfig.class })
@SpringBootApplication
public class OpenViduServer implements JsonRpcConfigurer {

	public static final String KMSS_URIS_PROPERTY = "kms.uris";
	public static final String KMSS_URIS_DEFAULT = "[ \"ws://localhost:8888/kurento\" ]";

	@Value("${kms.uris}")
	private String KMSS_CUSTOM_URIS;

	public static String publicUrl;

	private static final Logger log = LoggerFactory.getLogger(OpenViduServer.class);

	@Bean
	@ConditionalOnMissingBean
	public KurentoClientProvider kmsManager() {

		JsonArray kmsUris = getPropertyJson(KMSS_URIS_PROPERTY, KMSS_URIS_DEFAULT, JsonArray.class);
		List<String> kmsWsUris = JsonUtils.toStringList(kmsUris);

		if ((KMSS_CUSTOM_URIS != null) && (!KMSS_CUSTOM_URIS.isEmpty())) {
			List<String> uris = new Gson().fromJson(KMSS_CUSTOM_URIS, List.class);
			kmsWsUris.addAll(0, uris);
		}

		if (kmsWsUris.isEmpty()) {
			throw new IllegalArgumentException(KMSS_URIS_PROPERTY + " should contain at least one kms url");
		}

		String firstKmsWsUri = kmsWsUris.get(0);

		if (firstKmsWsUri.equals("autodiscovery")) {
			log.info("Using autodiscovery rules to locate KMS on every pipeline");
			return new AutodiscoveryKurentoClientProvider();
		} else {
			log.info("Configuring Kurento Room Server to use first of the following kmss: " + kmsWsUris);
			return new FixedOneKmsManager(firstKmsWsUri);
		}
	}

	@Bean
	@ConditionalOnMissingBean
	public JsonRpcNotificationService notificationService() {
		return new JsonRpcNotificationService();
	}

	@Bean
	@ConditionalOnMissingBean
	public NotificationRoomHandler defaultNotificationRoomHandler() {
		return new DefaultNotificationRoomHandler(notificationService());
	}

	@Bean
	@ConditionalOnMissingBean
	public RoomManager roomManager() {
		return new RoomManager();
	}

	@Bean
	@ConditionalOnMissingBean
	public NotificationRoomManager notificationRoomManager() {
		return new NotificationRoomManager();
	}

	@Bean
	@ConditionalOnMissingBean
	public JsonRpcUserControl userControl() {
		return new JsonRpcUserControl();
	}

	@Bean
	@ConditionalOnMissingBean
	public RoomJsonRpcHandler roomHandler() {
		return new RoomJsonRpcHandler();
	}

	@Override
	public void registerJsonRpcHandlers(JsonRpcHandlerRegistry registry) {
		registry.addHandler(roomHandler().withPingWatchdog(true), "/room");
	}

	@Bean
	public ServletServerContainerFactoryBean createWebSocketContainer() {
		ServletServerContainerFactoryBean container = new ServletServerContainerFactoryBean();
		container.setMaxTextMessageBufferSize(1000000); // chars
		container.setMaxBinaryMessageBufferSize(1000000); // bytes
		return container;
	}

	public static void main(String[] args) throws Exception {
		ConfigurableApplicationContext context = start(args);
		OpenviduConfiguration openviduConf = context.getBean(OpenviduConfiguration.class);
		OpenViduServer.publicUrl = "wss://localhost:" + openviduConf.getServerPort();

		if (openviduConf.getOpenViduPublicUrl().equals("ngrok")) {
			try {
				NgrokController ngrok = new NgrokController();
				System.out.println();
				System.out.println("        PUBLIC IP        ");
				System.out.println("-------------------------");
				System.out.println(ngrok.getNgrokAppUrl());
				System.out.println("-------------------------");
				System.out.println();
				OpenViduServer.publicUrl = ngrok.getNgrokServerUrl().replaceFirst("https://", "wss://");
			} catch (Exception e) {
				System.out.println("   No ngrok connection   ");
			}
		} else if (!openviduConf.getOpenViduPublicUrl().equals("local")) {
			System.out.println("CUSTOM URL");
			OpenViduServer.publicUrl = openviduConf.getOpenViduPublicUrl().replaceFirst("https://", "wss://");
			if (!OpenViduServer.publicUrl.startsWith("wss://")) {
				OpenViduServer.publicUrl = "wss://" + OpenViduServer.publicUrl;
			}
		}
	}

	public static ConfigurableApplicationContext start(String[] args) {
		log.info("Using /dev/urandom for secure random generation");
		System.setProperty("java.security.egd", "file:/dev/./urandom");
		return SpringApplication.run(OpenViduServer.class, args);
	}
}