package com.xmzs.midjourney.wss;

import com.xmzs.midjourney.ProxyProperties;
import com.neovisionaries.ws.client.ProxySettings;
import com.neovisionaries.ws.client.WebSocketFactory;
import org.apache.logging.log4j.util.Strings;

public interface WebSocketStarter {

	void setTrying(boolean trying);

	void start() throws Exception;

	default WebSocketFactory createWebSocketFactory(ProxyProperties.ProxyConfig proxy) {
		WebSocketFactory webSocketFactory = new WebSocketFactory().setConnectionTimeout(10000);
		if (Strings.isNotBlank(proxy.getHost())) {
			ProxySettings proxySettings = webSocketFactory.getProxySettings();
			proxySettings.setHost(proxy.getHost());
			proxySettings.setPort(proxy.getPort());
		}
		return webSocketFactory;
	}
}