import { EventEmitter } from 'events'

import { Keys } from './constants'
import { WebSocketHandlerAdapter } from "../netty"
import { HttpRequestHandler } from './httprequest'
import { TextWebSocketFrameHandler } from './text_websocket_frame'

const CharsetUtil = Java.type('io.netty.util.CharsetUtil')

const HttpServerCodec = Java.type('io.netty.handler.codec.http.HttpServerCodec')
const ChunkedWriteHandler = Java.type('io.netty.handler.stream.ChunkedWriteHandler')
const HttpObjectAggregator = Java.type('io.netty.handler.codec.http.HttpObjectAggregator')
const WebSocketServerProtocolHandler = Java.type('io.netty.handler.codec.http.websocketx.WebSocketServerProtocolHandler')

export class WebSocketHandler extends WebSocketHandlerAdapter {
    private event: EventEmitter;
    constructor(event: EventEmitter) {
        super()
        this.event = event;
    }
    channelRead(ctx: any, msg: any) {
        msg.markReaderIndex()
        let message: string = msg.toString(CharsetUtil.UTF_8)
        let channel = ctx.channel()
        let pipeline = channel.pipeline()
        if (message.indexOf('HTTP/1.1') > 0) {
            pipeline.names().forEach(f => {
                if (f == Keys.Handler || f.indexOf(Keys.Default) > -1) { return }
                pipeline.remove(f)
            })
            pipeline.addLast('http', new HttpServerCodec())
            pipeline.addLast('chunk', new ChunkedWriteHandler())
            pipeline.addLast('httpobj', new HttpObjectAggregator(64 * 1024))
            pipeline.addLast('http_request', new HttpRequestHandler().getHandler())
            pipeline.addLast('websocket', new WebSocketServerProtocolHandler("/ws", true))
            pipeline.addLast('websocket_handler', new TextWebSocketFrameHandler(this.event).getHandler())
        }
        pipeline.remove(Keys.Handler)
        msg.resetReaderIndex()
        ctx.fireChannelRead(msg)
    }
}