/*
 * Copyright (c) 2009 The openGion Project.
 *
 * 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 org.opengion.hayabusa.servlet;

import jakarta.websocket.OnOpen;
import jakarta.websocket.OnClose;
import jakarta.websocket.OnMessage;
import jakarta.websocket.OnError;
import jakarta.websocket.Session;
import jakarta.websocket.EndpointConfig;
import jakarta.websocket.CloseReason;
import jakarta.websocket.server.ServerEndpoint;
import java.io.IOException;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.nio.ByteBuffer;
import java.awt.image.BufferedImage;
import javax.imageio.ImageIO;
import java.util.Set;
import java.util.Collections;
import java.util.concurrent.ConcurrentHashMap;

/**
 * http://enterprisegeeks.hatenablog.com/entry/2015/12/17/104815
 *
 * WebSocketBasicEndpoint.java のソースを参照しています。
 *
 * 設定が、いくつか必要です。
 *   ① /wsdemo をサーバーエンドポイントのURLにしているため
 *      WEB-INF/web.xml の security-constraint の web-resource-collection の
 *      url-pattern に、/wsdemo を追加する必要がある。
 *        <url-pattern>/wsdemo/*</url-pattern>
 *
 *   ② コンパイル時(build.xml)のクラスパスの設定に、
 *      <pathelement path="${env.CATALINA_HOME}/lib/websocket-api.jar" />
 *      を追加する必要がある。
 */
@ServerEndpoint("/wsdemo")
public class WebSocketDemo {

        /** クライアントからの全接続を保持するセット */
        /**
         * クライアントからの全接続を保持するセット
         *
         * @og.rev 6.8.5.0 (2018/01/09) PMD Variables that are final and static should be all capitals。sessSET → SESS_SET
         */
        // 8.5.4.2 (2024/01/12) PMD 7.0.0 UseDiamondOperator 対応
//      private static final Set<Session> SESS_SET = Collections.newSetFromMap( new ConcurrentHashMap<Session, Boolean>() );
        private static final Set<Session> SESS_SET = Collections.newSetFromMap( new ConcurrentHashMap<>() );

        /**
         * デフォルトコンストラクター
         *
         */
        public WebSocketDemo() { super(); }             // これも、自動的に呼ばれるが、空のメソッドを作成すると警告されるので、明示的にしておきます。

        /**
         * 2.クライアントからの接続時にコールされる。
         *
         * 引数は以下が設定可能だが、メソッド内で使用しないなら省略できる。
         *
         * @og.rev 6.8.5.0 (2018/01/09) PMD Variables that are final and static should be all capitals。sessSET → SESS_SET
         *
         * @param client クライアントの接続情報
         * @param config 設定情報
         */
        @OnOpen
        public void onOpen( final Session client, final EndpointConfig config ) {
                System.out.println( client.getId() + " was connected." );
                printSession( client );

                SESS_SET.add( client );
        }

        /**
         * 3.クライアントの切断時にコールされる
         *
         * 引数は使用しなければ省略可能。
         *
         * @og.rev 6.8.5.0 (2018/01/09) PMD Variables that are final and static should be all capitals。sessSET → SESS_SET
         *
         * @param client 接続
         * @param reason 切断理由
         */
        @OnClose
        public void onClose( final Session client, final CloseReason reason ) {
                System.out.println( client.getId() + " was closed by "
                                                                +               reason.getCloseCode()
                                                                + "[" + reason.getCloseCode().getCode()+"]" );
                printSession( client );

                SESS_SET.remove( client );
        }

        /**
         * 4.エラー時にコールされる。
         *
         * 引数は使用しなければ省略可能。
         * @param client クライアント接続
         * @param error エラー
         */
        @OnError
        public void onError( final Session client, final Throwable error ) {
                System.out.println( client.getId() + " was error." );
                error.printStackTrace();
        }

        /**
         * 5.テキストメッセージ受信時の処理
         *
         * 全クライアントにメッセージを送信する。（サンプル）
         *
         * 引数は使用しなければ省略可能。
         * @param text クライアントから送信されたテキスト
         * @param client 接続情報
         * @throws IOException  なんらかの入出力例外の発生を通知するシグナルを発生させます。
         */
        @OnMessage
        public void onMessage( final String text, final Session client ) throws IOException {
                System.out.println( client.getId() + " was message." + text );

        //      // sessionが持つendPointに紐付く session を取り出す。
        //      for( final Session other : client.getOpenSessions() ) {
        //              other.getBasicRemote().sendText( text );
        //      }
                broadCast( text );              // Setで管理しているopenされたsession (元とかぶる)
        }

        /**
         * すべてのクライアントにメッセージを送信する。
         *
         * @og.rev 6.8.5.0 (2018/01/09) PMD Variables that are final and static should be all capitals。sessSET → SESS_SET
         *
         * @param text クライアントに送信するテキスト
         */
        private static void broadCast( final String text ) {
                for( final Session client : SESS_SET) {
                        client.getOpenSessions().forEach( sess -> sess.getAsyncRemote().sendText( text ) );
                }
        }

        /**
         * 6.バイナリ受信時の処理
         *
         * 送信元に画像を変換して送り返す。
         *
         * 引数は使用しなければ省略可能。
         * @param buf クライアントから送信されたバイナリ
         * @param client 接続情報
         * @throws IOException  なんらかの入出力例外の発生を通知するシグナルを発生させます。
         */
        @OnMessage
        public void onMessage( final ByteBuffer buf, final Session client ) throws IOException {
                client.getBasicRemote().sendBinary( grayScall( buf ) );
        }

        /**
         * 7.画像をグレースケールに変換する。本筋とは関係ない。
         *
         * @param input 引数のバイトバッファー
         * @return グレースケールに変換されたバイトバッファー
         * @throws IOException 入出力ｴﾗｰ
         */
        private ByteBuffer grayScall( final ByteBuffer input ) throws IOException {
                final BufferedImage img = ImageIO.read( new ByteArrayInputStream( input.array() ) );
                final BufferedImage glay = new BufferedImage( img.getWidth(), img.getHeight(), BufferedImage.TYPE_BYTE_GRAY );
                glay.getGraphics().drawImage( img, 0, 0, null );
                final ByteArrayOutputStream bos = new ByteArrayOutputStream();
                ImageIO.write( glay, "png", bos );

                return ByteBuffer.wrap( bos.toByteArray() );
        }

        /**
         * Session のセッションを標準出力に表示します。
         *
         * デバッグ用の仮メソッドです。
         *
         * @param session セッションオブジェクト
         */
        private void printSession( final Session session ) {
                System.out.println( " ID    = " + session.getId() );
        //      System.out.println( " Query = " + session.getQueryString() );           // 引数部分のみ
                System.out.println( " URI   = " + session.getRequestURI() );            // URI すべて

                System.out.println( " RequestParameter : " );                                           // 引数部分を、展開して、Map化
                session.getRequestParameterMap().forEach( (k,v) -> {
                                System.out.println( "  Key=" + k );
                                System.out.println( "  Val=" + java.util.Arrays.toString( v.toArray( new String[v.size()] ) ) );
                        }
                );
        }
}