src/TortoiseProc/PatchListCtrl.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2009-2013, 2015-2016, 2018-2019, 2021-2023 - TortoiseGit
   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 // PathListCtrl.cpp : implementation file
  20 //
  21
  22 #include "stdafx.h"
  23 #include "TortoiseProc.h"
  24 #include "PatchListCtrl.h"
  25 #include "IconMenu.h"
  26 #include "AppUtils.h"
  27 #include "Git.h"
  28 #include "AppUtils.h"
  29 // CPatchListCtrl
  30
  31 IMPLEMENT_DYNAMIC(CPatchListCtrl, CListCtrl)
  32
  33 CPatchListCtrl::CPatchListCtrl()
  34 {
  35 }
  36
  37 CPatchListCtrl::~CPatchListCtrl()
  38 {
  39 }
  40
  41 void CPatchListCtrl::PreSubclassWindow()
  42 {
  43         __super::PreSubclassWindow();
  44
  45         // use the default font, create a copy of it and
  46         // change the copy to BOLD (leave the rest of the font
  47         // the same)
  48         LOGFONT lf = { 0 };
  49         GetFont()->GetLogFont(&lf);
  50         lf.lfWeight = FW_BOLD;
  51         m_boldFont.CreateFontIndirect(&lf);
  52 }
  53
  54 BEGIN_MESSAGE_MAP(CPatchListCtrl, CListCtrl)
  55         ON_NOTIFY_REFLECT(NM_DBLCLK, &CPatchListCtrl::OnNMDblclk)
  56         ON_WM_CONTEXTMENU()
  57         ON_NOTIFY_REFLECT(NM_CUSTOMDRAW, &CPatchListCtrl::OnNMCustomdraw)
  58         ON_WM_DROPFILES()
  59 END_MESSAGE_MAP()
  60
  61
  62
  63 // CPatchListCtrl message handlers
  64
  65
  66
  67 void CPatchListCtrl::OnNMDblclk(NMHDR *pNMHDR, LRESULT *pResult)
  68 {
  69         LPNMITEMACTIVATE pNMItemActivate = reinterpret_cast<LPNMITEMACTIVATE>(pNMHDR);
  70         *pResult = 0;
  71
  72         CPoint point(pNMItemActivate->ptAction);
  73         UINT uFlags = 0;
  74         HitTest(point, &uFlags);
  75         if (uFlags == LVHT_ONITEMSTATEICON)
  76                 return;
  77
  78         CString path=GetItemText(pNMItemActivate->iItem,0);
  79         CTGitPath gitpath;
  80         gitpath.SetFromWin(path);
  81
  82         CAppUtils::StartUnifiedDiffViewer(path, gitpath.GetFilename(), 0, !!(GetAsyncKeyState(VK_SHIFT) & 0x8000));
  83 }
  84
  85 void CPatchListCtrl::OnContextMenu(CWnd* /*pWnd*/, CPoint point)
  86 {
  87         int selected=this->GetSelectedCount();
  88         POSITION pos=this->GetFirstSelectedItemPosition();
  89         auto index = this->GetNextSelectedItem(pos);
  90
  91         CIconMenu popup;
  92         if (popup.CreatePopupMenu())
  93         {
  94                 if(selected == 1)
  95                 {
  96                         if( m_ContextMenuMask&GetMenuMask(MENU_VIEWPATCH))
  97                                 popup.AppendMenuIcon(MENU_VIEWPATCH, IDS_MENU_VIEWPATCH, 0);
  98
  99                         if( m_ContextMenuMask&GetMenuMask(MENU_VIEWWITHMERGE))
 100                                 popup.AppendMenuIcon(MENU_VIEWWITHMERGE, IDS_MENU_VIEWWITHMERGE, 0);
 101
 102                         popup.SetDefaultItem(MENU_VIEWPATCH, FALSE);
 103                 }
 104                 if(selected >= 1)
 105                 {
 106                         if( m_ContextMenuMask&GetMenuMask(MENU_SENDMAIL))
 107                                 popup.AppendMenuIcon(MENU_SENDMAIL, IDS_MENU_SENDMAIL, IDI_MENUSENDMAIL);
 108
 109                         if( m_ContextMenuMask&GetMenuMask(MENU_APPLY))
 110                                 popup.AppendMenuIcon(MENU_APPLY, IDS_MENU_APPLY, 0);
 111                 }
 112
 113                 int cmd = popup.TrackPopupMenu(TPM_RETURNCMD | TPM_LEFTALIGN | TPM_NONOTIFY, point.x, point.y, this);
 114
 115                 switch (cmd)
 116                 {
 117                 case MENU_VIEWPATCH:
 118                         {
 119                                 CString path=GetItemText(index,0);
 120                                 CTGitPath gitpath;
 121                                 gitpath.SetFromWin(path);
 122
 123                                 CAppUtils::StartUnifiedDiffViewer(path, gitpath.GetFilename(), 0, !!(GetAsyncKeyState(VK_SHIFT) & 0x8000));
 124                                 break;
 125                         }
 126                 case MENU_VIEWWITHMERGE:
 127                         {
 128                                 CString path=GetItemText(index,0);
 129                                 CTGitPath gitpath;
 130                                 gitpath.SetFromWin(path);
 131
 132                                 CTGitPath dir;
 133                                 dir.SetFromGit(g_Git.m_CurrentDir);
 134
 135                                 CAppUtils::StartExtPatch(gitpath,dir);
 136                                 break;
 137                         }
 138                 case MENU_SENDMAIL:
 139                         {
 140                                 LaunchProc(L"sendmail");
 141                                 break;
 142                         }
 143                 case MENU_APPLY:
 144                         {
 145                                 LaunchProc(L"importpatch");
 146
 147                                 break;
 148                         }
 149                 default:
 150                         break;
 151                 }
 152         }
 153 }
 154
 155 int CPatchListCtrl::LaunchProc(const CString& command)
 156 {
 157         CString tempfile=GetTempFile();
 158         if (tempfile.IsEmpty())
 159         {
 160                 MessageBox(L"Could not create temp file.", L"TortoiseGit", MB_OK | MB_ICONERROR);
 161                 return -1;
 162         }
 163
 164         CTGitPathList paths;
 165         POSITION pos = GetFirstSelectedItemPosition();
 166         while(pos)
 167         {
 168                 int index = this->GetNextSelectedItem(pos);
 169                 paths.AddPath(GetItemText(index, 0));
 170         }
 171         if (!paths.WriteToFile(tempfile, false))
 172         {
 173                 MessageBox(L"Could not write to temp file.", L"TortoiseGit", MB_OK | MB_ICONERROR);
 174                 return -1;
 175         }
 176
 177         CString cmd = L"/command:";
 178         cmd += command;
 179         cmd += L" /pathfile:\"";
 180         cmd += tempfile;
 181         cmd += L"\" /deletepathfile";
 182         if (!CAppUtils::RunTortoiseGitProc(cmd))
 183         {
 184                 MessageBox(L"Could not start TortoiseGitProc.exe.", L"TortoiseGit", MB_ICONERROR);
 185                 return -1;
 186         }
 187         return 0;
 188 }
 189
 190 void CPatchListCtrl::OnNMCustomdraw(NMHDR *pNMHDR, LRESULT *pResult)
 191 {
 192         NMLVCUSTOMDRAW *pNMCD = reinterpret_cast<NMLVCUSTOMDRAW*>(pNMHDR);
 193
 194         *pResult = 0;
 195
 196         switch (pNMCD->nmcd.dwDrawStage)
 197         {
 198         case CDDS_PREPAINT:
 199                 {
 200                         *pResult = CDRF_NOTIFYITEMDRAW;
 201                         return;
 202                 }
 203         case CDDS_ITEMPREPAINT:
 204                 {
 205                         // This is the prepaint stage for an item. Here's where we set the
 206                         // item's text color.
 207
 208                         // Tell Windows to send draw notifications for each subitem.
 209                         *pResult = CDRF_NOTIFYSUBITEMDRAW;
 210
 211                         DWORD_PTR data = this->GetItemData(static_cast<int>(pNMCD->nmcd.dwItemSpec));
 212                         if(data & (STATUS_APPLY_FAIL | STATUS_APPLY_SUCCESS | STATUS_APPLY_SKIP))
 213                                 pNMCD->clrTextBk = RGB(200,200,200);
 214
 215                         switch(data & STATUS_MASK)
 216                         {
 217                         case STATUS_APPLY_SUCCESS:
 218                                 pNMCD->clrText = RGB(0,128,0);
 219                                 break;
 220                         case STATUS_APPLY_FAIL:
 221                                 pNMCD->clrText = RGB(255,0,0);
 222                                 break;
 223                         case STATUS_APPLY_SKIP:
 224                                 pNMCD->clrText = RGB(128,64,0);
 225                                 break;
 226                         }
 227
 228                         if(data & STATUS_APPLYING)
 229                         {
 230                                 SelectObject(pNMCD->nmcd.hdc, m_boldFont.GetSafeHandle());
 231                                 *pResult = CDRF_NOTIFYSUBITEMDRAW | CDRF_NEWFONT;
 232                         }
 233
 234                 }
 235                 break;
 236         }
 237         *pResult = CDRF_DODEFAULT;
 238 }
 239
 240 void CPatchListCtrl::OnDropFiles(HDROP hDropInfo)
 241 {
 242         UINT nNumFiles = DragQueryFile(hDropInfo, 0xFFFFFFFF, nullptr, 0);
 243         for (UINT i = 0; i < nNumFiles; ++i)
 244         {
 245                 CString file;
 246                 DragQueryFile(hDropInfo, i, CStrBuf(file, MAX_PATH), MAX_PATH);
 247                 if (PathIsDirectory(file))
 248                         continue;
 249
 250                 // no duplicates
 251                 LVFINDINFO lvInfo;
 252                 lvInfo.flags = LVFI_STRING;
 253                 lvInfo.psz = file;
 254                 if (FindItem(&lvInfo, -1) != -1)
 255                         continue;
 256
 257                 int index = InsertItem(GetItemCount(), file);
 258                 if (index >= 0)
 259                         SetCheck(index, true);
 260         }
 261         DragFinish(hDropInfo);
 262         SetColumnWidth(0, LVSCW_AUTOSIZE);
 263 }