usbloadergx/source/themes/ThemeDownloader.cpp
dimok321 2c268af0f9 *fixed possible crash on theme downloader
*changed fat/ntfs/ext installer to use the header title as foldername instead of wiitdb title (this is the way it is in the wbfs managers)
*Fixed language changing for wiitdb titles / coverdownload even without available .lang file which is loaded
*Added reload of game titles on language change
*Added check for valid config files. If config file is not valid it will not be used. Current valid configs are R1031 to all. (You configs will be reseted with this. If you want to save them, move them and merge them afterwards manually.)
*Fixed possible crash on mounting
*Changed default settings for device free space display to off, because fat32 partitions are very slow on that. Added a warning when on FAT partition and trying to enable this option.
*Updated langauge files
2011-01-06 18:59:45 +00:00

437 lines
13 KiB
C++

/****************************************************************************
* Theme_Downloader
* USB Loader GX 2009
*
* Theme downloader for USB Loader GX
*
* Theme_Downloader.cpp
***************************************************************************/
#include <string.h>
#include <unistd.h>
#include "ThemeDownloader.h"
#include "language/gettext.h"
#include "prompts/PromptWindows.h"
#include "prompts/ProgressWindow.h"
#include "FileOperations/DirList.h"
#include "network/networkops.h"
#include "themes/CTheme.h"
#include "FileOperations/fileops.h"
#include "sys.h"
#include "network/FileDownloader.h"
#include "network/http.h"
#include "menu/menus.h"
#include "ZipFile.h"
#include "utils/ShowError.h"
#include "utils/tools.h"
#include "gecko.h"
ThemeDownloader::ThemeDownloader()
: FlyingButtonsMenu(tr("Theme Downloader"))
{
ThemeList = NULL;
delete MainButtonImgData;
delete MainButtonImgOverData;
ParentMenu = MENU_SETTINGS;
ThemeListURL = "http://wii.spiffy360.com/themes.php?xml=1&category=1&adult=";
if(Settings.godmode)
ThemeListURL += "1";
else
ThemeListURL += "0";
MainButtonImgData = Resources::GetImageData("theme_box.png");
MainButtonImgOverData = NULL;
urlTxt = new GuiText(tr( "Themes by www.spiffy360.com" ), 22, (GXColor) {255, 255, 255, 255});
urlTxt->SetAlignment(ALIGN_LEFT, ALIGN_TOP);
urlTxt->SetPosition(350, 12);
Append(urlTxt);
for(int i = 0; i < 4; ++i)
ThemePreviews[i] = NULL;
}
ThemeDownloader::~ThemeDownloader()
{
HaltGui();
for(u32 i = 0; i < MainButton.size(); ++i)
Remove(MainButton[i]);
Remove(urlTxt);
delete urlTxt;
delete ThemeList;
for(int i = 0; i < 4; ++i)
delete ThemePreviews[i];
}
int ThemeDownloader::Run()
{
ThemeDownloader * Menu = new ThemeDownloader();
mainWindow->Append(Menu);
Menu->ShowMenu();
int returnMenu = MENU_NONE;
while((returnMenu = Menu->MainLoop()) == MENU_NONE);
delete Menu;
return returnMenu;
}
void ThemeDownloader::SetMainButton(int position, const char * ButtonText, GuiImageData * imageData, GuiImageData * themeImg)
{
if(position >= (int) MainButton.size())
{
MainButtonImg.resize(position+1);
MainButtonImgOver.resize(position+1);
MainButtonTxt.resize(position+1);
MainButton.resize(position+1);
}
MainButtonImg[position] = new GuiImage(imageData);
MainButtonImgOver[position] = new GuiImage(themeImg);
MainButtonImgOver[position]->SetScale(0.4);
MainButtonImgOver[position]->SetPosition(50, -45);
MainButtonTxt[position] = new GuiText(ButtonText, 18, ( GXColor ) {0, 0, 0, 255});
MainButtonTxt[position]->SetAlignment(ALIGN_CENTER, ALIGN_TOP);
MainButtonTxt[position]->SetPosition(0, 10);
MainButtonTxt[position]->SetMaxWidth(imageData->GetWidth() - 10, DOTTED);
MainButton[position] = new GuiButton(imageData->GetWidth(), imageData->GetHeight());
MainButton[position]->SetAlignment(ALIGN_LEFT, ALIGN_TOP);
MainButton[position]->SetSoundOver(btnSoundOver);
MainButton[position]->SetSoundClick(btnSoundClick);
MainButton[position]->SetImage(MainButtonImg[position]);
MainButton[position]->SetImageOver(MainButtonImg[position]);
MainButton[position]->SetIcon(MainButtonImgOver[position]);
MainButton[position]->SetLabel(MainButtonTxt[position]);
MainButton[position]->SetTrigger(trigA);
MainButton[position]->SetEffectGrow();
switch(position % 4)
{
case 0:
MainButton[position]->SetPosition(90, 75);
break;
case 1:
MainButton[position]->SetPosition(340, 75);
break;
case 2:
MainButton[position]->SetPosition(90, 230);
break;
case 3:
MainButton[position]->SetPosition(340, 230);
break;
default:
break;
}
}
GuiImageData * ThemeDownloader::GetImageData(int theme)
{
GuiImageData * ImageData = NULL;
char filepath[300];
snprintf(filepath, sizeof(filepath), "%s/tmp/%s.jpg", Settings.theme_path, ThemeList->GetThemeTitle(theme));
if (!CheckFile(filepath))
{
struct block file = downloadfile(ThemeList->GetImageLink(theme));
char storepath[300];
snprintf(storepath, sizeof(storepath), "%s/tmp/", Settings.theme_path);
CreateSubfolder(storepath);
if (file.data)
{
ImageData = new GuiImageData(file.data, file.size, false);
FILE *storefile = fopen(filepath, "wb");
if(storefile)
{
fwrite(file.data, 1, file.size, storefile);
fclose(storefile);
}
free(file.data);
}
}
else
ImageData = new GuiImageData(filepath);
return ImageData;
}
void ThemeDownloader::SetupMainButtons()
{
ResumeGui();
if (!IsNetworkInit() && !NetworkInitPrompt())
{
ShowError("Could not initialize network!");
return;
}
ShowProgress(tr("Downloading pagelist:"), "www.spiffy360.com", tr("Please wait..."), 0, 1);
ThemeList = new Theme_List(ThemeListURL.c_str());
if (ThemeList->GetThemeCount() == 0)
{
WindowPrompt(tr( "No themes found on the site." ), 0, "OK");
returnMenu = MENU_SETTINGS;
}
for(int i = 0; i < ThemeList->GetThemeCount(); ++i)
{
SetMainButton(i, ThemeList->GetThemeTitle(i), MainButtonImgData, NULL);
}
}
void ThemeDownloader::AddMainButtons()
{
if(!ThemeList)
SetupMainButtons();
HaltGui();
for(u32 i = 0; i < MainButton.size(); ++i)
Remove(MainButton[i]);
ResumeGui();
int FirstItem = currentPage*4;
int n = 0;
for(int i = FirstItem; i < (int) MainButton.size() && i < FirstItem+4; ++i)
{
delete ThemePreviews[n];
ShowProgress(tr("Downloading image:"), 0, ThemeList->GetThemeTitle(i), n, 4);
ThemePreviews[n] = GetImageData(i);
MainButtonImgOver[i]->SetImage(ThemePreviews[n]);
n++;
}
ProgressStop();
HaltGui();
FlyingButtonsMenu::AddMainButtons();
}
void ThemeDownloader::MainButtonClicked(int button)
{
//! TODO: Clean me
const char * title = ThemeList->GetThemeTitle(button);
const char * author = ThemeList->GetThemeAuthor(button);
GuiImageData *thumbimageData = ThemePreviews[button % 4];
const char *downloadlink = ThemeList->GetDownloadLink(button);
gprintf("\nTheme_Prompt(%s ,%s, <DATA>, %s)", title, author, downloadlink);
bool leave = false;
int result = 0;
GuiImageData btnOutline(Resources::GetFile("button_dialogue_box.png"), Resources::GetFileSize("button_dialogue_box.png"));
GuiImageData dialogBox(Resources::GetFile("theme_dialogue_box.png"), Resources::GetFileSize("theme_dialogue_box.png"));
GuiImage dialogBoxImg(&dialogBox);
GuiWindow promptWindow(dialogBox.GetWidth(), dialogBox.GetHeight());
promptWindow.SetAlignment(ALIGN_CENTRE, ALIGN_MIDDLE);
promptWindow.SetPosition(0, -10);
GuiTrigger trigA;
trigA.SetSimpleTrigger(-1, WPAD_BUTTON_A | WPAD_CLASSIC_BUTTON_A, PAD_BUTTON_A);
GuiTrigger trigB;
trigB.SetButtonOnlyTrigger(-1, WPAD_BUTTON_B | WPAD_CLASSIC_BUTTON_B, PAD_BUTTON_B);
GuiText titleTxt(tr( "Theme Title:" ), 18, thColor("r=0 g=0 b=0 a=255 - prompt windows text color"));
titleTxt.SetAlignment(ALIGN_LEFT, ALIGN_TOP);
titleTxt.SetPosition(230, 30);
GuiText titleTxt2(title, 18, thColor("r=0 g=0 b=0 a=255 - prompt windows text color"));
titleTxt2.SetAlignment(ALIGN_LEFT, ALIGN_TOP);
titleTxt2.SetPosition(230, 50);
titleTxt2.SetMaxWidth(dialogBox.GetWidth() - 220, WRAP);
GuiText authorTxt(tr( "Author:" ), 18, thColor("r=0 g=0 b=0 a=255 - prompt windows text color"));
authorTxt.SetAlignment(ALIGN_LEFT, ALIGN_TOP);
authorTxt.SetPosition(230, 100);
GuiText authorTxt2(author, 18, thColor("r=0 g=0 b=0 a=255 - prompt windows text color"));
authorTxt2.SetAlignment(ALIGN_LEFT, ALIGN_TOP);
authorTxt2.SetPosition(230, 120);
authorTxt2.SetMaxWidth(dialogBox.GetWidth() - 220, DOTTED);
GuiText downloadBtnTxt(tr( "Download" ), 22, thColor("r=0 g=0 b=0 a=255 - prompt windows text color"));
downloadBtnTxt.SetMaxWidth(btnOutline.GetWidth() - 30);
GuiImage downloadBtnImg(&btnOutline);
if (Settings.wsprompt)
{
downloadBtnTxt.SetWidescreen(Settings.widescreen);
downloadBtnImg.SetWidescreen(Settings.widescreen);
}
GuiButton downloadBtn(&downloadBtnImg, &downloadBtnImg, ALIGN_RIGHT, ALIGN_TOP, -5, 170, &trigA, btnSoundOver, btnSoundClick2, 1);
downloadBtn.SetLabel(&downloadBtnTxt);
downloadBtn.SetScale(0.9);
GuiText backBtnTxt(tr( "Back" ), 22, thColor("r=0 g=0 b=0 a=255 - prompt windows text color"));
backBtnTxt.SetMaxWidth(btnOutline.GetWidth() - 30);
GuiImage backBtnImg(&btnOutline);
if (Settings.wsprompt)
{
backBtnTxt.SetWidescreen(Settings.widescreen);
backBtnImg.SetWidescreen(Settings.widescreen);
}
GuiButton backBtn(&backBtnImg, &backBtnImg, ALIGN_RIGHT, ALIGN_TOP, -5, 220, &trigA, btnSoundOver, btnSoundClick2, 1);
backBtn.SetLabel(&backBtnTxt);
backBtn.SetTrigger(&trigB);
backBtn.SetScale(0.9);
GuiImage ThemeImage(thumbimageData);
ThemeImage.SetAlignment(ALIGN_LEFT, ALIGN_TOP);
ThemeImage.SetPosition(20, 10);
ThemeImage.SetScale(0.8);
promptWindow.Append(&dialogBoxImg);
promptWindow.Append(&ThemeImage);
promptWindow.Append(&titleTxt);
promptWindow.Append(&titleTxt2);
promptWindow.Append(&authorTxt);
promptWindow.Append(&authorTxt2);
promptWindow.Append(&downloadBtn);
promptWindow.Append(&backBtn);
HaltGui();
promptWindow.SetEffect(EFFECT_SLIDE_TOP | EFFECT_SLIDE_IN, 50);
mainWindow->SetState(STATE_DISABLED);
mainWindow->Append(&promptWindow);
mainWindow->ChangeFocus(&promptWindow);
ResumeGui();
while (!leave)
{
VIDEO_WaitVSync();
if (shutdown)
Sys_Shutdown();
else if (reset)
Sys_Reboot();
if (downloadBtn.GetState() == STATE_CLICKED)
{
int choice = WindowPrompt(tr( "Do you want to download this theme?" ), title, tr( "Yes" ), tr( "Cancel" ));
if (choice)
{
result = DownloadTheme(downloadlink, title);
if (result == 2)
returnMenu = MENU_THEMEDOWNLOADER;;
}
mainWindow->SetState(STATE_DISABLED);
promptWindow.SetState(STATE_DEFAULT);
mainWindow->ChangeFocus(&promptWindow);
downloadBtn.ResetState();
}
else if (backBtn.GetState() == STATE_CLICKED)
{
leave = true;
backBtn.ResetState();
}
}
promptWindow.SetEffect(EFFECT_SLIDE_TOP | EFFECT_SLIDE_OUT, 50);
while (promptWindow.GetEffect() > 0) usleep(100);
HaltGui();
mainWindow->Remove(&promptWindow);
mainWindow->SetState(STATE_DEFAULT);
ResumeGui();
}
int ThemeDownloader::DownloadTheme(const char *url, const char *title)
{
if (!url) return -1;
if(!CreateSubfolder(Settings.theme_path))
{
ShowError(tr("Can't create path: %s"), Settings.theme_path);
return -1;
}
DirList oldDir(Settings.theme_path);
char filepath[300];
snprintf(filepath, sizeof(filepath), "%s/TempTheme.zip", Settings.theme_path);
int ret = DownloadFileToPath(url, filepath, false);
if(ret < 1024)
{
ShowError(tr("Error when downloading file: %i"), ret);
return -2;
}
ZipFile zipfile(filepath);
int result = zipfile.ExtractAll(Settings.theme_path);
if(result < 0)
{
WindowPrompt(tr( "Failed to extract." ), tr( "Unsupported format, try to extract manually TempTheme.zip." ), tr( "OK" ));
return -3;
}
remove(filepath);
DirList newDir(Settings.theme_path);
int choice = WindowPrompt(tr( "Successfully extracted theme." ), tr( "Do you want to apply it now?" ), tr( "Yes" ), tr( "No" ));
if (choice == 0)
return -1;
char Filename[255];
memset(Filename, 0, sizeof(Filename));
for(int i = 0; i < newDir.GetFilecount(); ++i)
{
const char * FilenameNew = newDir.GetFilename(i);
if(!FilenameNew)
continue;
const char * FileExt = strrchr(FilenameNew, '.');
if(!FileExt || strcasecmp(FileExt, ".them") != 0)
continue;
bool Found = false;
for(int j = 0; j < oldDir.GetFilecount(); ++j)
{
const char * FilenameOld = oldDir.GetFilename(j);
if(!FilenameOld)
continue;
if(strcasecmp(FilenameNew, FilenameOld) == 0)
Found = true;
}
if(!Found)
{
snprintf(Filename, sizeof(Filename), FilenameNew);
break;
}
}
if(Filename[0] == 0)
{
WindowPrompt(tr( "ERROR: Can't set up theme." ), tr( "The .them file was not found in the zip." ), tr( "OK" ));
return -1;
}
char real_themepath[1024];
snprintf(real_themepath, sizeof(real_themepath), "%s/%s", Settings.theme_path, Filename);
if (Theme::Load(real_themepath))
{
snprintf(Settings.theme, sizeof(Settings.theme), real_themepath);
result = 2;
}
return result;
}