Dactyloidae/mobile/ios/Client/Frontend/Reader/ReaderModeUtils.swift

/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

import Foundation

struct ReaderModeUtils {

    static let DomainPrefixesToSimplify = ["www.", "mobile.", "m.", "blog."]

    static func simplifyDomain(_ domain: String) -> String {
        for prefix in DomainPrefixesToSimplify {
            if domain.hasPrefix(prefix) {
                return domain.substring(from: domain.characters.index(domain.startIndex, offsetBy: prefix.characters.count))
            }
        }
        return domain
    }

    static func generateReaderContent(_ readabilityResult: ReadabilityResult, initialStyle: ReaderModeStyle) -> String? {
        if let stylePath = Bundle.main.path(forResource: "Reader", ofType: "css") {
            do {
                let css = try NSString(contentsOfFile: stylePath, encoding: String.Encoding.utf8.rawValue)
                if let tmplPath = Bundle.main.path(forResource: "Reader", ofType: "html") {
                    do {
                        let tmpl = try NSMutableString(contentsOfFile: tmplPath, encoding: String.Encoding.utf8.rawValue)
                        tmpl.replaceOccurrences(of: "%READER-CSS%", with: css as String,
                            options: NSString.CompareOptions(), range: NSRange(location: 0, length: tmpl.length))

                        tmpl.replaceOccurrences(of: "%READER-STYLE%", with: initialStyle.encode(),
                            options: NSString.CompareOptions(), range: NSRange(location: 0, length: tmpl.length))

                        tmpl.replaceOccurrences(of: "%READER-DOMAIN%", with: simplifyDomain(readabilityResult.domain),
                            options: NSString.CompareOptions(), range: NSRange(location: 0, length: tmpl.length))

                        tmpl.replaceOccurrences(of: "%READER-URL%", with: readabilityResult.url,
                            options: NSString.CompareOptions(), range: NSRange(location: 0, length: tmpl.length))

                        tmpl.replaceOccurrences(of: "%READER-TITLE%", with: readabilityResult.title,
                            options: NSString.CompareOptions(), range: NSRange(location: 0, length: tmpl.length))

                        tmpl.replaceOccurrences(of: "%READER-CREDITS%", with: readabilityResult.credits,
                            options: NSString.CompareOptions(), range: NSRange(location: 0, length: tmpl.length))

                        tmpl.replaceOccurrences(of: "%READER-CONTENT%", with: readabilityResult.content,
                            options: NSString.CompareOptions(), range: NSRange(location: 0, length: tmpl.length))

                        return tmpl as String
                    } catch _ {
                    }
                }
            } catch _ {
            }
        }
        return nil
    }
}