contrib/issue-tracker-plugins/ExampleAtlPlugin/Provider.cpp

   1 #include "stdafx.h"
   2 #include "Provider.h"
   3 #include "WaitDialog.h"
   4
   5 CProvider::CProvider()
   6 {
   7 }
   8
   9 HRESULT CProvider::FinalConstruct()
  10 {
  11         return S_OK;
  12 }
  13
  14 void CProvider::FinalRelease()
  15 {
  16 }
  17
  18 CProvider::parameters_t CProvider::ParseParameters(BSTR parameters) const
  19 {
  20         CString temp(parameters);
  21
  22         parameters_t result;
  23
  24         // TODO: Handle quoting and stuff
  25         int pos = 0;
  26         CString token = temp.Tokenize(_T(";"), pos);
  27         while (token != _T(""))
  28         {
  29                 int x = token.Find('=');
  30                 CString name = token.Left(x);
  31                 CString value = token.Mid(x + 1);
  32
  33                 result[name] = value;
  34
  35                 token = temp.Tokenize(_T(";"), pos);
  36         }
  37
  38         return result;
  39 }
  40
  41 HRESULT STDMETHODCALLTYPE CProvider::ValidateParameters(
  42     /* [in] */ HWND hParentWnd,
  43     /* [in] */ BSTR parameters,
  44     /* [retval][out] */ VARIANT_BOOL *valid)
  45 {
  46         // Don't bother validating the parameters yet.
  47         *valid = VARIANT_TRUE;
  48         return S_OK;
  49 }
  50
  51 HRESULT STDMETHODCALLTYPE CProvider::GetLinkText(
  52     /* [in] */ HWND hParentWnd,
  53     /* [in] */ BSTR parameters,
  54     /* [retval][out] */ BSTR *linkText)
  55 {
  56         parameters_t params = ParseParameters(parameters);
  57         CString prompt = params[CString("Prompt")];
  58         if (prompt.IsEmpty())
  59                 *linkText = SysAllocString(L"Choose Task...");
  60         else
  61                 *linkText = prompt.AllocSysString();
  62
  63         return S_OK;
  64 }
  65
  66 HRESULT STDMETHODCALLTYPE CProvider::GetCommitMessage(
  67     /* [in] */ HWND hParentWnd,
  68     /* [in] */ BSTR parameters,
  69     /* [in] */ BSTR commonRoot,
  70     /* [in] */ SAFEARRAY * pathList,
  71     /* [in] */ BSTR originalMessage,
  72     /* [retval][out] */ BSTR *newMessage)
  73 {
  74         return GetCommitMessage2(hParentWnd, parameters, NULL, commonRoot, pathList, originalMessage, newMessage);
  75 }
  76
  77 HRESULT STDMETHODCALLTYPE CProvider::GetCommitMessage2(
  78         /* [in] */ HWND hParentWnd,
  79         /* [in] */ BSTR parameters,
  80         /* [in] */ BSTR commonURL,
  81         /* [in] */ BSTR commonRoot,
  82         /* [in] */ SAFEARRAY * pathList,
  83         /* [in] */ BSTR originalMessage,
  84         /* [retval][out] */ BSTR *newMessage)
  85 {
  86         USES_CONVERSION;
  87
  88         if (commonURL)
  89         {
  90                 // do something with the common root url
  91                 // if necessary
  92         }
  93
  94         parameters_t params = ParseParameters(parameters);
  95         CString commandLine = params[CString("CommandLine")];
  96
  97         if (commandLine.IsEmpty())
  98         {
  99                 MessageBox(hParentWnd, _T("CommandLine parameter is empty"), _T("ExampleAtlPlugin"), MB_OK | MB_ICONERROR);
 100                 return S_OK;
 101         }
 102
 103         TCHAR szTempPath[MAX_PATH];
 104         GetTempPath(ARRAYSIZE(szTempPath), szTempPath);
 105
 106         // Create a temporary file to hold the path list, and write the list to the file.
 107         TCHAR szPathListTempFile[MAX_PATH];
 108         GetTempFileName(szTempPath, _T("svn"), 0, szPathListTempFile);
 109
 110         DWORD bytesWritten;
 111         HANDLE hPathListFile = CreateFile(szPathListTempFile, GENERIC_READ | GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, NULL);
 112
 113         LONG a, b;
 114         if (FAILED(SafeArrayGetLBound(pathList, 1, &a)) || (FAILED(SafeArrayGetUBound(pathList, 1, &b))))
 115                 return E_FAIL;
 116
 117         for (LONG i = a; i <= b; ++i)
 118         {
 119                 BSTR path = NULL;
 120                 SafeArrayGetElement(pathList, &i, &path);
 121
 122                 CStringA line = OLE2A(path);
 123                 line += "\r\n";
 124
 125                 WriteFile(hPathListFile, line, line.GetLength(), &bytesWritten, NULL);
 126         }
 127
 128         CloseHandle(hPathListFile);
 129
 130         TCHAR szOriginalMessageTempFile[MAX_PATH];
 131         GetTempFileName(szTempPath, _T("svn"), 0, szOriginalMessageTempFile);
 132
 133         HANDLE hOriginalMessageFile = CreateFile(szOriginalMessageTempFile, GENERIC_READ | GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, NULL);
 134
 135         CStringA temp = OLE2A(originalMessage);
 136         WriteFile(hOriginalMessageFile, temp, temp.GetLength(), &bytesWritten, NULL);
 137
 138         CloseHandle(hOriginalMessageFile);
 139
 140         commandLine.AppendFormat(_T(" \"%s\" \"%ls\" \"%s\""), szPathListTempFile, commonRoot, szOriginalMessageTempFile);
 141
 142         CString message = originalMessage;
 143         CWaitDialog dlg(commandLine);
 144         if (dlg.DoModal() == IDOK)
 145         {
 146                 HANDLE hOrig = CreateFile(szOriginalMessageTempFile, GENERIC_READ | GENERIC_WRITE, 0, NULL, OPEN_EXISTING, 0, NULL);
 147                 DWORD cb = GetFileSize(hOrig, NULL);
 148                 BYTE *buffer = (BYTE *)malloc(cb + 1);
 149                 memset(buffer, 0, cb + 1);
 150                 DWORD bytesRead;
 151                 ReadFile(hOrig, buffer, cb, &bytesRead, NULL);
 152                 CloseHandle(hOrig);
 153
 154                 message = A2T((const char *)buffer);
 155         }
 156
 157         DeleteFile(szPathListTempFile);
 158         DeleteFile(szOriginalMessageTempFile);
 159
 160         *newMessage = message.AllocSysString();
 161         return S_OK;
 162 }
 163
 164 HRESULT STDMETHODCALLTYPE CProvider::OnCommitFinished (
 165         /* [in] */ HWND hParentWnd,
 166         /* [in] */ BSTR commonRoot,
 167         /* [in] */ SAFEARRAY * pathList,
 168         /* [in] */ BSTR logMessage,
 169         /* [in] */ ULONG revision,
 170         /* [out, retval] */ BSTR * error)
 171 {
 172         CString err = _T("Test error string");
 173         *error = err.AllocSysString();
 174         return S_OK;
 175 }
 176
 177 HRESULT STDMETHODCALLTYPE CProvider::HasOptions(
 178                                    /* [out, retval] */ VARIANT_BOOL *ret)
 179 {
 180         MessageBox(NULL, _T("test"), _T("test"), MB_ICONERROR);
 181         *ret = VARIANT_FALSE;
 182         MessageBox(NULL, _T("test2"), _T("test2"), MB_ICONERROR);
 183         return S_OK;
 184 }
 185
 186 // this method is called if HasOptions() returned true before.
 187 // Use this to show a custom dialog so the user doesn't have to
 188 // create the parameters string manually
 189 HRESULT STDMETHODCALLTYPE CProvider::ShowOptionsDialog(
 190                                                   /* [in] */ HWND hParentWnd,
 191                                                   /* [in] */ BSTR parameters,
 192                                                   /* [out, retval] */ BSTR * newparameters
 193                                                   )
 194 {
 195         // we don't show an options dialog
 196         return E_NOTIMPL;
 197 }