/*
 * Copyright (c) 2014, 2020 Oracle and/or its affiliates and others.
 * All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v. 2.0, which is available at
 * http://www.eclipse.org/legal/epl-2.0.
 *
 * This Source Code may also be made available under the following Secondary
 * Licenses when the conditions for such availability set forth in the
 * Eclipse Public License v. 2.0 are satisfied: GNU General Public License,
 * version 2 with the GNU Classpath Exception, which is available at
 * https://www.gnu.org/software/classpath/license.html.
 *
 * SPDX-License-Identifier: EPL-2.0 OR GPL-2.0 WITH Classpath-exception-2.0
 */

package com.sun.ts.tests.websocket.ee.jakarta.websocket.session11.server;

import java.util.HashSet;
import java.util.LinkedList;
import java.util.List;
import java.util.Set;

import com.sun.ts.tests.websocket.common.stringbean.StringBeanBinaryDecoder;
import com.sun.ts.tests.websocket.common.stringbean.StringBeanBinaryStreamDecoder;
import com.sun.ts.tests.websocket.common.stringbean.StringBeanTextDecoder;
import com.sun.ts.tests.websocket.common.stringbean.StringBeanTextStreamDecoder;
import com.sun.ts.tests.websocket.ee.jakarta.websocket.session11.common.LinkedListHashSetTextDecoder;
import com.sun.ts.tests.websocket.ee.jakarta.websocket.session11.common.StringListTextDecoder;
import com.sun.ts.tests.websocket.ee.jakarta.websocket.session11.common.TypeEnum;

import jakarta.websocket.Decoder;
import jakarta.websocket.Endpoint;
import jakarta.websocket.server.ServerApplicationConfig;
import jakarta.websocket.server.ServerEndpointConfig;

public class AppConfig implements ServerApplicationConfig {

  @Override
  public Set<ServerEndpointConfig> getEndpointConfigs(
      Set<Class<? extends Endpoint>> endpointClasses) {
    Set<ServerEndpointConfig> set = new HashSet<>();
    addServerEndpoint(set, LinkedListHashSetTextDecoder.class,
        TypeEnum.LINKEDLIST_HASHSET_TEXT);
    addServerEndpoint(set, StringListTextDecoder.class, TypeEnum.LIST_TEXT);
    addServerEndpoint(set, StringBeanTextDecoder.class, TypeEnum.STRINGBEAN);
    addServerEndpoint(set, StringBeanTextStreamDecoder.class,
        TypeEnum.STRINGBEANSTREAM);
    addServerEndpoint(set, StringBeanBinaryDecoder.class,
        TypeEnum.STRINGBEANBINARY);
    addServerEndpoint(set, StringBeanBinaryStreamDecoder.class,
        TypeEnum.STRINGBEANBINARYSTREAM);
    addServerEndpoint(set, null, TypeEnum.STRING_WHOLE);
    addServerEndpoint(set, null, TypeEnum.STRING_PARTIAL);
    addServerEndpoint(set, null, TypeEnum.PONG);
    addServerEndpoint(set, null, TypeEnum.BYTEBUFFER_WHOLE);
    addServerEndpoint(set, null, TypeEnum.BYTEARRAY_WHOLE);
    addServerEndpoint(set, null, TypeEnum.BYTEBUFFER_PARTIAL);
    addServerEndpoint(set, null, TypeEnum.BYTEARRAY_PARTIAL);
    addServerEndpoint(set, null, TypeEnum.INPUTSTREAM);
    addServerEndpoint(set, null, TypeEnum.READER);
    return set;
  }

  private Set<ServerEndpointConfig> addServerEndpoint(
      Set<ServerEndpointConfig> set, Class<? extends Decoder> decoder,
      final TypeEnum typeEnum) {
    List<Class<? extends Decoder>> decoders = new LinkedList<>();
    decoders.add(decoder);
    ServerEndpointConfig.Configurator configurator = new ServerEndpointConfig.Configurator() {
      @SuppressWarnings("unchecked")
      @Override
      public <T> T getEndpointInstance(Class<T> endpointClass)
          throws InstantiationException {
        return (T) new WSCServerEndpoint(typeEnum);
      }
    };
    ServerEndpointConfig.Builder builder = ServerEndpointConfig.Builder
        .create(WSCServerEndpoint.class, "/" + typeEnum.name().toLowerCase());
    if (decoder != null)
      builder = builder.decoders(decoders);
    ServerEndpointConfig config = builder.configurator(configurator).build();
    set.add(config);
    return set;
  }

  @Override
  public Set<Class<?>> getAnnotatedEndpointClasses(Set<Class<?>> scanned) {
    return scanned;
  }
}
