如何在 Swift 中挂钩（调配）方法？

Question

是否有任何简单的方法可以在 Swift 中挂钩（调配）方法？

Answer 1

这个框架可以提供帮助： https://github.com/623637646/SwiftHook

如何使用

例如，这是您的 class

class MyObject {
    @objc dynamic func noArgsNoReturnFunc() {
    }
    @objc dynamic func sumFunc(a: Int, b: Int) -> Int {
        return a + b
    }
    @objc dynamic class func classMethodNoArgsNoReturnFunc() {
    }
}

方法的@objc和dynamic是必须的

class 不必从 NSObject 继承。 如果 class 是由 Objective-C 写的，直接钩住就行了

1. 在执行指定实例的方法之前执行钩子关闭。

let object = MyObject()
let token = try? hookBefore(object: object, selector: #selector(MyObject.noArgsNoReturnFunc)) {
    // run your code
    print("hooked!")
}
object.noArgsNoReturnFunc()
token?.cancelHook() // cancel the hook

2. 执行指定实例的方法后执行钩子关闭。 并获取参数。

let object = MyObject()
let token = try? hookAfter(object: object, selector: #selector(MyObject.sumFunc(a:b:)), closure: { a, b in
    // get the arguments of the function
    print("arg1 is \(a)") // arg1 is 3
    print("arg2 is \(b)") // arg2 is 4
    } as @convention(block) (Int, Int) -> Void)
_ = object.sumFunc(a: 3, b: 4)
token?.cancelHook() // cancel the hook

关键字@convention(block)是必要的

用于钩在before和after 。 闭包的参数必须为空或与方法相同。 返回类型必须为void

3. 完全覆盖指定实例的方法。 您可以使用相同的参数或不同的参数调用 original。 如果您愿意，甚至不要调用原始方法。

let object = MyObject()
let token = try? hookInstead(object: object, selector: #selector(MyObject.sumFunc(a:b:)), closure: { original, a, b in
    // get the arguments of the function
    print("arg1 is \(a)") // arg1 is 3
    print("arg2 is \(b)") // arg2 is 4

    // run original function
    let result = original(a, b) // Or change the parameters: let result = original(-1, -2)
    print("original result is \(result)") // result = 7
    return 9
    } as @convention(block) ((Int, Int) -> Int, Int, Int) -> Int)
let result = object.sumFunc(a: 3, b: 4) // result
print("hooked result is \(result)") // result = 9
token?.cancelHook() // cancel the hook

换成instead 。 闭包的第一个参数必须是与方法具有相同类型的闭包。 rest 参数和返回类型必须与方法相同。

4. 在执行 class 的所有实例的方法之前执行钩子关闭。

let token = try? hookBefore(targetClass: MyObject.self, selector: #selector(MyObject.noArgsNoReturnFunc)) {
    // run your code
    print("hooked!")
}
MyObject().noArgsNoReturnFunc()
token?.cancelHook() // cancel the hook

5. 在执行 class 方法之前执行钩子关闭。

let token = try? hookClassMethodBefore(targetClass: MyObject.self, selector: #selector(MyObject.classMethodNoArgsNoReturnFunc)) {
    // run your code
    print("hooked!")
}
MyObject.classMethodNoArgsNoReturnFunc()
token?.cancelHook() // cancel the hook