/*-
    * #%L
    * Spring Stomp Server
    * ===============================================================
    * Copyright (C) 2020 Brabenetz Harald, Austria
    * ===============================================================
    * 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.
   * #L%
   */
  package net.brabenetz.app.springstompserver.testtools;
  
  import org.slf4j.Logger;
  import org.slf4j.LoggerFactory;
  import org.springframework.messaging.converter.MappingJackson2MessageConverter;
  import org.springframework.messaging.simp.stomp.StompFrameHandler;
  import org.springframework.messaging.simp.stomp.StompHeaders;
  import org.springframework.web.socket.client.WebSocketClient;
  import org.springframework.web.socket.client.standard.StandardWebSocketClient;
  import org.springframework.web.socket.messaging.WebSocketStompClient;
  import org.springframework.web.socket.sockjs.client.SockJsClient;
  import org.springframework.web.socket.sockjs.client.Transport;
  import org.springframework.web.socket.sockjs.client.WebSocketTransport;
  
  import java.lang.reflect.Type;
  import java.util.ArrayList;
  import java.util.List;
  import java.util.function.BiConsumer;
  
  public class WebSocketUtils {
  
      static final Logger LOG = LoggerFactory.getLogger(WebSocketUtils.class);
  
      public static WebSocketStompClient createStompClient() {
          WebSocketClient webSocketClient = new StandardWebSocketClient();
          WebSocketStompClient stompClient = new WebSocketStompClient(webSocketClient);
          stompClient.setMessageConverter(new MappingJackson2MessageConverter());
          return stompClient;
      }
  
      public static WebSocketStompClient createStompOverSocketJsClient() {
          List<Transport> transports = new ArrayList<>();
          transports.add(new WebSocketTransport(new StandardWebSocketClient()));
          WebSocketClient transport = new SockJsClient(transports);
          WebSocketStompClient stompClient = new WebSocketStompClient(transport);
          stompClient.setMessageConverter(new MappingJackson2MessageConverter());
          return stompClient;
      }
  
      public static StompFrameHandler createStompFrameHandler(BiConsumer<StompHeaders, String> handleFrameConsumer) {
          return new StompFrameHandler() {
  
              @Override
              public Type getPayloadType(StompHeaders headers) {
                  return String.class;
              }
  
              @Override
              public void handleFrame(StompHeaders headers, Object payload) {
                  handleFrameConsumer.accept(headers, (String) payload);
              }
  
          };
      }
  }