import Foundation

import XCTest

var deviceLanguage = ""

var locale = ""

@MainActor

func setupSnapshot(_ app: XCUIApplication, waitForAnimations: Bool = true) {

Snapshot.setupSnapshot(app, waitForAnimations: waitForAnimations)

}

@MainActor

func snapshot(_ name: String, waitForLoadingIndicator: Bool) {

if waitForLoadingIndicator {

Snapshot.snapshot(name)

/// - Parameters:

/// - name: The name of the snapshot

/// - timeout: Amount of seconds to wait until the network loading indicator disappears. Pass `0` if you don't want to wait.

@MainActor

func snapshot(_ name: String, timeWaitingForIdle timeout: TimeInterval = 20) {

Snapshot.snapshot(name, timeWaitingForIdle: timeout)

}

}

@objcMembers

@MainActor

open class Snapshot: NSObject {

static var app: XCUIApplication?

static var waitForAnimations = true

static var cacheDirectory: URL?

static var screenshotsDirectory: URL? {

return cacheDirectory?.appendingPathComponent("screenshots", isDirectory: true)

}

static var deviceLanguage = ""

static var currentLocale = ""

open class func setupSnapshot(_ app: XCUIApplication, waitForAnimations: Bool = true) {

do {

let trimCharacterSet = CharacterSet.whitespacesAndNewlines

locale = try String(contentsOf: path, encoding: .utf8).trimmingCharacters(in: trimCharacterSet)

currentLocale = try String(contentsOf: path, encoding: .utf8).trimmingCharacters(in: trimCharacterSet)

} catch {

NSLog("Couldn't detect/set locale...")

}

if locale.isEmpty && !deviceLanguage.isEmpty {

locale = Locale(identifier: deviceLanguage).identifier

if currentLocale.isEmpty && !deviceLanguage.isEmpty {

currentLocale = Locale(identifier: deviceLanguage).identifier

}

if !locale.isEmpty {

app.launchArguments += ["-AppleLocale", "\"\(locale)\""]

if !currentLocale.isEmpty {

app.launchArguments += ["-AppleLocale", "\"\(currentLocale)\""]

}

}

}

let screenshot = XCUIScreen.main.screenshot()

#if os(iOS)

#if os(iOS) && !targetEnvironment(macCatalyst)

let image = XCUIDevice.shared.orientation.isLandscape ? fixLandscapeOrientation(image: screenshot.image) : screenshot.image

#else

let image = screenshot.image

let path = screenshotsDir.appendingPathComponent("\(simulator)-\(name).png")

#if swift(<5.0)

UIImagePNGRepresentation(image)?.write(to: path, options: .atomic)

try UIImagePNGRepresentation(image)?.write(to: path, options: .atomic)

#else

try image.pngData()?.write(to: path, options: .atomic)

#endif

return self.containing(isNetworkLoadingIndicator)

}

@MainActor

var deviceStatusBars: XCUIElementQuery {

guard let app = Snapshot.app else {

fatalError("XCUIApplication is not set. Please call setupSnapshot(app) before snapshot().")

// Please don't remove the lines below

// They are used to detect outdated configuration files

// SnapshotHelperVersion [1.27]

// SnapshotHelperVersion [1.30]