src/TortoiseProc/PatchListCtrl.cpp

   1 // PathListCtrl.cpp : implementation file
   2 //
   3
   4 #include "stdafx.h"
   5 #include "TortoiseProc.h"
   6 #include "PatchListCtrl.h"
   7 #include "iconmenu.h"
   8 #include "AppUtils.h"
   9 #include "git.h"
  10 #include "AppUtils.h"
  11 // CPatchListCtrl
  12
  13 IMPLEMENT_DYNAMIC(CPatchListCtrl, CListCtrl)
  14
  15 CPatchListCtrl::CPatchListCtrl()
  16 {
  17         m_ContextMenuMask=0xFFFFFFFF;
  18
  19         HFONT hFont = (HFONT)GetStockObject(DEFAULT_GUI_FONT);
  20         LOGFONT lf = {0};
  21         GetObject(hFont, sizeof(LOGFONT), &lf);
  22         lf.lfWeight = FW_BOLD;
  23         m_boldFont = CreateFontIndirect(&lf);
  24
  25 }
  26
  27 CPatchListCtrl::~CPatchListCtrl()
  28 {
  29         if (m_boldFont)
  30                 DeleteObject(m_boldFont);
  31 }
  32
  33
  34 BEGIN_MESSAGE_MAP(CPatchListCtrl, CListCtrl)
  35         ON_NOTIFY_REFLECT(NM_DBLCLK, &CPatchListCtrl::OnNMDblclk)
  36         ON_WM_CONTEXTMENU()
  37         ON_NOTIFY_REFLECT(NM_CUSTOMDRAW, &CPatchListCtrl::OnNMCustomdraw)
  38 END_MESSAGE_MAP()
  39
  40
  41
  42 // CPatchListCtrl message handlers
  43
  44
  45
  46 void CPatchListCtrl::OnNMDblclk(NMHDR *pNMHDR, LRESULT *pResult)
  47 {
  48         LPNMITEMACTIVATE pNMItemActivate = reinterpret_cast<LPNMITEMACTIVATE>(pNMHDR);
  49
  50         CString path=GetItemText(pNMItemActivate->iItem,0);
  51         CTGitPath gitpath;
  52         gitpath.SetFromWin(path);
  53
  54         CAppUtils::StartUnifiedDiffViewer(path,gitpath.GetFilename());
  55
  56         *pResult = 0;
  57 }
  58
  59 void CPatchListCtrl::OnContextMenu(CWnd* /*pWnd*/, CPoint point)
  60 {
  61         int selected=this->GetSelectedCount();
  62         int index=0;
  63         POSITION pos=this->GetFirstSelectedItemPosition();
  64         index=this->GetNextSelectedItem(pos);
  65
  66         CIconMenu popup;
  67         if (popup.CreatePopupMenu())
  68         {
  69                 if(selected == 1)
  70                 {
  71                         if( m_ContextMenuMask&GetMenuMask(MENU_VIEWPATCH))
  72                                 popup.AppendMenuIcon(MENU_VIEWPATCH, IDS_MENU_VIEWPATCH, 0);
  73
  74                         if( m_ContextMenuMask&GetMenuMask(MENU_VIEWWITHMERGE))
  75                                 popup.AppendMenuIcon(MENU_VIEWWITHMERGE, IDS_MENU_VIEWWITHMERGE, 0);
  76
  77                         popup.SetDefaultItem(MENU_VIEWPATCH, FALSE);
  78                 }
  79                 if(selected >= 1)
  80                 {
  81                         if( m_ContextMenuMask&GetMenuMask(MENU_SENDMAIL))
  82                                 popup.AppendMenuIcon(MENU_SENDMAIL, IDS_MENU_SENDMAIL, IDI_MENUSENDMAIL);
  83
  84                         if( m_ContextMenuMask&GetMenuMask(MENU_APPLY))
  85                                 popup.AppendMenuIcon(MENU_APPLY, IDS_MENU_APPLY, 0);
  86                 }
  87
  88                 int cmd = popup.TrackPopupMenu(TPM_RETURNCMD | TPM_LEFTALIGN | TPM_NONOTIFY, point.x, point.y, this, 0);
  89
  90                 switch (cmd)
  91                 {
  92                 case MENU_VIEWPATCH:
  93                         {
  94
  95                                 CString path=GetItemText(index,0);
  96                                 CTGitPath gitpath;
  97                                 gitpath.SetFromWin(path);
  98
  99                                 CAppUtils::StartUnifiedDiffViewer(path,gitpath.GetFilename());
 100                                 break;
 101                         }
 102                 case MENU_VIEWWITHMERGE:
 103                         {
 104                                 CString path=GetItemText(index,0);
 105                                 CTGitPath gitpath;
 106                                 gitpath.SetFromWin(path);
 107
 108                                 CTGitPath dir;
 109                                 dir.SetFromGit(g_Git.m_CurrentDir);
 110
 111                                 CAppUtils::StartExtPatch(gitpath,dir);
 112                                 break;
 113                         }
 114                 case MENU_SENDMAIL:
 115                         {
 116                                 LaunchProc(_T("sendmail"));
 117                                 break;
 118                         }
 119                 case MENU_APPLY:
 120                         {
 121                                 LaunchProc(_T("importpatch"));
 122
 123                                 break;
 124                         }
 125                 default:
 126                         break;
 127                 }
 128         }
 129 }
 130
 131 int CPatchListCtrl::LaunchProc(const CString& command)
 132 {
 133         CString tempfile=GetTempFile();
 134         POSITION pos=this->GetFirstSelectedItemPosition();
 135         CFile file;
 136         file.Open(tempfile,CFile::modeWrite|CFile::modeCreate);
 137
 138         while(pos)
 139         {
 140                 int index = this->GetNextSelectedItem(pos);
 141                 CString one=this->GetItemText(index,0);
 142                 file.Write(one.GetBuffer(),sizeof(TCHAR)*one.GetLength());
 143                 file.Write(_T("\n"),sizeof(TCHAR)*1);
 144         }
 145
 146         file.Close();
 147
 148         CString cmd = command;
 149         cmd +=_T(" /pathfile:\"");
 150         cmd += tempfile;
 151         cmd += _T("\" /deletepathfile");
 152         CAppUtils::RunTortoiseProc(cmd);
 153         return 0;
 154 }
 155
 156 void CPatchListCtrl::OnNMCustomdraw(NMHDR *pNMHDR, LRESULT *pResult)
 157 {
 158         NMLVCUSTOMDRAW *pNMCD = reinterpret_cast<NMLVCUSTOMDRAW*>(pNMHDR);
 159
 160         *pResult = 0;
 161
 162
 163         switch (pNMCD->nmcd.dwDrawStage)
 164         {
 165         case CDDS_PREPAINT:
 166                 {
 167                         *pResult = CDRF_NOTIFYITEMDRAW;
 168                         return;
 169                 }
 170                 break;
 171         case CDDS_ITEMPREPAINT:
 172                 {
 173                         // This is the prepaint stage for an item. Here's where we set the
 174                         // item's text color.
 175
 176                         // Tell Windows to send draw notifications for each subitem.
 177                         *pResult = CDRF_NOTIFYSUBITEMDRAW;
 178
 179                         DWORD_PTR data = this->GetItemData((int)pNMCD->nmcd.dwItemSpec);
 180                         if(data & (STATUS_APPLY_FAIL | STATUS_APPLY_SUCCESS | STATUS_APPLY_SKIP))
 181                         {
 182                                 pNMCD->clrTextBk = RGB(200,200,200);
 183                         }
 184
 185                         switch(data & STATUS_MASK)
 186                         {
 187                         case STATUS_APPLY_SUCCESS:
 188                                 pNMCD->clrText = RGB(0,128,0);
 189                                 break;
 190                         case STATUS_APPLY_FAIL:
 191                                 pNMCD->clrText = RGB(255,0,0);
 192                                 break;
 193                         case STATUS_APPLY_SKIP:
 194                                 pNMCD->clrText = RGB(128,64,0);
 195                                 break;
 196                         }
 197
 198                         if(data & STATUS_APPLYING)
 199                         {
 200                                 SelectObject(pNMCD->nmcd.hdc, m_boldFont);
 201                                 *pResult = CDRF_NOTIFYSUBITEMDRAW | CDRF_NEWFONT;
 202                         }
 203
 204                 }
 205                 break;
 206         }
 207         *pResult = CDRF_DODEFAULT;
 208 }