stefanrenne/RxSonosLib

//
//  AEXMLDocument+Create.swift
//  RxSonosLib
//
//  Created by Stefan Renne on 12/03/2018.
//  Copyright © 2018 Uberweb. All rights reserved.
//

import Foundation
import AEXML

extension AEXMLDocument {
    class func create(_ data: Data) throws -> AEXMLElement? {
        guard let cleanString = try String(data: data, encoding: .utf8)?.removeXmlNamespace() else { return nil }
        return try? AEXMLDocument(xml: cleanString)
    }
    
    class func create(_ string: String?) throws -> AEXMLElement? {
        guard let cleanString = try string?.removeXmlNamespace() else { return nil }
        let xml = try? AEXMLDocument(xml: "<root>" + cleanString + "</root>")
        return xml?["root"]
    }
}

extension AEXMLElement {
    
    func mapMetaItems() -> [[String: String]] {
        return self["DIDL-Lite"].children.map { (item) -> [String: String] in
            return item.children.reduce(into: [String: String](), { (dict, row) in
                dict[row.name] = row.string
                row.attributes.forEach({ (key, value) in
                    dict["\(row.name)\(key)"] = value
                })
            })
        }
    }
}

extension String {
    func mapMetaItem() throws -> [String: String]? {
        return try AEXMLDocument.create(self)?.mapMetaItems().first
    }
    func mapMetaItems() throws -> [[String: String]]? {
        return try AEXMLDocument.create(self)?.mapMetaItems()
    }
}

fileprivate extension String {
    
    func removeXmlNamespace() throws -> String {
        return try validateXml().replace(pattern: "<([a-zA-Z]+:)", with: "<").replace(pattern: "</([a-zA-Z]+:)", with: "</")
    }
    
    func replace(pattern: String, with template: String) -> String {
        guard let regex = try? NSRegularExpression(pattern: pattern, options: .caseInsensitive) else {
            return self
        }
        return regex.stringByReplacingMatches(in: self, options: [], range: NSRange(0..<count), withTemplate: template)
    }
    
}