Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 3 additions & 3 deletions Documents/README.zh-Hans.md
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@ class TestObject { // No need to inherit from NSObject.

let obj = TestObject()

let token = try hookBefore(object: obj, selector: #selector(TestObject.testMethod)) {
let token = try ObjectHook(obj).hookBefore(#selector(TestObject.testMethod)) {
print("Before executing `testMethod`")
}

Expand All @@ -39,7 +39,7 @@ class TestObject {

let obj = TestObject()

let token = try hookAfter(object: obj, selector: #selector(TestObject.testMethod(_:)), closure: { obj, sel, parameter in
let token = try ObjectHook(obj).hookAfter(#selector(TestObject.testMethod(_:)), closure: { obj, sel, parameter in
print("After executing `testMethod` with parameter: \(parameter)")
} as @convention(block) ( // Using `@convention(block)` to declares a Swift closure as an Objective-C block
AnyObject, // `obj` Instance
Expand All @@ -65,7 +65,7 @@ class Math {

let math = Math()

try hookInstead(object: math, selector: #selector(Math.double(_:)), closure: { original, obj, sel, number in
try ObjectHook(math).hook(#selector(Math.double(_:)), closure: { original, obj, sel, number in
print("Before executing `double`")
let originalResult = original(obj, sel, number)
print("After executing `double`, got result \(originalResult)")
Expand Down
2 changes: 1 addition & 1 deletion Example/MacCocoapodsDynamicExample/ViewController.swift
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ class ViewController: NSViewController {
func hook() {
// hook
do {
try hookBefore(object: self, selector: #selector(NSViewController.viewDidAppear), closure: { viewController, _ in
try ObjectHook(self).hookBefore(#selector(NSViewController.viewDidAppear), closure: { viewController, _ in
print("ViewController did appear")
print("Title: \(viewController.title ?? "")")
})
Expand Down
2 changes: 1 addition & 1 deletion Example/MacCocoapodsStaticExample/ViewController.swift
Original file line number Diff line number Diff line change
Expand Up @@ -25,7 +25,7 @@ class ViewController: NSViewController {
func hook() {
// hook
do {
try hookBefore(object: self, selector: #selector(NSViewController.viewDidAppear), closure: { viewController, _ in
try ObjectHook(self).hookBefore(#selector(NSViewController.viewDidAppear), closure: { viewController, _ in
print("ViewController did appear")
print("Title: \(viewController.title ?? "")")
})
Expand Down
2 changes: 1 addition & 1 deletion Example/SPMExample/ContentView.swift
Original file line number Diff line number Diff line change
Expand Up @@ -30,7 +30,7 @@ struct ContentView: View {
let obj = MyObject()
// hook
do {
try hookBefore(object: obj, selector: #selector(MyObject.test), closure: { viewController, _ in
try ObjectHook(obj).hookBefore(#selector(MyObject.test), closure: { viewController, _ in
print("hooked")
})
} catch {
Expand Down
2 changes: 1 addition & 1 deletion Example/iOSCocoapodsDynamicExample/ViewController.swift
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ class ViewController: UIViewController {
func hook() {
// hook
do {
try hookBefore(object: self, selector: #selector(UIViewController.viewDidAppear(_:)), closure: { viewController, _ in
try ObjectHook(self).hookBefore(#selector(UIViewController.viewDidAppear(_:)), closure: { viewController, _ in
print("ViewController did appear")
print("Title: \(viewController.title ?? "")")
})
Expand Down
2 changes: 1 addition & 1 deletion Example/iOSCocoapodsStaticExample/ViewController.swift
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ class ViewController: UIViewController {
func hook() {
// hook
do {
try hookBefore(object: self, selector: #selector(UIViewController.viewDidAppear(_:)), closure: { viewController, _ in
try ObjectHook(self).hookBefore(#selector(UIViewController.viewDidAppear(_:)), closure: { viewController, _ in
print("ViewController did appear")
print("Title: \(viewController.title ?? "")")
})
Expand Down
6 changes: 3 additions & 3 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ class TestObject { // No need to inherit from NSObject.

let obj = TestObject()

let token = try hookBefore(object: obj, selector: #selector(TestObject.testMethod)) {
let token = try ObjectHook(obj).hookBefore(#selector(TestObject.testMethod)) {
print("Before executing `testMethod`")
}

Expand All @@ -40,7 +40,7 @@ class TestObject {

let obj = TestObject()

let token = try hookAfter(object: obj, selector: #selector(TestObject.testMethod(_:)), closure: { obj, sel, parameter in
let token = try ObjectHook(obj).hookAfter(#selector(TestObject.testMethod(_:)), closure: { obj, sel, parameter in
print("After executing `testMethod` with parameter: \(parameter)")
} as @convention(block) ( // Using `@convention(block)` to declares a Swift closure as an Objective-C block
AnyObject, // `obj` Instance
Expand All @@ -66,7 +66,7 @@ class Math {

let math = Math()

try hookInstead(object: math, selector: #selector(Math.double(_:)), closure: { original, obj, sel, number in
try ObjectHook(math).hook(#selector(Math.double(_:)), closure: { original, obj, sel, number in
print("Before executing `double`")
let originalResult = original(obj, sel, number)
print("After executing `double`, got result \(originalResult)")
Expand Down
112 changes: 112 additions & 0 deletions SwiftHook/Classes/Hook/NSObject+SelectorName.swift
Original file line number Diff line number Diff line change
@@ -0,0 +1,112 @@
//
// NSObject+SelectorName.swift
//
//
// Created by Florian Zand on 06.05.25.
//

import Foundation

extension PartialKeyPath {
func getterName() throws -> String where Root: AnyObject {
guard let getterName = _kvcKeyPathString else {
throw SwiftHookError.nonObjcProperty
}
return getterName
}

func setterName() throws -> String where Root: AnyObject {
guard let setterName = NSObject.setterName(for: try getterName(), _class: Root.self) else {
throw SwiftHookError.nonObjcProperty
}
return setterName
}

func getterName<T>() throws -> String where Root == T.Type, T: AnyObject {
guard let getterName = _kvcKeyPathString else {
throw SwiftHookError.nonObjcProperty
}
return getterName
}

func setterName<T>() throws -> String where Root == T.Type, T: AnyObject {
guard let setterName = NSObject.setterName(for: try getterName(), _class: T.self) else {
throw SwiftHookError.nonObjcProperty
}
return setterName
}
}

fileprivate extension NSObject {
static func setterName(for getterName: String, _class: AnyClass) -> String? {
var names: [String] = []
if getterName.hasPrefix("is") {
names.append("set\(getterName.dropFirst(2).uppercasedFirst()):")
} else if getterName.hasPrefix("get") {
names.append("set\(getterName.dropFirst(3).uppercasedFirst()):")
}
names.append("set\(getterName.uppercasedFirst()):")
for name in names {
if class_respondsToSelector(_class, NSSelectorFromString(name)) {
return name
}
}

let getterSelector = Selector(getterName)
var currentClass: AnyClass? = _class
while let c = currentClass {
var propertyCount: UInt32 = 0
guard let properties = class_copyPropertyList(c, &propertyCount) else {
currentClass = class_getSuperclass(c)
continue
}
defer { free(properties) }

for i in 0..<propertyCount {
let property = properties[Int(i)]
let nameCStr = property_getName(property)
let propName = String(cString: nameCStr)

let getterSel: Selector
if let getterAttr = property.attribute(for: "G") {
getterSel = Selector(getterAttr)
} else {
getterSel = Selector(propName)
}

if getterSel == getterSelector {
if let setterAttr = property.attribute(for: "S") {
return setterAttr
} else {
return "set\(propName.uppercasedFirst()):"
}
}
}
currentClass = class_getSuperclass(c)
}
return nil
}
}

fileprivate extension objc_property_t {
func attribute(for key: String) -> String? {
var count: UInt32 = 0
guard let attrs = property_copyAttributeList(self, &count) else { return nil }
defer { free(attrs) }
for i in 0..<count {
let attr = attrs[Int(i)]
if String(cString: attr.name) == key {
return String(cString: attr.value)
}
}
return nil
}
}


fileprivate extension StringProtocol {
func uppercasedFirst() -> String {
if isEmpty { return String(self) }
return prefix(1).uppercased() + dropFirst()
}
}
Loading