Home | History | Annotate | Download | only in win
      1 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
      2 // Use of this source code is governed by a BSD-style license that can be
      3 // found in the LICENSE file.
      4 
      5 #ifndef BASE_WIN_SHORTCUT_H_
      6 #define BASE_WIN_SHORTCUT_H_
      7 
      8 #include <windows.h>
      9 
     10 #include "base/files/file_path.h"
     11 #include "base/logging.h"
     12 #include "base/strings/string16.h"
     13 
     14 namespace base {
     15 namespace win {
     16 
     17 enum ShortcutOperation {
     18   // Create a new shortcut (overwriting if necessary).
     19   SHORTCUT_CREATE_ALWAYS = 0,
     20   // Overwrite an existing shortcut (fails if the shortcut doesn't exist).
     21   // If the arguments are not specified on the new shortcut, keep the old
     22   // shortcut's arguments.
     23   SHORTCUT_REPLACE_EXISTING,
     24   // Update specified properties only on an existing shortcut.
     25   SHORTCUT_UPDATE_EXISTING,
     26 };
     27 
     28 // Properties for shortcuts. Properties set will be applied to the shortcut on
     29 // creation/update, others will be ignored.
     30 // Callers are encouraged to use the setters provided which take care of
     31 // setting |options| as desired.
     32 struct ShortcutProperties {
     33   enum IndividualProperties {
     34     PROPERTIES_TARGET = 1 << 0,
     35     PROPERTIES_WORKING_DIR = 1 << 1,
     36     PROPERTIES_ARGUMENTS = 1 << 2,
     37     PROPERTIES_DESCRIPTION = 1 << 3,
     38     PROPERTIES_ICON = 1 << 4,
     39     PROPERTIES_APP_ID = 1 << 5,
     40     PROPERTIES_DUAL_MODE = 1 << 6,
     41   };
     42 
     43   ShortcutProperties() : icon_index(-1), dual_mode(false), options(0U) {}
     44 
     45   void set_target(const FilePath& target_in) {
     46     target = target_in;
     47     options |= PROPERTIES_TARGET;
     48   }
     49 
     50   void set_working_dir(const FilePath& working_dir_in) {
     51     working_dir = working_dir_in;
     52     options |= PROPERTIES_WORKING_DIR;
     53   }
     54 
     55   void set_arguments(const string16& arguments_in) {
     56     // Size restriction as per MSDN at http://goo.gl/TJ7q5.
     57     DCHECK(arguments_in.size() < MAX_PATH);
     58     arguments = arguments_in;
     59     options |= PROPERTIES_ARGUMENTS;
     60   }
     61 
     62   void set_description(const string16& description_in) {
     63     // Size restriction as per MSDN at http://goo.gl/OdNQq.
     64     DCHECK(description_in.size() < MAX_PATH);
     65     description = description_in;
     66     options |= PROPERTIES_DESCRIPTION;
     67   }
     68 
     69   void set_icon(const FilePath& icon_in, int icon_index_in) {
     70     icon = icon_in;
     71     icon_index = icon_index_in;
     72     options |= PROPERTIES_ICON;
     73   }
     74 
     75   void set_app_id(const string16& app_id_in) {
     76     app_id = app_id_in;
     77     options |= PROPERTIES_APP_ID;
     78   }
     79 
     80   void set_dual_mode(bool dual_mode_in) {
     81     dual_mode = dual_mode_in;
     82     options |= PROPERTIES_DUAL_MODE;
     83   }
     84 
     85   // The target to launch from this shortcut. This is mandatory when creating
     86   // a shortcut.
     87   FilePath target;
     88   // The name of the working directory when launching the shortcut.
     89   FilePath working_dir;
     90   // The arguments to be applied to |target| when launching from this shortcut.
     91   // The length of this string must be less than MAX_PATH.
     92   string16 arguments;
     93   // The localized description of the shortcut.
     94   // The length of this string must be less than MAX_PATH.
     95   string16 description;
     96   // The path to the icon (can be a dll or exe, in which case |icon_index| is
     97   // the resource id).
     98   FilePath icon;
     99   int icon_index;
    100   // The app model id for the shortcut (Win7+).
    101   string16 app_id;
    102   // Whether this is a dual mode shortcut (Win8+).
    103   bool dual_mode;
    104   // Bitfield made of IndividualProperties. Properties set in |options| will be
    105   // set on the shortcut, others will be ignored.
    106   uint32 options;
    107 };
    108 
    109 // This method creates (or updates) a shortcut link at |shortcut_path| using the
    110 // information given through |properties|.
    111 // Ensure you have initialized COM before calling into this function.
    112 // |operation|: a choice from the ShortcutOperation enum.
    113 // If |operation| is SHORTCUT_REPLACE_EXISTING or SHORTCUT_UPDATE_EXISTING and
    114 // |shortcut_path| does not exist, this method is a no-op and returns false.
    115 BASE_EXPORT bool CreateOrUpdateShortcutLink(
    116     const FilePath& shortcut_path,
    117     const ShortcutProperties& properties,
    118     ShortcutOperation operation);
    119 
    120 // Resolve Windows shortcut (.LNK file)
    121 // This methods tries to resolve a shortcut .LNK file. The path of the shortcut
    122 // to resolve is in |shortcut_path|. If |target_path| is not NULL, the target
    123 // will be resolved and placed in |target_path|. If |args| is not NULL, the
    124 // arguments will be retrieved and placed in |args|. The function returns true
    125 // if all requested fields are found successfully.
    126 // Callers can safely use the same variable for both |shortcut_path| and
    127 // |target_path|.
    128 BASE_EXPORT bool ResolveShortcut(const FilePath& shortcut_path,
    129                                  FilePath* target_path,
    130                                  string16* args);
    131 
    132 // Pins a shortcut to the Windows 7 taskbar. The shortcut file must already
    133 // exist and be a shortcut that points to an executable.
    134 BASE_EXPORT bool TaskbarPinShortcutLink(const wchar_t* shortcut);
    135 
    136 // Unpins a shortcut from the Windows 7 taskbar. The shortcut must exist and
    137 // already be pinned to the taskbar.
    138 BASE_EXPORT bool TaskbarUnpinShortcutLink(const wchar_t* shortcut);
    139 
    140 }  // namespace win
    141 }  // namespace base
    142 
    143 #endif  // BASE_WIN_SHORTCUT_H_
    144