Swift to xcpretty screenshot saving

XCTestCase+ScreenshotAssert.swift

import XCTest

extension XCTestCase {
    func assertFailScreenshot(_ closure: @autoclosure () -> Bool, message: String? = nil, file: String = #file, line: UInt = #line, function: String = #function) {
        if !closure() {
            takeScreenshot(description: message, file: file, line: line, function: function)
            if let message = message {
                XCTFail(message)
            } else {
                XCTFail()
            }
        }
    }
}

XCTestCase+TakeScreenshot.swift

import XCTest

extension XCTestCase {

    func takeScreenshot(description: String? = nil, file: String = #file, line: UInt = #line, function: String = #function) {
        guard let fileUrl = URL(string: file) else {
            return
        }

        let projectPath = fileUrl.deletingLastPathComponent().deletingLastPathComponent()
        let filename = screenshotFilename(description: description, fileUrl: fileUrl, line: line, function: function)

        let screenshot = XCUIScreen.main.screenshot()
        let saveUrl = projectPath
            .appendingPathComponent("build")
            .appendingPathComponent("reports")
            .appendingPathComponent(filename)
        let fileURLWithPath = URL(fileURLWithPath: saveUrl.path)
        try! FileManager.default.createDirectory(at: fileURLWithPath.deletingLastPathComponent(), withIntermediateDirectories: true, attributes: nil)
        try! screenshot.pngRepresentation.write(to: fileURLWithPath)
    }

    private func screenshotFilename(description: String?, fileUrl: URL, line: UInt, function: String) -> String {
        let testCase = fileUrl.deletingPathExtension().lastPathComponent
        let trimmedFunction = function.trimmingCharacters(in: .punctuationCharacters)

        let bundleName = Bundle.main.infoDictionary?["CFBundleName"] as? String
        let prefix = bundleName?.split(separator: "-").first ?? ""

        var result = "\(prefix).\(testCase)_\(trimmedFunction)_\(line)"
        if let description = description {
            result += " \(description)"
        }
        result += ".png"
        return result
    }
}