py-rs 0.1.1

#![allow(dead_code)]
use std::{collections::HashMap, sync::Arc};

#[cfg(feature = "serde-compat")]
use serde::Serialize;
use py_rs::PY;

#[derive(PY)]
#[py(export, export_to = "self_referential/")]
struct HasT {
    t: &'static T<'static>,
}

#[derive(PY)]
#[py(export, export_to = "self_referential/")]
struct T<'a> {
    t_box: Box<T<'a>>,
    self_box: Box<Self>,

    t_ref: &'a T<'a>,
    self_ref: &'a Self,

    t_arc: Arc<T<'a>>,
    self_arc: Arc<Self>,

    #[py(inline)]
    has_t: HasT,
}

#[test]
fn named() {
    assert_eq!(
        T::decl(),
        "type T = { \
            t_box: T, \
            self_box: T, \
            t_ref: T, \
            self_ref: T, \
            t_arc: T, \
            self_arc: T, \
            has_t: { t: T, }, \
         };"
    );
}

#[derive(PY)]
#[py(export, export_to = "self_referential/", rename = "E")]
enum ExternallyTagged {
    A(Box<ExternallyTagged>),
    B(&'static ExternallyTagged),
    C(Box<Self>),
    D(&'static Self),
    E(
        Box<ExternallyTagged>,
        Box<Self>,
        &'static ExternallyTagged,
        &'static Self,
    ),
    F {
        a: Box<Self>,
        b: &'static ExternallyTagged,
        c: HashMap<String, ExternallyTagged>,
        d: Option<Arc<ExternallyTagged>>,
        #[py(optional = nullable)]
        e: Option<Arc<ExternallyTagged>>,
        #[py(optional)]
        f: Option<Arc<ExternallyTagged>>,
    },

    G(
        Vec<ExternallyTagged>,
        [&'static ExternallyTagged; 1024],
        HashMap<String, ExternallyTagged>,
    ),
}

#[test]
fn enum_externally_tagged() {
    assert_eq!(
        ExternallyTagged::decl(),
       "type E = { \"A\": E } | \
                 { \"B\": E } | \
                 { \"C\": E } | \
                 { \"D\": E } | \
                 { \"E\": [E, E, E, E] } | \
                 { \"F\": { a: E, b: E, c: { [key in string]?: E }, d: E | null, e?: E | null, f?: E, } } | \
                 { \"G\": [Array<E>, Array<E>, { [key in string]?: E }] };"
    );
}

#[derive(PY)]
#[cfg_attr(feature = "serde-compat", derive(Serialize))]
#[py(rename = "I")]
#[cfg_attr(feature = "serde-compat", serde(tag = "tag"))]
#[cfg_attr(not(feature = "serde-compat"), py(tag = "tag"))]
enum InternallyTagged {
    A(Box<InternallyTagged>),
    B(&'static InternallyTagged),
    C(Box<Self>),
    D(&'static Self),
    E(Vec<Self>),
    F {
        a: Box<Self>,
        b: &'static InternallyTagged,
        c: HashMap<InternallyTagged, InternallyTagged>,
        d: Option<&'static InternallyTagged>,
        #[py(optional = nullable)]
        e: Option<&'static InternallyTagged>,
        #[py(optional)]
        f: Option<&'static InternallyTagged>,
    },
}

// NOTE: The generated type is actually not valid PY here, since the indirections rust enforces for recursive types
//       gets lost during the translation to TypeScript (e.g "Box<T>" => "T").
#[test]
fn enum_internally_tagged() {
    assert_eq!(
        InternallyTagged::decl(),
        "type I = { \"tag\": \"A\" } & I | \
                  { \"tag\": \"B\" } & I | \
                  { \"tag\": \"C\" } & I | \
                  { \"tag\": \"D\" } & I | \
                  { \"tag\": \"E\" } & Array<I> | \
                  { \"tag\": \"F\", a: I, b: I, c: { [key in I]?: I }, d: I | null, e?: I | null, f?: I, };"
    );
}

#[derive(PY)]
#[cfg_attr(feature = "serde-compat", derive(Serialize))]
#[py(export, export_to = "self_referential/", rename = "A")]
#[cfg_attr(feature = "serde-compat", serde(tag = "tag", content = "content"))]
#[cfg_attr(not(feature = "serde-compat"), py(tag = "tag", content = "content"))]
enum AdjacentlyTagged {
    A(Box<AdjacentlyTagged>),
    B(&'static AdjacentlyTagged),
    C(Box<Self>),
    D(&'static Self),
    E(Vec<Self>),
    F {
        a: Box<Self>,
        b: &'static AdjacentlyTagged,
        c: HashMap<String, AdjacentlyTagged>,
        d: Option<&'static AdjacentlyTagged>,
        #[py(optional = nullable)]
        e: Option<&'static AdjacentlyTagged>,
        #[py(optional)]
        f: Option<&'static AdjacentlyTagged>,
    },
    G(
        Vec<Self>,
        [&'static AdjacentlyTagged; 4],
        HashMap<String, AdjacentlyTagged>,
    ),
}

// NOTE: The generated type is actually not valid PY here, since the indirections rust enforces for recursive types
//       gets lost during the translation to TypeScript (e.g "Box<T>" => "T").
#[test]
fn enum_adjacently_tagged() {
    assert_eq!(
        AdjacentlyTagged::decl(),
        "type A = { \"tag\": \"A\", \"content\": A } | \
                  { \"tag\": \"B\", \"content\": A } | \
                  { \"tag\": \"C\", \"content\": A } | \
                  { \"tag\": \"D\", \"content\": A } | \
                  { \"tag\": \"E\", \"content\": Array<A> } | \
                  { \
                     \"tag\": \"F\", \
                     \"content\": { \
                         a: A, \
                         b: A, \
                         c: { [key in string]?: A }, \
                         d: A | null, \
                         e?: A | null, \
                         f?: A, \
                     } \
                  } | \
                  { \
                     \"tag\": \"G\", \
                     \"content\": [\
                        Array<A>, \
                        [A, A, A, A], \
                        { [key in string]?: A }\
                     ] \
                  };"
    );
}