SkeletonView/Sources/Debug/SkeletonDebug.swift

// Copyright © 2018 SkeletonView. All rights reserved.

import Foundation
import UIKit

enum SkeletonEnvironmentKey: String {
    case debugMode = "SKELETON_DEBUG"
}

extension Dictionary {
    subscript (_ key: SkeletonEnvironmentKey) -> Value? {
        // swiftlint:disable:next force_cast
        return self[key.rawValue as! Key]
    }
}

func printSkeletonHierarchy(in view: UIView) {
    skeletonLog(view.skeletonHierarchy())
}

func skeletonLog(_ message: String) {
    if ProcessInfo.processInfo.environment[.debugMode] != nil {
        print(message)
    }
}

extension UIView {
    public var skeletonDescription: String {
        var description = "<\(type(of: self)): \(Unmanaged.passUnretained(self).toOpaque())"
        let subSkeletons = subviewsSkeletonables
        if !subSkeletons.isEmpty {
            description += " | (\(subSkeletons.count)) subSkeletons"
        }
        if isSkeletonable {
            description += " | ☠️ "
        }
        return description + ">"
    }

    public func skeletonHierarchy(index: Int = 0) -> String {
        var description = index == 0 ? "\n ⬇⬇ ☠️ Root view hierarchy with Skeletons ⬇⬇ \n" : ""
        description += "\(index == 0 ? "\n" : 3.whitespaces) \(skeletonDescription) \n"
        subviewsToSkeleton.forEach {
            description += (index + 2).whitespaces
            description += $0.skeletonHierarchy(index: index + 1)
        }
        return description
    }
}