src/Utils/MiscUI/FileDropEdit.h

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2003-2008 - TortoiseSVN
   4
   5 // This program is free software; you can redistribute it and/or
   6 // modify it under the terms of the GNU General Public License
   7 // as published by the Free Software Foundation; either version 2
   8 // of the License, or (at your option) any later version.
   9
  10 // This program is distributed in the hope that it will be useful,
  11 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13 // GNU General Public License for more details.
  14
  15 // You should have received a copy of the GNU General Public License
  16 // along with this program; if not, write to the Free Software Foundation,
  17 // 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
  18 //
  19 #pragma once
  20
  21 #include "../DragDropImpl.h"
  22 #include "../UnicodeUtils.h"
  23
  24 /**
  25  * \ingroup Utils
  26  * helper class to turn a control into a file drop target
  27  */
  28 class CFileDropTarget : public CIDropTarget
  29 {
  30 public:
  31         CFileDropTarget(HWND hTargetWnd):CIDropTarget(hTargetWnd){}
  32         virtual bool OnDrop(FORMATETC* pFmtEtc, STGMEDIUM& medium, DWORD * /*pdwEffect*/, POINTL /*pt*/)
  33         {
  34                 if(pFmtEtc->cfFormat == CF_TEXT && medium.tymed == TYMED_ISTREAM)
  35                 {
  36                         if(medium.pstm != NULL)
  37                         {
  38                                 const int BUF_SIZE = 10000;
  39                                 char buff[BUF_SIZE+1];
  40                                 ULONG cbRead=0;
  41                                 HRESULT hr = medium.pstm->Read(buff, BUF_SIZE, &cbRead);
  42                                 if( SUCCEEDED(hr) && cbRead > 0 && cbRead < BUF_SIZE)
  43                                 {
  44                                         buff[cbRead]=0;
  45                                         LRESULT nLen = ::SendMessage(m_hTargetWnd, WM_GETTEXTLENGTH, 0, 0);
  46                                         ::SendMessage(m_hTargetWnd, EM_SETSEL, nLen, -1);
  47                                         std::wstring str = CUnicodeUtils::StdGetUnicode(std::string(buff));
  48                                         ::SendMessage(m_hTargetWnd, EM_REPLACESEL, TRUE, (LPARAM)str.c_str());
  49                                 }
  50                                 else
  51                                         for(;(hr==S_OK && cbRead >0) && SUCCEEDED(hr) ;)
  52                                         {
  53                                                 buff[cbRead]=0;
  54                                                 LRESULT nLen = ::SendMessage(m_hTargetWnd, WM_GETTEXTLENGTH, 0, 0);
  55                                                 ::SendMessage(m_hTargetWnd, EM_SETSEL, nLen, -1);
  56                                                 std::wstring str = CUnicodeUtils::StdGetUnicode(std::string(buff));
  57                                                 ::SendMessage(m_hTargetWnd, EM_REPLACESEL, TRUE, (LPARAM)str.c_str());
  58                                                 cbRead=0;
  59                                                 hr = medium.pstm->Read(buff, BUF_SIZE, &cbRead);
  60                                         }
  61                         }
  62                 }
  63                 if(pFmtEtc->cfFormat == CF_UNICODETEXT && medium.tymed == TYMED_ISTREAM)
  64                 {
  65                         if(medium.pstm != NULL)
  66                         {
  67                                 const int BUF_SIZE = 10000;
  68                                 TCHAR buff[BUF_SIZE+1];
  69                                 ULONG cbRead=0;
  70                                 HRESULT hr = medium.pstm->Read(buff, BUF_SIZE, &cbRead);
  71                                 if( SUCCEEDED(hr) && cbRead > 0 && cbRead < BUF_SIZE)
  72                                 {
  73                                         buff[cbRead]=0;
  74                                         LRESULT nLen = ::SendMessage(m_hTargetWnd, WM_GETTEXTLENGTH, 0, 0);
  75                                         ::SendMessage(m_hTargetWnd, EM_SETSEL, nLen, -1);
  76                                         ::SendMessage(m_hTargetWnd, EM_REPLACESEL, TRUE, (LPARAM)buff);
  77                                 }
  78                                 else
  79                                         for(;(hr==S_OK && cbRead >0) && SUCCEEDED(hr) ;)
  80                                         {
  81                                                 buff[cbRead]=0;
  82                                                 LRESULT nLen = ::SendMessage(m_hTargetWnd, WM_GETTEXTLENGTH, 0, 0);
  83                                                 ::SendMessage(m_hTargetWnd, EM_SETSEL, nLen, -1);
  84                                                 ::SendMessage(m_hTargetWnd, EM_REPLACESEL, TRUE, (LPARAM)buff);
  85                                                 cbRead=0;
  86                                                 hr = medium.pstm->Read(buff, BUF_SIZE, &cbRead);
  87                                         }
  88                         }
  89                 }
  90                 if(pFmtEtc->cfFormat == CF_TEXT && medium.tymed == TYMED_HGLOBAL)
  91                 {
  92                         char* pStr = (char*)GlobalLock(medium.hGlobal);
  93                         if(pStr != NULL)
  94                         {
  95                                 LRESULT nLen = ::SendMessage(m_hTargetWnd, WM_GETTEXTLENGTH, 0, 0);
  96                                 ::SendMessage(m_hTargetWnd, EM_SETSEL, nLen, -1);
  97                                 std::wstring str = CUnicodeUtils::StdGetUnicode(std::string(pStr));
  98                                 ::SendMessage(m_hTargetWnd, EM_REPLACESEL, TRUE, (LPARAM)str.c_str());
  99                         }
 100                         GlobalUnlock(medium.hGlobal);
 101                 }
 102                 if(pFmtEtc->cfFormat == CF_UNICODETEXT && medium.tymed == TYMED_HGLOBAL)
 103                 {
 104                         WCHAR* pStr = (WCHAR*)GlobalLock(medium.hGlobal);
 105                         if(pStr != NULL)
 106                         {
 107                                 LRESULT nLen = ::SendMessage(m_hTargetWnd, WM_GETTEXTLENGTH, 0, 0);
 108                                 ::SendMessage(m_hTargetWnd, EM_SETSEL, nLen, -1);
 109                                 ::SendMessage(m_hTargetWnd, EM_REPLACESEL, TRUE, (LPARAM)pStr);
 110                         }
 111                         GlobalUnlock(medium.hGlobal);
 112                 }
 113                 if(pFmtEtc->cfFormat == CF_HDROP && medium.tymed == TYMED_HGLOBAL)
 114                 {
 115                         HDROP hDrop = (HDROP)GlobalLock(medium.hGlobal);
 116                         if(hDrop != NULL)
 117                         {
 118                                 TCHAR szFileName[MAX_PATH];
 119
 120                                 UINT cFiles = DragQueryFile(hDrop, 0xFFFFFFFF, NULL, 0);
 121                                 for(UINT i = 0; i < cFiles; ++i)
 122                                 {
 123                                         DragQueryFile(hDrop, i, szFileName, sizeof(szFileName));
 124                                         ::SendMessage(m_hTargetWnd, WM_SETTEXT, 0, (LPARAM)szFileName);
 125                                 }
 126                                 //DragFinish(hDrop); // base class calls ReleaseStgMedium
 127                         }
 128                         GlobalUnlock(medium.hGlobal);
 129                 }
 130                 return true; //let base free the medium
 131         }
 132
 133 };
 134
 135
 136 /**
 137  * \ingroup Utils
 138  * Enhancement for a CEdit control which allows the edit control to have files
 139  * dropped onto it and fill in the path of that dropped file.
 140  */
 141 class CFileDropEdit : public CEdit
 142 {
 143         DECLARE_DYNAMIC(CFileDropEdit)
 144
 145 public:
 146         CFileDropEdit();
 147         virtual ~CFileDropEdit();
 148
 149 protected:
 150         DECLARE_MESSAGE_MAP()
 151
 152         CFileDropTarget * m_pDropTarget;
 153         virtual void PreSubclassWindow();
 154 };
 155
 156