// NeL - MMORPG Framework // Copyright (C) 2010 Winch Gate Property Limited // // This program is free software: you can redistribute it and/or modify // it under the terms of the GNU Affero General Public License as // published by the Free Software Foundation, either version 3 of the // License, or (at your option) any later version. // // This program is distributed in the hope that it will be useful, // but WITHOUT ANY WARRANTY; without even the implied warranty of // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the // GNU Affero General Public License for more details. // // You should have received a copy of the GNU Affero General Public License // along with this program. If not, see . // branch_patcherDlg.cpp : implementation file // #include "stdafx.h" #include "branch_patcher.h" #include "branch_patcherDlg.h" #include "shlobj.h" #include "direct.h" #ifdef _DEBUG #define new DEBUG_NEW #undef THIS_FILE static char THIS_FILE[] = __FILE__; #endif ///////////////////////////////////////////////////////////////////////////// // CBranch_patcherDlg dialog extern CBranch_patcherApp theApp; const CString TEMP_DIFF_FILE = "C:\\tempFile.diff"; const CString DIFF_ERRORS = "C:\\diffLog.txt"; const CString PATCH_RESULT = "C:\\patchResult.txt"; const CString PATCH_ERRORS = "C:\\patchErrors.txt"; CBranch_patcherDlg::CBranch_patcherDlg(CWnd* pParent /*=NULL*/) : CDialog(CBranch_patcherDlg::IDD, pParent) { //{{AFX_DATA_INIT(CBranch_patcherDlg) m_SrcDir = _T(""); m_DestDir = _T(""); m_Filename = _T(""); m_Tokens = _T(""); m_SrcDirLabel = _T(""); m_TargetDirLabel = _T(""); //}}AFX_DATA_INIT m_hIcon = AfxGetApp()->LoadIcon(IDR_MAINFRAME); m_Display = NULL; } void CBranch_patcherDlg::DoDataExchange(CDataExchange* pDX) { CDialog::DoDataExchange(pDX); //{{AFX_DATA_MAP(CBranch_patcherDlg) DDX_Text(pDX, IDC_SRCDIR, m_SrcDir); DDX_Text(pDX, IDC_DESTDIR, m_DestDir); DDX_Text(pDX, IDC_Filename, m_Filename); DDX_Text(pDX, IDC_CurrentTokens, m_Tokens); DDX_Text(pDX, IDC_SrcDirLabel, m_SrcDirLabel); DDX_Text(pDX, IDC_TargetDirLabel, m_TargetDirLabel); //}}AFX_DATA_MAP } BEGIN_MESSAGE_MAP(CBranch_patcherDlg, CDialog) //{{AFX_MSG_MAP(CBranch_patcherDlg) ON_WM_PAINT() ON_WM_QUERYDRAGICON() ON_BN_CLICKED(IDC_ButtonSetSrcDir, OnButtonSetSrcDir) ON_BN_CLICKED(IDC_ButtonSetDestDir, OnButtonSetDestDir) ON_BN_CLICKED(IDC_ButtonPatch, OnButtonPatch) ON_BN_CLICKED(IDC_DoPatch, OnDoPatch) ON_WM_SIZE() ON_WM_CLOSE() ON_BN_CLICKED(IDC_ButtonExtractTokens, OnButtonExtractTokens) ON_BN_CLICKED(IDC_ButtonClearTokens, OnButtonClearTokens) //}}AFX_MSG_MAP END_MESSAGE_MAP() ///////////////////////////////////////////////////////////////////////////// // CBranch_patcherDlg message handlers BOOL CBranch_patcherDlg::OnInitDialog() { CDialog::OnInitDialog(); SetIcon(m_hIcon, TRUE); // Set big icon SetIcon(m_hIcon, FALSE); // Set small icon // Extra initialization here RECT cltRect; GetClientRect( &cltRect ), m_Display = new CRichEditCtrl(); m_Display->Create( WS_CHILD|WS_VISIBLE|WS_BORDER|WS_HSCROLL|WS_VSCROLL|ES_AUTOHSCROLL|ES_AUTOVSCROLL|ES_MULTILINE, CRect( 20, 180, cltRect.right-20, cltRect.bottom-20 ), this, 1 ); // Initialize directories loadConfiguration(); processCommandLine(); displayTokens(); EnteringTokens = false; m_SrcDirLabel = "Source Dir"; m_TargetDirLabel = "Target Dir"; UpdateData( false ); ((CButton*)GetDlgItem( IDC_DoPatch ))->EnableWindow( FALSE ); return TRUE; // return TRUE unless you set the focus to a control } // If you add a minimize button to your dialog, you will need the code below // to draw the icon. For MFC applications using the document/view model, // this is automatically done for you by the framework. void CBranch_patcherDlg::OnPaint() { if (IsIconic()) { CPaintDC dc(this); // device context for painting SendMessage(WM_ICONERASEBKGND, (WPARAM) dc.GetSafeHdc(), 0); // Center icon in client rectangle int cxIcon = GetSystemMetrics(SM_CXICON); int cyIcon = GetSystemMetrics(SM_CYICON); CRect rect; GetClientRect(&rect); int x = (rect.Width() - cxIcon + 1) / 2; int y = (rect.Height() - cyIcon + 1) / 2; // Draw the icon dc.DrawIcon(x, y, m_hIcon); } else { CDialog::OnPaint(); } } HCURSOR CBranch_patcherDlg::OnQueryDragIcon() { return (HCURSOR) m_hIcon; } void CBranch_patcherDlg::setSrcDirectory( const CString& s ) { m_SrcDir = s; UpdateData( false ); } void CBranch_patcherDlg::setDestDirectory( const CString& s ) { m_DestDir = s; UpdateData( false ); } void CBranch_patcherDlg::OnButtonSetSrcDir() { DirDialog.m_strTitle = "Please choose the SOURCE directory"; if ( DirDialog.DoBrowse() == TRUE ) { setSrcDirectory( DirDialog.m_strPath ); guessDestDirectory(); } } void CBranch_patcherDlg::OnButtonSetDestDir() { DirDialog.m_strTitle = "Please choose the TARGET directory"; if ( DirDialog.DoBrowse() == TRUE ) { setDestDirectory( DirDialog.m_strPath ); } } CDirDialog::CDirDialog() {//////////////////////////////////////////// } CDirDialog::~CDirDialog() {/////////////////////////////////////////// } int CDirDialog::DoBrowse () {///////////////////////////////////////// LPMALLOC pMalloc; if (SHGetMalloc (&pMalloc)!= NOERROR) { return 0; } BROWSEINFO bInfo; LPITEMIDLIST pidl; ZeroMemory ( (PVOID) &bInfo,sizeof (BROWSEINFO)); if (!m_strInitDir.IsEmpty ()) { OLECHAR olePath[MAX_PATH]; ULONG chEaten; ULONG dwAttributes; HRESULT hr; LPSHELLFOLDER pDesktopFolder; // // Get a pointer to the Desktop's IShellFolder interface. // if (SUCCEEDED(SHGetDesktopFolder(&pDesktopFolder))) { // // IShellFolder::ParseDisplayName requires the file name be in Unicode. // MultiByteToWideChar(CP_ACP, MB_PRECOMPOSED, m_strInitDir.GetBuffer (MAX_PATH), -1, olePath, MAX_PATH); m_strInitDir.ReleaseBuffer (-1); // // Convert the path to an ITEMIDLIST. // hr = pDesktopFolder->ParseDisplayName(NULL, NULL, olePath, &chEaten, &pidl, &dwAttributes); if (FAILED(hr)) { pMalloc ->Free (pidl); pMalloc ->Release (); return 0; } bInfo.pidlRoot = pidl; } } bInfo.hwndOwner = NULL; bInfo.pszDisplayName = m_strPath.GetBuffer (MAX_PATH); bInfo.lpszTitle = (m_strTitle.IsEmpty()) ? "Open":m_strTitle; bInfo.ulFlags = BIF_RETURNFSANCESTORS|BIF_RETURNONLYFSDIRS; if ((pidl = ::SHBrowseForFolder (&bInfo)) == NULL) { return 0; } m_strPath.ReleaseBuffer (); m_iImageIndex = bInfo.iImage; if (::SHGetPathFromIDList(pidl,m_strPath.GetBuffer (MAX_PATH)) == FALSE) { pMalloc ->Free (pidl); pMalloc ->Release (); return 0; } m_strPath.ReleaseBuffer (); pMalloc ->Free (pidl); pMalloc ->Release (); return 1; } /* * Adapted from function by Jonah Bishop */ BOOL SendTextToClipboard(CString source) { // Return value is TRUE if the text was sent // Return value is FALSE if something went wrong if(OpenClipboard(NULL)) { HGLOBAL clipbuffer; char* buffer; EmptyClipboard(); // Empty whatever's already there clipbuffer = GlobalAlloc(GMEM_DDESHARE, source.GetLength()+1); buffer = (char*)GlobalLock(clipbuffer); strcpy(buffer, LPCSTR(source)); GlobalUnlock(clipbuffer); SetClipboardData(CF_TEXT, clipbuffer); // Send the data CloseClipboard(); // VERY IMPORTANT return TRUE; } return FALSE; } void CBranch_patcherDlg::displayMessage( const CString& msg, bool insertAtTop ) { if ( insertAtTop ) m_Display->SetSel( 0, 0 ); else m_Display->SetSel( 0, -1 ); m_Display->ReplaceSel( msg ); SaveDiff = false; } void CBranch_patcherDlg::OnButtonPatch() { UpdateData( true ); CString diffCmdLine; diffCmdLine.Format( "cvs.exe diff -c > %s 2> %s", TEMP_DIFF_FILE, DIFF_ERRORS ); // needs a valid cvs login before! and cvs.exe in the path CString text; text.Format( "Get diff from directory %s?\n\nCommand (choose No to copy it into the clipboard):\n%s", m_SrcDir, diffCmdLine ); int result; if ( (result = ::MessageBox( m_hWnd, text, "Confirmation", MB_YESNOCANCEL | MB_ICONQUESTION )) == IDYES ) { if ( _chdir( m_SrcDir ) == 0 ) { system( diffCmdLine ); displayFile( TEMP_DIFF_FILE ); SaveDiff = true; colorizeDiff(); m_Display->LineScroll( 0 ); ((CButton*)GetDlgItem( IDC_DoPatch ))->EnableWindow( TRUE ); if ( (m_Display->GetLineCount() == 0) || (m_Display->GetLineCount() == 1 && m_Display->LineLength(0)<2) ) { displayFile( DIFF_ERRORS ); displayMessage( "Diff is empty.\r\nIf this is not the expected result:\r\n- check if the source directory is part of a CVS tree\r\n- check if cvs.exe is in your PATH\r\n- check if you are logged to the cvs server with 'cvs login' (set your home cvs directory in the HOME environment variable if needed)\r\n- check if C:\\ has enough free space and access rights to write a file.\n\nHere is the log:\n\n", true ); } else { m_Filename = TEMP_DIFF_FILE + ":"; UpdateData( false ); } } else { displayMessage( "Source directory not found" ); } } else if ( result == IDNO ) { SendTextToClipboard( diffCmdLine ); } } static unsigned long CALLBACK MyStreamInCallback(DWORD_PTR dwCookie, LPBYTE pbBuff, LONG cb, LONG *pcb) { CFile* pFile = (CFile*) dwCookie; *pcb = pFile->Read(pbBuff, cb); return 0; } static unsigned long CALLBACK MyStreamOutCallback(DWORD_PTR dwCookie, LPBYTE pbBuff, LONG cb, LONG *pcb) { CFile* pFile = (CFile*) dwCookie; pFile->Write(pbBuff, cb); *pcb = cb; return 0; } void CBranch_patcherDlg::displayFile( const CString& filename ) { CFile cFile( filename, CFile::modeRead ); EDITSTREAM es; es.dwCookie = (DWORD_PTR) &cFile; es.pfnCallback = MyStreamInCallback; m_Display->StreamIn( SF_TEXT, es ); } void CBranch_patcherDlg::saveFile( const CString& filename ) { CFile cFile( filename, CFile::modeCreate | CFile::modeWrite ); EDITSTREAM es; es.dwCookie = (DWORD_PTR) &cFile; es.pfnCallback = MyStreamOutCallback; m_Display->StreamOut( SF_TEXT, es ); } void CBranch_patcherDlg::colorizeDiff() { CHARFORMAT blue; ZeroMemory( &blue, sizeof(blue) ); blue.cbSize = sizeof(blue); blue.dwMask = CFM_COLOR; blue.crTextColor = RGB(0,0,0xFF); CHARFORMAT red; ZeroMemory( &red, sizeof(red) ); red.cbSize = sizeof(red); red.dwMask = CFM_COLOR; red.crTextColor = RGB(0xFF,0,0); CHARFORMAT green; ZeroMemory( &green, sizeof(green) ); green.cbSize = sizeof(green); green.dwMask = CFM_COLOR; green.crTextColor = RGB(0,0x7F,0); for ( int i=0; i!=m_Display->GetLineCount(); ++i ) { int c = m_Display->LineIndex( i ); int l = m_Display->LineLength( c ); m_Display->SetSel( c, c+l ); CString s = m_Display->GetSelText(); if ( ! s.IsEmpty() ) { if ( s.Left(2) == "+ " ) { m_Display->SetSelectionCharFormat( blue ); } else if ( s.Left(2) == "- " ) { m_Display->SetSelectionCharFormat( red ); } else if ( s.Left(2) == "! " ) { m_Display->SetSelectionCharFormat( green ); } } } } void CBranch_patcherDlg::OnDoPatch() { UpdateData( true ); if ( SaveDiff ) { // Save the diff from the richedit saveFile( TEMP_DIFF_FILE ); } // Apply the patch CString patchCmdLine, concatOutput, delPatchErrors; patchCmdLine.Format( "%spatch.exe -c -p%u --verbose < %s > %s 2> %s", PatchExeDir, CvsDiffDirLevel, TEMP_DIFF_FILE, PATCH_RESULT, PATCH_ERRORS ); // needs patch.exe in the path concatOutput.Format( "copy %s+%s %s", PATCH_RESULT, PATCH_ERRORS, PATCH_RESULT ); delPatchErrors.Format( "del %s", PATCH_ERRORS ); CString text; text.Format( "Patch diff to directory %s?\n\nCommand (choose No to copy it into the clipboard):\n%s", m_DestDir, patchCmdLine ); int result; if ( (result = ::MessageBox( m_hWnd, text, "Confirmation", MB_YESNOCANCEL | MB_ICONQUESTION )) == IDYES ) { if ( _chdir( m_DestDir ) == 0 ) { system( patchCmdLine ); system( concatOutput ); system( delPatchErrors ); displayFile( PATCH_RESULT ); SaveDiff = false; m_Display->LineScroll( 0 ); if ( (m_Display->GetLineCount() == 0) || (m_Display->GetLineCount() == 1 && m_Display->LineLength(0)<2) ) { CString s; s.Format( "Nothing was patched.\r\nIf this is not the expected result:\r\n- check if the good patch.exe is in %s\r\n- check if %s exists (generated by previous diff)\r\n- check if C:\\ has enough free space and access rights to write a file.", TEMP_DIFF_FILE ); displayMessage( s ); } else { m_Filename = PATCH_RESULT + ":"; UpdateData( false ); } } else { displayMessage( "Target directory not found" ); } } else if ( result == IDNO ) { SendTextToClipboard( patchCmdLine ); } } void CBranch_patcherDlg::OnSize(UINT nType, int cx, int cy) { CDialog::OnSize(nType, cx, cy); if ( m_Display ) { RECT cltRect; GetClientRect( &cltRect ); CRect dispRect; m_Display->MoveWindow( 20, 180, cltRect.right-40, cltRect.bottom-200, true ); } } void CBranch_patcherDlg::OnClose() { saveConfiguration(); CDialog::OnClose(); } void CBranch_patcherDlg::processCommandLine() { CString cmdLine = theApp.m_lpCmdLine; if ( ! cmdLine.IsEmpty() ) { setSrcDirectory( cmdLine ); guessDestDirectory(); } } void CBranch_patcherDlg::guessDestDirectory() { if ( hasTokens() ) { CString dir = m_SrcDir; if ( dir.Find( "\\"+Token1+"\\", 0 ) != -1 ) { dir.Replace( "\\"+Token1+"\\", "\\"+Token2+"\\" ); setDestDirectory( dir ); } else if ( dir.Find( "\\"+Token2+"\\", 0 ) != -1 ) { dir.Replace( "\\"+Token2+"\\", "\\"+Token1+"\\" ); setDestDirectory( dir ); } } } void CBranch_patcherDlg::extractDirTokens() { int beginOfToken1, beginOfToken2, endOfToken1, endOfToken2; CString text; UpdateData( true ); // Search backward from the end until a different substring is found int c1 = m_SrcDir.GetLength()-1; int c2 = m_DestDir.GetLength()-1; while ( (c1 >= 0) && (c2 >= 0) && (m_SrcDir[c1] == m_DestDir[c2]) ) { --c1; --c2; } // Test if both strings are identical if ( (c1 < 0) || (c2 < 0) ) { Token1 = m_SrcDir; Token2 = m_DestDir; return; } endOfToken1 = c1+1; endOfToken2 = c2+1; // Search forward from the beginning until a different substring is found c1 = 0; c2 = 0; while ( (c1 < m_SrcDir.GetLength()) && (c2 < m_DestDir.GetLength()) && (m_SrcDir[c1] == m_DestDir[c2]) ) { ++c1; ++c2; } if ( (c1 == m_SrcDir.GetLength()) || (c2 == m_DestDir.GetLength()) ) { return; // both strings are identical (should not occur again) } // If one of the token is empty, expand both downto the closest backslash if ( (c1 == endOfToken1) || (c2 == endOfToken2) ) { --c1; while ( (c1 >= 0) && (m_SrcDir[c1] != '\\') ) { --c1; } ++c1; --c2; while ( (c2 >= 0) && (m_DestDir[c2] != '\\') ) { --c2; } ++c2; } beginOfToken1 = c1; beginOfToken2 = c2; Token1 = m_SrcDir.Mid( beginOfToken1, endOfToken1-beginOfToken1 ); Token2 = m_DestDir.Mid( beginOfToken2, endOfToken2-beginOfToken2 ); //endExtract: /*if ( hasTokens() ) { text.Format( "The two branch tokens '%s' and '%s' are now stored", Token1, Token2 ); ::MessageBox( m_hWnd, text, "Tokens found", MB_OK | MB_ICONINFORMATION ); return; }*/ //notfound: //::MessageBox( m_hWnd, "Tokens not found in the directories", "Extracting tokens", MB_OK | MB_ICONEXCLAMATION ); } void CBranch_patcherDlg::loadConfiguration() { // Read the dest directory from the registry free( (void*)theApp.m_pszRegistryKey ); theApp.m_pszRegistryKey = _tcsdup( _T("Nevrax") ); CString savedSrcDir, savedTargetDir, token1, token2; if ( m_SrcDir.IsEmpty() ) { savedSrcDir = theApp.GetProfileString( _T(""), _T("SourceDir") ); if ( ! savedSrcDir.IsEmpty() ) { setSrcDirectory( savedSrcDir ); } } savedTargetDir = theApp.GetProfileString( _T(""), _T("TargetDir") ); if ( ! savedTargetDir.IsEmpty() ) { setDestDirectory( savedTargetDir ); } Token1 = theApp.GetProfileString( _T(""), _T("Token1") ); Token2 = theApp.GetProfileString( _T(""), _T("Token2") ); PatchExeDir = theApp.GetProfileString( _T(""), _T("PatchExeDir") ); CvsDiffDirLevel = theApp.GetProfileInt( _T(""), _T("CvsDiffDirLevel"), 1 ); // 0 for old version of CVS, 1 for new version of CVS } void CBranch_patcherDlg::saveConfiguration() { UpdateData( true ); if ( ! EnteringTokens ) { theApp.WriteProfileString( _T(""), _T("SourceDir"), m_SrcDir ); theApp.WriteProfileString( _T(""), _T("TargetDir"), m_DestDir ); } theApp.WriteProfileString( _T(""), _T("Token1"), Token1 ); theApp.WriteProfileString( _T(""), _T("Token2"), Token2 ); } void CBranch_patcherDlg::OnButtonExtractTokens() { if ( ! EnteringTokens ) { EnteringTokens = true; extractDirTokens(); SrcDirBackup = m_SrcDir; TargetDirBackup = m_DestDir; m_SrcDir = Token1; m_DestDir = Token2; m_SrcDirLabel = "Enter Token 1"; m_TargetDirLabel = "Enter Token 2"; m_Filename = "The tokens above were extracted from the directories."; ((CButton*)GetDlgItem( IDC_ButtonExtractTokens ))->SetWindowText( "Store Tokens" ); GetDlgItem( IDC_TopText )->ShowWindow( SW_HIDE ); GetDlgItem( IDC_ButtonClearTokens )->EnableWindow( FALSE ); GetDlgItem( IDC_ButtonPatch )->ShowWindow( SW_HIDE ); GetDlgItem( IDC_ButtonPatch )->EnableWindow( FALSE ); GetDlgItem( IDC_DoPatch )->ShowWindow( SW_HIDE ); GetDlgItem( IDC_Group )->ShowWindow( SW_HIDE ); UpdateData( false ); } else { UpdateData( true ); EnteringTokens = false; Token1 = m_SrcDir; Token2 = m_DestDir; m_SrcDirLabel = "Source Dir"; m_TargetDirLabel = "Target Dir"; m_SrcDir = SrcDirBackup; m_DestDir = TargetDirBackup; m_Filename.Empty(); ((CButton*)GetDlgItem( IDC_ButtonExtractTokens ))->SetWindowText( _T("Enter Tokens") ); GetDlgItem( IDC_TopText )->ShowWindow( SW_SHOW ); GetDlgItem( IDC_ButtonClearTokens )->EnableWindow( TRUE ); GetDlgItem( IDC_ButtonPatch )->ShowWindow( SW_SHOW ); GetDlgItem( IDC_ButtonPatch )->EnableWindow( TRUE ); GetDlgItem( IDC_DoPatch )->ShowWindow( SW_SHOW ); GetDlgItem( IDC_Group )->ShowWindow( SW_SHOW ); displayTokens(); } } void CBranch_patcherDlg::OnButtonClearTokens() { Token1.Empty(); Token2.Empty(); displayTokens(); } bool CBranch_patcherDlg::hasTokens() const { return ! (Token1.IsEmpty() || Token2.IsEmpty()); } void CBranch_patcherDlg::displayTokens() { ((CButton*)GetDlgItem( IDC_ButtonClearTokens ))->EnableWindow( hasTokens()?TRUE:FALSE ); if ( hasTokens() ) { m_Tokens = "Tokens: '" + Token1 + "' and '" + Token2 + "'"; } else { m_Tokens = "No token"; } UpdateData( false ); }