#include "stdafx.h"

#include "AppManager.h"

#ifdef ENABLE_DYNAMIC_RHOBUNDLE
#include "rho/net/NetRequest.h"

#include "common/RhodesApp.h"

#include "unzip/unzip.h"
#include "ext/rho/rhoruby.h"
#include "common/AutoPointer.h"

extern "C" wchar_t* wce_mbtowc(const char* a);

CAppManager::CAppManager(void)
{
}

CAppManager::~CAppManager(void)
{
}

bool CAppManager::RemoveFolder(LPCTSTR szPath) {

	if (szPath) {
		SHFILEOPSTRUCT fop;
		
		fop.hwnd = NULL;
		fop.wFunc = FO_DELETE;		
		fop.pFrom = szPath;
		fop.pTo = NULL;
		fop.fFlags = FOF_SILENT | FOF_NOCONFIRMATION | FOF_NOERRORUI | FOF_NOCONFIRMMKDIR;
		
		int result = SHFileOperation(&fop);
		return result == 0;
	}
	return false;
}

bool CAppManager::RestartClient(HWND hwnd) {
	TCHAR module[MAX_PATH];
	GetModuleFileName(NULL,module,MAX_PATH);
	HINSTANCE res = ShellExecute(hwnd, L"open", module, L"-restarting", NULL, SW_SHOWNORMAL);
	return false;
}

void CAppManager::ReloadRhoBundle(HWND hwnd, const char* szUrl, const char* szZipPassword)
{
	if ( szUrl )
	{
		//get zip file with rhodes
		//DWORD dwDataSize = 0;
        rho::net::CNetRequest request;
        NetResponse(resp, request.pullData(szUrl, null));
        DWORD dwDataSize = resp.getDataSize();
		//char* zipData = request.doRequest( L"GET",const_cast<char*>(szUrl),NULL,0,NULL,0,false,true,false,&dwDataSize);
        const char* zipData = resp.getCharData();
		if ( zipData && dwDataSize > 0 )
		{
			LPWSTR rootw = wce_mbtowc(rho_native_rhopath());
			rootw[wcslen(rootw)-1] = '\\';
			bool ret = false;
			if (wcslen(rootw)<MAX_PATH) {
				TCHAR name[MAX_PATH+2];
				wsprintf(name,L"%sapps%c",rootw,'\0');
				if( RemoveFolder(name) ) {
					wsprintf(name,L"%slib%c",rootw,'\0');

					ret = RemoveFolder(name);
				}
			}

			//TODO: Add error handling to unzip code
			if( ret ) {
				ZIPENTRY ze; 
				// Open zip file
				HZIP hz = OpenZip(const_cast<char*>(zipData), dwDataSize, szZipPassword);
			
				if ( hz ) {
					//Stop HTTP Server
					RHODESAPP().stopApp();

					// Set base for unziping
					SetUnzipBaseDir(hz, rootw);
				
					// Get info about the zip
					// -1 gives overall information about the zipfile
					GetZipItem(hz,-1,&ze);
					int numitems = ze.index;
				
					// Iterate through items and unzip them
					for (int zi = 0; zi<numitems; zi++)
					{ 
						// fetch individual details, e.g. the item's name.
						GetZipItem(hz,zi,&ze); 
						// unzip item
						UnzipItem(hz, zi, ze.name);         
					}
				
					CloseZip(hz);
					
					//Show MessageBox
					MessageBox(hwnd, _T("Rhobundle has been updated successfully.\nApplication will be restarted."), _T("Information"), MB_OK | MB_ICONINFORMATION );
					//Restart client
					RestartClient(hwnd); 
					//Close main window and client
					::PostMessage(hwnd,WM_CLOSE,0,0);
				} else {
					MessageBox(hwnd, _T("Can't unzip loaded rhobundle."), _T("Stop"), MB_OK | MB_ICONSTOP );
				}
			} else {
				MessageBox(hwnd, _T("Can't remove old version of rhobundle. However, it may be corrupted. Exit application and reinstall rhobundle manualy."), _T("Stop"), MB_OK | MB_ICONSTOP );
			}

			if ( rootw )
				free(rootw);

		} else {
			MessageBox(hwnd, _T("Error loading rhobundle."), _T("Stop"), MB_OK | MB_ICONSTOP );
		}

//		if ( zipData )
//			free( zipData );
	}

}

#endif