Как подключать (swizzle) методы в Swift?

Question

Есть ли простой способ перехвата (swizzle) методов в Swift?

Answer 1

Эта структура может помочь: https://github.com/623637646/SwiftHook

Как пользоваться

Например, это ваш класс

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 необходимы

Класс не должен наследовать от NSObject. Если класс написан 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. Полностью переопределить метод для указанного экземпляра. Вы можете вызвать оригинал с одинаковыми или разными параметрами. Даже не вызывайте оригинальный метод, если хотите.

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. Первым аргументом замыкания должно быть замыкание, которое имеет те же типы, что и метод. Остальные аргументы и возвращаемый тип должны быть такими же, как и у метода.

4. Выполните закрытие ловушки перед выполнением метода всех экземпляров класса.

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

5. Выполните закрытие перехватчика перед выполнением метода класса.

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