reactos/rosapps/downloader/download.c

/* PROJECT:     ReactOS Downloader (was GetFirefox)
 * LICENSE:     GPL - See COPYING in the top level directory
 * FILE:        base/applications/downloader/download.c
 * PURPOSE:     Displaying a download dialog
 * COPYRIGHT:   Copyright 2001 John R. Sheets (for CodeWeavers)
 *              Copyright 2004 Mike McCormack (for CodeWeavers)
 *              Copyright 2005 Ge van Geldorp (gvg@reactos.org)
 *	            Copyright 2007 Dmitry Chapyshev (lentind@yandex.ru)
 */
/*
 * Based on Wine dlls/shdocvw/shdocvw_main.c
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library 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
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 */

#define COBJMACROS
#define WIN32_NO_STATUS
#include <windows.h>
#include <commctrl.h>
#include <shlobj.h>
#include <shlwapi.h>
#include <urlmon.h>

#include "resources.h"
#include "structures.h"

#define NDEBUG
#include <debug.h>

extern struct Application* SelectedApplication;
extern WCHAR Strings [STRING_COUNT][MAX_STRING_LENGHT];

typedef struct _IBindStatusCallbackImpl
{
	const IBindStatusCallbackVtbl *vtbl;
	LONG ref;
	HWND hDialog;
	BOOL *pbCancelled;
} IBindStatusCallbackImpl;

static HRESULT WINAPI
dlQueryInterface(IBindStatusCallback* This, REFIID riid, void** ppvObject)
{
	if (NULL == ppvObject)
	{
		return E_POINTER;
	}

	if (IsEqualIID(riid, &IID_IUnknown) || IsEqualIID(riid, &IID_IBindStatusCallback))
	{
		IBindStatusCallback_AddRef( This );
		*ppvObject = This;
		return S_OK;
	}

	return E_NOINTERFACE;
}

static ULONG WINAPI
dlAddRef(IBindStatusCallback* iface)
{
	IBindStatusCallbackImpl *This = (IBindStatusCallbackImpl *) iface;

	return InterlockedIncrement(&This->ref);
}

static ULONG WINAPI
dlRelease(IBindStatusCallback* iface)
{
	IBindStatusCallbackImpl *This = (IBindStatusCallbackImpl *) iface;
	DWORD ref = InterlockedDecrement(&This->ref);

	if( !ref )
	{
		DestroyWindow( This->hDialog );
		HeapFree(GetProcessHeap(), 0, This);
	}

	return ref;
}

static HRESULT WINAPI
dlOnStartBinding(IBindStatusCallback* iface, DWORD dwReserved, IBinding* pib)
{
	DPRINT1("OnStartBinding not implemented\n");

	return S_OK;
}

static HRESULT WINAPI
dlGetPriority(IBindStatusCallback* iface, LONG* pnPriority)
{
	DPRINT1("GetPriority not implemented\n");

	return S_OK;
}

static HRESULT WINAPI
dlOnLowResource( IBindStatusCallback* iface, DWORD reserved)
{
	DPRINT1("OnLowResource not implemented\n");

	return S_OK;
}

static HRESULT WINAPI
dlOnProgress(IBindStatusCallback* iface, ULONG ulProgress,
			ULONG ulProgressMax, ULONG ulStatusCode, LPCWSTR szStatusText)
{
	IBindStatusCallbackImpl *This = (IBindStatusCallbackImpl *) iface;
	HWND Item;
	LONG r;
	WCHAR OldText[100];

	Item = GetDlgItem(This->hDialog, IDC_PROGRESS);
	if (NULL != Item && 0 != ulProgressMax)
	{
		SendMessageW(Item, PBM_SETPOS, (ulProgress * 100) / ulProgressMax, 0);
	}

	Item = GetDlgItem(This->hDialog, IDC_STATUS);
	if (NULL != Item && NULL != szStatusText)
	{
		SendMessageW(Item, WM_GETTEXT, sizeof(OldText) / sizeof(OldText[0]),
					(LPARAM) OldText);
		if (sizeof(OldText) / sizeof(OldText[0]) - 1 <= wcslen(OldText) || 0 != wcscmp(OldText, szStatusText))
		{
			SendMessageW(Item, WM_SETTEXT, 0, (LPARAM) szStatusText);
		}
	}

	SetLastError(0);
	r = GetWindowLongPtrW(This->hDialog, GWLP_USERDATA);
	if (0 != r || 0 != GetLastError())
	{
		*This->pbCancelled = TRUE;
		DPRINT("Cancelled\n");
		return E_ABORT;
	}

	return S_OK;
}

static HRESULT WINAPI
dlOnStopBinding(IBindStatusCallback* iface, HRESULT hresult, LPCWSTR szError)
{
	DPRINT1("OnStopBinding not implemented\n");

	return S_OK;
}

static HRESULT WINAPI
dlGetBindInfo(IBindStatusCallback* iface, DWORD* grfBINDF, BINDINFO* pbindinfo)
{
	DPRINT1("GetBindInfo not implemented\n");

	return S_OK;
}

static HRESULT WINAPI
dlOnDataAvailable(IBindStatusCallback* iface, DWORD grfBSCF,
				DWORD dwSize, FORMATETC* pformatetc, STGMEDIUM* pstgmed)
{
	DPRINT1("OnDataAvailable implemented\n");

	return S_OK;
}

static HRESULT WINAPI
dlOnObjectAvailable(IBindStatusCallback* iface, REFIID riid, IUnknown* punk)
{
	DPRINT1("OnObjectAvailable implemented\n");

	return S_OK;
}

static const IBindStatusCallbackVtbl dlVtbl =
{
	dlQueryInterface,
	dlAddRef,
	dlRelease,
	dlOnStartBinding,
	dlGetPriority,
	dlOnLowResource,
	dlOnProgress,
	dlOnStopBinding,
	dlGetBindInfo,
	dlOnDataAvailable,
	dlOnObjectAvailable
};

static IBindStatusCallback*
CreateDl(HWND Dlg, BOOL *pbCancelled)
{
	IBindStatusCallbackImpl *This;

	This = HeapAlloc(GetProcessHeap(), 0, sizeof(IBindStatusCallbackImpl));
	This->vtbl = &dlVtbl;
	This->ref = 1;
	This->hDialog = Dlg;
	This->pbCancelled = pbCancelled;

	return (IBindStatusCallback*) This;
}

static DWORD WINAPI
ThreadFunc(LPVOID Context)
{
	//static const WCHAR szUrl[] = DownloadUrl;
	IBindStatusCallback *dl;
	WCHAR path[MAX_PATH];
	LPWSTR p;
	STARTUPINFOW si;
	PROCESS_INFORMATION pi;
	HWND Dlg = (HWND) Context;
	DWORD r;
	BOOL bCancelled = FALSE;
	BOOL bTempfile = FALSE;
	HKEY hKey;
	DWORD dwSize = MAX_PATH;

	/* built the path for the download */
	p = wcsrchr(SelectedApplication->Location, L'/');
	if (NULL == p)
	{
		goto end;
	}
  
	/* Create default download path */
	if (GetWindowsDirectory(path, sizeof(path) / sizeof(WCHAR)))
	{
		WCHAR DPath[256];
		int i;
		for (i = 0; i < 4; i++)
		{
			if (i == 3)
			{
				DPath[i] = '\0';
				break;
			}
			DPath[i] = path[i];
		}
		LoadString(GetModuleHandle(NULL), IDS_DOWNLOAD_FOLDER, path, sizeof(path) / sizeof(WCHAR));
		wcscat((LPWSTR)DPath, path);
		wcscpy(path, DPath);
	}
	
	if (RegOpenKey(HKEY_LOCAL_MACHINE,
				  TEXT("Software\\ReactOS\\Downloader"),
				  &hKey) == ERROR_SUCCESS)
	{
		if ((RegQueryValueEx(hKey,
						L"DownloadFolder",
						NULL,
						NULL,
						(LPBYTE)&path,
						&dwSize) != ERROR_SUCCESS) && (path[0] == 0))
		{
			goto end;
		}
	}
	
	

	if (GetFileAttributes(path) == 0xFFFFFFFF)
		if (!CreateDirectory((LPCTSTR)path,NULL))
		{
			goto end;
		}
	wcscat(path, L"\\");
	wcscat(path, p + 1);

	/* download it */
	bTempfile = TRUE;
	dl = CreateDl(Context, &bCancelled);
	r = URLDownloadToFileW(NULL, SelectedApplication->Location, path, 0, dl);
	if (NULL != dl)
	{
		IBindStatusCallback_Release(dl);
	}
	if (S_OK != r)
	{
		MessageBoxW(0,Strings[IDS_DOWNLOAD_ERROR],0,0);
		goto end;
	}
	else if (bCancelled)
	{
		goto end;
	}
	ShowWindow(Dlg, SW_HIDE);

	/* run it */
	memset(&si, 0, sizeof(si));
	si.cb = sizeof(si);
	r = CreateProcessW(path, NULL, NULL, NULL, 0, 0, NULL, NULL, &si, &pi);
	if (0 == r)
	{
		goto end;
	}
	CloseHandle(pi.hThread);
	WaitForSingleObject(pi.hProcess, INFINITE);
	CloseHandle(pi.hProcess);

	end:
		if (bTempfile)
		{
			if (bCancelled)
				DeleteFileW(path);
			else
			{
				DWORD dwSize = sizeof(DWORD);
				DWORD dwValue, dwType = REG_DWORD;
				if (RegQueryValueEx(hKey,
									L"DeleteInstaller",
									NULL,
									&dwType,
									(LPBYTE)&dwValue,
									&dwSize) == ERROR_SUCCESS)
					if (dwValue == 0x1)
						DeleteFileW(path);
				RegCloseKey(hKey);
			}
		}
	EndDialog(Dlg, 0);
	return 0;
}

INT_PTR CALLBACK
DownloadProc(HWND Dlg, UINT Msg, WPARAM wParam, LPARAM lParam)
{
	HANDLE Thread;
	DWORD ThreadId;
	HWND Item;

	switch (Msg)
	{
		case WM_INITDIALOG:/*
      Icon = LoadIconW((HINSTANCE) GetWindowLongPtr(Dlg, GWLP_HINSTANCE),
                       MAKEINTRESOURCEW(IDI_ICON_MAIN));
      if (NULL != Icon)
        {
          SendMessageW(Dlg, WM_SETICON, ICON_BIG, (LPARAM) Icon);
          SendMessageW(Dlg, WM_SETICON, ICON_SMALL, (LPARAM) Icon);
        }*/
			SetWindowLongPtrW(Dlg, GWLP_USERDATA, 0);
			Item = GetDlgItem(Dlg, IDC_PROGRESS);
			if (NULL != Item)
			{
				SendMessageW(Item, PBM_SETRANGE, 0, MAKELPARAM(0,100));
				SendMessageW(Item, PBM_SETPOS, 0, 0);
			}/*
      Item = GetDlgItem(Dlg, IDC_REMOVE);
      if (NULL != Item)
        {
          if (GetShortcutName(ShortcutName) &&
              INVALID_FILE_ATTRIBUTES != GetFileAttributesW(ShortcutName))
            {
              SendMessageW(Item, BM_SETCHECK, BST_CHECKED, 0);
            }
          else
            {
              SendMessageW(Item, BM_SETCHECK, BST_UNCHECKED, 0);
              ShowWindow(Item, SW_HIDE);
            }
        }*/
			Thread = CreateThread(NULL, 0, ThreadFunc, Dlg, 0, &ThreadId);
			if (NULL == Thread)
			{
				return FALSE;
			}
			CloseHandle(Thread);
			return TRUE;

		case WM_COMMAND:
			if (wParam == IDCANCEL)
			{
				SetWindowLongPtrW(Dlg, GWLP_USERDATA, 1);
				PostMessage(Dlg, WM_CLOSE, 0, 0);
			}
			return FALSE;

		case WM_CLOSE:
			EndDialog(Dlg, 0);
			return TRUE;

		default:
			return FALSE;
	}
}