覆盖窗口关闭行为

Posted 2023-02-27

【中文标题】覆盖窗口关闭行为

【英文标题】：Override window close behavior

【发布时间】：2011-09-12 01:50:39

【问题描述】：

我想捕捉所有关闭某些特定现有 Cocoa 窗口的尝试并添加一些自己的处理程序（这可能确实会关闭它或做一些不同的事情）。

我有不同的解决方案来做到这一点。一个是：

我想在运行时用自己的关闭小部件替换现有 Cocoa 窗口的窗口关闭按钮，我可以在其中添加一些自己的代码。

现在，我有这个代码：

import objc
_NSThemeCloseWidget = objc.lookUpClass("_NSThemeCloseWidget")

def find_close_widget(window):
    contentView = window.contentView()
    grayFrame = contentView.superview()
    for i in range(len(grayFrame.subviews())):
        v = grayFrame.subviews()[i]
        if isinstance(v, _NSThemeCloseWidget):
            return v, i, grayFrame

class CustomCloseWidget(_NSThemeCloseWidget):
    pass

def replace_close_widget(window, clazz=CustomCloseWidget):
    v, i, grayFrame = find_close_widget(window)
    newv = clazz.alloc().init()
    grayFrame.subviews()[i] = newv

但是，这似乎不太正确。 （它崩溃了。）

【问题讨论】：

是否有理由不使用 NSWindowDelegate 协议，例如windowShouldClose 或 windowWillClose 如果你想修改关闭行为？

我同意 - 这就是代表的目的。所以你可以加入决策管道.​​.....

我无法在需要的地方设置新的委托。但是，我可以连接到 windowShouldClose 左右。但是我怎样才能真正改变那里的行为呢？ （它不仅在某些情况下不应该关闭它，有时还应该执行一些操作。）

@Albert：你是说你已经有一个窗口代表了吗？如果是这样，那么您实际上已经完成了：您现在需要做的就是让该对象与任何需要控制窗口是否应该关闭/响应您提到的操作的对象对话。

@Peter：有一个委托，但它来自其他我无法更改的代码。

【参考方案1】：

关闭小部件并不是关闭窗口的唯一方法。有一个公共 API 可以获取小部件，因此您无需遍历框架视图的子视图，但无论如何这是错误的路径。

正确的方法是让一个对象成为窗口的委托，并在那里interfere with the window's closure。理想情况下，您应该在创建窗口和订购窗口之间设置窗口的委托。

【讨论】：

在我的情况下，我无法更改窗口的委托。我无法保证是否有其他代码我无法控制哪些代码会将其更改回来，以及其他代码会因此而中断。

【参考方案2】：

我现在要走另一条路。这部分与 Chrome 相关，但可以很容易地在其他地方采用。我想尽早捕获一些关闭窗口的操作，以避免任何其他清理导致窗口处于奇怪状态。

def check_close_callback(obj):
    # check ...
    return True # or:
    return False

import objc
BrowserWindowController = objc.lookUpClass("BrowserWindowController")

# copied from objc.signature to avoid warning
def my_signature(signature, **kw):
    from objc._objc import selector
    kw['signature'] = signature
    def makeSignature(func):
        return selector(func, **kw)
    return makeSignature

windowWillCloseSig = "c12@0:4@8" # BrowserWindowController.windowWillClose_.signature
commandDispatchSig = "v12@0:4@8"
class BrowserWindowController(objc.Category(BrowserWindowController)):
    @my_signature(windowWillCloseSig)
    def myWindowShouldClose_(self, sender):
        print "myWindowShouldClose", self, sender
        if not check_close_callback(self): return objc.NO
        return self.myWindowShouldClose_(sender) # this is no recursion when we exchanged the methods

    @my_signature(commandDispatchSig)
    def myCommandDispatch_(self, cmd):
        try: print "myCommandDispatch_", self, cmd
        except: pass # like <type 'exceptions.UnicodeEncodeError'>: 'ascii' codec can't encode character u'\u2026' in position 37: ordinal not in range(128)
        if cmd.tag() == 34015: # IDC_CLOSE_TAB
            if not check_close_callback(self): return           
        self.myCommandDispatch_(cmd)

from ctypes import *
capi = pythonapi

# id objc_getClass(const char *name)
capi.objc_getClass.restype = c_void_p
capi.objc_getClass.argtypes = [c_char_p]

# SEL sel_registerName(const char *str)
capi.sel_registerName.restype = c_void_p
capi.sel_registerName.argtypes = [c_char_p]

def capi_get_selector(name):
    return c_void_p(capi.sel_registerName(name))

# Method class_getInstanceMethod(Class aClass, SEL aSelector)
# Will also search superclass for implementations.
capi.class_getInstanceMethod.restype = c_void_p
capi.class_getInstanceMethod.argtypes = [c_void_p, c_void_p]

# void method_exchangeImplementations(Method m1, Method m2)
capi.method_exchangeImplementations.restype = None
capi.method_exchangeImplementations.argtypes = [c_void_p, c_void_p]

def method_exchange(className, origSelName, newSelName):
    clazz = capi.objc_getClass(className)
    origMethod = capi.class_getInstanceMethod(clazz, capi_get_selector(origSelName))
    newMethod = capi.class_getInstanceMethod(clazz, capi_get_selector(newSelName))
    capi.method_exchangeImplementations(origMethod, newMethod)

def hook_into_windowShouldClose():
    method_exchange("BrowserWindowController", "windowShouldClose:", "myWindowShouldClose:")

def hook_into_commandDispatch():
    method_exchange("BrowserWindowController", "commandDispatch:", "myCommandDispatch:")

此代码来自here 和here。