src/TortoiseProc/RefLogDlg.cpp

   1 // TortoiseGit - a Windows shell extension for easy version control
   2
   3 // Copyright (C) 2009-2014 - 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 = !g_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         //this->m_logEntries.ParserFromLog();
 266         m_RefList.SetRedraw(false);
 267
 268         ParserFromRefLog(ref, m_RefList.m_RevCache);
 269
 270         m_RefList.SetItemCountEx((int)m_RefList.m_RevCache.size());
 271
 272         this->m_RefList.m_arShownList.RemoveAll();
 273
 274         for (unsigned int i = 0; i < m_RefList.m_RevCache.size(); ++i)
 275         {
 276                 GitRev *rev = &m_RefList.m_RevCache[i];
 277                 rev->m_IsFull = TRUE;
 278                 this->m_RefList.m_arShownList.Add(rev);
 279         }
 280
 281         m_RefList.SetRedraw(true);
 282
 283         m_RefList.Invalidate();
 284
 285         if (ref == _T("refs/stash"))
 286         {
 287                 GetDlgItem(IDC_REFLOG_BUTTONCLEARSTASH)->ShowWindow(SW_SHOW);
 288                 GetDlgItem(IDC_REFLOG_BUTTONCLEARSTASH)->EnableWindow((m_RefList.m_arShownList.GetSize() > 0));
 289         }
 290         else
 291                 GetDlgItem(IDC_REFLOG_BUTTONCLEARSTASH)->ShowWindow(SW_HIDE);
 292 }
 293
 294 BOOL CRefLogDlg::PreTranslateMessage(MSG* pMsg)
 295 {
 296         if (pMsg->message == WM_KEYDOWN && pMsg->wParam == VK_F5)
 297                 Refresh();
 298         else if (pMsg->message == WM_KEYDOWN && (pMsg->wParam == VK_F3 || (pMsg->wParam == 'F' && (GetAsyncKeyState(VK_CONTROL) & 0x8000))))
 299                 OnFind();
 300
 301         return CResizableStandAloneDialog::PreTranslateMessage(pMsg);
 302 }
 303
 304 void CRefLogDlg::Refresh()
 305 {
 306         STRING_VECTOR list;
 307         list.push_back(_T("HEAD"));
 308         if (g_Git.GetRefList(list))
 309                 MessageBox(g_Git.GetGitLastErr(_T("Could not get all refs.")), _T("TortoiseGit"), MB_ICONERROR);
 310
 311         m_ChooseRef.SetList(list);
 312
 313         if (m_CurrentBranch.IsEmpty())
 314         {
 315                 m_CurrentBranch.Format(_T("refs/heads/%s"), g_Git.GetCurrentBranch());
 316                 m_ChooseRef.SetCurSel(0); /* Choose HEAD */
 317         }
 318         else
 319         {
 320                 bool found = false;
 321                 for (int i = 0; i < (int)list.size(); ++i)
 322                 {
 323                         if(list[i] == m_CurrentBranch)
 324                         {
 325                                 m_ChooseRef.SetCurSel(i);
 326                                 found = true;
 327                                 break;
 328                         }
 329                 }
 330                 if (!found)
 331                         m_ChooseRef.SetCurSel(0);
 332         }
 333
 334         m_RefList.m_RevCache.clear();
 335
 336         OnCbnSelchangeRef();
 337 }
 338
 339 void CRefLogDlg::OnFind()
 340 {
 341         m_nSearchLine = 0;
 342         m_pFindDialog = new CFindReplaceDialog();
 343         m_pFindDialog->Create(TRUE, _T(""), NULL, FR_DOWN | FR_HIDEWHOLEWORD | FR_HIDEUPDOWN, this);
 344 }
 345
 346 LRESULT CRefLogDlg::OnFindDialogMessage(WPARAM /*wParam*/, LPARAM /*lParam*/)
 347 {
 348         ASSERT(m_pFindDialog != NULL);
 349
 350         if (m_RefList.m_arShownList.IsEmpty())
 351                 return 0;
 352
 353         // If the FR_DIALOGTERM flag is set,
 354         // invalidate the handle identifying the dialog box.
 355         if (m_pFindDialog->IsTerminating())
 356         {
 357                         m_pFindDialog = NULL;
 358                         return 0;
 359         }
 360
 361         // If the FR_FINDNEXT flag is set,
 362         // call the application-defined search routine
 363         // to search for the requested string.
 364         if (m_pFindDialog->FindNext())
 365         {
 366                 //read data from dialog
 367                 CString findString = m_pFindDialog->GetFindString();
 368
 369                 bool bFound = false;
 370                 bool bCaseSensitive = !!(m_pFindDialog->m_nFlags & FR_MATCHCASE);
 371
 372                 if (!bCaseSensitive)
 373                         findString.MakeLower();
 374
 375                 int i = m_nSearchLine;
 376                 if (i < 0 || i >= m_RefList.m_arShownList.GetCount())
 377                         i = 0;
 378
 379                 do
 380                 {
 381                         GitRev * data = (GitRev*)m_RefList.m_arShownList.SafeGetAt(i);
 382
 383                         CString str;
 384                         str += data->m_Ref;
 385                         str += _T("\n");
 386                         str += data->m_RefAction;
 387                         str += _T("\n");
 388                         str += data->m_CommitHash.ToString();
 389                         str += _T("\n");
 390                         str += data->GetSubject();
 391                         str += _T("\n");
 392                         str += data->GetBody();
 393                         str += _T("\n");
 394
 395                         if (!bCaseSensitive)
 396                                 str.MakeLower();
 397
 398                         if (str.Find(findString) >= 0)
 399                                 bFound = true;
 400
 401                         ++i;
 402                         if(!bFound && i >= m_RefList.m_arShownList.GetCount())
 403                                 i=0;
 404                 } while (i != m_nSearchLine && (!bFound));
 405
 406                 if (bFound)
 407                 {
 408                         m_RefList.SetHotItem(i - 1);
 409                         m_RefList.EnsureVisible(i - 1, FALSE);
 410                         m_nSearchLine = i;
 411                 }
 412                 else
 413                         MessageBox(_T("\"") + findString + _T("\" ") + CString(MAKEINTRESOURCE(IDS_NOTFOUND)), _T("TortoiseGit"), MB_ICONINFORMATION);
 414         }
 415
 416         return 0;
 417 }