diff --git a/Source/Armchair.swift b/Source/Armchair.swift index d1d5e91..a3a22ac 100644 --- a/Source/Armchair.swift +++ b/Source/Armchair.swift @@ -572,6 +572,9 @@ public typealias ArmchairShouldIncrementClosure = () -> Bool public func onDidDisplayAlert(_ didDisplayAlertClosure: ArmchairClosure?) { Manager.defaultManager.didDisplayAlertClosure = didDisplayAlertClosure } +public func customAlertClosure(_ customAlertClosure: ArmchairClosureCustomAlert?) { + Manager.defaultManager.customAlertClosure = customAlertClosure +} public func onDidDeclineToRate(_ didDeclineToRateClosure: ArmchairClosure?) { Manager.defaultManager.didDeclineToRateClosure = didDeclineToRateClosure }