Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-08 10:44:09 -04:00
Code Issues Packages Projects Releases Wiki Activity

Read arbitrary strings from stdin

Browse Source
This commit is contained in:
Kovid Goyal 2016-05-03 20:50:25 +05:30
parent d8fe21d156
commit 770c076500
2 changed files with 35 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

38
setup/installer/windows/file_dialogs.cpp
View File

@ -14,7 +14,7 @@
#define REPORTERR(x) { PRINTERR(x); ret = 1; goto error; }
#define CALLCOM(x, err) hr = x; if(FAILED(hr)) REPORTERR(err)
int show_dialog(HWND parent, bool save_dialog) {
int show_dialog(HWND parent, bool save_dialog, LPWSTR title) {
int ret = 0;
IFileDialog *pfd = NULL;
IShellItem *psiResult = NULL;
@ -27,6 +27,7 @@ int show_dialog(HWND parent, bool save_dialog) {
CALLCOM(pfd->GetOptions(&dwFlags), "Failed to get options")
dwFlags |= FOS_FORCEFILESYSTEM;
CALLCOM(pfd->SetOptions(dwFlags), "Failed to set options")
if (title != NULL) { CALLCOM(pfd->SetTitle(title), "Failed to set title") }
hr = pfd->Show(parent);
if (hr == HRESULT_FROM_WIN32(ERROR_CANCELLED)) goto error;
if (FAILED(hr)) REPORTERR("Failed to show dialog")
@ -49,31 +50,60 @@ bool read_bytes(size_t sz, char* buf, bool allow_incomplete=false) {
return true;
}
bool from_utf8(size_t sz, const char *src, LPWSTR* ans) {
int asz = MultiByteToWideChar(CP_UTF8, MB_ERR_INVALID_CHARS, src, (int)sz, NULL, 0);
if (!asz) { PRINTERR("Failed to get size of UTF-8 string"); return false; }
*ans = (LPWSTR)calloc(asz+1, sizeof(wchar_t));
if(*ans == NULL) { PRINTERR("Out of memory!"); return false; }
asz = MultiByteToWideChar(CP_UTF8, MB_ERR_INVALID_CHARS, src, (int)sz, *ans, asz);
if (!asz) { PRINTERR("Failed to convert UTF-8 string"); return false; }
return true;
}
static char* rsbuf = NULL;
bool read_string(unsigned short sz, LPWSTR* ans) {
if(rsbuf == NULL) {
rsbuf = (char*)calloc(65537, sizeof(char));
if(rsbuf == NULL) { PRINTERR("Out of memory!"); return false; }
}
memset(rsbuf, 0, 65537);
if (!read_bytes(sz, rsbuf)) return false;
if (!from_utf8(sz, rsbuf, ans)) return false;
return true;
}
#define READ(x, y) if (!read_bytes((x), (y))) return 1;
#define CHECK_KEY(x) (key_size == sizeof(x) - 1 && memcmp(buf, x, sizeof(x) - 1) == 0)
#define READSTR(x) READ(sizeof(unsigned short), buf); if(!read_string(*((unsigned short*)buf), &x)) return 1;
int WINAPI wWinMain(HINSTANCE hInstance, HINSTANCE hPrevInstance, PWSTR pCmdLine, int nCmdShow) {
char buf[257];
size_t key_size = 0;
HWND parent = NULL;
bool save_dialog = false;
unsigned short len = 0;
LPWSTR title = NULL;
while(!feof(stdin)) {
memset(buf, 0, sizeof(buf));
if(!read_bytes(1, buf, true)) { if (feof(stdin)) break; return 1;}
key_size = (size_t)buf[0];
READ(key_size, buf);
if (key_size == 4 && memcmp(buf, "HWND", 4) == 0) {
if CHECK_KEY("HWND") {
READ(sizeof(HWND), buf);
if (sizeof(HWND) == 8) parent = (HWND)*((__int64*)buf);
else if (sizeof(HWND) == 4) parent = (HWND)*((__int32*)buf);
else { fprintf(stderr, "Unknown pointer size: %d", sizeof(HWND)); return 1;}
else { fprintf(stderr, "Unknown pointer size: %d", sizeof(HWND)); fflush(stderr); return 1;}
}
else if CHECK_KEY("TITLE") { READSTR(title) }
else {
PRINTERR("Unknown key");
return 1;
}
}
return show_dialog(parent, save_dialog);
return show_dialog(parent, save_dialog, title);
}

2
src/calibre/gui2/win_file_dialogs.py
View File

@ -81,7 +81,7 @@ if __name__ == '__main__':
q = QMainWindow()
def clicked():
print(run_file_dialog(b)), sys.stdout.flush()
print(run_file_dialog(b, 'Testing dialogs')), sys.stdout.flush()
b = QPushButton('click me')
b.clicked.connect(clicked)