Files
itgmania212121/src/ScreenPackages.cpp
T

736 lines
18 KiB
C++
Raw Normal View History

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.
*/