src/TortoiseProc/ProgressDlg.cpp

   1 // ProgressDlg.cpp : implementation file
   2 //
   3
   4 #include "stdafx.h"
   5 #include "TortoiseProc.h"
   6 #include "ProgressDlg.h"
   7 #include "Git.h"
   8 #include "atlconv.h"
   9 // CProgressDlg dialog
  10
  11 IMPLEMENT_DYNAMIC(CProgressDlg, CResizableStandAloneDialog)
  12
  13 CProgressDlg::CProgressDlg(CWnd* pParent /*=NULL*/)
  14         : CResizableStandAloneDialog(CProgressDlg::IDD, pParent), m_bShowCommand(true), m_bAutoCloseOnSuccess(false), m_bAbort(false), m_bDone(false)
  15 {
  16         m_pThread = NULL;
  17         m_bAltAbortPress=false;
  18         m_bBufferAll=false;
  19 }
  20
  21 CProgressDlg::~CProgressDlg()
  22 {
  23         if(m_pThread != NULL)
  24         {
  25                 delete m_pThread;
  26         }
  27 }
  28
  29 void CProgressDlg::DoDataExchange(CDataExchange* pDX)
  30 {
  31         CDialog::DoDataExchange(pDX);
  32         DDX_Control(pDX, IDC_CURRENT, this->m_CurrentWork);
  33         DDX_Control(pDX, IDC_TITLE_ANIMATE, this->m_Animate);
  34         DDX_Control(pDX, IDC_RUN_PROGRESS, this->m_Progress);
  35         DDX_Control(pDX, IDC_LOG, this->m_Log);
  36 }
  37
  38
  39 BEGIN_MESSAGE_MAP(CProgressDlg, CResizableStandAloneDialog)
  40         ON_MESSAGE(MSG_PROGRESSDLG_UPDATE_UI, OnProgressUpdateUI)
  41         ON_BN_CLICKED(IDOK, &CProgressDlg::OnBnClickedOk)
  42         ON_BN_CLICKED(IDC_PROGRESS_BUTTON1,&CProgressDlg::OnBnClickedButton1)
  43 END_MESSAGE_MAP()
  44
  45 BOOL CProgressDlg::OnInitDialog()
  46 {
  47         CResizableStandAloneDialog::OnInitDialog();
  48
  49         AddAnchor(IDC_TITLE_ANIMATE, TOP_LEFT, TOP_RIGHT);
  50         AddAnchor(IDC_RUN_PROGRESS, TOP_LEFT,TOP_RIGHT);
  51         AddAnchor(IDC_LOG, TOP_LEFT,BOTTOM_RIGHT);
  52
  53         AddAnchor(IDOK,BOTTOM_RIGHT);
  54         AddAnchor(IDCANCEL,BOTTOM_RIGHT);
  55         AddAnchor(IDC_PROGRESS_BUTTON1,BOTTOM_RIGHT);
  56
  57         this->GetDlgItem(IDC_PROGRESS_BUTTON1)->ShowWindow(SW_HIDE);
  58         m_Animate.Open(IDR_DOWNLOAD);
  59
  60         CString InitialText;
  61         if ( !m_PreText.IsEmpty() )
  62         {
  63                 InitialText = m_PreText + _T("\r\n");
  64         }
  65 #if 0
  66         if (m_bShowCommand && (!m_GitCmd.IsEmpty() ))
  67         {
  68                 InitialText += m_GitCmd+_T("\r\n\r\n");
  69         }
  70 #endif
  71         m_Log.SetWindowTextW(InitialText);
  72         m_CurrentWork.SetWindowTextW(_T(""));
  73
  74         m_pThread = AfxBeginThread(ProgressThreadEntry, this, THREAD_PRIORITY_NORMAL,0,CREATE_SUSPENDED);
  75         if (m_pThread==NULL)
  76         {
  77 //              ReportError(CString(MAKEINTRESOURCE(IDS_ERR_THREADSTARTFAILED)));
  78         }
  79         else
  80         {
  81                 m_pThread->m_bAutoDelete = FALSE;
  82                 m_pThread->ResumeThread();
  83         }
  84
  85         if(!m_Title.IsEmpty())
  86                 this->SetWindowText(m_Title);
  87         return TRUE;
  88 }
  89
  90 UINT CProgressDlg::ProgressThreadEntry(LPVOID pVoid)
  91 {
  92         return ((CProgressDlg*)pVoid)->ProgressThread();
  93 }
  94
  95 //static function, Share with SyncDialog
  96 UINT CProgressDlg::RunCmdList(CWnd *pWnd,std::vector<CString> &cmdlist,bool bShowCommand,CString *pfilename,bool *bAbort,std::vector<TCHAR>*pdata)
  97 {
  98         UINT ret=0;
  99
 100         PROCESS_INFORMATION pi;
 101         HANDLE hRead;
 102
 103         pWnd->PostMessage(MSG_PROGRESSDLG_UPDATE_UI,MSG_PROGRESSDLG_START,0);
 104
 105         if(pdata)
 106                 pdata->clear();
 107
 108         for(int i=0;i<cmdlist.size();i++)
 109         {
 110                 if(cmdlist[i].IsEmpty())
 111                         continue;
 112
 113                 if (bShowCommand)
 114                 {
 115                         CString str;
 116                         str+= cmdlist[i]+_T("\n\n");
 117                         for(int j=0;j<str.GetLength();j++)
 118                         {
 119                                 if(pdata)
 120                                         pdata->push_back(str[j]);
 121                                 else
 122                                         pWnd->PostMessage(MSG_PROGRESSDLG_UPDATE_UI,MSG_PROGRESSDLG_RUN,str[j]);
 123                         }
 124                         if(pdata)
 125                                 pWnd->PostMessage(MSG_PROGRESSDLG_UPDATE_UI,MSG_PROGRESSDLG_RUN,0);
 126                 }
 127
 128                 g_Git.RunAsync(cmdlist[i],&pi, &hRead,pfilename);
 129
 130                 DWORD readnumber;
 131                 char buffer[2];
 132                 CString output;
 133                 while(ReadFile(hRead,buffer,1,&readnumber,NULL))
 134                 {
 135                         buffer[readnumber]=0;
 136
 137                         if(pdata)
 138                         {
 139                                 pdata->push_back((TCHAR)buffer[0]);
 140
 141                                 if(buffer[0] == '\r' || buffer[0] == '\n')
 142                                         pWnd->PostMessage(MSG_PROGRESSDLG_UPDATE_UI,MSG_PROGRESSDLG_RUN,0);
 143                         }else
 144                                 pWnd->PostMessage(MSG_PROGRESSDLG_UPDATE_UI,MSG_PROGRESSDLG_RUN,buffer[0]);
 145                 }
 146
 147                 CloseHandle(pi.hThread);
 148
 149                 WaitForSingleObject(pi.hProcess, INFINITE);
 150
 151                 DWORD status=0;
 152                 if(!GetExitCodeProcess(pi.hProcess,&status) || *bAbort)
 153                 {
 154                         CloseHandle(pi.hProcess);
 155
 156                         CloseHandle(hRead);
 157
 158                         pWnd->PostMessage(MSG_PROGRESSDLG_UPDATE_UI,MSG_PROGRESSDLG_FAILED,0);
 159                         return GIT_ERROR_GET_EXIT_CODE;
 160                 }
 161                 ret |= status;
 162         }
 163
 164         CloseHandle(pi.hProcess);
 165
 166         CloseHandle(hRead);
 167
 168         pWnd->PostMessage(MSG_PROGRESSDLG_UPDATE_UI,MSG_PROGRESSDLG_END,0);
 169
 170         return ret;
 171
 172 }
 173
 174 UINT CProgressDlg::ProgressThread()
 175 {
 176
 177         m_GitCmdList.push_back(m_GitCmd);
 178
 179         CString *pfilename;
 180
 181         if(m_LogFile.IsEmpty())
 182                 pfilename=NULL;
 183         else
 184                 pfilename=&m_LogFile;
 185
 186         m_GitStatus = RunCmdList(this,m_GitCmdList,m_bShowCommand,pfilename,&m_bAbort,&this->m_Databuf);;
 187         return 0;
 188 }
 189
 190 LRESULT CProgressDlg::OnProgressUpdateUI(WPARAM wParam,LPARAM lParam)
 191 {
 192         if(wParam == MSG_PROGRESSDLG_START)
 193         {
 194                 m_BufStart = 0 ;
 195                 m_Animate.Play(0,-1,-1);
 196                 this->DialogEnableWindow(IDOK,FALSE);
 197         }
 198         if(wParam == MSG_PROGRESSDLG_END || wParam == MSG_PROGRESSDLG_FAILED)
 199         {
 200                 if(m_bBufferAll)
 201                 {
 202                         m_Databuf.push_back(0);
 203                         InsertCRLF();
 204                         m_Log.SetWindowText(&m_Databuf[0]);
 205                 }
 206                 m_BufStart=0;
 207                 this->m_Databuf.clear();
 208
 209                 m_bDone = true;
 210                 m_Animate.Stop();
 211                 m_Progress.SetPos(100);
 212                 this->DialogEnableWindow(IDOK,TRUE);
 213
 214                 CString err;
 215                 err.Format(_T("\r\nFailed 0x%x(git return wrong reture code at sometime )\r\n"),m_GitStatus);
 216                 if(this->m_GitStatus)
 217                         InsertColorText(this->m_Log,err,RGB(255,0,0));
 218                 else
 219                         InsertColorText(this->m_Log,_T("\r\nSuccess\r\n"),RGB(0,0,255));
 220
 221                 if(wParam == MSG_PROGRESSDLG_END && m_GitStatus == 0)
 222                 {
 223                         if(m_bAutoCloseOnSuccess)
 224                                 EndDialog(IDOK);
 225
 226                         if(!m_changeAbortButtonOnSuccessTo.IsEmpty())
 227                         {
 228                                 GetDlgItem(IDC_PROGRESS_BUTTON1)->SetWindowText(m_changeAbortButtonOnSuccessTo);
 229                                 GetDlgItem(IDC_PROGRESS_BUTTON1)->ShowWindow(SW_SHOW);
 230                                 GetDlgItem(IDCANCEL)->ShowWindow(SW_HIDE);
 231                                 //Set default button is "close" rather than "push"
 232                                 this->SendMessage(WM_NEXTDLGCTL, (WPARAM)GetDlgItem(IDOK)->m_hWnd, TRUE);
 233                         }
 234                         else
 235                                 DialogEnableWindow(IDCANCEL, FALSE);
 236                 }
 237                 else
 238                         DialogEnableWindow(IDCANCEL, FALSE);
 239         }
 240
 241         if(!m_bBufferAll)
 242         {
 243                 if(lParam == 0)
 244                 {
 245                         for(int i=this->m_BufStart;i<this->m_Databuf.size();i++)
 246                         {
 247                                 ParserCmdOutput(this->m_Databuf[m_BufStart]);
 248                                 m_BufStart++;
 249                         }
 250                 }else
 251                         ParserCmdOutput((TCHAR)lParam);
 252         }
 253         return 0;
 254 }
 255
 256 //static function, Share with SyncDialog
 257 int CProgressDlg::FindPercentage(CString &log)
 258 {
 259         int s1=log.Find(_T('%'));
 260         if(s1<0)
 261                 return -1;
 262
 263         int s2=s1-1;
 264         for(int i=s1-1;i>=0;i--)
 265         {
 266                 if(log[i]>=_T('0') && log[i]<=_T('9'))
 267                         s2=i;
 268                 else
 269                         break;
 270         }
 271         return _ttol(log.Mid(s2,s1-s2));
 272 }
 273
 274 #if 0
 275 void CProgressDlg::ParserCmdOutput(TCHAR ch)
 276 {
 277         TRACE(_T("%c"),ch);
 278         if( ch == _T('\r') || ch == _T('\n'))
 279         {
 280                 TRACE(_T("End Char %s \r\n"),ch==_T('\r')?_T("lf"):_T(""));
 281                 TRACE(_T("End Char %s \r\n"),ch==_T('\n')?_T("cr"):_T(""));
 282
 283                 int lines=m_Log.GetLineCount();
 284
 285                 if(ch == _T('\r'))
 286                 {
 287                         int start=m_Log.LineIndex(lines-1);
 288                         int length=m_Log.LineLength(lines-1);
 289                         m_Log.SetSel( start,start+length);
 290                         m_Log.ReplaceSel(m_LogText);
 291
 292                 }else
 293                 {
 294                         m_Log.SetSel(m_Log.GetWindowTextLength(),
 295                                              m_Log.GetWindowTextLength());
 296                         m_Log.ReplaceSel(CString(_T("\r\n"))+m_LogText);
 297                 }
 298
 299                 if( lines > 500 ) //limited log length
 300                 {
 301                         int end=m_Log.LineIndex(1);
 302                         m_Log.SetSel(0,end);
 303                         m_Log.ReplaceSel(_T(""));
 304                 }
 305                 m_Log.LineScroll(m_Log.GetLineCount());
 306
 307                 int s1=m_LogText.Find(_T(':'));
 308                 int s2=m_LogText.Find(_T('%'));
 309                 if(s1>0 && s2>0)
 310                 {
 311                         this->m_CurrentWork.SetWindowTextW(m_LogText.Left(s1));
 312                         int pos=FindPercentage(m_LogText);
 313                         TRACE(_T("Pos %d\r\n"),pos);
 314                         if(pos>0)
 315                                 this->m_Progress.SetPos(pos);
 316                 }
 317
 318                 m_LogText=_T("");
 319
 320         }else
 321         {
 322                 m_LogText+=ch;
 323         }
 324
 325 }
 326 #endif
 327
 328 void CProgressDlg::ParserCmdOutput(TCHAR ch)
 329 {
 330         ParserCmdOutput(this->m_Log,this->m_Progress,this->m_LogText,ch,&this->m_CurrentWork);
 331 }
 332 void CProgressDlg::ParserCmdOutput(CRichEditCtrl &log,CProgressCtrl &progressctrl,CString &oneline, TCHAR ch, CWnd *CurrentWork)
 333 {
 334         //TRACE(_T("%c"),ch);
 335         TRACE(_T("%c"),ch);
 336         if( ch == _T('\r') || ch == _T('\n'))
 337         {
 338                 TRACE(_T("End Char %s \r\n"),ch==_T('\r')?_T("lf"):_T(""));
 339                 TRACE(_T("End Char %s \r\n"),ch==_T('\n')?_T("cr"):_T(""));
 340
 341                 int lines=log.GetLineCount();
 342
 343                 if(ch == _T('\r'))
 344                 {
 345                         int start=log.LineIndex(lines-1);
 346                         int length=log.LineLength(lines-1);
 347                         log.SetSel( start,start+length);
 348                         log.ReplaceSel(oneline);
 349
 350                 }else
 351                 {
 352                         log.SetSel(log.GetWindowTextLength(),
 353                                              log.GetWindowTextLength());
 354                         log.ReplaceSel(CString(_T("\r\n"))+oneline);
 355                 }
 356
 357                 if( lines > 500 ) //limited log length
 358                 {
 359                         int end=log.LineIndex(1);
 360                         log.SetSel(0,end);
 361                         log.ReplaceSel(_T(""));
 362                 }
 363                 log.LineScroll(log.GetLineCount());
 364
 365                 int s1=oneline.Find(_T(':'));
 366                 int s2=oneline.Find(_T('%'));
 367                 if(s1>0 && s2>0)
 368                 {
 369                         if(CurrentWork)
 370                                 CurrentWork->SetWindowTextW(oneline.Left(s1));
 371
 372                         int pos=FindPercentage(oneline);
 373                         TRACE(_T("Pos %d\r\n"),pos);
 374                         if(pos>0)
 375                                 progressctrl.SetPos(pos);
 376                 }
 377
 378                 oneline=_T("");
 379
 380         }else
 381         {
 382                 oneline+=ch;
 383         }
 384
 385 }
 386 void CProgressDlg::RemoveLastLine(CString &str)
 387 {
 388         int start;
 389         start=str.ReverseFind(_T('\n'));
 390         if(start>0)
 391                 str=str.Left(start);
 392         return;
 393 }
 394 // CProgressDlg message handlers
 395
 396 void CProgressDlg::OnBnClickedOk()
 397 {
 398         // TODO: Add your control notification handler code here
 399         m_Log.GetWindowText(this->m_LogText);
 400         OnOK();
 401 }
 402
 403 void CProgressDlg::OnBnClickedButton1()
 404 {
 405         this->EndDialog(IDC_PROGRESS_BUTTON1);
 406
 407 }
 408 void CProgressDlg::OnCancel()
 409 {
 410         m_bAbort = true;
 411         if(m_bDone)
 412         {
 413                 CResizableStandAloneDialog::OnCancel();
 414                 return;
 415         }
 416
 417         if( g_Git.m_CurrentGitPi.hProcess )
 418         {
 419                 if(::GenerateConsoleCtrlEvent(CTRL_C_EVENT,0))
 420                 {
 421                         ::WaitForSingleObject(g_Git.m_CurrentGitPi.hProcess ,10000);
 422                 }else
 423                 {
 424                         int error=::GetLastError();
 425                 }
 426
 427                 HANDLE  hProcessHandle = ::OpenProcess( PROCESS_TERMINATE, FALSE,g_Git.m_CurrentGitPi.dwProcessId);
 428                 if( hProcessHandle )
 429                         if(!::TerminateProcess(hProcessHandle,-1) )
 430                         {
 431                                 int error =::GetLastError();
 432                         }
 433         }
 434
 435         ::WaitForSingleObject(g_Git.m_CurrentGitPi.hProcess ,10000);
 436         CResizableStandAloneDialog::OnCancel();
 437
 438 }
 439
 440 void CProgressDlg::InsertCRLF()
 441 {
 442         for(int i=0;i<m_Databuf.size();i++)
 443         {
 444                 if(m_Databuf[i]==_T('\n'))
 445                 {
 446                         if(i==0 || m_Databuf[i-1]!= _T('\r'))
 447                         {
 448                                 m_Databuf.insert(m_Databuf.begin()+i,_T('\r'));
 449                                 i++;
 450                         }
 451                 }
 452         }
 453 }
 454
 455 void CProgressDlg::InsertColorText(CRichEditCtrl &edit,CString text,COLORREF rgb)
 456 {
 457         CHARFORMAT old,cf;
 458         edit.GetDefaultCharFormat(cf);
 459         old=cf;
 460         cf.dwMask|=CFM_COLOR;
 461         cf.crTextColor=rgb;
 462         cf.dwEffects|=CFE_BOLD;
 463         cf.dwEffects &= ~CFE_AUTOCOLOR ;
 464         edit.SetSel(edit.GetTextLength()-1,edit.GetTextLength());
 465         edit.ReplaceSel(text);
 466         edit.SetSel(edit.LineIndex(edit.GetLineCount()-2),edit.GetTextLength());
 467         edit.SetSelectionCharFormat(cf);
 468         edit.SetSel(edit.GetTextLength(),edit.GetTextLength());
 469         edit.SetDefaultCharFormat(old);
 470         edit.LineScroll(edit.GetLineCount());
 471 }
 472