1 // ProgressDlg.cpp : implementation file
\r
5 #include "TortoiseProc.h"
\r
6 #include "ProgressDlg.h"
\r
9 // CProgressDlg dialog
\r
11 IMPLEMENT_DYNAMIC(CProgressDlg, CResizableStandAloneDialog)
\r
13 CProgressDlg::CProgressDlg(CWnd* pParent /*=NULL*/)
\r
14 : CResizableStandAloneDialog(CProgressDlg::IDD, pParent), m_bShowCommand(true), m_bAutoCloseOnSuccess(false), m_bAbort(false), m_bDone(false)
\r
17 m_bAltAbortPress=false;
\r
21 CProgressDlg::~CProgressDlg()
\r
23 if(m_pThread != NULL)
\r
29 void CProgressDlg::DoDataExchange(CDataExchange* pDX)
\r
31 CDialog::DoDataExchange(pDX);
\r
32 DDX_Control(pDX, IDC_CURRENT, this->m_CurrentWork);
\r
33 DDX_Control(pDX, IDC_TITLE_ANIMATE, this->m_Animate);
\r
34 DDX_Control(pDX, IDC_RUN_PROGRESS, this->m_Progress);
\r
35 DDX_Control(pDX, IDC_LOG, this->m_Log);
\r
39 BEGIN_MESSAGE_MAP(CProgressDlg, CResizableStandAloneDialog)
\r
40 ON_MESSAGE(MSG_PROGRESSDLG_UPDATE_UI, OnProgressUpdateUI)
\r
41 ON_BN_CLICKED(IDOK, &CProgressDlg::OnBnClickedOk)
\r
42 ON_BN_CLICKED(IDC_PROGRESS_BUTTON1,&CProgressDlg::OnBnClickedButton1)
\r
45 BOOL CProgressDlg::OnInitDialog()
\r
47 CResizableStandAloneDialog::OnInitDialog();
\r
49 AddAnchor(IDC_TITLE_ANIMATE, TOP_LEFT, TOP_RIGHT);
\r
50 AddAnchor(IDC_RUN_PROGRESS, TOP_LEFT,TOP_RIGHT);
\r
51 AddAnchor(IDC_LOG, TOP_LEFT,BOTTOM_RIGHT);
\r
53 AddAnchor(IDOK,BOTTOM_RIGHT);
\r
54 AddAnchor(IDCANCEL,BOTTOM_RIGHT);
\r
55 AddAnchor(IDC_PROGRESS_BUTTON1,BOTTOM_RIGHT);
\r
57 this->GetDlgItem(IDC_PROGRESS_BUTTON1)->ShowWindow(SW_HIDE);
\r
58 m_Animate.Open(IDR_DOWNLOAD);
\r
60 CString InitialText;
\r
61 if ( !m_PreText.IsEmpty() )
\r
63 InitialText = m_PreText + _T("\r\n");
\r
66 if (m_bShowCommand && (!m_GitCmd.IsEmpty() ))
\r
68 InitialText += m_GitCmd+_T("\r\n\r\n");
\r
71 m_Log.SetWindowTextW(InitialText);
\r
72 m_CurrentWork.SetWindowTextW(_T(""));
\r
74 m_pThread = AfxBeginThread(ProgressThreadEntry, this, THREAD_PRIORITY_NORMAL,0,CREATE_SUSPENDED);
\r
75 if (m_pThread==NULL)
\r
77 // ReportError(CString(MAKEINTRESOURCE(IDS_ERR_THREADSTARTFAILED)));
\r
81 m_pThread->m_bAutoDelete = FALSE;
\r
82 m_pThread->ResumeThread();
\r
85 if(!m_Title.IsEmpty())
\r
86 this->SetWindowText(m_Title);
\r
90 UINT CProgressDlg::ProgressThreadEntry(LPVOID pVoid)
\r
92 return ((CProgressDlg*)pVoid)->ProgressThread();
\r
95 //static function, Share with SyncDialog
\r
96 UINT CProgressDlg::RunCmdList(CWnd *pWnd,std::vector<CString> &cmdlist,bool bShowCommand,CString *pfilename,bool *bAbort,std::vector<TCHAR>*pdata)
\r
100 PROCESS_INFORMATION pi;
\r
103 pWnd->PostMessage(MSG_PROGRESSDLG_UPDATE_UI,MSG_PROGRESSDLG_START,0);
\r
108 for(int i=0;i<cmdlist.size();i++)
\r
110 if(cmdlist[i].IsEmpty())
\r
116 str+= cmdlist[i]+_T("\n\n");
\r
117 for(int j=0;j<str.GetLength();j++)
\r
120 pdata->push_back(str[j]);
\r
122 pWnd->PostMessage(MSG_PROGRESSDLG_UPDATE_UI,MSG_PROGRESSDLG_RUN,str[j]);
\r
125 pWnd->PostMessage(MSG_PROGRESSDLG_UPDATE_UI,MSG_PROGRESSDLG_RUN,0);
\r
128 g_Git.RunAsync(cmdlist[i],&pi, &hRead,pfilename);
\r
133 while(ReadFile(hRead,buffer,1,&readnumber,NULL))
\r
135 buffer[readnumber]=0;
\r
139 pdata->push_back((TCHAR)buffer[0]);
\r
141 if(buffer[0] == '\r' || buffer[0] == '\n')
\r
142 pWnd->PostMessage(MSG_PROGRESSDLG_UPDATE_UI,MSG_PROGRESSDLG_RUN,0);
\r
144 pWnd->PostMessage(MSG_PROGRESSDLG_UPDATE_UI,MSG_PROGRESSDLG_RUN,buffer[0]);
\r
147 CloseHandle(pi.hThread);
\r
149 WaitForSingleObject(pi.hProcess, INFINITE);
\r
152 if(!GetExitCodeProcess(pi.hProcess,&status) || *bAbort)
\r
154 CloseHandle(pi.hProcess);
\r
156 CloseHandle(hRead);
\r
158 pWnd->PostMessage(MSG_PROGRESSDLG_UPDATE_UI,MSG_PROGRESSDLG_FAILED,0);
\r
159 return GIT_ERROR_GET_EXIT_CODE;
\r
164 CloseHandle(pi.hProcess);
\r
166 CloseHandle(hRead);
\r
168 pWnd->PostMessage(MSG_PROGRESSDLG_UPDATE_UI,MSG_PROGRESSDLG_END,0);
\r
174 UINT CProgressDlg::ProgressThread()
\r
177 m_GitCmdList.push_back(m_GitCmd);
\r
179 CString *pfilename;
\r
181 if(m_LogFile.IsEmpty())
\r
184 pfilename=&m_LogFile;
\r
186 m_GitStatus = RunCmdList(this,m_GitCmdList,m_bShowCommand,pfilename,&m_bAbort,&this->m_Databuf);;
\r
190 LRESULT CProgressDlg::OnProgressUpdateUI(WPARAM wParam,LPARAM lParam)
\r
192 if(wParam == MSG_PROGRESSDLG_START)
\r
195 m_Animate.Play(0,-1,-1);
\r
196 this->DialogEnableWindow(IDOK,FALSE);
\r
198 if(wParam == MSG_PROGRESSDLG_END || wParam == MSG_PROGRESSDLG_FAILED)
\r
202 m_Databuf.push_back(0);
\r
204 m_Log.SetWindowText(&m_Databuf[0]);
\r
207 this->m_Databuf.clear();
\r
211 m_Progress.SetPos(100);
\r
212 this->DialogEnableWindow(IDOK,TRUE);
\r
214 if(wParam == MSG_PROGRESSDLG_END && m_GitStatus == 0)
\r
216 if(m_bAutoCloseOnSuccess)
\r
219 if(!m_changeAbortButtonOnSuccessTo.IsEmpty())
\r
221 GetDlgItem(IDC_PROGRESS_BUTTON1)->SetWindowText(m_changeAbortButtonOnSuccessTo);
\r
222 GetDlgItem(IDC_PROGRESS_BUTTON1)->ShowWindow(SW_SHOW);
\r
223 GetDlgItem(IDCANCEL)->ShowWindow(SW_HIDE);
\r
224 //Set default button is "close" rather than "push"
\r
225 this->SendMessage(WM_NEXTDLGCTL, (WPARAM)GetDlgItem(IDOK)->m_hWnd, TRUE);
\r
228 DialogEnableWindow(IDCANCEL, FALSE);
\r
231 DialogEnableWindow(IDCANCEL, FALSE);
\r
238 for(int i=this->m_BufStart;i<this->m_Databuf.size();i++)
\r
240 ParserCmdOutput(this->m_Databuf[m_BufStart]);
\r
244 ParserCmdOutput((TCHAR)lParam);
\r
249 //static function, Share with SyncDialog
\r
250 int CProgressDlg::FindPercentage(CString &log)
\r
252 int s1=log.Find(_T('%'));
\r
257 for(int i=s1-1;i>=0;i--)
\r
259 if(log[i]>=_T('0') && log[i]<=_T('9'))
\r
264 return _ttol(log.Mid(s2,s1-s2));
\r
267 void CProgressDlg::ParserCmdOutput(TCHAR ch)
\r
269 TRACE(_T("%c"),ch);
\r
270 if( ch == _T('\r') || ch == _T('\n'))
\r
272 TRACE(_T("End Char %s \r\n"),ch==_T('\r')?_T("lf"):_T(""));
\r
273 TRACE(_T("End Char %s \r\n"),ch==_T('\n')?_T("cr"):_T(""));
\r
276 m_Log.GetWindowTextW(text);
\r
279 RemoveLastLine(text);
\r
281 text+=_T("\r\n")+m_LogText;
\r
282 m_Log.SetWindowTextW(text);
\r
284 m_Log.LineScroll(m_Log.GetLineCount());
\r
286 int s1=m_LogText.Find(_T(':'));
\r
287 int s2=m_LogText.Find(_T('%'));
\r
290 this->m_CurrentWork.SetWindowTextW(m_LogText.Left(s1));
\r
291 int pos=FindPercentage(m_LogText);
\r
292 TRACE(_T("Pos %d\r\n"),pos);
\r
294 this->m_Progress.SetPos(pos);
\r
305 void CProgressDlg::RemoveLastLine(CString &str)
\r
308 start=str.ReverseFind(_T('\n'));
\r
310 str=str.Left(start);
\r
313 // CProgressDlg message handlers
\r
315 void CProgressDlg::OnBnClickedOk()
\r
317 // TODO: Add your control notification handler code here
\r
318 m_Log.GetWindowText(this->m_LogText);
\r
322 void CProgressDlg::OnBnClickedButton1()
\r
324 this->EndDialog(IDC_PROGRESS_BUTTON1);
\r
327 void CProgressDlg::OnCancel()
\r
332 CResizableStandAloneDialog::OnCancel();
\r
336 if( g_Git.m_CurrentGitPi.hProcess )
\r
338 if(::GenerateConsoleCtrlEvent(CTRL_C_EVENT,0))
\r
340 ::WaitForSingleObject(g_Git.m_CurrentGitPi.hProcess ,10000);
\r
343 int error=::GetLastError();
\r
346 HANDLE hProcessHandle = ::OpenProcess( PROCESS_TERMINATE, FALSE,g_Git.m_CurrentGitPi.dwProcessId);
\r
347 if( hProcessHandle )
\r
348 if(!::TerminateProcess(hProcessHandle,-1) )
\r
350 int error =::GetLastError();
\r
354 ::WaitForSingleObject(g_Git.m_CurrentGitPi.hProcess ,10000);
\r
355 CResizableStandAloneDialog::OnCancel();
\r
359 void CProgressDlg::InsertCRLF()
\r
361 for(int i=0;i<m_Databuf.size();i++)
\r
363 if(m_Databuf[i]==_T('\n'))
\r
365 if(i==0 || m_Databuf[i-1]!= _T('\r'))
\r
367 m_Databuf.insert(m_Databuf.begin()+i,_T('\r'));
\r