desktop.cc

   1 /*
   2  * Copyright (c) 2000, 2001 Red Hat, Inc.
   3  *
   4  *     This program is free software; you can redistribute it and/or modify
   5  *     it under the terms of the GNU General Public License as published by
   6  *     the Free Software Foundation; either version 2 of the License, or
   7  *     (at your option) any later version.
   8  *
   9  *     A copy of the GNU General Public License can be found at
  10  *     http://www.gnu.org/
  11  *
  12  * Written by DJ Delorie <dj@cygnus.com>
  13  *
  14  */
  15
  16 /* The purpose of this file is to manage all the desktop setup, such
  17    as start menu, batch files, desktop icons, and shortcuts.  Note
  18    that unlike other do_* functions, this one is called directly from
  19    install.cc */
  20
  21 #include "win32.h"
  22 #include <shlobj.h>
  23 #include "desktop.h"
  24 #include "propsheet.h"
  25
  26 #include <stdio.h>
  27 #include <stdlib.h>
  28 #include <unistd.h>
  29
  30 #include "ini.h"
  31 #include "resource.h"
  32 #include "state.h"
  33 #include "dialog.h"
  34 #include "mount.h"
  35 #include "mklink2.h"
  36 #include "package_db.h"
  37 #include "package_meta.h"
  38 #include "filemanip.h"
  39 #include "io_stream.h"
  40 #include "getopt++/BoolOption.h"
  41 #include "LogFile.h"
  42
  43 BoolOption NoShortcutsOption (false, 'n', "no-shortcuts", IDS_HELPTEXT_NO_SHORTCUTS);
  44 BoolOption NoStartMenuOption (false, 'N', "no-startmenu", IDS_HELPTEXT_NO_STARTMENU);
  45 static BoolOption NoDesktopOption (false, 'd', "no-desktop", IDS_HELPTEXT_NO_DESKTOP);
  46
  47 /* Lines starting with '@' are conditionals - include 'N' for NT,
  48    '5' for Win95, '8' for Win98, '*' for all, like this:
  49         echo foo
  50         @N8
  51         echo NT or 98
  52         @*
  53    */
  54
  55 static std::string batname;
  56 static ControlAdjuster::ControlInfo DesktopControlsInfo[] = {
  57   {IDC_DESKTOP_SEPARATOR,       CP_STRETCH, CP_BOTTOM},
  58   {IDC_STATUS,                  CP_LEFT, CP_BOTTOM},
  59   {IDC_STATUS_HEADER,           CP_LEFT, CP_BOTTOM},
  60   {0, CP_LEFT, CP_TOP}
  61 };
  62
  63 DesktopSetupPage::DesktopSetupPage ()
  64 {
  65   sizeProcessor.AddControlInfo (DesktopControlsInfo);
  66 }
  67
  68 static void
  69 make_link (const std::string& linkpath,
  70            const std::string& title,
  71            const std::string& target,
  72            const std::string& arg,
  73            const std::string& icon)
  74 {
  75   std::string fname = linkpath + "/" + title + ".lnk";
  76
  77   if (_access (fname.c_str(), 0) == 0)
  78     return;                     /* already exists */
  79
  80   LogBabblePrintf ("make_link %s, %s, %s\n",
  81        fname.c_str(), title.c_str(), target.c_str());
  82
  83   io_stream::mkpath_p (PATH_TO_FILE, std::string ("file://") + fname, 0);
  84
  85   std::string exepath;
  86   std::string argbuf;
  87
  88   exepath = target;
  89   argbuf = arg;
  90
  91   LogBabblePrintf ("make_link_2 (%s, %s, %s, %s)",
  92        exepath.c_str(), argbuf.c_str(),
  93        icon.c_str(), fname.c_str());
  94   make_link_2 (exepath.c_str(), argbuf.c_str(),
  95                icon.c_str(), fname.c_str());
  96 }
  97
  98 const char *startmenusuffix()
  99 {
 100   if (!is_64bit && (WowNativeMachine() != IMAGE_FILE_MACHINE_I386))
 101     return " (32-bit)";
 102 #ifdef __x86_64__
 103   if (WowNativeMachine() != IMAGE_FILE_MACHINE_AMD64)
 104     return " (x86_64)";
 105 #endif
 106   else
 107     return "";
 108 }
 109
 110 static const char *startmenudir()
 111 {
 112   return "/Cygwin";
 113 }
 114
 115 static void
 116 start_menu (const std::string& title, const std::string& target,
 117             const std::string& arg, const std::string& iconpath)
 118 {
 119   /* Special folders always < MAX_PATH. */
 120   char path[MAX_PATH];
 121   LPITEMIDLIST id;
 122   int issystem = (root_scope == IDC_ROOT_SYSTEM) ? 1 : 0;
 123   SHGetSpecialFolderLocation (NULL,
 124                               issystem ? CSIDL_COMMON_PROGRAMS :
 125                               CSIDL_PROGRAMS, &id);
 126   SHGetPathFromIDList (id, path);
 127   strncat (path, startmenudir(), MAX_PATH - strlen(path) - 1);
 128   strncat (path, startmenusuffix(), MAX_PATH - strlen(path) - 1);
 129   LogBabblePrintf ("Program directory for program link: %s", path);
 130   make_link (path, title, target, arg, iconpath);
 131 }
 132
 133 static void
 134 desktop_icon (const std::string& title, const std::string& target,
 135               const std::string& arg, const std::string& iconpath)
 136 {
 137   /* Special folders always < MAX_PATH. */
 138   char path[MAX_PATH];
 139   LPITEMIDLIST id;
 140   int issystem = (root_scope == IDC_ROOT_SYSTEM) ? 1 : 0;
 141   SHGetSpecialFolderLocation (NULL,
 142                               issystem ? CSIDL_COMMON_DESKTOPDIRECTORY :
 143                               CSIDL_DESKTOPDIRECTORY, &id);
 144   SHGetPathFromIDList (id, path);
 145   LogBabblePrintf ("Desktop directory for desktop link: %s", path);
 146   make_link (path, title, target, arg, iconpath);
 147 }
 148
 149 static void
 150 make_cygwin_bat ()
 151 {
 152   batname = backslash (cygpath ("/Cygwin.bat"));
 153   FILE *bat;
 154
 155   size_t len = batname.size () + 7;
 156   WCHAR wname[len];
 157   mklongpath (wname, batname.c_str (), len);
 158
 159   /* if the batch file exists, don't overwrite it */
 160   if (GetFileAttributesW (wname) != INVALID_FILE_ATTRIBUTES)
 161     return;
 162
 163   bat = nt_wfopen (wname, "wt", 0755);
 164
 165   if (!bat)
 166     return;
 167
 168   fprintf (bat, "@echo off\n\n");
 169
 170   fprintf (bat, "%.2s\n", get_root_dir ().c_str());
 171   fprintf (bat, "chdir %s\n\n",
 172            replace(backslash(get_root_dir() + "/bin"), "%", "%%").c_str());
 173
 174   fprintf (bat, "bash --login -i\n");
 175
 176   fclose (bat);
 177 }
 178
 179 static void
 180 save_icon (std::string &iconpath, const char *resource_name)
 181 {
 182   HRSRC rsrc = FindResource (NULL, resource_name, "FILE");
 183   if (rsrc == NULL)
 184     {
 185       Log (LOG_PLAIN) << "FindResource failed" << GetLastError() << endLog;
 186       return;
 187     }
 188
 189   HGLOBAL res = LoadResource (NULL, rsrc);
 190   char *data = (char *) LockResource (res);
 191   int len = SizeofResource (NULL, rsrc);
 192
 193   FILE *f;
 194   WIN32_FILE_ATTRIBUTE_DATA attr;
 195
 196   size_t ilen = iconpath.size () + 7;
 197   WCHAR wname[ilen];
 198   mklongpath (wname, iconpath.c_str (), ilen);
 199   if (GetFileAttributesExW (wname, GetFileExInfoStandard, &attr)
 200       && attr.dwFileAttributes != INVALID_FILE_ATTRIBUTES
 201       && attr.nFileSizeLow > 7022)      /* Size of old icon */
 202     return;
 203
 204   f = nt_wfopen (wname, "wb", 0644);
 205   if (f)
 206     {
 207       fwrite (data, 1, len, f);
 208       fclose (f);
 209     }
 210 }
 211
 212 #define TARGET          "/bin/mintty"
 213 #define DEFAULTICON     "/Cygwin.ico"
 214 #define TERMINALICON    "/Cygwin-Terminal.ico"
 215 #define TERMINALTITLE   (is_64bit ? "Cygwin64 Terminal" \
 216                                   : "Cygwin Terminal")
 217
 218 static void
 219 do_desktop_setup ()
 220 {
 221   std::string target = backslash (cygpath (TARGET));
 222   std::string defaulticon = backslash (cygpath (DEFAULTICON));
 223   std::string terminalicon = backslash (cygpath (TERMINALICON));
 224
 225   save_icon (defaulticon, "CYGWIN.ICON");
 226   save_icon (terminalicon, "CYGWIN-TERMINAL.ICON");
 227
 228   make_cygwin_bat ();
 229
 230   if (root_menu)
 231     start_menu (TERMINALTITLE, target, "-i " TERMINALICON " -", terminalicon);
 232
 233   if (root_desktop)
 234     desktop_icon (TERMINALTITLE, target, "-i " TERMINALICON " -", terminalicon);
 235 }
 236
 237 static int da[] = { IDC_ROOT_DESKTOP, 0 };
 238 static int ma[] = { IDC_ROOT_MENU, 0 };
 239
 240 static void
 241 check_if_enable_next (HWND h)
 242 {
 243   EnableWindow (GetDlgItem (h, IDOK), 1);
 244 }
 245
 246 static void
 247 set_status (HWND h)
 248 {
 249   std::wstring fmt = LoadStringW(Logger ().getExitMsg ());
 250   std::wstring buf = format(fmt, backslash (Logger ().getFileName (LOG_BABBLE)).c_str ());
 251   eset (h, IDC_STATUS, buf);
 252 }
 253
 254 static void
 255 load_dialog (HWND h)
 256 {
 257   if (source == IDC_SOURCE_DOWNLOAD)
 258     {
 259       // Don't need the checkboxes
 260       EnableWindow (GetDlgItem (h, IDC_ROOT_DESKTOP), FALSE);
 261       EnableWindow (GetDlgItem (h, IDC_ROOT_MENU), FALSE);
 262       ShowWindow (GetDlgItem(h, IDC_DESKTOP_HEADER_DOWNLOAD), SW_SHOW);
 263       ShowWindow (GetDlgItem(h, IDC_DESKTOP_HEADER_TITLE_DOWNLOAD), SW_SHOW);
 264       ShowWindow (GetDlgItem(h, IDC_DESKTOP_HEADER_INSTALL), SW_HIDE);
 265       ShowWindow (GetDlgItem(h, IDC_DESKTOP_HEADER_TITLE_INSTALL), SW_HIDE);
 266      }
 267   else
 268     {
 269       EnableWindow (GetDlgItem (h, IDC_ROOT_DESKTOP), TRUE);
 270       EnableWindow (GetDlgItem (h, IDC_ROOT_MENU), TRUE);
 271       ShowWindow (GetDlgItem(h, IDC_DESKTOP_HEADER_DOWNLOAD), SW_HIDE);
 272       ShowWindow (GetDlgItem(h, IDC_DESKTOP_HEADER_TITLE_DOWNLOAD), SW_HIDE);
 273       ShowWindow (GetDlgItem(h, IDC_DESKTOP_HEADER_INSTALL), SW_SHOW);
 274       ShowWindow (GetDlgItem(h, IDC_DESKTOP_HEADER_TITLE_INSTALL), SW_SHOW);
 275       rbset (h, da, root_desktop);
 276       rbset (h, ma, root_menu);
 277     }
 278   check_if_enable_next (h);
 279   set_status (h);
 280 }
 281
 282 static int
 283 check_desktop (const std::string title, const std::string target)
 284 {
 285   /* Special folders always < MAX_PATH. */
 286   char path[MAX_PATH];
 287   LPITEMIDLIST id;
 288   int issystem = (root_scope == IDC_ROOT_SYSTEM) ? 1 : 0;
 289   SHGetSpecialFolderLocation (NULL,
 290                               issystem ? CSIDL_COMMON_DESKTOPDIRECTORY :
 291                               CSIDL_DESKTOPDIRECTORY, &id);
 292   SHGetPathFromIDList (id, path);
 293   LogBabblePrintf ("Desktop directory for desktop link: %s", path);
 294   std::string fname = std::string(path) + "/" + title + ".lnk";
 295
 296   if (_access (fname.c_str(), 0) == 0)
 297     return 0;                   /* already exists */
 298
 299   return IDC_ROOT_DESKTOP;
 300 }
 301
 302 static int
 303 check_startmenu (const std::string title, const std::string target)
 304 {
 305   /* Special folders always < MAX_PATH. */
 306   char path[MAX_PATH];
 307   LPITEMIDLIST id;
 308   int issystem = (root_scope == IDC_ROOT_SYSTEM) ? 1 : 0;
 309   SHGetSpecialFolderLocation (NULL,
 310                               issystem ? CSIDL_COMMON_PROGRAMS :
 311                               CSIDL_PROGRAMS, &id);
 312   SHGetPathFromIDList (id, path);
 313   LogBabblePrintf ("Program directory for program link: %s", path);
 314   strcat (path, startmenudir());
 315   strcat (path, startmenusuffix());
 316   std::string fname = std::string(path) + "/" + title + ".lnk";
 317
 318   if (_access (fname.c_str(), 0) == 0)
 319     return 0;                   /* already exists */
 320
 321   return IDC_ROOT_MENU;
 322 }
 323
 324 static void
 325 save_dialog (HWND h)
 326 {
 327   root_desktop = rbget (h, da);
 328   root_menu = rbget (h, ma);
 329 }
 330
 331 static BOOL
 332 dialog_cmd (HWND h, int id, HWND hwndctl, UINT code)
 333 {
 334   switch (id)
 335     {
 336
 337     case IDC_ROOT_DESKTOP:
 338     case IDC_ROOT_MENU:
 339       save_dialog (h);
 340       check_if_enable_next (h);
 341       break;
 342     }
 343   return 0;
 344 }
 345
 346 bool
 347 DesktopSetupPage::Create ()
 348 {
 349   return PropertyPage::Create (NULL, dialog_cmd, IDD_DESKTOP);
 350 }
 351
 352 void
 353 DesktopSetupPage::OnInit ()
 354 {
 355   SetDlgItemFont(IDC_STATUS_HEADER, "MS Shell Dlg", 8, FW_BOLD);
 356 }
 357
 358 void
 359 DesktopSetupPage::OnActivate ()
 360 {
 361   if (NoShortcutsOption || source == IDC_SOURCE_DOWNLOAD)
 362     {
 363       root_desktop = root_menu = 0;
 364     }
 365   else
 366     {
 367       std::string target = backslash (cygpath (TARGET));
 368
 369       if (NoStartMenuOption)
 370         {
 371           root_menu = 0;
 372         }
 373       else
 374         {
 375           root_menu = check_startmenu (TERMINALTITLE, target);
 376         }
 377
 378       if (NoDesktopOption)
 379         {
 380           root_desktop = 0;
 381         }
 382       else
 383         {
 384           root_desktop = check_desktop (TERMINALTITLE, target);
 385         }
 386     }
 387
 388   load_dialog (GetHWND ());
 389 }
 390
 391 long
 392 DesktopSetupPage::OnBack ()
 393 {
 394   HWND h = GetHWND ();
 395   save_dialog (h);
 396   return IDD_CHOOSE;
 397 }
 398
 399 bool
 400 DesktopSetupPage::OnFinish ()
 401 {
 402   HWND h = GetHWND ();
 403   save_dialog (h);
 404   if (source != IDC_SOURCE_DOWNLOAD)
 405     do_desktop_setup ();
 406
 407   return true;
 408 }
 409
 410 long
 411 DesktopSetupPage::OnUnattended ()
 412 {
 413   Window::PostMessageNow (WM_APP_UNATTENDED_FINISH);
 414   // GetOwner ()->PressButton(PSBTN_FINISH);
 415   return -1;
 416 }
 417
 418 bool
 419 DesktopSetupPage::OnMessageApp (UINT uMsg, WPARAM wParam, LPARAM lParam)
 420 {
 421   switch (uMsg)
 422     {
 423     case WM_APP_UNATTENDED_FINISH:
 424       {
 425         GetOwner ()->PressButton(PSBTN_FINISH);
 426         break;
 427       }
 428     default:
 429       {
 430         // Not handled
 431         return false;
 432       }
 433     }
 434
 435   return true;
 436 }