// Copyright (c) 2018 Emmanuel Gil Peyrot <linkmauve@linkmauve.fr>
//
// 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/.

use helpers::{Base64, TrimmedPlainText};

generate_attribute!(Mechanism, "mechanism", {
    Plain => "PLAIN",
    ScramSha1 => "SCRAM-SHA-1",
    ScramSha1Plus => "SCRAM-SHA-1-PLUS",
    ScramSha256 => "SCRAM-SHA-256",
    ScramSha256Plus => "SCRAM-SHA-256-PLUS",
    Anonymous => "ANONYMOUS",
});

generate_element!(Auth, "auth", SASL,
    attributes: [
        mechanism: Mechanism = "mechanism" => required
    ],
    text: (
        data: Base64<Vec<u8>>
    )
);

generate_element!(Challenge, "challenge", SASL,
    text: (
        data: Base64<Vec<u8>>
    )
);

generate_element!(Response, "response", SASL,
    text: (
        data: Base64<Vec<u8>>
    )
);

generate_element!(Success, "success", SASL,
    text: (
        data: Base64<Vec<u8>>
    )
);

generate_element!(Failure, "failure", SASL,
    text: (
        data: TrimmedPlainText<String>
    )
);

#[cfg(test)]
mod tests {
    use super::*;
    use try_from::TryFrom;
    use minidom::Element;

    #[test]
    fn test_simple() {
        let elem: Element = "<auth xmlns='urn:ietf:params:xml:ns:xmpp-sasl' mechanism='PLAIN'/>".parse().unwrap();
        let auth = Auth::try_from(elem).unwrap();
        assert_eq!(auth.mechanism, Mechanism::Plain);
        assert!(auth.data.is_empty());
    }
}