diff options
Diffstat (limited to 'plugins/PasteIt/PasteToWeb.cpp')
| -rw-r--r-- | plugins/PasteIt/PasteToWeb.cpp | 577 |
1 files changed, 577 insertions, 0 deletions
diff --git a/plugins/PasteIt/PasteToWeb.cpp b/plugins/PasteIt/PasteToWeb.cpp new file mode 100644 index 0000000000..a7cc9a3b5f --- /dev/null +++ b/plugins/PasteIt/PasteToWeb.cpp @@ -0,0 +1,577 @@ +/*
+Paste It plugin
+Copyright (C) 2011 Krzysztof Kral
+
+This program is free software; you can redistribute it and/or
+modify it under the terms of the GNU General Public License
+as published by the Free Software Foundation version 2
+of the License.
+
+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 General Public License for more details.
+
+You should have received a copy of the GNU General Public License
+along with this program. If not, see <http://www.gnu.org/licenses/>.
+*/
+
+#include "StdAfx.h"
+#include "PasteToWeb.h"
+#include "Options.h"
+#include "resource.h"
+
+extern HINSTANCE hInst;
+
+PasteToWeb::PasteToWeb()
+{
+}
+
+
+PasteToWeb::~PasteToWeb()
+{
+}
+
+struct FromClipboardData
+{
+ std::wstring content;
+ int page;
+ std::wstring format;
+};
+
+struct FromFileData
+{
+ char* content;
+ int contentLen;
+ wchar_t* contentW;
+ UINT codepage;
+ std::wstring *fileName;
+ int page;
+ std::wstring format;
+ bool sendFileName;
+};
+
+INT_PTR CALLBACK DlgProcFromClipboard(HWND hwndDlg, UINT msg, WPARAM wParam, LPARAM lParam)
+{
+ switch (msg)
+ {
+ case WM_INITDIALOG:
+ {
+ TranslateDialogDefault(hwndDlg);
+ FromClipboardData* data = (FromClipboardData*)lParam;
+ SetWindowLongPtr(hwndDlg, GWLP_USERDATA, lParam);
+ int ts = 4;
+ Edit_SetTabStops(GetDlgItem(hwndDlg, IDC_CLIPBOARD_DATA), 1, &ts);
+ SetDlgItemText(hwndDlg, IDC_CLIPBOARD_DATA, data->content.c_str());
+ int sel = 0;
+ int i = 0;
+ std::wstring &defFormat = Options::instance->webOptions[data->page]->defFormatId;
+ HWND cb = GetDlgItem(hwndDlg, IDC_FORMAT);
+ for(std::list<PasteFormat>::iterator it = Options::instance->webOptions[data->page]->formats.begin(); it != Options::instance->webOptions[data->page]->formats.end(); ++it)
+ {
+ ComboBox_AddString(cb, it->name.c_str());
+ if(it->id == defFormat)
+ sel = i;
+ ++i;
+ }
+ if(!Options::instance->webOptions[data->page]->formats.empty())
+ {
+ ComboBox_SetCurSel(cb, sel);
+ }
+ return TRUE;
+ }
+ case WM_COMMAND:
+ {
+ if (HIWORD(wParam) == BN_CLICKED)
+ {
+ if (LOWORD(wParam) == IDOK)
+ {
+ FromClipboardData *clipboardData = (FromClipboardData*)GetWindowLongPtr(hwndDlg, GWLP_USERDATA);
+ int sel = ComboBox_GetCurSel(GetDlgItem(hwndDlg, IDC_FORMAT));
+ for(std::list<PasteFormat>::iterator it = Options::instance->webOptions[clipboardData->page]->formats.begin(); it != Options::instance->webOptions[clipboardData->page]->formats.end(); ++it)
+ {
+ if(sel-- <= 0)
+ {
+ clipboardData->format = it->id;
+ break;
+ }
+ }
+
+ EndDialog(hwndDlg, IDC_BTN_OK);
+ }
+ else if (LOWORD(wParam) == IDCANCEL)
+ {
+ EndDialog(hwndDlg, IDCANCEL);
+ }
+ }
+
+ break;
+ }
+ }
+
+ return FALSE;
+}
+
+void RecodeDlg(HWND hwndDlg)
+{
+ ShowWindow(GetDlgItem(hwndDlg,IDC_RECODE),SW_HIDE);
+ FromFileData *fromFileData = (FromFileData*)GetWindowLongPtr(hwndDlg, GWLP_USERDATA);
+ unsigned int cp = Options::GetCodepageCB(GetDlgItem(hwndDlg, IDC_CODEPAGE), false, fromFileData->codepage);
+ mir_free(fromFileData->contentW);
+ int cbLen = 0;
+ if(cp == 1200 || cp == 1201)
+ {
+ // UTF-16
+ cbLen = fromFileData->contentLen / 2;
+ }
+ else
+ {
+ cbLen = MultiByteToWideChar( cp, 0, fromFileData->content, fromFileData->contentLen, NULL, 0 );
+ }
+
+ fromFileData->contentW = ( wchar_t* )mir_alloc( sizeof( wchar_t )*(cbLen+1));
+ if ( fromFileData->contentW != NULL )
+ {
+ if(cp == 1200)
+ {
+ memcpy_s(fromFileData->contentW, sizeof( wchar_t )*(cbLen+1), fromFileData->content, sizeof( wchar_t )*cbLen);
+ }
+ else if(cp == 1201)
+ {
+ for(int i = 0 ; i < cbLen; ++i)
+ {
+ fromFileData->contentW[i] = ((unsigned char)fromFileData->content[i*2] << 8) | (unsigned char)fromFileData->content[i*2 + 1];
+ }
+ }
+ else
+ {
+ MultiByteToWideChar( cp, 0, fromFileData->content, fromFileData->contentLen, fromFileData->contentW, cbLen );
+ }
+
+ fromFileData->contentW[ cbLen ] = 0;
+ SetDlgItemText(hwndDlg, IDC_FILE_DATA, fromFileData->contentW);
+ }
+ else
+ {
+ SetDlgItemText(hwndDlg, IDC_FILE_DATA, _T(""));
+ }
+
+ fromFileData->codepage = cp;
+}
+
+INT_PTR CALLBACK DlgProcFromFile(HWND hwndDlg, UINT msg, WPARAM wParam, LPARAM lParam)
+{
+ switch (msg)
+ {
+ case WM_INITDIALOG:
+ {
+ TranslateDialogDefault(hwndDlg);
+ FromFileData *fromFileData = (FromFileData*)lParam;
+ int ts = 4;
+ Edit_SetTabStops(GetDlgItem(hwndDlg, IDC_FILE_DATA), 1, &ts);
+ SetDlgItemText(hwndDlg, IDC_FILE_DATA, fromFileData->contentW);
+ SetDlgItemText(hwndDlg, IDC_FILE_PATH, fromFileData->fileName->c_str());
+ Options::InitCodepageCB(GetDlgItem(hwndDlg, IDC_CODEPAGE), fromFileData->codepage);
+ SetWindowLongPtr(hwndDlg, GWLP_USERDATA, lParam);
+ ShowWindow(GetDlgItem(hwndDlg,IDC_RECODE),SW_HIDE);
+
+ int sel = 0;
+ int i = 0;
+ std::wstring &defFormat = Options::instance->webOptions[fromFileData->page]->defFormatId;
+ HWND cb = GetDlgItem(hwndDlg, IDC_FORMAT);
+ for(std::list<PasteFormat>::iterator it = Options::instance->webOptions[fromFileData->page]->formats.begin(); it != Options::instance->webOptions[fromFileData->page]->formats.end(); ++it)
+ {
+ ComboBox_AddString(cb, it->name.c_str());
+ if(it->id == defFormat)
+ sel = i;
+ ++i;
+ }
+ if(!Options::instance->webOptions[fromFileData->page]->formats.empty())
+ {
+ ComboBox_SetCurSel(cb, sel);
+ }
+
+ if(Options::instance->webOptions[fromFileData->page]->isSendFileName)
+ {
+ bool sendFileName = Options::instance->webOptions[fromFileData->page]->sendFileName;
+ CheckDlgButton(hwndDlg, IDC_AUTOFORMAT, sendFileName ? 1 : 0);
+ Button_Enable(GetDlgItem(hwndDlg, IDC_FORMAT), sendFileName ? 0 : 1);
+ Button_Enable(GetDlgItem(hwndDlg, IDC_FORMATTEXT), sendFileName ? 0 : 1);
+ }
+ else
+ Button_Enable(GetDlgItem(hwndDlg, IDC_AUTOFORMAT), FALSE);
+ return TRUE;
+ }
+ case WM_COMMAND:
+ {
+ if (HIWORD(wParam) == BN_CLICKED)
+ {
+ if (LOWORD(wParam) == IDOK)
+ {
+ FromFileData *fromFileData = (FromFileData*)GetWindowLongPtr(hwndDlg, GWLP_USERDATA);
+ int sel = ComboBox_GetCurSel(GetDlgItem(hwndDlg, IDC_FORMAT));
+ for(std::list<PasteFormat>::iterator it = Options::instance->webOptions[fromFileData->page]->formats.begin(); it != Options::instance->webOptions[fromFileData->page]->formats.end(); ++it)
+ {
+ if(sel-- <= 0)
+ {
+ fromFileData->format = it->id;
+ break;
+ }
+ }
+
+ if(Options::instance->webOptions[fromFileData->page]->isSendFileName)
+ fromFileData->sendFileName = IsDlgButtonChecked(hwndDlg, IDC_AUTOFORMAT) ? true : false;
+ else
+ fromFileData->sendFileName = false;
+
+ EndDialog(hwndDlg, IDC_BTN_OK);
+ }
+ else if (LOWORD(wParam) == IDCANCEL)
+ {
+ EndDialog(hwndDlg, IDCANCEL);
+ }
+ else if(LOWORD(wParam) == IDC_RECODE)
+ {
+ RecodeDlg(hwndDlg);
+ }
+ else if(LOWORD(wParam) == IDC_AUTOFORMAT)
+ {
+ UINT sendFileName = IsDlgButtonChecked(hwndDlg, IDC_AUTOFORMAT);
+ Button_Enable(GetDlgItem(hwndDlg, IDC_FORMAT), sendFileName ? 0 : 1);
+ Button_Enable(GetDlgItem(hwndDlg, IDC_FORMATTEXT), sendFileName ? 0 : 1);
+ }
+ }
+ else if(LOWORD(wParam) == IDC_CODEPAGE)
+ {
+ if(HIWORD(wParam) == CBN_SELCHANGE)
+ {
+ RecodeDlg(hwndDlg);
+ }
+ else if(HIWORD(wParam) == CBN_EDITCHANGE)
+ {
+ ShowWindow(GetDlgItem(hwndDlg,IDC_RECODE),SW_SHOW);
+ }
+ else if(HIWORD(wParam) == CBN_KILLFOCUS)
+ {
+ FromFileData *fromFileData = (FromFileData*)GetWindowLongPtr(hwndDlg, GWLP_USERDATA);
+ Options::GetCodepageCB(GetDlgItem(hwndDlg, IDC_CODEPAGE), true, fromFileData->codepage);
+ }
+ }
+
+ break;
+ }
+ }
+
+ return FALSE;
+}
+
+void PasteToWeb::FromClipboard()
+{
+ szFileLink[0] = 0;
+ error = NULL;
+ std::wstring str;
+ BOOL isFile = 0;
+ if(OpenClipboard(NULL))
+ {
+ HANDLE obj = GetClipboardData(CF_UNICODETEXT);
+ if(obj != NULL)
+ {
+ LPCWSTR wStr = (LPCWSTR)GlobalLock(obj);
+ str.append(wStr, wStr + wcslen(wStr));
+ GlobalUnlock(obj);
+ // Sometimes clipboard CF_UNICODETEXT format returns only 2 characters,
+ // to fix this I check if CF_TEXT contains more characters,
+ // if this is true, this mean that CF_UNICODETEXT is invalid.
+ obj = GetClipboardData(CF_TEXT);
+ if(obj != NULL)
+ {
+ LPCSTR cStr = (LPCSTR)GlobalLock(obj);
+ if(strlen(cStr) > str.length())
+ {
+ str = L"";
+ LPWSTR wStr = mir_a2u_cp(cStr, CP_ACP);
+ str.append(wStr, wStr + wcslen(wStr));
+ mir_free(wStr);
+ }
+ GlobalUnlock(obj);
+ }
+ }
+ else
+ {
+ obj = GetClipboardData(CF_TEXT);
+ if(obj != NULL)
+ {
+ LPCSTR cStr = (LPCSTR)GlobalLock(obj);
+ LPWSTR wStr = mir_a2u_cp(cStr, CP_ACP);
+ str.append(wStr, wStr + wcslen(wStr));
+ mir_free(wStr);
+ GlobalUnlock(obj);
+ }
+ else
+ {
+ obj = GetClipboardData(CF_HDROP);
+ if(obj != NULL)
+ {
+ LPDROPFILES df = (LPDROPFILES) GlobalLock(obj);
+ isFile = 1;
+ if(df->fWide)
+ {
+ // Unicode
+ WCHAR* file = (WCHAR*)((BYTE*)obj + df->pFiles);
+ size_t len = wcslen(file);
+ if(*(file + len + 1) == L'\0')
+ {
+ str.append(file, file + len);
+ }
+ else
+ {
+ error = TranslateT("You can only paste 1 file");
+ }
+ }
+ else
+ {
+ // ANSI
+ char* file = (char*)obj + df->pFiles;
+ size_t len = strlen(file);
+ if(*(file + len + 1) == '\0')
+ {
+ LPWSTR wStr = mir_a2u_cp(file, CP_ACP);
+ str.append(wStr, wStr + wcslen(wStr));
+ mir_free(wStr);
+ }
+ else
+ {
+ error = TranslateT("You can only paste 1 file");
+ }
+ }
+ GlobalUnlock(obj);
+ }
+ }
+ }
+
+ CloseClipboard();
+ }
+
+ if(str.length() > 0)
+ {
+ if(isFile)
+ FromFile(str);
+ else
+ {
+ FromClipboardData data;
+ data.content = str;
+ data.page = pageIndex;
+ if (Options::instance->confDlg && DialogBoxParam (hInst, MAKEINTRESOURCE(IDD_DLG_FROM_CLIPBOARD), 0, DlgProcFromClipboard, (LPARAM)&data) != IDC_BTN_OK)
+ return;
+
+ SendToServer(str, L"", data.format);
+ }
+ }
+ else if(error == NULL)
+ {
+ error = TranslateT("Cannot get data from clipboard");
+ }
+}
+
+void PasteToWeb::FromFile(std::wstring file)
+{
+ error = NULL;
+ szFileLink[0] = 0;
+ HANDLE hFile = CreateFile(file.c_str(), GENERIC_READ, FILE_SHARE_READ, NULL, OPEN_EXISTING, 0, NULL);
+ FromFileData fromFileData;
+ fromFileData.content = NULL;
+ fromFileData.contentLen = 0;
+ fromFileData.fileName = &file;
+ if(hFile != INVALID_HANDLE_VALUE)
+ {
+ LARGE_INTEGER fileSize;
+ if(GetFileSizeEx(hFile, &fileSize))
+ {
+ if(fileSize.QuadPart <= 10485760LL)
+ {
+ if(fileSize.QuadPart > 512000LL)
+ {
+ _stprintf_s(bufErr, 1024, TranslateT("File size is %dKB, do you realy want to paste such large file?"), fileSize.LowPart / 1024);
+ if(MessageBox(NULL, bufErr, TranslateT("Are You sure?"), MB_YESNO | MB_ICONQUESTION) != IDYES)
+ {
+ CloseHandle(hFile);
+ return;
+ }
+ }
+ DWORD readed;
+ fromFileData.contentLen = fileSize.LowPart;
+ fromFileData.content = (char*)mir_alloc(fromFileData.contentLen);
+ if(!ReadFile(hFile, fromFileData.content, fromFileData.contentLen, &readed, NULL))
+ {
+ mir_free(fromFileData.content);
+ fromFileData.content = NULL;
+ fromFileData.contentLen = 0;
+ _stprintf_s(bufErr, 1024, TranslateT("Cannot read file '%s'"), file.c_str());
+ error = bufErr;
+ }
+ }
+ else
+ {
+ error = TranslateT("File size is larger then 10MB, cannot be send");
+ }
+ }
+
+ CloseHandle(hFile);
+ }
+ else
+ {
+ _stprintf_s(bufErr, 1024, TranslateT("Cannot open file '%s'"), file.c_str());
+ error = bufErr;
+ }
+
+ if(fromFileData.content != NULL)
+ {
+ int cbLen = 0;
+ bool isDefTranslation = true;
+ if(Options::instance->autoUTF)
+ {
+ isDefTranslation = false;
+ fromFileData.codepage = CP_UTF8;
+ cbLen = MultiByteToWideChar( fromFileData.codepage, MB_ERR_INVALID_CHARS, fromFileData.content, fromFileData.contentLen, NULL, 0 );
+ if(cbLen == 0)
+ {
+ int errorN = GetLastError();
+ if(errorN == ERROR_NO_UNICODE_TRANSLATION)
+ {
+ isDefTranslation = true;
+ }
+ }
+ }
+
+ if(isDefTranslation)
+ {
+ fromFileData.codepage = Options::instance->codepage;
+ if(fromFileData.codepage == 1200 || fromFileData.codepage == 1201)
+ {
+ // UTF-16
+ cbLen = fromFileData.contentLen / 2;
+ }
+ else
+ {
+ cbLen = MultiByteToWideChar( fromFileData.codepage, 0, fromFileData.content, fromFileData.contentLen, NULL, 0 );
+ }
+ }
+
+ if(cbLen > 0)
+ {
+ fromFileData.contentW = ( wchar_t* )mir_alloc( sizeof( wchar_t )*(cbLen+1));
+ if ( fromFileData.contentW != NULL )
+ {
+ if(fromFileData.codepage == 1200)
+ {
+ memcpy_s(fromFileData.contentW, sizeof( wchar_t )*(cbLen+1), fromFileData.content, sizeof( wchar_t )*cbLen);
+ }
+ else if(fromFileData.codepage == 1201)
+ {
+ for(int i = 0 ; i < cbLen; ++i)
+ {
+ fromFileData.contentW[i] = (fromFileData.content[i*2] << 8) | fromFileData.content[i*2 + 1];
+ }
+ }
+ else
+ {
+ MultiByteToWideChar( fromFileData.codepage, 0, fromFileData.content, fromFileData.contentLen, fromFileData.contentW, cbLen );
+ }
+
+ fromFileData.contentW[ cbLen ] = 0;
+ fromFileData.page = pageIndex;
+ if (!Options::instance->confDlg || DialogBoxParam (hInst, MAKEINTRESOURCE(IDD_DLG_FROM_FILE), 0, DlgProcFromFile, (LPARAM)&fromFileData) == IDC_BTN_OK)
+ {
+ std::wstring fileName;
+ std::wstring::size_type pos1 = file.find_last_of(L'\\');
+ std::wstring::size_type pos2 = file.find_last_of(L'/');
+ if(pos2 > pos1 && pos2 < file.length())
+ pos1 = pos2;
+ if(pos1 >= 0 && pos1 + 1 < file.length())
+ fileName = file.substr(pos1 + 1);
+ else
+ fileName = file;
+ SendToServer(fromFileData.contentW, fromFileData.sendFileName ? fileName : L"", fromFileData.format);
+ }
+ mir_free(fromFileData.contentW);
+ }
+ }
+ else
+ {
+ _stprintf_s(bufErr, 1024, TranslateT("File '%s' is empty"), file.c_str());
+ error = bufErr;
+ }
+ mir_free(fromFileData.content);
+ }
+}
+
+extern HANDLE g_hNetlibUser;
+
+wchar_t* PasteToWeb::SendToWeb(char* url, std::map<std::string, std::string>& headers, std::wstring content)
+{
+ wchar_t* resCont = NULL;
+ int cbLen = WideCharToMultiByte( CP_UTF8, 0, content.c_str(), -1, NULL, 0, NULL, NULL );
+ char* contentBytes = ( char* )mir_alloc(cbLen);
+ if ( contentBytes == NULL )
+ return resCont;
+
+ WideCharToMultiByte( CP_UTF8, 0, content.c_str(), -1, contentBytes, cbLen, NULL, NULL );
+ --cbLen;
+
+ int nHeaders = 0;
+ for(std::map<std::string, std::string>::iterator it = headers.begin(); it != headers.end(); ++it)
+ {
+ ++nHeaders;
+ }
+
+ NETLIBHTTPREQUEST nlhr={0};
+ NETLIBHTTPHEADER* httpHeaders = new NETLIBHTTPHEADER[nHeaders];
+ nlhr.cbSize=sizeof(nlhr);
+ nlhr.requestType=REQUEST_POST;
+ nlhr.flags=NLHRF_NODUMPSEND|NLHRF_DUMPASTEXT|NLHPIF_HTTP11;
+ nlhr.szUrl=url;
+ nlhr.headers = httpHeaders;
+ nlhr.pData = contentBytes;
+ nlhr.dataLength = cbLen;
+ nHeaders = 0;
+ std::list<char*> mallBuf;
+ for(std::map<std::string, std::string>::iterator it = headers.begin(); it != headers.end(); ++it)
+ {
+ char* b1 = new char[it->first.length() + 1];
+ char* b2 = new char[it->second.length() + 1];
+ strcpy_s(b1, it->first.length() + 1, it->first.c_str());
+ strcpy_s(b2, it->second.length() + 1, it->second.c_str());
+ httpHeaders[nHeaders].szName = b1;
+ httpHeaders[nHeaders].szValue = b2;
+ mallBuf.push_back(b1);
+ mallBuf.push_back(b2);
+ ++nHeaders;
+ }
+
+ nlhr.headersCount = nHeaders;
+ NETLIBHTTPREQUEST* nlhrReply = (NETLIBHTTPREQUEST*)CallService(MS_NETLIB_HTTPTRANSACTION,(WPARAM)g_hNetlibUser,(LPARAM)&nlhr);
+ if(nlhrReply != NULL)
+ {
+ if(nlhrReply->resultCode == 200)
+ {
+ int resLen = MultiByteToWideChar(CP_UTF8, 0, nlhrReply->pData, nlhrReply->dataLength, NULL, 0);
+ ++resLen;
+ resCont = ( wchar_t* )mir_alloc(resLen * sizeof(wchar_t));
+ if ( resCont != NULL )
+ {
+ resLen = MultiByteToWideChar( CP_UTF8, 0, nlhrReply->pData, nlhrReply->dataLength, resCont, resLen);
+ resCont[resLen] = 0;
+ }
+ }
+
+ CallService(MS_NETLIB_FREEHTTPREQUESTSTRUCT,0,(LPARAM)nlhrReply);
+ }
+ delete httpHeaders;
+ for(std::list<char*>::iterator it = mallBuf.begin(); it != mallBuf.end(); ++it)
+ {
+ delete *it;
+ }
+
+ mir_free(contentBytes);
+ return resCont;
+}
|