src/TortoiseProc/ProgressDlg.cpp

   1 // ProgressDlg.cpp : implementation file\r
   2 //\r
   3 \r
   4 #include "stdafx.h"\r
   5 #include "TortoiseProc.h"\r
   6 #include "ProgressDlg.h"\r
   7 #include "Git.h"\r
   8 #include "atlconv.h"\r
   9 // CProgressDlg dialog\r
  10 \r
  11 IMPLEMENT_DYNAMIC(CProgressDlg, CResizableStandAloneDialog)\r
  12 \r
  13 CProgressDlg::CProgressDlg(CWnd* pParent /*=NULL*/)\r
  14         : CResizableStandAloneDialog(CProgressDlg::IDD, pParent), m_bShowCommand(true), m_bAutoCloseOnSuccess(false)\r
  15 {\r
  16 \r
  17 }\r
  18 \r
  19 CProgressDlg::~CProgressDlg()\r
  20 {\r
  21         if(m_pThread != NULL)\r
  22         {\r
  23                 delete m_pThread;\r
  24         }\r
  25 }\r
  26 \r
  27 void CProgressDlg::DoDataExchange(CDataExchange* pDX)\r
  28 {\r
  29         CDialog::DoDataExchange(pDX);\r
  30         DDX_Control(pDX, IDC_CURRENT, this->m_CurrentWork);\r
  31         DDX_Control(pDX, IDC_TITLE_ANIMATE, this->m_Animate);\r
  32         DDX_Control(pDX, IDC_RUN_PROGRESS, this->m_Progress);\r
  33         DDX_Control(pDX, IDC_LOG, this->m_Log);\r
  34 }\r
  35 \r
  36 \r
  37 BEGIN_MESSAGE_MAP(CProgressDlg, CResizableStandAloneDialog)\r
  38         ON_MESSAGE(MSG_PROGRESSDLG_UPDATE_UI, OnProgressUpdateUI)\r
  39         ON_BN_CLICKED(IDOK, &CProgressDlg::OnBnClickedOk)\r
  40 END_MESSAGE_MAP()\r
  41 \r
  42 BOOL CProgressDlg::OnInitDialog()\r
  43 {\r
  44         CResizableStandAloneDialog::OnInitDialog();\r
  45 \r
  46         AddAnchor(IDC_TITLE_ANIMATE, TOP_LEFT, TOP_RIGHT);\r
  47         AddAnchor(IDC_RUN_PROGRESS, TOP_LEFT,TOP_RIGHT);\r
  48         AddAnchor(IDC_LOG, TOP_LEFT,BOTTOM_RIGHT);\r
  49 \r
  50         AddAnchor(IDOK,BOTTOM_RIGHT);\r
  51         AddAnchor(IDCANCEL,BOTTOM_RIGHT);\r
  52 \r
  53         m_Animate.Open(IDR_DOWNLOAD);\r
  54         \r
  55         CString InitialText;\r
  56         if ( !m_PreText.IsEmpty() )\r
  57         {\r
  58                 InitialText = m_PreText + _T("\r\n");\r
  59         }\r
  60         if (m_bShowCommand && (!m_GitCmd.IsEmpty() ))\r
  61         {\r
  62                 InitialText += m_GitCmd+_T("\r\n\r\n");\r
  63         }\r
  64         m_Log.SetWindowTextW(InitialText);\r
  65         m_CurrentWork.SetWindowTextW(_T(""));\r
  66 \r
  67         m_pThread = AfxBeginThread(ProgressThreadEntry, this, THREAD_PRIORITY_NORMAL,0,CREATE_SUSPENDED);\r
  68         if (m_pThread==NULL)\r
  69         {\r
  70 //              ReportError(CString(MAKEINTRESOURCE(IDS_ERR_THREADSTARTFAILED)));\r
  71         }\r
  72         else\r
  73         {\r
  74                 m_pThread->m_bAutoDelete = FALSE;\r
  75                 m_pThread->ResumeThread();\r
  76         }\r
  77 \r
  78         if(!m_Title.IsEmpty())\r
  79                 this->SetWindowText(m_Title);\r
  80         return TRUE;\r
  81 }\r
  82 \r
  83 UINT CProgressDlg::ProgressThreadEntry(LPVOID pVoid)\r
  84 {\r
  85         return ((CProgressDlg*)pVoid)->ProgressThread();\r
  86 }\r
  87 \r
  88 UINT CProgressDlg::ProgressThread()\r
  89 {\r
  90         PROCESS_INFORMATION pi;\r
  91         HANDLE hRead;\r
  92 \r
  93         this->PostMessage(MSG_PROGRESSDLG_UPDATE_UI,MSG_PROGRESSDLG_START,0);\r
  94 \r
  95         CString *pfilename;\r
  96         if(m_LogFile.IsEmpty())\r
  97                 pfilename=NULL;\r
  98         else\r
  99                 pfilename=&m_LogFile;\r
 100 \r
 101         m_GitCmdList.push_back(m_GitCmd);\r
 102 \r
 103         m_GitStatus =0;\r
 104 \r
 105         for(int i=0;i<m_GitCmdList.size();i++)\r
 106         {\r
 107                 if(m_GitCmdList[i].IsEmpty())\r
 108                         continue;\r
 109 \r
 110                 if (m_bShowCommand && m_GitCmdList[i]!= m_GitCmd)\r
 111                 {\r
 112                         CString str;\r
 113                         str+= m_GitCmdList[i]+_T("\r\n\r\n");\r
 114                         for(int j=0;j<str.GetLength();j++)\r
 115                                 this->PostMessage(MSG_PROGRESSDLG_UPDATE_UI,MSG_PROGRESSDLG_RUN,str[j]);\r
 116                 }\r
 117 \r
 118                 g_Git.RunAsync(this->m_GitCmdList[i],&pi, &hRead,pfilename);\r
 119 \r
 120                 DWORD readnumber;\r
 121                 char buffer[2];\r
 122                 CString output;\r
 123                 while(ReadFile(hRead,buffer,1,&readnumber,NULL))\r
 124                 {\r
 125                         buffer[readnumber]=0;\r
 126                         this->PostMessage(MSG_PROGRESSDLG_UPDATE_UI,MSG_PROGRESSDLG_RUN,(TCHAR)buffer[0]);\r
 127                 }\r
 128         \r
 129                 CloseHandle(pi.hThread);\r
 130 \r
 131                 WaitForSingleObject(pi.hProcess, INFINITE);\r
 132                 \r
 133                 DWORD status=0;\r
 134                 if(!GetExitCodeProcess(pi.hProcess,&status))\r
 135                 {\r
 136                         CloseHandle(pi.hProcess);\r
 137 \r
 138                         CloseHandle(hRead);\r
 139 \r
 140                         this->PostMessage(MSG_PROGRESSDLG_UPDATE_UI,MSG_PROGRESSDLG_FAILED,0);\r
 141                         return GIT_ERROR_GET_EXIT_CODE;\r
 142                 }\r
 143                 m_GitStatus |= status;\r
 144         }\r
 145 \r
 146         CloseHandle(pi.hProcess);\r
 147 \r
 148         CloseHandle(hRead);\r
 149 \r
 150         this->PostMessage(MSG_PROGRESSDLG_UPDATE_UI,MSG_PROGRESSDLG_END,0);\r
 151 \r
 152         return 0;\r
 153 }\r
 154 \r
 155 LRESULT CProgressDlg::OnProgressUpdateUI(WPARAM wParam,LPARAM lParam)\r
 156 {\r
 157         if(wParam == MSG_PROGRESSDLG_START)\r
 158         {\r
 159                 m_Animate.Play(0,-1,-1);\r
 160                 this->DialogEnableWindow(IDOK,FALSE);\r
 161         }\r
 162         if(wParam == MSG_PROGRESSDLG_END || wParam == MSG_PROGRESSDLG_FAILED)\r
 163         {\r
 164                 m_Animate.Stop();\r
 165                 m_Progress.SetPos(100);\r
 166                 this->DialogEnableWindow(IDOK,TRUE);\r
 167                 if(m_bAutoCloseOnSuccess && wParam == MSG_PROGRESSDLG_END)\r
 168                         EndDialog(IDOK);\r
 169         }\r
 170 \r
 171         if(lParam != 0)\r
 172                 ParserCmdOutput((TCHAR)lParam);\r
 173 \r
 174         return 0;\r
 175 }\r
 176 int CProgressDlg::FindPercentage(CString &log)\r
 177 {\r
 178         int s1=log.Find(_T('%'));\r
 179         if(s1<0)\r
 180                 return -1;\r
 181 \r
 182         int s2=s1-1;\r
 183         for(int i=s1-1;i>=0;i--)\r
 184         {\r
 185                 if(log[i]>=_T('0') && log[i]<=_T('9'))\r
 186                         s2=i;\r
 187                 else\r
 188                         break;\r
 189         }\r
 190         return _ttol(log.Mid(s2,s1-s2));\r
 191 }\r
 192 \r
 193 void CProgressDlg::ParserCmdOutput(TCHAR ch)\r
 194 {\r
 195         TRACE(_T("%c"),ch);\r
 196         if( ch == _T('\r') || ch == _T('\n'))\r
 197         {\r
 198                 TRACE(_T("End Char %s \r\n"),ch==_T('\r')?_T("lf"):_T(""));\r
 199                 TRACE(_T("End Char %s \r\n"),ch==_T('\n')?_T("cr"):_T(""));\r
 200 \r
 201                 CString text;\r
 202                 m_Log.GetWindowTextW(text);\r
 203                 if(ch == _T('\r'))\r
 204                 {\r
 205                         RemoveLastLine(text);\r
 206                 }\r
 207                 text+=_T("\r\n")+m_LogText;\r
 208                 m_Log.SetWindowTextW(text);\r
 209                 \r
 210                 m_Log.LineScroll(m_Log.GetLineCount());\r
 211 \r
 212                 int s1=m_LogText.Find(_T(':'));\r
 213                 int s2=m_LogText.Find(_T('%'));\r
 214                 if(s1>0 && s2>0)\r
 215                 {\r
 216                         this->m_CurrentWork.SetWindowTextW(m_LogText.Left(s1));\r
 217                         int pos=FindPercentage(m_LogText);\r
 218                         TRACE(_T("Pos %d\r\n"),pos);\r
 219                         if(pos>0)\r
 220                                 this->m_Progress.SetPos(pos);\r
 221                 }\r
 222 \r
 223                 m_LogText=_T("");\r
 224 \r
 225         }else\r
 226         {\r
 227                 m_LogText+=ch;\r
 228         }\r
 229 \r
 230 }\r
 231 void CProgressDlg::RemoveLastLine(CString &str)\r
 232 {\r
 233         int start;\r
 234         start=str.ReverseFind(_T('\n'));\r
 235         if(start>0)\r
 236                 str=str.Left(start);\r
 237         return;\r
 238 }\r
 239 // CProgressDlg message handlers\r
 240 \r
 241 void CProgressDlg::OnBnClickedOk()\r
 242 {\r
 243         // TODO: Add your control notification handler code here\r
 244         m_Log.GetWindowText(this->m_LogText);\r
 245         OnOK();\r
 246 }\r