2011-03-17 01:47:30 -04:00
|
|
|
#include "global.h"
|
|
|
|
|
|
|
|
|
|
#if !defined(WITHOUT_NETWORKING)
|
|
|
|
|
#include "ScreenPackages.h"
|
|
|
|
|
#include "GameConstantsAndTypes.h"
|
|
|
|
|
#include "GameState.h"
|
|
|
|
|
#include "ThemeManager.h"
|
|
|
|
|
#include "RageDisplay.h"
|
|
|
|
|
#include "RageLog.h"
|
|
|
|
|
#include "RageUtil.h"
|
|
|
|
|
#include "RageFile.h"
|
|
|
|
|
#include "ScreenTextEntry.h"
|
|
|
|
|
#include "ScreenManager.h"
|
|
|
|
|
#include <cstdlib>
|
|
|
|
|
#include "LocalizedString.h"
|
|
|
|
|
|
|
|
|
|
AutoScreenMessage( SM_BackFromURL );
|
|
|
|
|
|
|
|
|
|
REGISTER_SCREEN_CLASS( ScreenPackages );
|
|
|
|
|
|
|
|
|
|
static LocalizedString VISIT_URL( "ScreenPackages", "VisitURL" );
|
|
|
|
|
void ScreenPackages::Init()
|
|
|
|
|
{
|
|
|
|
|
EXISTINGBG_WIDTH.Load(m_sName, "PackagesBGWidth");
|
|
|
|
|
WEBBG_WIDTH.Load(m_sName, "WebBGWidth");
|
|
|
|
|
NUM_PACKAGES_SHOW.Load(m_sName, "NumPackagesShow");
|
|
|
|
|
NUM_LINKS_SHOW.Load(m_sName, "NumLinksShow");
|
|
|
|
|
DEFAULT_URL.Load(m_sName, "DefaultUrl");
|
|
|
|
|
ScreenWithMenuElements::Init();
|
|
|
|
|
|
|
|
|
|
m_iPackagesPos = 0;
|
|
|
|
|
m_iLinksPos = 0;
|
|
|
|
|
m_iDLorLST = 0;
|
|
|
|
|
m_bIsDownloading = false;
|
|
|
|
|
m_bCanDL = THEME->GetMetricB( m_sName, "CanDL" );
|
|
|
|
|
m_sStatus = "";
|
|
|
|
|
m_fLastUpdate = 0;
|
|
|
|
|
m_iTotalBytes = 0;
|
|
|
|
|
m_iDownloaded = 0;
|
|
|
|
|
|
|
|
|
|
FOREACH_PlayerNumber( pn )
|
|
|
|
|
GAMESTATE->JoinPlayer( pn );
|
|
|
|
|
|
|
|
|
|
m_sprExistingBG.Load( THEME->GetPathG( m_sName, "PackagesBG" ) );
|
|
|
|
|
m_sprExistingBG->SetName( "PackagesBG" );
|
|
|
|
|
LOAD_ALL_COMMANDS_AND_SET_XY_AND_ON_COMMAND( m_sprExistingBG );
|
|
|
|
|
this->AddChild( m_sprExistingBG );
|
|
|
|
|
|
|
|
|
|
m_sprWebBG.Load( THEME->GetPathG( m_sName, "WebBG" ) );
|
|
|
|
|
m_sprWebBG->SetName( "WebBG" );
|
|
|
|
|
LOAD_ALL_COMMANDS_AND_SET_XY_AND_ON_COMMAND( m_sprWebBG );
|
|
|
|
|
this->AddChild( m_sprWebBG );
|
|
|
|
|
|
|
|
|
|
COMMAND( m_sprExistingBG, "Back" );
|
|
|
|
|
COMMAND( m_sprWebBG, "Away" );
|
|
|
|
|
|
|
|
|
|
// m_fOutputFile.
|
|
|
|
|
m_textPackages.LoadFromFont( THEME->GetPathF(m_sName,"default") );
|
|
|
|
|
m_textPackages.SetShadowLength( 0 );
|
|
|
|
|
m_textPackages.SetName( "Packages" );
|
|
|
|
|
m_textPackages.SetMaxWidth( EXISTINGBG_WIDTH );
|
|
|
|
|
LOAD_ALL_COMMANDS_AND_SET_XY_AND_ON_COMMAND( m_textPackages );
|
|
|
|
|
this->AddChild( &m_textPackages );
|
|
|
|
|
RefreshPackages();
|
|
|
|
|
|
|
|
|
|
m_textWeb.LoadFromFont( THEME->GetPathF(m_sName,"default") );
|
|
|
|
|
m_textWeb.SetShadowLength( 0 );
|
|
|
|
|
m_textWeb.SetName( "Web" );
|
|
|
|
|
m_textWeb.SetMaxWidth( WEBBG_WIDTH );
|
|
|
|
|
LOAD_ALL_COMMANDS_AND_SET_XY_AND_ON_COMMAND( m_textWeb );
|
|
|
|
|
this->AddChild( &m_textWeb);
|
|
|
|
|
m_Links.push_back( " " ); // what is this here for? -aj
|
|
|
|
|
m_LinkTitles.push_back( VISIT_URL );
|
|
|
|
|
|
|
|
|
|
m_textURL.LoadFromFont( THEME->GetPathF( m_sName,"default") );
|
|
|
|
|
m_textURL.SetShadowLength( 0 );
|
|
|
|
|
m_textURL.SetName( "WebURL" );
|
|
|
|
|
LOAD_ALL_COMMANDS_AND_SET_XY_AND_ON_COMMAND( m_textURL );
|
|
|
|
|
this->AddChild( &m_textURL );
|
|
|
|
|
UpdateLinksList();
|
|
|
|
|
|
|
|
|
|
m_sprWebSel.SetName( "WebSel" );
|
|
|
|
|
m_sprWebSel.Load( THEME->GetPathG( m_sName, "WebSel" ) );
|
|
|
|
|
LOAD_ALL_COMMANDS_AND_SET_XY_AND_ON_COMMAND( m_sprWebSel );
|
|
|
|
|
this->AddChild( &m_sprWebSel );
|
|
|
|
|
|
|
|
|
|
m_sprDLBG.SetName( "Download" );
|
|
|
|
|
m_sprDLBG.Load( THEME->GetPathG( m_sName, "DownloadBG" ) );
|
|
|
|
|
LOAD_ALL_COMMANDS_AND_SET_XY_AND_ON_COMMAND( m_sprDLBG );
|
|
|
|
|
this->AddChild( &m_sprDLBG );
|
|
|
|
|
|
|
|
|
|
//(HTTP ELEMENTS)
|
|
|
|
|
m_sprDL.SetName( "Download" );
|
|
|
|
|
m_sprDL.Load( THEME->GetPathG( m_sName, "Download" ) );
|
|
|
|
|
LOAD_ALL_COMMANDS_AND_SET_XY_AND_ON_COMMAND( m_sprDL );
|
|
|
|
|
this->AddChild( &m_sprDL );
|
|
|
|
|
m_sprDL.RunCommands( THEME->GetMetricA( m_sName, "DownloadProgressCommand" ) );
|
|
|
|
|
|
|
|
|
|
m_textStatus.LoadFromFont( THEME->GetPathF(m_sName,"default") );
|
|
|
|
|
m_textStatus.SetShadowLength( 0 );
|
|
|
|
|
m_textStatus.SetName( "DownloadStatus" );
|
|
|
|
|
LOAD_ALL_COMMANDS_AND_SET_XY_AND_ON_COMMAND( m_textStatus );
|
|
|
|
|
this->AddChild( &m_textStatus );
|
|
|
|
|
|
|
|
|
|
// if the default url isn't empty, load it.
|
|
|
|
|
if( !DEFAULT_URL.GetValue().empty() )
|
|
|
|
|
EnterURL( DEFAULT_URL );
|
|
|
|
|
|
|
|
|
|
UpdateProgress();
|
|
|
|
|
|
|
|
|
|
// Workaround: For some reason, the first download sometimes
|
|
|
|
|
// corrupts; by opening and closing the RageFile, this
|
|
|
|
|
// problem does not occur. Go figure?
|
|
|
|
|
|
|
|
|
|
// XXX: This is a really dirty work around!
|
|
|
|
|
// Why does RageFile do this?
|
|
|
|
|
|
|
|
|
|
// It's always some strange number of bytes at the end of the
|
|
|
|
|
// file when it corrupts.
|
|
|
|
|
m_fOutputFile.Open( "Packages/dummy.txt", RageFile::WRITE );
|
|
|
|
|
m_fOutputFile.Close();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void ScreenPackages::HandleScreenMessage( const ScreenMessage SM )
|
|
|
|
|
{
|
|
|
|
|
if( SM == SM_GoToPrevScreen )
|
|
|
|
|
{
|
|
|
|
|
SCREENMAN->SetNewScreen( THEME->GetMetric (m_sName, "PrevScreen") );
|
|
|
|
|
}
|
|
|
|
|
else if( SM ==SM_GoToNextScreen )
|
|
|
|
|
{
|
|
|
|
|
SCREENMAN->SetNewScreen( THEME->GetMetric (m_sName, "NextScreen") );
|
|
|
|
|
}
|
|
|
|
|
else if( SM == SM_BackFromURL )
|
|
|
|
|
{
|
|
|
|
|
if ( !ScreenTextEntry::s_bCancelledLast )
|
|
|
|
|
EnterURL( ScreenTextEntry::s_sLastAnswer );
|
|
|
|
|
}
|
|
|
|
|
ScreenWithMenuElements::HandleScreenMessage( SM );
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static LocalizedString DOWNLOAD_PROGRESS( "ScreenPackages", "DL @ %d KB/s" );
|
|
|
|
|
void ScreenPackages::Update( float fDeltaTime )
|
|
|
|
|
{
|
|
|
|
|
HTTPUpdate();
|
|
|
|
|
|
|
|
|
|
m_fLastUpdate += fDeltaTime;
|
|
|
|
|
if ( m_fLastUpdate >= 1.0 )
|
|
|
|
|
{
|
|
|
|
|
if ( m_bIsDownloading && m_bGotHeader )
|
|
|
|
|
m_sStatus = ssprintf( DOWNLOAD_PROGRESS.GetValue(), int((m_iDownloaded-m_bytesLastUpdate)/1024) );
|
|
|
|
|
|
|
|
|
|
m_bytesLastUpdate = m_iDownloaded;
|
|
|
|
|
UpdateProgress();
|
|
|
|
|
m_fLastUpdate = 0;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
ScreenWithMenuElements::Update(fDeltaTime);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static LocalizedString ENTER_URL ("ScreenPackages","Enter URL");
|
|
|
|
|
void ScreenPackages::MenuStart( const InputEventPlus &input )
|
|
|
|
|
{
|
|
|
|
|
if ( m_iDLorLST == 1 )
|
|
|
|
|
{
|
|
|
|
|
if ( m_iLinksPos == 0 )
|
|
|
|
|
ScreenTextEntry::TextEntry( SM_BackFromURL, ENTER_URL, "http://", 255 );
|
|
|
|
|
else
|
|
|
|
|
EnterURL( m_Links[m_iLinksPos] );
|
|
|
|
|
}
|
|
|
|
|
ScreenWithMenuElements::MenuStart( input );
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void ScreenPackages::MenuUp( const InputEventPlus &input )
|
|
|
|
|
{
|
|
|
|
|
if ( m_bIsDownloading )
|
|
|
|
|
return;
|
|
|
|
|
if ( m_iDLorLST == 0)
|
|
|
|
|
{
|
|
|
|
|
if ( m_iPackagesPos > 0)
|
|
|
|
|
{
|
|
|
|
|
m_iPackagesPos--;
|
|
|
|
|
UpdatePackagesList();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
if ( m_iLinksPos > 0 )
|
|
|
|
|
{
|
|
|
|
|
m_iLinksPos--;
|
|
|
|
|
UpdateLinksList();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
ScreenWithMenuElements::MenuUp( input );
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void ScreenPackages::MenuDown( const InputEventPlus &input )
|
|
|
|
|
{
|
|
|
|
|
if ( m_bIsDownloading )
|
|
|
|
|
return;
|
|
|
|
|
|
|
|
|
|
if ( m_iDLorLST == 0)
|
|
|
|
|
{
|
|
|
|
|
if( (unsigned) m_iPackagesPos < m_Packages.size() - 1 )
|
|
|
|
|
{
|
|
|
|
|
m_iPackagesPos++;
|
|
|
|
|
UpdatePackagesList();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
if( (unsigned) m_iLinksPos < m_LinkTitles.size() - 1 )
|
|
|
|
|
{
|
|
|
|
|
m_iLinksPos++;
|
|
|
|
|
UpdateLinksList();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
ScreenWithMenuElements::MenuDown( input );
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void ScreenPackages::MenuLeft( const InputEventPlus &input )
|
|
|
|
|
{
|
|
|
|
|
if ( m_bIsDownloading )
|
|
|
|
|
return;
|
|
|
|
|
if ( !m_bCanDL )
|
|
|
|
|
return;
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
m_sprExistingBG.StopTweening();
|
|
|
|
|
m_sprWebBG.StopTweening();
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
if ( m_iDLorLST == 0 )
|
|
|
|
|
{
|
|
|
|
|
m_iDLorLST = 1;
|
|
|
|
|
COMMAND( m_sprExistingBG, "Away" );
|
|
|
|
|
COMMAND( m_sprWebBG, "Back" );
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
m_iDLorLST = 0;
|
|
|
|
|
COMMAND( m_sprExistingBG, "Back" );
|
|
|
|
|
COMMAND( m_sprWebBG, "Away" );
|
|
|
|
|
}
|
|
|
|
|
ScreenWithMenuElements::MenuLeft( input );
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void ScreenPackages::MenuRight( const InputEventPlus &input )
|
|
|
|
|
{
|
|
|
|
|
if ( m_bIsDownloading )
|
|
|
|
|
return;
|
|
|
|
|
|
|
|
|
|
// Huh?
|
|
|
|
|
//MenuLeft( input );
|
|
|
|
|
|
|
|
|
|
if ( m_iDLorLST == 1 )
|
|
|
|
|
{
|
|
|
|
|
m_iDLorLST = 0;
|
|
|
|
|
COMMAND( m_sprExistingBG, "Away" );
|
|
|
|
|
COMMAND( m_sprWebBG, "Back" );
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
m_iDLorLST = 1;
|
|
|
|
|
COMMAND( m_sprExistingBG, "Back" );
|
|
|
|
|
COMMAND( m_sprWebBG, "Away" );
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
ScreenWithMenuElements::MenuRight( input );
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static LocalizedString DOWNLOAD_CANCELLED( "ScreenPackages", "Download cancelled." );
|
|
|
|
|
void ScreenPackages::MenuBack( const InputEventPlus &input )
|
|
|
|
|
{
|
|
|
|
|
if ( m_bIsDownloading )
|
|
|
|
|
{
|
|
|
|
|
SCREENMAN->SystemMessage( DOWNLOAD_CANCELLED );
|
|
|
|
|
CancelDownload( );
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
TweenOffScreen();
|
|
|
|
|
Cancel( SM_GoToPrevScreen );
|
|
|
|
|
ScreenWithMenuElements::MenuBack( input );
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void ScreenPackages::TweenOffScreen( )
|
|
|
|
|
{
|
|
|
|
|
OFF_COMMAND( m_sprExistingBG );
|
|
|
|
|
OFF_COMMAND( m_sprWebBG );
|
|
|
|
|
OFF_COMMAND( m_sprWebSel );
|
|
|
|
|
OFF_COMMAND( m_textPackages );
|
|
|
|
|
OFF_COMMAND( m_textWeb );
|
|
|
|
|
OFF_COMMAND( m_sprDL );
|
|
|
|
|
OFF_COMMAND( m_sprDLBG );
|
|
|
|
|
OFF_COMMAND( m_textStatus );
|
|
|
|
|
|
|
|
|
|
m_fOutputFile.Close();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void ScreenPackages::RefreshPackages()
|
|
|
|
|
{
|
|
|
|
|
GetDirListing( "Packages/*.zip", m_Packages, false, false );
|
|
|
|
|
GetDirListing( "Packages/*.smzip", m_Packages, false, false );
|
|
|
|
|
|
|
|
|
|
if ( m_iPackagesPos < 0 )
|
|
|
|
|
m_iPackagesPos = 0;
|
|
|
|
|
|
|
|
|
|
if( (unsigned) m_iPackagesPos >= m_Packages.size() )
|
|
|
|
|
m_iPackagesPos = m_Packages.size() - 1;
|
|
|
|
|
|
|
|
|
|
UpdatePackagesList();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void ScreenPackages::UpdatePackagesList()
|
|
|
|
|
{
|
|
|
|
|
RString TempText="";
|
|
|
|
|
int min = m_iPackagesPos-NUM_PACKAGES_SHOW;
|
|
|
|
|
int max = m_iPackagesPos+NUM_PACKAGES_SHOW;
|
|
|
|
|
for (int i=min; i<max; i++ )
|
|
|
|
|
{
|
|
|
|
|
if ( i >= 0 && (unsigned) i < m_Packages.size() )
|
|
|
|
|
TempText += m_Packages[i] + '\n';
|
|
|
|
|
else
|
|
|
|
|
TempText += '\n';
|
|
|
|
|
}
|
|
|
|
|
m_textPackages.SetText( TempText );
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void ScreenPackages::UpdateLinksList()
|
|
|
|
|
{
|
|
|
|
|
RString TempText="";
|
|
|
|
|
if( (unsigned) m_iLinksPos >= m_LinkTitles.size() )
|
|
|
|
|
m_iLinksPos = m_LinkTitles.size() - 1;
|
|
|
|
|
int min = m_iLinksPos-NUM_LINKS_SHOW;
|
|
|
|
|
int max = m_iLinksPos+NUM_LINKS_SHOW;
|
|
|
|
|
for (int i=min; i<max; i++ )
|
|
|
|
|
{
|
|
|
|
|
if( i >= 0 && (unsigned) i < m_LinkTitles.size() )
|
|
|
|
|
TempText += m_LinkTitles[i] + '\n';
|
|
|
|
|
else
|
|
|
|
|
TempText += '\n';
|
|
|
|
|
}
|
|
|
|
|
m_textWeb.SetText( TempText );
|
|
|
|
|
if ( m_iLinksPos >= 0 && (unsigned) m_iLinksPos < m_Links.size() )
|
|
|
|
|
m_textURL.SetText( m_Links[m_iLinksPos] );
|
|
|
|
|
else
|
|
|
|
|
m_textURL.SetText( " " );
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void ScreenPackages::HTMLParse()
|
|
|
|
|
{
|
|
|
|
|
m_Links.clear();
|
|
|
|
|
m_LinkTitles.clear();
|
|
|
|
|
m_Links.push_back( " " );
|
|
|
|
|
m_LinkTitles.push_back( VISIT_URL.GetValue() );
|
|
|
|
|
|
|
|
|
|
// XXX: VERY DIRTY HTML PARSER!
|
|
|
|
|
// Only designed to find links on websites.
|
|
|
|
|
size_t i = m_sBUFFER.find( "<A " );
|
|
|
|
|
size_t j = 0;
|
|
|
|
|
size_t k = 0;
|
|
|
|
|
size_t l = 0;
|
|
|
|
|
size_t m = 0;
|
|
|
|
|
|
|
|
|
|
for ( int mode = 0; mode < 2; mode++ )
|
|
|
|
|
{
|
|
|
|
|
while ( i != string::npos )
|
|
|
|
|
{
|
|
|
|
|
k = m_sBUFFER.find( ">", i+1 );
|
|
|
|
|
l = m_sBUFFER.find( "HREF", i+1);
|
|
|
|
|
m = m_sBUFFER.find( "=", l );
|
|
|
|
|
|
|
|
|
|
if( k == string::npos || l == string::npos || m == string::npos || l > k || m > k ) //no "href" in this tag.
|
|
|
|
|
{
|
|
|
|
|
if ( mode == 0 )
|
|
|
|
|
i = m_sBUFFER.find( "<A ", i+1 );
|
|
|
|
|
else
|
|
|
|
|
i = m_sBUFFER.find( "<a ", i+1 );
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
l = m_sBUFFER.find( "</", m+1 );
|
|
|
|
|
|
|
|
|
|
// Special case: There is exactly one extra tag in the link.
|
|
|
|
|
j = m_sBUFFER.find( ">", k+1 );
|
|
|
|
|
if ( j < l )
|
|
|
|
|
k = j;
|
|
|
|
|
|
|
|
|
|
RString TempLink = StripOutContainers( m_sBUFFER.substr(m+1,k-m-1) );
|
|
|
|
|
// xxx: handle https? -aj
|
|
|
|
|
if ( TempLink.substr(0,7).compare("http://") != 0 )
|
|
|
|
|
TempLink = m_sBaseAddress + TempLink;
|
|
|
|
|
|
|
|
|
|
RString TempTitle = m_sBUFFER.substr( k+1, l-k-1 );
|
|
|
|
|
|
|
|
|
|
m_Links.push_back( TempLink );
|
|
|
|
|
m_LinkTitles.push_back( TempTitle );
|
|
|
|
|
|
|
|
|
|
if ( mode == 0 )
|
|
|
|
|
i = m_sBUFFER.find( "<A ", i+1 );
|
|
|
|
|
else
|
|
|
|
|
i = m_sBUFFER.find( "<a ", i+1 );
|
|
|
|
|
}
|
|
|
|
|
if ( mode == 0 )
|
|
|
|
|
i = m_sBUFFER.find( "<a " );
|
|
|
|
|
}
|
|
|
|
|
UpdateLinksList();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
RString ScreenPackages::StripOutContainers( const RString & In )
|
|
|
|
|
{
|
|
|
|
|
if( In.size() == 0 )
|
|
|
|
|
return RString();
|
|
|
|
|
|
|
|
|
|
unsigned i = 0;
|
|
|
|
|
char t = In.at(i);
|
|
|
|
|
while( t == ' ' && i < In.length() )
|
|
|
|
|
{
|
|
|
|
|
t = In.at(++i);
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if( t == '\"' || t == '\'' )
|
|
|
|
|
{
|
|
|
|
|
unsigned j = i+1;
|
|
|
|
|
char u = In.at(j);
|
|
|
|
|
while( u != t && j < In.length() )
|
|
|
|
|
{
|
|
|
|
|
u = In.at(++j);
|
|
|
|
|
}
|
|
|
|
|
if( j == i )
|
|
|
|
|
return StripOutContainers( In.substr(i+1) );
|
|
|
|
|
else
|
|
|
|
|
return StripOutContainers( In.substr(i+1, j-i-1) );
|
|
|
|
|
}
|
|
|
|
|
return In.substr( i );
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void ScreenPackages::UpdateProgress()
|
|
|
|
|
{
|
|
|
|
|
float DownloadedRatio;
|
|
|
|
|
|
|
|
|
|
if( m_iTotalBytes < 1 )
|
|
|
|
|
DownloadedRatio = 0;
|
|
|
|
|
else
|
|
|
|
|
DownloadedRatio = float(m_iDownloaded) / float(m_iTotalBytes);
|
|
|
|
|
|
|
|
|
|
float dLY = DownloadedRatio / 2 * m_sprDLBG.GetUnzoomedWidth();
|
|
|
|
|
|
|
|
|
|
m_sprDL.SetX( m_sprDLBG.GetX() - m_sprDLBG.GetUnzoomedWidth() / 2.0f + dLY );
|
|
|
|
|
m_sprDL.SetWidth( dLY * 2 );
|
|
|
|
|
|
|
|
|
|
m_textStatus.SetText( m_sStatus );
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static LocalizedString DOWNLOAD_FAILED( "ScreenPackages", "Failed." );
|
|
|
|
|
void ScreenPackages::CancelDownload( )
|
|
|
|
|
{
|
|
|
|
|
m_wSocket.close();
|
|
|
|
|
m_bIsDownloading = false;
|
|
|
|
|
m_iDownloaded = 0;
|
|
|
|
|
m_bGotHeader = false;
|
|
|
|
|
m_fOutputFile.Close();
|
|
|
|
|
m_sStatus = DOWNLOAD_FAILED.GetValue();
|
|
|
|
|
m_sBUFFER = "";
|
|
|
|
|
if( !FILEMAN->Remove( "Packages/" + m_sEndName ) )
|
|
|
|
|
SCREENMAN->SystemMessage( "Packages/" + m_sEndName );
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static LocalizedString INVALID_URL( "ScreenPackages", "Invalid URL." );
|
|
|
|
|
static LocalizedString FILE_ALREADY_EXISTS( "ScreenPackages", "File Already Exists" );
|
|
|
|
|
static LocalizedString FAILED_TO_CONNECT( "ScreenPackages", "Failed to connect." );
|
|
|
|
|
static LocalizedString HEADER_SENT( "ScreenPackages", "Header Sent." );
|
|
|
|
|
void ScreenPackages::EnterURL( const RString & sURL )
|
|
|
|
|
{
|
|
|
|
|
RString Proto;
|
|
|
|
|
RString Server;
|
|
|
|
|
int Port=80;
|
|
|
|
|
RString sAddress;
|
|
|
|
|
|
|
|
|
|
if( !ParseHTTPAddress( sURL, Proto, Server, Port, sAddress ) )
|
|
|
|
|
{
|
|
|
|
|
m_sStatus = INVALID_URL.GetValue();
|
|
|
|
|
UpdateProgress();
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Determine if this is a website, or a package?
|
|
|
|
|
// Criteria: does it end with *zip?
|
|
|
|
|
if( sAddress.Right(3).CompareNoCase("zip") == 0 )
|
|
|
|
|
m_bIsPackage=true;
|
|
|
|
|
else
|
|
|
|
|
m_bIsPackage = false;
|
|
|
|
|
|
|
|
|
|
m_sBaseAddress = "http://" + Server;
|
|
|
|
|
if( Port != 80 )
|
|
|
|
|
m_sBaseAddress += ssprintf( ":%d", Port );
|
|
|
|
|
m_sBaseAddress += "/";
|
|
|
|
|
|
|
|
|
|
if( sAddress.Right(1) != "/" )
|
|
|
|
|
{
|
|
|
|
|
m_sEndName = Basename( sAddress );
|
|
|
|
|
m_sBaseAddress += Dirname( sAddress );
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
m_sEndName = "";
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
// Open the file...
|
|
|
|
|
|
|
|
|
|
// First find out if a file by this name already exists
|
|
|
|
|
// if so, then we gotta ditch out.
|
|
|
|
|
// XXX: This should be fixed by a prompt or something?
|
|
|
|
|
|
|
|
|
|
// if we are not talking about a file, let's not worry
|
|
|
|
|
if( m_sEndName != "" && m_bIsPackage )
|
|
|
|
|
{
|
|
|
|
|
vector<RString> AddTo;
|
|
|
|
|
GetDirListing( "Packages/"+m_sEndName, AddTo, false, false );
|
|
|
|
|
if ( AddTo.size() > 0 )
|
|
|
|
|
{
|
|
|
|
|
m_sStatus = FILE_ALREADY_EXISTS.GetValue();
|
|
|
|
|
UpdateProgress();
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if( !m_fOutputFile.Open( "Packages/"+m_sEndName, RageFile::WRITE | RageFile::STREAMED ) )
|
|
|
|
|
{
|
|
|
|
|
m_sStatus = m_fOutputFile.GetError();
|
|
|
|
|
UpdateProgress();
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
// Continue...
|
|
|
|
|
|
|
|
|
|
sAddress = URLEncode( StripOutContainers(sAddress) );
|
|
|
|
|
|
|
|
|
|
if ( sAddress != "/" )
|
|
|
|
|
sAddress = "/" + sAddress;
|
|
|
|
|
|
|
|
|
|
m_wSocket.close();
|
|
|
|
|
m_wSocket.create();
|
|
|
|
|
|
|
|
|
|
m_wSocket.blocking = true;
|
|
|
|
|
|
|
|
|
|
if( !m_wSocket.connect( Server, (short) Port ) )
|
|
|
|
|
{
|
|
|
|
|
m_sStatus = FAILED_TO_CONNECT.GetValue();
|
|
|
|
|
UpdateProgress();
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
//Produce HTTP header
|
|
|
|
|
|
|
|
|
|
RString Header="";
|
|
|
|
|
|
|
|
|
|
Header = "GET "+sAddress+" HTTP/1.0\r\n";
|
|
|
|
|
Header+= "Host: " + Server + "\r\n";
|
|
|
|
|
Header+= "Connection: closed\r\n\r\n";
|
|
|
|
|
|
|
|
|
|
m_wSocket.SendData( Header.c_str(), Header.length() );
|
|
|
|
|
m_sStatus = HEADER_SENT.GetValue();
|
|
|
|
|
m_wSocket.blocking = false;
|
|
|
|
|
m_bIsDownloading = true;
|
|
|
|
|
m_sBUFFER = "";
|
|
|
|
|
m_bGotHeader = false;
|
|
|
|
|
UpdateProgress();
|
|
|
|
|
return;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static size_t FindEndOfHeaders( const RString &buf )
|
|
|
|
|
{
|
|
|
|
|
size_t iPos1 = buf.find( "\n\n" );
|
|
|
|
|
size_t iPos2 = buf.find( "\r\n\r\n" );
|
|
|
|
|
LOG->Trace("end: %u, %u", unsigned(iPos1), unsigned(iPos2));
|
|
|
|
|
if( iPos1 != string::npos && (iPos2 == string::npos || iPos2 > iPos1) )
|
|
|
|
|
return iPos1 + 2;
|
|
|
|
|
else if( iPos2 != string::npos && (iPos1 == string::npos || iPos1 > iPos2) )
|
|
|
|
|
return iPos2 + 4;
|
|
|
|
|
else
|
|
|
|
|
return string::npos;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
static LocalizedString WAITING_FOR_HEADER( "ScreenPackages", "Waiting for header." );
|
|
|
|
|
void ScreenPackages::HTTPUpdate()
|
|
|
|
|
{
|
|
|
|
|
if( !m_bIsDownloading )
|
|
|
|
|
return;
|
|
|
|
|
|
|
|
|
|
int BytesGot=0;
|
|
|
|
|
// Keep this as a code block
|
|
|
|
|
// as there may be need to "if" it out some time.
|
|
|
|
|
/* If you need a conditional for a large block of code, stick it in
|
|
|
|
|
* a function and return. */
|
|
|
|
|
while(1)
|
|
|
|
|
{
|
|
|
|
|
char Buffer[1024];
|
|
|
|
|
int iSize = m_wSocket.ReadData( Buffer, 1024 );
|
|
|
|
|
if( iSize <= 0 )
|
|
|
|
|
break;
|
|
|
|
|
|
|
|
|
|
m_sBUFFER.append( Buffer, iSize );
|
|
|
|
|
BytesGot += iSize;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if( !m_bGotHeader )
|
|
|
|
|
{
|
|
|
|
|
m_sStatus = WAITING_FOR_HEADER.GetValue();
|
|
|
|
|
// We don't know if we are using unix-style or dos-style
|
|
|
|
|
size_t iHeaderEnd = FindEndOfHeaders( m_sBUFFER );
|
|
|
|
|
if( iHeaderEnd == m_sBUFFER.npos )
|
|
|
|
|
return;
|
|
|
|
|
|
|
|
|
|
// "HTTP/1.1 200 OK"
|
|
|
|
|
size_t i = m_sBUFFER.find(" ");
|
|
|
|
|
size_t j = m_sBUFFER.find(" ",i+1);
|
|
|
|
|
size_t k = m_sBUFFER.find("\n",j+1);
|
|
|
|
|
if ( i == string::npos || j == string::npos || k == string::npos )
|
|
|
|
|
{
|
|
|
|
|
m_iResponseCode = -100;
|
|
|
|
|
m_sResponseName = "Malformed response.";
|
|
|
|
|
return;
|
|
|
|
|
}
|
2011-05-11 15:58:31 -04:00
|
|
|
m_iResponseCode = StringToInt(m_sBUFFER.substr(i+1,j-i));
|
2011-03-17 01:47:30 -04:00
|
|
|
m_sResponseName = m_sBUFFER.substr( j+1, k-j );
|
|
|
|
|
|
|
|
|
|
i = m_sBUFFER.find("Content-Length:");
|
|
|
|
|
j = m_sBUFFER.find("\n", i+1 );
|
|
|
|
|
|
|
|
|
|
if( i != string::npos )
|
2011-05-11 15:58:31 -04:00
|
|
|
m_iTotalBytes = StringToInt(m_sBUFFER.substr(i+16,j-i));
|
2011-03-17 01:47:30 -04:00
|
|
|
else
|
|
|
|
|
m_iTotalBytes = -1; //We don't know, so go until disconnect
|
|
|
|
|
|
|
|
|
|
m_bGotHeader = true;
|
|
|
|
|
m_sBUFFER.erase( 0, iHeaderEnd );
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if( m_bIsPackage )
|
|
|
|
|
{
|
|
|
|
|
m_iDownloaded += m_sBUFFER.length();
|
|
|
|
|
m_fOutputFile.Write( m_sBUFFER );
|
|
|
|
|
m_sBUFFER = "";
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
m_iDownloaded = m_sBUFFER.length();
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if ( ( m_iTotalBytes <= m_iDownloaded && m_iTotalBytes != -1 ) ||
|
|
|
|
|
//We have the full doc. (And we knew how big it was)
|
|
|
|
|
( m_iTotalBytes == -1 &&
|
|
|
|
|
( m_wSocket.state == EzSockets::skERROR || m_wSocket.state == EzSockets::skDISCONNECTED ) ) )
|
|
|
|
|
//We didn't know how big it was, and were disconnected
|
|
|
|
|
//So that means we have it all.
|
|
|
|
|
{
|
|
|
|
|
m_wSocket.close();
|
|
|
|
|
m_bIsDownloading = false;
|
|
|
|
|
m_bGotHeader=false;
|
|
|
|
|
m_sStatus = ssprintf( "Done ;%dB", int(m_iDownloaded) );
|
|
|
|
|
|
|
|
|
|
if( m_iResponseCode < 200 || m_iResponseCode >= 400 )
|
|
|
|
|
{
|
|
|
|
|
m_sStatus = ssprintf( "%ld", m_iResponseCode ) + m_sResponseName;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
{
|
|
|
|
|
if( m_bIsPackage && m_iResponseCode < 300 )
|
|
|
|
|
{
|
|
|
|
|
m_fOutputFile.Close();
|
|
|
|
|
RefreshPackages();
|
|
|
|
|
m_iDownloaded = 0;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
HTMLParse();
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
bool ScreenPackages::ParseHTTPAddress( const RString &URL, RString &sProto, RString &sServer, int &iPort, RString &sAddress )
|
|
|
|
|
{
|
|
|
|
|
// [PROTO://]SERVER[:PORT][/URL]
|
|
|
|
|
|
|
|
|
|
Regex re(
|
|
|
|
|
"^([A-Z]+)://" // [0]: HTTP://
|
|
|
|
|
"([^/:]+)" // [1]: a.b.com
|
|
|
|
|
"(:([0-9]+))?" // [2], [3]: :1234 (optional, default 80)
|
|
|
|
|
"(/(.*))?$"); // [4], [5]: /foo.html (optional)
|
|
|
|
|
vector<RString> asMatches;
|
|
|
|
|
if( !re.Compare( URL, asMatches ) )
|
|
|
|
|
return false;
|
|
|
|
|
ASSERT( asMatches.size() == 6 );
|
|
|
|
|
|
|
|
|
|
sProto = asMatches[0];
|
|
|
|
|
sServer = asMatches[1];
|
|
|
|
|
if( asMatches[3] != "" )
|
|
|
|
|
{
|
2011-05-11 15:58:31 -04:00
|
|
|
iPort = StringToInt(asMatches[3]);
|
2011-03-17 01:47:30 -04:00
|
|
|
if( iPort == 0 )
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
else
|
|
|
|
|
iPort = 80;
|
|
|
|
|
|
|
|
|
|
sAddress = asMatches[5];
|
|
|
|
|
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
#endif
|
|
|
|
|
/*
|
|
|
|
|
* (c) 2004 Charles Lohr
|
|
|
|
|
* All rights reserved.
|
|
|
|
|
*
|
|
|
|
|
* Permission is hereby granted, free of charge, to any person obtaining a
|
|
|
|
|
* copy of this software and associated documentation files (the
|
|
|
|
|
* "Software"), to deal in the Software without restriction, including
|
|
|
|
|
* without limitation the rights to use, copy, modify, merge, publish,
|
|
|
|
|
* distribute, and/or sell copies of the Software, and to permit persons to
|
|
|
|
|
* whom the Software is furnished to do so, provided that the above
|
|
|
|
|
* copyright notice(s) and this permission notice appear in all copies of
|
|
|
|
|
* the Software and that both the above copyright notice(s) and this
|
|
|
|
|
* permission notice appear in supporting documentation.
|
|
|
|
|
*
|
|
|
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
|
|
|
|
|
* OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
|
|
|
|
|
* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT OF
|
|
|
|
|
* THIRD PARTY RIGHTS. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR HOLDERS
|
|
|
|
|
* INCLUDED IN THIS NOTICE BE LIABLE FOR ANY CLAIM, OR ANY SPECIAL INDIRECT
|
|
|
|
|
* OR CONSEQUENTIAL DAMAGES, OR ANY DAMAGES WHATSOEVER RESULTING FROM LOSS
|
|
|
|
|
* OF USE, DATA OR PROFITS, WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR
|
|
|
|
|
* OTHER TORTIOUS ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR
|
|
|
|
|
* PERFORMANCE OF THIS SOFTWARE.
|
|
|
|
|
*/
|