ryzom/tools/client/ryzom_installer/src/main.cpp

   1 // Ryzom - MMORPG Framework <http://dev.ryzom.com/projects/ryzom/>
   2 // Copyright (C) 2010  Winch Gate Property Limited
   3 //
   4 // This program is free software: you can redistribute it and/or modify
   5 // it under the terms of the GNU Affero General Public License as
   6 // published by the Free Software Foundation, either version 3 of the
   7 // License, or (at your option) any later version.
   8 //
   9 // This program is distributed in the hope that it will be useful,
  10 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  11 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  12 // GNU Affero General Public License for more details.
  13 //
  14 // You should have received a copy of the GNU Affero General Public License
  15 // along with this program.  If not, see <http://www.gnu.org/licenses/>.
  16
  17 #include "stdpch.h"
  18 #include "mainwindow.h"
  19 #include "configfile.h"
  20 #include "migratedialog.h"
  21 #include "installdialog.h"
  22 #include "uninstalldialog.h"
  23 #include "operationdialog.h"
  24 #include "utils.h"
  25
  26 #ifdef HAVE_CONFIG_H
  27 #include "config.h"
  28 #endif
  29
  30 #ifdef QT_STATICPLUGIN
  31
  32 #include <QtPlugin>
  33
  34 #if defined(Q_OS_WIN32)
  35         Q_IMPORT_PLUGIN(QWindowsIntegrationPlugin)
  36 #elif defined(Q_OS_MAC)
  37         Q_IMPORT_PLUGIN(QCocoaIntegrationPlugin)
  38 #elif defined(Q_OS_UNIX)
  39         Q_IMPORT_PLUGIN(QXcbIntegrationPlugin)
  40 #endif
  41
  42         Q_IMPORT_PLUGIN(QICOPlugin)
  43
  44 #endif
  45
  46 #ifdef DEBUG_NEW
  47         #define new DEBUG_NEW
  48 #endif
  49
  50 // copy all specified files from current directory to destination directory
  51 bool copyInstallerFiles(const QStringList &files, const QString &destination)
  52 {
  53         QString path = QApplication::applicationDirPath();
  54
  55         foreach(const QString &file, files)
  56         {
  57                 // convert to absolute path
  58                 QString srcPath = path + "/" + file;
  59                 QString dstPath = destination + "/" + file;
  60
  61                 if (QFile::exists(srcPath))
  62                 {
  63                         if (QFile::exists(dstPath))
  64                         {
  65                                 if (!QFile::remove(dstPath))
  66                                 {
  67                                         nlwarning("Unable to delete %s", Q2C(dstPath));
  68                                 }
  69                         }
  70
  71                         if (!QFile::copy(srcPath, dstPath))
  72                         {
  73                                 nlwarning("Unable to copy %s to %s", Q2C(srcPath), Q2C(dstPath));
  74
  75                                 return false;
  76                         }
  77                 }
  78         }
  79
  80         return true;
  81 }
  82
  83 int main(int argc, char *argv[])
  84 {
  85 #if defined(_MSC_VER) && defined(_DEBUG)
  86         _CrtSetDbgFlag (_CRTDBG_ALLOC_MEM_DF | _CRTDBG_LEAK_CHECK_DF);
  87 #endif
  88
  89 #ifdef Q_OS_WIN
  90         CCOMHelper comHelper;
  91 #endif
  92
  93         NLMISC::CApplicationContext appContext;
  94
  95         QApplication app(argc, argv);
  96
  97         QApplication::setApplicationName("Ryzom");
  98         QApplication::setApplicationVersion(RYZOM_VERSION);
  99         QApplication::setWindowIcon(QIcon(":/icons/ryzom.ico"));
 100
 101         // remove first argument because it's not really an argument
 102         QStringList args = QApplication::arguments();
 103         args.removeFirst();
 104
 105         QLocale locale = QLocale::system();
 106
 107         // load application translations
 108         QTranslator localTranslator;
 109         if (localTranslator.load(locale, "ryzom_installer", "_", ":/translations"))
 110         {
 111                 QApplication::installTranslator(&localTranslator);
 112         }
 113
 114         // load Qt default translations
 115         QTranslator qtTranslator;
 116         if (qtTranslator.load(locale, "qtbase", "_", ":/translations"))
 117         {
 118                 QApplication::installTranslator(&qtTranslator);
 119         }
 120
 121         // define commandline arguments
 122         QCommandLineParser parser;
 123         parser.setApplicationDescription(QApplication::tr("Installation and launcher tool for Ryzom"));
 124         parser.addHelpOption();
 125
 126         QCommandLineOption uninstallOption(QStringList() << "u" << "uninstall", QApplication::tr("Uninstall"));
 127         parser.addOption(uninstallOption);
 128
 129         QCommandLineOption silentOption(QStringList() << "s" << "silent", QApplication::tr("Silent mode"));
 130         parser.addOption(silentOption);
 131
 132         QCommandLineOption versionOption(QStringList() << "v" << "version", QApplication::tr("Version"));
 133         parser.addOption(versionOption);
 134
 135         QCommandLineOption installOption(QStringList() << "i" << "install", QApplication::tr("Install itself"));
 136         parser.addOption(installOption);
 137
 138         // process the actual command line arguments given by the user
 139         parser.process(app);
 140
 141         // don't need to load config file for version
 142         if (parser.isSet(versionOption))
 143         {
 144                 printf("Ryzom Installer %s (built on %s)\nCopyright (C) %s\n", RYZOM_VERSION, BUILD_DATE, COPYRIGHT);
 145
 146                 return 0;
 147         }
 148
 149         // instanciate ConfigFile
 150         CConfigFile config;
 151
 152         bool res = config.load();
 153
 154         // init log
 155         CLogHelper logHelper(config.getInstallationDirectory().isEmpty() ? config.getNewInstallationDirectory():config.getInstallationDirectory());
 156
 157         nlinfo("Launched %s", Q2C(config.getInstallerCurrentFilePath()));
 158
 159         OperationStep step = res ? config.getInstallNextStep():DisplayNoServerError;
 160
 161         if (step == DisplayNoServerError)
 162         {
 163                 QMessageBox::critical(NULL, QApplication::tr("Error"), QApplication::tr("Unable to find ryzom_installer.ini"));
 164                 return 1;
 165         }
 166
 167 #if defined(Q_OS_WIN) && !defined(_DEBUG)
 168         // under Windows, Ryzom Installer should always be copied in TEMP directory
 169         QString tempPath = QStandardPaths::writableLocation(QStandardPaths::TempLocation);
 170
 171         // check if launched from TEMP directory
 172         if (step == Done && !config.getInstallerCurrentDirPath().startsWith(tempPath))
 173         {
 174                 nlinfo("Not launched from TEMP directory");
 175
 176                 // try to delete all temporary installers
 177                 QDir tempDir(tempPath);
 178
 179                 QStringList filter;
 180                 filter << "ryzom_installer_*";
 181
 182                 QStringList dirs = tempDir.entryList(filter, QDir::Dirs);
 183
 184                 foreach(const QString &dir, dirs)
 185                 {
 186                         // delete each directory
 187                         QDir dirToRemove(tempDir);
 188                         dirToRemove.cd(dir);
 189                         dirToRemove.removeRecursively();
 190
 191                         nlinfo("Delete directory %s", Q2C(dir));
 192                 }
 193
 194                 tempPath += QString("/ryzom_installer_%1").arg(QDateTime::currentMSecsSinceEpoch());
 195
 196                 nlinfo("Creating directory %s", Q2C(tempPath));
 197
 198                 // copy installer and required files to TEMP directory
 199                 if (QDir().mkdir(tempPath) && copyInstallerFiles(config.getInstallerRequiredFiles(), tempPath))
 200                 {
 201                         QString tempFile = tempPath + "/" + QFileInfo(config.getInstallerCurrentFilePath()).fileName();
 202
 203                         nlinfo("Launching %s", Q2C(tempFile));
 204
 205                         // launch copy in TEMP directory with same arguments
 206                         if (QProcess::startDetached(tempFile, args, tempPath)) return 0;
 207
 208                         nlwarning("Unable to launch %s", Q2C(tempFile));
 209                 }
 210         }
 211 #endif
 212
 213         // use product name from ryzom_installer.ini
 214         if (!config.getProductName().isEmpty()) QApplication::setApplicationName(config.getProductName());
 215
 216         if (parser.isSet(uninstallOption))
 217         {
 218                 nlinfo("Uninstalling...");
 219
 220                 SComponents components;
 221
 222                 // add all servers by default
 223                 for (int i = 0; i < config.getServersCount(); ++i)
 224                 {
 225                         components.servers << config.getServer(i).id;
 226                 }
 227
 228                 // show uninstall wizard dialog if not in silent mode
 229                 if (!parser.isSet(silentOption))
 230                 {
 231                         CUninstallDialog dialog;
 232
 233                         dialog.setSelectedComponents(components);
 234
 235                         // exit if press Cancel button or close dialog
 236                         if (!dialog.exec()) return 0;
 237
 238                         components = dialog.getSelectedCompenents();
 239                 }
 240
 241                 COperationDialog dialog;
 242
 243                 dialog.setCurrentServerId(config.getProfile().server);
 244                 dialog.setOperation(OperationUninstall);
 245                 dialog.setUninstallComponents(components);
 246
 247                 // TODO: set all components to uninstall
 248
 249                 return dialog.exec() ? 0 : 1;
 250         }
 251
 252         if (step == ShowMigrateWizard)
 253         {
 254                 nlinfo("Display migration dialog");
 255 #ifdef Q_OS_WIN32
 256                 CMigrateDialog dialog;
 257
 258                 if (!dialog.exec()) return 1;
 259
 260                 step = config.getInstallNextStep();
 261 #else
 262                 nlwarning("Migration disabled under Linux and OS X");
 263 #endif
 264         }
 265         else if (step == ShowInstallWizard)
 266         {
 267                 nlinfo("Display installation dialog");
 268
 269                 CInstallDialog dialog;
 270
 271                 if (!dialog.exec()) return 1;
 272
 273                 step = config.getInstallNextStep();
 274         }
 275
 276         nlinfo("Next step is %s", Q2C(stepToString(step)));
 277
 278         bool restartInstaller = false;
 279
 280         if (step != Done)
 281         {
 282                 COperationDialog dialog;
 283                 dialog.setCurrentServerId(config.getProfile().server);
 284                 dialog.setOperation(config.getSrcServerDirectory().isEmpty() ? OperationInstall:OperationMigrate);
 285
 286                 if (!dialog.exec()) return 1;
 287
 288                 step = config.getInstallNextStep();
 289
 290                 nlinfo("Last step is %s", Q2C(stepToString(step)));
 291
 292                 if (step == LaunchInstalledInstaller)
 293                 {
 294                         // restart more recent installed Installer version
 295                         restartInstaller = true;
 296                 }
 297                 else if (step == Done)
 298                 {
 299 #if defined(Q_OS_WIN) && !defined(_DEBUG)
 300                         // restart Installer, so it could be copied in TEMP and allowed to update itself
 301                         restartInstaller = true;
 302 #endif
 303                 }
 304         }
 305
 306         if (restartInstaller)
 307         {
 308 #ifndef _DEBUG
 309                 nlinfo("Restart Installer %s", Q2C(config.getInstallerInstalledFilePath()));
 310
 311 #ifndef Q_OS_WIN32
 312                 // fix executable permissions under UNIX
 313                 QFile::setPermissions(config.getInstallerInstalledFilePath(), QFile::permissions(config.getInstallerInstalledFilePath()) | QFile::ExeGroup | QFile::ExeUser | QFile::ExeOther);
 314 #endif
 315
 316                 if (QProcess::startDetached(config.getInstallerInstalledFilePath(), args, config.getInstallationDirectory())) return 0;
 317
 318                 nlwarning("Unable to restart Installer %s", Q2C(config.getInstallerInstalledFilePath()));
 319 #endif
 320         }
 321
 322         CMainWindow mainWindow;
 323         mainWindow.show();
 324
 325         return QApplication::exec();
 326 }