src/TortoiseProc/RefLogDlg.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2009-2015 - 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 // RefLogDlg.cpp : implementation file
  20 //
  21
  22 #include "stdafx.h"
  23 #include "resource.h"
  24 #include "RefLogDlg.h"
  25 #include "Git.h"
  26 #include "AppUtils.h"
  27 #include "MessageBox.h"
  28 #include "UnicodeUtils.h"
  29
  30 // CRefLogDlg dialog
  31
  32 IMPLEMENT_DYNAMIC(CRefLogDlg, CResizableStandAloneDialog)
  33
  34 UINT CRefLogDlg::m_FindDialogMessage = ::RegisterWindowMessage(FINDMSGSTRING);
  35
  36 CRefLogDlg::CRefLogDlg(CWnd* pParent /*=NULL*/)
  37         : CResizableStandAloneDialog(CRefLogDlg::IDD, pParent)
  38         , m_pFindDialog(NULL)
  39         , m_nSearchLine(0)
  40 {
  41
  42 }
  43
  44 CRefLogDlg::~CRefLogDlg()
  45 {
  46 }
  47
  48 void CRefLogDlg::DoDataExchange(CDataExchange* pDX)
  49 {
  50         CDialog::DoDataExchange(pDX);
  51         DDX_Control(pDX, IDC_COMBOBOXEX_REF, m_ChooseRef);
  52         DDX_Control(pDX, IDC_REFLOG_LIST, m_RefList);
  53 }
  54
  55
  56 BEGIN_MESSAGE_MAP(CRefLogDlg, CResizableStandAloneDialog)
  57         ON_BN_CLICKED(IDOK, &CRefLogDlg::OnBnClickedOk)
  58         ON_BN_CLICKED(IDC_REFLOG_BUTTONCLEARSTASH, &CRefLogDlg::OnBnClickedClearStash)
  59         ON_CBN_SELCHANGE(IDC_COMBOBOXEX_REF,   &CRefLogDlg::OnCbnSelchangeRef)
  60         ON_MESSAGE(MSG_REFLOG_CHANGED,OnRefLogChanged)
  61         ON_REGISTERED_MESSAGE(m_FindDialogMessage, OnFindDialogMessage)
  62 END_MESSAGE_MAP()
  63
  64 LRESULT CRefLogDlg::OnRefLogChanged(WPARAM /*wParam*/, LPARAM /*lParam*/)
  65 {
  66         m_RefList.m_RevCache.clear();
  67         OnCbnSelchangeRef();
  68         return 0;
  69 }
  70
  71 BOOL CRefLogDlg::OnInitDialog()
  72 {
  73         CResizableStandAloneDialog::OnInitDialog();
  74         CAppUtils::MarkWindowAsUnpinnable(m_hWnd);
  75
  76         AddAnchor(IDOK,BOTTOM_RIGHT);
  77         AddAnchor(IDCANCEL,BOTTOM_RIGHT);
  78         AddAnchor(IDC_REFLOG_BUTTONCLEARSTASH, BOTTOM_LEFT);
  79         AddAnchor(IDC_REFLOG_LIST,TOP_LEFT,BOTTOM_RIGHT);
  80         AddAnchor(IDHELP, BOTTOM_RIGHT);
  81         AddAnchor(IDC_COMBOBOXEX_REF, TOP_LEFT, TOP_RIGHT);
  82
  83         AddOthersToAnchor();
  84         this->EnableSaveRestore(_T("RefLogDlg"));
  85
  86         CString sWindowTitle;
  87         GetWindowText(sWindowTitle);
  88         CAppUtils::SetWindowTitle(m_hWnd, g_Git.m_CurrentDir, sWindowTitle);
  89
  90         m_ChooseRef.SetMaxHistoryItems(0x7FFFFFFF);
  91
  92         m_RefList.m_hasWC = !GitAdminDir::IsBareRepo(g_Git.m_CurrentDir);
  93
  94         this->m_RefList.InsertRefLogColumn();
  95
  96         Refresh();
  97
  98         return TRUE;
  99 }
 100 // CRefLogDlg message handlers
 101
 102 void CRefLogDlg::OnBnClickedOk()
 103 {
 104         if (m_RefList.GetSelectedCount() == 1)
 105         {
 106                 // get the selected row
 107                 POSITION pos = m_RefList.GetFirstSelectedItemPosition();
 108                 int selIndex = m_RefList.GetNextSelectedItem(pos);
 109                 if (selIndex < m_RefList.m_arShownList.GetCount())
 110                 {
 111                         // all ok, pick up the revision
 112                         GitRev* pLogEntry = reinterpret_cast<GitRev *>(m_RefList.m_arShownList.GetAt(selIndex));
 113                         // extract the hash
 114                         m_SelectedHash = pLogEntry->m_CommitHash;
 115                 }
 116         }
 117
 118         OnOK();
 119 }
 120 void CRefLogDlg::OnBnClickedClearStash()
 121 {
 122         if (CMessageBox::Show(this->GetSafeHwnd(), IDS_PROC_DELETEALLSTASH, IDS_APPNAME, 2, IDI_QUESTION, IDS_DELETEBUTTON, IDS_ABORTBUTTON) == 1)
 123         {
 124                 CString cmdOut;
 125                 if (g_Git.Run(_T("git.exe stash clear"), &cmdOut, CP_UTF8))
 126                 {
 127                         MessageBox(cmdOut, _T("TortoiseGit"), MB_ICONERROR);
 128                         return;
 129                 }
 130
 131                 m_RefList.m_RevCache.clear();
 132
 133                 OnCbnSelchangeRef();
 134         }
 135 }
 136
 137 int AddToRefLoglist(unsigned char * /*osha1*/, unsigned char *nsha1, const char * /*name*/, unsigned long time, int /*sz*/, const char *msg, void *data)
 138 {
 139         std::vector<GitRev> *vector = (std::vector<GitRev> *)data;
 140         GitRev rev;
 141         rev.m_CommitHash = (char *)nsha1;
 142         rev.GetCommitterDate() = CTime(time);
 143
 144         CString one;
 145         g_Git.StringAppend(&one, (BYTE *)msg);
 146
 147         int message = one.Find(_T(":"), 0);
 148         if (message > 0)
 149         {
 150                 rev.m_RefAction = one.Left(message);
 151                 rev.GetSubject() = one.Mid(message + 1);
 152         }
 153
 154         vector->insert(vector->begin(), rev);
 155
 156         return 0;
 157 }
 158
 159 int ParserFromRefLog(CString ref, std::vector<GitRev> &refloglist)
 160 {
 161         refloglist.clear();
 162         if (g_Git.m_IsUseLibGit2)
 163         {
 164                 CAutoRepository repo(g_Git.GetGitRepository());
 165                 if (!repo)
 166                 {
 167                         MessageBox(nullptr, CGit::GetLibGit2LastErr(_T("Could not open repository.")), _T("TortoiseGit"), MB_ICONERROR);
 168                         return -1;
 169                 }
 170
 171                 CAutoReflog reflog;
 172                 if (git_reflog_read(reflog.GetPointer(), repo, CUnicodeUtils::GetUTF8(ref)) < 0)
 173                 {
 174                         MessageBox(nullptr, CGit::GetLibGit2LastErr(_T("Could not read reflog.")), _T("TortoiseGit"), MB_ICONERROR);
 175                         return -1;
 176                 }
 177
 178                 for (size_t i = 0; i < git_reflog_entrycount(reflog); ++i)
 179                 {
 180                         const git_reflog_entry *entry = git_reflog_entry_byindex(reflog, i);
 181                         if (!entry)
 182                                 continue;
 183
 184                         GitRev rev;
 185                         rev.m_CommitHash = (char *)git_reflog_entry_id_new(entry)->id;
 186                         rev.m_Ref.Format(_T("%s@{%d}"), ref, i);
 187                         rev.GetCommitterDate() = CTime(git_reflog_entry_committer(entry)->when.time);
 188                         if (git_reflog_entry_message(entry) != nullptr)
 189                         {
 190                                 CString one;
 191                                 g_Git.StringAppend(&one, (BYTE *)git_reflog_entry_message(entry));
 192                                 int message = one.Find(_T(":"), 0);
 193                                 if (message > 0)
 194                                 {
 195                                         rev.m_RefAction = one.Left(message);
 196                                         rev.GetSubject() = one.Mid(message + 1);
 197                                 }
 198                         }
 199                         refloglist.push_back(rev);
 200                 }
 201         }
 202         else if (g_Git.m_IsUseGitDLL)
 203         {
 204                 git_for_each_reflog_ent(CUnicodeUtils::GetUTF8(ref), AddToRefLoglist, &refloglist);
 205                 for (size_t i = 0; i < refloglist.size(); ++i)
 206                         refloglist[i].m_Ref.Format(_T("%s@{%d}"), ref, i);
 207         }
 208         else
 209         {
 210                 CString cmd, out;
 211                 GitRev rev;
 212                 cmd.Format(_T("git.exe reflog show --pretty=\"%%H %%gD: %%gs\" --date=raw %s"), ref);
 213                 if (g_Git.Run(cmd, &out, NULL, CP_UTF8))
 214                         return -1;
 215
 216                 int i = 0;
 217                 CString prefix = ref + _T("@{");
 218                 int pos = 0;
 219                 while (pos >= 0)
 220                 {
 221                         CString one = out.Tokenize(_T("\n"), pos);
 222                         int refPos = one.Find(_T(' '), 0);
 223                         if (refPos < 0)
 224                                 continue;
 225
 226                         rev.Clear();
 227
 228                         CString hashStr = one.Left(refPos);
 229                         rev.m_CommitHash = hashStr;
 230                         rev.m_Ref.Format(_T("%s@{%d}"), ref, i++);
 231                         int prefixPos = one.Find(prefix, refPos + 1);
 232                         if (prefixPos != refPos + 1)
 233                                 continue;
 234
 235                         int spacePos = one.Find(_T(' '), prefixPos + prefix.GetLength() + 1);
 236                         if (spacePos < 0)
 237                                 continue;
 238
 239                         CString timeStr = one.Mid(prefixPos + prefix.GetLength(), spacePos - prefixPos - prefix.GetLength());
 240                         rev.GetCommitterDate() = CTime(_ttoi(timeStr));
 241                         int action = one.Find(_T("}: "), spacePos + 1);
 242                         if (action > 0)
 243                         {
 244                                 action += 2;
 245                                 int message = one.Find(_T(":"), action);
 246                                 if (message > 0)
 247                                 {
 248                                         rev.m_RefAction = one.Mid(action + 1, message - action - 1);
 249                                         rev.GetSubject() = one.Right(one.GetLength() - message - 1);
 250                                 }
 251                         }
 252
 253                         refloglist.push_back(rev);
 254                 }
 255         }
 256         return 0;
 257 }
 258
 259
 260 void CRefLogDlg::OnCbnSelchangeRef()
 261 {
 262         CString ref=m_ChooseRef.GetString();
 263         m_RefList.ClearText();
 264
 265         m_RefList.SetRedraw(false);
 266
 267         ParserFromRefLog(ref, m_RefList.m_RevCache);
 268
 269         m_RefList.SetItemCountEx((int)m_RefList.m_RevCache.size());
 270
 271         this->m_RefList.m_arShownList.RemoveAll();
 272
 273         for (unsigned int i = 0; i < m_RefList.m_RevCache.size(); ++i)
 274         {
 275                 GitRev *rev = &m_RefList.m_RevCache[i];
 276                 rev->m_IsFull = TRUE;
 277                 this->m_RefList.m_arShownList.Add(rev);
 278         }
 279
 280         m_RefList.SetRedraw(true);
 281
 282         m_RefList.Invalidate();
 283
 284         if (ref == _T("refs/stash"))
 285         {
 286                 GetDlgItem(IDC_REFLOG_BUTTONCLEARSTASH)->ShowWindow(SW_SHOW);
 287                 GetDlgItem(IDC_REFLOG_BUTTONCLEARSTASH)->EnableWindow((m_RefList.m_arShownList.GetSize() > 0));
 288         }
 289         else
 290                 GetDlgItem(IDC_REFLOG_BUTTONCLEARSTASH)->ShowWindow(SW_HIDE);
 291 }
 292
 293 BOOL CRefLogDlg::PreTranslateMessage(MSG* pMsg)
 294 {
 295         if (pMsg->message == WM_KEYDOWN && pMsg->wParam == VK_F5)
 296                 Refresh();
 297         else if (pMsg->message == WM_KEYDOWN && (pMsg->wParam == VK_F3 || (pMsg->wParam == 'F' && (GetAsyncKeyState(VK_CONTROL) & 0x8000))))
 298                 OnFind();
 299
 300         return CResizableStandAloneDialog::PreTranslateMessage(pMsg);
 301 }
 302
 303 void CRefLogDlg::Refresh()
 304 {
 305         STRING_VECTOR list;
 306         list.push_back(_T("HEAD"));
 307         if (g_Git.GetRefList(list))
 308                 MessageBox(g_Git.GetGitLastErr(_T("Could not get all refs.")), _T("TortoiseGit"), MB_ICONERROR);
 309
 310         m_ChooseRef.SetList(list);
 311
 312         if (m_CurrentBranch.IsEmpty())
 313         {
 314                 m_CurrentBranch.Format(_T("refs/heads/%s"), g_Git.GetCurrentBranch());
 315                 m_ChooseRef.SetCurSel(0); /* Choose HEAD */
 316         }
 317         else
 318         {
 319                 bool found = false;
 320                 for (int i = 0; i < (int)list.size(); ++i)
 321                 {
 322                         if(list[i] == m_CurrentBranch)
 323                         {
 324                                 m_ChooseRef.SetCurSel(i);
 325                                 found = true;
 326                                 break;
 327                         }
 328                 }
 329                 if (!found)
 330                         m_ChooseRef.SetCurSel(0);
 331         }
 332
 333         m_RefList.m_RevCache.clear();
 334
 335         OnCbnSelchangeRef();
 336 }
 337
 338 void CRefLogDlg::OnFind()
 339 {
 340         m_nSearchLine = 0;
 341         m_pFindDialog = new CFindReplaceDialog();
 342         m_pFindDialog->Create(TRUE, _T(""), NULL, FR_DOWN | FR_HIDEWHOLEWORD | FR_HIDEUPDOWN, this);
 343 }
 344
 345 LRESULT CRefLogDlg::OnFindDialogMessage(WPARAM /*wParam*/, LPARAM /*lParam*/)
 346 {
 347         ASSERT(m_pFindDialog != NULL);
 348
 349         if (m_RefList.m_arShownList.IsEmpty())
 350                 return 0;
 351
 352         // If the FR_DIALOGTERM flag is set,
 353         // invalidate the handle identifying the dialog box.
 354         if (m_pFindDialog->IsTerminating())
 355         {
 356                         m_pFindDialog = NULL;
 357                         return 0;
 358         }
 359
 360         // If the FR_FINDNEXT flag is set,
 361         // call the application-defined search routine
 362         // to search for the requested string.
 363         if (m_pFindDialog->FindNext())
 364         {
 365                 //read data from dialog
 366                 CString findString = m_pFindDialog->GetFindString();
 367
 368                 bool bFound = false;
 369                 bool bCaseSensitive = !!(m_pFindDialog->m_nFlags & FR_MATCHCASE);
 370
 371                 if (!bCaseSensitive)
 372                         findString.MakeLower();
 373
 374                 int i = m_nSearchLine;
 375                 if (i < 0 || i >= m_RefList.m_arShownList.GetCount())
 376                         i = 0;
 377
 378                 do
 379                 {
 380                         GitRev * data = (GitRev*)m_RefList.m_arShownList.SafeGetAt(i);
 381
 382                         CString str;
 383                         str += data->m_Ref;
 384                         str += _T("\n");
 385                         str += data->m_RefAction;
 386                         str += _T("\n");
 387                         str += data->m_CommitHash.ToString();
 388                         str += _T("\n");
 389                         str += data->GetSubject();
 390                         str += _T("\n");
 391                         str += data->GetBody();
 392                         str += _T("\n");
 393
 394                         if (!bCaseSensitive)
 395                                 str.MakeLower();
 396
 397                         if (str.Find(findString) >= 0)
 398                                 bFound = true;
 399
 400                         ++i;
 401                         if(!bFound && i >= m_RefList.m_arShownList.GetCount())
 402                                 i=0;
 403                 } while (i != m_nSearchLine && (!bFound));
 404
 405                 if (bFound)
 406                 {
 407                         m_RefList.SetHotItem(i - 1);
 408                         m_RefList.EnsureVisible(i - 1, FALSE);
 409                         m_nSearchLine = i;
 410                 }
 411                 else
 412                         MessageBox(_T("\"") + findString + _T("\" ") + CString(MAKEINTRESOURCE(IDS_NOTFOUND)), _T("TortoiseGit"), MB_ICONINFORMATION);
 413         }
 414
 415         return 0;
 416 }