dlgupdate.cpp

/***************************************************************************
                           dlgupdate.cpp
                           -------------------
    begin                : 2004-10-30 12:45
    copyright            : (C) 2004-2005 by Dominik Haumann
    email                : dhaumann@users.sourceforge.net
 ***************************************************************************/

/***************************************************************************
 This program is free software; you can redistribute it and/or
 modify it under the terms of the GNU General Public License
 as published by the Free Software Foundation; either version 2
 of the License, or (at your option) any later version.

 This program is distributed in the hope that it will be useful,
 but WITHOUT ANY WARRANTY; without even the implied warranty of
 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 GNU General Public License for more details.

 You should have received a copy of the GNU General Public License
 along with this program; if not, write to the Free Software
 Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 ***************************************************************************/

#include "dlgupdate.h"
#include "profilemanager.h"
#include "md5.h"
#include "parser.h"

#include <sstream>

#include <wx/file.h>
#include <wx/radiobut.h>

namespace WDS
{

BEGIN_EVENT_TABLE(CDlgUpdate, wxDialog)
    EVT_BUTTON( wxID_OK, CDlgUpdate::OnOK )
    EVT_BUTTON( wxID_CANCEL, CDlgUpdate::OnCancel )
    EVT_TIMER( IDT_TIMER, CDlgUpdate::OnTimer )
END_EVENT_TABLE()
/*----------------------------------------------------------------------------/
  Constructor / Destructor
-----------------------------------------------------------------------------*/

CDlgUpdate::CDlgUpdate( wxWindow *pParent, CProfile* profile )
    : wxDialog(pParent, -1, "Update von phpmap - WDSMap",
    wxDefaultPosition, wxSize(400, 300), wxDEFAULT_DIALOG_STYLE | wxRESIZE_BORDER ),
    m_timer(this, IDT_TIMER), m_pProfile(profile)
{
    //BEGIN Sizer
    // contains all three containers
    wxBoxSizer *sTop   = new wxBoxSizer( wxVERTICAL );

    // Phpmap GroupBox
    wxFlexGridSizer *sGrid = new wxFlexGridSizer( 2, 2, 5, 5 );
    sGrid->AddGrowableCol(1);

    // are in Profile GroupBox
    wxBoxSizer *sDlAvg = new wxBoxSizer( wxHORIZONTAL );
    // Buttons: Close, Cancel
    wxBoxSizer *sButtons = new wxBoxSizer( wxHORIZONTAL );
    //END Sizer

    //BEGIN Controls
    wxStaticText *pLblServer = new wxStaticText( this, -1, "Server:" );
    wxTextCtrl   *pEdtSource = new wxTextCtrl  ( this, -1, m_pProfile->GetServer(),
            wxDefaultPosition, wxDefaultSize, wxTE_READONLY );

    wxStaticText *pLblProgress = new wxStaticText( this, -1, "Progress:" );
    m_pProgessBar = new wxGauge( this, IDC_PROGRESS_BAR, 10 );

    m_pLblTransferred = new wxStaticText( this, -1, "Total: 0 bytes übertragen." );
    m_pLblAverageSpeed= new wxStaticText( this, -1, "00.0 kb/s" );
    m_pLblStatus      = new wxStaticText( this, -1, "Status: Verbindung zum Server herstellen." );

    wxStaticLine *pLine = new wxStaticLine( this, -1 );

    m_pBtnCancel = new wxButton( this, wxID_CANCEL, "Abbruch" );
    m_pBtnOK     = new wxButton( this, wxID_OK, "OK" );
    //END Controls


    //BEGIN sizer setup
    sGrid->Add( pLblServer,   0, wxALIGN_CENTER_VERTICAL | wxTOP | wxLEFT, 5 );
    sGrid->Add( pEdtSource,   0, wxEXPAND | wxTOP | wxRIGHT, 5 );
    sGrid->Add( pLblProgress, 0, wxALIGN_CENTER_VERTICAL | wxLEFT, 5 );
    sGrid->Add( m_pProgessBar,0, wxEXPAND | wxRIGHT, 5 );

    sDlAvg->Add( m_pLblTransferred , 1, wxALIGN_CENTER_VERTICAL | wxLEFT | wxRIGHT, 5 );
    sDlAvg->Add( m_pLblAverageSpeed, 0, wxALIGN_RIGHT | wxRIGHT, 5 );

    sButtons->Add( m_pBtnCancel, 0, wxALIGN_RIGHT | wxRIGHT, 5 );
    sButtons->Add( m_pBtnOK, 0, wxALIGN_RIGHT, 0 );

    sTop->Add( sGrid , 0, wxEXPAND | wxALL, 5 );
    sTop->Add( sDlAvg, 0, wxEXPAND | wxLEFT | wxRIGHT | wxBOTTOM, 5 );
    sTop->Add( m_pLblStatus, 0, wxEXPAND | wxALIGN_CENTER_VERTICAL | wxLEFT | wxRIGHT, 10 );
    sTop->Add( pLine, 0, wxEXPAND | wxALIGN_CENTER_VERTICAL | wxALL, 5 );
    sTop->Add( sButtons, 0, wxALIGN_RIGHT | wxLEFT | wxRIGHT | wxBOTTOM, 10 );
    //END sizer setup

    SetAutoLayout( true );
    SetSizer( sTop );

    sTop->SetSizeHints( this );
    SetClientSize( 400, GetSize().GetHeight() );

    m_pBtnOK->SetFocus( );
    m_pBtnOK->SetDefault( );
    m_pBtnOK->Enable( false );

    // set state, httprequester and start timer
    m_state = US_INIT;
    m_httpRequest = new HTTP::GetRequest;
    m_timer.Start( 50 );
}

CDlgUpdate::~CDlgUpdate()
{
    if ( m_httpRequest )
        delete m_httpRequest;
  // all widgets are automatically deleted.
}

/*----------------------------------------------------------------------------/
  Member Functions
-----------------------------------------------------------------------------*/

const HTTP::GetRequest* CDlgUpdate::GetHTTPRequest()
{
    return m_httpRequest;
}

//
//BEGIN EVENT HANDLER
//
void CDlgUpdate::OnTimer( wxTimerEvent& event )
{
    wxString url, sport;
    //long port;
    static wxStopWatch sw;
    static bool bGotSize = false;

    switch ( m_state ) {

    case US_INIT:
        m_httpRequest->setUserAgent( string("WDSMap/") + WDSMAP_VERSION );
        m_httpRequest->addHeader( "WDSMap-User", m_pProfile->GetUser().c_str() );
        m_httpRequest->addHeader( "WDSMap-Password", MD5::hash( m_pProfile->GetPassword().c_str() ) );
        m_httpRequest->addHeader( "WDSMap-Type", "update" );
        m_httpRequest->addHeader( "WDSMap-Version", 1 );
        m_httpRequest->addHeader( "WDSMap-Timestamp", m_pProfile->GetTimestamp() );

        bGotSize = false;

        cout << "INIT" << endl;
        m_state = US_CONNECT;
        break;

    case US_CONNECT:
        cout << "CONNECT" << endl;
        url   = m_pProfile->GetServer();
        sport = url.AfterFirst(':');
        long nport;
        (sport.BeforeFirst('/')).ToLong(&nport);

        if ( m_httpRequest->connect( (url.BeforeFirst(':')).c_str(), int(nport) ) ) {
            m_pLblStatus->SetLabel( "Status: Verbindung zum Server herstellen." );
            m_state = US_REQUEST;
        } else {
            m_pLblStatus->SetLabel( "Fehler: Verbindung zum Server kann nicht hergestellt werden." );
            m_state = US_ERROR;
        }
        break;

    case US_REQUEST:
        cout << "REQUEST" << endl;
        url  = m_pProfile->GetServer().AfterFirst('/');
        url += "export.php";

        if ( m_httpRequest->request( url.c_str() ) ) {
            sw.Start( 0 );
            m_pLblStatus->SetLabel( "Status: Daten werden übertragen." );
            m_state = US_PROCESS_DATA;
        } else {
            m_pLblStatus->SetLabel( "Fehler: Bei der Datenübertragung ist ein Fehler aufgetreten." );
            m_state = US_ERROR;
        }
        break;

    case US_PROCESS_DATA:
        cout << "PROCESS" << endl;
        if ( ! m_httpRequest->process() ) m_state = US_MAYBE_SUCCESS;

        if ( ! bGotSize ) {
            // ok, read "Update-Size: <size>;" from http-header
            string rawData = m_httpRequest->getRaw();
            unsigned int pos = rawData.find( "\r\nphpmap-DataSize: " );

            if ( pos != string::npos ) {
                pos += 19;
                // assume "Update-Size: <size>;" already transferred, which is very much the case
                int end = rawData.find( "\r\n", pos );
                if ( end != string::npos) {
                    int size;
                    stringstream ss;
                    ss << rawData.substr(pos, end - pos);
                    ss >> size;
                    m_pProgessBar->SetRange( size );
                    bGotSize = true;
                }
            }
        }
        m_pLblAverageSpeed->SetLabel(
            wxString::Format("%.1f kb/s", double (m_httpRequest->getRawSize())
            / (1024.0 * double(sw.Time()/1000.0))) );

        m_pLblTransferred->SetLabel( wxString::Format( "Total: %d bytes übertragen.",
                                                      m_httpRequest->getRawSize() ) );

        m_pProgessBar->SetValue( m_httpRequest->getRawSize() );
        break;

    case US_MAYBE_SUCCESS:
        cout << "MAYBESUCCESS" << endl;
        if ( ! m_httpRequest->success() )
        {
            m_pLblStatus->SetLabel( "Fehler: Unbekannter Fehler aufgetreten." );
            m_state = US_ERROR;
        }
        else
        {
//          cout << m_httpRequest->getRaw() << endl;
            m_timer.Stop();
            sw.Pause();

            m_pProgessBar->SetRange( 10 );
            m_pProgessBar->SetValue( 10 );
            if ( CheckDownload() ) {
                m_pLblStatus->SetLabel( "Status: Fertig." );
                m_pBtnOK->Enable( true );
            }
        }
        break;

    case US_ERROR:
        cout << m_httpRequest->getError() << endl;
        m_timer.Stop();
        sw.Pause();
        break;

    default: break;
        // should not happen
    }
}

bool CDlgUpdate::CheckDownload( )
{
    if ( m_httpRequest->getHeaderAsInt("phpmap-ReturnCode") == 0 ) {
        // success

        int version = m_httpRequest->getHeaderAsInt( "phpmap-Version" );
        // test version
        if ( version != 1 ) {
            m_pLblStatus->SetLabel( wxString("Fehler: Falsche Version (")
                    + m_httpRequest->getHeaderAsString("phpmap-Version").c_str()
                    + ")." );
            return false;
        }

        int timestamp = m_httpRequest->getHeaderAsInt( "phpmap-Timestamp" );
        // set new timestamp
        // FIXME: uncomment this before release!
        //m_pProfile->SetTimestamp( timestamp );

        m_strUpdate = m_httpRequest->getData();

        return true;
    } else {
        m_pLblStatus->SetLabel( wxString("Fehler: ")
                + m_httpRequest->getHeaderAsString("phpmap-ErrorMessage").c_str()
                + " (#"
                + m_httpRequest->getHeaderAsString("phpmap-ReturnCode").c_str()
                + ")." );
    }

    return false;
}

void CDlgUpdate::OnCancel( wxCommandEvent &event )
{
    m_timer.Stop();
    EndModal( wxID_CANCEL );

    event.Skip( );
}

void CDlgUpdate::OnOK( wxCommandEvent &event )
{
    if ( m_httpRequest ) {
        if( m_httpRequest->success() ) {
            EndModal( wxID_OK );
        } else {
            EndModal( wxID_CANCEL );
        }
    }

    event.Skip( );
}
//
//END EVENT HANDLER
//

}

// kate: space-indent off; tab-width 4; replace-tabs off;

---