1 files changed, 472 insertions, 0 deletions
diff --git a/util/install/win/shell.cpp b/util/install/win/shell.cpp
new file mode 100644
index 0000000..b1303a7
--- /dev/null
+++ b/util/install/win/shell.cpp
@@ -0,0 +1,472 @@
+/****************************************************************************
+**
+** Copyright (C) 2009 Nokia Corporation and/or its subsidiary(-ies).
+** Contact: Qt Software Information (qt-info@nokia.com)
+**
+** This file is part of the utils of the Qt Toolkit.
+**
+** $QT_BEGIN_LICENSE:LGPL$
+** No Commercial Usage
+** This file contains pre-release code and may not be distributed.
+** You may use this file in accordance with the terms and conditions
+** contained in the either Technology Preview License Agreement or the
+** Beta Release License Agreement.
+**
+** GNU Lesser General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU Lesser
+** General Public License version 2.1 as published by the Free Software
+** Foundation and appearing in the file LICENSE.LGPL included in the
+** packaging of this file.  Please review the following information to
+** ensure the GNU Lesser General Public License version 2.1 requirements
+** will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html.
+**
+** In addition, as a special exception, Nokia gives you certain
+** additional rights. These rights are described in the Nokia Qt LGPL
+** Exception version 1.0, included in the file LGPL_EXCEPTION.txt in this
+** package.
+**
+** GNU General Public License Usage
+** Alternatively, this file may be used under the terms of the GNU
+** General Public License version 3.0 as published by the Free Software
+** Foundation and appearing in the file LICENSE.GPL included in the
+** packaging of this file.  Please review the following information to
+** ensure the GNU General Public License version 3.0 requirements will be
+** met: http://www.gnu.org/copyleft/gpl.html.
+**
+** If you are unsure which license is appropriate for your use, please
+** contact the sales department at qt-sales@nokia.com.
+** $QT_END_LICENSE$
+**
+****************************************************************************/
+#include "shell.h"
+#include "environment.h"
+#include "dialogs/folderdlgimpl.h"
+#include <qnamespace.h>
+#include <qdir.h>
+#include <qlibrary.h>
+#if defined(Q_OS_WIN32)
+#include <windows.h>
+#include <shlobj.h>
+#else
+#include <sys/param.h>
+#include <sys/mount.h>
+#endif
+
+static const char* folder_closed_xpm[]={
+    "16 16 9 1",
+    "g c #808080",
+    "b c #c0c000",
+    "e c #c0c0c0",
+    "# c #000000",
+    "c c #ffff00",
+    ". c None",
+    "a c #585858",
+    "f c #a0a0a4",
+    "d c #ffffff",
+    "..###...........",
+    ".#abc##.........",
+    ".#daabc#####....",
+    ".#ddeaabbccc#...",
+    ".#dedeeabbbba...",
+    ".#edeeeeaaaab#..",
+    ".#deeeeeeefe#ba.",
+    ".#eeeeeeefef#ba.",
+    ".#eeeeeefeff#ba.",
+    ".#eeeeefefff#ba.",
+    ".##geefeffff#ba.",
+    "...##gefffff#ba.",
+    ".....##fffff#ba.",
+    ".......##fff#b##",
+    ".........##f#b##",
+    "...........####."
+};
+
+static const char* folder_open_xpm[]={
+    "16 16 11 1",
+    "# c #000000",
+    "g c #c0c0c0",
+    "e c #303030",
+    "a c #ffa858",
+    "b c #808080",
+    "d c #a0a0a4",
+    "f c #585858",
+    "c c #ffdca8",
+    "h c #dcdcdc",
+    "i c #ffffff",
+    ". c None",
+    "....#ab##.......",
+    "....###.........",
+    "....#acab####...",
+    "###.#acccccca#..",
+    "#ddefaaaccccca#.",
+    "#bdddbaaaacccab#",
+    ".eddddbbaaaacab#",
+    ".#bddggdbbaaaab#",
+    "..edgdggggbbaab#",
+    "..#bgggghghdaab#",
+    "...ebhggghicfab#",
+    "....#edhhiiidab#",
+    "......#egiiicfb#",
+    "........#egiibb#",
+    "..........#egib#",
+    "............#ee#"
+};
+
+static const char* file_xpm []={
+    "16 16 7 1",
+    "# c #000000",
+    "b c #ffffff",
+    "e c #000000",
+    "d c #404000",
+    "c c #c0c000",
+    "a c #ffffc0",
+    ". c None",
+    "................",
+    ".........#......",
+    "......#.#a##....",
+    ".....#b#bbba##..",
+    "....#b#bbbabbb#.",
+    "...#b#bba##bb#..",
+    "..#b#abb#bb##...",
+    ".#a#aab#bbbab##.",
+    "#a#aaa#bcbbbbbb#",
+    "#ccdc#bcbbcbbb#.",
+    ".##c#bcbbcabb#..",
+    "...#acbacbbbe...",
+    "..#aaaacaba#....",
+    "...##aaaaa#.....",
+    ".....##aa#......",
+    ".......##......."
+};
+
+static const char* info_xpm[] = {
+    "16 16 6 1",
+    "# c #0000ff",
+    "a c #6868ff",
+    "b c #d0d0ff",
+    "c c #ffffff",
+    "- c #000000",
+    ". c none",
+    ".....------.....",
+    "...--######--...",
+    "..-###acca###-..",
+    ".-####cccc####-.",
+    ".-####acca####-.",
+    "-##############-",
+    "-######bcc#####-",
+    "-####ccccc#####-",
+    "-#####cccc#####-",
+    "-#####cccc#####-",
+    "-#####cccc#####-",
+    ".-####cccc####-.",
+    ".-###cccccc###-.",
+    "..-##########-..",
+    "...--#######-...",
+    ".....------....."
+};
+
+static QPixmap* closedImage = NULL;
+static QPixmap* openImage = NULL;
+static QPixmap* fileImage = NULL;
+static QPixmap* infoImage = NULL;
+
+#if defined(Q_OS_WIN32)
+typedef BOOL (WINAPI *PtrSHGetPathFromIDListW)(LPITEMIDLIST,LPWSTR);
+static PtrSHGetPathFromIDListW ptrSHGetPathFromIDListW = 0;
+
+static void resolveLibs()
+{
+    static bool triedResolve = false;
+
+    if ( !triedResolve ) {
+	triedResolve = true;
+	if( int( qWinVersion() ) & int( Qt::WV_NT_based ) ) {
+	    QLibrary lib("shell32");
+	    lib.setAutoUnload( false );
+	    ptrSHGetPathFromIDListW = (PtrSHGetPathFromIDListW) lib.resolve( "SHGetPathFromIDListW" );
+	}
+    }
+}
+#endif
+
+WinShell::WinShell()
+{
+#if defined(Q_OS_WIN32)
+    HRESULT hr;
+    LPITEMIDLIST item;
+#endif
+
+    localProgramsFolderName.clear();
+    commonProgramsFolderName.clear();
+    windowsFolderName.clear();
+
+#if defined(Q_OS_WIN32)
+    resolveLibs();
+    if( ptrSHGetPathFromIDListW && int( qWinVersion() ) & int( Qt::WV_NT_based ) ) {
+	ushort buffer[MAX_PATH];
+	if( SUCCEEDED( hr = SHGetSpecialFolderLocation( NULL, CSIDL_PROGRAMS, &item ) ) ) {
+	    if( ptrSHGetPathFromIDListW( item, (wchar_t*) buffer ) ) {
+		localProgramsFolderName = QString::fromUcs2( buffer );
+		if( SUCCEEDED( hr = SHGetSpecialFolderLocation( NULL, CSIDL_COMMON_PROGRAMS, &item ) ) ) {
+		    if( ptrSHGetPathFromIDListW( item, (wchar_t*) buffer ) )
+			commonProgramsFolderName = QString::fromUcs2( buffer );
+		    else
+			qDebug( "Could not get name of common programs folder" );
+		}
+		else
+		    qDebug( "Could not get common programs folder location" );
+
+		if( GetWindowsDirectoryW( (wchar_t*) buffer, MAX_PATH ) )
+		    windowsFolderName = QString::fromUcs2( buffer );
+		else
+		    qDebug( "Could not get Windows directory" );
+	    }
+	    else
+		qDebug( "Could not get name of programs folder" );
+	}
+	else
+	    qDebug( "Could not get programs folder location" );
+    }
+    else {
+	QByteArray buffer( MAX_PATH );
+	if( SUCCEEDED( hr = SHGetSpecialFolderLocation( NULL, CSIDL_PROGRAMS, &item ) ) ) {
+	    if( SHGetPathFromIDListA( item, buffer.data() ) ) {
+		localProgramsFolderName = buffer.data();
+		commonProgramsFolderName = buffer.data();
+	    }
+	    else
+		qDebug( "Could not get name of programs folder" );
+	}
+	else
+	    qDebug( "Could not get programs folder location" );
+    }
+#endif
+
+    closedImage = new QPixmap( folder_closed_xpm );
+    openImage = new QPixmap( folder_open_xpm );
+    fileImage = new QPixmap( file_xpm );
+    infoImage = new QPixmap( info_xpm );
+}
+
+WinShell::~WinShell()
+{
+}
+
+QString WinShell::selectFolder( QString folderName, bool common )
+{
+    QStringList folders;
+    FolderDlgImpl dlg;
+
+    if( common )
+	dlg.setup( commonProgramsFolderName, folderName );
+    else
+	dlg.setup( localProgramsFolderName, folderName );
+
+    if( dlg.exec() ) {
+	return dlg.getFolderName();
+    }
+    else
+	return folderName;
+}
+
+QString WinShell::createFolder( QString folderName, bool common )
+{
+    QDir folderDir;
+    QString folderPath;
+
+    if( common )
+	folderPath = commonProgramsFolderName + QString( "\\" ) + folderName;
+    else
+	folderPath = localProgramsFolderName + QString( "\\" ) + folderName;
+
+    folderDir.setPath( folderPath );
+
+    if( !folderDir.exists( folderPath ) )
+	if( !createDir( folderPath ) )
+	    return QString();
+
+    return folderPath;
+}
+
+
+#if defined(Q_OS_WIN32)
+HRESULT WinShell::createShortcut( QString folderName, bool, QString shortcutName, QString target, QString description, QString arguments, QString wrkDir )
+{
+    IPersistFile* linkFile;
+    HRESULT hr;
+
+    // Add .lnk to shortcut name if needed
+    if( shortcutName.right( 4 ) != ".lnk" )
+	shortcutName += ".lnk";
+
+    folderName = QEnvironment::getFSFileName( folderName );
+    if( int( qWinVersion() ) & int( Qt::WV_NT_based ) ) {
+        IShellLinkW* link;
+	if( SUCCEEDED( hr = CoCreateInstance( CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, IID_IShellLinkW, (void**)&link ) ) ) {
+	    if( SUCCEEDED( hr = link->QueryInterface( IID_IPersistFile, (void**)&linkFile ) ) ) {
+		link->SetPath( (const wchar_t*) target.ucs2() );
+		QString _wrkDir = wrkDir;
+		if( !_wrkDir.length() ) {
+		    _wrkDir = QDir::toNativeSeparators( target );
+		    // remove the filename
+		    int pos = _wrkDir.findRev( '\\' );
+		    if ( pos > 0 )
+			_wrkDir = _wrkDir.left( pos );
+		    else
+			_wrkDir = "";
+		}
+
+		link->SetWorkingDirectory( (const wchar_t*) _wrkDir.ucs2() );
+		if( description.length() )
+		    link->SetDescription( (const wchar_t*) description.ucs2() );
+		if( arguments.length() )
+		    link->SetArguments( (const wchar_t*)  arguments.ucs2() );
+
+		hr = linkFile->Save( (const wchar_t*)  QString( folderName + QString( "\\" ) + shortcutName ).ucs2(), false );
+
+		linkFile->Release();
+	    }
+	    else
+		qDebug( "Could not get link file interface" );
+
+	    link->Release();
+	}
+	else
+	    qDebug( "Could not instantiate link object" );
+    }
+    else {
+        IShellLinkA* link;
+	if( SUCCEEDED( hr = CoCreateInstance( CLSID_ShellLink, NULL, CLSCTX_INPROC_SERVER, IID_IShellLinkA, (void**)&link ) ) ) {
+	    if( SUCCEEDED( hr = link->QueryInterface( IID_IPersistFile, (void**)&linkFile ) ) ) {
+		link->SetPath( target.local8Bit().data() );
+		QString wrkDir = QDir::toNativeSeparators( target );
+
+		// remove the filename
+		int pos = wrkDir.findRev( '\\' );
+		if ( pos > 0 )
+		    wrkDir = wrkDir.left( pos );
+		else
+		    wrkDir = "";
+
+		link->SetWorkingDirectory( wrkDir );
+		if( description.length() )
+		    link->SetDescription( description.local8Bit() );
+		if( arguments.length() )
+		    link->SetArguments( arguments.local8Bit() );
+
+		hr = linkFile->Save( (const wchar_t*)  QString( folderName + QString( "\\" ) + shortcutName ).ucs2(), false );
+
+		linkFile->Release();
+	    }
+	    else
+		qDebug( "Could not get link file interface" );
+
+	    link->Release();
+	}
+	else
+	    qDebug( "Could not instantiate link object" );
+    }
+
+    return hr;
+}
+#endif
+
+#if defined(Q_OS_WIN32)
+void WinShell::createInternetShortcut( QString folderName, bool, QString shortcutName, QString url )
+{
+    // Add .url to shortcut name if needed
+    if( shortcutName.right( 4 ) != ".url" )
+	shortcutName += ".url";
+
+    // ### maybe we should use some Microsoft API instead (IShellLink, e.g.)?
+    QDir dir( folderName );
+    QFile f( dir.filePath( shortcutName ) );
+    if ( f.open( IO_WriteOnly | IO_Translate ) ) {
+	QTextStream ts( &f );
+	ts  << "[InternetShortcut]" << endl
+	    << "URL=" << url;
+    }
+}
+#endif
+
+bool WinShell::createDir( QString fullPath )
+{
+    QStringList hierarchy = QStringList::split( QString( "\\" ), fullPath );
+    QString pathComponent, tmpPath;
+    QDir dirTmp;
+    bool success;
+
+    for( QStringList::Iterator it = hierarchy.begin(); it != hierarchy.end(); ++it ) {
+	pathComponent = *it + "\\";
+	tmpPath += pathComponent;
+	success = dirTmp.mkdir( tmpPath );
+    }
+    return success;
+}
+
+QPixmap* WinShell::getClosedFolderImage()
+{
+    return closedImage;
+}
+
+QPixmap* WinShell::getOpenFolderImage()
+{
+    return openImage;
+}
+
+QPixmap* WinShell::getFileImage()
+{
+    return fileImage;
+}
+
+QPixmap* WinShell::getInfoImage()
+{
+    return infoImage;
+}
+
+#if defined(Q_OS_WIN32)
+QString WinShell::OLESTR2QString( LPOLESTR str )
+{
+    QString tmp;
+
+    for( int i = 0; str[ i ]; i++ )
+	tmp += QChar( str[ i ] );
+
+    return tmp;
+}
+#endif
+
+/*!
+  Returns the free space for the directory.  The space is returned in bytes,
+  and should only be considered valid for this particular directory.
+*/
+#if defined(Q_OS_WIN32)
+ULARGE_INTEGER WinShell::dirFreeSpace( QString dirPath )
+{
+    QString drive = dirPath.left( dirPath.find( '\\' ) );
+    ULARGE_INTEGER freeSpace;
+
+    freeSpace.QuadPart = 0;
+
+    if( GetProcAddress( GetModuleHandleA( "kernel32.dll" ), "GetDiskFreeSpaceExA" ) ) {
+	ULARGE_INTEGER ulBytesAvailable, ulBytesTotal;
+	if( GetDiskFreeSpaceExA( drive.local8Bit(), &ulBytesAvailable, &ulBytesTotal, NULL ) )
+	    freeSpace = ulBytesAvailable;
+    }
+    else if( GetProcAddress( GetModuleHandleA( "kernel32.dll" ), "GetDiskFreeSpaceA" ) ) {
+	DWORD dwSPC, dwBPS, dwClusters, dwTotalClusters;
+	if( GetDiskFreeSpaceA( drive.local8Bit(), &dwSPC, &dwBPS, &dwClusters, &dwTotalClusters ) )
+	    freeSpace.QuadPart = dwSPC * dwBPS * dwClusters;
+    }
+    return freeSpace;
+}
+#elif defined(Q_OS_MAC)
+long  WinShell::dirFreeSpace( QString dirPath )
+{
+    struct statfs buf;
+    if (statfs( dirPath.local8Bit(), &buf ) != -1)
+	return buf.f_bavail * buf.f_bsize;
+    return 0;
+}
+#endif
+