#include "WebSocket.h"
#include "HttpRequest.h"
#include <iostream>

using namespace Network;
using namespace WebSocket;
using namespace Framework;

__declspec( dllexport ) Frame &Frame::operator+=( const Frame &b ) // baut frames zusammen welche zu einer nachricht gehören
{
    fin = b.fin;
    if( opcode == 0 )
        opcode = b.opcode;
    dataLength += b.dataLength;
    if( dataLength )
    {
        char *data = new char[ dataLength ];
        if( data )
            memcpy( data, this->data, dataLength - b.dataLength );
        if( b.data )
            memcpy( data + dataLength, b.data, b.dataLength );
        delete[] this->data;
        this->data = data;
    }
    else
    {
        delete[] this->data;
        this->data = 0;
    }
    return *this;
}


__declspec( dllexport ) WebSocketClient::WebSocketClient( const char *path, const char *host, unsigned short port )
    : Thread()
{
    queue = new Array< Frame >();
    callback = 0;
    klient = 0;
    this->path = path;
    this->host = host;
    this->port = port;
    lastPingFrame = 0;
    nextClose = 0;
}

WebSocketClient::~WebSocketClient()
{
    disconnect();
    while( queue->getEintragAnzahl() )
    {
        Frame f = queue->get( 0 );
        delete[] f.data;
        queue->remove( 0 );
    }
    queue->release();
}

__declspec( dllexport ) void WebSocketClient::setMessageCallback( std::function< void( WebSocketClient *, __int64 size, const char *data, DataType typ ) > callback )
{
    this->callback = callback;
}

__declspec( dllexport ) bool WebSocketClient::connect()
{
    char allowedKeyChars[] = { 'a', 'b','c','d','e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm',
        'n', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y', 'z', '0', '1', '2', '3',
        '4', '5', '6', '7', '8', '9', 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K',
        'L', 'M', 'N', 'O', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', 0 };
    __int64 numKeyChars = strlen( allowedKeyChars );
    if( !klient )
    {
        Text message = "GET ";
        message += path;
        message += " HTTP/1.1\r\n";
        message += "Host: ";
        message += host;
        message += "\r\nUpgrade: websocket\r\n";
        message += "Connection: Upgrade\r\n";
        char *key = new char[ 25 ];
        key[ 24 ] = 0;
        key[ 23 ] = '=';
        key[ 22 ] = '=';
        for( int i = 0; i < 22; i++ )
            key[ i ] = allowedKeyChars[ (int)( gen.rand() * numKeyChars ) ];
        message += "Sec-WebSocket-Key: ";
        message += key;
        delete[] key;
        message += "\r\nSec-WebSocket-Version: 13\r\n\r\n";
        klient = new Klient();
        if( !klient->verbinde( port, host ) )
        {
            klient = klient->release();
            return 0;
        }
        klient->sende( message, message.getLength() );
        Text answer;
        int br = 0;
        do
        {
            char buff[ 2 ];
            buff[ 1 ] = 0;
            if( klient->getNachricht( buff, 1 ) )
                answer += buff;
            else
                break;
            if( buff[ 0 ] == '\n' )
                br++;
            else if( buff[ 0 ] != '\r' )
                br = 0;
        } while( br < 2 && klient->hatNachricht( 1000 ) );
        HTTP::Answer *handshakeResponse = new HTTP::Answer( answer );
        if( handshakeResponse->getStatusCode() != 101 )
        {
            handshakeResponse->release();
            klient = klient->release();
            return 0;
        }
        handshakeResponse->release();
        start();
        return 1;
    }
    else
        return 1;
}

__declspec( dllexport ) bool WebSocketClient::send( __int64 size, const char *data, DataType typ )
{
    Frame f;
    f.fin = 1;
    f.rsv1 = 0;
    f.rsv2 = 0;
    f.rsv3 = 0;
    f.mask = 1;
    if( typ == TEXT )
        f.opcode = 1;
    else
        f.opcode = 2;
    f.dataLength = size;
    f.data = new char[ f.dataLength ];
    memcpy( f.data, data, f.dataLength );
    f.key[ 0 ] = (unsigned char)(gen.rand() * 256);
    f.key[ 1 ] = (unsigned char)(gen.rand() * 256);
    f.key[ 2 ] = (unsigned char)(gen.rand() * 256);
    f.key[ 3 ] = (unsigned char)(gen.rand() * 256);
    c.lock();
    queue->add( f );
    c.unlock();
    return 1;
}

__declspec( dllexport ) void WebSocketClient::thread()
{
    while( klient )
    {
        c2.lock();
        if( !klient )
        {
            c2.unlock();
            return;
        }
        if( klient->hatNachricht( 100 ) )
        {
            bool ok = 1;
            c2.unlock();
            bool first = 1;
            Frame m;
            unsigned char byte;
            do
            {
                c2.lock();
                if( !klient )
                {
                    c2.unlock();
                    return;
                }
                Frame message;
                ok &= klient->getNachricht( (char*)&byte, 1 );
                message.fin = ( byte & 0x80 ) != 0;
                message.rsv1 = ( byte & 0x40 ) != 0;
                message.rsv2 = ( byte & 0x20 ) != 0;
                message.rsv3 = ( byte & 0x10 ) != 0;
                message.opcode = byte & 0xF;
                ok &= klient->getNachricht( (char*)&byte, 1 );
                message.mask = ( byte & 0x80 ) != 0;
                message.dataLength = byte & 0x7F;
                if( message.dataLength == 126 )
                {
                    ok &= klient->getNachricht( (char*)&byte, 1 );
                    message.dataLength = byte << 8;
                    ok &= klient->getNachricht( (char*)&byte, 1 );
                    message.dataLength |= byte;
                }
                else if( message.dataLength == 127 )
                {
                    ok &= klient->getNachricht( (char*)&byte, 1 );
                    message.dataLength = (__int64)byte << 56;
                    ok &= klient->getNachricht( (char*)&byte, 1 );
                    message.dataLength |= (__int64)byte << 48;
                    ok &= klient->getNachricht( (char*)&byte, 1 );
                    message.dataLength |= (__int64)byte << 40;
                    ok &= klient->getNachricht( (char*)&byte, 1 );
                    message.dataLength |= (__int64)byte << 32;
                    ok &= klient->getNachricht( (char*)&byte, 1 );
                    message.dataLength |= (__int64)byte << 24;
                    ok &= klient->getNachricht( (char*)&byte, 1 );
                    message.dataLength |= (__int64)byte << 16;
                    ok &= klient->getNachricht( (char*)&byte, 1 );
                    message.dataLength |= (__int64)byte << 8;
                    ok &= klient->getNachricht( (char*)&byte, 1 );
                    message.dataLength |= (__int64)byte;
                }
                if( message.mask )
                {
                    ok &= klient->getNachricht( (char*)&byte, 1 );
                    message.key[ 0 ] = byte;
                    ok &= klient->getNachricht( (char*)&byte, 1 );
                    message.key[ 1 ] = byte;
                    ok &= klient->getNachricht( (char*)&byte, 1 );
                    message.key[ 2 ] = byte;
                    ok &= klient->getNachricht( (char*)&byte, 1 );
                    message.key[ 3 ] = byte;
                }
                if( !ok )
                    message.dataLength = 1;
                message.data = 0;
                if( message.dataLength )
                    message.data = new char[ message.dataLength ];
                for( int i = 0; i < message.dataLength; i++ )
                {
                    ok &= klient->getNachricht( (char*)&byte, 1 );
                    if( message.mask )
                        message.data[ i ] = byte ^ message.key[ i % 4 ];
                    else
                        message.data[ i ] = byte;
                }
                c2.unlock();
                if( first )
                    m = message;
                else
                {
                    m += message;
                    delete[] message.data;
                }
                first = 0;
                if( !ok )
                    break;
            } while( !m.fin );
            if( !ok )
            {
                delete[] m.data;
                return;
            }
            if( m.opcode == 0x9 )
            {
                m.opcode = 0xA;
                m.key[ 0 ] = (unsigned char)( gen.rand() * 256 );
                m.key[ 1 ] = (unsigned char)( gen.rand() * 256 );
                m.key[ 2 ] = (unsigned char)( gen.rand() * 256 );
                m.key[ 3 ] = (unsigned char)( gen.rand() * 256 );
                queue->add( m );
            }
            else if( m.opcode == 0xA )
            {
                delete[] m.data;
            }
            else if( m.opcode == 0x8 )
            {
                if( nextClose )
                {
                    delete[] m.data;
                    c2.lock();
                    klient->trenne();
                    klient = klient->release();
                    c2.unlock();
                    return;
                }
                else
                {
                    m.key[ 0 ] = (unsigned char)( gen.rand() * 256 );
                    m.key[ 1 ] = (unsigned char)( gen.rand() * 256 );
                    m.key[ 2 ] = (unsigned char)( gen.rand() * 256 );
                    m.key[ 3 ] = (unsigned char)( gen.rand() * 256 );
                    queue->add( m );
                    nextClose = 1;
                }
            }
            else if( callback )
            {
                callback( this, m.dataLength, m.data, m.opcode == 1 ? TEXT : BINARY );
                delete[] m.data;
            }
        }
        else
        {
            c2.unlock();
            c.lock();
            while( queue->getEintragAnzahl() )
            {
                Frame f = queue->get( 0 );
                c.unlock();
                unsigned char byte = ( f.fin ? 1 : 0 ) << 7;
                byte |= ( f.rsv1 ? 1 : 0 ) << 6;
                byte |= ( f.rsv2 ? 1 : 0 ) << 5;
                byte |= ( f.rsv3 ? 1 : 0 ) << 4;
                byte |= f.opcode & 0xF;
                c2.lock();
                if( !klient )
                {
                    c2.unlock();
                    return;
                }
                klient->sende( (char*)&byte, 1 );
                byte = ( f.mask ? 1 : 0 ) << 7;
                if( f.dataLength < 126 )
                    byte |= (unsigned char)f.dataLength & 0x7F;
                else if( f.dataLength <= 32767 )
                    byte |= 126;
                else
                    byte |= 127;
                klient->sende( (char*)&byte, 1 );
                if( f.dataLength >= 126 )
                {
                    if( f.dataLength <= 32767 )
                    {
                        byte = (unsigned char)( f.dataLength >> 8 );
                        klient->sende( (char*)&byte, 1 );
                        byte = (unsigned char)f.dataLength & 0xFF;
                        klient->sende( (char*)&byte, 1 );
                    }
                    else
                    {
                        byte = (unsigned char)( f.dataLength >> 56 );
                        klient->sende( (char*)&byte, 1 );
                        byte = (unsigned char)( f.dataLength >> 48 );
                        klient->sende( (char*)&byte, 1 );
                        byte = (unsigned char)( f.dataLength >> 40 );
                        klient->sende( (char*)&byte, 1 );
                        byte = (unsigned char)( f.dataLength >> 32 );
                        klient->sende( (char*)&byte, 1 );
                        byte = (unsigned char)( f.dataLength >> 24 );
                        klient->sende( (char*)&byte, 1 );
                        byte = (unsigned char)( f.dataLength >> 16 );
                        klient->sende( (char*)&byte, 1 );
                        byte = (unsigned char)( f.dataLength >> 8 );
                        klient->sende( (char*)&byte, 1 );
                        byte = (unsigned char)f.dataLength & 0xFF;
                        klient->sende( (char*)&byte, 1 );
                    }
                }
                if( f.mask )
                {
                    byte = (unsigned char)f.key[ 0 ];
                    klient->sende( (char*)&byte, 1 );
                    byte = (unsigned char)f.key[ 1 ];
                    klient->sende( (char*)&byte, 1 );
                    byte = (unsigned char)f.key[ 2 ];
                    klient->sende( (char*)&byte, 1 );
                    byte = (unsigned char)f.key[ 3 ];
                    klient->sende( (char*)&byte, 1 );
                }
                if( f.dataLength )
                {
                    for( int i = 0; i < f.dataLength; i++ )
                    {
                        if( f.mask )
                            byte = (unsigned char)f.data[ i ] ^ f.key[ i % 4 ];
                        else
                            byte = (unsigned char)f.data[ i ];
                        klient->sende( (char*)&byte, 1 );
                    }
                }
                c2.unlock();
                delete[] f.data;
                c.lock();
                queue->remove( 0 );
                if( f.opcode == 0x8 )
                {
                    if( nextClose )
                    {
                        c.unlock();
                        c2.lock();
                        klient->trenne();
                        klient = klient->release();
                        c2.unlock();
                        return;
                    }
                    else
                    {
                        nextClose = 1;
                    }
                }
            }
            c.unlock();
        }
    }
}

__declspec( dllexport ) void WebSocketClient::disconnect()
{
    if( !klient )
        return;
    c2.lock();
    klient->trenne();
    klient = klient->release();
    c2.unlock();
    warteAufThread( 1000 );
}

__declspec( dllexport ) bool WebSocketClient::isConnected() const
{
    return klient != 0;
}