//===----------------------------------------------------------------------===//
//
// This source file is part of the Hummingbird server framework project
//
// Copyright (c) 2021-2021 the Hummingbird authors
// Licensed under Apache License v2.0
//
// See LICENSE.txt for license information
// See hummingbird/CONTRIBUTORS.txt for the list of Hummingbird authors
//
// SPDX-License-Identifier: Apache-2.0
//
//===----------------------------------------------------------------------===//

/// Protocol for content types
public protocol HBMustacheContentType {
    /// escape text for this content type eg for HTML replace "<" with "&lt;"
    func escapeText(_ text: String) -> String
}

/// Text content type where no character is escaped
struct HBTextContentType: HBMustacheContentType {
    func escapeText(_ text: String) -> String {
        return text
    }
}

/// HTML content where text is escaped for HTML output
struct HBHTMLContentType: HBMustacheContentType {
    func escapeText(_ text: String) -> String {
        return text.htmlEscape()
    }
}

/// Map of strings to content types.
///
/// The string is read from the "CONTENT_TYPE" pragma `{{% CONTENT_TYPE: type}}`. Replace type with
/// the content type required. The default available types are `TEXT` and `HTML`. You can register your own
/// with `HBMustacheContentTypes.register`.
public enum HBMustacheContentTypes {
    static func get(_ name: String) -> HBMustacheContentType? {
        return self.types[name]
    }

    /// Register new content type
    /// - Parameters:
    ///   - contentType: Content type
    ///   - name: String to identify it
    public static func register(_ contentType: HBMustacheContentType, named name: String) {
        self.types[name] = contentType
    }

    static var types: [String: HBMustacheContentType] = [
        "HTML": HBHTMLContentType(),
        "TEXT": HBTextContentType(),
    ]
}