use oasgen::OaSchema;
use serde::{Deserialize, Serialize};
use oasgen::generate_openapi;

#[derive(OaSchema, Serialize, Deserialize)]
pub enum Duration {
    Days(u32),
    Months(u32),
}

#[derive(OaSchema, Serialize, Deserialize)]
pub enum ExternallyTagged {
    A(i32),
    B,
    C { test: i32 },
    D(Duration),
    E,
}

#[derive(OaSchema, Serialize, Deserialize)]
#[serde(tag = "type")]
pub enum InternallyTagged {
    // A(i32), internally tagged does not support tuple variants that do not contain a struct
    B,
    C { test: i32 },
    D(Duration),
    E,
}

#[derive(OaSchema, Serialize, Deserialize)]
#[serde(tag = "t", content = "c")]
pub enum AdjacentlyTagged {
    A(i32),
    B,
    C { test: i32 },
    D(Duration),
    E,
}

#[derive(OaSchema, Serialize, Deserialize)]
#[serde(untagged)]
pub enum Untagged {
    A(i32),
    B,
    C { test: i32 },
    D(Duration),
    E,
}

#[derive(OaSchema, Serialize, Deserialize)]
pub struct Foo {
    externally_tagged: ExternallyTagged,
    #[oasgen(inline)]
    externally_tagged_inline: ExternallyTagged,

    internally_tagged: InternallyTagged,
    #[oasgen(inline)]
    internally_tagged_inline: InternallyTagged,

    adjacently_tagged: AdjacentlyTagged,
    #[oasgen(inline)]
    adjacently_tagged_inline: AdjacentlyTagged,

    untagged: Untagged,
    #[oasgen(inline)]
    untagged_inline: Untagged,
}

fn main() {
    use pretty_assertions::assert_eq;
    let schema = generate_openapi();
    let spec = serde_yaml::to_string(&schema).unwrap();
    assert_eq!(spec, include_str!("06-complex-enum.yaml"));
}