v/vlib/clipboard/clipboard_windows.v

163 lines
4.4 KiB
V
Raw Normal View History

module clipboard
import time
2020-04-04 15:37:13 +02:00
struct WndClassEx {
cbSize u32
style u32
lpfnWndProc voidptr
2020-04-04 15:37:13 +02:00
cbClsExtra int
cbWndExtra int
hInstance C.HINSTANCE
hIcon C.HICON
hCursor C.HCURSOR
hbrBackground C.HBRUSH
lpszMenuName &u16 // LPCWSTR
lpszClassName &u16
2020-04-04 15:37:13 +02:00
hIconSm &u16
}
2019-11-24 11:22:57 +01:00
2020-04-04 15:37:13 +02:00
fn C.RegisterClassEx(class WndClassEx) int
fn C.GetClipboardOwner() &C.HWND
fn C.CreateWindowEx(dwExStyle i64, lpClassName &u16, lpWindowName &u16, dwStyle i64, x int, y int, nWidth int, nHeight int, hWndParent i64, hMenu voidptr, hInstance voidptr, lpParam voidptr) &C.HWND
2019-11-24 13:29:11 +01:00
//fn C.MultiByteToWideChar(CodePage u32, dwFlags u16, lpMultiByteStr byteptr, cbMultiByte int, lpWideCharStr u16, cchWideChar int) int
fn C.EmptyClipboard()
fn C.CloseClipboard()
2020-04-04 15:37:13 +02:00
fn C.GlobalAlloc(uFlag u32, size i64) C.HGLOBAL
fn C.GlobalFree(buf C.HGLOBAL)
fn C.GlobalLock(buf C.HGLOBAL)
fn C.GlobalUnlock(buf C.HGLOBAL)
fn C.SetClipboardData(uFormat u32, data voidptr) C.HANDLE
fn C.GetClipboardData(uFormat u32) C.HANDLE
2020-04-04 15:37:13 +02:00
fn C.DefWindowProc(hwnd C.HWND, msg u32, wParam C.WPARAM, lParam C.LPARAM) C.LRESULT
fn C.SetLastError(error i64)
2020-04-04 15:37:13 +02:00
fn C.OpenClipboard(hwnd C.HWND) int
fn C.DestroyWindow(hwnd C.HWND)
2020-04-04 15:37:13 +02:00
struct Clipboard {
max_retries int
retry_delay int
mut:
2020-04-04 15:37:13 +02:00
hwnd C.HWND
2019-12-07 13:51:00 +01:00
foo int // TODO remove
}
fn (cb &Clipboard) get_clipboard_lock() bool {
mut retries := cb.max_retries
mut last_error := u32(0)
for {
retries--
if retries < 0 {
break
}
2020-04-04 15:37:13 +02:00
last_error = C.GetLastError()
if C.OpenClipboard(cb.hwnd) > 0 {
return true
} else if last_error != u32(C.ERROR_ACCESS_DENIED) {
return false
}
time.sleep(cb.retry_delay)
}
2020-04-04 15:37:13 +02:00
C.SetLastError(last_error)
return false
}
fn new_clipboard() &Clipboard {
mut cb := &Clipboard {
max_retries: 5
retry_delay: 5
}
2020-04-04 15:37:13 +02:00
class_name := "clipboard"
wndclass := WndClassEx {
cbSize: sizeof(WndClassEx)
lpfnWndProc: voidptr(&C.DefWindowProc)
lpszClassName: class_name.to_wide()
lpszMenuName: 0
hIconSm: 0
}
2020-04-04 15:37:13 +02:00
if C.RegisterClassEx(&wndclass) == 0 && C.GetLastError() != u32(C.ERROR_CLASS_ALREADY_EXISTS) {
println("Failed registering class.")
}
2020-04-04 15:37:13 +02:00
hwnd := C.CreateWindowEx(0, wndclass.lpszClassName, wndclass.lpszClassName, 0, 0, 0, 0, 0, C.HWND_MESSAGE, C.NULL, C.NULL, C.NULL)
if hwnd == C.NULL {
println("Error creating window!")
}
cb.hwnd = hwnd
return cb
}
fn (cb &Clipboard) check_availability() bool {
2020-04-04 15:37:13 +02:00
return cb.hwnd != C.HWND(C.NULL)
}
fn (cb &Clipboard) has_ownership() bool {
2020-04-04 15:37:13 +02:00
return C.GetClipboardOwner() == cb.hwnd
}
2019-12-07 13:51:00 +01:00
fn (cb mut Clipboard) clear() {
if !cb.get_clipboard_lock() {return}
2020-04-04 15:37:13 +02:00
C.EmptyClipboard()
C.CloseClipboard()
2019-12-07 13:51:00 +01:00
cb.foo = 0
}
2019-12-07 13:51:00 +01:00
fn (cb mut Clipboard) free(){
2020-04-04 15:37:13 +02:00
C.DestroyWindow(cb.hwnd)
2019-12-07 13:51:00 +01:00
cb.foo = 0
}
// the string.to_wide doesn't work with SetClipboardData, don't know why
2020-04-04 15:37:13 +02:00
fn to_wide(text string) &C.HGLOBAL {
len_required := C.MultiByteToWideChar(C.CP_UTF8, C.MB_ERR_INVALID_CHARS, text.str, text.len + 1, C.NULL, 0)
buf := C.GlobalAlloc(C.GMEM_MOVEABLE, sizeof(u16) * len_required)
if buf != C.HGLOBAL(C.NULL) {
mut locked := &u16(C.GlobalLock(buf))
C.MultiByteToWideChar(C.CP_UTF8, C.MB_ERR_INVALID_CHARS, text.str, text.len + 1, locked, len_required)
locked[len_required - 1] = u16(0)
2020-04-04 15:37:13 +02:00
C.GlobalUnlock(buf)
}
return buf
}
2019-12-07 13:51:00 +01:00
fn (cb mut Clipboard) set_text(text string) bool {
cb.foo = 0
buf := to_wide(text)
if !cb.get_clipboard_lock() {
2020-04-04 15:37:13 +02:00
C.GlobalFree(buf)
return false
} else {
2019-11-24 11:22:57 +01:00
// EmptyClipboard must be called to properly update clipboard ownership
2020-04-04 15:37:13 +02:00
C.EmptyClipboard()
if C.SetClipboardData(C.CF_UNICODETEXT, buf) == C.HANDLE(C.NULL) {
println("SetClipboardData: Failed.")
2020-04-04 15:37:13 +02:00
C.CloseClipboard()
C.GlobalFree(buf)
return false
}
}
2019-11-24 11:22:57 +01:00
// CloseClipboard appears to change the sequence number...
2020-04-04 15:37:13 +02:00
C.CloseClipboard()
return true
}
2019-12-07 13:51:00 +01:00
fn (cb mut Clipboard) get_text() string {
cb.foo = 0
if !cb.get_clipboard_lock() {
return ""
}
2020-04-04 15:37:13 +02:00
h_data := C.GetClipboardData(C.CF_UNICODETEXT)
if h_data == C.HANDLE(C.NULL) {
C.CloseClipboard()
return ""
}
2020-04-04 15:37:13 +02:00
str := string_from_wide(&u16(C.GlobalLock(h_data)))
C.GlobalUnlock(h_data)
return str
2019-11-18 11:10:31 +01:00
}
pub fn new_primary() &Clipboard {
panic('Primary clipboard is not supported on non-Linux systems.')
}