From 6101a95d390556023aaccb6b5cb72a0f2b440ae0 Mon Sep 17 00:00:00 2001 From: Arman Arutiunov Date: Wed, 4 Nov 2020 15:24:59 +0000 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20Guard=20Method=20values=20in=20s?= =?UTF-8?q?wizzle()?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- SwiftLintXcode/SaveHook.swift | 7 +++++-- 1 file changed, 5 insertions(+), 2 deletions(-) diff --git a/SwiftLintXcode/SaveHook.swift b/SwiftLintXcode/SaveHook.swift index 6642f2a..e66ab92 100644 --- a/SwiftLintXcode/SaveHook.swift +++ b/SwiftLintXcode/SaveHook.swift @@ -24,8 +24,11 @@ final class SaveHook { if swizzled { return } swizzled = true - let fromMethod = class_getInstanceMethod(NSDocument.self, #selector(NSDocument.save(withDelegate:didSave:contextInfo:))) - let toMethod = class_getInstanceMethod(NSDocument.self, #selector(NSDocument.swiftLintXcodeSaveDocument(delegate:didSaveSelector:contextInfo:))) + guard let fromMethod = class_getInstanceMethod(NSDocument.self, #selector(NSDocument.save(withDelegate:didSave:contextInfo:))), + let toMethod = class_getInstanceMethod(NSDocument.self, #selector(NSDocument.swiftLintXcodeSaveDocument(delegate:didSaveSelector:contextInfo:))) else { + return + } + method_exchangeImplementations(fromMethod, toMethod) }