#[macro_use]
extern crate serde_derive;

use serde_plain2::derive_fromstr_from_deserialize;

use std::str::FromStr;

#[derive(Deserialize, PartialEq, Eq, Debug)]
#[serde(rename_all = "snake_case")]
pub enum Test {
    FooBarBaz,
    BlahBlah,
}

#[derive(Deserialize, PartialEq, Eq, Debug)]
#[serde(rename_all = "snake_case")]
pub enum FooType {
    FooA,
    FooB,
}
#[derive(Deserialize, PartialEq, Eq, Debug)]
#[serde(rename_all = "snake_case")]
pub enum BarType {
    BarA,
    BarB,
}

#[derive(Deserialize, PartialEq, Eq, Debug)]
#[serde(untagged)]
pub enum TestUntagged {
    Foo(FooType),
    Bar(BarType),
    Other(String),
}

derive_fromstr_from_deserialize!(TestUntagged);

#[derive(Deserialize, PartialEq, Eq, Debug)]
pub struct NewInt(pub i32);

impl FromStr for Test {
    type Err = serde_plain2::Error;
    fn from_str(value: &str) -> Result<Test, serde_plain2::Error> {
        serde_plain2::from_str(value)
    }
}

#[test]
fn test_basics() {
    assert_eq!(serde_plain2::from_str::<&str>("aha").unwrap(), "aha");
    assert_eq!(serde_plain2::from_str::<i32>("42").unwrap(), 42);
    assert!(serde_plain2::from_str::<bool>("true").unwrap());
    assert!(!serde_plain2::from_str::<bool>("false").unwrap());
    assert_eq!(serde_plain2::from_str::<Option<()>>("").unwrap(), None);
    assert_eq!(
        serde_plain2::from_str::<Option<String>>("42").unwrap(),
        Some("42".into())
    );
    assert_eq!(
        serde_plain2::from_str::<Option<&str>>("42").unwrap(),
        Some("42")
    );
    assert_eq!(
        serde_plain2::from_str::<Test>("blah_blah").unwrap(),
        Test::BlahBlah
    );
    assert_eq!(serde_plain2::from_str::<NewInt>("42").unwrap(), NewInt(42));
    serde_plain2::from_str::<serde::de::IgnoredAny>("blah_blah")
        .expect("doesn't fail on #[serde(skip)] fields");
}

#[test]
fn test_untagged_enum() {
    let pairs = [
        ("foo_a", TestUntagged::Foo(FooType::FooA)),
        ("foo_b", TestUntagged::Foo(FooType::FooB)),
        ("bar_a", TestUntagged::Bar(BarType::BarA)),
        ("bar_b", TestUntagged::Bar(BarType::BarB)),
        ("other", TestUntagged::Other("other".to_string())),
    ];

    for (string, expected) in pairs.iter() {
        assert_eq!(string.parse::<TestUntagged>().unwrap(), *expected);
    }
}

#[test]
fn test_from_str() {
    assert_eq!("foo_bar_baz".parse::<Test>().unwrap(), Test::FooBarBaz);
}