Show More
@@ -0,0 +1,104 b'' | |||
|
1 | ||
|
2 | # obj-c boilerplate from appnope, used under BSD 2-clause | |
|
3 | ||
|
4 | import ctypes | |
|
5 | import ctypes.util | |
|
6 | ||
|
7 | objc = ctypes.cdll.LoadLibrary(ctypes.util.find_library('objc')) | |
|
8 | CoreFoundation = ctypes.cdll.LoadLibrary(ctypes.util.find_library('CoreFoundation')) | |
|
9 | # Cocoa = ctypes.cdll.LoadLibrary(ctypes.util.find_library('Cocoa')) | |
|
10 | ||
|
11 | void_p = ctypes.c_void_p | |
|
12 | ||
|
13 | objc.objc_getClass.restype = void_p | |
|
14 | objc.sel_registerName.restype = void_p | |
|
15 | objc.objc_msgSend.restype = void_p | |
|
16 | objc.objc_msgSend.argtypes = [void_p, void_p] | |
|
17 | ||
|
18 | msg = objc.objc_msgSend | |
|
19 | ||
|
20 | def _utf8(s): | |
|
21 | """ensure utf8 bytes""" | |
|
22 | if not isinstance(s, bytes): | |
|
23 | s = s.encode('utf8') | |
|
24 | return s | |
|
25 | ||
|
26 | def n(name): | |
|
27 | """create a selector name (for methods)""" | |
|
28 | return objc.sel_registerName(_utf8(name)) | |
|
29 | ||
|
30 | def C(classname): | |
|
31 | """get an ObjC Class by name""" | |
|
32 | return objc.objc_getClass(_utf8(classname)) | |
|
33 | ||
|
34 | # CoreFoundation calls we will use: | |
|
35 | CFFileDescriptorCreate = CoreFoundation.CFFileDescriptorCreate | |
|
36 | CFFileDescriptorCreate.restype = void_p | |
|
37 | CFFileDescriptorCreate.argtypes = [void_p, ctypes.c_int, ctypes.c_bool, void_p] | |
|
38 | ||
|
39 | CFFileDescriptorGetNativeDescriptor = CoreFoundation.CFFileDescriptorGetNativeDescriptor | |
|
40 | CFFileDescriptorGetNativeDescriptor.restype = ctypes.c_int | |
|
41 | CFFileDescriptorGetNativeDescriptor.argtypes = [void_p] | |
|
42 | ||
|
43 | CFFileDescriptorEnableCallBacks = CoreFoundation.CFFileDescriptorEnableCallBacks | |
|
44 | CFFileDescriptorEnableCallBacks.restype = None | |
|
45 | CFFileDescriptorEnableCallBacks.argtypes = [void_p, ctypes.c_ulong] | |
|
46 | ||
|
47 | CFFileDescriptorCreateRunLoopSource = CoreFoundation.CFFileDescriptorCreateRunLoopSource | |
|
48 | CFFileDescriptorCreateRunLoopSource.restype = void_p | |
|
49 | CFFileDescriptorCreateRunLoopSource.argtypes = [void_p, void_p, void_p] | |
|
50 | ||
|
51 | CFRunLoopGetCurrent = CoreFoundation.CFRunLoopGetCurrent | |
|
52 | CFRunLoopGetCurrent.restype = void_p | |
|
53 | ||
|
54 | CFRunLoopAddSource = CoreFoundation.CFRunLoopAddSource | |
|
55 | CFRunLoopAddSource.restype = None | |
|
56 | CFRunLoopAddSource.argtypes = [void_p, void_p, void_p] | |
|
57 | ||
|
58 | CFRelease = CoreFoundation.CFRelease | |
|
59 | CFRelease.restype = None | |
|
60 | CFRelease.argtypes = [void_p] | |
|
61 | ||
|
62 | CFFileDescriptorInvalidate = CoreFoundation.CFFileDescriptorInvalidate | |
|
63 | CFFileDescriptorInvalidate.restype = None | |
|
64 | CFFileDescriptorInvalidate.argtypes = [void_p] | |
|
65 | ||
|
66 | ||
|
67 | # From CFFileDescriptor.h | |
|
68 | kCFFileDescriptorReadCallBack = 1 | |
|
69 | kCFRunLoopCommonModes = void_p.in_dll(CoreFoundation, 'kCFRunLoopCommonModes') | |
|
70 | ||
|
71 | def _NSApp(): | |
|
72 | """Return the global NSApplication instance (NSApp)""" | |
|
73 | return msg(C('NSApplication'), n('sharedApplication')) | |
|
74 | ||
|
75 | def _input_callback(fdref, flags, info): | |
|
76 | """Callback to fire when there's input to be read""" | |
|
77 | CFFileDescriptorInvalidate(fdref) | |
|
78 | CFRelease(fdref) | |
|
79 | NSApp = _NSApp() | |
|
80 | msg(NSApp, n('stop:'), NSApp) | |
|
81 | ||
|
82 | _c_callback_func_type = ctypes.CFUNCTYPE(None, void_p, void_p, void_p) | |
|
83 | _c_callback = _c_callback_func_type(_input_callback) | |
|
84 | ||
|
85 | def _stop_on_read(fd): | |
|
86 | """Register callback to stop eventloop when there's data on fd""" | |
|
87 | fdref = CFFileDescriptorCreate(None, fd, False, _c_callback, None) | |
|
88 | CFFileDescriptorEnableCallBacks(fdref, kCFFileDescriptorReadCallBack) | |
|
89 | source = CFFileDescriptorCreateRunLoopSource(None, fdref, 0) | |
|
90 | loop = CFRunLoopGetCurrent() | |
|
91 | CFRunLoopAddSource(loop, source, kCFRunLoopCommonModes) | |
|
92 | CFRelease(source) | |
|
93 | ||
|
94 | def inputhook(context): | |
|
95 | """Inputhook for Cocoa (NSApp)""" | |
|
96 | NSApp = _NSApp() | |
|
97 | window_count = msg( | |
|
98 | msg(NSApp, n('windows')), | |
|
99 | n('count') | |
|
100 | ) | |
|
101 | if not window_count: | |
|
102 | return | |
|
103 | _stop_on_read(context.fileno()) | |
|
104 | msg(NSApp, n('run')) |
General Comments 0
You need to be logged in to leave comments.
Login now