#include "../KSGKlient.h"
#include "../KSGServer.h"
#include <Datei.h>
#include "../../Global/Variablen.h"
#include "../Keys.h"

// Inhalt der KartenKlient Klasse aus KSGKlient.h
// Konstruktor
KartenKlient::KartenKlient()
{
    verbunden = 0;
    klient = 0;
    fehler = new Text( "" );
    klientId = getKlientId();
    ref = 1;
}

// Destruktor
KartenKlient::~KartenKlient()
{
    cs.lock();
    if( verbunden )
        trenne();
    fehler = fehler->release();
    cs.unlock();
}

// nicht constant
bool KartenKlient::verbinde( unsigned short port, char *ip ) // verbindet ich mit dem Karten Server
{
    if( klient && verbunden && klient->getServerPort() == port && Text( ip ).istGleich( klient->getServerIp() ) )
        return 1;
    if( klient || verbunden )
        trenne();
    klient = new Klient();
    int keyLen = 0;
    char *key = 0;
    Keys::getServerKey( &key, keyLen, Keys::KARTEN, Keys::SENDEN );
    klient->setSendeKey( key, keyLen );
    delete[] key;
    Keys::getServerKey( &key, keyLen, Keys::KARTEN, Keys::EMPFANGEN );
    klient->setEmpfangKey( key, keyLen );
    delete[] key;
    if( !klient->verbinde( (unsigned short)port, ip ) )
    {
        klient = klient->release();
        fehler->setText( "Fehler beim verbinden mit dem Editor Server." );
        return 0;
    }
    klient->sendeEncrypted( "\1", 1 );
    klient->sendeEncrypted( (char*)&klientId, 4 );
    char ret = 0;
    klient->getNachrichtEncrypted( &ret, 1 );
    if( ret == 3 )
    {
        char l�n = 0;
        klient->getNachrichtEncrypted( &l�n, 1 );
        char *nachricht = new char[ l�n + 1 ];
        nachricht[ l�n ] = 0;
        klient->getNachrichtEncrypted( nachricht, l�n );
        fehler->setText( nachricht );
        delete[]nachricht;
        trenne();
        return 0;
    }
    if( ret == 1 )
    {
        char *sl = 0;
        char slL�n = getSchl�ssel( &sl );
        klient->setSendeKey( sl, slL�n );
        klient->setEmpfangKey( sl, slL�n );
        delete[] sl;
        verbunden = 1;
    }
    return 1;
}

bool KartenKlient::downloadKarte( int id ) // l�d die Karte herunter
{
    cs.lock();
    if( !verbunden )
    {
        cs.unlock();
        return 0;
    }
    klient->sendeEncrypted( "\x6", 1 );
    char ret = 0;
    klient->getNachrichtEncrypted( &ret, 1 );
    if( ret == 1 )
    {
        klient->sendeEncrypted( (char*)&id, 4 );
        klient->getNachrichtEncrypted( &ret, 1 );
        if( ret == 2 )
        { // update
            nachLogin->zNachrichtenListe()->addNachricht( new Text( "Fehler" ), new Text( "Die Karte wird momentan aktualisiert." ), new Text( "Ok" ) );
            cs.unlock();
            return 0;
        }
        else if( ret == 1 )
        {
            int anz = 0;
            klient->getNachrichtEncrypted( (char*)&anz, 4 );
            for( int i = 0; i < anz; i++ )
            {
                char l�n = 0;
                klient->getNachrichtEncrypted( &l�n, 1 );
                if( !l�n )
                    continue;
                char *pf = new char[ l�n + 1 ];
                pf[ l�n ] = 0;
                klient->getNachrichtEncrypted( pf, l�n );
                __int64 gr��e = 0;
                klient->getNachrichtEncrypted( (char*)&gr��e, 8 );
                char *buffer = new char[ 2048 ];
                Text pfad = "data/tmp/Karten/";
                pfad += id;
                pfad += "/spiel";
                pfad += pf;
                delete[] pf;
                Datei d;
                d.setDatei( pfad );
                d.erstellen();
                d.open( Datei::Style::schreiben );
                while( gr��e )
                {
                    int l = gr��e >= 2048 ? 2048 : (int)gr��e;
                    klient->getNachricht( buffer, l );
                    d.schreibe( buffer, l );
                    gr��e -= l;
                }
                d.close();
                delete[] buffer;
            }
        }
    }
    if( ret == 3 )
    {
        char byte = 0;
        klient->getNachrichtEncrypted( &byte, 1 );
        char *f = new char[ byte + 1 ];
        f[ byte ] = 0;
        klient->getNachrichtEncrypted( f, byte );
        fehler->setText( f );
        delete[] f;
        cs.unlock();
        return 0;
    }
    cs.unlock();
    return 1;
}

bool KartenKlient::downloadKarteTitel( int id ) // l�d das Titelbild einer Karte herunter
{
    cs.lock();
    if( !verbunden )
    {
        cs.unlock();
        return 0;
    }
    klient->sendeEncrypted( "\x8", 1 );
    char ret = 0;
    klient->getNachrichtEncrypted( &ret, 1 );
    if( ret == 1 )
    {
        klient->sendeEncrypted( (char*)&id, 4 );
        klient->getNachrichtEncrypted( &ret, 1 );
        if( ret == 2 )
        { // update
            nachLogin->zNachrichtenListe()->addNachricht( new Text( "Fehler" ), new Text( "Die Karte wird momentan aktualisiert." ), new Text( "Ok" ) );
            cs.unlock();
            return 0;
        }
        else if( ret == 1 )
        {
            __int64 gr��e = 0;
            klient->getNachrichtEncrypted( (char*)&gr��e, 8 );
            char *buffer = new char[ 2048 ];
            Text pfad = "data/tmp/Karten/";
            pfad += id;
            pfad += "/titel.ltdb";
            Datei d;
            d.setDatei( pfad );
            d.erstellen();
            d.open( Datei::Style::schreiben );
            while( gr��e )
            {
                int l = gr��e >= 2048 ? 2048 : (int)gr��e;
                klient->getNachricht( buffer, l );
                d.schreibe( buffer, l );
                gr��e -= l;
            }
            d.close();
            delete[] buffer;
        }
    }
    if( ret == 3 )
    {
        char byte = 0;
        klient->getNachrichtEncrypted( &byte, 1 );
        char *f = new char[ byte + 1 ];
        f[ byte ] = 0;
        klient->getNachrichtEncrypted( f, byte );
        fehler->setText( f );
        delete[] f;
        cs.unlock();
        return 0;
    }
    cs.unlock();
    return 1;
}

bool KartenKlient::downloadKarteBeschreibung( int id ) // l�d die Beschreibung einer Karte herunter
{
    cs.lock();
    if( !verbunden )
    {
        cs.unlock();
        return 0;
    }
    klient->sendeEncrypted( "\x9", 1 );
    char ret = 0;
    klient->getNachrichtEncrypted( &ret, 1 );
    if( ret == 1 )
    {
        klient->sendeEncrypted( (char*)&id, 4 );
        klient->getNachrichtEncrypted( &ret, 1 );
        if( ret == 2 )
        { // update
            nachLogin->zNachrichtenListe()->addNachricht( new Text( "Fehler" ), new Text( "Die Karte wird momentan aktualisiert." ), new Text( "Ok" ) );
            cs.unlock();
            return 0;
        }
        else if( ret == 1 )
        {
            __int64 gr��e = 0;
            klient->getNachrichtEncrypted( (char*)&gr��e, 8 );
            char *buffer = new char[ 2048 ];
            Text pfad = "data/tmp/Karten/";
            pfad += id;
            pfad += "/beschreibung.ksgs";
            Datei d;
            d.setDatei( pfad );
            d.erstellen();
            d.open( Datei::Style::schreiben );
            while( gr��e )
            {
                int l = gr��e >= 2048 ? 2048 : (int)gr��e;
                klient->getNachricht( buffer, l );
                d.schreibe( buffer, l );
                gr��e -= l;
            }
            d.close();
            delete[] buffer;
        }
    }
    if( ret == 3 )
    {
        char byte = 0;
        klient->getNachrichtEncrypted( &byte, 1 );
        char *f = new char[ byte + 1 ];
        f[ byte ] = 0;
        klient->getNachrichtEncrypted( f, byte );
        fehler->setText( f );
        delete[] f;
        cs.unlock();
        return 0;
    }
    cs.unlock();
    return 1;
}

bool KartenKlient::downloadKarteMinimap( int id ) // l�d die Minimap einer Karte herunter
{
    cs.lock();
    if( !verbunden )
    {
        cs.unlock();
        return 0;
    }
    klient->sendeEncrypted( "\xA", 1 );
    char ret = 0;
    klient->getNachrichtEncrypted( &ret, 1 );
    if( ret == 1 )
    {
        klient->sendeEncrypted( (char*)&id, 4 );
        klient->getNachrichtEncrypted( &ret, 1 );
        if( ret == 2 )
        { // update
            nachLogin->zNachrichtenListe()->addNachricht( new Text( "Fehler" ), new Text( "Die Karte wird momentan aktualisiert." ), new Text( "Ok" ) );
            cs.unlock();
            return 0;
        }
        else if( ret == 1 )
        {
            __int64 gr��e = 0;
            klient->getNachrichtEncrypted( (char*)&gr��e, 8 );
            char *buffer = new char[ 2048 ];
            Text pfad = "data/tmp/Karten/";
            pfad += id;
            pfad += "/minimap.ltdb";
            Datei d;
            d.setDatei( pfad );
            d.erstellen();
            d.open( Datei::Style::schreiben );
            while( gr��e )
            {
                int l = gr��e >= 2048 ? 2048 : (int)gr��e;
                klient->getNachricht( buffer, l );
                d.schreibe( buffer, l );
                gr��e -= l;
            }
            d.close();
            delete[] buffer;
        }
    }
    if( ret == 3 )
    {
        char byte = 0;
        klient->getNachrichtEncrypted( &byte, 1 );
        char *f = new char[ byte + 1 ];
        f[ byte ] = 0;
        klient->getNachrichtEncrypted( f, byte );
        fehler->setText( f );
        delete[] f;
        cs.unlock();
        return 0;
    }
    cs.unlock();
    return 1;
}

bool KartenKlient::downloadKarteLadebild( int id ) // l�d das Ladebild einer Karte herunter
{
    cs.lock();
    if( !verbunden )
    {
        cs.unlock();
        return 0;
    }
    klient->sendeEncrypted( "\xB", 1 );
    char ret = 0;
    klient->getNachrichtEncrypted( &ret, 1 );
    if( ret == 1 )
    {
        klient->sendeEncrypted( (char*)&id, 4 );
        klient->getNachrichtEncrypted( &ret, 1 );
        if( ret == 2 )
        { // update
            nachLogin->zNachrichtenListe()->addNachricht( new Text( "Fehler" ), new Text( "Die Karte wird momentan aktualisiert." ), new Text( "Ok" ) );
            cs.unlock();
            return 0;
        }
        else if( ret == 1 )
        {
            __int64 gr��e = 0;
            klient->getNachrichtEncrypted( (char*)&gr��e, 8 );
            char *buffer = new char[ 2048 ];
            Text pfad = "data/tmp/Karten/";
            pfad += id;
            pfad += "/ladebild.ltdb";
            Datei d;
            d.setDatei( pfad );
            d.erstellen();
            d.open( Datei::Style::schreiben );
            while( gr��e )
            {
                int l = gr��e >= 2048 ? 2048 : (int)gr��e;
                klient->getNachricht( buffer, l );
                d.schreibe( buffer, l );
                gr��e -= l;
            }
            d.close();
            delete[] buffer;
        }
    }
    if( ret == 3 )
    {
        char byte = 0;
        klient->getNachrichtEncrypted( &byte, 1 );
        char *f = new char[ byte + 1 ];
        f[ byte ] = 0;
        klient->getNachrichtEncrypted( f, byte );
        fehler->setText( f );
        delete[] f;
        cs.unlock();
        return 0;
    }
    cs.unlock();
    return 1;
}

bool KartenKlient::keepAlive() // Erh�lt die Verbindung aufrecht
{
    if( !verbunden )
        return 0;
    char res = 0;
    if( !cs.tryLock() )
        return 1;
    klient->sendeEncrypted( "\x5", 1 );
    klient->getNachrichtEncrypted( &res, 1 );
    cs.unlock();
    if( res != 1 )
    {
        trenne();
        fehler->setText( "Verbindung unterbrochen: 'Keep Alive' nicht erfolgreich." );
    }
    return res == 1;
}

bool KartenKlient::trenne() // trennt sich von dem Editor Server
{
    if( !klient || !verbunden )
    {
        if( klient )
            klient = klient->release();
        return 1;
    }
    char serverReturn;
    klient->sendeEncrypted( "\4", 1 );
    klient->getNachrichtEncrypted( &serverReturn, 1 );
    if( serverReturn == 3 )
    {
        char l�n = 0;
        klient->getNachrichtEncrypted( &l�n, 1 );
        char *nachricht = new char[ l�n + 1 ];
        nachricht[ l�n ] = 0;
        klient->getNachrichtEncrypted( nachricht, l�n );
        delete[]nachricht;
    }
    klient->sendeEncrypted( "\3", 1 );
    klient->getNachrichtEncrypted( &serverReturn, 1 );
    klient->trenne();
    klient = klient->release();
    return 1;
}

// constant
bool KartenKlient::istVerbunden() const // pr�ft, ob mit Editor Server verbunden
{
    return verbunden;
}

char *KartenKlient::getLetzterFehler() const // gibt den Letzten Fehlertext zu�ck
{
    return fehler->getText();
}

// Reference Counting
KartenKlient *KartenKlient::getThis()
{
    ref++;
    return this;
}

KartenKlient *KartenKlient::release()
{
    ref--;
    if( !ref )
        delete this;
    return 0;
}