Make sure buffer is large enough for the nul terminator
[TortoiseGit.git] / src / TortoiseProc / RefLogDlg.cpp
blob1c7f77d7f97e76dc9dd63c817b67de0742b32c66
1 // TortoiseGit - a Windows shell extension for easy version control
3 // Copyright (C) 2009-2015 - TortoiseGit
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.
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.
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.
19 // RefLogDlg.cpp : implementation file
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"
30 // CRefLogDlg dialog
32 IMPLEMENT_DYNAMIC(CRefLogDlg, CResizableStandAloneDialog)
34 UINT CRefLogDlg::m_FindDialogMessage = ::RegisterWindowMessage(FINDMSGSTRING);
36 CRefLogDlg::CRefLogDlg(CWnd* pParent /*=NULL*/)
37 : CResizableStandAloneDialog(CRefLogDlg::IDD, pParent)
38 , m_pFindDialog(NULL)
39 , m_nSearchLine(0)
44 CRefLogDlg::~CRefLogDlg()
48 void CRefLogDlg::DoDataExchange(CDataExchange* pDX)
50 CDialog::DoDataExchange(pDX);
51 DDX_Control(pDX, IDC_COMBOBOXEX_REF, m_ChooseRef);
52 DDX_Control(pDX, IDC_REFLOG_LIST, m_RefList);
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()
64 LRESULT CRefLogDlg::OnRefLogChanged(WPARAM /*wParam*/, LPARAM /*lParam*/)
66 m_RefList.m_RevCache.clear();
67 OnCbnSelchangeRef();
68 return 0;
71 BOOL CRefLogDlg::OnInitDialog()
73 CResizableStandAloneDialog::OnInitDialog();
74 CAppUtils::MarkWindowAsUnpinnable(m_hWnd);
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);
83 AddOthersToAnchor();
84 this->EnableSaveRestore(_T("RefLogDlg"));
86 CString sWindowTitle;
87 GetWindowText(sWindowTitle);
88 CAppUtils::SetWindowTitle(m_hWnd, g_Git.m_CurrentDir, sWindowTitle);
90 m_ChooseRef.SetMaxHistoryItems(0x7FFFFFFF);
92 m_RefList.m_hasWC = !GitAdminDir::IsBareRepo(g_Git.m_CurrentDir);
94 this->m_RefList.InsertRefLogColumn();
96 Refresh();
98 return TRUE;
100 // CRefLogDlg message handlers
102 void CRefLogDlg::OnBnClickedOk()
104 if (m_RefList.GetSelectedCount() == 1)
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())
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;
118 OnOK();
120 void CRefLogDlg::OnBnClickedClearStash()
122 if (CMessageBox::Show(this->GetSafeHwnd(), IDS_PROC_DELETEALLSTASH, IDS_APPNAME, 2, IDI_QUESTION, IDS_DELETEBUTTON, IDS_ABORTBUTTON) == 1)
124 CString cmdOut;
125 if (g_Git.Run(_T("git.exe stash clear"), &cmdOut, CP_UTF8))
127 MessageBox(cmdOut, _T("TortoiseGit"), MB_ICONERROR);
128 return;
131 m_RefList.m_RevCache.clear();
133 OnCbnSelchangeRef();
137 void CRefLogDlg::OnCbnSelchangeRef()
139 CString ref=m_ChooseRef.GetString();
140 m_RefList.ClearText();
142 m_RefList.SetRedraw(false);
144 CString err;
145 if (GitRevLoglist::GetRefLog(ref, m_RefList.m_RevCache, err))
146 MessageBox(_T("Error while loading reflog.\n") + err, _T("TortoiseGit"), MB_ICONERROR);
148 m_RefList.SetItemCountEx((int)m_RefList.m_RevCache.size());
150 this->m_RefList.m_arShownList.RemoveAll();
152 for (unsigned int i = 0; i < m_RefList.m_RevCache.size(); ++i)
154 GitRevLoglist* rev = &m_RefList.m_RevCache[i];
155 rev->m_IsFull = TRUE;
156 this->m_RefList.m_arShownList.Add(rev);
159 m_RefList.SetRedraw(true);
161 m_RefList.Invalidate();
163 if (ref == _T("refs/stash"))
165 GetDlgItem(IDC_REFLOG_BUTTONCLEARSTASH)->ShowWindow(SW_SHOW);
166 GetDlgItem(IDC_REFLOG_BUTTONCLEARSTASH)->EnableWindow((m_RefList.m_arShownList.GetSize() > 0));
168 else
169 GetDlgItem(IDC_REFLOG_BUTTONCLEARSTASH)->ShowWindow(SW_HIDE);
172 BOOL CRefLogDlg::PreTranslateMessage(MSG* pMsg)
174 if (pMsg->message == WM_KEYDOWN && pMsg->wParam == VK_F5)
175 Refresh();
176 else if (pMsg->message == WM_KEYDOWN && (pMsg->wParam == VK_F3 || (pMsg->wParam == 'F' && (GetAsyncKeyState(VK_CONTROL) & 0x8000))))
177 OnFind();
179 return CResizableStandAloneDialog::PreTranslateMessage(pMsg);
182 void CRefLogDlg::Refresh()
184 STRING_VECTOR list;
185 list.push_back(_T("HEAD"));
186 if (g_Git.GetRefList(list))
187 MessageBox(g_Git.GetGitLastErr(_T("Could not get all refs.")), _T("TortoiseGit"), MB_ICONERROR);
189 m_ChooseRef.SetList(list);
191 if (m_CurrentBranch.IsEmpty())
193 m_CurrentBranch.Format(_T("refs/heads/%s"), g_Git.GetCurrentBranch());
194 m_ChooseRef.SetCurSel(0); /* Choose HEAD */
196 else
198 bool found = false;
199 for (int i = 0; i < (int)list.size(); ++i)
201 if(list[i] == m_CurrentBranch)
203 m_ChooseRef.SetCurSel(i);
204 found = true;
205 break;
208 if (!found)
209 m_ChooseRef.SetCurSel(0);
212 m_RefList.m_RevCache.clear();
214 OnCbnSelchangeRef();
217 void CRefLogDlg::OnFind()
219 m_nSearchLine = 0;
220 m_pFindDialog = new CFindReplaceDialog();
221 m_pFindDialog->Create(TRUE, _T(""), NULL, FR_DOWN | FR_HIDEWHOLEWORD | FR_HIDEUPDOWN, this);
224 LRESULT CRefLogDlg::OnFindDialogMessage(WPARAM /*wParam*/, LPARAM /*lParam*/)
226 ASSERT(m_pFindDialog != NULL);
228 if (m_RefList.m_arShownList.IsEmpty())
229 return 0;
231 // If the FR_DIALOGTERM flag is set,
232 // invalidate the handle identifying the dialog box.
233 if (m_pFindDialog->IsTerminating())
235 m_pFindDialog = NULL;
236 return 0;
239 // If the FR_FINDNEXT flag is set,
240 // call the application-defined search routine
241 // to search for the requested string.
242 if (m_pFindDialog->FindNext())
244 //read data from dialog
245 CString findString = m_pFindDialog->GetFindString();
247 bool bFound = false;
248 bool bCaseSensitive = !!(m_pFindDialog->m_nFlags & FR_MATCHCASE);
250 if (!bCaseSensitive)
251 findString.MakeLower();
253 int i = m_nSearchLine;
254 if (i < 0 || i >= m_RefList.m_arShownList.GetCount())
255 i = 0;
259 GitRevLoglist* data = (GitRevLoglist*)m_RefList.m_arShownList.SafeGetAt(i);
261 CString str;
262 str += data->m_Ref;
263 str += _T("\n");
264 str += data->m_RefAction;
265 str += _T("\n");
266 str += data->m_CommitHash.ToString();
267 str += _T("\n");
268 str += data->GetSubject();
269 str += _T("\n");
270 str += data->GetBody();
271 str += _T("\n");
273 if (!bCaseSensitive)
274 str.MakeLower();
276 if (str.Find(findString) >= 0)
277 bFound = true;
279 ++i;
280 if(!bFound && i >= m_RefList.m_arShownList.GetCount())
281 i=0;
282 } while (i != m_nSearchLine && (!bFound));
284 if (bFound)
286 m_RefList.SetHotItem(i - 1);
287 m_RefList.EnsureVisible(i - 1, FALSE);
288 m_nSearchLine = i;
290 else
291 MessageBox(_T("\"") + findString + _T("\" ") + CString(MAKEINTRESOURCE(IDS_NOTFOUND)), _T("TortoiseGit"), MB_ICONINFORMATION);
294 return 0;