开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 书籍源码 > 查看源码
ShortCutDlg.cpp
资源名称：c++时尚编程百例__81-110.rar　[点击查看]
上传用户：qzzxgm
上传日期：2009-12-14
资源大小：1882k
文件大小：13k
源码类别：

书籍源码

开发平台：

Visual C++

ShortCutDlg.cpp：源码内容
  1. // ShortCutDlg.cpp : implementation file
  2. //
  4. #include "stdafx.h"
  5. #include "ShortCut.h"
  6. #include "ShortCutDlg.h"
  7. #include "NameDlg.h"
  8. #ifdef _DEBUG
  9. #define new DEBUG_NEW
  10. #undef THIS_FILE
  11. static char THIS_FILE[] = __FILE__;
  12. #endif
  14. /////////////////////////////////////////////////////////////////////////////
  15. // CAboutDlg dialog used for App About
  17. class CAboutDlg : public CDialog
  18. {
  19. public:
  20. CAboutDlg();
  22. // Dialog Data
  23. //{{AFX_DATA(CAboutDlg)
  24. enum { IDD = IDD_ABOUTBOX };
  25. //}}AFX_DATA
  27. // ClassWizard generated virtual function overrides
  28. //{{AFX_VIRTUAL(CAboutDlg)
  29. protected:
  30. virtual void DoDataExchange(CDataExchange* pDX);    // DDX/DDV support
  31. //}}AFX_VIRTUAL
  33. // Implementation
  34. protected:
  35. //{{AFX_MSG(CAboutDlg)
  36. //}}AFX_MSG
  37. DECLARE_MESSAGE_MAP()
  38. };
  40. CAboutDlg::CAboutDlg() : CDialog(CAboutDlg::IDD)
  41. {
  42. //{{AFX_DATA_INIT(CAboutDlg)
  43. //}}AFX_DATA_INIT
  44. }
  46. void CAboutDlg::DoDataExchange(CDataExchange* pDX)
  47. {
  48. CDialog::DoDataExchange(pDX);
  49. //{{AFX_DATA_MAP(CAboutDlg)
  50. //}}AFX_DATA_MAP
  51. }
  53. BEGIN_MESSAGE_MAP(CAboutDlg, CDialog)
  54. //{{AFX_MSG_MAP(CAboutDlg)
  55. // No message handlers
  56. //}}AFX_MSG_MAP
  57. END_MESSAGE_MAP()
  59. /////////////////////////////////////////////////////////////////////////////
  60. // CShortCutDlg dialog
  62. CShortCutDlg::CShortCutDlg(CWnd* pParent /*=NULL*/)
  63. : CDialog(CShortCutDlg::IDD, pParent)
  64. {
  65. //{{AFX_DATA_INIT(CShortCutDlg)
  66. m_nLocation = 0;
  67. //}}AFX_DATA_INIT
  68. // Note that LoadIcon does not require a subsequent DestroyIcon in Win32
  69. m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME);
  70. }
  72. void CShortCutDlg::DoDataExchange(CDataExchange* pDX)
  73. {
  74. CDialog::DoDataExchange(pDX);
  75. //{{AFX_DATA_MAP(CShortCutDlg)
  76. DDX_Radio(pDX, IDC_RADIO1, m_nLocation);
  77. //}}AFX_DATA_MAP
  78. }
  80. BEGIN_MESSAGE_MAP(CShortCutDlg, CDialog)
  81. //{{AFX_MSG_MAP(CShortCutDlg)
  82. ON_WM_SYSCOMMAND()
  83. ON_WM_PAINT()
  84. ON_WM_QUERYDRAGICON()
  85. ON_BN_CLICKED(IDC_CREATEGROUP, OnCreateGroup)
  86. ON_BN_CLICKED(IDC_CREATEITEM, OnCreateItem)
  87. ON_BN_CLICKED(IDC_DELETEGROUP, OnDeleteGroup)
  88. ON_BN_CLICKED(IDC_DELETEITEM, OnDeleteItem)
  89. ON_WM_DESTROY()
  90. //}}AFX_MSG_MAP
  91. END_MESSAGE_MAP()
  93. /////////////////////////////////////////////////////////////////////////////
  94. // CShortCutDlg message handlers
  96. BOOL CShortCutDlg::OnInitDialog()
  97. {
  98. CDialog::OnInitDialog();
  100. // Add "About..." menu item to system menu.
  102. // IDM_ABOUTBOX must be in the system command range.
  103. ASSERT((IDM_ABOUTBOX & 0xFFF0) == IDM_ABOUTBOX);
  104. ASSERT(IDM_ABOUTBOX < 0xF000);
  106. CMenu* pSysMenu = GetSystemMenu(FALSE);
  107. if (pSysMenu != NULL)
  108. {
  109. CString strAboutMenu;
  110. strAboutMenu.LoadString(IDS_ABOUTBOX);
  111. if (!strAboutMenu.IsEmpty())
  112. {
  113. pSysMenu->AppendMenu(MF_SEPARATOR);
  114. pSysMenu->AppendMenu(MF_STRING, IDM_ABOUTBOX, strAboutMenu);
  115. }
  116. }
  118. // Set the icon for this dialog.  The framework does this automatically
  119. //  when the application's main window is not a dialog
  120. SetIcon(m_hIcon, TRUE); // Set big icon
  121. SetIcon(m_hIcon, FALSE); // Set small icon
  123. // TODO: Add extra initialization here
  124. CoInitialize (NULL);
  125.     
  126. return TRUE;  // return TRUE  unless you set the focus to a control
  127. }
  129. void CShortCutDlg::OnSysCommand(UINT nID, LPARAM lParam)
  130. {
  131. if ((nID & 0xFFF0) == IDM_ABOUTBOX)
  132. {
  133. CAboutDlg dlgAbout;
  134. dlgAbout.DoModal();
  135. }
  136. else
  137. {
  138. CDialog::OnSysCommand(nID, lParam);
  139. }
  140. }
  142. // If you add a minimize button to your dialog, you will need the code below
  143. //  to draw the icon.  For MFC applications using the document/view model,
  144. //  this is automatically done for you by the framework.
  146. void CShortCutDlg::OnPaint() 
  147. {
  148. if (IsIconic())
  149. {
  150. CPaintDC dc(this); // device context for painting
  152. SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0);
  154. // Center icon in client rectangle
  155. int cxIcon = GetSystemMetrics(SM_CXICON);
  156. int cyIcon = GetSystemMetrics(SM_CYICON);
  157. CRect rect;
  158. GetClientRect(&rect);
  159. int x = (rect.Width() - cxIcon + 1) / 2;
  160. int y = (rect.Height() - cyIcon + 1) / 2;
  162. // Draw the icon
  163. dc.DrawIcon(x, y, m_hIcon);
  164. }
  165. else
  166. {
  167. CDialog::OnPaint();
  168. }
  169. }
  171. // The system calls this to obtain the cursor to display while the user drags
  172. //  the minimized window.
  173. HCURSOR CShortCutDlg::OnQueryDragIcon()
  174. {
  175. return (HCURSOR) m_hIcon;
  176. }
  177. //浏览文件夹
  178. BOOL CShortCutDlg::BrowseForFolder(
  179.     LPITEMIDLIST pidlRoot,//浏览开始处的PIDL
  180.     LPITEMIDLIST *ppidlDestination,
  181.                 //浏览结束时所选择的PIDL
  182.     LPCSTR lpszTitle)//浏览对话框中的提示文字
  183. {    
  184. BROWSEINFO BrInfo ;
  186.     ZeroMemory( &BrInfo, sizeof(BrInfo)) ;
  187.     BrInfo.hwndOwner = HWND_DESKTOP ;
  188.     BrInfo.pidlRoot = pidlRoot ;
  189.     BrInfo.lpszTitle = lpszTitle ;
  191.     //浏览文件夹
  192.     *ppidlDestination= SHBrowseForFolder(&BrInfo);
  193.     //用户选择了取消按钮
  194.     if(NULL == *ppidlDestination)
  195.         return FALSE ;
  196.     return TRUE ;
  197. }
  199. //取得快捷方式的目标应用程序名
  200. BOOL CShortCutDlg::SelectMenuItem( LPSTR szFileName)
  201. {
  202.     OPENFILENAME ofn ;
  203.     static CHAR szFilter[] = "Programs*.exe" ;
  205.     ZeroMemory(&ofn, sizeof( OPENFILENAME)) ;
  206.     ofn.lStructSize = sizeof( OPENFILENAME) ;
  207.     ofn.hwndOwner = HWND_DESKTOP;
  208.     ofn.lpstrFilter = szFilter ;
  209.     ofn.nFilterIndex = 0 ;
  210.     ofn.nMaxFile = MAX_PATH ;
  211.     ofn.lpstrTitle = "请选择目标应用程序:" ;
  212.     ofn.lpstrFile = szFileName ;
  213.     ofn.Flags = OFN_FILEMUSTEXIST | 
  214.             OFN_PATHMUSTEXIST | OFN_EXPLORER ;
  215.     //文件浏览
  216.     if(!GetOpenFileName( &ofn))//选择了取消按钮
  217.         return FALSE ;
  218.     return TRUE ;
  219. }
  220. //取得要创建的快捷方式的名字
  221. BOOL CShortCutDlg::GetShortcutCrt(LPSTR szPath)
  222. {
  223.     LPITEMIDLIST pidlBeginAt, pidlDestination ;
  224.     
  225.     // 取得开始菜单或桌面的PIDL
  226.     SHGetSpecialFolderLocation( HWND_DESKTOP, 
  227.                 nBeginAt, &pidlBeginAt) ;
  228.     // 取得要创建的快捷方式所在的位置
  229.     if( !BrowseForFolder(pidlBeginAt,                         &pidlDestination, 
  230.                 "请选择快捷方式所在的位置："))
  231.         return FALSE ;
  232.     // 把PIDL转换为路径名
  233.     SHGetPathFromIDList( pidlDestination, szPath) ;
  234.     // 取得快捷方式名称
  235.     CNameDlg name_dlg;
  236. if(name_dlg.DoModal() == IDCANCEL)    
  237.         return FALSE ;
  238.     //把快捷方式名和扩展名.LNK添加到路径名后
  239.     //形成完整的快捷方式数据文件名
  240.     wsprintf(szPath+lstrlen(szPath),"\%s.lnk", 
  241.                 name_dlg.m_strName) ;
  242.     return TRUE ;
  243. }
  244. //创建快捷方式
  245. BOOL CShortCutDlg::CreateLink ( 
  246.     LPSTR szPath,//快捷方式的目标应用程序名
  247.     LPSTR szLink)//快捷方式的数据文件名(*.lnk)
  248. {
  249. HRESULT hres ;
  250. IShellLink * psl ;
  251. IPersistFile* ppf ;
  252. WORD wsz[ MAX_PATH] ;
  253.     //创建一个IShellLink实例
  254.     hres = CoCreateInstance( CLSID_ShellLink, NULL,
  255.         CLSCTX_INPROC_SERVER, IID_IShellLink,
  256.         (void **)&psl) ;
  257. if( FAILED( hres))
  258.         return FALSE ;
  259.     //设置目标应用程序
  260.     psl -> SetPath( szPath) ;
  261.     //设置快捷键(此处设为Shift+Ctrl+'R')
  262.     psl -> SetHotkey( MAKEWORD( 'R',
  263. HOTKEYF_SHIFT |HOTKEYF_CONTROL)) ;
  264.     //从IShellLink获取其IPersistFile接口
  265.     //用于保存快捷方式的数据文件 (*.lnk)
  266.     hres = psl -> QueryInterface( IID_IPersistFile,
  267. (void**)&ppf) ;
  268. if( FAILED( hres))
  269.         return FALSE ;
  270. // 确保数据文件名为ANSI格式
  271. MultiByteToWideChar( CP_ACP, 0, szLink, -1, 
  272. wsz, MAX_PATH) ;
  273.     //调用IPersistFile::Save
  274.     //保存快捷方式的数据文件 (*.lnk)
  275. hres = ppf -> Save( wsz, STGM_READWRITE) ;
  276.     //释放IPersistFile和IShellLink接口
  277. ppf -> Release( ) ;
  278. psl -> Release( ) ;
  279. return TRUE;
  280. }
  281. BOOL CShortCutDlg::DeleteFolder( LPSTR pszFolder)
  282. {
  283.     SHFILEOPSTRUCT fos ;
  284.     
  285.     ZeroMemory( &fos, sizeof( fos)) ;
  286.     fos.hwnd = HWND_DESKTOP;
  287.     fos.wFunc = FO_DELETE ;
  288.     fos.fFlags = FOF_SILENT | FOF_ALLOWUNDO ;
  289.     fos.pFrom = pszFolder ;
  291.     // 删除文件夹及其内容
  292.     if( 0 != SHFileOperation( &fos))
  293.         return FALSE ;
  294.     return TRUE;
  295. }
  297. //取得要删除的快捷方式
  298. BOOL CShortCutDlg::GetShortcutDel (
  299.         LPSTR lpszInitDir,//选择文件的开始目录
  300.         LPSTR lpszShortcut)//快捷方式名
  301. {
  302.     OPENFILENAME ofn ;
  303.     char szFilter[] = "Shortcuts*.lnk" ;
  305.     ZeroMemory(&ofn,sizeof( OPENFILENAME));
  306.     ofn.lStructSize = sizeof( OPENFILENAME) ;
  307.     ofn.hwndOwner = HWND_DESKTOP ;
  308.     ofn.lpstrFilter = szFilter ;
  309.     ofn.nFilterIndex = 0 ;
  310.     ofn.nMaxFile = MAX_PATH ;
  311.     ofn.lpstrTitle = "请选择要删除的快捷方式：" ;
  312.     ofn.lpstrFile = lpszShortcut;
  313.     ofn.lpstrInitialDir = lpszInitDir ;
  314.     ofn.Flags = OFN_FILEMUSTEXIST | 
  315.         OFN_PATHMUSTEXIST | OFN_EXPLORER |
  316.         OFN_NODEREFERENCELINKS ;
  317.     //文件浏览
  318.     if(! GetOpenFileName( &ofn))//选择了取消按钮
  319.         return FALSE ;
  320.     return TRUE ;
  321. }
  322. //删除快捷方式的数据文件 (*.lnk)
  323. BOOL CShortCutDlg::DeleteLink( LPSTR lpszShortcut)
  324. {
  325.     SHFILEOPSTRUCT fos ;
  327.     ZeroMemory( &fos, sizeof(fos)) ;
  328.     fos.hwnd = HWND_DESKTOP ;
  329.     fos.wFunc = FO_DELETE ;
  330.     fos.pFrom = lpszShortcut;
  331.     fos.pTo = NULL ;
  332.     fos.fFlags = FOF_SILENT | FOF_ALLOWUNDO ;
  333.     //删除快捷方式（*.lnk)
  334.     if( 0 != SHFileOperation( &fos))
  335.         return FALSE ;
  336.     return TRUE ;
  337. }
  338. // 通知shell有关变化
  339. void CShortCutDlg::NotifyShell(LONG wEventId,//事件标志
  340.             LPSTR szPath)//路径
  341. {    
  342.     SHChangeNotify( wEventId,
  343.                     SHCNF_FLUSH | SHCNF_PATH,
  344.                     szPath,0);
  345.     //取得szPath的父目录
  346.     char* p;
  347.     for( p=szPath+lstrlen(szPath)-1;
  348.             *p != '\'; 
  349.             p--);
  350.     *p='';
  351.     SHChangeNotify(SHCNE_UPDATEDIR
  352.             |SHCNE_INTERRUPT, 
  353.             SHCNF_FLUSH | SHCNF_PATH,szPath,0);
  354. }
  356. void CShortCutDlg::OnCreateGroup() 
  357. {
  358. UpdateData(TRUE);
  359. if(m_nLocation==0)
  360. {
  361. //设置起始文件夹为桌面
  362. nBeginAt=CSIDL_DESKTOPDIRECTORY ;    
  363. }
  364. else
  365. {
  366. //设置起始文件夹为“开始”菜单
  367. nBeginAt=CSIDL_STARTMENU;
  368. }
  370. LPITEMIDLIST pidlBeginAt, pidlDestination ;
  371.     char szPath[ MAX_PATH] ;
  373.     // 取得开始菜单或桌面的PIDL
  374.     SHGetSpecialFolderLocation( HWND_DESKTOP, 
  375.             nBeginAt, &pidlBeginAt) ;
  376.     // 取得新建文件夹的父文件夹
  377.     if( !BrowseForFolder(pidlBeginAt , 
  378.             &pidlDestination,
  379.             "请选择新建文件夹/菜单组的位置："))
  380.         return ;
  381.     // 把PIDL转换为路径名
  382.     SHGetPathFromIDList( pidlDestination, szPath) ;
  383.     //取得新建文件夹的名字
  384.     CNameDlg name_dlg;
  385.     if(name_dlg.DoModal() == IDCANCEL)
  386.         return;
  387.     //形成完整的新建文件夹名
  388.     wsprintf(szPath+lstrlen(szPath),"\%s",
  389.                 name_dlg.m_strName);
  390.     //创建文件夹（子目录）
  391.     if( !CreateDirectory( szPath, NULL))
  392.     {
  393.         MessageBox( "创建文件夹失败！") ;
  394.         return ;
  395.     }
  396.     // 通知shell有关变化
  397.     NotifyShell( SHCNE_MKDIR|SHCNE_INTERRUPT,
  398.                 szPath);
  400. }
  402. void CShortCutDlg::OnCreateItem() 
  403. {
  404. UpdateData(TRUE);
  405. if(m_nLocation==0)
  406. {
  407. //设置起始文件夹为桌面
  408. nBeginAt=CSIDL_DESKTOPDIRECTORY ;    
  409. }
  410. else
  411. {
  412. //设置起始文件夹为“开始”菜单
  413. nBeginAt=CSIDL_STARTMENU;
  414. }
  416. char szPath[MAX_PATH]="";
  417.                 //快捷方式的目标应用程序名
  418.     char szLink[MAX_PATH]="";
  419.                 //快捷方式的数据文件名
  420.     // 取得快捷方式的目标应用程序名
  421.     if( !SelectMenuItem( szPath))
  422.         return ;
  423.     // 取得新建快捷方式所在的文件夹
  424.     //并形成其数据文件名
  425.     if( !GetShortcutCrt( szLink))
  426.         return ;
  427.     // 创建快捷方式
  428.     if(!CreateLink( szPath, szLink) )
  429.         return;
  430.     // 通知shell有关变化
  431.     NotifyShell( SHCNE_CREATE|SHCNE_INTERRUPT, 
  432.                 szLink) ;
  434. }
  436. void CShortCutDlg::OnDeleteGroup() 
  437. {
  438. UpdateData(TRUE);
  439. if(m_nLocation==0)
  440. {
  441. //设置起始文件夹为桌面
  442. nBeginAt=CSIDL_DESKTOPDIRECTORY ;    
  443. }
  444. else
  445. {
  446. //设置起始文件夹为“开始”菜单
  447. nBeginAt=CSIDL_STARTMENU;
  449. }
  451. LPITEMIDLIST pidlBeginAt, pidlFolder ;
  452.     char szPath[MAX_PATH]="";
  454.     // 取得开始菜单或桌面的PIDL
  455.     SHGetSpecialFolderLocation( HWND_DESKTOP, 
  456.                 nBeginAt, &pidlBeginAt) ;
  457.     // 取得要删除的文件夹
  458.     if( !BrowseForFolder( pidlBeginAt, &pidlFolder, 
  459.             "请选择要删除的文件夹/菜单组："))
  460.         return ;
  461.     // 把PIDL转化为路径名
  462.     SHGetPathFromIDList( pidlFolder, szPath) ;
  463.     // 删除文件夹
  464.     if(!DeleteFolder( szPath))
  465.         return ;
  466.     // 通知shell有关变化
  467.     NotifyShell( SHCNE_RMDIR|SHCNE_INTERRUPT, 
  468.                     szPath) ;
  470. }
  472. void CShortCutDlg::OnDeleteItem() 
  473. {
  474. UpdateData(TRUE);
  475. if(m_nLocation==0)
  476. {
  477. //设置起始文件夹为桌面
  478. nBeginAt=CSIDL_DESKTOPDIRECTORY ;    
  479. }
  480. else
  481. {
  482. //设置起始文件夹为“开始”菜单
  483. nBeginAt=CSIDL_STARTMENU;
  485. }
  487. LPITEMIDLIST pidlBeginAt ;
  488.     char szShortcut[ MAX_PATH]="",
  489.         szPath[ MAX_PATH]="";
  491.     // 取得开始菜单或桌面的PIDL
  492.     SHGetSpecialFolderLocation( HWND_DESKTOP, 
  493.                 nBeginAt, &pidlBeginAt) ;
  494.     // 把PIDL转化为路径名
  495.     SHGetPathFromIDList( pidlBeginAt, szPath) ;
  496.     // 取得要删除的快捷方式
  497.     if( !GetShortcutDel( szPath, szShortcut))
  498.         return ;
  499.     // 删除快捷方式
  500.     if( !DeleteLink(szShortcut))
  501.         return ;
  502.     // 通知SHELL有关改变
  503.     NotifyShell( SHCNE_DELETE|SHCNE_INTERRUPT, 
  504.                 szShortcut) ;
  506. }
  508. void CShortCutDlg::OnDestroy() 
  509. {
  510. CoUninitialize();
  511.     
  512. CDialog::OnDestroy();
  513. }