/*
    * Copyright 2020 The Netty Project
    *
    * The Netty Project licenses this file to you 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:
    *
    *   https://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.netty.example.stomp.websocket;
  
  import io.netty.util.AttributeKey;
  import io.netty.util.internal.StringUtil;
  
  import java.util.ArrayList;
  import java.util.List;
  
  public enum StompVersion {
  
      STOMP_V11("1.1", "v11.stomp"),
  
      STOMP_V12("1.2", "v12.stomp");
  
      public static final AttributeKey<StompVersion> CHANNEL_ATTRIBUTE_KEY = AttributeKey.valueOf("stomp_version");
      public static final String SUB_PROTOCOLS;
  
      static {
          List<String> subProtocols = new ArrayList<String>(values().length);
          for (StompVersion stompVersion : values()) {
              subProtocols.add(stompVersion.subProtocol);
          }
  
          SUB_PROTOCOLS = StringUtil.join(",", subProtocols).toString();
      }
  
      private final String version;
      private final String subProtocol;
  
      StompVersion(String version, String subProtocol) {
          this.version = version;
          this.subProtocol = subProtocol;
      }
  
      public String version() {
          return version;
      }
  
      public String subProtocol() {
          return subProtocol;
      }
  
      public static StompVersion findBySubProtocol(String subProtocol) {
          if (subProtocol != null) {
              for (StompVersion stompVersion : values()) {
                  if (stompVersion.subProtocol().equals(subProtocol)) {
                      return stompVersion;
                  }
              }
          }
  
          throw new IllegalArgumentException("Not found StompVersion for '" + subProtocol + "'");
      }
  }